dxf_to_ply.py 2.92 KB
Newer Older
Clement Pinard's avatar
Clement Pinard committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
import numpy as np
import ezdxf
import meshio
from path import Path
from argparse import ArgumentParser, ArgumentDefaultsHelpFormatter
from tqdm import tqdm


def dxf2numpy(dxf_file, centroid):
    print("Opening dxf file...")
    file = ezdxf.readfile(dxf_file)
    msp = file.modelspace()
    print("extracting edges from dxf file...")
    num_edges = sum(1 for _ in msp.query('POLYLINE'))
    edges = np.zeros((num_edges, 2, 3))

    for i, pl in tqdm(enumerate(msp.query('POLYLINE')), total=num_edges):
        start, end = pl.points()
        edges[i, 0] = start
        edges[i, 1] = end
    edges -= centroid
    return edges.astype(np.float32)


def edges2triangles(edges):
    vertices, edge_indices = np.unique(edges.reshape(-1, 3), axis=0, return_inverse=True)
    edge_indices = edge_indices.reshape(-1, 2)

    vertex_tree = {}

    def add_entry(tree, loc, target):
        if loc not in tree:
            tree[loc] = set()
        tree[loc].add(target)

    print("Constructing vertex tree...")
    for seg in tqdm(edge_indices):
        i_start, i_end = seg
        add_entry(vertex_tree, i_start, i_end)
        add_entry(vertex_tree, i_end, i_start)

    faces_set = set()

    print("Detecting triangles...")

    def sub_dict(tree, indices):
        return {k: tree[k] for k in indices if k in tree}

    for v1, leaf1 in tqdm(vertex_tree.items()):
        for v2, leaf2 in sub_dict(vertex_tree, leaf1).items():
            for v3, leaf3 in sub_dict(vertex_tree, leaf2).items():
                if v1 in leaf3:
                    faces_set.add(frozenset([v1, v2, v3]))

    faces = np.zeros((len(faces_set), 3), dtype=np.int32)
    for i, f in enumerate(faces_set):
        faces[i] = list(f)

    return vertices, faces


parser = ArgumentParser(description='convert a dxf file with only edges to a faced mesh, only counting triangles',
                        formatter_class=ArgumentDefaultsHelpFormatter)

parser.add_argument('--dxf', default="manoir.dxf",
                    help='dxf file, must contain the wireframe')
parser.add_argument('--centroid_path', default="centroid.txt",
                    help='txt containing the centroid computed with las2ply.py')
parser.add_argument('--output', default=None,
                    help="output file name. By default, will be dxf path with \".dxf\" replace with \"ply\"")


def main():
    args = parser.parse_args()
    if args.centroid_path is not None:
        centroid = np.loadtxt(args.centroid_path)
    else:
        centroid = np.zeros(3)

    if args.output is None:
        output_name = Path(args.dxf).stripext()
        output_path = str(output_name + ".ply")
    else:
        output_path = args.output

    edges = dxf2numpy(args.dxf, centroid)
    vertices, faces = edges2triangles(edges)

    cells = {
        "triangle": faces
    }
    meshio.write_points_cells(
        output_path,
        vertices,
        cells,
        file_format='ply-ascii'
    )


if __name__ == '__main__':
    main()