src/XGUI/XGUI_SelectionMgr.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
   2
   3 #include "XGUI_SelectionMgr.h"
   4
   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"
  11
  12 #ifndef HAVE_SALOME
  13 #include <AppElements_MainWindow.h>
  14 #endif
  15
  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
  23 #include <SelectMgr_ListIteratorOfListOfFilter.hxx>
  24
  25 XGUI_SelectionMgr::XGUI_SelectionMgr(XGUI_Workshop* theParent)
  26     : QObject(theParent),
  27       myWorkshop(theParent)
  28 {
  29   mySelection = new XGUI_Selection(myWorkshop);
  30 }
  31
  32 XGUI_SelectionMgr::~XGUI_SelectionMgr()
  33 {
  34   delete mySelection;
  35 }
  36
  37 //**************************************************************
  38 void XGUI_SelectionMgr::connectViewers()
  39 {
  40   connect(myWorkshop->objectBrowser(), SIGNAL(selectionChanged()), this,
  41           SLOT(onObjectBrowserSelection()));
  42
  43   //Connect to other viewers
  44   connect(myWorkshop->viewer(), SIGNAL(selectionChanged()), this, SLOT(onViewerSelection()));
  45 }
  46
  47 //**************************************************************
  48 void XGUI_SelectionMgr::setSelectedOwners(const SelectMgr_IndexedMapOfOwner& theSelectedOwners,
  49                                           bool isUpdateViewer)
  50 {
  51   SelectMgr_IndexedMapOfOwner aSelectedOwners;
  52   selection()->selectedOwners(aSelectedOwners);
  53
  54   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
  55   if (!aContext.IsNull()) {
  56     for  (Standard_Integer i = 1, n = theSelectedOwners.Extent(); i <= n; i++)  {
  57       Handle(SelectMgr_EntityOwner) anOwner = theSelectedOwners(i);
  58       if (aSelectedOwners.FindIndex(anOwner) > 0)
  59         continue;
  60
  61       aContext->AddOrRemoveSelected(anOwner, isUpdateViewer);
  62     }
  63   }
  64 }
  65
  66 //**************************************************************
  67 void XGUI_SelectionMgr::updateSelectedOwners(bool isUpdateViewer)
  68 {
  69   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
  70   if (aContext.IsNull())
  71     return;
  72
  73   const SelectMgr_ListOfFilter& aFilters = aContext->Filters();
  74
  75   SelectMgr_IndexedMapOfOwner anOwnersToDeselect;
  76
  77   SelectMgr_ListIteratorOfListOfFilter anIt(aFilters);
  78   for (; anIt.More(); anIt.Next()) {
  79     Handle(SelectMgr_Filter) aFilter = anIt.Value();
  80     for (aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected()) {
  81       Handle(SelectMgr_EntityOwner) anOwner = aContext->SelectedOwner();
  82       if (!aFilter->IsOk(anOwner))
  83         anOwnersToDeselect.Add(aContext->SelectedOwner());
  84     }
  85   }
  86
  87   setSelectedOwners(anOwnersToDeselect, false);
  88
  89   if (isUpdateViewer)
  90     aContext->UpdateCurrentViewer();
  91 }
  92
  93 //**************************************************************
  94 void XGUI_SelectionMgr::onObjectBrowserSelection()
  95 {
  96   QList<ModuleBase_ViewerPrs> aSelectedPrs =
  97              myWorkshop->selector()->selection()->getSelected(ModuleBase_ISelection::Browser);
  98
  99   QList<ModuleBase_ViewerPrs> aTmpList = aSelectedPrs;
 100   ObjectPtr aObject;
 101   FeaturePtr aFeature;
 102   foreach(ModuleBase_ViewerPrs aPrs, aTmpList) {
 103     aObject = aPrs.object();
 104     if (aObject.get()) {
 105       aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(aObject);
 106       if (aFeature.get()) {
 107         const std::list<std::shared_ptr<ModelAPI_Result>> aResList = aFeature->results();
 108         std::list<ResultPtr>::const_iterator aIt;
 109         for (aIt = aResList.cbegin(); aIt != aResList.cend(); ++aIt) {
 110           aSelectedPrs.append(ModuleBase_ViewerPrs((*aIt), GeomShapePtr(), NULL));
 111         }
 112       }
 113     }
 114   }
 115   XGUI_Displayer* aDisplayer = myWorkshop->displayer();
 116   aDisplayer->setSelected(aSelectedPrs);
 117   emit selectionChanged();
 118 }
 119
 120 //**************************************************************
 121 void XGUI_SelectionMgr::onViewerSelection()
 122 {
 123   QObjectPtrList aFeatures;
 124   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
 125   if (!aContext.IsNull()) {
 126     QList<ModuleBase_ViewerPrs> aPresentations = selection()->getSelected(ModuleBase_ISelection::Viewer);
 127     foreach(ModuleBase_ViewerPrs aPrs, aPresentations) {
 128       if (aPrs.object().get())
 129         aFeatures.append(aPrs.object());
 130     }
 131   }
 132   // Add features by selected results
 133   QObjectPtrList aTmpList = aFeatures;
 134   ResultPtr aResult;
 135   FeaturePtr aFeature;
 136   foreach(ObjectPtr aObj, aTmpList) {
 137     aResult = std::dynamic_pointer_cast<ModelAPI_Result>(aObj);
 138     if (aResult.get()) {
 139       aFeature = ModelAPI_Feature::feature(aResult);
 140       if (aFeature.get() && (!aFeatures.contains(aFeature)))
 141         aFeatures.append(aFeature);
 142     }
 143   }
 144   bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
 145   myWorkshop->objectBrowser()->setObjectsSelected(aFeatures);
 146   myWorkshop->objectBrowser()->blockSignals(aBlocked);
 147
 148   emit selectionChanged();
 149 }
 150
 151 //**************************************************************
 152 void XGUI_SelectionMgr::updateSelectionBy(const ModuleBase_ISelection::SelectionPlace& thePlace)
 153 {
 154   QList<ModuleBase_ViewerPrs> aSelectedPrs =
 155                myWorkshop->selector()->selection()->getSelected(thePlace);
 156   if (thePlace == ModuleBase_ISelection::Browser) {
 157     XGUI_Displayer* aDisplayer = myWorkshop->displayer();
 158     aDisplayer->setSelected(aSelectedPrs);
 159   }
 160
 161 }
 162 //**************************************************************
 163 void XGUI_SelectionMgr::clearSelection()
 164 {
 165   QObjectPtrList aFeatures;
 166   bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
 167   myWorkshop->objectBrowser()->setObjectsSelected(aFeatures);
 168   myWorkshop->objectBrowser()->blockSignals(aBlocked);
 169
 170   QList<ModuleBase_ViewerPrs> aSelectedPrs =
 171              myWorkshop->selector()->selection()->getSelected(ModuleBase_ISelection::Browser);
 172
 173   XGUI_Displayer* aDisplayer = myWorkshop->displayer();
 174   aDisplayer->setSelected(aSelectedPrs);
 175
 176   emit selectionChanged();
 177 }