Function get_element_data

src/ifc2ca/ca2ifc.py:28–117

(model, name, element)

def get_element_data(model, name, element):
    if element["geometry_type"] == "Edge":
        for i, cell_block in enumerate(model.cells):
            if cell_block.type == "line":
                cell_tags = model.cell_data["cell_tags"][i]
                break
        rows = []
        for i_row, i in enumerate(cell_tags):
            if i == 0:
                continue
            tags = model.cell_tags[i]
            for tag in tags:
                if tag == name:
                    # print(i_row, i)
                    rows.append(i_row)
                    break

        points = list(set(flatten([cell_block.data[c] for c in rows])))
        points.sort(key=lambda p: np.linalg.norm(model.points[p] - np.array(element["origin"])))
        coords = [np.round(model.points[p], 4).tolist() for p in points]
        local_coords = [
            [float(round(np.linalg.norm(model.points[p] - np.array(element["origin"])), 4))] for p in points
        ]

        return {
            "name": name,
            "points": points,
            "coords": coords,
            "local_coords": local_coords,
        }

    elif element["geometry_type"] == "Face":
        triangle_cell_tags = None
        quad_cell_tags = None
        for i, cell_block in enumerate(model.cells):
            if cell_block.type == "triangle":
                triangle_cell_tags = model.cell_data["cell_tags"][i]
                break

        if triangle_cell_tags is not None:
            rows = []
            for i_row, i in enumerate(triangle_cell_tags):
                if i == 0:
                    continue
                tags = model.cell_tags[i]
                for tag in tags:
                    if tag == name:
                        # print(i_row, i)
                        rows.append(i_row)
                        break
            if not len(rows):
                points = []
            else:
                points = list(flatten([cell_block.data[c] for c in rows]))

        for i, cell_block in enumerate(model.cells):
            if cell_block.type == "quad":
                quad_cell_tags = model.cell_data["cell_tags"][i]