Salome HOME
83fa8010dd1dce83aedec88d74947ef59887dbd6
[modules/shaper.git] / 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  QList<ModuleBase_ViewerPrs>& theValues, 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     //if (aSelected.size() > 0) {
428     foreach (ModuleBase_ViewerPrs aPrs, theValues) {
429       const TopoDS_Shape& aShape = aPrs.shape();
430       if (!aShape.IsNull()) {
431         aContext->AddOrRemoveSelected(aShape, false);
432       } else {
433         ObjectPtr anObject = aPrs.object();
434         ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
435         if (aResult.get() && isVisible(aResult)) {
436           AISObjectPtr anObj = myResult2AISObjectMap[aResult];
437           Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
438           if (!anAIS.IsNull()) {
439             // The methods are replaced in order to provide multi-selection, e.g. restore selection
440             // by activating multi selector widget. It also gives an advantage that the multi
441             // selection in OB gives multi-selection in the viewer
442             //aContext->SetSelected(anAIS, false);
443             // The selection in the context was cleared, so the method sets the objects are selected
444             aContext->AddOrRemoveSelected(anAIS, false);
445           }
446         }
447       }
448     }
449   } else {
450     aContext->UnhilightCurrents();
451     aContext->ClearCurrents();
452     foreach (ModuleBase_ViewerPrs aPrs, theValues) {
453       ObjectPtr anObject = aPrs.object();
454       ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObject);
455       if (aResult.get() && isVisible(aResult)) {
456         AISObjectPtr anObj = myResult2AISObjectMap[aResult];
457         Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
458         if (!anAIS.IsNull())
459           aContext->SetCurrentObject(anAIS, false);
460       }
461     }
462   }
463   if (isUpdateViewer)
464     updateViewer();
465 }
466
467 void XGUI_Displayer::clearSelected()
468 {
469   Handle(AIS_InteractiveContext) aContext = AISContext();
470   if (aContext) {
471     aContext->UnhilightCurrents(false);
472     aContext->ClearSelected();
473   }
474 }
475
476 void XGUI_Displayer::eraseAll(const bool isUpdateViewer)
477 {
478   Handle(AIS_InteractiveContext) aContext = AISContext();
479   if (!aContext.IsNull()) {
480     foreach (ObjectPtr aObj, myResult2AISObjectMap.keys()) {
481       AISObjectPtr aAISObj = myResult2AISObjectMap[aObj];
482       // erase an object
483       Handle(AIS_InteractiveObject) anIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
484       if (!anIO.IsNull()) {
485         emit beforeObjectErase(aObj, aAISObj);
486         aContext->Remove(anIO, false);
487       }
488     }
489     if (isUpdateViewer)
490       updateViewer();
491   }
492   myResult2AISObjectMap.clear();
493 }
494
495 void XGUI_Displayer::openLocalContext()
496 {
497   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
498   if (aContext.IsNull())
499     return;
500   // Open local context if there is no one
501   if (!aContext->HasOpenedContext()) {
502     // Preserve selected objects
503     //AIS_ListOfInteractive aAisList;
504     //for (aContext->InitCurrent(); aContext->MoreCurrent(); aContext->NextCurrent())
505     //  aAisList.Append(aContext->Current());
506
507     // get the filters from the global context and append them to the local context
508     // a list of filters in the global context is not cleared and should be cleared here
509     SelectMgr_ListOfFilter aFilters;
510     aFilters.Assign(aContext->Filters());
511     // it is important to remove the filters in the global context, because there is a code
512     // in the closeLocalContex, which restore the global context filters
513     aContext->RemoveFilters();
514
515     //aContext->ClearCurrents();
516     aContext->OpenLocalContext();
517     //aContext->NotUseDisplayedObjects();
518
519     //myUseExternalObjects = false;
520
521     SelectMgr_ListIteratorOfListOfFilter aIt(aFilters);
522     for (;aIt.More(); aIt.Next()) {
523       aContext->AddFilter(aIt.Value());
524     }
525     // Restore selection
526     //AIS_ListIteratorOfListOfInteractive aIt2(aAisList);
527     //for(; aIt2.More(); aIt2.Next()) {
528     //  aContext->SetSelected(aIt2.Value(), false);
529     //}
530   }
531 }
532
533 void XGUI_Displayer::closeLocalContexts(const bool isUpdateViewer)
534 {
535   Handle(AIS_InteractiveContext) aContext = AISContext();
536   if ( (!aContext.IsNull()) && (aContext->HasOpenedContext()) ) {
537     // Preserve selected objects
538     //AIS_ListOfInteractive aAisList;
539     //for (aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected())
540     //  aAisList.Append(aContext->SelectedInteractive());
541
542     // get the filters from the local context and append them to the global context
543     // a list of filters in the local context is cleared
544     SelectMgr_ListOfFilter aFilters;
545     aFilters.Assign(aContext->Filters());
546
547     //aContext->ClearSelected();
548     aContext->CloseAllContexts(false);
549
550     // Redisplay all object if they were displayed in localContext
551     Handle(AIS_InteractiveObject) aAISIO;
552     foreach (AISObjectPtr aAIS, myResult2AISObjectMap) {
553       aAISIO = aAIS->impl<Handle(AIS_InteractiveObject)>();
554       if (aContext->DisplayStatus(aAISIO) != AIS_DS_Displayed) {
555         aContext->Display(aAISIO, false);
556         aContext->SetDisplayMode(aAISIO, Shading, false);
557       }
558     }
559
560     // Append the filters from the local selection in the global selection context
561     SelectMgr_ListIteratorOfListOfFilter aIt(aFilters);
562     for (;aIt.More(); aIt.Next()) {
563       Handle(SelectMgr_Filter) aFilter = aIt.Value();
564       aContext->AddFilter(aFilter);
565     }
566
567     if (isUpdateViewer)
568       updateViewer();
569     //myUseExternalObjects = false;
570
571     // Restore selection
572     //AIS_ListIteratorOfListOfInteractive aIt2(aAisList);
573     //for(; aIt2.More(); aIt2.Next()) {
574     //  if (aContext->IsDisplayed(aIt2.Value()))
575     //    aContext->SetCurrentObject(aIt2.Value(), false);
576     //}
577   }
578 }
579
580 AISObjectPtr XGUI_Displayer::getAISObject(ObjectPtr theObject) const
581 {
582   AISObjectPtr anIO;
583   if (myResult2AISObjectMap.contains(theObject))
584     anIO = myResult2AISObjectMap[theObject];
585   return anIO;
586 }
587
588 ObjectPtr XGUI_Displayer::getObject(const AISObjectPtr& theIO) const
589 {
590   Handle(AIS_InteractiveObject) aRefAIS = theIO->impl<Handle(AIS_InteractiveObject)>();
591   return getObject(aRefAIS);
592 }
593
594 ObjectPtr XGUI_Displayer::getObject(const Handle(AIS_InteractiveObject)& theIO) const
595 {
596   ObjectPtr aFeature;
597   foreach (ObjectPtr anObj, myResult2AISObjectMap.keys()) {
598     AISObjectPtr aAIS = myResult2AISObjectMap[anObj];
599     Handle(AIS_InteractiveObject) anAIS = aAIS->impl<Handle(AIS_InteractiveObject)>();
600     if (anAIS == theIO)
601       return anObj;
602   }
603   return aFeature;
604 }
605
606 bool XGUI_Displayer::enableUpdateViewer(const bool isEnabled)
607 {
608   bool aWasEnabled = myEnableUpdateViewer;
609
610   myEnableUpdateViewer = isEnabled;
611
612   return aWasEnabled;
613 }
614
615 void XGUI_Displayer::updateViewer() const
616 {
617   Handle(AIS_InteractiveContext) aContext = AISContext();
618   if (!aContext.IsNull() && myEnableUpdateViewer)
619     aContext->UpdateCurrentViewer();
620 }
621
622 Handle(AIS_InteractiveContext) XGUI_Displayer::AISContext() const
623 {
624   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
625   if ((!aContext.IsNull()) && (!aContext->HasOpenedContext())) {
626     aContext->OpenLocalContext();
627   }
628   return aContext;
629 }
630
631 Handle(SelectMgr_AndFilter) XGUI_Displayer::GetFilter()
632 {
633   Handle(AIS_InteractiveContext) aContext = AISContext();
634   if (myAndFilter.IsNull() && !aContext.IsNull()) {
635     myAndFilter = new SelectMgr_AndFilter();
636     aContext->AddFilter(myAndFilter);
637   }
638   return myAndFilter;
639 }
640
641 void XGUI_Displayer::displayAIS(AISObjectPtr theAIS, bool isUpdate)
642 {
643   Handle(AIS_InteractiveContext) aContext = AISContext();
644   if (aContext.IsNull())
645     return;
646   Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
647   if (!anAISIO.IsNull()) {
648     aContext->Display(anAISIO, isUpdate);
649     if (aContext->HasOpenedContext()) {
650       //if (myUseExternalObjects) {
651         if (myActiveSelectionModes.size() == 0)
652           aContext->Activate(anAISIO);
653         else {
654           foreach(int aMode, myActiveSelectionModes) {
655             aContext->Activate(anAISIO, aMode);
656           }
657         }
658       //}
659     }
660   }
661 }
662
663 void XGUI_Displayer::eraseAIS(AISObjectPtr theAIS, const bool isUpdate)
664 {
665   Handle(AIS_InteractiveContext) aContext = AISContext();
666   if (aContext.IsNull())
667     return;
668   Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
669   if (!anAISIO.IsNull()) {
670     aContext->Remove(anAISIO, isUpdate);
671   }
672 }
673
674
675 void XGUI_Displayer::setDisplayMode(ObjectPtr theObject, DisplayMode theMode, bool toUpdate)
676 {
677   if (theMode == NoMode)
678     return;
679
680   Handle(AIS_InteractiveContext) aContext = AISContext();
681   if (aContext.IsNull())
682     return;
683
684   AISObjectPtr aAISObj = getAISObject(theObject);
685   if (!aAISObj)
686     return;
687
688   Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
689   bool aCanBeShaded = ::canBeShaded(aAISIO);
690   // In order to avoid extra closing/opening context
691   SelectMgr_IndexedMapOfOwner aSelectedOwners;
692   if (aCanBeShaded) {
693     myWorkshop->selector()->selection()->selectedOwners(aSelectedOwners);
694     closeLocalContexts(false);
695   }
696   aContext->SetDisplayMode(aAISIO, theMode, false);
697   // Redisplay in order to update new mode because it could be not computed before
698   aContext->Redisplay(aAISIO, false);
699   if (aCanBeShaded) {
700     openLocalContext();
701     activateObjects(myActiveSelectionModes);
702     myWorkshop->selector()->setSelectedOwners(aSelectedOwners, false);
703   }
704   if (toUpdate)
705     updateViewer();
706 }
707
708 XGUI_Displayer::DisplayMode XGUI_Displayer::displayMode(ObjectPtr theObject) const
709 {
710   Handle(AIS_InteractiveContext) aContext = AISContext();
711   if (aContext.IsNull())
712     return NoMode;
713
714   AISObjectPtr aAISObj = getAISObject(theObject);
715   if (!aAISObj)
716     return NoMode;
717
718   Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
719   return (XGUI_Displayer::DisplayMode) aAISIO->DisplayMode();
720 }
721
722 void XGUI_Displayer::addSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
723 {
724   Handle(AIS_InteractiveContext) aContext = AISContext();
725   if (aContext.IsNull())
726     return;
727   const SelectMgr_ListOfFilter& aFilters = aContext->Filters();
728   SelectMgr_ListIteratorOfListOfFilter aIt(aFilters);
729   for (; aIt.More(); aIt.Next()) {
730     if (theFilter.Access() == aIt.Value().Access())
731       return;
732   }
733   Handle(SelectMgr_CompositionFilter) aCompFilter = GetFilter();
734   const SelectMgr_ListOfFilter& aStoredFilters = aCompFilter->StoredFilters();
735   for (aIt.Initialize(aStoredFilters); aIt.More(); aIt.Next()) {
736     if (theFilter.Access() == aIt.Value().Access())
737       return;
738   }
739   aCompFilter->Add(theFilter);
740 #ifdef DEBUG_SELECTION_FILTERS
741   int aCount = GetFilter()->StoredFilters().Extent();
742   qDebug(QString("addSelectionFilter: filters.count() = %1").arg(aCount).toStdString().c_str());
743 #endif
744 }
745
746 void XGUI_Displayer::removeSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
747 {
748   Handle(AIS_InteractiveContext) aContext = AISContext();
749   if (aContext.IsNull())
750     return;
751   Handle(SelectMgr_AndFilter) aCompositeFilter = GetFilter();
752   if (aCompositeFilter->IsIn(theFilter))
753     aCompositeFilter->Remove(theFilter);
754 #ifdef DEBUG_SELECTION_FILTERS
755   int aCount = GetFilter()->StoredFilters().Extent();
756   qDebug(QString("removeSelectionFilter: filters.count() = %1").arg(aCount).toStdString().c_str());
757 #endif
758 }
759
760 void XGUI_Displayer::removeFilters()
761 {
762   Handle(AIS_InteractiveContext) aContext = AISContext();
763   if (aContext.IsNull())
764     return;
765   GetFilter()->Clear();
766 }
767
768 void XGUI_Displayer::showOnly(const QObjectPtrList& theList)
769 {
770   QObjectPtrList aDispList = myResult2AISObjectMap.keys();
771   foreach(ObjectPtr aObj, aDispList) {
772     if (!theList.contains(aObj))
773       erase(aObj, false);
774   }
775   foreach(ObjectPtr aObj, theList) {
776     if (!isVisible(aObj))
777       display(aObj, false);
778   }
779   updateViewer();
780 }
781
782 bool XGUI_Displayer::canBeShaded(ObjectPtr theObject) const
783
784   if (!isVisible(theObject))
785     return false;
786
787   AISObjectPtr aAISObj = getAISObject(theObject);
788   if (aAISObj.get() == NULL)
789     return false;
790
791   Handle(AIS_InteractiveObject) anAIS = aAISObj->impl<Handle(AIS_InteractiveObject)>();
792   return ::canBeShaded(anAIS);
793 }
794
795 void XGUI_Displayer::activate(const Handle(AIS_InteractiveObject)& theIO,
796                               const QIntList& theModes) const
797 {
798   Handle(AIS_InteractiveContext) aContext = AISContext();
799   if (aContext.IsNull() || theIO.IsNull())
800     return;
801
802   // deactivate object in all modes, which are not in the list of activation
803   // It seems that after the IO deactivation the selected state of the IO's owners
804   // is modified in OCC(version: 6.8.0) and the selection of the object later is lost.
805   // By this reason, the number of the IO deactivate is decreased and the object is deactivated
806   // only if there is a difference in the current modes and the parameters modes.
807   // If the selection problem happens again, it is possible to write a test scenario and create
808   // a bug. The bug steps are the following:
809   // Create two IO, activate them in 5 modes, select the first IO, deactivate 3 modes for both,
810   // with clicked SHIFT select the second object. The result is the selection of the first IO is lost.
811   TColStd_ListOfInteger aTColModes;
812   aContext->ActivatedModes(theIO, aTColModes);
813   TColStd_ListIteratorOfListOfInteger itr( aTColModes );
814   QIntList aModesActivatedForIO;
815   //bool isDeactivated = false;
816   for (; itr.More(); itr.Next() ) {
817     Standard_Integer aMode = itr.Value();
818     if (!theModes.contains(aMode)) {
819 #ifdef DEBUG_ACTIVATE
820       qDebug(QString("deactivate: %1").arg(aMode).toStdString().c_str());
821 #endif
822       aContext->Deactivate(theIO, aMode);
823       //isDeactivated = true;
824     }
825     else {
826       aModesActivatedForIO.append(aMode);
827 #ifdef DEBUG_ACTIVATE
828       qDebug(QString("  active: %1").arg(aMode).toStdString().c_str());
829 #endif
830     }
831   }
832 #ifdef DEBUG_USE_CLEAR_OUTDATED_SELECTION
833   if (isDeactivated) {
834     aContext->LocalContext()->ClearOutdatedSelection(theIO, true);
835     updateViewer();
836   }
837 #endif
838   // loading the interactive object allowing the decomposition
839   if (aTColModes.IsEmpty())
840     aContext->Load(theIO, -1, true);
841
842   Handle(AIS_Trihedron) aTrihedron = Handle(AIS_Trihedron)::DownCast(theIO);
843   //Deactivate trihedron which can be activated in local selector
844   if (aTrihedron.IsNull()) {
845     //aContext->Load(anAISIO, -1, true);
846     // In order to clear active modes list
847     if (theModes.size() == 0) {
848       //aContext->Load(anAISIO, 0, true);
849       aContext->Activate(theIO);
850 #ifdef DEBUG_ACTIVATE
851       qDebug("activate in all modes");
852 #endif
853     } else {
854       foreach(int aMode, theModes) {
855         //aContext->Load(anAISIO, aMode, true);
856         if (!aModesActivatedForIO.contains(aMode)) {
857           aContext->Activate(theIO, aMode);
858 #ifdef DEBUG_ACTIVATE
859           qDebug(QString("activate: %1").arg(aMode).toStdString().c_str());
860 #endif
861         }
862       }
863     }
864   }
865 }
866
867 bool XGUI_Displayer::customizeObject(ObjectPtr theObject)
868 {
869   AISObjectPtr anAISObj = getAISObject(theObject);
870   // correct the result's color it it has the attribute
871   ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
872
873   // Customization of presentation
874   GeomCustomPrsPtr aCustomPrs;
875   FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
876   if (aFeature.get() != NULL) {
877     GeomCustomPrsPtr aCustPrs = std::dynamic_pointer_cast<GeomAPI_ICustomPrs>(aFeature);
878     if (aCustPrs.get() != NULL)
879       aCustomPrs = aCustPrs;
880   }
881   if (aCustomPrs.get() == NULL) {
882     // we ignore presentable not customized objects
883     GeomPresentablePtr aPrs = std::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
884     if (aPrs.get() != NULL)
885       return false;
886     aCustomPrs = myCustomPrs;
887   }
888   return aCustomPrs->customisePresentation(aResult, anAISObj, myCustomPrs);
889 }
890
891
892 QColor XGUI_Displayer::setObjectColor(ObjectPtr theObject, const QColor& theColor, bool toUpdate)
893 {
894   if (!isVisible(theObject))
895     return Qt::black;
896
897   AISObjectPtr anAISObj = getAISObject(theObject);
898   int aR, aG, aB;
899   anAISObj->getColor(aR, aG, aB);
900   anAISObj->setColor(theColor.red(), theColor.green(), theColor.blue());
901   if (toUpdate)
902     updateViewer();
903   return QColor(aR, aG, aB);
904 }