src/XGUI/XGUI_Displayer.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
   2
   3 // File:        XGUI_Displayer.cpp
   4 // Created:     20 Apr 2014
   5 // Author:      Natalia ERMOLAEVA
   6
   7 #include "XGUI_Displayer.h"
   8 #include "XGUI_Workshop.h"
   9 #include "XGUI_ViewerProxy.h"
  10 #include "XGUI_SelectionMgr.h"
  11 #include "XGUI_Selection.h"
  12 #include "XGUI_CustomPrs.h"
  13
  14 #include <AppElements_Viewer.h>
  15
  16 #include <ModelAPI_Document.h>
  17 #include <ModelAPI_Data.h>
  18 #include <ModelAPI_Object.h>
  19 #include <ModelAPI_Tools.h>
  20 #include <ModelAPI_AttributeIntArray.h>
  21
  22 #include <ModuleBase_ResultPrs.h>
  23
  24 #include <GeomAPI_Shape.h>
  25 #include <GeomAPI_IPresentable.h>
  26 #include <GeomAPI_ICustomPrs.h>
  27
  28 #include <AIS_InteractiveContext.hxx>
  29 #include <AIS_LocalContext.hxx>
  30 #include <AIS_ListOfInteractive.hxx>
  31 #include <AIS_ListIteratorOfListOfInteractive.hxx>
  32 #include <AIS_DimensionSelectionMode.hxx>
  33 #include <AIS_Shape.hxx>
  34 #include <AIS_Dimension.hxx>
  35 #include <TColStd_ListIteratorOfListOfInteger.hxx>
  36 #include <SelectMgr_ListOfFilter.hxx>
  37 #include <SelectMgr_ListIteratorOfListOfFilter.hxx>
  38
  39 #include <TColStd_MapOfTransient.hxx>
  40 #include <TColStd_MapIteratorOfMapOfTransient.hxx>
  41
  42 #include <set>
  43
  44 const int MOUSE_SENSITIVITY_IN_PIXEL = 10;  ///< defines the local context mouse selection sensitivity
  45
  46 //#define DEBUG_DISPLAY
  47 //#define DEBUG_ACTIVATE
  48 //#define DEBUG_FEATURE_REDISPLAY
  49 //#define DEBUG_SELECTION_FILTERS
  50 //#define DEBUG_USE_CLEAR_OUTDATED_SELECTION
  51
  52 // Workaround for bug #25637
  53 void displayedObjects(const Handle(AIS_InteractiveContext)& theAIS, AIS_ListOfInteractive& theList)
  54 {
  55   // Get from null point
  56   theAIS->DisplayedObjects(theList, true);
  57   if (theAIS->HasOpenedContext()) {
  58     // get from local context
  59     const Handle(AIS_LocalContext)& aLC = theAIS->LocalContext();
  60     TColStd_MapOfTransient aMap;
  61     int NbDisp = aLC->DisplayedObjects(aMap);
  62     TColStd_MapIteratorOfMapOfTransient aIt(aMap);
  63
  64     Handle(AIS_InteractiveObject) curIO;
  65     Handle(Standard_Transient) Tr;
  66     for(; aIt.More(); aIt.Next()){
  67       Tr = aIt.Key();
  68       curIO = *((Handle(AIS_InteractiveObject)*) &Tr);
  69       theList.Append(curIO);
  70     }
  71   }
  72 }
  73
  74
  75 XGUI_Displayer::XGUI_Displayer(XGUI_Workshop* theWorkshop)
  76   : myWorkshop(theWorkshop)
  77 {
  78   enableUpdateViewer(true);
  79   myCustomPrs = std::shared_ptr<GeomAPI_ICustomPrs>(new XGUI_CustomPrs());
  80 }
  81
  82 XGUI_Displayer::~XGUI_Displayer()
  83 {
  84 }
  85
  86 bool XGUI_Displayer::isVisible(ObjectPtr theObject) const
  87 {
  88   return myResult2AISObjectMap.contains(theObject);
  89 }
  90
  91 void XGUI_Displayer::display(ObjectPtr theObject, bool isUpdateViewer)
  92 {
  93   if (isVisible(theObject)) {
  94     redisplay(theObject, isUpdateViewer);
  95   } else {
  96 #ifdef DEBUG_DISPLAY
  97     FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
  98     if (aFeature.get() != NULL) {
  99       qDebug(QString("display feature: %1, displayed: %2").
 100         arg(aFeature->data()->name().c_str()).
 101         arg(displayedObjects().size()).toStdString().c_str());
 102     }
 103 #endif
 104     AISObjectPtr anAIS;
 105
 106     GeomPresentablePtr aPrs = std::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
 107     bool isShading = false;
 108     if (aPrs.get() != NULL) {
 109       anAIS = aPrs->getAISObject(anAIS);
 110     } else {
 111       ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
 112       if (aResult.get() != NULL) {
 113         std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(aResult);
 114         if (aShapePtr.get() != NULL) {
 115           anAIS = AISObjectPtr(new GeomAPI_AISObject());
 116           anAIS->setImpl(new Handle(AIS_InteractiveObject)(new ModuleBase_ResultPrs(aResult)));
 117           //anAIS->createShape(aShapePtr);
 118           isShading = true;
 119         }
 120       }
 121     }
 122     if (anAIS)
 123       display(theObject, anAIS, isShading, isUpdateViewer);
 124   }
 125 }
 126
 127 bool canBeShaded(Handle(AIS_InteractiveObject) theAIS)
 128 {
 129   Handle(AIS_Shape) aShapePrs = Handle(AIS_Shape)::DownCast(theAIS);
 130   if (!aShapePrs.IsNull()) {
 131     TopoDS_Shape aShape = aShapePrs->Shape();
 132     TopAbs_ShapeEnum aType = aShape.ShapeType();
 133     if ((aType == TopAbs_VERTEX) || (aType == TopAbs_EDGE) || (aType == TopAbs_WIRE))
 134       return false;
 135     else {
 136       // Check that the presentation is not a sketch
 137       Handle(ModuleBase_ResultPrs) aPrs = Handle(ModuleBase_ResultPrs)::DownCast(theAIS);
 138       if (!aPrs.IsNull())
 139         return !aPrs->isSketchMode();
 140       return true;
 141     }
 142   }
 143   return false;
 144 }
 145
 146 void XGUI_Displayer::display(ObjectPtr theObject, AISObjectPtr theAIS,
 147                              bool isShading, bool isUpdateViewer)
 148 {
 149   Handle(AIS_InteractiveContext) aContext = AISContext();
 150   if (aContext.IsNull())
 151     return;
 152
 153   Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
 154   if (!anAISIO.IsNull()) {
 155     myResult2AISObjectMap[theObject] = theAIS;
 156     bool aCanBeShaded = ::canBeShaded(anAISIO);
 157     // In order to avoid extra closing/opening context
 158     SelectMgr_IndexedMapOfOwner aSelectedOwners;
 159     if (aCanBeShaded) {
 160       myWorkshop->selector()->selection()->selectedOwners(aSelectedOwners);
 161       closeLocalContexts(false);
 162     }
 163     aContext->Display(anAISIO, false);
 164     aContext->SetDisplayMode(anAISIO, isShading? Shading : Wireframe, false);
 165     if (isShading)
 166       anAISIO->Attributes()->SetFaceBoundaryDraw( Standard_True );
 167     emit objectDisplayed(theObject, theAIS);
 168
 169     bool isCustomized = customizeObject(theObject);
 170     if (isCustomized)
 171       aContext->Redisplay(anAISIO, false);
 172
 173     if (aCanBeShaded) {
 174       openLocalContext();
 175       activateObjects(myActiveSelectionModes);
 176       myWorkshop->selector()->setSelectedOwners(aSelectedOwners, false);
 177     }
 178     else
 179       activate(anAISIO, myActiveSelectionModes);
 180  }
 181   if (isUpdateViewer)
 182     updateViewer();
 183 }
 184
 185 void XGUI_Displayer::erase(ObjectPtr theObject, const bool isUpdateViewer)
 186 {
 187   if (!isVisible(theObject))
 188     return;
 189
 190   Handle(AIS_InteractiveContext) aContext = AISContext();
 191   if (aContext.IsNull())
 192     return;
 193   AISObjectPtr anObject = myResult2AISObjectMap[theObject];
 194   if (anObject) {
 195     Handle(AIS_InteractiveObject) anAIS = anObject->impl<Handle(AIS_InteractiveObject)>();
 196     if (!anAIS.IsNull()) {
 197       emit beforeObjectErase(theObject, anObject);
 198       aContext->Remove(anAIS, isUpdateViewer);
 199     }
 200   }
 201   myResult2AISObjectMap.remove(theObject);
 202 }
 203
 204 void XGUI_Displayer::redisplay(ObjectPtr theObject, bool isUpdateViewer)
 205 {
 206   if (!isVisible(theObject))
 207     return;
 208
 209   AISObjectPtr aAISObj = getAISObject(theObject);
 210   Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
 211
 212   GeomPresentablePtr aPrs = std::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
 213   if (aPrs) {
 214     AISObjectPtr aAIS_Obj = aPrs->getAISObject(aAISObj);
 215     if (!aAIS_Obj) {
 216       erase(theObject, isUpdateViewer);
 217       return;
 218     }
 219     if (aAIS_Obj != aAISObj) {
 220       myResult2AISObjectMap[theObject] = aAIS_Obj;
 221     }
 222     aAISIO = aAIS_Obj->impl<Handle(AIS_InteractiveObject)>();
 223   }
 224
 225   if (!aAISIO.IsNull()) {
 226     Handle(AIS_InteractiveContext) aContext = AISContext();
 227     if (aContext.IsNull())
 228       return;
 229     // Check that the visualized shape is the same and the redisplay is not necessary
 230     // Redisplay of AIS object leads to this object selection compute and the selection
 231     // in the browser is lost
 232
 233     // this check is not necessary anymore because the selection store/restore is realized
 234     // before and after the values modification.
 235     // Moreother, this check avoids customize and redisplay presentation if the presentable
 236     // parameter is changed.
 237     bool isEqualShapes = false;
 238     ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
 239     if (aResult.get() != NULL) {
 240       Handle(AIS_Shape) aShapePrs = Handle(AIS_Shape)::DownCast(aAISIO);
 241       if (!aShapePrs.IsNull()) {
 242         std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(aResult);
 243         if (aShapePtr.get()) {
 244           const TopoDS_Shape& aOldShape = aShapePrs->Shape();
 245           isEqualShapes = aOldShape.IsEqual(aShapePtr->impl<TopoDS_Shape>());
 246         }
 247       }
 248     }
 249     // Customization of presentation
 250     bool isCustomized = customizeObject(theObject);
 251     #ifdef DEBUG_FEATURE_REDISPLAY
 252       //qDebug(QString("Redisplay: %1, isEqualShapes=%2, isCustomized=%3").
 253       //  arg(!isEqualShapes || isCustomized).arg(isEqualShapes).arg(isCustomized).toStdString().c_str());
 254     #endif
 255     if (!isEqualShapes || isCustomized) {
 256       aContext->Redisplay(aAISIO, false);
 257       #ifdef DEBUG_FEATURE_REDISPLAY
 258       //qDebug("  Redisplay happens");
 259       #endif
 260       if (isUpdateViewer)
 261         updateViewer();
 262     }
 263   }
 264 }
 265
 266 void XGUI_Displayer::deactivate(ObjectPtr theObject)
 267 {
 268   if (isVisible(theObject)) {
 269     Handle(AIS_InteractiveContext) aContext = AISContext();
 270     if (aContext.IsNull())
 271       return;
 272
 273     AISObjectPtr anObj = myResult2AISObjectMap[theObject];
 274     Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
 275     aContext->Deactivate(anAIS);
 276 #ifdef DEBUG_USE_CLEAR_OUTDATED_SELECTION
 277     aContext->LocalContext()->ClearOutdatedSelection(anAIS, true);
 278     updateViewer();
 279 #endif
 280   }
 281 }
 282
 283 /*void XGUI_Displayer::activate(ObjectPtr theFeature)
 284 {
 285   activate(theFeature, myActiveSelectionModes);
 286 }
 287
 288 void XGUI_Displayer::activate(ObjectPtr theObject, const QIntList& theModes)
 289 {
 290 #ifdef DEBUG_ACTIVATE
 291     FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
 292
 293     if (aFeature.get() != NULL) {
 294       QIntList aModes;
 295       getModesOfActivation(theObject, aModes);
 296
 297
 298       qDebug(QString("activate feature: %1, theModes: %2, myActiveSelectionModes: %3, getModesOf: %4").
 299         arg(aFeature->data()->name().c_str()).
 300         arg(theModes.size()).
 301         arg(myActiveSelectionModes.size()).
 302         arg(aModes.size()).toStdString().c_str());
 303     }
 304 #endif
 305
 306   if (isVisible(theObject)) {
 307     Handle(AIS_InteractiveContext) aContext = AISContext();
 308     if (aContext.IsNull())
 309       return;
 310
 311     AISObjectPtr anObj = myResult2AISObjectMap[theObject];
 312     Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
 313
 314     activate(anAIS, theModes);
 315   }
 316 }*/
 317
 318 void XGUI_Displayer::getModesOfActivation(ObjectPtr theObject, QIntList& theModes)
 319 {
 320   if (!isVisible(theObject))
 321     return;
 322
 323   Handle(AIS_InteractiveContext) aContext = AISContext();
 324   if (aContext.IsNull())
 325     return;
 326
 327   AISObjectPtr aAISObj = getAISObject(theObject);
 328
 329   if (aAISObj.get() != NULL) {
 330     Handle(AIS_InteractiveObject) anAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
 331     TColStd_ListOfInteger aTColModes;
 332     aContext->ActivatedModes(anAISIO, aTColModes);
 333     TColStd_ListIteratorOfListOfInteger itr( aTColModes );
 334     for (; itr.More(); itr.Next() ) {
 335       theModes.append(itr.Value());
 336     }
 337   }
 338 }
 339
 340 void XGUI_Displayer::activateObjects(const QIntList& theModes)
 341 {
 342 #ifdef DEBUG_ACTIVATE
 343   qDebug(QString("activate all features: theModes: %2, myActiveSelectionModes: %3").
 344     arg(theModes.size()).
 345     arg(myActiveSelectionModes.size()).
 346     toStdString().c_str());
 347 #endif
 348   // In order to avoid doblications of selection modes
 349   QIntList aNewModes;
 350   foreach (int aMode, theModes) {
 351     if (!aNewModes.contains(aMode))
 352       aNewModes.append(aMode);
 353   }
 354   myActiveSelectionModes = aNewModes;
 355   Handle(AIS_InteractiveContext) aContext = AISContext();
 356   if (aContext.IsNull())
 357     return;
 358   // Open local context if there is no one
 359   if (!aContext->HasOpenedContext())
 360     return;
 361
 362   //aContext->UseDisplayedObjects();
 363   //myUseExternalObjects = true;
 364
 365   AIS_ListOfInteractive aPrsList;
 366   ::displayedObjects(aContext, aPrsList);
 367
 368   Handle(AIS_Trihedron) aTrihedron;
 369   AIS_ListIteratorOfListOfInteractive aLIt(aPrsList);
 370   Handle(AIS_InteractiveObject) anAISIO;
 371   for(aLIt.Initialize(aPrsList); aLIt.More(); aLIt.Next()){
 372     anAISIO = aLIt.Value();
 373     activate(anAISIO, myActiveSelectionModes);
 374   }
 375 }
 376
 377
 378 void XGUI_Displayer::deactivateObjects()
 379 {
 380   myActiveSelectionModes.clear();
 381   Handle(AIS_InteractiveContext) aContext = AISContext();
 382   // Open local context if there is no one
 383   if (!aContext->HasOpenedContext())
 384     return;
 385
 386   //aContext->NotUseDisplayedObjects();
 387   AIS_ListOfInteractive aPrsList;
 388   ::displayedObjects(aContext, aPrsList);
 389
 390   AIS_ListIteratorOfListOfInteractive aLIt;
 391   //Handle(AIS_Trihedron) aTrihedron;
 392   Handle(AIS_InteractiveObject) anAISIO;
 393   for(aLIt.Initialize(aPrsList); aLIt.More(); aLIt.Next()){
 394     anAISIO = aLIt.Value();
 395     aContext->Deactivate(anAISIO);
 396 #ifdef DEBUG_USE_CLEAR_OUTDATED_SELECTION
 397     aContext->LocalContext()->ClearOutdatedSelection(anAISIO, true);
 398     updateViewer();
 399 #endif
 400   }
 401 }
 402
 403 bool XGUI_Displayer::isActive(ObjectPtr theObject) const
 404 {
 405   Handle(AIS_InteractiveContext) aContext = AISContext();
 406   if (aContext.IsNull())
 407     return false;
 408   if (!isVisible(theObject))
 409     return false;
 410
 411   AISObjectPtr anObj = myResult2AISObjectMap[theObject];
 412   Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
 413
 414   TColStd_ListOfInteger aModes;
 415   aContext->ActivatedModes(anAIS, aModes);
 416   return aModes.Extent() > 0;
 417 }
 418
 419 void XGUI_Displayer::setSelected(const QObjectPtrList& theResults, const bool isUpdateViewer)
 420 {
 421   Handle(AIS_InteractiveContext) aContext = AISContext();
 422   if (aContext.IsNull())
 423     return;
 424   if (aContext->HasOpenedContext()) {
 425     aContext->UnhilightSelected();
 426     aContext->ClearSelected();
 427     foreach(ObjectPtr aResult, theResults) {
 428       if (isVisible(aResult)) {
 429         AISObjectPtr anObj = myResult2AISObjectMap[aResult];
 430         Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
 431         if (!anAIS.IsNull()) {
 432           // The methods are replaced in order to provide multi-selection, e.g. restore selection
 433           // by activating multi selector widget. It also gives an advantage that the multi
 434           // selection in OB gives multi-selection in the viewer
 435           //aContext->SetSelected(anAIS, false);
 436           // The selection in the context was cleared, so the method sets the objects are selected
 437           aContext->AddOrRemoveSelected(anAIS, false);
 438         }
 439       }
 440     }
 441   } else {
 442     aContext->UnhilightCurrents();
 443     aContext->ClearCurrents();
 444     foreach(ObjectPtr aResult, theResults) {
 445       if (isVisible(aResult)) {
 446         AISObjectPtr anObj = myResult2AISObjectMap[aResult];
 447         Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
 448         if (!anAIS.IsNull())
 449           aContext->SetCurrentObject(anAIS, false);
 450       }
 451     }
 452   }
 453   if (isUpdateViewer)
 454     updateViewer();
 455 }
 456
 457 void XGUI_Displayer::setSelected(const  QList<ModuleBase_ViewerPrs>& theValues, bool isUpdateViewer)
 458 {
 459   Handle(AIS_InteractiveContext) aContext = AISContext();
 460   if (aContext.IsNull())
 461     return;
 462   if (aContext->HasOpenedContext()) {
 463     aContext->UnhilightSelected();
 464     aContext->ClearSelected();
 465     //if (aSelected.size() > 0) {
 466     foreach (ModuleBase_ViewerPrs aPrs, theValues) {
 467     //    if (isValidSelection(aPrs)) {
 468     //foreach(ObjectPtr aResult, theResults) {
 469       ObjectPtr anObject = aPrs.object();
 470       ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
 471       if (aResult.get() && isVisible(aResult)) {
 472         AISObjectPtr anObj = myResult2AISObjectMap[aResult];
 473         Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
 474         if (!anAIS.IsNull()) {
 475           const TopoDS_Shape& aShape = aPrs.shape();
 476           if (!aShape.IsNull()) {
 477             aContext->AddOrRemoveSelected(aShape, false);
 478           }
 479           else {
 480             // The methods are replaced in order to provide multi-selection, e.g. restore selection
 481             // by activating multi selector widget. It also gives an advantage that the multi
 482             // selection in OB gives multi-selection in the viewer
 483             //aContext->SetSelected(anAIS, false);
 484             // The selection in the context was cleared, so the method sets the objects are selected
 485             aContext->AddOrRemoveSelected(anAIS, false);
 486           }
 487         }
 488       }
 489     }
 490   } else {
 491     aContext->UnhilightCurrents();
 492     aContext->ClearCurrents();
 493     //foreach(ObjectPtr aResult, theResults) {
 494     foreach (ModuleBase_ViewerPrs aPrs, theValues) {
 495       ObjectPtr anObject = aPrs.object();
 496       ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
 497       if (aResult.get() && isVisible(aResult)) {
 498         AISObjectPtr anObj = myResult2AISObjectMap[aResult];
 499         Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
 500         if (!anAIS.IsNull())
 501           aContext->SetCurrentObject(anAIS, false);
 502       }
 503     }
 504   }
 505   if (isUpdateViewer)
 506     updateViewer();
 507 }
 508
 509 void XGUI_Displayer::clearSelected()
 510 {
 511   Handle(AIS_InteractiveContext) aContext = AISContext();
 512   if (aContext) {
 513     aContext->UnhilightCurrents(false);
 514     aContext->ClearSelected();
 515   }
 516 }
 517
 518 void XGUI_Displayer::eraseAll(const bool isUpdateViewer)
 519 {
 520   Handle(AIS_InteractiveContext) aContext = AISContext();
 521   if (!aContext.IsNull()) {
 522     foreach (ObjectPtr aObj, myResult2AISObjectMap.keys()) {
 523       AISObjectPtr aAISObj = myResult2AISObjectMap[aObj];
 524       // erase an object
 525       Handle(AIS_InteractiveObject) anIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
 526       if (!anIO.IsNull()) {
 527         emit beforeObjectErase(aObj, aAISObj);
 528         aContext->Remove(anIO, false);
 529       }
 530     }
 531     if (isUpdateViewer)
 532       updateViewer();
 533   }
 534   myResult2AISObjectMap.clear();
 535 }
 536
 537 void XGUI_Displayer::openLocalContext()
 538 {
 539   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
 540   if (aContext.IsNull())
 541     return;
 542   // Open local context if there is no one
 543   if (!aContext->HasOpenedContext()) {
 544     // Preserve selected objects
 545     //AIS_ListOfInteractive aAisList;
 546     //for (aContext->InitCurrent(); aContext->MoreCurrent(); aContext->NextCurrent())
 547     //  aAisList.Append(aContext->Current());
 548
 549     // get the filters from the global context and append them to the local context
 550     // a list of filters in the global context is not cleared and should be cleared here
 551     SelectMgr_ListOfFilter aFilters;
 552     aFilters.Assign(aContext->Filters());
 553     // it is important to remove the filters in the global context, because there is a code
 554     // in the closeLocalContex, which restore the global context filters
 555     aContext->RemoveFilters();
 556
 557     //aContext->ClearCurrents();
 558     aContext->OpenLocalContext();
 559     //aContext->NotUseDisplayedObjects();
 560
 561     //myUseExternalObjects = false;
 562
 563     SelectMgr_ListIteratorOfListOfFilter aIt(aFilters);
 564     for (;aIt.More(); aIt.Next()) {
 565       aContext->AddFilter(aIt.Value());
 566     }
 567     // Restore selection
 568     //AIS_ListIteratorOfListOfInteractive aIt2(aAisList);
 569     //for(; aIt2.More(); aIt2.Next()) {
 570     //  aContext->SetSelected(aIt2.Value(), false);
 571     //}
 572   }
 573 }
 574
 575 void XGUI_Displayer::closeLocalContexts(const bool isUpdateViewer)
 576 {
 577   Handle(AIS_InteractiveContext) aContext = AISContext();
 578   if ( (!aContext.IsNull()) && (aContext->HasOpenedContext()) ) {
 579     // Preserve selected objects
 580     //AIS_ListOfInteractive aAisList;
 581     //for (aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected())
 582     //  aAisList.Append(aContext->SelectedInteractive());
 583
 584     // get the filters from the local context and append them to the global context
 585     // a list of filters in the local context is cleared
 586     SelectMgr_ListOfFilter aFilters;
 587     aFilters.Assign(aContext->Filters());
 588
 589     //aContext->ClearSelected();
 590     aContext->CloseAllContexts(false);
 591
 592     // Redisplay all object if they were displayed in localContext
 593     Handle(AIS_InteractiveObject) aAISIO;
 594     foreach (AISObjectPtr aAIS, myResult2AISObjectMap) {
 595       aAISIO = aAIS->impl<Handle(AIS_InteractiveObject)>();
 596       if (aContext->DisplayStatus(aAISIO) != AIS_DS_Displayed) {
 597         aContext->Display(aAISIO, false);
 598         aContext->SetDisplayMode(aAISIO, Shading, false);
 599       }
 600     }
 601
 602     // Append the filters from the local selection in the global selection context
 603     SelectMgr_ListIteratorOfListOfFilter aIt(aFilters);
 604     for (;aIt.More(); aIt.Next()) {
 605       Handle(SelectMgr_Filter) aFilter = aIt.Value();
 606       aContext->AddFilter(aFilter);
 607     }
 608
 609     if (isUpdateViewer)
 610       updateViewer();
 611     //myUseExternalObjects = false;
 612
 613     // Restore selection
 614     //AIS_ListIteratorOfListOfInteractive aIt2(aAisList);
 615     //for(; aIt2.More(); aIt2.Next()) {
 616     //  if (aContext->IsDisplayed(aIt2.Value()))
 617     //    aContext->SetCurrentObject(aIt2.Value(), false);
 618     //}
 619   }
 620 }
 621
 622 AISObjectPtr XGUI_Displayer::getAISObject(ObjectPtr theObject) const
 623 {
 624   AISObjectPtr anIO;
 625   if (myResult2AISObjectMap.contains(theObject))
 626     anIO = myResult2AISObjectMap[theObject];
 627   return anIO;
 628 }
 629
 630 ObjectPtr XGUI_Displayer::getObject(const AISObjectPtr& theIO) const
 631 {
 632   Handle(AIS_InteractiveObject) aRefAIS = theIO->impl<Handle(AIS_InteractiveObject)>();
 633   return getObject(aRefAIS);
 634 }
 635
 636 ObjectPtr XGUI_Displayer::getObject(const Handle(AIS_InteractiveObject)& theIO) const
 637 {
 638   ObjectPtr aFeature;
 639   foreach (ObjectPtr anObj, myResult2AISObjectMap.keys()) {
 640     AISObjectPtr aAIS = myResult2AISObjectMap[anObj];
 641     Handle(AIS_InteractiveObject) anAIS = aAIS->impl<Handle(AIS_InteractiveObject)>();
 642     if (anAIS == theIO)
 643       return anObj;
 644   }
 645   return aFeature;
 646 }
 647
 648 bool XGUI_Displayer::enableUpdateViewer(const bool isEnabled)
 649 {
 650   bool aWasEnabled = myEnableUpdateViewer;
 651
 652   myEnableUpdateViewer = isEnabled;
 653
 654   return aWasEnabled;
 655 }
 656
 657 void XGUI_Displayer::updateViewer() const
 658 {
 659   Handle(AIS_InteractiveContext) aContext = AISContext();
 660   if (!aContext.IsNull() && myEnableUpdateViewer)
 661     aContext->UpdateCurrentViewer();
 662 }
 663
 664 Handle(AIS_InteractiveContext) XGUI_Displayer::AISContext() const
 665 {
 666   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
 667   if ((!aContext.IsNull()) && (!aContext->HasOpenedContext())) {
 668     aContext->OpenLocalContext();
 669   }
 670   return aContext;
 671 }
 672
 673 Handle(SelectMgr_AndFilter) XGUI_Displayer::GetFilter()
 674 {
 675   Handle(AIS_InteractiveContext) aContext = AISContext();
 676   if (myAndFilter.IsNull() && !aContext.IsNull()) {
 677     myAndFilter = new SelectMgr_AndFilter();
 678     aContext->AddFilter(myAndFilter);
 679   }
 680   return myAndFilter;
 681 }
 682
 683 void XGUI_Displayer::displayAIS(AISObjectPtr theAIS, bool isUpdate)
 684 {
 685   Handle(AIS_InteractiveContext) aContext = AISContext();
 686   if (aContext.IsNull())
 687     return;
 688   Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
 689   if (!anAISIO.IsNull()) {
 690     aContext->Display(anAISIO, isUpdate);
 691     if (aContext->HasOpenedContext()) {
 692       //if (myUseExternalObjects) {
 693         if (myActiveSelectionModes.size() == 0)
 694           aContext->Activate(anAISIO);
 695         else {
 696           foreach(int aMode, myActiveSelectionModes) {
 697             aContext->Activate(anAISIO, aMode);
 698           }
 699         }
 700       //}
 701     }
 702   }
 703 }
 704
 705 void XGUI_Displayer::eraseAIS(AISObjectPtr theAIS, const bool isUpdate)
 706 {
 707   Handle(AIS_InteractiveContext) aContext = AISContext();
 708   if (aContext.IsNull())
 709     return;
 710   Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
 711   if (!anAISIO.IsNull()) {
 712     aContext->Remove(anAISIO, isUpdate);
 713   }
 714 }
 715
 716
 717 void XGUI_Displayer::setDisplayMode(ObjectPtr theObject, DisplayMode theMode, bool toUpdate)
 718 {
 719   if (theMode == NoMode)
 720     return;
 721
 722   Handle(AIS_InteractiveContext) aContext = AISContext();
 723   if (aContext.IsNull())
 724     return;
 725
 726   AISObjectPtr aAISObj = getAISObject(theObject);
 727   if (!aAISObj)
 728     return;
 729
 730   Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
 731   bool aCanBeShaded = ::canBeShaded(aAISIO);
 732   // In order to avoid extra closing/opening context
 733   SelectMgr_IndexedMapOfOwner aSelectedOwners;
 734   if (aCanBeShaded) {
 735     myWorkshop->selector()->selection()->selectedOwners(aSelectedOwners);
 736     closeLocalContexts(false);
 737   }
 738   aContext->SetDisplayMode(aAISIO, theMode, false);
 739   // Redisplay in order to update new mode because it could be not computed before
 740   aContext->Redisplay(aAISIO, false);
 741   if (aCanBeShaded) {
 742     openLocalContext();
 743     activateObjects(myActiveSelectionModes);
 744     myWorkshop->selector()->setSelectedOwners(aSelectedOwners, false);
 745   }
 746   if (toUpdate)
 747     updateViewer();
 748 }
 749
 750 XGUI_Displayer::DisplayMode XGUI_Displayer::displayMode(ObjectPtr theObject) const
 751 {
 752   Handle(AIS_InteractiveContext) aContext = AISContext();
 753   if (aContext.IsNull())
 754     return NoMode;
 755
 756   AISObjectPtr aAISObj = getAISObject(theObject);
 757   if (!aAISObj)
 758     return NoMode;
 759
 760   Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
 761   return (XGUI_Displayer::DisplayMode) aAISIO->DisplayMode();
 762 }
 763
 764 void XGUI_Displayer::addSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
 765 {
 766   Handle(AIS_InteractiveContext) aContext = AISContext();
 767   if (aContext.IsNull())
 768     return;
 769   const SelectMgr_ListOfFilter& aFilters = aContext->Filters();
 770   SelectMgr_ListIteratorOfListOfFilter aIt(aFilters);
 771   for (; aIt.More(); aIt.Next()) {
 772     if (theFilter.Access() == aIt.Value().Access())
 773       return;
 774   }
 775   Handle(SelectMgr_CompositionFilter) aCompFilter = GetFilter();
 776   const SelectMgr_ListOfFilter& aStoredFilters = aCompFilter->StoredFilters();
 777   for (aIt.Initialize(aStoredFilters); aIt.More(); aIt.Next()) {
 778     if (theFilter.Access() == aIt.Value().Access())
 779       return;
 780   }
 781   aCompFilter->Add(theFilter);
 782 #ifdef DEBUG_SELECTION_FILTERS
 783   int aCount = GetFilter()->StoredFilters().Extent();
 784   qDebug(QString("addSelectionFilter: filters.count() = %1").arg(aCount).toStdString().c_str());
 785 #endif
 786 }
 787
 788 void XGUI_Displayer::removeSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
 789 {
 790   Handle(AIS_InteractiveContext) aContext = AISContext();
 791   if (aContext.IsNull())
 792     return;
 793   Handle(SelectMgr_AndFilter) aCompositeFilter = GetFilter();
 794   if (aCompositeFilter->IsIn(theFilter))
 795     aCompositeFilter->Remove(theFilter);
 796 #ifdef DEBUG_SELECTION_FILTERS
 797   int aCount = GetFilter()->StoredFilters().Extent();
 798   qDebug(QString("removeSelectionFilter: filters.count() = %1").arg(aCount).toStdString().c_str());
 799 #endif
 800 }
 801
 802 void XGUI_Displayer::removeFilters()
 803 {
 804   Handle(AIS_InteractiveContext) aContext = AISContext();
 805   if (aContext.IsNull())
 806     return;
 807   GetFilter()->Clear();
 808 }
 809
 810 void XGUI_Displayer::showOnly(const QObjectPtrList& theList)
 811 {
 812   QObjectPtrList aDispList = myResult2AISObjectMap.keys();
 813   foreach(ObjectPtr aObj, aDispList) {
 814     if (!theList.contains(aObj))
 815       erase(aObj, false);
 816   }
 817   foreach(ObjectPtr aObj, theList) {
 818     if (!isVisible(aObj))
 819       display(aObj, false);
 820   }
 821   updateViewer();
 822 }
 823
 824 bool XGUI_Displayer::canBeShaded(ObjectPtr theObject) const
 825 {
 826   if (!isVisible(theObject))
 827     return false;
 828
 829   AISObjectPtr aAISObj = getAISObject(theObject);
 830   if (aAISObj.get() == NULL)
 831     return false;
 832
 833   Handle(AIS_InteractiveObject) anAIS = aAISObj->impl<Handle(AIS_InteractiveObject)>();
 834   return ::canBeShaded(anAIS);
 835 }
 836
 837 void XGUI_Displayer::activate(const Handle(AIS_InteractiveObject)& theIO,
 838                               const QIntList& theModes) const
 839 {
 840   Handle(AIS_InteractiveContext) aContext = AISContext();
 841   if (aContext.IsNull() || theIO.IsNull())
 842     return;
 843
 844   // deactivate object in all modes, which are not in the list of activation
 845   // It seems that after the IO deactivation the selected state of the IO's owners
 846   // is modified in OCC(version: 6.8.0) and the selection of the object later is lost.
 847   // By this reason, the number of the IO deactivate is decreased and the object is deactivated
 848   // only if there is a difference in the current modes and the parameters modes.
 849   // If the selection problem happens again, it is possible to write a test scenario and create
 850   // a bug. The bug steps are the following:
 851   // Create two IO, activate them in 5 modes, select the first IO, deactivate 3 modes for both,
 852   // with clicked SHIFT select the second object. The result is the selection of the first IO is lost.
 853   TColStd_ListOfInteger aTColModes;
 854   aContext->ActivatedModes(theIO, aTColModes);
 855   TColStd_ListIteratorOfListOfInteger itr( aTColModes );
 856   QIntList aModesActivatedForIO;
 857   //bool isDeactivated = false;
 858   for (; itr.More(); itr.Next() ) {
 859     Standard_Integer aMode = itr.Value();
 860     if (!theModes.contains(aMode)) {
 861 #ifdef DEBUG_ACTIVATE
 862       qDebug(QString("deactivate: %1").arg(aMode).toStdString().c_str());
 863 #endif
 864       aContext->Deactivate(theIO, aMode);
 865       //isDeactivated = true;
 866     }
 867     else {
 868       aModesActivatedForIO.append(aMode);
 869 #ifdef DEBUG_ACTIVATE
 870       qDebug(QString("  active: %1").arg(aMode).toStdString().c_str());
 871 #endif
 872     }
 873   }
 874 #ifdef DEBUG_USE_CLEAR_OUTDATED_SELECTION
 875   if (isDeactivated) {
 876     aContext->LocalContext()->ClearOutdatedSelection(theIO, true);
 877     updateViewer();
 878   }
 879 #endif
 880   // loading the interactive object allowing the decomposition
 881   if (aTColModes.IsEmpty())
 882     aContext->Load(theIO, -1, true);
 883
 884   Handle(AIS_Trihedron) aTrihedron = Handle(AIS_Trihedron)::DownCast(theIO);
 885   //Deactivate trihedron which can be activated in local selector
 886   if (aTrihedron.IsNull()) {
 887     //aContext->Load(anAISIO, -1, true);
 888     // In order to clear active modes list
 889     if (theModes.size() == 0) {
 890       //aContext->Load(anAISIO, 0, true);
 891       aContext->Activate(theIO);
 892 #ifdef DEBUG_ACTIVATE
 893       qDebug("activate in all modes");
 894 #endif
 895     } else {
 896       foreach(int aMode, theModes) {
 897         //aContext->Load(anAISIO, aMode, true);
 898         if (!aModesActivatedForIO.contains(aMode)) {
 899           aContext->Activate(theIO, aMode);
 900 #ifdef DEBUG_ACTIVATE
 901           qDebug(QString("activate: %1").arg(aMode).toStdString().c_str());
 902 #endif
 903         }
 904       }
 905     }
 906   }
 907 }
 908
 909 bool XGUI_Displayer::customizeObject(ObjectPtr theObject)
 910 {
 911   AISObjectPtr anAISObj = getAISObject(theObject);
 912   // correct the result's color it it has the attribute
 913   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
 914
 915   // Customization of presentation
 916   GeomCustomPrsPtr aCustomPrs;
 917   FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
 918   if (aFeature.get() != NULL) {
 919     GeomCustomPrsPtr aCustPrs = std::dynamic_pointer_cast<GeomAPI_ICustomPrs>(aFeature);
 920     if (aCustPrs.get() != NULL)
 921       aCustomPrs = aCustPrs;
 922   }
 923   if (aCustomPrs.get() == NULL) {
 924     // we ignore presentable not customized objects
 925     GeomPresentablePtr aPrs = std::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
 926     if (aPrs.get() != NULL)
 927       return false;
 928     aCustomPrs = myCustomPrs;
 929   }
 930   return aCustomPrs->customisePresentation(aResult, anAISObj, myCustomPrs);
 931 }
 932
 933
 934 QColor XGUI_Displayer::setObjectColor(ObjectPtr theObject, const QColor& theColor, bool toUpdate)
 935 {
 936   if (!isVisible(theObject))
 937     return Qt::black;
 938
 939   AISObjectPtr anAISObj = getAISObject(theObject);
 940   int aR, aG, aB;
 941   anAISObj->getColor(aR, aG, aB);
 942   anAISObj->setColor(theColor.red(), theColor.green(), theColor.blue());
 943   if (toUpdate)
 944     updateViewer();
 945   return QColor(aR, aG, aB);
 946 }