src/PartSet/PartSet_ExternalPointsMgr.cpp

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