Method extract_geometry

src/exterior-shell-extractor/main.py:440–619

(self, **kwargs)

    @utils.trace
    # @profile
    def extract_geometry(self, **kwargs):
        # not only align facets part of the (potentially concave) input polyhedron, but also align facets resulting from the convex decomposition
        ALIGN_INNER = True

        s = ifcopenshell.geom.settings(
            USE_WORLD_COORDS=False,
            # ITERATOR_OUTPUT=ifcopenshell.ifcopenshell_wrapper.NATIVE,
            ITERATOR_OUTPUT=ifcopenshell.ifcopenshell_wrapper.TRIANGULATED,
            DISABLE_OPENING_SUBTRACTIONS=True,
        )


        its = []
        fffs = []

        data = model_geometry()


        for f in self.fs:
            if kwargs.keys() == {'include'}:
                kwargs2 = {'include': [e for e in kwargs['include'] if e.wrapped_data.file == f]}
            else:
                kwargs2 = kwargs
            it = ifcopenshell.geom.iterator(s, f, geometry_library="cgal", **kwargs2)
            
            if not it.initialize():
                # print(ifcopenshell.get_log())
                # exit(1)
                continue
            
            # convex decomposition is expensive, geometries can be shared, apply product-level transformations after CD and cache results pre-transform
            cd_cache = {}

            while True:
                elem = it.get()
                elem_g_id = elem.geometry.id
                  

                if f[int(elem_g_id.split("-")[0])].RepresentationIdentifier != "Box":
                    print(f"[{utils.get_mem()} MB]", "reading", f[elem.id])

                    elem = it.get_native()
                    elem2 = None
                    for i in range(elem.geometry.size()):
                        elem_i = elem.geometry.item(i)
                        repitem = f[elem.geometry.item_id(i)]

                        if elem_i.num_vertices() < 6:
                            # try and detect single faces used sometime for glass panes which can't
                            # be represented as halfspace intersection and need to be 'solidified'
                            fs = elem_i.facets()
                            axes_ = [f.axis() for f in fs]
                            axes = list(map(utils.to_tuple, axes_))
                            if all(ax == axes[0] for ax in axes):
                                ff = ifcopenshell.file(schema=f.schema)
                                ff.add(*f.by_type("IfcProject"))
                                nelem = ff.add(f[elem.id])
                                body = [rep for rep in nelem.Representation.Representations if rep.RepresentationIdentifier == "Body"][0]