# *************************************************************************** # * * # * Copyright (c) 2016 - Bernd Hahnebach * # * * # * 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__ = "_ViewProviderFemMeshRegion" __author__ = "Bernd Hahnebach" __url__ = "http://www.freecadweb.org" ## @package ViewProviderFemMeshRegion # \ingroup FEM import FreeCAD import FreeCADGui # for the panel import femmesh.meshtools as FemMeshTools from PySide import QtCore from PySide import QtGui class _ViewProviderFemMeshRegion: "A View Provider for the FemMeshRegion object" def __init__(self, vobj): vobj.Proxy = self def getIcon(self): return ":/icons/fem-femmesh-region.svg" def attach(self, vobj): from pivy import coin self.ViewObject = vobj self.Object = vobj.Object self.standard = coin.SoGroup() vobj.addDisplayMode(self.standard, "Standard") def getDisplayModes(self, obj): return ["Standard"] def getDefaultDisplayMode(self): return "Standard" def updateData(self, obj, prop): return def onChanged(self, vobj, prop): return def setEdit(self, vobj, mode=0): # hide all meshes for o in FreeCAD.ActiveDocument.Objects: if o.isDerivedFrom("Fem::FemMeshObject"): o.ViewObject.hide() # show task panel taskd = _TaskPanelFemMeshRegion(self.Object) taskd.obj = vobj.Object FreeCADGui.Control.showDialog(taskd) return True def unsetEdit(self, vobj, mode=0): FreeCADGui.Control.closeDialog() return def doubleClicked(self, vobj): guidoc = FreeCADGui.getDocument(vobj.Object.Document) # check if another VP is in edit mode, https://forum.freecadweb.org/viewtopic.php?t=13077#p104702 if not guidoc.getInEdit(): guidoc.setEdit(vobj.Object.Name) else: from PySide.QtGui import QMessageBox message = 'Active Task Dialog found! Please close this one before open a new one!' QMessageBox.critical(None, "Error in tree view", message) FreeCAD.Console.PrintError(message + '\n') return True def __getstate__(self): return None def __setstate__(self, state): return None class _TaskPanelFemMeshRegion: '''The TaskPanel for editing References property of FemMeshRegion objects''' def __init__(self, obj): FreeCADGui.Selection.clearSelection() self.sel_server = None self.obj = obj self.selection_mode_solid = False self.selection_mode_std_print_message = "Select Faces, Edges and Vertices by single click on them to add them to the list." self.selection_mode_solid_print_message = "Select Solids by single click on a Face or Edge which belongs to the Solid, to add the Solid to the list." self.obj_notvisible = [] self.form = FreeCADGui.PySideUic.loadUi(FreeCAD.getHomePath() + "Mod/Fem/Resources/ui/MeshRegion.ui") QtCore.QObject.connect(self.form.if_elelen, QtCore.SIGNAL("valueChanged(Base::Quantity)"), self.elelen_changed) QtCore.QObject.connect(self.form.rb_standard, QtCore.SIGNAL("toggled(bool)"), self.choose_selection_mode_standard) QtCore.QObject.connect(self.form.rb_solid, QtCore.SIGNAL("toggled(bool)"), self.choose_selection_mode_solid) QtCore.QObject.connect(self.form.pushButton_Reference, QtCore.SIGNAL("clicked()"), self.add_references) self.form.list_References.itemSelectionChanged.connect(self.select_clicked_reference_shape) self.form.list_References.setContextMenuPolicy(QtCore.Qt.CustomContextMenu) self.form.list_References.connect(self.form.list_References, QtCore.SIGNAL("customContextMenuRequested(QPoint)"), self.references_list_right_clicked) self.get_meshregion_props() self.update() def accept(self): self.setback_listobj_visibility() self.set_meshregion_props() if self.sel_server: FreeCADGui.Selection.removeObserver(self.sel_server) FreeCADGui.ActiveDocument.resetEdit() FreeCAD.ActiveDocument.recompute() return True def reject(self): self.setback_listobj_visibility() if self.sel_server: FreeCADGui.Selection.removeObserver(self.sel_server) FreeCADGui.ActiveDocument.resetEdit() return True def get_meshregion_props(self): self.elelen = self.obj.CharacteristicLength self.references = [] if self.obj.References: self.tuplereferences = self.obj.References self.get_references() def set_meshregion_props(self): self.obj.References = self.references self.obj.CharacteristicLength = self.elelen def update(self): 'fills the widgets' self.form.if_elelen.setText(self.elelen.UserString) self.rebuild_list_References() def elelen_changed(self, base_quantity_value): self.elelen = base_quantity_value def choose_selection_mode_standard(self, state): self.selection_mode_solid = not state if self.sel_server and not self.selection_mode_solid: print(self.selection_mode_std_print_message) def choose_selection_mode_solid(self, state): self.selection_mode_solid = state if self.sel_server and self.selection_mode_solid: print(self.selection_mode_solid_print_message) def get_references(self): for ref in self.tuplereferences: for elem in ref[1]: self.references.append((ref[0], elem)) def references_list_right_clicked(self, QPos): self.form.contextMenu = QtGui.QMenu() menu_item_remove_selected = self.form.contextMenu.addAction("Remove selected reference") menu_item_remove_all = self.form.contextMenu.addAction("Remove all references") if not self.references: menu_item_remove_selected.setDisabled(True) menu_item_remove_all.setDisabled(True) self.form.connect(menu_item_remove_selected, QtCore.SIGNAL("triggered()"), self.remove_selected_reference) self.form.connect(menu_item_remove_all, QtCore.SIGNAL("triggered()"), self.remove_all_references) parentPosition = self.form.list_References.mapToGlobal(QtCore.QPoint(0, 0)) self.form.contextMenu.move(parentPosition + QPos) self.form.contextMenu.show() def remove_selected_reference(self): if not self.references: return currentItemName = str(self.form.list_References.currentItem().text()) currentRow = self.form.list_References.currentRow() for ref in self.references: refname_to_compare_listentry = ref[0].Name + ':' + ref[1] if refname_to_compare_listentry == currentItemName: self.references.remove(ref) self.rebuild_list_References(currentRow) def remove_all_references(self): self.references = [] self.rebuild_list_References() def add_references(self): '''Called if Button add_reference is triggered''' # in constraints EditTaskPanel the selection is active as soon as the taskpanel is open # here the addReference button EditTaskPanel has to be triggered to start selection mode self.setback_listobj_visibility() FreeCADGui.Selection.clearSelection() # start SelectionObserver and parse the function to add the References to the widget if self.selection_mode_solid: # print message on button click print_message = self.selection_mode_solid_print_message else: print_message = self.selection_mode_std_print_message if not self.sel_server: # if we do not check, we would start a new SelectionObserver on every click on addReference button # but close only one SelectionObserver on leaving the task panel from . import FemSelectionObserver self.sel_server = FemSelectionObserver.FemSelectionObserver(self.selectionParser, print_message) def selectionParser(self, selection): print('selection: ', selection[0].Shape.ShapeType, ' ', selection[0].Name, ' ', selection[1]) if hasattr(selection[0], "Shape") and selection[1]: elt = selection[0].Shape.getElement(selection[1]) if self.selection_mode_solid: # in solid selection mode use edges and faces for selection of a solid solid_to_add = None if elt.ShapeType == 'Edge': found_edge = False for i, s in enumerate(selection[0].Shape.Solids): for e in s.Edges: if elt.isSame(e): if not found_edge: solid_to_add = str(i + 1) else: FreeCAD.Console.PrintMessage('Edge belongs to more than one solid\n') solid_to_add = None found_edge = True elif elt.ShapeType == 'Face': found_face = False for i, s in enumerate(selection[0].Shape.Solids): for e in s.Faces: if elt.isSame(e): if not found_face: solid_to_add = str(i + 1) else: FreeCAD.Console.PrintMessage('Face belongs to more than one solid\n') solid_to_add = None found_edge = True if solid_to_add: selection = (selection[0], 'Solid' + solid_to_add) print('selection element changed to Solid: ', selection[0].Shape.ShapeType, ' ', selection[0].Name, ' ', selection[1]) else: return if selection not in self.references: self.references.append(selection) self.rebuild_list_References() else: FreeCAD.Console.PrintMessage(selection[0].Name + ' --> ' + selection[1] + ' is in reference list already!\n') def rebuild_list_References(self, current_row=0): self.form.list_References.clear() items = [] for ref in self.references: item_name = ref[0].Name + ':' + ref[1] items.append(item_name) for listItemName in sorted(items): self.form.list_References.addItem(listItemName) if current_row > self.form.list_References.count() - 1: # first row is 0 current_row = self.form.list_References.count() - 1 if self.form.list_References.count() > 0: self.form.list_References.setCurrentItem(self.form.list_References.item(current_row)) def select_clicked_reference_shape(self): self.setback_listobj_visibility() if self.sel_server: FreeCADGui.Selection.removeObserver(self.sel_server) self.sel_server = None if not self.sel_server: if not self.references: return currentItemName = str(self.form.list_References.currentItem().text()) for ref in self.references: refname_to_compare_listentry = ref[0].Name + ':' + ref[1] if refname_to_compare_listentry == currentItemName: # print('found: shape: ' + ref[0].Name + ' element: ' + ref[1]) if not ref[0].ViewObject.Visibility: self.obj_notvisible.append(ref[0]) ref[0].ViewObject.Visibility = True FreeCADGui.Selection.clearSelection() if ref[1].startswith('Solid') and (ref[0].Shape.ShapeType == 'Compound' or ref[0].Shape.ShapeType == 'CompSolid'): # selection of Solids of Compounds or CompSolids is not possible, because a Solid is no Subelement # since only Subelements can be selected, we gone select all Faces of such an Solids solid = FemMeshTools.get_element(ref[0], ref[1]) # the method getElement(element) does not return Solid elements if not solid: return faces = [] for fs in solid.Faces: # find these faces in ref[0] for i, fref in enumerate(ref[0].Shape.Faces): if fs.isSame(fref): fref_elstring = 'Face' + str(i + 1) if fref_elstring not in faces: faces.append(fref_elstring) for f in faces: FreeCADGui.Selection.addSelection(ref[0], f) else: # Selection of all other element types is supported FreeCADGui.Selection.addSelection(ref[0], ref[1]) def setback_listobj_visibility(self): '''set back Visibility of the list objects ''' FreeCADGui.Selection.clearSelection() for obj in self.obj_notvisible: obj.ViewObject.Visibility = False self.obj_notvisible = []