helpers/shape_processor.py

Refer to the Model Explore Helper Implementation

# $Id$
#
# Copyright (C) 2008-2014, Roman Lygin. All rights reserved.
# Copyright (C) 2014-2025, CADEX. All rights reserved.
#
# This file is part of the Manufacturing Toolkit software.
#
# You may use this file under the terms of the BSD license as follows:
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions are met:
# * Redistributions of source code must retain the above copyright notice,
#   this list of conditions and the following disclaimer.
# * Redistributions in binary form must reproduce the above copyright notice,
#   this list of conditions and the following disclaimer in the documentation
#   and/or other materials provided with the distribution.
#
# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
# AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
# IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
# ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE
# LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
# CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
# SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
# INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
# CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
# ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
# POSSIBILITY OF SUCH DAMAGE.
 
from abc import abstractmethod
 
import manufacturingtoolkit.CadExMTK as mtk
 
class ShapeProcessor(mtk.ModelData_ModelElementVoidVisitor):
    def __init__(self):
        super().__init__()
        self.myPartIndex = 0
 
    def VisitPart(self, thePart: mtk.ModelData_Part):
        aPartName = "noname" if thePart.Name().IsEmpty() else thePart.Name()
        
        aBodyList = thePart.Bodies()
        i = 0
        for aBody in aBodyList:
            aShapeIt = mtk.ModelData_ShapeIterator(aBody)
            for aShape in aShapeIt:
                if aShape.Type() == mtk.ShapeType_Solid:
                    print("Part #", self.myPartIndex, " [\"", aPartName, "\"] - solid #", i, " has:", sep="")
                    i += 1
                    self.ProcessSolid(mtk.ModelData_Solid.Cast(aShape))
                elif aShape.Type() == mtk.ShapeType_Shell:
                    print("Part #", self.myPartIndex, " [\"", aPartName, "\"] - shell #", i, " has:", sep="")
                    i += 1
                    self.ProcessShell(mtk.ModelData_Shell.Cast (aShape))
        self.myPartIndex += 1
 
    @abstractmethod
    def ProcessSolid(self, theSolid: mtk.ModelData_Solid):
        pass
 
    @abstractmethod
    def ProcessShell(self, theShell: mtk.ModelData_Shell):
        pass
 
class SolidProcessor(mtk.ModelData_ModelElementVoidVisitor):
    def __init__(self):
        super().__init__()
        self.myPartIndex = 0
 
    def VisitPart(self, thePart: mtk.ModelData_Part):
        aPartName = "noname" if thePart.Name().IsEmpty() else thePart.Name()
 
        aBodyList = thePart.Bodies()
        i = 0
        for aBody in aBodyList:
            aShapeIt = mtk.ModelData_ShapeIterator(aBody)
            for aShape in aShapeIt:
                if aShape.Type() == mtk.ShapeType_Solid:
                    print("Part #", self.myPartIndex, " [\"", aPartName, "\"] - solid #", i, " has:", sep="")
                    i += 1
                    self.ProcessSolid (mtk.ModelData_Solid.Cast (aShape))
        self.myPartIndex += 1
 
    @abstractmethod
    def ProcessSolid(self, theSolid: mtk.ModelData_Solid):
        pass
 
class SolidAndMeshProcessor(mtk.ModelData_ModelElementVoidVisitor):
    def __init__(self):
        super().__init__()
        self.myPartIndex = 0
        self.myShapeIndex = 0
 
    def VisitPart(self, thePart: mtk.ModelData_Part):
        aPartName = "noname" if thePart.Name().IsEmpty() else thePart.Name()
 
        myShapeIndex = 0
        aBodyList = thePart.Bodies()
        for aBody in aBodyList:
            if mtk.ModelData_MeshBody.CompareType(aBody):
                aMeshBody = mtk.ModelData_MeshBody.Cast(aBody)
                self._ProcessMeshBody(aMeshBody, aPartName)
            elif mtk.ModelData_SolidBody.CompareType(aBody):
                aSolid = mtk.ModelData_SolidBody.Cast(aBody).Solid()
                print("Part #", self.myPartIndex, " [\"", aPartName, "\"] - solid #", myShapeIndex, " has:", sep="")
                self.ProcessSolid (aSolid, aPartName)
                self.myShapeIndex += 1
        self.myPartIndex += 1
 
    @abstractmethod
    def ProcessSolid(self, theSolid: mtk.ModelData_Solid, thePartName: mtk.UTF16String):
        pass
 
    @abstractmethod
    def ProcessITS(self, theITS: mtk.ModelData_IndexedTriangleSet, thePartName: mtk.UTF16String):
        pass
 
    def _ProcessMeshBody(self, theMeshBody: mtk.ModelData_MeshBody, thePartName: mtk.UTF16String):
        aMeshShapes = theMeshBody.Shapes()
        for aMeshShape in aMeshShapes:
            if(mtk.ModelData_IndexedTriangleSet.CompareType(aMeshShape)):
                print("Part #", self.myPartIndex, " [\"", thePartName, "\"] - ITS #", self.myShapeIndex, " has:", sep="")
                anITS = mtk.ModelData_IndexedTriangleSet.Cast(aMeshShape)
                self.ProcessITS(anITS, thePartName)
                self.myShapeIndex += 1