【dxf】（1）dxf 转 geojson

最新推荐文章于 2024-06-13 23:55:30 发布

GIS程序媛—椰子

最新推荐文章于 2024-06-13 23:55:30 发布

阅读量853

点赞数 2

分类专栏：文件格式解析文章标签： python 开发语言

本文链接：https://blog.csdn.net/OTTOkongbai/article/details/129366458

版权

文件格式解析专栏收录该内容

5 篇文章 0 订阅

订阅专栏

import json
import os

import ezdxf
import pyproj
from ezdxf.addons import geo

# XX请改成自己的数据
crs = pyproj.CRS('EPSG:XX')
wgs84 = pyproj.CRS('EPSG:4326')


def transformer(data, crs, wgs84):
    result = []
    if type(data) == tuple:
        project = pyproj.Transformer.from_crs(crs, wgs84, always_xy=True)
        wgs84_pts = project.transform(data[0], data[1])
        result.append(wgs84_pts)
    else:
        for points in data:
            if type(points) == list:
                points = transformer(points, crs, wgs84)
                result.append(points)
            else:
                project = pyproj.Transformer.from_crs(crs, wgs84, always_xy=True)
                wgs84_pts = project.transform(points[0], points[1])
                result.append(wgs84_pts)
    return result


def printEzdxf(file, output, layernames):
    doc = ezdxf.readfile(file)
    msp = doc.modelspace()
    features = []
    for layername in layernames:
        tracks = msp.query('*[layer=="' + layername + '"]')
        n = 0
        for track in tracks:
            print("dxftype: ", track.DXFTYPE)
            features.append(export_geojson(track, doc))
            n = n + 1
            print('percent: ', n / len(tracks) * 100)
    geo = {
        "type": "FeatureCollection",
        "features": features
    }
    with open(output, "wt", encoding="utf8") as fp:
        json.dump(geo, fp, indent=2)

    print("output: ", output)
 
 
def export_geojson(entity, doc):
    try:
        geo_proxy = ezdxf.addons.geo.proxy(entity)
        geometry = geo_proxy.__geo_interface__
        coordinates = transformer(geometry['coordinates'], pyproj.CRS('EPSG:3414'), pyproj.CRS('EPSG:4326'))
        print('geometry-after: ', coordinates)
        layer = doc.layers.get(str(entity.dxf.layer).lower())
        geo = {
            "type": "Feature",
            "geometry": {
                "type": geometry['type'],
                "coordinates": coordinates
            },
            "properties": {
                "handle": str(entity.dxf.handle),
                "type": str(entity.DXFTYPE),
                "layer": str(entity.dxf.layer),
                "color": str(layer.color),
            }
        }
        if entity.DXFTYPE == 'LINE':
            geo['properties']['linetype'] = str(layer.dxf.linetype)
            geo['properties']['lineweight'] = str(layer.dxf.lineweight)
        if entity.DXFTYPE == 'LWPOLYLINE':
            geo['properties']['linetype'] = str(layer.dxf.linetype)
            geo['properties']['lineweight'] = str(layer.dxf.lineweight)
        if entity.DXFTYPE == 'POLYLINE':
            geo['properties']['linetype'] = str(layer.dxf.linetype)
            geo['properties']['lineweight'] = str(layer.dxf.lineweight)
        if entity.DXFTYPE == 'CIRCLE':
            geo['properties']['center'] = str(entity.dxf.center)
            geo['properties']['radius'] = str(entity.dxf.radius)
        if entity.DXFTYPE == 'ARC':
            geo['properties']['center'] = str(entity.dxf.center)
            geo['properties']['radius'] = str(entity.dxf.radius)
            geo['properties']['start_angle'] = str(entity.dxf.start_angle)
            geo['properties']['end_angle'] = str(entity.dxf.end_angle)
        return geo
    except Exception as e:
        print("!!! Error found in export_geojson(): ", e)
        pass


if __name__ == "__main__":
    try:
        path = 'XX'
        file = 'XX.dxf'
        layername = 'XX'
        print('file-start: ', file)
        printEzdxf(os.path.join(path, file), file.split("\\")[-1].split(".")[0] + ".json", layername)
        print('file-done: ', file)
        print("### main(): done")
    except Exception as e:
        print("!!! Error found: ", e)
        pass