374 lines
14 KiB
Python
374 lines
14 KiB
Python
# ***************************************************************************
|
|
# * Copyright (c) 2015 Przemo Fiszt <przemo@firszt.eu> *
|
|
# * Copyright (c) 2016 Bernd Hahnebach <bernd@bimstatik.org> *
|
|
# * *
|
|
# * This file is part of the FreeCAD CAx development system. *
|
|
# * *
|
|
# * This program is free software; you can redistribute it and/or modify *
|
|
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
# * as published by the Free Software Foundation; either version 2 of *
|
|
# * the License, or (at your option) any later version. *
|
|
# * for detail see the LICENCE text file. *
|
|
# * *
|
|
# * This program is distributed in the hope that it will be useful, *
|
|
# * but WITHOUT ANY WARRANTY; without even the implied warranty of *
|
|
# * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the *
|
|
# * GNU Library General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Library General Public *
|
|
# * License along with this program; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
__title__ = "FreeCAD FEM command base class"
|
|
__author__ = "Przemo Firszt, Bernd Hahnebach"
|
|
__url__ = "https://www.freecadweb.org"
|
|
|
|
## @package manager
|
|
# \ingroup FEM
|
|
# \brief FreeCAD FEM command base class
|
|
|
|
import FreeCAD
|
|
|
|
from femtools.femutils import is_of_type
|
|
|
|
if FreeCAD.GuiUp:
|
|
from PySide import QtCore
|
|
import FreeCADGui
|
|
import FemGui
|
|
|
|
|
|
class CommandManager(object):
|
|
|
|
def __init__(self):
|
|
|
|
self.command = "FEM" + self.__class__.__name__
|
|
self.pixmap = self.command
|
|
self.menuetext = self.__class__.__name__.lstrip("_")
|
|
self.accel = ""
|
|
self.tooltip = "Creates a {}".format(self.menuetext)
|
|
self.resources = None
|
|
|
|
self.is_active = None
|
|
self.do_activated = None
|
|
self.selobj = None
|
|
self.selobj2 = None
|
|
self.active_analysis = None
|
|
|
|
def GetResources(self):
|
|
if self.resources is None:
|
|
self.resources = {
|
|
"Pixmap": self.pixmap,
|
|
"MenuText": QtCore.QT_TRANSLATE_NOOP(self.command, self.menuetext),
|
|
"Accel": self.accel,
|
|
"ToolTip": QtCore.QT_TRANSLATE_NOOP(self.command, self.tooltip)
|
|
}
|
|
return self.resources
|
|
|
|
def IsActive(self):
|
|
if not self.is_active:
|
|
active = False
|
|
elif self.is_active == "always":
|
|
active = True
|
|
elif self.is_active == "with_document":
|
|
active = FreeCADGui.ActiveDocument is not None
|
|
elif self.is_active == "with_analysis":
|
|
active = (
|
|
FemGui.getActiveAnalysis() is not None
|
|
and self.active_analysis_in_active_doc()
|
|
)
|
|
elif self.is_active == "with_results":
|
|
active = (
|
|
FemGui.getActiveAnalysis() is not None
|
|
and self.active_analysis_in_active_doc()
|
|
and (self.results_present() or self.result_mesh_present())
|
|
)
|
|
elif self.is_active == "with_selresult":
|
|
active = (
|
|
# on import of Frd file in a empty document not Analysis will be there
|
|
FreeCADGui.ActiveDocument is not None
|
|
and self.result_selected()
|
|
)
|
|
elif self.is_active == "with_part_feature":
|
|
active = (
|
|
FreeCADGui.ActiveDocument is not None
|
|
and self.part_feature_selected()
|
|
)
|
|
elif self.is_active == "with_femmesh":
|
|
active = (
|
|
FreeCADGui.ActiveDocument is not None
|
|
and self.femmesh_selected()
|
|
)
|
|
elif self.is_active == "with_gmsh_femmesh":
|
|
active = (
|
|
FreeCADGui.ActiveDocument is not None
|
|
and self.gmsh_femmesh_selected()
|
|
)
|
|
elif self.is_active == "with_femmesh_andor_res":
|
|
active = (
|
|
FreeCADGui.ActiveDocument is not None
|
|
and self.with_femmesh_andor_res_selected()
|
|
)
|
|
elif self.is_active == "with_material":
|
|
active = (
|
|
FemGui.getActiveAnalysis() is not None
|
|
and self.active_analysis_in_active_doc()
|
|
and self.material_selected()
|
|
)
|
|
elif self.is_active == "with_material_solid":
|
|
active = (
|
|
FemGui.getActiveAnalysis() is not None
|
|
and self.active_analysis_in_active_doc()
|
|
and self.material_solid_selected()
|
|
)
|
|
elif self.is_active == "with_solver":
|
|
active = (
|
|
FemGui.getActiveAnalysis() is not None
|
|
and self.active_analysis_in_active_doc()
|
|
and self.solver_selected()
|
|
)
|
|
elif self.is_active == "with_solver_elmer":
|
|
active = (
|
|
FemGui.getActiveAnalysis() is not None
|
|
and self.active_analysis_in_active_doc()
|
|
and self.solver_elmer_selected()
|
|
)
|
|
elif self.is_active == "with_analysis_without_solver":
|
|
active = (
|
|
FemGui.getActiveAnalysis() is not None
|
|
and self.active_analysis_in_active_doc()
|
|
and not self.analysis_has_solver()
|
|
)
|
|
return active
|
|
|
|
def Activated(self):
|
|
if self.do_activated == "add_obj_on_gui_noset_edit":
|
|
self.add_obj_on_gui_noset_edit(self.__class__.__name__.lstrip("_"))
|
|
elif self.do_activated == "add_obj_on_gui_set_edit":
|
|
self.add_obj_on_gui_set_edit(self.__class__.__name__.lstrip("_"))
|
|
elif self.do_activated == "add_obj_on_gui_selobj_noset_edit":
|
|
self.add_obj_on_gui_selobj_noset_edit(self.__class__.__name__.lstrip("_"))
|
|
elif self.do_activated == "add_obj_on_gui_selobj_set_edit":
|
|
self.add_obj_on_gui_selobj_set_edit(self.__class__.__name__.lstrip("_"))
|
|
# in all other cases Activated is implemented it the command class
|
|
|
|
def results_present(self):
|
|
results = False
|
|
analysis_members = FemGui.getActiveAnalysis().Group
|
|
for o in analysis_members:
|
|
if o.isDerivedFrom("Fem::FemResultObject"):
|
|
results = True
|
|
return results
|
|
|
|
def result_mesh_present(self):
|
|
result_mesh = False
|
|
analysis_members = FemGui.getActiveAnalysis().Group
|
|
for o in analysis_members:
|
|
if is_of_type(o, "Fem::MeshResult"):
|
|
result_mesh = True
|
|
return result_mesh
|
|
|
|
def result_selected(self):
|
|
sel = FreeCADGui.Selection.getSelection()
|
|
if len(sel) == 1 and sel[0].isDerivedFrom("Fem::FemResultObject"):
|
|
self.selobj = sel[0]
|
|
return True
|
|
return False
|
|
|
|
def part_feature_selected(self):
|
|
sel = FreeCADGui.Selection.getSelection()
|
|
if len(sel) == 1 and sel[0].isDerivedFrom("Part::Feature"):
|
|
self.selobj = sel[0]
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def femmesh_selected(self):
|
|
sel = FreeCADGui.Selection.getSelection()
|
|
if len(sel) == 1 and sel[0].isDerivedFrom("Fem::FemMeshObject"):
|
|
self.selobj = sel[0]
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def gmsh_femmesh_selected(self):
|
|
sel = FreeCADGui.Selection.getSelection()
|
|
if len(sel) == 1 and is_of_type(sel[0], "Fem::FemMeshGmsh"):
|
|
self.selobj = sel[0]
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def material_selected(self):
|
|
sel = FreeCADGui.Selection.getSelection()
|
|
if len(sel) == 1 and sel[0].isDerivedFrom("App::MaterialObjectPython"):
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def material_solid_selected(self):
|
|
sel = FreeCADGui.Selection.getSelection()
|
|
if (
|
|
len(sel) == 1
|
|
and sel[0].isDerivedFrom("App::MaterialObjectPython")
|
|
and hasattr(sel[0], "Category")
|
|
and sel[0].Category == "Solid"
|
|
):
|
|
self.selobj = sel[0]
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def with_femmesh_andor_res_selected(self):
|
|
sel = FreeCADGui.Selection.getSelection()
|
|
if len(sel) == 1 and sel[0].isDerivedFrom("Fem::FemMeshObject"):
|
|
self.selobj = sel[0]
|
|
return True
|
|
elif len(sel) == 2:
|
|
if sel[0].isDerivedFrom("Fem::FemMeshObject"):
|
|
if sel[1].isDerivedFrom("Fem::FemResultObject"):
|
|
self.selobj = sel[0] # mesh
|
|
self.selobj2 = sel[1] # res
|
|
return True
|
|
else:
|
|
return False
|
|
elif sel[1].isDerivedFrom("Fem::FemMeshObject"):
|
|
if sel[0].isDerivedFrom("Fem::FemResultObject"):
|
|
self.selobj = sel[1] # mesh
|
|
self.selobj2 = sel[0] # res
|
|
return True
|
|
else:
|
|
return False
|
|
else:
|
|
return False
|
|
else:
|
|
return False
|
|
|
|
def active_analysis_in_active_doc(self):
|
|
analysis = FemGui.getActiveAnalysis()
|
|
if analysis.Document is FreeCAD.ActiveDocument:
|
|
self.active_analysis = analysis
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def solver_selected(self):
|
|
sel = FreeCADGui.Selection.getSelection()
|
|
if len(sel) == 1 and sel[0].isDerivedFrom("Fem::FemSolverObjectPython"):
|
|
self.selobj = sel[0]
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def solver_elmer_selected(self):
|
|
sel = FreeCADGui.Selection.getSelection()
|
|
if len(sel) == 1 and is_of_type(sel[0], "Fem::SolverElmer"):
|
|
self.selobj = sel[0]
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def analysis_has_solver(self):
|
|
solver = False
|
|
analysis_members = FemGui.getActiveAnalysis().Group
|
|
for o in analysis_members:
|
|
if o.isDerivedFrom("Fem::FemSolverObjectPython"):
|
|
solver = True
|
|
if solver is True:
|
|
return True
|
|
else:
|
|
return False
|
|
|
|
def hide_meshes_show_parts_constraints(self):
|
|
if FreeCAD.GuiUp:
|
|
for acnstrmesh in FemGui.getActiveAnalysis().Group:
|
|
if "Constraint" in acnstrmesh.TypeId:
|
|
acnstrmesh.ViewObject.Visibility = True
|
|
if "Mesh" in acnstrmesh.TypeId:
|
|
# OvG: Hide meshes and show constraints and meshed part
|
|
# e.g. on purging results
|
|
acnstrmesh.ViewObject.Visibility = False
|
|
|
|
# ****************************************************************************************
|
|
# methods to add the objects to the document in FreeCADGui mode
|
|
|
|
def add_obj_on_gui_set_edit(self, objtype):
|
|
FreeCAD.ActiveDocument.openTransaction(
|
|
"Create Fem{}"
|
|
.format(objtype)
|
|
)
|
|
FreeCADGui.addModule(
|
|
"ObjectsFem"
|
|
)
|
|
FreeCADGui.addModule(
|
|
"FemGui"
|
|
)
|
|
FreeCADGui.doCommand(
|
|
"FemGui.getActiveAnalysis().addObject(ObjectsFem."
|
|
"make{}(FreeCAD.ActiveDocument))"
|
|
.format(objtype)
|
|
)
|
|
# no other obj should be selected if we go in task panel
|
|
FreeCADGui.Selection.clearSelection()
|
|
FreeCADGui.doCommand(
|
|
"FreeCADGui.ActiveDocument.setEdit(FreeCAD.ActiveDocument.ActiveObject.Name)"
|
|
)
|
|
FreeCAD.ActiveDocument.recompute()
|
|
|
|
def add_obj_on_gui_noset_edit(self, objtype):
|
|
FreeCAD.ActiveDocument.openTransaction(
|
|
"Create Fem{}"
|
|
.format(objtype)
|
|
)
|
|
FreeCADGui.addModule(
|
|
"ObjectsFem"
|
|
)
|
|
FreeCADGui.addModule(
|
|
"FemGui"
|
|
)
|
|
FreeCADGui.doCommand(
|
|
"FemGui.getActiveAnalysis().addObject(ObjectsFem."
|
|
"make{}(FreeCAD.ActiveDocument))"
|
|
.format(objtype)
|
|
)
|
|
# FreeCAD.ActiveDocument.commitTransaction() # solver command class had this line
|
|
# no clear selection is done
|
|
FreeCAD.ActiveDocument.recompute()
|
|
|
|
def add_obj_on_gui_selobj_set_edit(self, objtype):
|
|
FreeCAD.ActiveDocument.openTransaction(
|
|
"Create Fem{}"
|
|
.format(objtype)
|
|
)
|
|
FreeCADGui.addModule(
|
|
"ObjectsFem"
|
|
)
|
|
FreeCADGui.doCommand(
|
|
"ObjectsFem.make{}("
|
|
"FreeCAD.ActiveDocument, FreeCAD.ActiveDocument.{})"
|
|
.format(objtype, self.selobj.Name)
|
|
)
|
|
FreeCADGui.Selection.clearSelection()
|
|
FreeCADGui.doCommand(
|
|
"FreeCADGui.ActiveDocument.setEdit(FreeCAD.ActiveDocument.ActiveObject.Name)"
|
|
)
|
|
FreeCAD.ActiveDocument.recompute()
|
|
|
|
def add_obj_on_gui_selobj_noset_edit(self, objtype):
|
|
FreeCAD.ActiveDocument.openTransaction(
|
|
"Create Fem{}"
|
|
.format(objtype)
|
|
)
|
|
FreeCADGui.addModule(
|
|
"ObjectsFem"
|
|
)
|
|
FreeCADGui.doCommand(
|
|
"ObjectsFem.make{}("
|
|
"FreeCAD.ActiveDocument, FreeCAD.ActiveDocument.{})"
|
|
.format(objtype, self.selobj.Name)
|
|
)
|
|
FreeCADGui.Selection.clearSelection()
|
|
FreeCAD.ActiveDocument.recompute()
|