Source code for moldflow.cad_manager

# SPDX-FileCopyrightText: 2025 Autodesk, Inc.
# SPDX-License-Identifier: Apache-2.0

"""
Usage:
    CADManager Class API Wrapper
"""

from .ent_list import EntList
from .vector import Vector
from .logger import process_log, LogMessage
from .helper import check_type, coerce_optional_dispatch
from .com_proxy import safe_com




class CADManager:
    """
    Wrapper for CADManager class of Moldflow Synergy.
    """

    def __init__(self, _cad_manager):
        """
        Initialize the CADManager with a CADManager instance from COM.

        Args:
            _cad_manager: The CADManager instance.
        """
        process_log(__name__, LogMessage.CLASS_INIT, locals(), name="CADManager")
        self.cad_manager = safe_com(_cad_manager)



    def create_entity_list(self) -> EntList:
        """
        Creates an empty EntList object
        When using this function, it will first ask for result invalidation.
        If you want to select entities without checking result, use StudyDoc.create_entity_list().

        Returns:
            EntList: The new entity list.
        """
        result = self.cad_manager.CreateEntityList
        if result is None:
            return None
        return EntList(result)




    def modify_cad_surfaces_by_normal(
        self, faces: EntList | None, transit_faces: EntList | None, distance: float
    ) -> bool:
        """
        Modify CAD faces by a given distance

        Args:
            faces (EntList | None): EntList object containing the faces to be modified
            transit_faces (EntList | None): EntList object containing the transit faces
                to be preserved
            distance (float): distance along input faces' normal direction

        Returns:
            bool: True if operation is successful; False otherwise
        """
        process_log(
            __name__, LogMessage.FUNCTION_CALL, locals(), name="modify_cad_surfaces_by_normal"
        )
        if faces is not None:
            check_type(faces, EntList)
        if transit_faces is not None:
            check_type(transit_faces, EntList)
        check_type(distance, (float, int))
        return self.cad_manager.ModifyCADSurfacesByNormal(
            coerce_optional_dispatch(faces, "ent_list"),
            coerce_optional_dispatch(transit_faces, "ent_list"),
            distance,
        )




    def modify_cad_surfaces_by_vector(
        self, faces: EntList | None, transit_faces: EntList | None, vector: Vector | None
    ) -> bool:
        """
        Modify CAD faces by a given vector

        Args:
            faces (EntList | None): EntList object containing the faces to be modified
            transit_faces (EntList | None): EntList object containing the transit
                faces to be preserved
            vector (Vector | None): Vector object that specifies the direction

        Returns:
            bool: True if operation is successful; False otherwise
        """
        process_log(
            __name__, LogMessage.FUNCTION_CALL, locals(), name="modify_cad_surfaces_by_vector"
        )
        if faces is not None:
            check_type(faces, EntList)
        if transit_faces is not None:
            check_type(transit_faces, EntList)
        if vector is not None:
            check_type(vector, Vector)
        return self.cad_manager.ModifyCADSurfacesByVector(
            coerce_optional_dispatch(faces, "ent_list"),
            coerce_optional_dispatch(transit_faces, "ent_list"),
            coerce_optional_dispatch(vector, "vector"),
        )