1 // Copyright (C) 2017 CEA/DEN, EDF R&D
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.
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.
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
17 // See http://www.salome-platform.org/ or
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
21 // File: PartSet_ExternalPointsMgr.cpp
22 // Created: 26 April 2017
23 // Author: Vitaly SMETANNIKOV
25 #include "PartSet_ExternalPointsMgr.h"
26 #include "PartSet_CenterPrs.h"
27 #include "PartSet_Tools.h"
29 #include <ModelAPI_Tools.h>
31 #include <ModuleBase_IWorkshop.h>
32 #include <ModuleBase_ViewerPrs.h>
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>
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>
46 PartSet_ExternalPointsMgr::PartSet_ExternalPointsMgr(ModuleBase_IWorkshop* theWorkshop,
47 const CompositeFeaturePtr& theSketch)
48 : QObject(theWorkshop), myWorkshop(theWorkshop), mySketch(theSketch)
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)));
55 connect(aDisplayer, SIGNAL(beforeObjectErase(ObjectPtr, AISObjectPtr)),
56 SLOT(onEraseObject(ObjectPtr, AISObjectPtr)));
58 updateCenterPresentations();
62 PartSet_ExternalPointsMgr::~PartSet_ExternalPointsMgr()
64 if (myPresentations.isEmpty())
66 XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
70 XGUI_Displayer* aDisplayer = aWorkshop->displayer();
71 QMapIterator<ObjectPtr, ListOfAIS> aIt(myPresentations);
72 while (aIt.hasNext()) {
74 ListOfAIS aAISList = aIt.value();
75 foreach (AISObjectPtr aAIS, aAISList) {
76 aDisplayer->eraseAIS(aAIS, false);
82 //******************************************************
83 QList<std::shared_ptr<ModuleBase_ViewerPrs>> PartSet_ExternalPointsMgr::findCircularEdgesInPlane()
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();
90 std::shared_ptr<GeomAPI_Pln> aPlane = plane();
91 foreach(ObjectPtr aObj, aDispObjects) {
92 if (myPresentations.contains(aObj))
95 // Do not process objects of the current sketch
96 if (isSketchObject(aObj))
99 ResultPtr aResObj = std::dynamic_pointer_cast<ModelAPI_Result>(aObj);
101 GeomShapePtr aShape = aResObj->shape();
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)) {
120 std::shared_ptr<ModuleBase_ViewerPrs>
121 aPrs(new ModuleBase_ViewerPrs(aResObj, aEdgeShape));
122 aResult.append(aPrs);
133 void PartSet_ExternalPointsMgr::updateCenterPresentations()
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()));
142 XGUI_Displayer* aDisplayer = aWorkshop->displayer();
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()));
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);
155 AISObjectPtr anAIS(new GeomAPI_AISObject());
156 anAIS->setImpl(new Handle(AIS_InteractiveObject)(aCentPrs));
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);
169 AISObjectPtr anAIS1(new GeomAPI_AISObject());
170 anAIS1->setImpl(new Handle(AIS_InteractiveObject)(aF1Prs));
171 aList.append(anAIS1);
173 AISObjectPtr anAIS2(new GeomAPI_AISObject());
174 anAIS2->setImpl(new Handle(AIS_InteractiveObject)(aF2Prs));
175 aList.append(anAIS2);
177 if (myPresentations.contains(aPrs->object()))
178 myPresentations[aPrs->object()].append(aList);
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);
189 std::shared_ptr<GeomAPI_Pln> PartSet_ExternalPointsMgr::plane() const
191 return PartSet_Tools::sketchPlane(mySketch);
194 void PartSet_ExternalPointsMgr::onDisplayObject(ObjectPtr theObj, AISObjectPtr theAIS)
196 updateCenterPresentations();
199 void PartSet_ExternalPointsMgr::onEraseObject(ObjectPtr theObj, AISObjectPtr theAIS)
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);
208 myPresentations.remove(theObj);
209 aDisplayer->updateViewer();
214 bool PartSet_ExternalPointsMgr::isSketchObject(const ObjectPtr& theRes) const
216 FeaturePtr aFeature = ModelAPI_Feature::feature(theRes);
219 CompositeFeaturePtr aComp = ModelAPI_Tools::compositeOwner(aFeature);
220 return aComp == mySketch;
223 void PartSet_ExternalPointsMgr::onSelectionChanged()
226 XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
227 disconnect(aWorkshop->selector(), SIGNAL(selectionChanged()),
228 this, SLOT(onSelectionChanged()));
229 updateCenterPresentations();