src/PartSet/PartSet_ExternalPointsMgr.cpp

   1 // Copyright (C) 2017  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
  18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
  19 //
  20
  21 // File:        PartSet_ExternalPointsMgr.cpp
  22 // Created:     26 April 2017
  23 // Author:      Vitaly SMETANNIKOV
  24
  25 #include "PartSet_ExternalPointsMgr.h"
  26 #include "PartSet_CenterPrs.h"
  27 #include "PartSet_Tools.h"
  28
  29 #include <ModelAPI_Tools.h>
  30
  31 #include <ModuleBase_IWorkshop.h>
  32 #include <ModuleBase_ViewerPrs.h>
  33
  34 #include <GeomAPI_Circ.h>
  35 #include <GeomAPI_Edge.h>
  36 #include <GeomAPI_Ellipse.h>
  37 #include <GeomAPI_Pnt.h>
  38 #include <GeomAPI_ShapeExplorer.h>
  39
  40 #include <XGUI_Tools.h>
  41 #include <XGUI_Displayer.h>
  42 #include <XGUI_Workshop.h>
  43 #include <XGUI_SelectionActivate.h>
  44 #include <XGUI_SelectionMgr.h>
  45
  46 PartSet_ExternalPointsMgr::PartSet_ExternalPointsMgr(ModuleBase_IWorkshop* theWorkshop,
  47                                                      const CompositeFeaturePtr& theSketch)
  48   : QObject(theWorkshop), myWorkshop(theWorkshop), mySketch(theSketch)
  49 {
  50   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
  51   XGUI_Displayer* aDisplayer = aWorkshop->displayer();
  52   connect(aDisplayer, SIGNAL(objectDisplayed(ObjectPtr, AISObjectPtr)),
  53                       SLOT(onDisplayObject(ObjectPtr, AISObjectPtr)));
  54
  55   connect(aDisplayer, SIGNAL(beforeObjectErase(ObjectPtr, AISObjectPtr)),
  56                       SLOT(onEraseObject(ObjectPtr, AISObjectPtr)));
  57
  58   updateCenterPresentations();
  59 }
  60
  61
  62 PartSet_ExternalPointsMgr::~PartSet_ExternalPointsMgr()
  63 {
  64   if (myPresentations.isEmpty())
  65     return;
  66   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
  67   if (!aWorkshop)
  68     return;
  69
  70   XGUI_Displayer* aDisplayer = aWorkshop->displayer();
  71   QMapIterator<ObjectPtr, ListOfAIS> aIt(myPresentations);
  72   while (aIt.hasNext()) {
  73     aIt.next();
  74     ListOfAIS aAISList = aIt.value();
  75     foreach (AISObjectPtr aAIS, aAISList) {
  76       aDisplayer->eraseAIS(aAIS, false);
  77     }
  78   }
  79 }
  80
  81
  82 //******************************************************
  83 QList<std::shared_ptr<ModuleBase_ViewerPrs>> PartSet_ExternalPointsMgr::findCircularEdgesInPlane()
  84 {
  85   QList<std::shared_ptr<ModuleBase_ViewerPrs>> aResult;
  86   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
  87   XGUI_Displayer* aDisplayer = aWorkshop->displayer();
  88   QObjectPtrList aDispObjects = aDisplayer->displayedObjects();
  89
  90   std::shared_ptr<GeomAPI_Pln> aPlane = plane();
  91   foreach(ObjectPtr aObj, aDispObjects) {
  92     if (myPresentations.contains(aObj))
  93       continue;
  94
  95     // Do not process objects of the current sketch
  96     if (isSketchObject(aObj))
  97       continue;
  98
  99     ResultPtr aResObj = std::dynamic_pointer_cast<ModelAPI_Result>(aObj);
 100     if (aResObj.get()) {
 101       GeomShapePtr aShape = aResObj->shape();
 102       if (aShape.get()) {
 103         GeomAPI_ShapeExplorer aExplorer(aShape, GeomAPI_Shape::EDGE);
 104         for(; aExplorer.more(); aExplorer.next()) {
 105           GeomShapePtr aEdgeShape = aExplorer.current();
 106           GeomAPI_Edge anEdge(aEdgeShape);
 107           if ((anEdge.isCircle() || anEdge.isArc() || anEdge.isEllipse()) &&
 108                anEdge.isInPlane(aPlane)) {
 109             bool isContains = false;
 110             // Check that edge is not already used.
 111             // It is possible that the same edge will be taken from different faces
 112             foreach(std::shared_ptr<ModuleBase_ViewerPrs> aPrs, aResult) {
 113               GeomAPI_Edge aUsedEdge(aPrs->shape());
 114               if (aUsedEdge.isEqual(aEdgeShape)) {
 115                 isContains = true;
 116                 break;
 117               }
 118             }
 119             if (!isContains) {
 120               std::shared_ptr<ModuleBase_ViewerPrs>
 121                 aPrs(new ModuleBase_ViewerPrs(aResObj, aEdgeShape));
 122               aResult.append(aPrs);
 123             }
 124           }
 125         }
 126       }
 127     }
 128   }
 129   return aResult;
 130 }
 131
 132
 133 void PartSet_ExternalPointsMgr::updateCenterPresentations()
 134 {
 135   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
 136   // Return if there is no plane defined
 137   if (!plane().get()) {
 138     connect(aWorkshop->selector(), SIGNAL(selectionChanged()),
 139                                    SLOT(onSelectionChanged()));
 140     return;
 141   }
 142   XGUI_Displayer* aDisplayer = aWorkshop->displayer();
 143
 144   QList<std::shared_ptr<ModuleBase_ViewerPrs>> aEdgesPrs = findCircularEdgesInPlane();
 145   foreach(std::shared_ptr<ModuleBase_ViewerPrs> aPrs, aEdgesPrs) {
 146     GeomEdgePtr aEdge(new GeomAPI_Edge(aPrs->shape()));
 147     ListOfAIS aList;
 148     if (aEdge->isArc() || aEdge->isCircle()) {
 149       GeomCirclePtr aCircle = aEdge->circle();
 150       GeomPointPtr aCenter = aCircle->center();
 151       Handle(PartSet_CenterPrs) aCentPrs =
 152         new PartSet_CenterPrs(aPrs->object(), aEdge, aCenter->impl<gp_Pnt>(),
 153                               ModelAPI_AttributeSelection::CIRCLE_CENTER);
 154
 155       AISObjectPtr anAIS(new GeomAPI_AISObject());
 156       anAIS->setImpl(new Handle(AIS_InteractiveObject)(aCentPrs));
 157       aList.append(anAIS);
 158     } else if (aEdge->isEllipse()) {
 159       GeomEllipsePtr aEllipse = aEdge->ellipse();
 160       GeomPointPtr aF1 = aEllipse->firstFocus();
 161       GeomPointPtr aF2 = aEllipse->secondFocus();
 162       Handle(PartSet_CenterPrs) aF1Prs =
 163         new PartSet_CenterPrs(aPrs->object(), aEdge, aF1->impl<gp_Pnt>(),
 164                               ModelAPI_AttributeSelection::ELLIPSE_FIRST_FOCUS);
 165       Handle(PartSet_CenterPrs) aF2Prs =
 166         new PartSet_CenterPrs(aPrs->object(), aEdge, aF2->impl<gp_Pnt>(),
 167                               ModelAPI_AttributeSelection::ELLIPSE_SECOND_FOCUS);
 168
 169       AISObjectPtr anAIS1(new GeomAPI_AISObject());
 170       anAIS1->setImpl(new Handle(AIS_InteractiveObject)(aF1Prs));
 171       aList.append(anAIS1);
 172
 173       AISObjectPtr anAIS2(new GeomAPI_AISObject());
 174       anAIS2->setImpl(new Handle(AIS_InteractiveObject)(aF2Prs));
 175       aList.append(anAIS2);
 176     }
 177     if (myPresentations.contains(aPrs->object()))
 178       myPresentations[aPrs->object()].append(aList);
 179     else
 180       myPresentations[aPrs->object()] = aList;
 181     foreach(AISObjectPtr anAIS, aList) {
 182       aDisplayer->displayAIS(anAIS, false);
 183       aWorkshop->selectionActivate()->activateAIS(anAIS->impl<Handle(AIS_InteractiveObject)>(),
 184         TopAbs_VERTEX, false);
 185     }
 186   }
 187 }
 188
 189 std::shared_ptr<GeomAPI_Pln> PartSet_ExternalPointsMgr::plane() const
 190 {
 191   return PartSet_Tools::sketchPlane(mySketch);
 192 }
 193
 194 void PartSet_ExternalPointsMgr::onDisplayObject(ObjectPtr theObj, AISObjectPtr theAIS)
 195 {
 196   updateCenterPresentations();
 197 }
 198
 199 void PartSet_ExternalPointsMgr::onEraseObject(ObjectPtr theObj, AISObjectPtr theAIS)
 200 {
 201   if (myPresentations.contains(theObj)) {
 202     XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
 203     XGUI_Displayer* aDisplayer = aWorkshop->displayer();
 204     ListOfAIS aList = myPresentations[theObj];
 205     foreach(AISObjectPtr aAIS, aList) {
 206       aDisplayer->eraseAIS(aAIS, false);
 207     }
 208     myPresentations.remove(theObj);
 209     aDisplayer->updateViewer();
 210   }
 211 }
 212
 213
 214 bool PartSet_ExternalPointsMgr::isSketchObject(const ObjectPtr& theRes) const
 215 {
 216   FeaturePtr aFeature = ModelAPI_Feature::feature(theRes);
 217   if (!aFeature.get())
 218     return false;
 219   CompositeFeaturePtr aComp = ModelAPI_Tools::compositeOwner(aFeature);
 220   return aComp == mySketch;
 221 }
 222
 223 void PartSet_ExternalPointsMgr::onSelectionChanged()
 224 {
 225   if (plane().get()) {
 226     XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
 227     disconnect(aWorkshop->selector(), SIGNAL(selectionChanged()),
 228                this, SLOT(onSelectionChanged()));
 229     updateCenterPresentations();
 230   }
 231 }