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>
31 const int MOUSE_SENSITIVITY_IN_PIXEL = 10; ///< defines the local context mouse selection sensitivity
33 XGUI_Displayer::XGUI_Displayer(XGUI_Workshop* theWorkshop)
35 myWorkshop = theWorkshop;
38 XGUI_Displayer::~XGUI_Displayer()
42 bool XGUI_Displayer::isVisible(ObjectPtr theObject) const
44 return myResult2AISObjectMap.find(theObject) != myResult2AISObjectMap.end();
47 void XGUI_Displayer::display(ObjectPtr theObject, bool isUpdateViewer)
49 if (isVisible(theObject)) {
50 redisplay(theObject, isUpdateViewer);
54 GeomPresentablePtr aPrs = boost::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
55 bool isShading = false;
57 anAIS = aPrs->getAISObject(AISObjectPtr());
59 ResultPtr aResult = boost::dynamic_pointer_cast<ModelAPI_Result>(theObject);
61 boost::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(aResult);
63 anAIS = AISObjectPtr(new GeomAPI_AISObject());
64 anAIS->setImpl(new Handle(AIS_InteractiveObject)(new ModuleBase_ResultPrs(aResult)));
65 //anAIS->createShape(aShapePtr);
71 display(theObject, anAIS, isShading, isUpdateViewer);
75 void XGUI_Displayer::display(ObjectPtr theObject, AISObjectPtr theAIS,
76 bool isShading, bool isUpdateViewer)
78 Handle(AIS_InteractiveContext) aContext = AISContext();
79 if (aContext.IsNull())
82 Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
83 if (!anAISIO.IsNull()) {
84 myResult2AISObjectMap[theObject] = theAIS;
85 aContext->Display(anAISIO, false);
86 aContext->SetDisplayMode(anAISIO, isShading? Shading : Wireframe, isUpdateViewer);
90 void XGUI_Displayer::erase(ObjectPtr theObject, const bool isUpdateViewer)
92 if (!isVisible(theObject))
95 Handle(AIS_InteractiveContext) aContext = AISContext();
96 if (aContext.IsNull())
98 AISObjectPtr anObject = myResult2AISObjectMap[theObject];
100 Handle(AIS_InteractiveObject) anAIS = anObject->impl<Handle(AIS_InteractiveObject)>();
101 if (!anAIS.IsNull()) {
102 aContext->Remove(anAIS, isUpdateViewer);
105 myResult2AISObjectMap.erase(theObject);
108 void XGUI_Displayer::redisplay(ObjectPtr theObject, bool isUpdateViewer)
110 if (!isVisible(theObject))
113 Handle(AIS_InteractiveObject) aAISIO;
114 AISObjectPtr aAISObj = getAISObject(theObject);
115 GeomPresentablePtr aPrs = boost::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
117 AISObjectPtr aAIS_Obj = aPrs->getAISObject(aAISObj);
118 if (aAISObj && !aAIS_Obj) {
119 erase(theObject, isUpdateViewer);
122 aAISIO = aAIS_Obj->impl<Handle(AIS_InteractiveObject)>();
124 ResultPtr aResult = boost::dynamic_pointer_cast<ModelAPI_Result>(theObject);
126 boost::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(aResult);
128 Handle(AIS_Shape) aAISShape = Handle(AIS_Shape)::DownCast(
129 aAISObj->impl<Handle(AIS_InteractiveObject)>());
130 if (!aAISShape.IsNull()) {
131 aAISShape->Set(aShapePtr->impl<TopoDS_Shape>());
137 if (!aAISIO.IsNull()) {
138 Handle(AIS_InteractiveContext) aContext = AISContext();
139 if (aContext.IsNull())
141 aContext->Redisplay(aAISIO, isUpdateViewer);
142 //if (aContext->HasOpenedContext()) {
143 // aContext->Load(aAISIO, -1, true/*allow decomposition*/);
148 void XGUI_Displayer::deactivate(ObjectPtr theObject)
150 if (isVisible(theObject)) {
151 Handle(AIS_InteractiveContext) aContext = AISContext();
152 if (aContext.IsNull())
155 AISObjectPtr anObj = myResult2AISObjectMap[theObject];
156 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
157 aContext->Deactivate(anAIS);
161 void XGUI_Displayer::activate(ObjectPtr theObject, const QIntList& theModes)
163 if (isVisible(theObject)) {
164 Handle(AIS_InteractiveContext) aContext = AISContext();
165 if (aContext.IsNull())
168 AISObjectPtr anObj = myResult2AISObjectMap[theObject];
169 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
170 if (aContext->HasOpenedContext()) {
171 aContext->Load(anAIS, -1, true);
173 if (theModes.size() > 0) {
174 foreach(int aMode, theModes) {
175 aContext->Activate(anAIS, aMode);
178 aContext->Activate(anAIS);
182 bool XGUI_Displayer::isActive(ObjectPtr theObject) const
184 Handle(AIS_InteractiveContext) aContext = AISContext();
185 if (aContext.IsNull())
187 if (!isVisible(theObject))
190 AISObjectPtr anObj = myResult2AISObjectMap.at(theObject);
191 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
193 TColStd_ListOfInteger aModes;
194 aContext->ActivatedModes(anAIS, aModes);
195 return aModes.Extent() > 0;
198 void XGUI_Displayer::stopSelection(const QList<ObjectPtr>& theResults, const bool isStop,
199 const bool isUpdateViewer)
201 Handle(AIS_InteractiveContext) aContext = AISContext();
202 if (aContext.IsNull())
205 Handle(AIS_Shape) anAIS;
206 QList<ObjectPtr>::const_iterator anIt = theResults.begin(), aLast = theResults.end();
208 for (; anIt != aLast; anIt++) {
210 if (isVisible(aFeature))
211 anAIS = Handle(AIS_Shape)::DownCast(
212 myResult2AISObjectMap[aFeature]->impl<Handle(AIS_InteractiveObject)>());
217 QColor aColor(Qt::white);
219 Quantity_Color(aColor.red() / 255., aColor.green() / 255., aColor.blue() / 255.,
223 QColor aColor(Qt::red);
225 Quantity_Color(aColor.red() / 255., aColor.green() / 255., aColor.blue() / 255.,
234 void XGUI_Displayer::setSelected(const QList<ObjectPtr>& theResults, const bool isUpdateViewer)
236 Handle(AIS_InteractiveContext) aContext = AISContext();
237 // we need to unhighligth objects manually in the current local context
238 // in couple with the selection clear (TODO)
239 Handle(AIS_LocalContext) aLocalContext = aContext->LocalContext();
240 if (!aLocalContext.IsNull())
241 aLocalContext->UnhilightLastDetected(myWorkshop->viewer()->activeView());
243 aContext->ClearSelected();
244 foreach(ObjectPtr aResult, theResults)
246 if (myResult2AISObjectMap.find(aResult) == myResult2AISObjectMap.end())
249 AISObjectPtr anObj = myResult2AISObjectMap[aResult];
251 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
253 aContext->AddOrRemoveSelected(anAIS, false);
261 void XGUI_Displayer::clearSelected()
263 Handle(AIS_InteractiveContext) aContext = AISContext();
265 aContext->UnhilightCurrents(false);
266 aContext->ClearSelected();
270 void XGUI_Displayer::eraseAll(const bool isUpdateViewer)
272 Handle(AIS_InteractiveContext) ic = AISContext();
276 ResultToAISMap::iterator aIt;
277 for (aIt = myResult2AISObjectMap.begin(); aIt != myResult2AISObjectMap.end(); aIt++) {
279 AISObjectPtr aAISObj = (*aIt).second;
280 Handle(AIS_InteractiveObject) anIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
282 ic->Remove(anIO, false);
284 myResult2AISObjectMap.clear();
289 void XGUI_Displayer::eraseDeletedResults(const bool isUpdateViewer)
291 Handle(AIS_InteractiveContext) aContext = AISContext();
292 if (aContext.IsNull())
295 ResultToAISMap::const_iterator aFIt = myResult2AISObjectMap.begin(), aFLast =
296 myResult2AISObjectMap.end();
297 std::list<ObjectPtr> aRemoved;
298 for (; aFIt != aFLast; aFIt++) {
299 ObjectPtr aFeature = (*aFIt).first;
300 if (!aFeature || !aFeature->data() || !aFeature->data()->isValid()) {
301 AISObjectPtr anObj = (*aFIt).second;
304 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
305 if (!anAIS.IsNull()) {
306 aContext->Remove(anAIS, false);
307 aRemoved.push_back(aFeature);
311 std::list<ObjectPtr>::const_iterator anIt = aRemoved.begin(), aLast = aRemoved.end();
312 for (; anIt != aLast; anIt++) {
313 myResult2AISObjectMap.erase(myResult2AISObjectMap.find(*anIt));
320 void XGUI_Displayer::openLocalContext()
322 Handle(AIS_InteractiveContext) aContext = AISContext();
323 if (aContext.IsNull())
325 // Open local context if there is no one
326 if (!aContext->HasOpenedContext()) {
327 aContext->ClearCurrents(false);
328 //aContext->OpenLocalContext(false/*use displayed objects*/, true/*allow shape decomposition*/);
329 aContext->OpenLocalContext();
330 aContext->NotUseDisplayedObjects();
334 void XGUI_Displayer::closeLocalContexts(const bool isUpdateViewer)
336 AISContext()->ClearSelected(false);
337 closeAllContexts(true);
340 AISObjectPtr XGUI_Displayer::getAISObject(ObjectPtr theObject) const
343 if (myResult2AISObjectMap.find(theObject) != myResult2AISObjectMap.end())
344 anIO = (myResult2AISObjectMap.find(theObject))->second;
348 ObjectPtr XGUI_Displayer::getObject(const AISObjectPtr& theIO) const
350 Handle(AIS_InteractiveObject) aRefAIS = theIO->impl<Handle(AIS_InteractiveObject)>();
351 return getObject(aRefAIS);
354 ObjectPtr XGUI_Displayer::getObject(const Handle(AIS_InteractiveObject)& theIO) const
357 ResultToAISMap::const_iterator aFIt = myResult2AISObjectMap.begin(), aFLast =
358 myResult2AISObjectMap.end();
359 for (; aFIt != aFLast && !aFeature; aFIt++) {
360 AISObjectPtr anObj = (*aFIt).second;
363 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
366 aFeature = (*aFIt).first;
371 void XGUI_Displayer::closeAllContexts(const bool isUpdateViewer)
373 Handle(AIS_InteractiveContext) ic = AISContext();
375 ic->CloseAllContexts(false);
381 void XGUI_Displayer::updateViewer()
383 Handle(AIS_InteractiveContext) ic = AISContext();
385 ic->UpdateCurrentViewer();
388 Handle(AIS_InteractiveContext) XGUI_Displayer::AISContext() const
390 return myWorkshop->viewer()->AISContext();
393 void XGUI_Displayer::display(AISObjectPtr theAIS, bool isUpdate)
395 Handle(AIS_InteractiveContext) aContext = AISContext();
396 Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
397 if (!anAISIO.IsNull())
398 aContext->Display(anAISIO, isUpdate);
401 void XGUI_Displayer::erase(AISObjectPtr theAIS, const bool isUpdate)
403 Handle(AIS_InteractiveContext) aContext = AISContext();
404 Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
405 if (!anAISIO.IsNull()) {
406 aContext->Remove(anAISIO, isUpdate);
410 void XGUI_Displayer::activateObjectsOutOfContext(const QIntList& theModes)
412 Handle(AIS_InteractiveContext) aContext = AISContext();
413 // Open local context if there is no one
414 if (!aContext->HasOpenedContext())
417 aContext->UseDisplayedObjects();
418 ResultToAISMap::iterator aIt;
419 Handle(AIS_InteractiveObject) anAISIO;
420 for (aIt = myResult2AISObjectMap.begin(); aIt != myResult2AISObjectMap.end(); aIt++) {
421 anAISIO = (*aIt).second->impl<Handle(AIS_InteractiveObject)>();
422 aContext->Load(anAISIO, -1, true);
423 if (theModes.size() == 0)
424 aContext->Activate(anAISIO);
426 foreach(int aMode, theModes) {
427 aContext->Activate(anAISIO, aMode);
434 void XGUI_Displayer::deactivateObjectsOutOfContext()
436 Handle(AIS_InteractiveContext) aContext = AISContext();
437 // Open local context if there is no one
438 if (!aContext->HasOpenedContext())
441 aContext->NotUseDisplayedObjects();
445 void XGUI_Displayer::setDisplayMode(ObjectPtr theObject, DisplayMode theMode, bool toUpdate)
447 if (theMode == NoMode)
450 Handle(AIS_InteractiveContext) aContext = AISContext();
451 if (aContext.IsNull())
454 AISObjectPtr aAISObj = getAISObject(theObject);
458 Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
459 aContext->SetDisplayMode(aAISIO, theMode, toUpdate);
462 void XGUI_Displayer::setSelectionModes(const QIntList& theModes)
464 Handle(AIS_InteractiveContext) aContext = AISContext();
465 if (aContext.IsNull())
467 if (!aContext->HasOpenedContext())
469 // Clear previous mode
470 const TColStd_ListOfInteger& aModes = aContext->ActivatedStandardModes();
471 if (!aModes.IsEmpty()) {
472 TColStd_ListOfInteger aMModes;
473 aMModes.Assign(aModes);
474 TColStd_ListIteratorOfListOfInteger it(aMModes);
475 for(; it.More(); it.Next()) {
476 aContext->DeactivateStandardMode((TopAbs_ShapeEnum)it.Value());
479 foreach(int aMode, theModes) {
480 aContext->ActivateStandardMode((TopAbs_ShapeEnum)aMode);
484 XGUI_Displayer::DisplayMode XGUI_Displayer::displayMode(ObjectPtr theObject) const
486 Handle(AIS_InteractiveContext) aContext = AISContext();
487 if (aContext.IsNull())
490 AISObjectPtr aAISObj = getAISObject(theObject);
494 Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
495 return (XGUI_Displayer::DisplayMode) aAISIO->DisplayMode();
498 void XGUI_Displayer::addSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
500 Handle(AIS_InteractiveContext) aContext = AISContext();
501 if (aContext.IsNull())
503 aContext->AddFilter(theFilter);
506 void XGUI_Displayer::removeSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
508 Handle(AIS_InteractiveContext) aContext = AISContext();
509 if (aContext.IsNull())
511 aContext->RemoveFilter(theFilter);