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

这段代码示例展示了如何使用Python和osm2gmr库从OpenStreetMap数据中提取建筑数据，并将其转换为一个3D场景。代码中包含了数据库连接设置、创建图层、添加字段、解析数据和创建要素的过程。需要注意的是，这个例子假设你已经有了一个名为your_area.osm的OpenStreetMap数据文件，并且osm2gmr库已经安装。