src/PartSet/PartSet_ExternalPointsMgr.cpp

   1 // Copyright (C) 2017-2021  CEA/DEN, EDF R&D
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 // File:        PartSet_ExternalPointsMgr.cpp
  21 // Created:     26 April 2017
  22 // Author:      Vitaly SMETANNIKOV
  23
  24 #include "PartSet_ExternalPointsMgr.h"
  25 #include "PartSet_CenterPrs.h"
  26 #include "PartSet_Tools.h"
  27
  28 #include <ModelAPI_Tools.h>
  29 #include <ModelAPI_ResultField.h>
  30 #include <ModelAPI_ResultGroup.h>
  31
  32 #include <ModuleBase_IWorkshop.h>
  33 #include <ModuleBase_ViewerPrs.h>
  34
  35 #include <GeomAPI_Circ.h>
  36 #include <GeomAPI_Edge.h>
  37 #include <GeomAPI_Ellipse.h>
  38 #include <GeomAPI_Pnt.h>
  39 #include <GeomAPI_ShapeExplorer.h>
  40
  41 #include <XGUI_Tools.h>
  42 #include <XGUI_Displayer.h>
  43 #include <XGUI_Workshop.h>
  44 #include <XGUI_SelectionActivate.h>
  45 #include <XGUI_SelectionMgr.h>
  46
  47 #ifdef WIN32
  48 #pragma warning(disable : 4456) // for nested foreach
  49 #endif
  50
  51 PartSet_ExternalPointsMgr::PartSet_ExternalPointsMgr(ModuleBase_IWorkshop* theWorkshop,
  52                                                      const CompositeFeaturePtr& theSketch)
  53   : QObject(theWorkshop), myWorkshop(theWorkshop), mySketch(theSketch)
  54 {
  55   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
  56   XGUI_Displayer* aDisplayer = aWorkshop->displayer();
  57   connect(aDisplayer, SIGNAL(objectDisplayed(ObjectPtr, AISObjectPtr)),
  58                       SLOT(onDisplayObject(ObjectPtr, AISObjectPtr)));
  59
  60   connect(aDisplayer, SIGNAL(beforeObjectErase(ObjectPtr, AISObjectPtr)),
  61                       SLOT(onEraseObject(ObjectPtr, AISObjectPtr)));
  62
  63   updateCenterPresentations();
  64 }
  65
  66
  67 PartSet_ExternalPointsMgr::~PartSet_ExternalPointsMgr()
  68 {
  69   if (myPresentations.isEmpty())
  70     return;
  71   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
  72   if (!aWorkshop)
  73     return;
  74
  75   XGUI_Displayer* aDisplayer = aWorkshop->displayer();
  76   QMapIterator<ObjectPtr, ListOfAIS> aIt(myPresentations);
  77   while (aIt.hasNext()) {
  78     aIt.next();
  79     ListOfAIS aAISList = aIt.value();
  80     foreach (AISObjectPtr aAIS, aAISList) {
  81       aDisplayer->eraseAIS(aAIS, false);
  82     }
  83   }
  84 }
  85
  86
  87 //******************************************************
  88 QList<std::shared_ptr<ModuleBase_ViewerPrs>> PartSet_ExternalPointsMgr::findCircularEdgesInPlane()
  89 {
  90   QList<std::shared_ptr<ModuleBase_ViewerPrs>> aResult;
  91   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
  92   XGUI_Displayer* aDisplayer = aWorkshop->displayer();
  93   QObjectPtrList aDispObjects = aDisplayer->displayedObjects();
  94
  95   std::shared_ptr<GeomAPI_Pln> aPlane = plane();
  96   foreach(ObjectPtr aObj, aDispObjects) {
  97     if (myPresentations.contains(aObj))
  98       continue;
  99
 100     // Do not process objects of the current sketch
 101     if (isSketchObject(aObj))
 102       continue;
 103
 104     ResultPtr aResObj = std::dynamic_pointer_cast<ModelAPI_Result>(aObj);
 105
 106     // Do not use Fields and groups in selection in sketcher
 107     if (!aResObj.get() ||
 108       (aResObj->groupName() == ModelAPI_ResultField::group()) ||
 109       (aResObj->groupName() == ModelAPI_ResultGroup::group()))
 110       continue;
 111
 112     if (aResObj.get()) {
 113       GeomShapePtr aShape = aResObj->shape();
 114       if (aShape.get()) {
 115         GeomAPI_ShapeExplorer aExplorer(aShape, GeomAPI_Shape::EDGE);
 116         for(; aExplorer.more(); aExplorer.next()) {
 117           GeomShapePtr aEdgeShape = aExplorer.current();
 118           GeomAPI_Edge anEdge(aEdgeShape);
 119           if ((anEdge.isCircle() || anEdge.isArc() || anEdge.isEllipse()) &&
 120                anEdge.isInPlane(aPlane)) {
 121             bool isContains = false;
 122             // Check that edge is not already used.
 123             // It is possible that the same edge will be taken from different faces
 124             foreach(std::shared_ptr<ModuleBase_ViewerPrs> aPrs, aResult) {
 125               GeomAPI_Edge aUsedEdge(aPrs->shape());
 126               if (aUsedEdge.isEqual(aEdgeShape)) {
 127                 isContains = true;
 128                 break;
 129               }
 130             }
 131             if (!isContains) {
 132               std::shared_ptr<ModuleBase_ViewerPrs>
 133                 aPrs(new ModuleBase_ViewerPrs(aResObj, aEdgeShape));
 134               aResult.append(aPrs);
 135             }
 136           }
 137         }
 138       }
 139     }
 140   }
 141   return aResult;
 142 }
 143
 144
 145 void PartSet_ExternalPointsMgr::updateCenterPresentations()
 146 {
 147   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
 148   // Return if there is no plane defined
 149   if (!plane().get()) {
 150     connect(aWorkshop->selector(), SIGNAL(selectionChanged()),
 151                                    SLOT(onSelectionChanged()));
 152     return;
 153   }
 154   XGUI_Displayer* aDisplayer = aWorkshop->displayer();
 155
 156   QList<std::shared_ptr<ModuleBase_ViewerPrs>> aEdgesPrs = findCircularEdgesInPlane();
 157   foreach(std::shared_ptr<ModuleBase_ViewerPrs> aPrs, aEdgesPrs) {
 158     GeomEdgePtr aEdge(new GeomAPI_Edge(aPrs->shape()));
 159     ListOfAIS aList;
 160     if (aEdge->isArc() || aEdge->isCircle()) {
 161       GeomCirclePtr aCircle = aEdge->circle();
 162       GeomPointPtr aCenter = aCircle->center();
 163       Handle(PartSet_CenterPrs) aCentPrs =
 164         new PartSet_CenterPrs(aPrs->object(), aEdge, aCenter->impl<gp_Pnt>(),
 165                               ModelAPI_AttributeSelection::CIRCLE_CENTER);
 166
 167       AISObjectPtr anAIS(new GeomAPI_AISObject());
 168       anAIS->setImpl(new Handle(AIS_InteractiveObject)(aCentPrs));
 169       aList.append(anAIS);
 170     } else if (aEdge->isEllipse()) {
 171       GeomEllipsePtr aEllipse = aEdge->ellipse();
 172       GeomPointPtr aF1 = aEllipse->firstFocus();
 173       GeomPointPtr aF2 = aEllipse->secondFocus();
 174       Handle(PartSet_CenterPrs) aF1Prs =
 175         new PartSet_CenterPrs(aPrs->object(), aEdge, aF1->impl<gp_Pnt>(),
 176                               ModelAPI_AttributeSelection::ELLIPSE_FIRST_FOCUS);
 177       Handle(PartSet_CenterPrs) aF2Prs =
 178         new PartSet_CenterPrs(aPrs->object(), aEdge, aF2->impl<gp_Pnt>(),
 179                               ModelAPI_AttributeSelection::ELLIPSE_SECOND_FOCUS);
 180
 181       AISObjectPtr anAIS1(new GeomAPI_AISObject());
 182       anAIS1->setImpl(new Handle(AIS_InteractiveObject)(aF1Prs));
 183       aList.append(anAIS1);
 184
 185       AISObjectPtr anAIS2(new GeomAPI_AISObject());
 186       anAIS2->setImpl(new Handle(AIS_InteractiveObject)(aF2Prs));
 187       aList.append(anAIS2);
 188     }
 189     if (myPresentations.contains(aPrs->object()))
 190       myPresentations[aPrs->object()].append(aList);
 191     else
 192       myPresentations[aPrs->object()] = aList;
 193     foreach(AISObjectPtr anAIS, aList) {
 194       aDisplayer->displayAIS(anAIS, false, 0, false);
 195       aWorkshop->selectionActivate()->activateAIS(anAIS->impl<Handle(AIS_InteractiveObject)>(),
 196         TopAbs_VERTEX, false);
 197     }
 198   }
 199 }
 200
 201 std::shared_ptr<GeomAPI_Pln> PartSet_ExternalPointsMgr::plane() const
 202 {
 203   return PartSet_Tools::sketchPlane(mySketch);
 204 }
 205
 206 void PartSet_ExternalPointsMgr::onDisplayObject(ObjectPtr theObj, AISObjectPtr theAIS)
 207 {
 208   updateCenterPresentations();
 209 }
 210
 211 void PartSet_ExternalPointsMgr::onEraseObject(ObjectPtr theObj, AISObjectPtr theAIS)
 212 {
 213   if (myPresentations.contains(theObj)) {
 214     XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
 215     XGUI_Displayer* aDisplayer = aWorkshop->displayer();
 216     ListOfAIS aList = myPresentations[theObj];
 217     foreach(AISObjectPtr aAIS, aList) {
 218       aDisplayer->eraseAIS(aAIS, false);
 219     }
 220     myPresentations.remove(theObj);
 221     aDisplayer->updateViewer();
 222   }
 223 }
 224
 225
 226 bool PartSet_ExternalPointsMgr::isSketchObject(const ObjectPtr& theRes) const
 227 {
 228   FeaturePtr aFeature = ModelAPI_Feature::feature(theRes);
 229   if (!aFeature.get())
 230     return false;
 231   CompositeFeaturePtr aComp = ModelAPI_Tools::compositeOwner(aFeature);
 232   return aComp == mySketch;
 233 }
 234
 235 void PartSet_ExternalPointsMgr::onSelectionChanged()
 236 {
 237   if (plane().get()) {
 238     XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
 239     disconnect(aWorkshop->selector(), SIGNAL(selectionChanged()),
 240                this, SLOT(onSelectionChanged()));
 241     updateCenterPresentations();
 242   }
 243 }