Initial commit: Complete project foundation with all documentation, scripts, and project structure
This commit is contained in:
134
scripts/import_osm_data.py
Executable file
134
scripts/import_osm_data.py
Executable file
@@ -0,0 +1,134 @@
|
||||
#!/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()
|
||||
|
||||
Reference in New Issue
Block a user