src/XGUI/XGUI_SelectionMgr.cpp

   1 // Copyright (C) 2014-2017  CEA/DEN, EDF R&D
   2 //
   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.
   7 //
   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.
  12 //
  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
  16 //
  17 // See http://www.salome-platform.org/ or
  18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
  19 //
  20
  21 #include "XGUI_SelectionMgr.h"
  22
  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"
  31
  32 #ifndef HAVE_SALOME
  33 #include <AppElements_MainWindow.h>
  34 #endif
  35
  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>
  44
  45 #include <GeomAPI_Shape.h>
  46
  47 #include <ModuleBase_ViewerPrs.h>
  48 #include <ModuleBase_Tools.h>
  49
  50 #include <SelectMgr_ListIteratorOfListOfFilter.hxx>
  51 #include <TopExp_Explorer.hxx>
  52 #include <TopoDS_Shape.hxx>
  53 #include <TopTools_MapOfShape.hxx>
  54
  55 #ifdef TINSPECTOR
  56 #include <inspector/VInspectorAPI_CallBack.hxx>
  57 #endif
  58
  59 #define OPTIMIZATION_LEVEL 50
  60
  61
  62 XGUI_SelectionMgr::XGUI_SelectionMgr(XGUI_Workshop* theParent)
  63     : QObject(theParent),
  64       myWorkshop(theParent)
  65 {
  66   mySelection = new XGUI_Selection(myWorkshop);
  67 }
  68
  69 XGUI_SelectionMgr::~XGUI_SelectionMgr()
  70 {
  71   delete mySelection;
  72 }
  73
  74 //**************************************************************
  75 void XGUI_SelectionMgr::connectViewers()
  76 {
  77   connect(myWorkshop->objectBrowser(), SIGNAL(selectionChanged()), this,
  78           SLOT(onObjectBrowserSelection()));
  79
  80   //Connect to other viewers
  81   connect(myWorkshop->viewer(), SIGNAL(selectionChanged()), this, SLOT(onViewerSelection()));
  82 }
  83
  84 //**************************************************************
  85 void XGUI_SelectionMgr::setSelectedOwners(const SelectMgr_IndexedMapOfOwner& theSelectedOwners,
  86                                           bool isUpdateViewer)
  87 {
  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);
  93
  94     for  (Standard_Integer i = 1, n = theSelectedOwners.Extent(); i <= n; i++)  {
  95       Handle(SelectMgr_EntityOwner) anOwner = theSelectedOwners(i);
  96
  97       aContext->AddOrRemoveSelected(anOwner, isUpdateViewer);
  98       #ifdef TINSPECTOR
  99       if (myWorkshop->displayer()->getCallBack())
 100         myWorkshop->displayer()->getCallBack()->AddOrRemoveSelected(anOwner);
 101       #endif
 102     }
 103   }
 104 }
 105
 106 //**************************************************************
 107 void XGUI_SelectionMgr::onObjectBrowserSelection()
 108 {
 109   QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
 110     myWorkshop->selector()->selection()->getSelected(ModuleBase_ISelection::Browser);
 111   XGUI_Displayer* aDisplayer = myWorkshop->displayer();
 112   if (!myWorkshop->operationMgr()->hasOperation()) {
 113
 114     QList<ModuleBase_ViewerPrsPtr> aTmpList = aSelectedPrs;
 115     ObjectPtr aObject;
 116     FeaturePtr aFeature;
 117     // Select all results of a selected feature in viewer
 118     foreach(ModuleBase_ViewerPrsPtr aPrs, aSelectedPrs) {
 119       aObject = aPrs->object();
 120       if (aObject.get()) {
 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)));
 128           }
 129         }
 130       }
 131     }
 132     aDisplayer->setSelected(aTmpList);
 133   } else {
 134     aDisplayer->setSelected(aSelectedPrs);
 135   }
 136   emit selectionChanged();
 137 }
 138
 139 //**************************************************************
 140 void XGUI_SelectionMgr::onViewerSelection()
 141 {
 142   QList<ModuleBase_ViewerPrsPtr> aValues;
 143   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
 144   if (!aContext.IsNull())
 145     aValues = selection()->getSelected(ModuleBase_ISelection::Viewer);
 146
 147   QObjectPtrList anObjects;
 148   convertToObjectBrowserSelection(aValues, anObjects);
 149
 150   bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
 151   myWorkshop->objectBrowser()->setObjectsSelected(anObjects);
 152   myWorkshop->objectBrowser()->blockSignals(aBlocked);
 153
 154   emit selectionChanged();
 155 }
 156
 157 //**************************************************************
 158 void XGUI_SelectionMgr::deselectPresentation(const Handle(AIS_InteractiveObject) theObject)
 159 {
 160   NCollection_List<Handle(SelectBasics_EntityOwner)> aResultOwners;
 161
 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
 166       continue;
 167     if (anOwner->Selectable() == theObject && anOwner->IsSelected())
 168       aResultOwners.Append(anOwner);
 169   }
 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);
 176   }
 177 }
 178
 179 //**************************************************************
 180 void XGUI_SelectionMgr::updateSelectionBy(const ModuleBase_ISelection::SelectionPlace& thePlace)
 181 {
 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);
 187   }
 188
 189 }
 190 //**************************************************************
 191 void XGUI_SelectionMgr::clearSelection()
 192 {
 193   QObjectPtrList aFeatures;
 194   bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
 195   myWorkshop->objectBrowser()->setObjectsSelected(aFeatures);
 196   myWorkshop->objectBrowser()->blockSignals(aBlocked);
 197
 198   QList<ModuleBase_ViewerPrsPtr> aSelectedPrs =
 199              myWorkshop->selector()->selection()->getSelected(ModuleBase_ISelection::Browser);
 200
 201   XGUI_Displayer* aDisplayer = myWorkshop->displayer();
 202   aDisplayer->setSelected(aSelectedPrs);
 203
 204   emit selectionChanged();
 205 }
 206 //**************************************************************
 207 void XGUI_SelectionMgr::setSelected(const QList<ModuleBase_ViewerPrsPtr>& theValues)
 208 {
 209   // update selection in Viewer
 210   XGUI_Displayer* aDisplayer = myWorkshop->displayer();
 211   aDisplayer->setSelected(theValues);
 212
 213   // update selection in Object Browser
 214   bool aBlocked = myWorkshop->objectBrowser()->blockSignals(true);
 215   QObjectPtrList anObjects;
 216   convertToObjectBrowserSelection(theValues, anObjects);
 217
 218   myWorkshop->objectBrowser()->setObjectsSelected(anObjects);
 219   myWorkshop->objectBrowser()->blockSignals(aBlocked);
 220 }
 221
 222 //**************************************************************
 223 void XGUI_SelectionMgr::convertToObjectBrowserSelection(
 224                                    const QList<ModuleBase_ViewerPrsPtr>& theValues,
 225                                    QObjectPtrList& theObjects)
 226 {
 227   theObjects.clear();
 228
 229   ResultPtr aResult;
 230   FeaturePtr aFeature;
 231   bool aHasOperation = (myWorkshop->operationMgr()->currentOperation() != 0);
 232   SessionPtr aMgr = ModelAPI_Session::get();
 233   DocumentPtr anActiveDocument = aMgr->activeDocument();
 234
 235   TopTools_MapOfShape aShapeMap;
 236   bool aToOptimize = (theValues.size() > OPTIMIZATION_LEVEL);
 237
 238   GeomShapePtr aShape;
 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());
 246         if (aResult.get()) {
 247           aShape = aPrs->shape();
 248           aTShape = aShape->impl<TopoDS_Shape>();
 249           if (aToOptimize) {
 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);
 257                 }
 258               }
 259             }
 260           }
 261           else {
 262             aFeature = anActiveDocument->producedByFeature(aResult, aShape);
 263           }
 264           if (aFeature.get() && (!theObjects.contains(aFeature)))
 265             theObjects.append(aFeature);
 266         }
 267       }
 268     }
 269   }
 270 }
 271
 272 std::list<FeaturePtr> XGUI_SelectionMgr::getSelectedFeatures()
 273 {
 274   std::list<FeaturePtr> aFeatures;
 275   QObjectPtrList aObjects = selection()->selectedObjects();
 276   if (aObjects.isEmpty())
 277     return aFeatures;
 278
 279   bool isPart = false;
 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()));
 285       if (!isPart)
 286         aFeatures.push_back(aFeature);
 287     }
 288   }
 289   return aFeatures;
 290 }
 291
 292 QList<TopoDS_Shape> XGUI_SelectionMgr::findAllShapes(const ResultPtr& theResult) const
 293 {
 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());
 303       }
 304     }
 305   }
 306   return aResult;
 307 }