从DXF文件中获取点坐标

最近使用CloudCompare比较多,在点云数据中标记出polygon之后,选取多个polygon文件保存出来的是.dxf 格式。

在网上查了查python用来解析.dxf 文件的是ezdxf模块,但是内容有些多,所以偷懒直接用vim 打开.dxf 文件,找到了其中关于点坐标的信息,还是有一定规律的。

从DXF文件中获取点坐标_第1张图片

以 "VERTEX" 开头,接下来的9行都按照一定的格式分布。

"POLYLINE_001" 是后面的001是用来区分不同的polygon的。

"10", "20", "30"  下面的一行分别对应3D空间中某个点的X,Y,Z 坐标。

其他的不清楚是什么意思了(如果有明白的欢迎留言)。

为了获取不同的polygon都包含哪些点的信息,写了如下的python脚本。

从一个.dxf中可能获取多个polygon 点集,分别按照编号生成单独的.txt文件,例如POLYGON_001.txt,点的坐标保留小数点后8位。

import os

input_path = '{YOUR_DXF_FILE_PATH}'
already_collect = dict()

with open(input_path, 'r') as f_read:
    vertex_index = []
    content = f_read.readlines()
    for idx, line in enumerate(content):
        if line == 'VERTEX\n':
            vertex_index.append(idx)
    for idx in vertex_index:
        if already_collect == {}:
            first_key = content[idx + 2].split('\n')[0]
            already_collect[first_key] = []
            
        current_key = content[idx + 2].split('\n')[0]
        x = content[idx + 4].split('\n')[0]
        y = content[idx + 6].split('\n')[0]
        z = content[idx + 8].split('\n')[0]
        if current_key not in already_collect.keys():
            already_collect[current_key] = ['{:.8f} {:.8f} {:.8f}\n'.format(float(x), float(y), float(z))]
        else:
            already_collect[current_key].append('{:.8f} {:.8f} {:.8f}\n'.format(float(x), float(y), float(z)))

    for key, lines in already_collect.items():
        output_file_path = os.path.join(os.path.dirname(input_path), key + '.txt')
        with open(output_file_path, 'w') as f:
            for line in lines:
                f.writelines(line)
            print('successful create {}'.format(output_file_path))


你可能感兴趣的:(tool,python)