1 // File: XGUI_Displayer.cpp
2 // Created: 20 Apr 2014
3 // Author: Natalia ERMOLAEVA
5 #include "XGUI_Displayer.h"
6 #include "XGUI_Viewer.h"
7 #include "XGUI_Workshop.h"
8 #include "XGUI_ViewerProxy.h"
10 #include <ModelAPI_Document.h>
11 #include <ModelAPI_Data.h>
12 #include <ModelAPI_Object.h>
13 #include <ModelAPI_Tools.h>
15 #include <ModuleBase_ResultPrs.h>
17 #include <GeomAPI_Shape.h>
18 #include <GeomAPI_IPresentable.h>
20 #include <AIS_InteractiveContext.hxx>
21 #include <AIS_LocalContext.hxx>
22 #include <AIS_ListOfInteractive.hxx>
23 #include <AIS_ListIteratorOfListOfInteractive.hxx>
24 #include <AIS_DimensionSelectionMode.hxx>
25 #include <AIS_Shape.hxx>
26 #include <AIS_Dimension.hxx>
27 #include <TColStd_ListIteratorOfListOfInteger.hxx>
28 #include <SelectMgr_ListOfFilter.hxx>
29 #include <SelectMgr_ListIteratorOfListOfFilter.hxx>
33 const int MOUSE_SENSITIVITY_IN_PIXEL = 10; ///< defines the local context mouse selection sensitivity
35 XGUI_Displayer::XGUI_Displayer(XGUI_Workshop* theWorkshop)
36 : myUseExternalObjects(false), myWorkshop(theWorkshop)
40 XGUI_Displayer::~XGUI_Displayer()
44 bool XGUI_Displayer::isVisible(ObjectPtr theObject) const
46 return myResult2AISObjectMap.contains(theObject);
49 void XGUI_Displayer::display(ObjectPtr theObject, bool isUpdateViewer)
51 if (isVisible(theObject)) {
52 redisplay(theObject, isUpdateViewer);
56 GeomPresentablePtr aPrs = std::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
57 bool isShading = false;
59 anAIS = aPrs->getAISObject(AISObjectPtr());
61 ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
63 std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(aResult);
65 anAIS = AISObjectPtr(new GeomAPI_AISObject());
66 anAIS->setImpl(new Handle(AIS_InteractiveObject)(new ModuleBase_ResultPrs(aResult)));
67 //anAIS->createShape(aShapePtr);
73 display(theObject, anAIS, isShading, isUpdateViewer);
77 void XGUI_Displayer::display(ObjectPtr theObject, AISObjectPtr theAIS,
78 bool isShading, bool isUpdateViewer)
80 Handle(AIS_InteractiveContext) aContext = AISContext();
81 if (aContext.IsNull())
84 Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
85 if (!anAISIO.IsNull()) {
86 myResult2AISObjectMap[theObject] = theAIS;
87 aContext->Display(anAISIO, false);
88 aContext->SetDisplayMode(anAISIO, isShading? Shading : Wireframe, isUpdateViewer);
89 if (aContext->HasOpenedContext()) {
90 if (myUseExternalObjects) {
91 if (myActiveSelectionModes.size() == 0)
92 aContext->Activate(anAISIO);
94 foreach(int aMode, myActiveSelectionModes) {
95 aContext->Activate(anAISIO, aMode);
103 void XGUI_Displayer::erase(ObjectPtr theObject, const bool isUpdateViewer)
105 if (!isVisible(theObject))
108 Handle(AIS_InteractiveContext) aContext = AISContext();
109 if (aContext.IsNull())
111 AISObjectPtr anObject = myResult2AISObjectMap[theObject];
113 Handle(AIS_InteractiveObject) anAIS = anObject->impl<Handle(AIS_InteractiveObject)>();
114 if (!anAIS.IsNull()) {
115 aContext->Remove(anAIS, isUpdateViewer);
118 myResult2AISObjectMap.remove(theObject);
121 void XGUI_Displayer::redisplay(ObjectPtr theObject, bool isUpdateViewer)
123 if (!isVisible(theObject))
126 AISObjectPtr aAISObj = getAISObject(theObject);
127 Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
129 GeomPresentablePtr aPrs = std::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
131 AISObjectPtr aAIS_Obj = aPrs->getAISObject(aAISObj);
133 erase(theObject, isUpdateViewer);
136 if (aAIS_Obj != aAISObj) {
137 myResult2AISObjectMap[theObject] = aAIS_Obj;
139 aAISIO = aAIS_Obj->impl<Handle(AIS_InteractiveObject)>();
142 if (!aAISIO.IsNull()) {
143 Handle(AIS_InteractiveContext) aContext = AISContext();
144 if (aContext.IsNull())
146 aContext->Redisplay(aAISIO, isUpdateViewer);
150 void XGUI_Displayer::deactivate(ObjectPtr theObject)
152 if (isVisible(theObject)) {
153 Handle(AIS_InteractiveContext) aContext = AISContext();
154 if (aContext.IsNull())
157 AISObjectPtr anObj = myResult2AISObjectMap[theObject];
158 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
159 aContext->Deactivate(anAIS);
163 void XGUI_Displayer::activate(ObjectPtr theFeature)
166 activate(theFeature, aModes);
169 void XGUI_Displayer::activate(ObjectPtr theObject, const QIntList& theModes)
171 if (isVisible(theObject)) {
172 Handle(AIS_InteractiveContext) aContext = AISContext();
173 if (aContext.IsNull())
176 AISObjectPtr anObj = myResult2AISObjectMap[theObject];
177 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
178 if (aContext->HasOpenedContext()) {
179 aContext->Load(anAIS, -1, true);
181 if (theModes.size() > 0) {
182 foreach(int aMode, theModes) {
183 aContext->Activate(anAIS, aMode);
186 aContext->Activate(anAIS);
190 bool XGUI_Displayer::isActive(ObjectPtr theObject) const
192 Handle(AIS_InteractiveContext) aContext = AISContext();
193 if (aContext.IsNull())
195 if (!isVisible(theObject))
198 AISObjectPtr anObj = myResult2AISObjectMap[theObject];
199 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
201 TColStd_ListOfInteger aModes;
202 aContext->ActivatedModes(anAIS, aModes);
203 return aModes.Extent() > 0;
206 void XGUI_Displayer::stopSelection(const QObjectPtrList& theResults, const bool isStop,
207 const bool isUpdateViewer)
209 Handle(AIS_InteractiveContext) aContext = AISContext();
210 if (aContext.IsNull())
213 Handle(AIS_Shape) anAIS;
214 QObjectPtrList::const_iterator anIt = theResults.begin(), aLast = theResults.end();
216 for (; anIt != aLast; anIt++) {
218 if (isVisible(aFeature))
219 anAIS = Handle(AIS_Shape)::DownCast(
220 myResult2AISObjectMap[aFeature]->impl<Handle(AIS_InteractiveObject)>());
225 QColor aColor(Qt::white);
227 Quantity_Color(aColor.red() / 255., aColor.green() / 255., aColor.blue() / 255.,
231 QColor aColor(Qt::red);
233 Quantity_Color(aColor.red() / 255., aColor.green() / 255., aColor.blue() / 255.,
242 void XGUI_Displayer::setSelected(const QObjectPtrList& theResults, const bool isUpdateViewer)
244 Handle(AIS_InteractiveContext) aContext = AISContext();
245 if (aContext.IsNull())
247 if (aContext->HasOpenedContext()) {
248 aContext->UnhilightSelected();
249 aContext->ClearSelected();
250 foreach(ObjectPtr aResult, theResults) {
251 if (isVisible(aResult)) {
252 AISObjectPtr anObj = myResult2AISObjectMap[aResult];
253 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
255 aContext->SetSelected(anAIS, false);
259 aContext->UnhilightCurrents();
260 aContext->ClearCurrents();
261 foreach(ObjectPtr aResult, theResults) {
262 if (isVisible(aResult)) {
263 AISObjectPtr anObj = myResult2AISObjectMap[aResult];
264 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
266 aContext->SetCurrentObject(anAIS, false);
275 void XGUI_Displayer::clearSelected()
277 Handle(AIS_InteractiveContext) aContext = AISContext();
279 aContext->UnhilightCurrents(false);
280 aContext->ClearSelected();
284 void XGUI_Displayer::eraseAll(const bool isUpdateViewer)
286 Handle(AIS_InteractiveContext) aContext = AISContext();
287 if (aContext.IsNull())
290 foreach (AISObjectPtr aAISObj, myResult2AISObjectMap) {
292 Handle(AIS_InteractiveObject) anIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
294 aContext->Remove(anIO, false);
296 myResult2AISObjectMap.clear();
301 void XGUI_Displayer::eraseDeletedResults(const bool isUpdateViewer)
303 Handle(AIS_InteractiveContext) aContext = AISContext();
304 if (aContext.IsNull())
307 QObjectPtrList aRemoved;
308 foreach (ObjectPtr aFeature, myResult2AISObjectMap.keys()) {
309 if (!aFeature || !aFeature->data() || !aFeature->data()->isValid()) {
310 AISObjectPtr anObj = myResult2AISObjectMap[aFeature];
313 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
314 if (!anAIS.IsNull()) {
315 aContext->Remove(anAIS, false);
316 aRemoved.append(aFeature);
320 foreach(ObjectPtr aObj, aRemoved) {
321 myResult2AISObjectMap.remove(aObj);
328 void XGUI_Displayer::openLocalContext()
330 Handle(AIS_InteractiveContext) aContext = AISContext();
331 if (aContext.IsNull())
333 // Open local context if there is no one
334 if (!aContext->HasOpenedContext()) {
335 // Preserve selected objects
336 //AIS_ListOfInteractive aAisList;
337 //for (aContext->InitCurrent(); aContext->MoreCurrent(); aContext->NextCurrent())
338 // aAisList.Append(aContext->Current());
340 // get the filters from the global context and append them to the local context
341 // a list of filters in the global context is not cleared and should be cleared here
342 SelectMgr_ListOfFilter aFilters;
343 aFilters.Assign(aContext->Filters());
344 // it is important to remove the filters in the global context, because there is a code
345 // in the closeLocalContex, which restore the global context filters
346 aContext->RemoveFilters();
348 aContext->ClearCurrents();
349 aContext->OpenLocalContext();
350 aContext->NotUseDisplayedObjects();
352 myUseExternalObjects = false;
353 myActiveSelectionModes.clear();
355 SelectMgr_ListIteratorOfListOfFilter aIt(aFilters);
356 for (;aIt.More(); aIt.Next()) {
357 aContext->AddFilter(aIt.Value());
360 //AIS_ListIteratorOfListOfInteractive aIt2(aAisList);
361 //for(; aIt2.More(); aIt2.Next()) {
362 // aContext->SetSelected(aIt2.Value(), false);
367 void XGUI_Displayer::closeLocalContexts(const bool isUpdateViewer)
369 Handle(AIS_InteractiveContext) aContext = AISContext();
370 if ( (!aContext.IsNull()) && (aContext->HasOpenedContext()) ) {
371 // Preserve selected objects
372 //AIS_ListOfInteractive aAisList;
373 //for (aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected())
374 // aAisList.Append(aContext->SelectedInteractive());
376 // get the filters from the local context and append them to the global context
377 // a list of filters in the local context is cleared
378 SelectMgr_ListOfFilter aFilters;
379 aFilters.Assign(aContext->Filters());
381 aContext->ClearSelected();
382 aContext->CloseAllContexts(false);
384 // Redisplay all object if they were displayed in localContext
385 Handle(AIS_InteractiveObject) aAISIO;
386 foreach (AISObjectPtr aAIS, myResult2AISObjectMap) {
387 aAISIO = aAIS->impl<Handle(AIS_InteractiveObject)>();
388 if (aContext->DisplayStatus(aAISIO) != AIS_DS_Displayed) {
389 aContext->Display(aAISIO, false);
390 aContext->SetDisplayMode(aAISIO, Shading, false);
394 // Append the filters from the local selection in the global selection context
395 SelectMgr_ListIteratorOfListOfFilter aIt(aFilters);
396 for (;aIt.More(); aIt.Next()) {
397 Handle(SelectMgr_Filter) aFilter = aIt.Value();
398 aContext->AddFilter(aFilter);
403 myUseExternalObjects = false;
404 myActiveSelectionModes.clear();
407 //AIS_ListIteratorOfListOfInteractive aIt2(aAisList);
408 //for(; aIt2.More(); aIt2.Next()) {
409 // if (aContext->IsDisplayed(aIt2.Value()))
410 // aContext->SetCurrentObject(aIt2.Value(), false);
415 AISObjectPtr XGUI_Displayer::getAISObject(ObjectPtr theObject) const
418 if (myResult2AISObjectMap.contains(theObject))
419 anIO = myResult2AISObjectMap[theObject];
423 ObjectPtr XGUI_Displayer::getObject(const AISObjectPtr& theIO) const
425 Handle(AIS_InteractiveObject) aRefAIS = theIO->impl<Handle(AIS_InteractiveObject)>();
426 return getObject(aRefAIS);
429 ObjectPtr XGUI_Displayer::getObject(const Handle(AIS_InteractiveObject)& theIO) const
432 foreach (ObjectPtr anObj, myResult2AISObjectMap.keys()) {
433 AISObjectPtr aAIS = myResult2AISObjectMap[anObj];
434 Handle(AIS_InteractiveObject) anAIS = aAIS->impl<Handle(AIS_InteractiveObject)>();
441 void XGUI_Displayer::updateViewer()
443 Handle(AIS_InteractiveContext) aContext = AISContext();
444 if (!aContext.IsNull())
445 aContext->UpdateCurrentViewer();
448 Handle(AIS_InteractiveContext) XGUI_Displayer::AISContext() const
450 return myWorkshop->viewer()->AISContext();
453 Handle(SelectMgr_AndFilter) XGUI_Displayer::GetFilter()
455 Handle(AIS_InteractiveContext) aContext = AISContext();
456 if (myAndFilter.IsNull() && !aContext.IsNull()) {
457 myAndFilter = new SelectMgr_AndFilter();
458 aContext->AddFilter(myAndFilter);
463 void XGUI_Displayer::displayAIS(AISObjectPtr theAIS, bool isUpdate)
465 Handle(AIS_InteractiveContext) aContext = AISContext();
466 Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
467 if (!anAISIO.IsNull()) {
468 aContext->Display(anAISIO, isUpdate);
469 if (aContext->HasOpenedContext()) {
470 if (myUseExternalObjects) {
471 if (myActiveSelectionModes.size() == 0)
472 aContext->Activate(anAISIO);
474 foreach(int aMode, myActiveSelectionModes) {
475 aContext->Activate(anAISIO, aMode);
483 void XGUI_Displayer::eraseAIS(AISObjectPtr theAIS, const bool isUpdate)
485 Handle(AIS_InteractiveContext) aContext = AISContext();
486 Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
487 if (!anAISIO.IsNull()) {
488 aContext->Remove(anAISIO, isUpdate);
492 void XGUI_Displayer::activateObjects(const QIntList& theModes)
494 Handle(AIS_InteractiveContext) aContext = AISContext();
495 // Open local context if there is no one
496 if (!aContext->HasOpenedContext())
499 aContext->UseDisplayedObjects();
500 myUseExternalObjects = true;
501 myActiveSelectionModes = theModes;
503 //Deactivate trihedron which can be activated in local selector
504 AIS_ListOfInteractive aPrsList;
505 aContext->DisplayedObjects(aPrsList, true);
507 Handle(AIS_Trihedron) aTrihedron;
508 AIS_ListIteratorOfListOfInteractive aLIt(aPrsList);
509 for(; aLIt.More(); aLIt.Next()){
510 aTrihedron = Handle(AIS_Trihedron)::DownCast(aLIt.Value());
511 if (!aTrihedron.IsNull()) {
512 aContext->Deactivate(aTrihedron);
517 //Activate all displayed objects with the module modes
518 //AIS_ListOfInteractive aPrsList;
519 //aContext->DisplayedObjects(aPrsList, true);
521 //AIS_ListIteratorOfListOfInteractive aLIt(aPrsList);
522 Handle(AIS_InteractiveObject) anAISIO;
523 for(aLIt.Initialize(aPrsList); aLIt.More(); aLIt.Next()){
524 anAISIO = aLIt.Value();
525 aTrihedron = Handle(AIS_Trihedron)::DownCast(anAISIO);
526 if (!aTrihedron.IsNull())
529 aContext->Load(anAISIO, -1, true);
530 if (theModes.size() == 0)
531 aContext->Activate(anAISIO);
533 foreach(int aMode, theModes) {
534 aContext->Activate(anAISIO, aMode);
541 void XGUI_Displayer::deactivateObjects()
543 Handle(AIS_InteractiveContext) aContext = AISContext();
544 // Open local context if there is no one
545 if (!aContext->HasOpenedContext())
548 aContext->NotUseDisplayedObjects();
552 void XGUI_Displayer::setDisplayMode(ObjectPtr theObject, DisplayMode theMode, bool toUpdate)
554 if (theMode == NoMode)
557 Handle(AIS_InteractiveContext) aContext = AISContext();
558 if (aContext.IsNull())
561 AISObjectPtr aAISObj = getAISObject(theObject);
565 Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
566 aContext->SetDisplayMode(aAISIO, theMode, toUpdate);
569 XGUI_Displayer::DisplayMode XGUI_Displayer::displayMode(ObjectPtr theObject) const
571 Handle(AIS_InteractiveContext) aContext = AISContext();
572 if (aContext.IsNull())
575 AISObjectPtr aAISObj = getAISObject(theObject);
579 Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
580 return (XGUI_Displayer::DisplayMode) aAISIO->DisplayMode();
583 void XGUI_Displayer::addSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
585 Handle(AIS_InteractiveContext) aContext = AISContext();
586 if (aContext.IsNull())
588 const SelectMgr_ListOfFilter& aFilters = aContext->Filters();
589 SelectMgr_ListIteratorOfListOfFilter aIt(aFilters);
590 for (; aIt.More(); aIt.Next()) {
591 if (theFilter.Access() == aIt.Value().Access())
594 GetFilter()->Add(theFilter);
597 void XGUI_Displayer::removeSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
599 Handle(AIS_InteractiveContext) aContext = AISContext();
600 if (aContext.IsNull())
602 GetFilter()->Remove(theFilter);
605 void XGUI_Displayer::removeFilters()
607 Handle(AIS_InteractiveContext) aContext = AISContext();
608 if (aContext.IsNull())
610 GetFilter()->Clear();
613 void XGUI_Displayer::showOnly(const QObjectPtrList& theList)
615 QObjectPtrList aDispList = myResult2AISObjectMap.keys();
616 foreach(ObjectPtr aObj, aDispList) {
617 if (!theList.contains(aObj))
620 foreach(ObjectPtr aObj, theList) {
621 if (!isVisible(aObj))
622 display(aObj, false);