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"
9 #include "XGUI_Tools.h"
11 #include <ModelAPI_Document.h>
12 #include <ModelAPI_Data.h>
14 #include <AIS_InteractiveContext.hxx>
15 #include <AIS_LocalContext.hxx>
16 #include <AIS_ListOfInteractive.hxx>
17 #include <AIS_ListIteratorOfListOfInteractive.hxx>
19 #include <AIS_Shape.hxx>
23 const int MOUSE_SENSITIVITY_IN_PIXEL = 10; ///< defines the local context mouse selection sensitivity
25 XGUI_Displayer::XGUI_Displayer(XGUI_Workshop* theWorkshop)
27 myWorkshop = theWorkshop;
30 XGUI_Displayer::~XGUI_Displayer()
34 bool XGUI_Displayer::IsVisible(boost::shared_ptr<ModelAPI_Feature> theFeature)
36 return myFeature2AISObjectMap.find(theFeature) != myFeature2AISObjectMap.end();
39 void XGUI_Displayer::Display(boost::shared_ptr<ModelAPI_Feature> theFeature,
40 const bool isUpdateViewer)
44 /*void XGUI_Displayer::Display(boost::shared_ptr<ModelAPI_Feature> theFeature,
45 const TopoDS_Shape& theShape, const bool isUpdateViewer)
47 Handle(AIS_InteractiveContext) aContext = AISContext();
49 Handle(AIS_Shape) anAIS = new AIS_Shape(theShape);
50 myFeature2AISObjectMap[theFeature] = anAIS;
52 aContext->Display(anAIS, Standard_False);
54 aContext->UpdateCurrentViewer();
58 std::list<XGUI_ViewerPrs> XGUI_Displayer::GetSelected(const int theShapeTypeToSkip)
60 std::set<boost::shared_ptr<ModelAPI_Feature> > aPrsFeatures;
61 std::list<XGUI_ViewerPrs> aPresentations;
63 Handle(AIS_InteractiveContext) aContext = AISContext();
64 for (aContext->InitSelected(); aContext->MoreSelected(); aContext->NextSelected()) {
65 Handle(AIS_InteractiveObject) anIO = aContext->SelectedInteractive();
66 TopoDS_Shape aShape = aContext->SelectedShape();
67 if (theShapeTypeToSkip >= 0 && !aShape.IsNull() && aShape.ShapeType() == theShapeTypeToSkip)
70 boost::shared_ptr<ModelAPI_Feature> aFeature = GetFeature(anIO);
71 if (aPrsFeatures.find(aFeature) != aPrsFeatures.end())
73 aPresentations.push_back(XGUI_ViewerPrs(aFeature, aShape));
74 aPrsFeatures.insert(aFeature);
76 return aPresentations;
79 std::list<XGUI_ViewerPrs> XGUI_Displayer::GetHighlighted(const int theShapeTypeToSkip)
81 std::set<boost::shared_ptr<ModelAPI_Feature> > aPrsFeatures;
82 std::list<XGUI_ViewerPrs> aPresentations;
84 Handle(AIS_InteractiveContext) aContext = AISContext();
85 for (aContext->InitDetected(); aContext->MoreDetected(); aContext->NextDetected()) {
86 Handle(AIS_InteractiveObject) anIO = aContext->DetectedInteractive();
87 TopoDS_Shape aShape = aContext->DetectedShape();
88 if (theShapeTypeToSkip >= 0 && !aShape.IsNull() && aShape.ShapeType() == theShapeTypeToSkip)
91 boost::shared_ptr<ModelAPI_Feature> aFeature = GetFeature(anIO);
92 if (aPrsFeatures.find(aFeature) != aPrsFeatures.end())
94 aPresentations.push_back(XGUI_ViewerPrs(aFeature, aShape));
95 aPrsFeatures.insert(aFeature);
98 return aPresentations;
101 void XGUI_Displayer::Erase(boost::shared_ptr<ModelAPI_Feature> theFeature,
102 const bool isUpdateViewer)
104 if (myFeature2AISObjectMap.find(theFeature) == myFeature2AISObjectMap.end())
107 Handle(AIS_InteractiveContext) aContext = AISContext();
108 Handle(AIS_InteractiveObject) anAIS = myFeature2AISObjectMap[theFeature];
109 Handle(AIS_Shape) anAISShape = Handle(AIS_Shape)::DownCast(anAIS);
110 if (!anAISShape.IsNull())
112 aContext->Erase(anAISShape);
114 myFeature2AISObjectMap.erase(theFeature);
117 aContext->UpdateCurrentViewer();
120 void XGUI_Displayer::Redisplay(boost::shared_ptr<ModelAPI_Feature> theFeature,
121 const TopoDS_Shape& theShape, const bool isUpdateViewer)
123 Handle(AIS_InteractiveContext) aContext = AISContext();
124 // Open local context if there is no one
125 if (!aContext->HasOpenedContext()) {
126 aContext->ClearCurrents(false);
127 aContext->OpenLocalContext(false/*use displayed objects*/, true/*allow shape decomposition*/);
128 // set mouse sensitivity
129 //aContext->SetSensitivityMode(StdSelect_SM_WINDOW);
130 //aContext->SetPixelTolerance(MOUSE_SENSITIVITY_IN_PIXEL);
132 // display or redisplay presentation
133 Handle(AIS_Shape) anAIS;
134 if (IsVisible(theFeature)) {
135 anAIS = Handle(AIS_Shape)::DownCast(myFeature2AISObjectMap[theFeature]);
136 if (!anAIS.IsNull()) {
137 // if the AIS object is displayed in the opened local context in some mode, additional
138 // AIS sub objects are created there. They should be rebuild for correct selecting.
139 // It is possible to correct it by closing local context before the shape set and opening
140 // after. Another workaround to thrown down the selection and reselecting the AIS.
141 // If there was a problem here, try the first solution with close/open local context.
142 anAIS->Set(theShape);
145 /*if (aContext->IsSelected(anAIS)) {
146 aContext->AddOrRemoveSelected(anAIS, false);
147 aContext->AddOrRemoveSelected(anAIS, false);
148 //aContext->SetSelected(anAIS, false);
153 anAIS = new AIS_Shape(theShape);
154 myFeature2AISObjectMap[theFeature] = anAIS;
155 aContext->Display(anAIS, false);
159 void XGUI_Displayer::ActivateInLocalContext(boost::shared_ptr<ModelAPI_Feature> theFeature,
160 const std::list<int>& theModes, const bool isUpdateViewer)
162 Handle(AIS_InteractiveContext) aContext = AISContext();
163 // Open local context if there is no one
164 if (!aContext->HasOpenedContext()) {
165 aContext->ClearCurrents(false);
166 aContext->OpenLocalContext(false/*use displayed objects*/, true/*allow shape decomposition*/);
168 // display or redisplay presentation
169 Handle(AIS_Shape) anAIS;
170 if (IsVisible(theFeature))
171 anAIS = Handle(AIS_Shape)::DownCast(myFeature2AISObjectMap[theFeature]);
173 // Activate selection of objects from prs
174 if (!anAIS.IsNull()) {
175 aContext->Load(anAIS, -1, true/*allow decomposition*/);
176 aContext->Deactivate(anAIS);
178 std::list<int>::const_iterator anIt = theModes.begin(), aLast = theModes.end();
179 for (; anIt != aLast; anIt++)
181 aContext->Activate(anAIS, AIS_Shape::SelectionMode((TopAbs_ShapeEnum)*anIt));
186 aContext->UpdateCurrentViewer();
189 void XGUI_Displayer::StopSelection(const std::list<XGUI_ViewerPrs>& theFeatures, const bool isStop,
190 const bool isUpdateViewer)
192 Handle(AIS_InteractiveContext) aContext = AISContext();
194 Handle(AIS_Shape) anAIS;
195 std::list<XGUI_ViewerPrs>::const_iterator anIt = theFeatures.begin(), aLast = theFeatures.end();
196 boost::shared_ptr<ModelAPI_Feature> aFeature;
197 for (; anIt != aLast; anIt++) {
198 aFeature = (*anIt).feature();
199 if (IsVisible(aFeature))
200 anAIS = Handle(AIS_Shape)::DownCast(myFeature2AISObjectMap[aFeature]);
205 QColor aColor(Qt::white);
206 anAIS->SetColor(Quantity_Color(aColor.red()/255., aColor.green()/255., aColor.blue()/255., Quantity_TOC_RGB));
210 QColor aColor(Qt::red);
211 anAIS->SetColor(Quantity_Color(aColor.red()/255., aColor.green()/255., aColor.blue()/255., Quantity_TOC_RGB));
216 aContext->UpdateCurrentViewer();
219 void XGUI_Displayer::SetSelected(const std::list<XGUI_ViewerPrs>& theFeatures, const bool isUpdateViewer)
221 Handle(AIS_InteractiveContext) aContext = AISContext();
223 std::list<XGUI_ViewerPrs>::const_iterator anIt = theFeatures.begin(), aLast = theFeatures.end();
224 boost::shared_ptr<ModelAPI_Feature> aFeature;
226 Handle(AIS_Shape) anAIS;
227 // we need to unhighligth objects manually in the current local context
228 // in couple with the selection clear (TODO)
229 Handle(AIS_LocalContext) aLocalContext = aContext->LocalContext();
230 if (!aLocalContext.IsNull())
231 aLocalContext->UnhilightLastDetected(myWorkshop->viewer()->activeView());
232 aContext->ClearSelected(false);
234 for (; anIt != aLast; anIt++) {
235 aFeature = (*anIt).feature();
236 if (IsVisible(aFeature))
237 anAIS = Handle(AIS_Shape)::DownCast(myFeature2AISObjectMap[aFeature]);
240 aContext->AddOrRemoveSelected(anAIS, false);
244 aContext->UpdateCurrentViewer();
247 void XGUI_Displayer::EraseAll(const bool isUpdateViewer)
249 Handle(AIS_InteractiveContext) ic = AISContext();
251 AIS_ListOfInteractive aList;
252 ic->DisplayedObjects(aList);
253 AIS_ListIteratorOfListOfInteractive anIter(aList);
254 for (; anIter.More(); anIter.Next()) {
255 if ((anIter.Value()->DynamicType() == STANDARD_TYPE(AIS_Trihedron)))
259 Handle(AIS_InteractiveObject) anIO = anIter.Value();
260 ic->Erase(anIO, false);
262 myFeature2AISObjectMap.clear();
264 ic->UpdateCurrentViewer();
267 void XGUI_Displayer::EraseDeletedFeatures(const bool isUpdateViewer)
269 Handle(AIS_InteractiveContext) aContext = AISContext();
271 FeatureToAISMap::const_iterator aFIt = myFeature2AISObjectMap.begin(),
272 aFLast = myFeature2AISObjectMap.end();
273 std::list<boost::shared_ptr<ModelAPI_Feature>> aRemoved;
274 for (; aFIt != aFLast; aFIt++)
276 boost::shared_ptr<ModelAPI_Feature> aFeature = (*aFIt).first;
277 if (!aFeature || !aFeature->data() || !aFeature->data()->isValid()) {
278 Handle(AIS_InteractiveObject) anAIS = (*aFIt).second;
279 if (!anAIS.IsNull()) {
280 aContext->Erase(anAIS, false);
281 aRemoved.push_back(aFeature);
285 std::list<boost::shared_ptr<ModelAPI_Feature>>::const_iterator anIt = aRemoved.begin(),
286 aLast = aRemoved.end();
287 for (; anIt != aLast; anIt++) {
288 myFeature2AISObjectMap.erase(myFeature2AISObjectMap.find(*anIt));
292 aContext->UpdateCurrentViewer();
295 void XGUI_Displayer::CloseLocalContexts(const bool isUpdateViewer)
297 CloseAllContexts(true);
300 boost::shared_ptr<ModelAPI_Feature> XGUI_Displayer::GetFeature(Handle(AIS_InteractiveObject) theIO)
302 boost::shared_ptr<ModelAPI_Feature> aFeature;
303 FeatureToAISMap::const_iterator aFIt = myFeature2AISObjectMap.begin(),
304 aFLast = myFeature2AISObjectMap.end();
305 for (; aFIt != aFLast && !aFeature; aFIt++) {
306 Handle(AIS_InteractiveObject) anAIS = (*aFIt).second;
309 aFeature = (*aFIt).first;
314 void XGUI_Displayer::CloseAllContexts(const bool isUpdateViewer)
316 Handle(AIS_InteractiveContext) ic = AISContext();
318 ic->CloseAllContexts(false);
320 ic->UpdateCurrentViewer();
324 void XGUI_Displayer::UpdateViewer()
326 Handle(AIS_InteractiveContext) ic = AISContext();
328 ic->UpdateCurrentViewer();
331 Handle(AIS_InteractiveContext) XGUI_Displayer::AISContext() const
333 return myWorkshop->viewer()->AISContext();