1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
3 #include "XGUI_SelectionMgr.h"
5 #include "XGUI_Workshop.h"
6 #include "XGUI_ObjectsBrowser.h"
7 #include "XGUI_SalomeConnector.h"
8 #include "XGUI_ViewerProxy.h"
9 #include "XGUI_Displayer.h"
10 #include "XGUI_Selection.h"
13 #include <AppElements_MainWindow.h>
16 #include <ModelAPI_Feature.h>
17 #include <ModelAPI_Session.h>
18 #include <ModelAPI_AttributeDocRef.h>
19 #include <ModelAPI_Data.h>
20 #include <ModelAPI_Result.h>
21 #include <ModelAPI_Object.h>
22 #include <ModelAPI_ResultCompSolid.h>
24 #include <ModuleBase_ViewerPrs.h>
25 #include <ModuleBase_Tools.h>
27 #include <SelectMgr_ListIteratorOfListOfFilter.hxx>
30 #include <VInspectorAPI_CallBack.h>
33 XGUI_SelectionMgr::XGUI_SelectionMgr(XGUI_Workshop* theParent)
37 mySelection = new XGUI_Selection(myWorkshop);
40 XGUI_SelectionMgr::~XGUI_SelectionMgr()
45 //**************************************************************
46 void XGUI_SelectionMgr::connectViewers()
48 connect(myWorkshop->objectBrowser(), SIGNAL(selectionChanged()), this,
49 SLOT(onObjectBrowserSelection()));
51 //Connect to other viewers
52 connect(myWorkshop->viewer(), SIGNAL(selectionChanged()), this, SLOT(onViewerSelection()));
55 //**************************************************************
56 void XGUI_SelectionMgr::setSelectedOwners(const SelectMgr_IndexedMapOfOwner& theSelectedOwners,
59 SelectMgr_IndexedMapOfOwner aSelectedOwners;
60 selection()->selectedOwners(aSelectedOwners);
62 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
63 if (!aContext.IsNull()) {
64 for (Standard_Integer i = 1, n = theSelectedOwners.Extent(); i <= n; i++) {
65 Handle(SelectMgr_EntityOwner) anOwner = theSelectedOwners(i);
66 if (aSelectedOwners.FindIndex(anOwner) > 0)
69 aContext->AddOrRemoveSelected(anOwner, isUpdateViewer);
71 if (myWorkshop->displayer()->getCallBack())
72 myWorkshop->displayer()->getCallBack()->AddOrRemoveSelected(anOwner);
78 //**************************************************************
79 void XGUI_SelectionMgr::updateSelectedOwners(bool isUpdateViewer)
81 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
82 if (aContext.IsNull())
85 const SelectMgr_ListOfFilter& aFilters = aContext->Filters();
87 SelectMgr_IndexedMapOfOwner anOwnersToDeselect;
89 SelectMgr_ListIteratorOfListOfFilter anIt(aFilters);
90 for (; anIt.More(); anIt.Next()) {
91 Handle(SelectMgr_Filter) aFilter = anIt.Value();
92 for (aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected()) {
93 Handle(SelectMgr_EntityOwner) anOwner = aContext->SelectedOwner();
94 if (!aFilter->IsOk(anOwner))
95 anOwnersToDeselect.Add(aContext->SelectedOwner());
99 setSelectedOwners(anOwnersToDeselect, false);
102 aContext->UpdateCurrentViewer();
105 //**************************************************************
106 void XGUI_SelectionMgr::onObjectBrowserSelection()
108 QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
109 myWorkshop->selector()->selection()->getSelected(ModuleBase_ISelection::Browser);
111 QList<ModuleBase_ViewerPrsPtr> aTmpList = aSelectedPrs;
114 foreach(ModuleBase_ViewerPrsPtr aPrs, aTmpList) {
115 aObject = aPrs->object();
117 aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObject);
118 if (aFeature.get()) {
119 const std::list<std::shared_ptr<ModelAPI_Result>> aResList = aFeature->results();
121 ResultCompSolidPtr aCompSolid;
122 std::list<ResultPtr>::const_iterator aIt;
123 for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
125 aSelectedPrs.append(std::shared_ptr<ModuleBase_ViewerPrs>(
126 new ModuleBase_ViewerPrs(aResult, GeomShapePtr(), NULL)));
127 aCompSolid = std::dynamic_pointer_cast<ModelAPI_ResultCompSolid>(aResult);
128 if (aCompSolid.get()) {
129 for (int i = 0; i < aCompSolid->numberOfSubs(); i++) {
130 ResultBodyPtr aResult = aCompSolid->subResult(i);
131 aSelectedPrs.append(std::shared_ptr<ModuleBase_ViewerPrs>(
132 new ModuleBase_ViewerPrs(aResult, aResult->shape(), NULL)));
139 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
140 aDisplayer->setSelected(aSelectedPrs);
141 emit selectionChanged();
144 //**************************************************************
145 void XGUI_SelectionMgr::onViewerSelection()
147 SessionPtr aMgr = ModelAPI_Session::get();
148 DocumentPtr anActiveDocument = aMgr->activeDocument();
149 QObjectPtrList aFeatures;
152 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
153 if (!aContext.IsNull()) {
154 QList<ModuleBase_ViewerPrsPtr> aPresentations =
155 selection()->getSelected(ModuleBase_ISelection::Viewer);
156 foreach(ModuleBase_ViewerPrsPtr aPrs, aPresentations) {
157 if (aPrs->object().get()) {
158 if (!aFeatures.contains(aPrs->object()))
159 aFeatures.append(aPrs->object());
160 if (aPrs->shape().get()) {
161 aResult = std::dynamic_pointer_cast<ModelAPI_Result>(aPrs->object());
163 aFeature = anActiveDocument->producedByFeature(aResult, aPrs->shape());
164 if (aFeature.get() && (!aFeatures.contains(aFeature)))
165 aFeatures.append(aFeature);
171 bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
172 myWorkshop->objectBrowser()->setObjectsSelected(aFeatures);
173 myWorkshop->objectBrowser()->blockSignals(aBlocked);
175 emit selectionChanged();
178 //**************************************************************
179 void XGUI_SelectionMgr::updateSelectionBy(const ModuleBase_ISelection::SelectionPlace& thePlace)
181 QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
182 myWorkshop->selector()->selection()->getSelected(thePlace);
183 if (thePlace == ModuleBase_ISelection::Browser) {
184 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
185 aDisplayer->setSelected(aSelectedPrs);
189 //**************************************************************
190 void XGUI_SelectionMgr::clearSelection()
192 QObjectPtrList aFeatures;
193 bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
194 myWorkshop->objectBrowser()->setObjectsSelected(aFeatures);
195 myWorkshop->objectBrowser()->blockSignals(aBlocked);
197 QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
198 myWorkshop->selector()->selection()->getSelected(ModuleBase_ISelection::Browser);
200 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
201 aDisplayer->setSelected(aSelectedPrs);
203 emit selectionChanged();