135 lines
3.8 KiB
Python
Executable File
135 lines
3.8 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
"""
|
|
Dubai Metaverse - OpenStreetMap Data Import Script
|
|
Imports OpenStreetMap data for Dubai Marina and converts to Unreal-compatible format
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import json
|
|
import argparse
|
|
from typing import List, Dict, Tuple
|
|
|
|
try:
|
|
import overpy
|
|
import geojson
|
|
except ImportError:
|
|
print("Error: Required packages not installed.")
|
|
print("Install with: pip install overpy geojson")
|
|
sys.exit(1)
|
|
|
|
|
|
def get_marina_buildings(api: overpy.Overpass) -> List[Dict]:
|
|
"""
|
|
Query OpenStreetMap for buildings in Dubai Marina area.
|
|
|
|
Note: Coordinates should be adjusted based on actual Dubai Marina location.
|
|
"""
|
|
# Dubai Marina approximate bounding box
|
|
# These coordinates should be verified and adjusted
|
|
query = """
|
|
[out:json][timeout:25];
|
|
(
|
|
way["building"](25.0750,55.1350,25.0850,55.1450);
|
|
);
|
|
out geom;
|
|
"""
|
|
|
|
try:
|
|
result = api.query(query)
|
|
buildings = []
|
|
|
|
for way in result.ways:
|
|
building = {
|
|
'id': way.id,
|
|
'nodes': [(node.lat, node.lon) for node in way.nodes],
|
|
'tags': way.tags
|
|
}
|
|
buildings.append(building)
|
|
|
|
return buildings
|
|
except Exception as e:
|
|
print(f"Error querying OpenStreetMap: {e}")
|
|
return []
|
|
|
|
|
|
def buildings_to_geojson(buildings: List[Dict], output_file: str):
|
|
"""Convert buildings to GeoJSON format."""
|
|
features = []
|
|
|
|
for building in buildings:
|
|
# Create polygon from nodes
|
|
coordinates = [[node[1], node[0]] for node in building['nodes']] # GeoJSON uses [lon, lat]
|
|
coordinates.append(coordinates[0]) # Close polygon
|
|
|
|
feature = {
|
|
'type': 'Feature',
|
|
'geometry': {
|
|
'type': 'Polygon',
|
|
'coordinates': [coordinates]
|
|
},
|
|
'properties': {
|
|
'id': building['id'],
|
|
'tags': building['tags']
|
|
}
|
|
}
|
|
features.append(feature)
|
|
|
|
geojson_data = {
|
|
'type': 'FeatureCollection',
|
|
'features': features
|
|
}
|
|
|
|
with open(output_file, 'w') as f:
|
|
json.dump(geojson_data, f, indent=2)
|
|
|
|
print(f"✓ Exported {len(features)} buildings to {output_file}")
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description='Import OpenStreetMap data for Dubai Marina')
|
|
parser.add_argument('--output', '-o', default='data/processed/dubai_marina_buildings.geojson',
|
|
help='Output GeoJSON file path')
|
|
parser.add_argument('--area', help='Custom bounding box (lat1,lon1,lat2,lon2)')
|
|
|
|
args = parser.parse_args()
|
|
|
|
# Create output directory if it doesn't exist
|
|
os.makedirs(os.path.dirname(args.output), exist_ok=True)
|
|
|
|
print("==========================================")
|
|
print("Dubai Metaverse - OSM Data Import")
|
|
print("==========================================")
|
|
print("")
|
|
print("Connecting to OpenStreetMap API...")
|
|
|
|
api = overpy.Overpass()
|
|
|
|
print("Querying Dubai Marina buildings...")
|
|
buildings = get_marina_buildings(api)
|
|
|
|
if not buildings:
|
|
print("⚠ No buildings found. Check coordinates and query.")
|
|
return
|
|
|
|
print(f"Found {len(buildings)} buildings")
|
|
print("")
|
|
print("Converting to GeoJSON...")
|
|
buildings_to_geojson(buildings, args.output)
|
|
|
|
print("")
|
|
print("==========================================")
|
|
print("Import Complete")
|
|
print "=========================================="
|
|
print("")
|
|
print("Next steps:")
|
|
print("1. Review GeoJSON file in GIS software (QGIS)")
|
|
print("2. Import to Unreal Engine using GIS import tools")
|
|
print("3. Generate building meshes from footprints")
|
|
print("")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
main()
|
|
|