IFCの形状情報の秘密に迫る

from OCC.Extend.ShapeFactory import translate_shp, rotate_shp_3_axis
from OCC.Extend.ShapeFactory import make_wire, make_face, make_extrusion
from OCC.Core.BRepBuilderAPI import BRepBuilderAPI_MakePolygon

def get_polyline(item):
    if item.is_a() == 'IfcPolyline':
        points = item.Points
        points = [point.Coordinates for point in points]
        return points
    else:
        print('not IfcPolyline !!')
        
def extract_ifc_extruded_shape(item):
    if item.is_a() == 'IfcExtrudedAreaSolid':
        depth = item.Depth 
        direction = item.ExtrudedDirection.DirectionRatios
        swept_area = item.SweptArea
        position = item.Position
        if swept_area.is_a() == 'IfcArbitraryClosedProfileDef':
            outer_curve = swept_area.OuterCurve
            if outer_curve.is_a() == 'IfcPolyline':
                points = get_polyline(outer_curve)
                return points, direction, depth
            else:
                print('not IfcPolyline !!')
        else:
            print('not IfcArbitraryClosedProfileDef !!')
    else:
        print('not IfcExtrudedAreaSolid !!')

def points2face(points, is_close=True):
    gp_points = [(gp_Pnt(*p) if len(p) == 3 else gp_Pnt(*p, 0)) for p in points]
    polygon = BRepBuilderAPI_MakePolygon()
    for p in gp_points:
        polygon.Add(p)

    # 必要であれば始点と終点をつなぐ
    if is_close:
        polygon.Close()

    face = BRepBuilderAPI_MakeFace(polygon.Wire()).Face()
    return face

def create_extrude_shape(points, direction, length):
    face = points2face(points, is_close=False)
    extrusion = make_extrusion(face, length, gp_Vec(*direction))
    return extrusion

shapes = []
walls = ifc.by_type('IfcWall')
for element in walls:
    placement = element.ObjectPlacement
    transformation_matrix = placement_to_transformation_matrix(placement)
    tr, angle = matrix_to_translation_angle(transformation_matrix)
    
    representations = element.Representation.Representations
    for item in representations[0].Items:
        ret = extract_ifc_extruded_shape(item)
        if ret is not None:
            points, direction, depth = ret
            shape = create_extrude_shape(*ret)
            rotated_shape = rotate_shp_3_axis(shape, *angle, "rad")
            trans_shape = translate_shp(rotated_shape, gp_Vec(*tr))
            shapes.append(trans_shape)

shapes = []
walls = ifc.by_type('IfcWall')
for element in walls:
    placement = element.ObjectPlacement
    transformation = placement_to_transformation(placement)
    
    representations = element.Representation.Representations
    for item in representations[0].Items:
        if item.is_a() == 'IfcExtrudedAreaSolid':
            shape = ifc_extruded_area_solid_to_shape(item)
            trans_shape = BRepBuilderAPI_Transform(shape, transformation).Shape()
            opening_shapes = get_opening_shape(element)
            for opening_shape in opening_shapes:
                trans_shape = BRepAlgoAPI_Cut(trans_shape, opening_shape).Shape()
                
            shapes.append(trans_shape)
        else:
            print(item.id(), item.is_a())

# IfcExtrudedAreaSolid から押し出し形状
def ifc_extruded_area_solid_to_shape(item):
    ret = extract_ifc_extruded_shape(item)
    if ret is not None:
        points, direction, depth, position = ret
        transformation = placement3d_to_transformation(position)
        shape = create_extrude_shape(points, direction, depth)
        trans_shape = BRepBuilderAPI_Transform(shape, transformation).Shape()
        return trans_shape

# IfcOpeningElement の形状作成
def get_opening_shape(element):
    has_openings = element.HasOpenings
    opening_shapes = []
    for opening in has_openings:
        placement = opening.RelatedOpeningElement.ObjectPlacement
        transformation = placement_to_transformation(placement)
    
        items = opening.RelatedOpeningElement.Representation.Representations[0].Items
        for item in items:
            shape = ifc_extruded_area_solid_to_shape(item)
            trans_shape = BRepBuilderAPI_Transform(shape, transformation).Shape()
            opening_shapes.append(trans_shape)

    return opening_shapes

# IfcLocalPlacement を 変換行列に
def placement_to_transformation(placement):
    transform = placement.RelativePlacement
    transformation = placement3d_to_transformation(transform)

    rel = placement.PlacementRelTo
    if rel is not None:
        transformation.PreMultiply(placement_to_transformation(rel))

    return transformation

# IfcAxis2Placement3D を 変換行列に
def placement3d_to_transformation(item):
    location, axis, direction = extract_ifc_placement3d(item)
    ax3 = gp_Ax3(gp_Pnt(*location), gp_Dir(*axis), gp_Dir(*direction))
    ax2 = gp.gp.XOY()
    xoy = gp_Ax3(ax2.Location(), ax2.Direction(), ax2.XDirection())

    transformation = gp_Trsf()
    transformation.SetTransformation(ax3, xoy)
    return transformation

# 点から面
def points2face(points, is_close=True):
    gp_points = [(gp_Pnt(*p) if len(p) == 3 else gp_Pnt(*p, 0)) for p in points]
    polygon = BRepBuilderAPI_MakePolygon()
    for p in gp_points:
        polygon.Add(p)

    # 必要であれば始点と終点をつなぐ
    if is_close:
        polygon.Close()

    face = BRepBuilderAPI_MakeFace(polygon.Wire()).Face()
    return face

# 押し出し形状
def create_extrude_shape(points, direction, length):
    face = points2face(points, is_close=True)
    extrusion = make_extrusion(face, length, gp_Vec(*direction))
    return extrusion

# IfcAxis2Placement3D から値取得
def extract_ifc_placement3d(item):
    return (
        item.Location.Coordinates,
        item.Axis.DirectionRatios,
        item.RefDirection.DirectionRatios,
    )

# IfcExtrudedAreaSolid から値取得
def extract_ifc_extruded_shape(item):
    if item.is_a() == 'IfcExtrudedAreaSolid':
        depth = item.Depth
        direction = item.ExtrudedDirection.DirectionRatios
        position = item.Position

        swept_area = item.SweptArea
        if swept_area.is_a() == 'IfcArbitraryClosedProfileDef':
            points = extract_ifc_arbitrary_closed(swept_area)
            if points is None:
                return
        elif swept_area.is_a() == 'IfcRectangleProfileDef':
            points = extract_ifc_rectangle(swept_area)
        else:
            print_not_defined(item)

        return points, direction, depth, position

    else:
        print_not_defined(item)

# IfcArbitraryClosedProfileDef から値取得
def extract_ifc_arbitrary_closed(item):
    outer_curve = item.OuterCurve
    if outer_curve.is_a() == 'IfcPolyline':
        points = get_polyline(outer_curve)
        return points
    else:
        print_not_defined(item)

# IfcRectangleProfileDef から値取得
def extract_ifc_rectangle(item):
    x, y = item.XDim / 2, item.YDim / 2
    points = [
        [-x, -y],
        [x, -y],
        [x, y],
        [-x, y],
    ]
    return points


# IfcPolyline から値取得
def get_polyline(item):
    if item.is_a() == 'IfcPolyline':
        points = item.Points
        points = [point.Coordinates for point in points]
        return points
    else:
        print_not_defined(item)

# 表示用
def print_not_defined(item):
    print(item.id(), item.is_a())