OpenStreetMap数据转3D场景【Python + PostgreSQL】

import os
import osm2gmr as ogr

# 设置数据库连接信息
dbname = 'gis'
user = 'postgres'
password = 'password'
host = 'localhost'
port = '5432'

# 创建连接
conn_string = "host='{}' dbname='{}' user='{}' password='{}' port='{}'".format(host, dbname, user, password, port)

# 创建一个新的数据库连接
conn = ogr.Open(conn_string, update=1)

# 创建一个新的图层
lyr = conn.CreateLayer('osm_buildings', geom_type=ogr.wkbMultiPolygon)

# 添加一个ID字段
id_field = ogr.FieldDefn('id', ogr.OFTInteger)
lyr.CreateField(id_field)

# 添加一个名称字段
name_field = ogr.FieldDefn('name', ogr.OFTString)
name_field.SetWidth(254)
lyr.CreateField(name_field)

# 从OpenStreetMap获取建筑数据
osm_data = osm2gmr.get_data('your_area.osm')

# 解析数据并创建要素
for feature in osm_data:
    if feature.type == 'way':
        geom = feature.geom
        fid = feature.id
        tags = feature.tags

        # 检查是否为建筑
        if 'building' in tags:
            feat = ogr.Feature(lyr.GetLayerDefn())
            feat.SetGeometry(geom)
            feat.SetField('id', fid)
            if 'name' in tags:
                feat.SetField('name', tags['name'])
            lyr.CreateFeature(feat)

# 关闭连接
conn = None