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>
26 #include <SelectMgr_ListIteratorOfListOfFilter.hxx>
28 XGUI_SelectionMgr::XGUI_SelectionMgr(XGUI_Workshop* theParent)
32 mySelection = new XGUI_Selection(myWorkshop);
35 XGUI_SelectionMgr::~XGUI_SelectionMgr()
40 //**************************************************************
41 void XGUI_SelectionMgr::connectViewers()
43 connect(myWorkshop->objectBrowser(), SIGNAL(selectionChanged()), this,
44 SLOT(onObjectBrowserSelection()));
46 //Connect to other viewers
47 connect(myWorkshop->viewer(), SIGNAL(selectionChanged()), this, SLOT(onViewerSelection()));
50 //**************************************************************
51 void XGUI_SelectionMgr::setSelectedOwners(const SelectMgr_IndexedMapOfOwner& theSelectedOwners,
54 SelectMgr_IndexedMapOfOwner aSelectedOwners;
55 selection()->selectedOwners(aSelectedOwners);
57 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
58 if (!aContext.IsNull()) {
59 for (Standard_Integer i = 1, n = theSelectedOwners.Extent(); i <= n; i++) {
60 Handle(SelectMgr_EntityOwner) anOwner = theSelectedOwners(i);
61 if (aSelectedOwners.FindIndex(anOwner) > 0)
64 aContext->AddOrRemoveSelected(anOwner, isUpdateViewer);
69 //**************************************************************
70 void XGUI_SelectionMgr::updateSelectedOwners(bool isUpdateViewer)
72 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
73 if (aContext.IsNull())
76 const SelectMgr_ListOfFilter& aFilters = aContext->Filters();
78 SelectMgr_IndexedMapOfOwner anOwnersToDeselect;
80 SelectMgr_ListIteratorOfListOfFilter anIt(aFilters);
81 for (; anIt.More(); anIt.Next()) {
82 Handle(SelectMgr_Filter) aFilter = anIt.Value();
83 for (aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected()) {
84 Handle(SelectMgr_EntityOwner) anOwner = aContext->SelectedOwner();
85 if (!aFilter->IsOk(anOwner))
86 anOwnersToDeselect.Add(aContext->SelectedOwner());
90 setSelectedOwners(anOwnersToDeselect, false);
93 aContext->UpdateCurrentViewer();
96 //**************************************************************
97 void XGUI_SelectionMgr::onObjectBrowserSelection()
99 QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
100 myWorkshop->selector()->selection()->getSelected(ModuleBase_ISelection::Browser);
102 QList<ModuleBase_ViewerPrsPtr> aTmpList = aSelectedPrs;
105 foreach(ModuleBase_ViewerPrsPtr aPrs, aTmpList) {
106 aObject = aPrs->object();
108 aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObject);
109 if (aFeature.get()) {
110 const std::list<std::shared_ptr<ModelAPI_Result>> aResList = aFeature->results();
112 ResultCompSolidPtr aCompSolid;
113 std::list<ResultPtr>::const_iterator aIt;
114 for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
116 aCompSolid = std::dynamic_pointer_cast<ModelAPI_ResultCompSolid>(aResult);
117 if (aCompSolid.get()) {
118 for (int i = 0; i < aCompSolid->numberOfSubs(); i++) {
119 aSelectedPrs.append(std::shared_ptr<ModuleBase_ViewerPrs>(
120 new ModuleBase_ViewerPrs(aCompSolid->subResult(i), GeomShapePtr(), NULL)));
123 aSelectedPrs.append(std::shared_ptr<ModuleBase_ViewerPrs>(
124 new ModuleBase_ViewerPrs(aResult, GeomShapePtr(), NULL)));
130 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
131 aDisplayer->setSelected(aSelectedPrs);
132 emit selectionChanged();
135 //**************************************************************
136 void XGUI_SelectionMgr::onViewerSelection()
138 SessionPtr aMgr = ModelAPI_Session::get();
139 DocumentPtr anActiveDocument = aMgr->activeDocument();
140 QObjectPtrList aFeatures;
143 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
144 if (!aContext.IsNull()) {
145 QList<ModuleBase_ViewerPrsPtr> aPresentations = selection()->getSelected(ModuleBase_ISelection::Viewer);
146 foreach(ModuleBase_ViewerPrsPtr aPrs, aPresentations) {
147 if (aPrs->object().get()) {
148 if (!aFeatures.contains(aPrs->object()))
149 aFeatures.append(aPrs->object());
150 if (aPrs->shape().get()) {
151 aResult = std::dynamic_pointer_cast<ModelAPI_Result>(aPrs->object());
153 aFeature = anActiveDocument->producedByFeature(aResult, aPrs->shape());
154 if (aFeature.get() && (!aFeatures.contains(aFeature)))
155 aFeatures.append(aFeature);
161 bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
162 myWorkshop->objectBrowser()->setObjectsSelected(aFeatures);
163 myWorkshop->objectBrowser()->blockSignals(aBlocked);
165 emit selectionChanged();
168 //**************************************************************
169 void XGUI_SelectionMgr::updateSelectionBy(const ModuleBase_ISelection::SelectionPlace& thePlace)
171 QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
172 myWorkshop->selector()->selection()->getSelected(thePlace);
173 if (thePlace == ModuleBase_ISelection::Browser) {
174 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
175 aDisplayer->setSelected(aSelectedPrs);
179 //**************************************************************
180 void XGUI_SelectionMgr::clearSelection()
182 QObjectPtrList aFeatures;
183 bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
184 myWorkshop->objectBrowser()->setObjectsSelected(aFeatures);
185 myWorkshop->objectBrowser()->blockSignals(aBlocked);
187 QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
188 myWorkshop->selector()->selection()->getSelected(ModuleBase_ISelection::Browser);
190 XGUI_Displayer* aDisplayer = myWorkshop->displayer();
191 aDisplayer->setSelected(aSelectedPrs);
193 emit selectionChanged();