osm2maya.py

import maya.cmds as cmds

import random

import xml.dom.minidom

# replace filepath with the absolute path of the OSM file you have downloaded
doc = xml.dom.minidom.parse(filepath)


# building:levels


all_ways = doc.getElementsByTagName("way")

buildings = []

for el in all_ways:
	for ch in el.childNodes:
    
		if ch.attributes:

			if 'k' in ch.attributes.keys() and 'building' ==  ch.attributes['k'].value:
				buildings.append(el)
    
				break


nodes = doc.getElementsByTagName("node")
id_to_tuple = {}
for node in nodes:
	id_val = node.attributes['id'].value
	if 'lon' in node.attributes.keys():
		(lon, lat) = (node.attributes['lon'].value, node.attributes['lat'].value)
		id_to_tuple[id_val] = (lon, lat)
		

all_buildings = []

for b in buildings:
	lst = []
	nds = b.getElementsByTagName('nd')
	for ch in nds:
		if ch.tagName == 'nd':
			node_id = ch.attributes['ref'].value
			lst.append(id_to_tuple[node_id])
	
	tags = b.getElementsByTagName('tag')
	level = 1
	for tag in tags:
	    if tag.tagName == 'tag':
	        if tag.attributes['k'].value == 'building:levels':
	            try:
	                level = int(tag.attributes['v'].value)
	            except:
	                level = 1
	
	 
	
	all_buildings.append((lst, level))


print(all_buildings[0])

all_buildings = sorted(all_buildings)

sz = len(all_buildings)

start_lon = float(all_buildings[int(sz/2)][0][0][0])
start_lat = float(all_buildings[int(sz/2)][0][0][1])


print(all_buildings[0])


def get_xy(lon, lat):
    lon = float(lon)
    lat = float(lat)
    mul = 111.321 * 1000 # meters
    diff_lon = lon - start_lon
    diff_lat = lat - start_lat
    return (diff_lon * mul, diff_lat*mul)
     
buildings_xy = []

for lst in all_buildings:
    tmp = [] 
    for i in lst[0]:
        tmp.append(get_xy(i[0], i[1]))
        
    # height froom levels
    h = lst[1]
    buildings_xy.append((tmp, h))

print(buildings_xy[0])



# Polygons ready, now use it below

cubeList  = cmds.ls('myCube*')
if len(cubeList) > 0:
    cmds.delete(cubeList)
   
   

all_poly = []

for lst in buildings_xy:
    tmp = []

    for i in lst[0]:
        (x,z) = i
        x/=-100
        z/=100
        y = 0
        tmp.append((x,y,z))
    h = lst[1]
    res = cmds.polyCreateFacet( p=tmp, name='buildingpoly#')
  
    all_poly.append(res)
    thickness = random.uniform(0.1, 0.2)
    assert h >= 0
    
    normals = cmds.polyInfo(res[0], fn=1)
    normal = float(normals[0].split(":")[1].split()[1])
    
    # For reversed normals, Just redraw in anticlockwise
    if normal < 0:
        cmds.delete(res[0])
        tmp.reverse()
        res = cmds.polyCreateFacet( p=tmp, name='buildingpoly#')
        normals = cmds.polyInfo(res[0], fn=1)
        normal = float(normals[0].split(":")[1].split()[1])
        
    
    cmds.polyExtrudeFacet(res[0], kft=1, thickness=(h / 10.0))