1 // Copyright (C) 2017-2022 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 email : webmaster.salome@opencascade.com
20 // File: PartSet_ExternalPointsMgr.cpp
21 // Created: 26 April 2017
22 // Author: Vitaly SMETANNIKOV
24 #include "PartSet_ExternalPointsMgr.h"
25 #include "PartSet_CenterPrs.h"
26 #include "PartSet_Tools.h"
28 #include <ModelAPI_Tools.h>
29 #include <ModelAPI_ResultField.h>
30 #include <ModelAPI_ResultGroup.h>
32 #include <ModuleBase_IWorkshop.h>
33 #include <ModuleBase_ViewerPrs.h>
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>
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>
48 #pragma warning(disable : 4456) // for nested foreach
51 PartSet_ExternalPointsMgr::PartSet_ExternalPointsMgr(ModuleBase_IWorkshop* theWorkshop,
52 const CompositeFeaturePtr& theSketch)
53 : QObject(theWorkshop), myWorkshop(theWorkshop), mySketch(theSketch)
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)));
60 connect(aDisplayer, SIGNAL(beforeObjectErase(ObjectPtr, AISObjectPtr)),
61 SLOT(onEraseObject(ObjectPtr, AISObjectPtr)));
63 updateCenterPresentations();
67 PartSet_ExternalPointsMgr::~PartSet_ExternalPointsMgr()
69 if (myPresentations.isEmpty())
71 XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
75 XGUI_Displayer* aDisplayer = aWorkshop->displayer();
76 QMapIterator<ObjectPtr, ListOfAIS> aIt(myPresentations);
77 while (aIt.hasNext()) {
79 ListOfAIS aAISList = aIt.value();
80 foreach (AISObjectPtr aAIS, aAISList) {
81 aDisplayer->eraseAIS(aAIS, false);
87 //******************************************************
88 QList<std::shared_ptr<ModuleBase_ViewerPrs>> PartSet_ExternalPointsMgr::findCircularEdgesInPlane()
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();
95 std::shared_ptr<GeomAPI_Pln> aPlane = plane();
96 foreach(ObjectPtr aObj, aDispObjects) {
97 if (myPresentations.contains(aObj))
100 // Do not process objects of the current sketch
101 if (isSketchObject(aObj))
104 ResultPtr aResObj = std::dynamic_pointer_cast<ModelAPI_Result>(aObj);
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()))
113 GeomShapePtr aShape = aResObj->shape();
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)) {
132 std::shared_ptr<ModuleBase_ViewerPrs>
133 aPrs(new ModuleBase_ViewerPrs(aResObj, aEdgeShape));
134 aResult.append(aPrs);
145 void PartSet_ExternalPointsMgr::updateCenterPresentations()
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()));
154 XGUI_Displayer* aDisplayer = aWorkshop->displayer();
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()));
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);
167 AISObjectPtr anAIS(new GeomAPI_AISObject());
168 anAIS->setImpl(new Handle(AIS_InteractiveObject)(aCentPrs));
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);
181 AISObjectPtr anAIS1(new GeomAPI_AISObject());
182 anAIS1->setImpl(new Handle(AIS_InteractiveObject)(aF1Prs));
183 aList.append(anAIS1);
185 AISObjectPtr anAIS2(new GeomAPI_AISObject());
186 anAIS2->setImpl(new Handle(AIS_InteractiveObject)(aF2Prs));
187 aList.append(anAIS2);
189 if (myPresentations.contains(aPrs->object()))
190 myPresentations[aPrs->object()].append(aList);
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);
201 std::shared_ptr<GeomAPI_Pln> PartSet_ExternalPointsMgr::plane() const
203 return PartSet_Tools::sketchPlane(mySketch);
206 void PartSet_ExternalPointsMgr::onDisplayObject(ObjectPtr theObj, AISObjectPtr theAIS)
208 updateCenterPresentations();
211 void PartSet_ExternalPointsMgr::onEraseObject(ObjectPtr theObj, AISObjectPtr theAIS)
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);
220 myPresentations.remove(theObj);
221 aDisplayer->updateViewer();
226 bool PartSet_ExternalPointsMgr::isSketchObject(const ObjectPtr& theRes) const
228 FeaturePtr aFeature = ModelAPI_Feature::feature(theRes);
231 CompositeFeaturePtr aComp = ModelAPI_Tools::compositeOwner(aFeature);
232 return aComp == mySketch;
235 void PartSet_ExternalPointsMgr::onSelectionChanged()
238 XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(myWorkshop);
239 disconnect(aWorkshop->selector(), SIGNAL(selectionChanged()),
240 this, SLOT(onSelectionChanged()));
241 updateCenterPresentations();