1 // Copyright (C) 2014-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 #include "XGUI_SelectionMgr.h"
23 #include "XGUI_Workshop.h"
24 #include "XGUI_ObjectsBrowser.h"
25 #include "XGUI_SalomeConnector.h"
26 #include "XGUI_ViewerProxy.h"
27 #include "XGUI_Displayer.h"
28 #include "XGUI_Selection.h"
29 #include "XGUI_OperationMgr.h"
30 #include "XGUI_SelectionActivate.h"
33 #include <AppElements_MainWindow.h>
36 #include <ModelAPI_Feature.h>
37 #include <ModelAPI_Session.h>
38 #include <ModelAPI_AttributeDocRef.h>
39 #include <ModelAPI_Data.h>
40 #include <ModelAPI_Result.h>
41 #include <ModelAPI_Object.h>
42 #include <ModelAPI_ResultBody.h>
43 #include <ModelAPI_Tools.h>
45 #include <GeomAPI_Shape.h>
47 #include <ModuleBase_ViewerPrs.h>
48 #include <ModuleBase_Tools.h>
50 #include <SelectMgr_ListIteratorOfListOfFilter.hxx>
51 #include <TopExp_Explorer.hxx>
52 #include <TopoDS_Shape.hxx>
53 #include <TopTools_MapOfShape.hxx>
56 #include <inspector/VInspectorAPI_CallBack.hxx>
59 #define OPTIMIZATION_LEVEL 50
62 XGUI_SelectionMgr::XGUI_SelectionMgr(XGUI_Workshop* theParent)
66 mySelection = new XGUI_Selection(myWorkshop);
69 XGUI_SelectionMgr::~XGUI_SelectionMgr()
74 //**************************************************************
75 void XGUI_SelectionMgr::connectViewers()
77 connect(myWorkshop->objectBrowser(), SIGNAL(selectionChanged()), this,
78 SLOT(onObjectBrowserSelection()));
80 //Connect to other viewers
81 connect(myWorkshop->viewer(), SIGNAL(selectionChanged()), this, SLOT(onViewerSelection()));
84 //**************************************************************
85 void XGUI_SelectionMgr::setSelectedOwners(const SelectMgr_IndexedMapOfOwner& theSelectedOwners,
88 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
89 if (!aContext.IsNull()) {
90 /// previous selection should be cleared, else there will be decomposition of selections:
91 /// as AddOrRemoveSelected inverts current selection
92 aContext->ClearSelected(false);
94 for (Standard_Integer i = 1, n = theSelectedOwners.Extent(); i <= n; i++) {
95 Handle(SelectMgr_EntityOwner) anOwner = theSelectedOwners(i);
97 aContext->AddOrRemoveSelected(anOwner, isUpdateViewer);
99 if (myWorkshop->displayer()->getCallBack())
100 myWorkshop->displayer()->getCallBack()->AddOrRemoveSelected(anOwner);
106 //**************************************************************
107 void XGUI_SelectionMgr::onObjectBrowserSelection()
109 QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
110 myWorkshop->selector()->selection()->getSelected(ModuleBase_ISelection::Browser);
111 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
112 if (!myWorkshop->operationMgr()->hasOperation()) {
114 QList<ModuleBase_ViewerPrsPtr> aTmpList = aSelectedPrs;
117 // Select all results of a selected feature in viewer
118 foreach(ModuleBase_ViewerPrsPtr aPrs, aSelectedPrs) {
119 aObject = aPrs->object();
121 aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObject);
122 if (aFeature.get()) {
123 std::list<ResultPtr> allRes;
124 ModelAPI_Tools::allResults(aFeature, allRes);
125 for(std::list<ResultPtr>::iterator aRes = allRes.begin(); aRes != allRes.end(); aRes++) {
126 aSelectedPrs.append(std::shared_ptr<ModuleBase_ViewerPrs>(
127 new ModuleBase_ViewerPrs(*aRes, GeomShapePtr(), NULL)));
132 aDisplayer->setSelected(aTmpList);
134 aDisplayer->setSelected(aSelectedPrs);
136 emit selectionChanged();
139 //**************************************************************
140 void XGUI_SelectionMgr::onViewerSelection()
142 QList<ModuleBase_ViewerPrsPtr> aValues;
143 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
144 if (!aContext.IsNull())
145 aValues = selection()->getSelected(ModuleBase_ISelection::Viewer);
147 QObjectPtrList anObjects;
148 convertToObjectBrowserSelection(aValues, anObjects);
150 bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
151 myWorkshop->objectBrowser()->setObjectsSelected(anObjects);
152 myWorkshop->objectBrowser()->blockSignals(aBlocked);
154 emit selectionChanged();
157 //**************************************************************
158 void XGUI_SelectionMgr::deselectPresentation(const Handle(AIS_InteractiveObject) theObject)
160 NCollection_List<Handle(SelectBasics_EntityOwner)> aResultOwners;
162 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
163 for (aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected()) {
164 Handle(SelectMgr_EntityOwner) anOwner = aContext->SelectedOwner();
165 if (anOwner.IsNull()) // TODO: check why it is possible
167 if (anOwner->Selectable() == theObject && anOwner->IsSelected())
168 aResultOwners.Append(anOwner);
170 NCollection_List<Handle(SelectBasics_EntityOwner)>::Iterator anOwnersIt (aResultOwners);
171 Handle(SelectMgr_EntityOwner) anOwner;
172 for (; anOwnersIt.More(); anOwnersIt.Next()) {
173 anOwner = Handle(SelectMgr_EntityOwner)::DownCast(anOwnersIt.Value());
174 if (!anOwner.IsNull())
175 aContext->AddOrRemoveSelected(anOwner, false);
179 //**************************************************************
180 void XGUI_SelectionMgr::updateSelectionBy(const ModuleBase_ISelection::SelectionPlace& thePlace)
182 QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
183 myWorkshop->selector()->selection()->getSelected(thePlace);
184 if (thePlace == ModuleBase_ISelection::Browser) {
185 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
186 aDisplayer->setSelected(aSelectedPrs);
190 //**************************************************************
191 void XGUI_SelectionMgr::clearSelection()
193 QObjectPtrList aFeatures;
194 bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
195 myWorkshop->objectBrowser()->setObjectsSelected(aFeatures);
196 myWorkshop->objectBrowser()->blockSignals(aBlocked);
198 QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
199 myWorkshop->selector()->selection()->getSelected(ModuleBase_ISelection::Browser);
201 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
202 aDisplayer->setSelected(aSelectedPrs);
204 emit selectionChanged();
206 //**************************************************************
207 void XGUI_SelectionMgr::setSelected(const QList<ModuleBase_ViewerPrsPtr>& theValues)
209 // update selection in Viewer
210 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
211 aDisplayer->setSelected(theValues);
213 // update selection in Object Browser
214 bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
215 QObjectPtrList anObjects;
216 convertToObjectBrowserSelection(theValues, anObjects);
218 myWorkshop->objectBrowser()->setObjectsSelected(anObjects);
219 myWorkshop->objectBrowser()->blockSignals(aBlocked);
222 //**************************************************************
223 void XGUI_SelectionMgr::convertToObjectBrowserSelection(
224 const QList<ModuleBase_ViewerPrsPtr>& theValues,
225 QObjectPtrList& theObjects)
231 bool aHasOperation = (myWorkshop->operationMgr()->currentOperation() != 0);
232 SessionPtr aMgr = ModelAPI_Session::get();
233 DocumentPtr anActiveDocument = aMgr->activeDocument();
235 TopTools_MapOfShape aShapeMap;
236 bool aToOptimize = (theValues.size() > OPTIMIZATION_LEVEL);
239 TopoDS_Shape aTShape;
240 foreach(ModuleBase_ViewerPrsPtr aPrs, theValues) {
241 if (aPrs->object().get()) {
242 if (!theObjects.contains(aPrs->object()))
243 theObjects.append(aPrs->object());
244 if (aPrs->shape().get() && (!aHasOperation)) {
245 aResult = std::dynamic_pointer_cast<ModelAPI_Result>(aPrs->object());
247 aShape = aPrs->shape();
248 aTShape = aShape->impl<TopoDS_Shape>();
250 if (!aShapeMap.Contains(aTShape)) {
251 aFeature = anActiveDocument->producedByFeature(aResult, aShape);
252 if (aFeature.get()) {
253 QList<TopoDS_Shape> aResList = findAllShapes(aResult);
254 foreach(TopoDS_Shape aShape, aResList) {
255 if (!aShapeMap.Contains(aShape))
256 aShapeMap.Add(aShape);
262 aFeature = anActiveDocument->producedByFeature(aResult, aShape);
264 if (aFeature.get() && (!theObjects.contains(aFeature)))
265 theObjects.append(aFeature);
272 std::list<FeaturePtr> XGUI_SelectionMgr::getSelectedFeatures()
274 std::list<FeaturePtr> aFeatures;
275 QObjectPtrList aObjects = selection()->selectedObjects();
276 if (aObjects.isEmpty())
280 foreach(ObjectPtr aObj, aObjects) {
281 FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObj);
282 if (aFeature.get()) {
283 ResultPtr aRes = aFeature->firstResult();
284 isPart = (aRes.get() && (aRes->groupName() == ModelAPI_ResultPart::group()));
286 aFeatures.push_back(aFeature);
292 QList<TopoDS_Shape> XGUI_SelectionMgr::findAllShapes(const ResultPtr& theResult) const
294 QIntList aModes = myWorkshop->selectionActivate()->activeSelectionModes();
295 GeomShapePtr aResShape = theResult->shape();
296 TopoDS_Shape aShape = aResShape->impl<TopoDS_Shape>();
297 QList<TopoDS_Shape> aResult;
298 foreach(int aShapeType, aModes) {
299 if (aShapeType < TopAbs_SHAPE) {
300 TopExp_Explorer aExp(aShape, (TopAbs_ShapeEnum)aShapeType);
301 for (; aExp.More(); aExp.Next()) {
302 aResult.append(aExp.Current());