1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: PartSet_ExternalPointsMgr.cpp
4 // Created: 26 April 2017
5 // Author: Vitaly SMETANNIKOV
7 #include "PartSet_ExternalPointsMgr.h"
8 #include "PartSet_CenterPrs.h"
9 #include "PartSet_Tools.h"
11 #include <ModelAPI_Tools.h>
13 #include <ModuleBase_IWorkshop.h>
14 #include <ModuleBase_ViewerPrs.h>
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>
22 #include <XGUI_Tools.h>
23 #include <XGUI_Displayer.h>
24 #include <XGUI_Workshop.h>
25 #include <XGUI_SelectionMgr.h>
27 PartSet_ExternalPointsMgr::PartSet_ExternalPointsMgr(ModuleBase_IWorkshop* theWorkshop,
28 const CompositeFeaturePtr& theSketch)
29 : QObject(theWorkshop), myWorkshop(theWorkshop), mySketch(theSketch)
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)));
36 connect(aDisplayer, SIGNAL(beforeObjectErase(ObjectPtr, AISObjectPtr)),
37 SLOT(onEraseObject(ObjectPtr, AISObjectPtr)));
39 updateCenterPresentations();
43 PartSet_ExternalPointsMgr::~PartSet_ExternalPointsMgr()
45 if (myPresentations.isEmpty())
47 XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
51 XGUI_Displayer* aDisplayer = aWorkshop->displayer();
52 QMapIterator<ObjectPtr, ListOfAIS> aIt(myPresentations);
53 while (aIt.hasNext()) {
55 ListOfAIS aAISList = aIt.value();
56 foreach (AISObjectPtr aAIS, aAISList) {
57 aDisplayer->eraseAIS(aAIS, false);
63 //******************************************************
64 QList<std::shared_ptr<ModuleBase_ViewerPrs>> PartSet_ExternalPointsMgr::findCircularEdgesInPlane()
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();
71 std::shared_ptr<GeomAPI_Pln> aPlane = plane();
72 foreach(ObjectPtr aObj, aDispObjects) {
73 if (myPresentations.contains(aObj))
76 // Do not process objects of the current sketch
77 if (isSketchObject(aObj))
80 ResultPtr aResObj = std::dynamic_pointer_cast<ModelAPI_Result>(aObj);
82 GeomShapePtr aShape = aResObj->shape();
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)) {
101 std::shared_ptr<ModuleBase_ViewerPrs>
102 aPrs(new ModuleBase_ViewerPrs(aResObj, aEdgeShape));
103 aResult.append(aPrs);
114 void PartSet_ExternalPointsMgr::updateCenterPresentations()
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()));
123 XGUI_Displayer* aDisplayer = aWorkshop->displayer();
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()));
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);
136 AISObjectPtr anAIS(new GeomAPI_AISObject());
137 anAIS->setImpl(new Handle(AIS_InteractiveObject)(aCentPrs));
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);
150 AISObjectPtr anAIS1(new GeomAPI_AISObject());
151 anAIS1->setImpl(new Handle(AIS_InteractiveObject)(aF1Prs));
152 aList.append(anAIS1);
154 AISObjectPtr anAIS2(new GeomAPI_AISObject());
155 anAIS2->setImpl(new Handle(AIS_InteractiveObject)(aF2Prs));
156 aList.append(anAIS2);
158 if (myPresentations.contains(aPrs->object()))
159 myPresentations[aPrs->object()].append(aList);
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);
169 std::shared_ptr<GeomAPI_Pln> PartSet_ExternalPointsMgr::plane() const
171 return PartSet_Tools::sketchPlane(mySketch);
174 void PartSet_ExternalPointsMgr::onDisplayObject(ObjectPtr theObj, AISObjectPtr theAIS)
176 updateCenterPresentations();
179 void PartSet_ExternalPointsMgr::onEraseObject(ObjectPtr theObj, AISObjectPtr theAIS)
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);
188 myPresentations.remove(theObj);
189 aDisplayer->updateViewer();
194 bool PartSet_ExternalPointsMgr::isSketchObject(const ObjectPtr& theRes) const
196 FeaturePtr aFeature = ModelAPI_Feature::feature(theRes);
199 CompositeFeaturePtr aComp = ModelAPI_Tools::compositeOwner(aFeature);
200 return aComp == mySketch;
203 void PartSet_ExternalPointsMgr::onSelectionChanged()
206 XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
207 disconnect(aWorkshop->selector(), SIGNAL(selectionChanged()),
208 this, SLOT(onSelectionChanged()));
209 updateCenterPresentations();