ExportLDraw/geometry_data.py at master · cuddlyogre/ExportLDraw · GitHub

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
class FaceData:
    """
    Raw vertex information
    """

    def __init__(self, vertices, color_code, texmap=None, pe_texmaps=None):
        self.vertices = vertices
        self.color_code = color_code
        self.texmap = texmap
        self.pe_texmaps = pe_texmaps

    # https://github.com/rredford/LdrawToObj/blob/802924fb8d42145c4f07c10824e3a7f2292a6717/LdrawData/LdrawToData.cs#L219
    # https://github.com/rredford/LdrawToObj/blob/802924fb8d42145c4f07c10824e3a7f2292a6717/LdrawData/LdrawToData.cs#L260
    @staticmethod
    def handle_vertex_winding(child_node, matrix, winding):
        vert_count = len(child_node.vertices)

        vertices = []
        if winding == "CW":
            if vert_count == 3:
                vertices = [
                    matrix @ child_node.vertices[0],
                    matrix @ child_node.vertices[2],
                    matrix @ child_node.vertices[1],
                ]
            elif vert_count == 4:
                vertices = [
                    matrix @ child_node.vertices[0],
                    matrix @ child_node.vertices[3],
                    matrix @ child_node.vertices[2],
                    matrix @ child_node.vertices[1],
                ]
                FaceData.fix_bowties(vertices)
        else:  # winding == "CCW" or winding is None:
            if vert_count == 3:
                vertices = [
                    matrix @ child_node.vertices[0],
                    matrix @ child_node.vertices[1],
                    matrix @ child_node.vertices[2],
                ]
            elif vert_count == 4:
                vertices = [
                    matrix @ child_node.vertices[0],
                    matrix @ child_node.vertices[1],
                    matrix @ child_node.vertices[2],
                    matrix @ child_node.vertices[3],
                ]
                FaceData.fix_bowties(vertices)

        return vertices

    # handle bowtie quadrilaterals - 6582.dat
    # https://github.com/TobyLobster/ImportLDraw/pull/65/commits/3d8cebee74bf6d0447b616660cc989e870f00085
    @staticmethod
    def fix_bowties(vertices):
        nA = (vertices[1] - vertices[0]).cross(vertices[2] - vertices[0])
        nB = (vertices[2] - vertices[1]).cross(vertices[3] - vertices[1])
        nC = (vertices[3] - vertices[2]).cross(vertices[0] - vertices[2])
        if nA.dot(nB) < 0:
            vertices[2], vertices[3] = vertices[3], vertices[2]
        elif nB.dot(nC) < 0:
            vertices[2], vertices[1] = vertices[1], vertices[2]


class GeometryData:
    """
    Raw mesh data used to build the final mesh.
    """

    def __init__(self):
        self.key = None
        self.file = None
        self.bfc_certified = None
        self.edge_data = []
        self.face_data = []
        self.line_data = []

    def add_edge_data(self, vertices, color_code):
        self.edge_data.append(FaceData(
            vertices=vertices,
            color_code=color_code,
        ))

    def add_face_data(self, vertices, color_code, texmap=None, pe_texmaps=None):
        self.face_data.append(FaceData(
            vertices=vertices,
            color_code=color_code,
            texmap=texmap,
            pe_texmaps=pe_texmaps,
        ))

    def add_line_data(self, vertices, color_code):
        self.line_data.append(FaceData(
            vertices=vertices,
            color_code=color_code,
        ))