1 // Copyright (C) 2014-2019 CEA/DEN, EDF R&D
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
11 // Lesser General Public License for more details.
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
20 #include "XGUI_Displayer.h"
22 #include "XGUI_FacesPanel.h"
23 #include "XGUI_Selection.h"
24 #include "XGUI_SelectionActivate.h"
25 #include "XGUI_SelectionMgr.h"
26 #include "XGUI_ViewerProxy.h"
27 #include "XGUI_Workshop.h"
30 #include <AppElements_Viewer.h>
33 #include <ModelAPI_Document.h>
34 #include <ModelAPI_Data.h>
35 #include <ModelAPI_Object.h>
36 #include <ModelAPI_Tools.h>
37 #include <ModelAPI_AttributeIntArray.h>
38 #include <ModelAPI_ResultBody.h>
39 #include <ModelAPI_ResultConstruction.h>
41 #include <ModuleBase_BRepOwner.h>
42 #include <ModuleBase_IModule.h>
43 #include <ModuleBase_Preferences.h>
44 #include <ModuleBase_Tools.h>
45 #include <ModuleBase_ViewerPrs.h>
46 #include <ModuleBase_IViewer.h>
48 #include <GeomAPI_Shape.h>
49 #include <GeomAPI_IPresentable.h>
50 #include <GeomAPI_ICustomPrs.h>
51 #include <GeomAPI_Pnt.h>
52 #include <GeomAPI_IScreenParams.h>
54 #include <SUIT_ResourceMgr.h>
56 #include <AIS_InteractiveContext.hxx>
57 #include <AIS_ListOfInteractive.hxx>
58 #include <AIS_ListIteratorOfListOfInteractive.hxx>
59 #include <AIS_DimensionSelectionMode.hxx>
60 #include <AIS_Shape.hxx>
61 #include <AIS_Dimension.hxx>
62 #include <AIS_Trihedron.hxx>
63 #ifdef BEFORE_TRIHEDRON_PATCH
64 #include <AIS_Axis.hxx>
65 #include <AIS_Plane.hxx>
66 #include <AIS_Point.hxx>
68 #include <AIS_Selection.hxx>
69 #include <Prs3d_Drawer.hxx>
70 #include <Prs3d_IsoAspect.hxx>
71 #include <SelectMgr_ListOfFilter.hxx>
72 #include <SelectMgr_ListIteratorOfListOfFilter.hxx>
73 #include <SelectMgr_SelectionManager.hxx>
74 #include <TColStd_ListIteratorOfListOfInteger.hxx>
76 #include <StdSelect_ViewerSelector3d.hxx>
78 #include <TColStd_MapOfTransient.hxx>
79 #include <TColStd_MapIteratorOfMapOfTransient.hxx>
82 #include <inspector/VInspectorAPI_CallBack.hxx>
85 #include <Events_Loop.h>
86 #include <ModelAPI_Events.h>
87 #include <Config_PropManager.h>
91 /// defines the local context mouse selection sensitivity
92 const int MOUSE_SENSITIVITY_IN_PIXEL = 10;
94 //#define DEBUG_DISPLAY
95 //#define DEBUG_FEATURE_REDISPLAY
96 //#define DEBUG_SELECTION_FILTERS
98 //#define DEBUG_OCCT_SHAPE_SELECTION
100 #define CLEAR_OUTDATED_SELECTION_BEFORE_REDISPLAY
102 //#define DEBUG_VIEWER_BLOCKED_COUNT
104 //**************************************************************
105 void displayedObjects(const Handle(AIS_InteractiveContext)& theAIS, AIS_ListOfInteractive& theList)
107 // Get from null point
108 theAIS->DisplayedObjects(theList, true);
111 QString qIntListInfo(const QIntList& theValues, const QString& theSeparator = QString(", "))
114 QIntList::const_iterator anIt = theValues.begin(), aLast = theValues.end();
115 for (; anIt != aLast; anIt++) {
116 anInfo.append(QString::number(*anIt));
118 return anInfo.join(theSeparator);
121 //**************************************************************
122 XGUI_Displayer::XGUI_Displayer(XGUI_Workshop* theWorkshop)
123 : myWorkshop(theWorkshop), myNeedUpdate(false),
124 myViewerBlockedRecursiveCount(0), myContextId(0)
128 //**************************************************************
129 XGUI_Displayer::~XGUI_Displayer()
133 //**************************************************************
134 bool XGUI_Displayer::isVisible(ObjectPtr theObject) const
136 return myResult2AISObjectMap.contains(theObject);
139 //**************************************************************
140 bool XGUI_Displayer::display(ObjectPtr theObject, bool theUpdateViewer)
142 bool aDisplayed = false;
143 if (isVisible(theObject)) {
144 aDisplayed = redisplay(theObject, theUpdateViewer);
147 GeomPresentablePtr aPrs = std::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
148 bool isShading = false;
149 if (aPrs.get() != NULL) {
150 GeomScreenParamsPtr aScreen = std::dynamic_pointer_cast<GeomAPI_IScreenParams>(theObject);
152 aScreen->setScreenPlane(getScreenPlane());
153 aScreen->setViewScale(getViewScale());
155 anAIS = aPrs->getAISObject(anAIS);
157 anAIS = myWorkshop->module()->createPresentation(theObject);
161 aDisplayed = display(theObject, anAIS, isShading, theUpdateViewer);
167 //**************************************************************
168 bool canBeShaded(Handle(AIS_InteractiveObject) theAIS, ModuleBase_IModule* theModule)
170 Handle(AIS_Shape) aShapePrs = Handle(AIS_Shape)::DownCast(theAIS);
171 if (!aShapePrs.IsNull()) {
172 TopoDS_Shape aShape = aShapePrs->Shape();
175 TopAbs_ShapeEnum aType = aShape.ShapeType();
176 if ((aType == TopAbs_VERTEX) || (aType == TopAbs_EDGE) || (aType == TopAbs_WIRE))
179 // Check that the presentation is not a sketch
180 return theModule->canBeShaded(theAIS);
186 //**************************************************************
187 bool XGUI_Displayer::display(ObjectPtr theObject, AISObjectPtr theAIS,
188 bool isShading, bool theUpdateViewer)
190 bool aDisplayed = false;
192 Handle(AIS_InteractiveContext) aContext = AISContext();
193 if (aContext.IsNull())
196 Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
197 if (!anAISIO.IsNull()) {
198 appendResultObject(theObject, theAIS);
200 //bool isCustomized = customizeObject(theObject);
202 int aDispMode = isShading? Shading : Wireframe;
204 anAISIO->Attributes()->SetFaceBoundaryDraw( Standard_True );
205 anAISIO->SetDisplayMode(aDispMode);
206 aContext->Display(anAISIO, aDispMode, 0, false, true, AIS_DS_Displayed);
208 if (getCallBack()) getCallBack()->Display(anAISIO);
212 emit objectDisplayed(theObject, theAIS);
213 selectionActivate()->activate(anAISIO, theUpdateViewer);
221 //**************************************************************
222 bool XGUI_Displayer::erase(ObjectPtr theObject, const bool theUpdateViewer)
224 bool aErased = false;
225 if (!isVisible(theObject))
228 Handle(AIS_InteractiveContext) aContext = AISContext();
229 if (aContext.IsNull())
232 AISObjectPtr anObject = myResult2AISObjectMap.value(theObject);
234 Handle(AIS_InteractiveObject) anAIS = anObject->impl<Handle(AIS_InteractiveObject)>();
235 if (!anAIS.IsNull()) {
236 emit beforeObjectErase(theObject, anObject);
237 aContext->Remove(anAIS, false/*update viewer*/);
239 if (getCallBack()) getCallBack()->Remove(anAIS);
244 myResult2AISObjectMap.remove(theObject);
247 std::ostringstream aPtrStr;
248 aPtrStr << theObject.get();
249 qDebug(QString("erase object: %1").arg(aPtrStr.str().c_str()).toStdString().c_str());
250 qDebug(getResult2AISObjectMapInfo().c_str());
259 //**************************************************************
260 bool XGUI_Displayer::redisplay(ObjectPtr theObject, bool theUpdateViewer)
262 bool aRedisplayed = false;
263 Handle(AIS_InteractiveContext) aContext = AISContext();
264 if (aContext.IsNull())
267 if (!isVisible(theObject))
270 AISObjectPtr aAISObj = getAISObject(theObject);
271 Handle(AIS_InteractiveObject) aAISIO;
273 GeomPresentablePtr aPrs = std::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
275 GeomScreenParamsPtr aScreen = std::dynamic_pointer_cast<GeomAPI_IScreenParams>(theObject);
277 aScreen->setScreenPlane(getScreenPlane());
278 aScreen->setViewScale(getViewScale());
280 AISObjectPtr aAIS_Obj = aPrs->getAISObject(aAISObj);
282 aRedisplayed = erase(theObject, theUpdateViewer);
285 if (aAIS_Obj != aAISObj) {
286 erase(theObject, theUpdateViewer);
287 appendResultObject(theObject, aAIS_Obj);
289 aAISIO = aAIS_Obj->impl<Handle(AIS_InteractiveObject)>();
292 aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
295 if (!aAISIO.IsNull()) {
296 // Check that the visualized shape is the same and the redisplay is not necessary
297 // Redisplay of AIS object leads to this object selection compute and the selection
298 // in the browser is lost
299 // this check is not necessary anymore because the selection store/restore is realized
300 // before and after the values modification.
301 // Moreother, this check avoids customize and redisplay presentation if the presentable
302 // parameter is changed.
303 //bool isEqualShapes = false;
304 //ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
305 //if (aResult.get() != NULL) {
306 // Handle(AIS_Shape) aShapePrs = Handle(AIS_Shape)::DownCast(aAISIO);
307 // if (!aShapePrs.IsNull()) {
308 // std::shared_ptr<GeomAPI_Shape> aShapePtr = ModelAPI_Tools::shape(aResult);
309 // if (aShapePtr.get()) {
310 // const TopoDS_Shape& aOldShape = aShapePrs->Shape();
311 // if (!aOldShape.IsNull())
312 // isEqualShapes = aOldShape.IsEqual(aShapePtr->impl<TopoDS_Shape>());
316 // Customization of presentation
317 //bool isCustomized = customizeObject(theObject);
318 #ifdef DEBUG_FEATURE_REDISPLAY
319 qDebug(QString("Redisplay: %1, isEqualShapes=%2").
320 arg(!isEqualShapes/* || isCustomized*/).arg(isEqualShapes)
321 .toStdString().c_str());
323 //if (!isEqualShapes/* || isCustomized*/) {
324 // /// if shapes are equal and presentation are customized, selection should be restored
325 // bool aNeedToRestoreSelection = isEqualShapes/* && isCustomized*/;
326 // if (aNeedToRestoreSelection)
327 if (aAISObj.get() && myWorkshop->facesPanel())
328 myWorkshop->facesPanel()->customizeObject(theObject, aAISObj);
330 myWorkshop->module()->storeSelection();
332 #ifdef CLEAR_OUTDATED_SELECTION_BEFORE_REDISPLAY
333 myWorkshop->selector()->deselectPresentation(aAISIO);
335 if (aContext->IsDisplayed(aAISIO))
336 aContext->Redisplay(aAISIO, false);
338 aContext->Display(aAISIO, false);
341 if (getCallBack()) getCallBack()->Redisplay(aAISIO);
344 //if (aNeedToRestoreSelection)
345 myWorkshop->module()->restoreSelection();
348 #ifdef DEBUG_FEATURE_REDISPLAY
349 qDebug(" Redisplay happens");
357 //**************************************************************
358 void XGUI_Displayer::redisplayObjects()
360 // redisplay objects visualized in the viewer
361 static Events_ID EVENT_DISP = Events_Loop::eventByName(EVENT_OBJECT_TO_REDISPLAY);
362 static const ModelAPI_EventCreator* aECreator = ModelAPI_EventCreator::get();
363 QObjectPtrList aDisplayed = displayedObjects();
364 QObjectPtrList::const_iterator anIt = aDisplayed.begin(), aLast = aDisplayed.end();
365 for (; anIt != aLast; anIt++) {
366 aECreator->sendUpdated(*anIt, EVENT_DISP);
368 XGUI_ViewerProxy* aViewer = myWorkshop->viewer();
369 if (aViewer->isColorScaleVisible()) {
370 aViewer->setupColorScale();
371 aViewer->setColorScaleShown(true);
373 Events_Loop::loop()->flush(EVENT_DISP);
376 //**************************************************************
377 void XGUI_Displayer::deactivateObjects(const QObjectPtrList& theObjList,
378 const bool theUpdateViewer)
380 //Handle(AIS_InteractiveObject) aTrihedron = getTrihedron();
381 //if (!aTrihedron.IsNull())
382 // deactivateAIS(aTrihedron);
384 QObjectPtrList::const_iterator anIt = theObjList.begin(), aLast = theObjList.end();
385 for (; anIt != aLast; anIt++) {
386 selectionActivate()->deactivate(*anIt, false);
388 //VSV It seems that there is no necessity to update viewer on deactivation
389 //if (theUpdateViewer)
393 //**************************************************************
394 bool XGUI_Displayer::isVisible(XGUI_Displayer* theDisplayer, const ObjectPtr& theObject)
396 bool aVisible = false;
397 GeomPresentablePtr aPrs = std::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
398 ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
399 if (aPrs.get() || aResult.get()) {
400 aVisible = theDisplayer->isVisible(theObject);
401 // compsolid is not visualized in the viewer,
402 // but should have presentation when all sub solids are
403 // visible. It is useful for highlight presentation where compsolid shape is selectable
404 if (!aVisible && aResult.get() && aResult->groupName() == ModelAPI_ResultBody::group()) {
405 ResultBodyPtr aCompsolidResult = std::dynamic_pointer_cast<ModelAPI_ResultBody>(aResult);
406 if (aCompsolidResult.get() != NULL) { // change colors for all sub-solids
407 int aNumberOfSubs = aCompsolidResult->numberOfSubs();
408 bool anAllSubsVisible = aNumberOfSubs > 0;
409 for(int i = 0; i < aNumberOfSubs && anAllSubsVisible; i++) {
410 anAllSubsVisible = theDisplayer->isVisible(aCompsolidResult->subResult(i));
412 aVisible = anAllSubsVisible;
416 // it is possible that feature is presentable and has results, so we should check visibility
417 // of results if presentation is not shown (e.g. Sketch Circle/Arc features)
419 // check if all results of the feature are visible
420 FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
421 std::list<ResultPtr> aResults;
422 ModelAPI_Tools::allResults(aFeature, aResults);
423 std::list<ResultPtr>::const_iterator aIt;
424 aVisible = !aResults.empty();
425 for (aIt = aResults.begin(); aIt != aResults.end(); ++aIt) {
426 aVisible = aVisible && theDisplayer->isVisible(*aIt);
432 //**************************************************************
433 void XGUI_Displayer::setSelected(const QList<ModuleBase_ViewerPrsPtr>& theValues,
434 bool theUpdateViewer)
436 Handle(AIS_InteractiveContext) aContext = AISContext();
437 if (aContext.IsNull())
439 aContext->UnhilightSelected(false);
440 aContext->ClearSelected(false);
442 if (getCallBack()) getCallBack()->ClearSelected();
444 NCollection_DataMap<TopoDS_Shape, NCollection_Map<Handle(AIS_InteractiveObject)>>
447 foreach (ModuleBase_ViewerPrsPtr aPrs, theValues) {
448 const GeomShapePtr& aGeomShape = aPrs->shape();
449 if (aGeomShape.get() && !aGeomShape->isNull()) {
450 const TopoDS_Shape& aShape = aGeomShape->impl<TopoDS_Shape>();
451 #ifdef DEBUG_OCCT_SHAPE_SELECTION
452 // problem 1: performance
453 // problem 2: IO is not specified, so the first found owner is selected, as a result
454 // it might belong to another result
455 aContext->AddOrRemoveSelected(aShape, false);
457 if (getCallBack()) getCallBack()->AddOrRemoveSelected(aShape);
460 NCollection_Map<Handle(AIS_InteractiveObject)> aPresentations;
461 if (aShapesToBeSelected.IsBound(aShape))
462 aPresentations = aShapesToBeSelected.Find(aShape);
463 ObjectPtr anObject = aPrs->object();
464 getPresentations(anObject, aPresentations);
466 aShapesToBeSelected.Bind(aShape, aPresentations);
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.value(aResult);
473 Handle(AIS_InteractiveObject) anAIS = anObj->impl<Handle(AIS_InteractiveObject)>();
474 if (!anAIS.IsNull()) {
475 // The methods are replaced in order to provide multi-selection, e.g. restore selection
476 // by activating multi selector widget. It also gives an advantage that the multi
477 // selection in OB gives multi-selection in the viewer
478 //aContext->SetSelected(anAIS, false);
479 // The selection in the context was cleared, so the method sets the objects are selected
480 aContext->AddOrRemoveSelected(anAIS, false);
482 if (getCallBack()) getCallBack()->AddOrRemoveSelected(anAIS);
488 if (!aShapesToBeSelected.IsEmpty())
489 XGUI_Displayer::AddOrRemoveSelectedShapes(aContext, aShapesToBeSelected);
495 //**************************************************************
496 void XGUI_Displayer::clearSelected(const bool theUpdateViewer)
498 Handle(AIS_InteractiveContext) aContext = AISContext();
499 if (!aContext.IsNull()) {
500 aContext->UnhilightSelected(false);//UnhilightCurrents(false);
501 aContext->ClearSelected(theUpdateViewer);
503 if (getCallBack()) getCallBack()->ClearSelected();
508 //**************************************************************
509 bool XGUI_Displayer::eraseAll(const bool theUpdateViewer)
511 bool aErased = false;
512 Handle(AIS_InteractiveContext) aContext = AISContext();
513 if (!aContext.IsNull()) {
515 foreach(ObjectPtr aObj, myResult2AISObjectMap.objects()) {
516 AISObjectPtr aAISObj = myResult2AISObjectMap.value(aObj);
518 foreach(ObjectPtr aObj, myResult2AISObjectMap.keys()) {
519 AISObjectPtr aAISObj = myResult2AISObjectMap[aObj];
522 Handle(AIS_InteractiveObject) anIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
523 if (!anIO.IsNull()) {
524 emit beforeObjectErase(aObj, aAISObj);
525 aContext->Remove(anIO, false/*update viewer*/);
527 if (getCallBack()) getCallBack()->Remove(anIO);
535 myResult2AISObjectMap.clear();
538 qDebug(getResult2AISObjectMapInfo().c_str());
544 //**************************************************************
545 AISObjectPtr XGUI_Displayer::getAISObject(ObjectPtr theObject) const
548 return myResult2AISObjectMap.value(theObject);
551 if (myResult2AISObjectMap.contains(theObject))
552 anIO = myResult2AISObjectMap[theObject];
557 //**************************************************************
558 ObjectPtr XGUI_Displayer::getObject(const AISObjectPtr& theIO) const
560 Handle(AIS_InteractiveObject) aRefAIS = theIO->impl<Handle(AIS_InteractiveObject)>();
561 return getObject(aRefAIS);
564 //**************************************************************
565 ObjectPtr XGUI_Displayer::getObject(const Handle(AIS_InteractiveObject)& theIO) const
568 ObjectPtr anObject = myResult2AISObjectMap.value(theIO);
571 ResultToAISMap::const_iterator aMapIter = myResult2AISObjectMap.cbegin();
572 for (; aMapIter != myResult2AISObjectMap.cend(); aMapIter++) {
573 const AISObjectPtr& aAIS = aMapIter.value();
574 Handle(AIS_InteractiveObject) anAIS = aAIS->impl<Handle(AIS_InteractiveObject)>();
576 anObject = aMapIter.key();
581 if (!anObject.get()) {
582 std::shared_ptr<GeomAPI_AISObject> anAISObj = AISObjectPtr(new GeomAPI_AISObject());
583 if (!theIO.IsNull()) {
584 anAISObj->setImpl(new Handle(AIS_InteractiveObject)(theIO));
586 anObject = myWorkshop->module()->findPresentedObject(anAISObj);
591 //**************************************************************
592 bool XGUI_Displayer::enableUpdateViewer(const bool isEnabled)
594 bool aWasEnabled = isUpdateEnabled();
596 myViewerBlockedRecursiveCount--;
598 myViewerBlockedRecursiveCount++;
600 #ifdef DEBUG_VIEWER_BLOCKED_COUNT
601 std::cout << "myViewerBlockedRecursiveCount = " << myViewerBlockedRecursiveCount << std::endl;
604 if (myNeedUpdate && isUpdateEnabled()) {
606 myNeedUpdate = false;
611 //**************************************************************
612 void XGUI_Displayer::updateViewer() const
614 Handle(AIS_InteractiveContext) aContext = AISContext();
616 #ifdef DEBUG_VIEWER_BLOCKED_COUNT
617 std::cout << "updateViewer: " << (myViewerBlockedRecursiveCount == 0 ? " done" : " later")
621 if (!aContext.IsNull() && isUpdateEnabled()) {
622 //myWorkshop->viewer()->Zfitall();
623 aContext->UpdateCurrentViewer();
629 //**************************************************************
630 Handle(AIS_InteractiveContext) XGUI_Displayer::AISContext() const
632 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
633 if (!aContext.IsNull() && (myContextId != aContext.get())) {
634 myContextId = aContext.get();
635 if (!myWorkshop->selectionActivate()->isTrihedronActive())
636 selectionActivate()->deactivateTrihedron(true);
637 aContext->DefaultDrawer()->VIsoAspect()->SetNumber(0);
638 aContext->DefaultDrawer()->UIsoAspect()->SetNumber(0);
640 //Handle(AIS_Trihedron) aTrihedron = myWorkshop->viewer()->trihedron();
641 //aTrihedron->getHighlightPointAspect()->SetScale(2.0);
642 //aTrihedron->getHighlightPointAspect()->SetTypeOfMarker(Aspect_TOM_O_STAR);
644 // Commented out according to discussion in bug #2825
645 //ModuleBase_IViewer::DefaultHighlightDrawer = aContext->HighlightStyle();
646 //Handle(Prs3d_Drawer) aSelStyle = aContext->SelectionStyle();
647 //double aDeflection =
648 // QString(ModelAPI_ResultConstruction::DEFAULT_DEFLECTION().c_str()).toDouble();
650 // aDeflection = Config_PropManager::real("Visualization", "construction_deflection");
653 //ModuleBase_IViewer::DefaultHighlightDrawer->SetDeviationCoefficient(aDeflection);
654 //aSelStyle->SetDeviationCoefficient(aDeflection);
659 //**************************************************************
660 void XGUI_Displayer::setSelectionColor(const std::vector<int>& theColor)
662 Handle(AIS_InteractiveContext) aContext = AISContext();
663 if (!aContext.IsNull()) {
664 Quantity_Color aQColor(theColor[0] / 255.,
666 theColor[2] / 255., Quantity_TOC_RGB);
667 aContext->SelectionStyle()->SetColor(aQColor);
668 aContext->HighlightStyle(Prs3d_TypeOfHighlight_LocalSelected)->SetColor(aQColor);
673 //**************************************************************
674 std::vector<int> XGUI_Displayer::selectionColor() const
676 std::vector<int> aColor;
677 Handle(AIS_InteractiveContext) aContext = AISContext();
678 if (!aContext.IsNull()) {
679 Quantity_Color aQColor = aContext->SelectionStyle()->Color();
680 aColor.push_back((int)(aQColor.Red() * 255));
681 aColor.push_back((int)(aQColor.Green() * 255));
682 aColor.push_back((int)(aQColor.Blue() * 255));
688 //**************************************************************
689 Handle(SelectMgr_AndFilter) XGUI_Displayer::GetFilter()
691 Handle(AIS_InteractiveContext) aContext = AISContext();
692 if (!aContext.IsNull() && myAndFilter.IsNull()) {
693 myAndFilter = new SelectMgr_AndFilter();
694 aContext->AddFilter(myAndFilter);
699 //**************************************************************
700 bool XGUI_Displayer::displayAIS(AISObjectPtr theAIS, const bool toActivateInSelectionModes,
701 const Standard_Integer theDisplayMode, bool theUpdateViewer)
703 bool aDisplayed = false;
704 Handle(AIS_InteractiveContext) aContext = AISContext();
705 Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
706 if (!aContext.IsNull() && !anAISIO.IsNull()) {
707 aContext->Display(anAISIO, theDisplayMode, 0, false/*update viewer*/, true, AIS_DS_Displayed);
709 if (getCallBack()) getCallBack()->Display(anAISIO);
712 aContext->Deactivate(anAISIO);
714 if (getCallBack()) getCallBack()->Deactivate(anAISIO);
716 aContext->Load(anAISIO);
718 if (getCallBack()) getCallBack()->Load(anAISIO);
720 if (toActivateInSelectionModes)
721 myWorkshop->selectionActivate()->activateOnDisplay(anAISIO, theUpdateViewer);
729 //**************************************************************
730 bool XGUI_Displayer::eraseAIS(AISObjectPtr theAIS, const bool theUpdateViewer)
732 bool aErased = false;
733 Handle(AIS_InteractiveContext) aContext = AISContext();
734 if (!aContext.IsNull()) {
735 Handle(AIS_InteractiveObject) anAISIO = theAIS->impl<Handle(AIS_InteractiveObject)>();
736 if (!anAISIO.IsNull() && aContext->IsDisplayed(anAISIO)) {
737 aContext->Remove(anAISIO, false/*update viewer*/);
739 if (getCallBack()) getCallBack()->Remove(anAISIO);
744 if (aErased && theUpdateViewer)
749 //**************************************************************
750 void XGUI_Displayer::setDisplayMode(ObjectPtr theObject, DisplayMode theMode, bool theUpdateViewer)
752 if (theMode == NoMode)
755 Handle(AIS_InteractiveContext) aContext = AISContext();
756 if (aContext.IsNull())
759 AISObjectPtr aAISObj = getAISObject(theObject);
763 Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
764 aContext->SetDisplayMode(aAISIO, theMode, false);
765 // Redisplay in order to update new mode because it could be not computed before
770 //**************************************************************
771 XGUI_Displayer::DisplayMode XGUI_Displayer::displayMode(ObjectPtr theObject) const
773 Handle(AIS_InteractiveContext) aContext = AISContext();
774 if (aContext.IsNull())
777 AISObjectPtr aAISObj = getAISObject(theObject);
781 Handle(AIS_InteractiveObject) aAISIO = aAISObj->impl<Handle(AIS_InteractiveObject)>();
782 return (XGUI_Displayer::DisplayMode) aAISIO->DisplayMode();
785 //**************************************************************
786 void XGUI_Displayer::deactivateSelectionFilters(const bool theAddFilterOnly)
788 Handle(AIS_InteractiveContext) aContext = AISContext();
789 if (!myAndFilter.IsNull()) {
791 if (!aContext.IsNull()) {
792 const SelectMgr_ListOfFilter& aFilters = aContext->Filters();
793 SelectMgr_ListIteratorOfListOfFilter anIt(aFilters);
794 for (; anIt.More() && !aFound; anIt.Next()) {
795 Handle(SelectMgr_Filter) aFilter = anIt.Value();
796 aFound = aFilter == myAndFilter;
799 aContext->RemoveFilter(myAndFilter);
801 myAndFilter.Nullify();
805 //**************************************************************
806 void XGUI_Displayer::addSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
808 Handle(AIS_InteractiveContext) aContext = AISContext();
809 if (aContext.IsNull() || hasSelectionFilter(theFilter))
812 Handle(SelectMgr_CompositionFilter) aCompositeFilter = GetFilter();
813 if (!aCompositeFilter.IsNull()) {
814 aCompositeFilter->Add(theFilter);
815 #ifdef DEBUG_SELECTION_FILTERS
816 int aCount = aCompositeFilter->StoredFilters().Extent();
817 qDebug(QString("addSelectionFilter: filters.count() = %1").arg(aCount).toStdString().c_str());
822 //**************************************************************
823 void XGUI_Displayer::removeSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
825 Handle(AIS_InteractiveContext) aContext = AISContext();
826 if (aContext.IsNull())
829 Handle(SelectMgr_AndFilter) aCompositeFilter = GetFilter();
830 if (!aCompositeFilter.IsNull() && aCompositeFilter->IsIn(theFilter)) {
831 aCompositeFilter->Remove(theFilter);
832 #ifdef DEBUG_SELECTION_FILTERS
833 int aCount = aCompositeFilter->StoredFilters().Extent();
834 qDebug(QString("removeSelectionFilter: filters.count() = %1")
835 .arg(aCount).toStdString().c_str());
840 //**************************************************************
841 bool XGUI_Displayer::hasSelectionFilter(const Handle(SelectMgr_Filter)& theFilter)
843 bool aFilterFound = false;
845 Handle(AIS_InteractiveContext) aContext = AISContext();
846 if (aContext.IsNull())
848 const SelectMgr_ListOfFilter& aFilters = aContext->Filters();
849 SelectMgr_ListIteratorOfListOfFilter aIt(aFilters);
850 for (; aIt.More() && !aFilterFound; aIt.Next()) {
851 if (theFilter.get() == aIt.Value().get())
854 Handle(SelectMgr_CompositionFilter) aCompositeFilter = GetFilter();
855 if (!aCompositeFilter.IsNull()) {
856 const SelectMgr_ListOfFilter& aStoredFilters = aCompositeFilter->StoredFilters();
857 for (aIt.Initialize(aStoredFilters); aIt.More() && !aFilterFound; aIt.Next()) {
858 if (theFilter.get() == aIt.Value().get())
865 //**************************************************************
866 void XGUI_Displayer::removeFilters()
868 Handle(AIS_InteractiveContext) aContext = AISContext();
869 if (aContext.IsNull())
872 Handle(SelectMgr_CompositionFilter) aCompositeFilter = GetFilter();
873 if (!aCompositeFilter.IsNull())
874 aCompositeFilter->Clear();
877 //**************************************************************
878 void XGUI_Displayer::showOnly(const QObjectPtrList& theList)
881 QObjectPtrList aDispList = myResult2AISObjectMap.objects();
883 QObjectPtrList aDispList = myResult2AISObjectMap.keys();
885 foreach(ObjectPtr aObj, aDispList) {
886 if (!theList.contains(aObj))
889 foreach(ObjectPtr aObj, theList) {
890 if (!isVisible(aObj))
891 display(aObj, false);
896 //**************************************************************
897 bool XGUI_Displayer::canBeShaded(ObjectPtr theObject) const
899 if (!isVisible(theObject))
902 AISObjectPtr aAISObj = getAISObject(theObject);
903 if (aAISObj.get() == NULL)
906 Handle(AIS_InteractiveObject) anAIS = aAISObj->impl<Handle(AIS_InteractiveObject)>();
907 return ::canBeShaded(anAIS, myWorkshop->module());
910 //**************************************************************
911 //bool XGUI_Displayer::customizeObject(ObjectPtr theObject)
913 // AISObjectPtr anAISObj = getAISObject(theObject);
914 // // correct the result's color it it has the attribute
915 // ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
917 // // Customization of presentation
918 // GeomCustomPrsPtr aCustomPrs;
919 // FeaturePtr aFeature = ModelAPI_Feature::feature(theObject);
920 // if (aFeature.get() != NULL) {
921 // GeomCustomPrsPtr aCustPrs = std::dynamic_pointer_cast<GeomAPI_ICustomPrs>(aFeature);
922 // if (aCustPrs.get() != NULL)
923 // aCustomPrs = aCustPrs;
925 // if (aCustomPrs.get() == NULL) {
926 // GeomPresentablePtr aPrs = std::dynamic_pointer_cast<GeomAPI_IPresentable>(theObject);
927 // // we ignore presentable not customized objects
928 // if (aPrs.get() == NULL)
929 // aCustomPrs = myCustomPrs;
931 // bool isCustomized = aCustomPrs.get() &&
932 // aCustomPrs->customisePresentation(aResult, anAISObj, myCustomPrs);
933 // isCustomized = myWorkshop->module()->afterCustomisePresentation(aResult, anAISObj, myCustomPrs)
936 // // update presentation state if faces panel is active
937 // if (anAISObj.get() && myWorkshop->facesPanel())
938 // isCustomized = myWorkshop->facesPanel()->customizeObject(theObject, anAISObj) || isCustomized;
940 // return isCustomized;
943 //**************************************************************
944 QColor XGUI_Displayer::setObjectColor(ObjectPtr theObject,
945 const QColor& theColor,
946 bool theUpdateViewer)
948 if (!isVisible(theObject))
951 AISObjectPtr anAISObj = getAISObject(theObject);
953 anAISObj->getColor(aR, aG, aB);
954 anAISObj->setColor(theColor.red(), theColor.green(), theColor.blue());
957 return QColor(aR, aG, aB);
960 //**************************************************************
961 void XGUI_Displayer::appendResultObject(ObjectPtr theObject, AISObjectPtr theAIS)
964 myResult2AISObjectMap.add(theObject, theAIS);
966 myResult2AISObjectMap[theObject] = theAIS;
970 std::ostringstream aPtrStr;
971 aPtrStr << theObject.get();
972 qDebug(QString("display object: %1").arg(aPtrStr.str().c_str()).toStdString().c_str());
973 qDebug(getResult2AISObjectMapInfo().c_str());
978 //**************************************************************
979 std::string XGUI_Displayer::getResult2AISObjectMapInfo() const
981 QStringList aContent;
983 foreach(ObjectPtr aObj, myResult2AISObjectMap.objects()) {
984 AISObjectPtr aAISObj = myResult2AISObjectMap.value(aObj);
986 foreach(ObjectPtr aObj, myResult2AISObjectMap.keys()) {
987 AISObjectPtr aAISObj = myResult2AISObjectMap[aObj];
989 std::ostringstream aPtrStr;
990 aPtrStr << "aObj = " << aObj.get() << ":";
991 aPtrStr << "anAIS = " << aAISObj.get() << ":";
992 aPtrStr << "[" << ModuleBase_Tools::objectInfo(aObj).toStdString().c_str() << "]";
994 aContent.append(aPtrStr.str().c_str());
996 return QString("myResult2AISObjectMap: size = %1\n%2\n").arg(myResult2AISObjectMap.size()).
997 arg(aContent.join("\n")).toStdString().c_str();
1001 //**************************************************************
1002 void XGUI_Displayer::getPresentations(const ObjectPtr& theObject,
1003 NCollection_Map<Handle(AIS_InteractiveObject)>& thePresentations)
1005 ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(theObject);
1006 if (aResult.get()) {
1007 AISObjectPtr aAISObj = getAISObject(aResult);
1008 if (aAISObj.get() == NULL) {
1009 // if result is a result of a composite feature, it is visualized by visualization of
1010 // composite children, so we should get one of this presentations
1011 ResultBodyPtr aCompSolid = std::dynamic_pointer_cast<ModelAPI_ResultBody>(aResult);
1012 if (aCompSolid.get() && aCompSolid->numberOfSubs() > 0) {
1013 aAISObj = getAISObject(aCompSolid->subResult(0));
1016 if (aAISObj.get() != NULL) {
1017 Handle(AIS_InteractiveObject) anAIS = aAISObj->impl<Handle(AIS_InteractiveObject)>();
1018 if (!anAIS.IsNull() && !thePresentations.Contains(anAIS))
1019 thePresentations.Add(anAIS);
1023 FeaturePtr aFeature = std::dynamic_pointer_cast<ModelAPI_Feature>(theObject);
1024 // find presentation of the feature
1025 AISObjectPtr aAISObj = getAISObject(aFeature);
1026 if (aAISObj.get() != NULL) {
1027 Handle(AIS_InteractiveObject) anAIS = aAISObj->impl<Handle(AIS_InteractiveObject)>();
1028 if (!anAIS.IsNull() && !thePresentations.Contains(anAIS))
1029 thePresentations.Add(anAIS);
1031 // find presentations of the feature results
1032 std::list<ResultPtr> aResults;
1033 ModelAPI_Tools::allResults(aFeature, aResults);
1034 std::list<ResultPtr>::const_iterator anIt = aResults.begin(), aLast = aResults.end();
1035 for (; anIt != aLast; ++anIt) {
1036 AISObjectPtr aAISObj = getAISObject(*anIt);
1037 if (aAISObj.get() != NULL) {
1038 Handle(AIS_InteractiveObject) anAIS = aAISObj->impl<Handle(AIS_InteractiveObject)>();
1039 if (!anAIS.IsNull() && !thePresentations.Contains(anAIS))
1040 thePresentations.Add(anAIS);
1046 //**************************************************************
1047 void XGUI_Displayer::displayTrihedron(bool theToDisplay) const
1049 Handle(AIS_InteractiveContext) aContext = AISContext();
1050 if (aContext.IsNull())
1053 Handle(AIS_Trihedron) aTrihedron = myWorkshop->viewer()->trihedron();
1055 XGUI_SelectionActivate* aSelectionActive = selectionActivate();
1057 if (!aContext->IsDisplayed(aTrihedron))
1058 aContext->Display(aTrihedron,
1060 -1 /* selection mode */,
1061 Standard_True /* update viewer*/,
1062 Standard_False /* allow decomposition */,
1063 AIS_DS_Displayed /* xdisplay status */);
1065 if (getCallBack()) getCallBack()->Display(aTrihedron);
1068 if (!aSelectionActive->isTrihedronActive())
1069 aSelectionActive->deactivateTrihedron(false);
1071 aSelectionActive->activate(aTrihedron, false);
1073 aSelectionActive->deactivateTrihedron(false);
1075 aContext->Erase(aTrihedron, Standard_True);
1077 if (getCallBack()) getCallBack()->Remove(aTrihedron);
1082 //**************************************************************
1083 void XGUI_Displayer::AddOrRemoveSelectedShapes(Handle(AIS_InteractiveContext) theContext,
1084 const NCollection_DataMap<TopoDS_Shape,
1085 NCollection_Map<Handle(AIS_InteractiveObject)>>& theShapesToBeSelected)
1087 NCollection_Map<Handle(AIS_InteractiveObject)> aCompsolidPresentations;
1088 NCollection_Map<Handle(AIS_InteractiveObject)> aSelectedPresentations;
1090 NCollection_List<Handle(SelectBasics_EntityOwner)> anActiveOwners;
1091 theContext->MainSelector()->ActiveOwners(anActiveOwners);
1092 NCollection_List<Handle(SelectBasics_EntityOwner)>::Iterator anOwnersIt (anActiveOwners);
1093 Handle(SelectMgr_EntityOwner) anOwner;
1095 /// It is very important to check that the owner is processed only once and has a map of
1096 /// processed owners because SetSelected works as a switch.
1097 /// If count of calls setSelectec is even, the object stays in the previous state
1098 /// (selected, deselected)
1099 /// OCCT: to write about the problem that active owners method returns one owner several times
1100 QList<size_t> aSelectedIds; // Remember of selected address in order to avoid duplicates
1101 for (; anOwnersIt.More(); anOwnersIt.Next()) {
1102 anOwner = Handle(SelectMgr_EntityOwner)::DownCast (anOwnersIt.Value());
1103 if (aSelectedIds.contains((size_t)anOwner.get()))
1105 aSelectedIds.append((size_t)anOwner.get());
1107 Handle(StdSelect_BRepOwner) BROwnr = Handle(StdSelect_BRepOwner)::DownCast(anOwner);
1108 if (!BROwnr.IsNull() && BROwnr->HasShape()) {
1109 const TopoDS_Shape& aShape = BROwnr->Shape();
1110 if (aShape.IsNull())
1113 Handle(ModuleBase_BRepOwner) aCustomOwner = Handle(ModuleBase_BRepOwner)::DownCast(anOwner);
1115 NCollection_DataMap<TopoDS_Shape, NCollection_Map<Handle(AIS_InteractiveObject)> >
1116 ::Iterator aShapeIt(theShapesToBeSelected);
1117 for (; aShapeIt.More(); aShapeIt.Next()) {
1118 const TopoDS_Shape& aParameterShape = aShapeIt.Key();
1119 // In case of compound we cannot rely on simple comparison method.
1120 // If the compound is generated by Group feature then this compound is alwais new.
1121 // So, we have to compare content of these compounds
1123 // isSame should be used here as it does not check orientation of shapes
1124 // despite on isEqual of shapes or IsBound for shape in QMap. Orientation is
1125 // different for Edges shapes in model shape and owner even if this is the same shape
1126 if (ModuleBase_Tools::isSameShape(aParameterShape, aShape)) {
1127 Handle(AIS_InteractiveObject) anOwnerPresentation =
1128 Handle(AIS_InteractiveObject)::DownCast(anOwner->Selectable());
1129 NCollection_Map<Handle(AIS_InteractiveObject)> aPresentations =
1130 theShapesToBeSelected.Find(aParameterShape);
1131 if (aPresentations.Contains(anOwnerPresentation)) {
1132 theContext->AddOrRemoveSelected(anOwner, Standard_False);
1133 anOwner->SetSelected(Standard_True);
1134 // collect selected presentations to do not select them if compsolid is selected
1135 if (!aSelectedPresentations.Contains(anOwnerPresentation))
1136 aSelectedPresentations.Add(anOwnerPresentation);
1139 else if (!aCustomOwner.IsNull()) { // CompSolid processing #2219
1140 // shape of owner is compound, but shape to be selected is compsolid, so
1141 // we need to compare shape to AIS presentation of owner(rule of the owner creation)
1142 Handle(AIS_Shape) anOwnerPresentation =
1143 Handle(AIS_Shape)::DownCast(anOwner->Selectable());
1144 const TopoDS_Shape& aPresentationShape = anOwnerPresentation->Shape();
1145 if (aParameterShape.IsSame(anOwnerPresentation->Shape()) &&
1146 !aCompsolidPresentations.Contains(anOwnerPresentation))
1147 aCompsolidPresentations.Add(anOwnerPresentation);
1152 // select CompSolid presentations if their owners was not selected yet
1153 NCollection_Map<Handle(AIS_InteractiveObject)>::Iterator anIt (aCompsolidPresentations);
1154 for (; anIt.More(); anIt.Next()) {
1155 if (aSelectedPresentations.Contains(anIt.Value()))
1157 theContext->AddOrRemoveSelected(anIt.Value(), Standard_False);
1161 //**************************************************************
1162 XGUI_SelectionActivate* XGUI_Displayer::selectionActivate() const
1164 return myWorkshop->selectionActivate();
1167 //**************************************************************
1168 GeomPlanePtr XGUI_Displayer::getScreenPlane() const
1170 GeomPlanePtr aResult;
1171 Handle(AIS_InteractiveContext) aContext = AISContext();
1172 if (!aContext.IsNull()) {
1173 Handle(V3d_Viewer) aViewer = aContext->CurrentViewer();
1174 Handle(V3d_View) aView;
1175 for (aViewer->InitActiveViews(); aViewer->MoreActiveViews(); aViewer->NextActiveViews()) {
1176 aView = aViewer->ActiveView();
1179 if (!aView.IsNull()) {
1180 double aEyeX, aEyeY, aEyeZ;
1181 aView->Eye(aEyeX, aEyeY, aEyeZ);
1183 double aProjX, aProjY, aProjZ;
1184 aView->Proj(aProjX, aProjY, aProjZ);
1186 GeomPointPtr aPnt = GeomPointPtr(new GeomAPI_Pnt(aEyeX, aEyeY, aEyeZ));
1187 GeomDirPtr aDir = GeomDirPtr(new GeomAPI_Dir(aProjX, aProjY, aProjZ));
1189 aResult = GeomPlanePtr(new GeomAPI_Pln(aPnt, aDir));
1195 double XGUI_Displayer::getViewScale() const
1197 Handle(AIS_InteractiveContext) aContext = AISContext();
1198 if (!aContext.IsNull()) {
1199 Handle(V3d_Viewer) aViewer = aContext->CurrentViewer();
1200 Handle(V3d_View) aView;
1201 for (aViewer->InitActiveViews(); aViewer->MoreActiveViews(); aViewer->NextActiveViews()) {
1202 aView = aViewer->ActiveView();
1205 return aView->Camera()->Scale();