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