RoomLayout3D_RandC/layout_structs.py at master · vevenom/RoomLayout3D_RandC · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
from collections import namedtuple

# Struct that describes the layout plane
LayoutPlane = namedtuple('LayoutPlane',
                           ['plane',
                            'mask',
                            'type',
                            ])
LayoutPlane.__new__.__defaults__ = (None,) * len(LayoutPlane._fields)

# Struct that describes the candidate layout component
CandidateLayoutComp = namedtuple('CandidateLayoutComp',
                           ['plane',
                            'mask',
                            'type',
                            'poly',
                            'poly_mask',
                            'cost',
                            'area'
                            ])
LayoutPlane.__new__.__defaults__ = (None,) * len(LayoutPlane._fields)