Пример #1
0
 def _checkEquation(self, obj):
     for o in obj.Document.Objects:
         if (
             femutils.is_derived_from(o, "Fem::FemSolverObject")
             and hasattr(o, "Group")
             and obj in o.Group
         ):
             if o in _machines:
                 _machines[o].reset()
Пример #2
0
 def _checkEquation(self, obj):
     for o in obj.Document.Objects:
         if (
             FemUtils.is_derived_from(o, "Fem::FemSolverObject")
             and hasattr(o, "Group")
             and obj in o.Group
         ):
             if o in _machines:
                 _machines[o].reset()
Пример #3
0
 def _checkAnalysis(self, obj):
     if FemUtils.is_derived_from(obj, "Fem::FemAnalysis"):
         deltaObjs = self._getAdded(obj)
         if deltaObjs:
             reset = False
             for o in deltaObjs:
                 if self._partOfModel(o):
                     reset = True
             if reset:
                 self._resetAll(obj)
Пример #4
0
 def _checkAnalysis(self, obj):
     if FemUtils.is_derived_from(obj, "Fem::FemAnalysis"):
         deltaObjs = self._getAdded(obj)
         if deltaObjs:
             reset = False
             for o in deltaObjs:
                 if self._partOfModel(o):
                     reset = True
             if reset:
                 self._resetAll(obj)
Пример #5
0
def run_analysis(doc, base_name, filepath=""):

    from os.path import join, exists
    from os import makedirs
    from tempfile import gettempdir as gettmp

    # recompute
    doc.recompute()

    # print(doc.Objects)
    # print([obj.Name for obj in doc.Objects])

    # filepath
    if filepath is "":
        filepath = join(gettmp(), "FEM_examples")
    if not exists(filepath):
        makedirs(filepath)

    # find solver
    # ATM we only support one solver, search for a frame work solver and run it
    for m in doc.Analysis.Group:
        from femtools.femutils import is_derived_from
        if (
            is_derived_from(m, "Fem::FemSolverObjectPython")
            and m.Proxy.Type is not "Fem::FemSolverCalculixCcxTools"
        ):
            solver = m
            break

    # we need a file name for the besides dir to work
    save_fc_file = join(filepath, (base_name + ".FCStd"))
    FreeCAD.Console.PrintMessage(
        "Save FreeCAD file for {} analysis to {}\n.".format(base_name, save_fc_file)
    )
    doc.saveAs(save_fc_file)

    # get analysis workig dir
    from femtools.femutils import get_beside_dir
    working_dir = get_beside_dir(solver)

    # run analysis
    from femsolver.run import run_fem_solver
    run_fem_solver(solver, working_dir)

    # save doc once again with results
    doc.save()
Пример #6
0
def run_analysis(doc, base_name, filepath=''):

    from os.path import join, exists
    from os import makedirs
    from tempfile import gettempdir as gettmp

    # recompute
    doc.recompute()

    # print(doc.Objects)
    # print([obj.Name for obj in doc.Objects])

    # filepath
    if filepath is '':
        filepath = join(gettmp(), 'FEM_examples')
    if not exists(filepath):
        makedirs(filepath)

    # find solver
    # ATM we only support one solver, search for a frame work solver and run it
    for m in doc.Analysis.Group:
        from femtools.femutils import is_derived_from
        if is_derived_from(m, "Fem::FemSolverObjectPython") and m.Proxy.Type is not 'Fem::FemSolverCalculixCcxTools':
            solver = m
            break

    # we need a file name for the besides dir to work
    save_fc_file = join(filepath, (base_name + '.FCStd'))
    FreeCAD.Console.PrintMessage('Save FreeCAD file for {} analysis to {}\n.'.format(base_name, save_fc_file))
    doc.saveAs(save_fc_file)

    # get analysis workig dir
    from femsolver.run import _getBesideDir as getpath
    working_dir = getpath(solver)

    # run analysis
    from femsolver.run import run_fem_solver
    run_fem_solver(solver, working_dir)

    # save doc once again with results
    doc.save()
Пример #7
0
    def test_femobjects_derivedfromfem(self):
        # try to add all possible True types from inheritance chain see https://forum.freecadweb.org/viewtopic.php?f=10&t=32625
        doc = self.active_doc

        from femtools.femutils import is_derived_from

        materialsolid = ObjectsFem.makeMaterialSolid(doc)
        mesh = ObjectsFem.makeMeshGmsh(doc)
        solverelmer = ObjectsFem.makeSolverElmer(doc)

        self.assertTrue(is_derived_from(ObjectsFem.makeAnalysis(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeAnalysis(doc), 'Fem::FemAnalysis'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintBearing(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintBearing(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintBearing(doc), 'Fem::ConstraintBearing'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintBodyHeatSource(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintBodyHeatSource(doc), 'Fem::ConstraintPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintBodyHeatSource(doc), 'Fem::ConstraintBodyHeatSource'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintContact(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintContact(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintContact(doc), 'Fem::ConstraintContact'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintDisplacement(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintDisplacement(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintDisplacement(doc), 'Fem::ConstraintDisplacement'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintElectrostaticPotential(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintElectrostaticPotential(doc), 'Fem::ConstraintPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintElectrostaticPotential(doc), 'Fem::ConstraintElectrostaticPotential'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintFixed(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintFixed(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintFixed(doc), 'Fem::ConstraintFixed'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintFlowVelocity(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintFlowVelocity(doc), 'Fem::ConstraintPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintFlowVelocity(doc), 'Fem::ConstraintFlowVelocity'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintFluidBoundary(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintFluidBoundary(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintFluidBoundary(doc), 'Fem::ConstraintFluidBoundary'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintForce(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintForce(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintForce(doc), 'Fem::ConstraintForce'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintGear(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintGear(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintGear(doc), 'Fem::ConstraintGear'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintHeatflux(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintHeatflux(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintHeatflux(doc), 'Fem::ConstraintHeatflux'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintInitialFlowVelocity(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintInitialFlowVelocity(doc), 'Fem::ConstraintPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintInitialFlowVelocity(doc), 'Fem::ConstraintInitialFlowVelocity'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintInitialTemperature(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintInitialTemperature(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintInitialTemperature(doc), 'Fem::ConstraintInitialTemperature'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintPlaneRotation(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintPlaneRotation(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintPlaneRotation(doc), 'Fem::ConstraintPlaneRotation'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintPressure(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintPressure(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintPressure(doc), 'Fem::ConstraintPressure'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintPulley(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintPulley(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintPulley(doc), 'Fem::ConstraintPulley'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintSelfWeight(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintSelfWeight(doc), 'Fem::ConstraintPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintSelfWeight(doc), 'Fem::ConstraintSelfWeight'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintTemperature(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintTemperature(doc), 'Fem::Constraint'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintTemperature(doc), 'Fem::ConstraintTemperature'))

        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintTransform(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeConstraintTransform(doc), 'Fem::ConstraintTransform'))

        self.assertTrue(is_derived_from(ObjectsFem.makeElementFluid1D(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeElementFluid1D(doc), 'Fem::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeElementFluid1D(doc), 'Fem::FemElementFluid1D'))

        self.assertTrue(is_derived_from(ObjectsFem.makeElementGeometry1D(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeElementGeometry1D(doc), 'Fem::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeElementGeometry1D(doc), 'Fem::FemElementGeometry1D'))

        self.assertTrue(is_derived_from(ObjectsFem.makeElementGeometry2D(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeElementGeometry2D(doc), 'Fem::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeElementGeometry2D(doc), 'Fem::FemElementGeometry2D'))

        self.assertTrue(is_derived_from(ObjectsFem.makeElementRotation1D(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeElementRotation1D(doc), 'Fem::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeElementRotation1D(doc), 'Fem::FemElementRotation1D'))

        self.assertTrue(is_derived_from(ObjectsFem.makeMaterialFluid(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMaterialFluid(doc), 'App::MaterialObjectPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMaterialFluid(doc), 'Fem::Material'))

        self.assertTrue(is_derived_from(materialsolid, 'App::DocumentObject'))
        self.assertTrue(is_derived_from(materialsolid, 'App::MaterialObjectPython'))
        self.assertTrue(is_derived_from(materialsolid, 'Fem::Material'))

        self.assertTrue(is_derived_from(ObjectsFem.makeMaterialMechanicalNonlinear(doc, materialsolid), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMaterialMechanicalNonlinear(doc, materialsolid), 'Fem::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMaterialMechanicalNonlinear(doc, materialsolid), 'Fem::MaterialMechanicalNonlinear'))

        self.assertTrue(is_derived_from(mesh, 'App::DocumentObject'))
        self.assertTrue(is_derived_from(mesh, 'Fem::FemMeshObjectPython'))
        self.assertTrue(is_derived_from(mesh, 'Fem::FemMeshGmsh'))

        self.assertTrue(is_derived_from(ObjectsFem.makeMeshBoundaryLayer(doc, mesh), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMeshBoundaryLayer(doc, mesh), 'Fem::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMeshBoundaryLayer(doc, mesh), 'Fem::FemMeshBoundaryLayer'))

        self.assertTrue(is_derived_from(ObjectsFem.makeMeshGroup(doc, mesh), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMeshGroup(doc, mesh), 'Fem::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMeshGroup(doc, mesh), 'Fem::FemMeshGroup'))

        self.assertTrue(is_derived_from(ObjectsFem.makeMeshRegion(doc, mesh), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMeshRegion(doc, mesh), 'Fem::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMeshRegion(doc, mesh), 'Fem::FemMeshRegion'))

        self.assertTrue(is_derived_from(ObjectsFem.makeMeshNetgen(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMeshNetgen(doc), 'Fem::FemMeshShapeNetgenObject'))

        self.assertTrue(is_derived_from(ObjectsFem.makeMeshResult(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMeshResult(doc), 'Fem::FemMeshObjectPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeMeshResult(doc), 'Fem::FemMeshResult'))

        self.assertTrue(is_derived_from(ObjectsFem.makeResultMechanical(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeResultMechanical(doc), 'Fem::FemResultObjectPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeResultMechanical(doc), 'Fem::FemResultMechanical'))

        self.assertTrue(is_derived_from(ObjectsFem.makeSolverCalculixCcxTools(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeSolverCalculixCcxTools(doc), 'Fem::FemSolverObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeSolverCalculixCcxTools(doc), 'Fem::FemSolverObjectPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeSolverCalculixCcxTools(doc), 'Fem::FemSolverCalculixCcxTools'))

        self.assertTrue(is_derived_from(ObjectsFem.makeSolverCalculix(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeSolverCalculix(doc), 'Fem::FemSolverObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeSolverCalculix(doc), 'Fem::FemSolverObjectPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeSolverCalculix(doc), 'Fem::FemSolverObjectCalculix'))

        self.assertTrue(is_derived_from(solverelmer, 'App::DocumentObject'))
        self.assertTrue(is_derived_from(solverelmer, 'Fem::FemSolverObject'))
        self.assertTrue(is_derived_from(solverelmer, 'Fem::FemSolverObjectPython'))
        self.assertTrue(is_derived_from(solverelmer, 'Fem::FemSolverObjectElmer'))

        self.assertTrue(is_derived_from(ObjectsFem.makeSolverZ88(doc), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeSolverZ88(doc), 'Fem::FemSolverObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeSolverZ88(doc), 'Fem::FemSolverObjectPython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeSolverZ88(doc), 'Fem::FemSolverObjectZ88'))

        self.assertTrue(is_derived_from(ObjectsFem.makeEquationElasticity(doc, solverelmer), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeEquationElasticity(doc, solverelmer), 'App::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeEquationElasticity(doc, solverelmer), 'Fem::FemEquationElmerElasticity'))

        self.assertTrue(is_derived_from(ObjectsFem.makeEquationElectrostatic(doc, solverelmer), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeEquationElectrostatic(doc, solverelmer), 'App::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeEquationElectrostatic(doc, solverelmer), 'Fem::FemEquationElmerElectrostatic'))

        self.assertTrue(is_derived_from(ObjectsFem.makeEquationFlow(doc, solverelmer), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeEquationFlow(doc, solverelmer), 'App::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeEquationFlow(doc, solverelmer), 'Fem::FemEquationElmerFlow'))

        self.assertTrue(is_derived_from(ObjectsFem.makeEquationFluxsolver(doc, solverelmer), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeEquationFluxsolver(doc, solverelmer), 'App::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeEquationFluxsolver(doc, solverelmer), 'Fem::FemEquationElmerFluxsolver'))

        self.assertTrue(is_derived_from(ObjectsFem.makeEquationHeat(doc, solverelmer), 'App::DocumentObject'))
        self.assertTrue(is_derived_from(ObjectsFem.makeEquationHeat(doc, solverelmer), 'App::FeaturePython'))
        self.assertTrue(is_derived_from(ObjectsFem.makeEquationHeat(doc, solverelmer), 'Fem::FemEquationElmerHeat'))
Пример #8
0
    def test_femobjects_derivedfromfem(self):
        # try to add all possible True types from inheritance chain see
        # https://forum.freecadweb.org/viewtopic.php?f=10&t=32625
        doc = self.active_doc

        from femtools.femutils import is_derived_from

        materialsolid = ObjectsFem.makeMaterialSolid(doc)
        mesh = ObjectsFem.makeMeshGmsh(doc)
        solverelmer = ObjectsFem.makeSolverElmer(doc)

        # FemAnalysis
        self.assertTrue(
            is_derived_from(ObjectsFem.makeAnalysis(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeAnalysis(doc), 'Fem::FemAnalysis'))

        # ConstraintBearing
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintBearing(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintBearing(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintBearing(doc),
                            'Fem::ConstraintBearing'))

        # ConstraintBodyHeatSource
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintBodyHeatSource(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintBodyHeatSource(doc),
                            'Fem::ConstraintPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintBodyHeatSource(doc),
                            'Fem::ConstraintBodyHeatSource'))

        # ConstraintContact
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintContact(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintContact(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintContact(doc),
                            'Fem::ConstraintContact'))

        # ConstraintDisplacement
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintDisplacement(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintDisplacement(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintDisplacement(doc),
                            'Fem::ConstraintDisplacement'))

        # ConstraintElectrostaticPotential
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeConstraintElectrostaticPotential(doc),
                'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeConstraintElectrostaticPotential(doc),
                'Fem::ConstraintPython'))
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeConstraintElectrostaticPotential(doc),
                'Fem::ConstraintElectrostaticPotential'))

        # ConstraintFixed
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintFixed(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintFixed(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintFixed(doc),
                            'Fem::ConstraintFixed'))

        # ConstraintFlowVelocity
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintFlowVelocity(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintFlowVelocity(doc),
                            'Fem::ConstraintPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintFlowVelocity(doc),
                            'Fem::ConstraintFlowVelocity'))

        # ConstraintFluidBoundary
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintFluidBoundary(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintFluidBoundary(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintFluidBoundary(doc),
                            'Fem::ConstraintFluidBoundary'))

        # ConstraintForce
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintForce(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintForce(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintForce(doc),
                            'Fem::ConstraintForce'))

        # ConstraintGear
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintGear(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintGear(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintGear(doc),
                            'Fem::ConstraintGear'))

        # ConstraintHeatflux
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintHeatflux(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintHeatflux(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintHeatflux(doc),
                            'Fem::ConstraintHeatflux'))

        # ConstraintInitialFlowVelocity
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintInitialFlowVelocity(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintInitialFlowVelocity(doc),
                            'Fem::ConstraintPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintInitialFlowVelocity(doc),
                            'Fem::ConstraintInitialFlowVelocity'))

        # ConstraintInitialTemperature
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintInitialTemperature(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintInitialTemperature(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintInitialTemperature(doc),
                            'Fem::ConstraintInitialTemperature'))

        # ConstraintPlaneRotation
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintPlaneRotation(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintPlaneRotation(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintPlaneRotation(doc),
                            'Fem::ConstraintPlaneRotation'))

        # ConstraintPressure
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintPressure(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintPressure(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintPressure(doc),
                            'Fem::ConstraintPressure'))

        # ConstraintPulley
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintPulley(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintPulley(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintPulley(doc),
                            'Fem::ConstraintPulley'))

        # ConstraintSelfWeight
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintSelfWeight(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintSelfWeight(doc),
                            'Fem::ConstraintPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintSelfWeight(doc),
                            'Fem::ConstraintSelfWeight'))

        # ConstraintTemperature
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintTemperature(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintTemperature(doc),
                            'Fem::Constraint'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintTemperature(doc),
                            'Fem::ConstraintTemperature'))

        # ConstraintTransform
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintTransform(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeConstraintTransform(doc),
                            'Fem::ConstraintTransform'))

        # FemElementFluid1D
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementFluid1D(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementFluid1D(doc),
                            'Fem::FeaturePython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementFluid1D(doc),
                            'Fem::FemElementFluid1D'))

        # FemElementGeometry1D
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementGeometry1D(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementGeometry1D(doc),
                            'Fem::FeaturePython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementGeometry1D(doc),
                            'Fem::FemElementGeometry1D'))

        # FemElementGeometry2D
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementGeometry2D(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementGeometry2D(doc),
                            'Fem::FeaturePython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementGeometry2D(doc),
                            'Fem::FemElementGeometry2D'))

        # FemElementRotation1D
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementRotation1D(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementRotation1D(doc),
                            'Fem::FeaturePython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeElementRotation1D(doc),
                            'Fem::FemElementRotation1D'))

        # Material
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMaterialFluid(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMaterialFluid(doc),
                            'App::MaterialObjectPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMaterialFluid(doc),
                            'Fem::Material'))

        # Material
        self.assertTrue(is_derived_from(materialsolid, 'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(materialsolid, 'App::MaterialObjectPython'))
        self.assertTrue(is_derived_from(materialsolid, 'Fem::Material'))

        # MaterialMechanicalNonlinear
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeMaterialMechanicalNonlinear(doc, materialsolid),
                'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeMaterialMechanicalNonlinear(doc, materialsolid),
                'Fem::FeaturePython'))
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeMaterialMechanicalNonlinear(doc, materialsolid),
                'Fem::MaterialMechanicalNonlinear'))

        # MaterialReinforced
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMaterialReinforced(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMaterialReinforced(doc),
                            'App::MaterialObjectPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMaterialReinforced(doc),
                            'Fem::MaterialReinforced'))

        # FemMeshGmsh
        self.assertTrue(is_derived_from(mesh, 'App::DocumentObject'))
        self.assertTrue(is_derived_from(mesh, 'Fem::FemMeshObjectPython'))
        self.assertTrue(is_derived_from(mesh, 'Fem::FemMeshGmsh'))

        # FemMeshBoundaryLayer
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshBoundaryLayer(doc, mesh),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshBoundaryLayer(doc, mesh),
                            'Fem::FeaturePython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshBoundaryLayer(doc, mesh),
                            'Fem::FemMeshBoundaryLayer'))

        # FemMeshGroup
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshGroup(doc, mesh),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshGroup(doc, mesh),
                            'Fem::FeaturePython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshGroup(doc, mesh),
                            'Fem::FemMeshGroup'))

        # FemMeshRegion
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshRegion(doc, mesh),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshRegion(doc, mesh),
                            'Fem::FeaturePython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshRegion(doc, mesh),
                            'Fem::FemMeshRegion'))

        # FemMeshShapeNetgenObject
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshNetgen(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshNetgen(doc),
                            'Fem::FemMeshShapeNetgenObject'))

        # FemMeshResult
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshResult(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshResult(doc),
                            'Fem::FemMeshObjectPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeMeshResult(doc),
                            'Fem::FemMeshResult'))

        # FemResultMechanical
        self.assertTrue(
            is_derived_from(ObjectsFem.makeResultMechanical(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeResultMechanical(doc),
                            'Fem::FemResultObjectPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeResultMechanical(doc),
                            'Fem::FemResultMechanical'))

        # FemSolverCalculixCcxTools
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverCalculixCcxTools(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverCalculixCcxTools(doc),
                            'Fem::FemSolverObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverCalculixCcxTools(doc),
                            'Fem::FemSolverObjectPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverCalculixCcxTools(doc),
                            'Fem::FemSolverCalculixCcxTools'))

        # FemSolverObjectCalculix
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverCalculix(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverCalculix(doc),
                            'Fem::FemSolverObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverCalculix(doc),
                            'Fem::FemSolverObjectPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverCalculix(doc),
                            'Fem::FemSolverObjectCalculix'))

        # FemSolverObjectElmer
        self.assertTrue(is_derived_from(solverelmer, 'App::DocumentObject'))
        self.assertTrue(is_derived_from(solverelmer, 'Fem::FemSolverObject'))
        self.assertTrue(
            is_derived_from(solverelmer, 'Fem::FemSolverObjectPython'))
        self.assertTrue(
            is_derived_from(solverelmer, 'Fem::FemSolverObjectElmer'))

        # FemSolverObjectZ88
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverZ88(doc),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverZ88(doc),
                            'Fem::FemSolverObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverZ88(doc),
                            'Fem::FemSolverObjectPython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeSolverZ88(doc),
                            'Fem::FemSolverObjectZ88'))

        # FemEquationElmerElasticity
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeEquationElasticity(doc, solverelmer),
                'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeEquationElasticity(doc, solverelmer),
                'App::FeaturePython'))
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeEquationElasticity(doc, solverelmer),
                'Fem::FemEquationElmerElasticity'))

        # FemEquationElmerElectrostatic
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeEquationElectrostatic(doc, solverelmer),
                'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeEquationElectrostatic(doc, solverelmer),
                'App::FeaturePython'))
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeEquationElectrostatic(doc, solverelmer),
                'Fem::FemEquationElmerElectrostatic'))

        # FemEquationElmerFlow
        self.assertTrue(
            is_derived_from(ObjectsFem.makeEquationFlow(doc, solverelmer),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeEquationFlow(doc, solverelmer),
                            'App::FeaturePython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeEquationFlow(doc, solverelmer),
                            'Fem::FemEquationElmerFlow'))

        # FemEquationElmerFluxsolver
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeEquationFluxsolver(doc, solverelmer),
                'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeEquationFluxsolver(doc, solverelmer),
                'App::FeaturePython'))
        self.assertTrue(
            is_derived_from(
                ObjectsFem.makeEquationFluxsolver(doc, solverelmer),
                'Fem::FemEquationElmerFluxsolver'))

        # FemEquationElmerHeat
        self.assertTrue(
            is_derived_from(ObjectsFem.makeEquationHeat(doc, solverelmer),
                            'App::DocumentObject'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeEquationHeat(doc, solverelmer),
                            'App::FeaturePython'))
        self.assertTrue(
            is_derived_from(ObjectsFem.makeEquationHeat(doc, solverelmer),
                            'Fem::FemEquationElmerHeat'))
Пример #9
0
 def _partOfModel(self, obj):
     for t in self._WHITELIST:
         if FemUtils.is_derived_from(obj, t):
             return True
     return False
Пример #10
0
    def test_femobjects_derivedfromfem(self):
        # try to add all possible True types from inheritance chain see
        # https://forum.freecadweb.org/viewtopic.php?f=10&t=32625
        doc = self.active_doc

        from femtools.femutils import is_derived_from

        # FemAnalysis
        analysis = ObjectsFem.makeAnalysis(doc)
        self.assertTrue(is_derived_from(analysis, "App::DocumentObject"))
        self.assertTrue(is_derived_from(analysis, "Fem::FemAnalysis"))

        # ConstraintBearing
        constraint_bearing = ObjectsFem.makeConstraintBearing(doc)
        self.assertTrue(
            is_derived_from(constraint_bearing, "App::DocumentObject"))
        self.assertTrue(is_derived_from(constraint_bearing, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_bearing, "Fem::ConstraintBearing"))

        # ConstraintBodyHeatSource
        constraint_body_heat_source = ObjectsFem.makeConstraintBodyHeatSource(
            doc)
        self.assertTrue(
            is_derived_from(constraint_body_heat_source,
                            "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_body_heat_source,
                            "Fem::ConstraintPython"))
        self.assertTrue(
            is_derived_from(constraint_body_heat_source,
                            "Fem::ConstraintBodyHeatSource"))

        # ConstraintContact
        constraint_contact = ObjectsFem.makeConstraintContact(doc)
        self.assertTrue(
            is_derived_from(constraint_contact, "App::DocumentObject"))
        self.assertTrue(is_derived_from(constraint_contact, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_contact, "Fem::ConstraintContact"))

        # ConstraintDisplacement
        constraint_dicplacement = ObjectsFem.makeConstraintDisplacement(doc)
        self.assertTrue(
            is_derived_from(constraint_dicplacement, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_dicplacement, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_dicplacement,
                            "Fem::ConstraintDisplacement"))

        # ConstraintElectrostaticPotential
        constraint_electorstatic_potential = ObjectsFem.makeConstraintElectrostaticPotential(
            doc)
        self.assertTrue(
            is_derived_from(constraint_electorstatic_potential,
                            "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_electorstatic_potential,
                            "Fem::ConstraintPython"))
        self.assertTrue(
            is_derived_from(constraint_electorstatic_potential,
                            "Fem::ConstraintElectrostaticPotential"))

        # ConstraintFixed
        constraint_fixed = ObjectsFem.makeConstraintFixed(doc)
        self.assertTrue(
            is_derived_from(constraint_fixed, "App::DocumentObject"))
        self.assertTrue(is_derived_from(constraint_fixed, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_fixed, "Fem::ConstraintFixed"))

        # ConstraintFlowVelocity
        constraint_flow_velocity = ObjectsFem.makeConstraintFlowVelocity(doc)
        self.assertTrue(
            is_derived_from(constraint_flow_velocity, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_flow_velocity, "Fem::ConstraintPython"))
        self.assertTrue(
            is_derived_from(constraint_flow_velocity,
                            "Fem::ConstraintFlowVelocity"))

        # ConstraintFluidBoundary
        constraint_fluid_boundary = ObjectsFem.makeConstraintFluidBoundary(doc)
        self.assertTrue(
            is_derived_from(constraint_fluid_boundary, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_fluid_boundary, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_fluid_boundary,
                            "Fem::ConstraintFluidBoundary"))

        # ConstraintForce
        constraint_force = ObjectsFem.makeConstraintForce(doc)
        self.assertTrue(
            is_derived_from(constraint_force, "App::DocumentObject"))
        self.assertTrue(is_derived_from(constraint_force, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_force, "Fem::ConstraintForce"))

        # ConstraintGear
        constraint_gear = ObjectsFem.makeConstraintGear(doc)
        self.assertTrue(is_derived_from(constraint_gear,
                                        "App::DocumentObject"))
        self.assertTrue(is_derived_from(constraint_gear, "Fem::Constraint"))
        self.assertTrue(is_derived_from(constraint_gear,
                                        "Fem::ConstraintGear"))

        # ConstraintHeatflux
        constraint_heat_flux = ObjectsFem.makeConstraintHeatflux(doc)
        self.assertTrue(
            is_derived_from(constraint_heat_flux, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_heat_flux, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_heat_flux, "Fem::ConstraintHeatflux"))

        # ConstraintInitialFlowVelocity
        constraint_initial_flow_velocity = ObjectsFem.makeConstraintInitialFlowVelocity(
            doc)
        self.assertTrue(
            is_derived_from(constraint_initial_flow_velocity,
                            "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_initial_flow_velocity,
                            "Fem::ConstraintPython"))
        self.assertTrue(
            is_derived_from(constraint_initial_flow_velocity,
                            "Fem::ConstraintInitialFlowVelocity"))

        # ConstraintInitialTemperature
        constraint_initial_temperature = ObjectsFem.makeConstraintInitialTemperature(
            doc)
        self.assertTrue(
            is_derived_from(constraint_initial_temperature,
                            "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_initial_temperature, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_initial_temperature,
                            "Fem::ConstraintInitialTemperature"))

        # ConstraintPlaneRotation
        constraint_plane_rotation = ObjectsFem.makeConstraintPlaneRotation(doc)
        self.assertTrue(
            is_derived_from(constraint_plane_rotation, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_plane_rotation, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_plane_rotation,
                            "Fem::ConstraintPlaneRotation"))

        # ConstraintPressure
        constraint_pressure = ObjectsFem.makeConstraintPressure(doc)
        self.assertTrue(
            is_derived_from(constraint_pressure, "App::DocumentObject"))
        self.assertTrue(is_derived_from(constraint_pressure,
                                        "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_pressure, "Fem::ConstraintPressure"))

        # ConstraintPulley
        constraint_pulley = ObjectsFem.makeConstraintPulley(doc)
        self.assertTrue(
            is_derived_from(constraint_pulley, "App::DocumentObject"))
        self.assertTrue(is_derived_from(constraint_pulley, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_pulley, "Fem::ConstraintPulley"))

        # ConstraintSelfWeight
        constraint_self_weight = ObjectsFem.makeConstraintSelfWeight(doc)
        self.assertTrue(
            is_derived_from(constraint_self_weight, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_self_weight, "Fem::ConstraintPython"))
        self.assertTrue(
            is_derived_from(constraint_self_weight,
                            "Fem::ConstraintSelfWeight"))

        # ConstraintTemperature
        constraint_temperature = ObjectsFem.makeConstraintTemperature(doc)
        self.assertTrue(
            is_derived_from(constraint_temperature, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_temperature, "Fem::Constraint"))
        self.assertTrue(
            is_derived_from(constraint_temperature,
                            "Fem::ConstraintTemperature"))

        # ConstraintTransform
        constraint_transform = ObjectsFem.makeConstraintTransform(doc)
        self.assertTrue(
            is_derived_from(constraint_transform, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(constraint_transform, "Fem::ConstraintTransform"))

        # FemElementFluid1D
        fluid1d = ObjectsFem.makeElementFluid1D(doc)
        self.assertTrue(is_derived_from(fluid1d, "App::DocumentObject"))
        self.assertTrue(is_derived_from(fluid1d, "Fem::FeaturePython"))
        self.assertTrue(is_derived_from(fluid1d, "Fem::FemElementFluid1D"))

        # FemElementGeometry1D
        geometry1d = ObjectsFem.makeElementGeometry1D(doc)
        self.assertTrue(is_derived_from(geometry1d, "App::DocumentObject"))
        self.assertTrue(is_derived_from(geometry1d, "Fem::FeaturePython"))
        self.assertTrue(
            is_derived_from(geometry1d, "Fem::FemElementGeometry1D"))

        # FemElementGeometry2D
        geometry2d = ObjectsFem.makeElementGeometry2D(doc)
        self.assertTrue(is_derived_from(geometry2d, "App::DocumentObject"))
        self.assertTrue(is_derived_from(geometry2d, "Fem::FeaturePython"))
        self.assertTrue(
            is_derived_from(geometry2d, "Fem::FemElementGeometry2D"))

        # FemElementRotation1D
        rotation1d = ObjectsFem.makeElementRotation1D(doc)
        self.assertTrue(is_derived_from(rotation1d, "App::DocumentObject"))
        self.assertTrue(is_derived_from(rotation1d, "Fem::FeaturePython"))
        self.assertTrue(
            is_derived_from(rotation1d, "Fem::FemElementRotation1D"))

        # Material Fluid
        material_fluid = ObjectsFem.makeMaterialFluid(doc)
        self.assertTrue(is_derived_from(material_fluid, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(material_fluid, "App::MaterialObjectPython"))
        self.assertTrue(is_derived_from(material_fluid, "Fem::Material"))

        # Material Solid
        material_solid = ObjectsFem.makeMaterialSolid(doc)
        self.assertTrue(is_derived_from(material_solid, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(material_solid, "App::MaterialObjectPython"))
        self.assertTrue(is_derived_from(material_solid, "Fem::Material"))

        # MaterialMechanicalNonlinear
        material_nonlinear = ObjectsFem.makeMaterialMechanicalNonlinear(
            doc, material_solid)
        self.assertTrue(
            is_derived_from(material_nonlinear, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(material_nonlinear, "Fem::FeaturePython"))
        self.assertTrue(
            is_derived_from(material_nonlinear,
                            "Fem::MaterialMechanicalNonlinear"))

        # MaterialReinforced
        material_reinforced = ObjectsFem.makeMaterialReinforced(doc)
        self.assertTrue(
            is_derived_from(material_reinforced, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(material_reinforced, "App::MaterialObjectPython"))
        self.assertTrue(
            is_derived_from(material_reinforced, "Fem::MaterialReinforced"))

        # FemMeshGmsh
        mesh_gmsh = ObjectsFem.makeMeshGmsh(doc)
        self.assertTrue(is_derived_from(mesh_gmsh, "App::DocumentObject"))
        self.assertTrue(is_derived_from(mesh_gmsh, "Fem::FemMeshObjectPython"))
        self.assertTrue(is_derived_from(mesh_gmsh, "Fem::FemMeshGmsh"))

        # FemMeshBoundaryLayer
        mesh_boundarylayer = ObjectsFem.makeMeshBoundaryLayer(doc, mesh_gmsh)
        self.assertTrue(
            is_derived_from(mesh_boundarylayer, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(mesh_boundarylayer, "Fem::FeaturePython"))
        self.assertTrue(
            is_derived_from(mesh_boundarylayer, "Fem::FemMeshBoundaryLayer"))

        # FemMeshGroup
        mesh_group = ObjectsFem.makeMeshGroup(doc, mesh_gmsh)
        self.assertTrue(is_derived_from(mesh_group, "App::DocumentObject"))
        self.assertTrue(is_derived_from(mesh_group, "Fem::FeaturePython"))
        self.assertTrue(is_derived_from(mesh_group, "Fem::FemMeshGroup"))

        # FemMeshRegion
        mesh_region = ObjectsFem.makeMeshRegion(doc, mesh_gmsh)
        self.assertTrue(is_derived_from(mesh_region, "App::DocumentObject"))
        self.assertTrue(is_derived_from(mesh_region, "Fem::FeaturePython"))
        self.assertTrue(is_derived_from(mesh_region, "Fem::FemMeshRegion"))

        # FemMeshShapeNetgenObject
        mesh_netgen = ObjectsFem.makeMeshNetgen(doc)
        self.assertTrue(is_derived_from(mesh_netgen, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(mesh_netgen, "Fem::FemMeshShapeNetgenObject"))

        # FemMeshResult
        mesh_result = ObjectsFem.makeMeshResult(doc)
        self.assertTrue(is_derived_from(mesh_result, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(mesh_result, "Fem::FemMeshObjectPython"))
        self.assertTrue(is_derived_from(mesh_result, "Fem::FemMeshResult"))

        # FemResultMechanical
        result_mechanical = ObjectsFem.makeResultMechanical(doc)
        self.assertTrue(
            is_derived_from(result_mechanical, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(result_mechanical, "Fem::FemResultObjectPython"))
        self.assertTrue(
            is_derived_from(result_mechanical, "Fem::FemResultMechanical"))

        # FemSolverCalculixCcxTools
        solver_ccxtools = ObjectsFem.makeSolverCalculixCcxTools(doc)
        self.assertTrue(is_derived_from(solver_ccxtools,
                                        "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(solver_ccxtools, "Fem::FemSolverObject"))
        self.assertTrue(
            is_derived_from(solver_ccxtools, "Fem::FemSolverObjectPython"))
        self.assertTrue(
            is_derived_from(solver_ccxtools, "Fem::FemSolverCalculixCcxTools"))

        # FemSolverObjectCalculix
        solver_calculix = ObjectsFem.makeSolverCalculix(doc)
        self.assertTrue(is_derived_from(solver_calculix,
                                        "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(solver_calculix, "Fem::FemSolverObject"))
        self.assertTrue(
            is_derived_from(solver_calculix, "Fem::FemSolverObjectPython"))
        self.assertTrue(
            is_derived_from(solver_calculix, "Fem::FemSolverObjectCalculix"))

        # FemSolverObjectElmer
        solver_elmer = ObjectsFem.makeSolverElmer(doc)
        self.assertTrue(is_derived_from(solver_elmer, "App::DocumentObject"))
        self.assertTrue(is_derived_from(solver_elmer, "Fem::FemSolverObject"))
        self.assertTrue(
            is_derived_from(solver_elmer, "Fem::FemSolverObjectPython"))
        self.assertTrue(
            is_derived_from(solver_elmer, "Fem::FemSolverObjectElmer"))

        # FemSolverObjectZ88
        solver_z88 = ObjectsFem.makeSolverZ88(doc)
        self.assertTrue(is_derived_from(solver_z88, "App::DocumentObject"))
        self.assertTrue(is_derived_from(solver_z88, "Fem::FemSolverObject"))
        self.assertTrue(
            is_derived_from(solver_z88, "Fem::FemSolverObjectPython"))
        self.assertTrue(is_derived_from(solver_z88, "Fem::FemSolverObjectZ88"))

        # FemEquationElmerElasticity
        equation_elasticity = ObjectsFem.makeEquationElasticity(
            doc, solver_elmer)
        self.assertTrue(
            is_derived_from(equation_elasticity, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(equation_elasticity, "App::FeaturePython"))
        self.assertTrue(
            is_derived_from(equation_elasticity,
                            "Fem::FemEquationElmerElasticity"))

        # FemEquationElmerElectrostatic
        equation_electrostatic = ObjectsFem.makeEquationElectrostatic(
            doc, solver_elmer)
        self.assertTrue(
            is_derived_from(equation_electrostatic, "App::DocumentObject"))
        self.assertTrue(
            is_derived_from(equation_electrostatic, "App::FeaturePython"))
        self.assertTrue(
            is_derived_from(equation_electrostatic,
                            "Fem::FemEquationElmerElectrostatic"))

        # FemEquationElmerFlow
        equation_flow = ObjectsFem.makeEquationFlow(doc, solver_elmer)
        self.assertTrue(is_derived_from(equation_flow, "App::DocumentObject"))
        self.assertTrue(is_derived_from(equation_flow, "App::FeaturePython"))
        self.assertTrue(
            is_derived_from(equation_flow, "Fem::FemEquationElmerFlow"))

        # FemEquationElmerFluxsolver
        equation_flux = ObjectsFem.makeEquationFluxsolver(doc, solver_elmer)
        self.assertTrue(is_derived_from(equation_flux, "App::DocumentObject"))
        self.assertTrue(is_derived_from(equation_flux, "App::FeaturePython"))
        self.assertTrue(
            is_derived_from(equation_flux, "Fem::FemEquationElmerFluxsolver"))

        # FemEquationElmerHeat
        equation_heat = ObjectsFem.makeEquationHeat(doc, solver_elmer)
        self.assertTrue(is_derived_from(equation_heat, "App::DocumentObject"))
        self.assertTrue(is_derived_from(equation_heat, "App::FeaturePython"))
        self.assertTrue(
            is_derived_from(equation_heat, "Fem::FemEquationElmerHeat"))

        fcc_print("doc objects count: {}, method: {}".format(
            len(doc.Objects),
            sys._getframe().f_code.co_name))
        # TODO: vtk post objs, thus 5 obj less than test_femobjects_make
        self.assertEqual(len(doc.Objects), testtools.get_defmake_count(False))
Пример #11
0
 def _partOfModel(self, obj):
     for t in self._WHITELIST:
         if FemUtils.is_derived_from(obj, t):
             return True
     return False