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 <PartSet_CustomPrs.h>
21 #include <PartSet_Module.h>
22 #include "PartSet_OperationPrs.h"
23 #include "PartSet_OverconstraintListener.h"
25 #include <XGUI_ModuleConnector.h>
26 #include <XGUI_Workshop.h>
27 #include <XGUI_Displayer.h>
29 #include <ModuleBase_IWorkshop.h>
30 #include <ModuleBase_IViewer.h>
31 #include <ModuleBase_Tools.h>
33 #include <Config_PropManager.h>
34 #include <Events_Loop.h>
35 #include <ModelAPI_Events.h>
37 #include <AIS_InteractiveContext.hxx>
38 #include <AIS_InteractiveObject.hxx>
39 #include <Prs3d_PointAspect.hxx>
41 //#define DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
43 PartSet_CustomPrs::PartSet_CustomPrs(ModuleBase_IWorkshop* theWorkshop)
44 : myWorkshop(theWorkshop), myFeature(FeaturePtr()), myPresentationIsEmpty(false),
47 Events_Loop* aLoop = Events_Loop::loop();
48 aLoop->registerListener(this, Events_Loop::eventByName(EVENT_EMPTY_OPERATION_PRESENTATION));
50 initPresentation(ModuleBase_IModule::CustomizeArguments);
51 initPresentation(ModuleBase_IModule::CustomizeResults);
52 initPresentation(ModuleBase_IModule::CustomizeHighlightedObjects);
54 myIsActive[ModuleBase_IModule::CustomizeArguments] = false;
55 myIsActive[ModuleBase_IModule::CustomizeResults] = false;
56 myIsActive[ModuleBase_IModule::CustomizeHighlightedObjects] = false;
59 bool PartSet_CustomPrs::isActive(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
61 return myIsActive[theFlag];
64 bool PartSet_CustomPrs::activate(const FeaturePtr& theFeature,
65 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
66 const bool theUpdateViewer)
68 #ifdef DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
72 myIsActive[theFlag] = true;
73 myFeature = theFeature;
75 bool isModified = false;
76 if (theFeature.get()) {
77 displayPresentation(theFlag, theUpdateViewer);
83 bool PartSet_CustomPrs::deactivate(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
84 const bool theUpdateViewer)
86 myIsActive[theFlag] = false;
87 erasePresentation(theFlag, theUpdateViewer);
91 bool PartSet_CustomPrs::displayPresentation(
92 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
93 const bool theUpdateViewer)
95 bool isModified = false;
97 if (myDisabledMode == theFlag)
100 // update the AIS objects content
101 AISObjectPtr aPresentation = getPresentation(theFlag, true);
102 Handle(AIS_InteractiveObject) anAISIO = aPresentation->impl<Handle(AIS_InteractiveObject)>();
103 Handle(PartSet_OperationPrs) anOperationPrs = Handle(PartSet_OperationPrs)::DownCast(anAISIO);
105 // do nothing if the feature can not be displayed [is moved from presentation, to be checked]
106 if (!myFeature.get())
109 QMap<ObjectPtr, QList<GeomShapePtr> > aFeatureShapes;
111 case ModuleBase_IModule::CustomizeArguments:
112 PartSet_OperationPrs::getFeatureShapes(myFeature, myWorkshop, aFeatureShapes);
114 case ModuleBase_IModule::CustomizeResults:
115 PartSet_OperationPrs::getResultShapes(myFeature, myWorkshop, aFeatureShapes);
116 PartSet_OperationPrs::getPresentationShapes(myFeature, myWorkshop, aFeatureShapes, false);
118 case ModuleBase_IModule::CustomizeHighlightedObjects:
119 PartSet_OperationPrs::getHighlightedShapes(myWorkshop, aFeatureShapes);
124 NCollection_DataMap<TopoDS_Shape, Handle(AIS_InteractiveObject)>& aShapeMap =
125 anOperationPrs->shapesMap();
126 PartSet_OperationPrs::fillShapeList(aFeatureShapes, myWorkshop, aShapeMap);
128 myPresentationIsEmpty = false;
129 // redisplay AIS objects
130 bool aRedisplayed = false;
131 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
132 if (!aContext.IsNull() && !aContext->IsDisplayed(anOperationPrs)) {
133 // when the feature can not be visualized in the module, the operation preview should not
134 // be visualized also
135 if (anOperationPrs->hasShapes() && myWorkshop->module()->canDisplayObject(myFeature)) {
136 // set color here because it can be changed in preferences
137 Quantity_Color aShapeColor = getShapeColor(theFlag);
138 anOperationPrs->setShapeColor(aShapeColor);
140 PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(myWorkshop->module());
141 XGUI_Workshop* aWorkshop = workshop();
142 aRedisplayed = aWorkshop->displayer()->displayAIS(myPresentations[theFlag],
143 false/*load object in selection*/, false);
144 aContext->SetZLayer(anOperationPrs, aModule->getVisualLayerId());
149 // when the feature can not be visualized in the module, the operation preview should not
150 // be visualized also
151 if (!anOperationPrs->hasShapes() || !myWorkshop->module()->canDisplayObject(myFeature)) {
152 aRedisplayed = erasePresentation(theFlag, false);
156 anOperationPrs->Redisplay();
161 if (myPresentationIsEmpty) {
162 aRedisplayed = erasePresentation(theFlag, false);
164 if (aRedisplayed && theUpdateViewer)
165 workshop()->displayer()->updateViewer();
170 bool PartSet_CustomPrs::erasePresentation(
171 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
172 const bool theUpdateViewer)
174 bool isErased = false;
175 XGUI_Workshop* aWorkshop = workshop();
176 if (myPresentations.contains(theFlag))
177 isErased = aWorkshop->displayer()->eraseAIS(myPresentations[theFlag], theUpdateViewer);
181 void PartSet_CustomPrs::clearPresentation(
182 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
184 AISObjectPtr aPresentation = getPresentation(theFlag, false);
185 if (aPresentation.get()) {
186 Handle(AIS_InteractiveObject) anAISIO = aPresentation->impl<Handle(AIS_InteractiveObject)>();
187 Handle(PartSet_OperationPrs) anOperationPrs = Handle(PartSet_OperationPrs)::DownCast(anAISIO);
189 anOperationPrs->shapesMap().Clear();
190 if (!anOperationPrs.IsNull())
191 anOperationPrs.Nullify();
192 myPresentations.remove(theFlag);
196 AISObjectPtr PartSet_CustomPrs::getPresentation(
197 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
198 const bool theToCreate)
200 Handle(PartSet_OperationPrs) aPresentation;
202 AISObjectPtr anOperationPrs;
203 if (myPresentations.contains(theFlag))
204 anOperationPrs = myPresentations[theFlag];
206 if (!anOperationPrs.get() && theToCreate) {
207 initPresentation(theFlag);
208 anOperationPrs = myPresentations[theFlag];
211 return anOperationPrs;
214 bool PartSet_CustomPrs::redisplay(const ObjectPtr& theObject,
215 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
216 const bool theUpdateViewer)
218 #ifdef DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
221 bool aRedisplayed = false;
222 if (myIsActive[theFlag])
223 aRedisplayed = displayPresentation(theFlag, theUpdateViewer);
228 void PartSet_CustomPrs::clearPrs()
230 clearPresentation(ModuleBase_IModule::CustomizeArguments);
231 clearPresentation(ModuleBase_IModule::CustomizeResults);
232 clearPresentation(ModuleBase_IModule::CustomizeHighlightedObjects);
235 void PartSet_CustomPrs::processEvent(const std::shared_ptr<Events_Message>& theMessage)
237 if (theMessage->eventID() == Events_Loop::eventByName(EVENT_EMPTY_OPERATION_PRESENTATION))
238 myPresentationIsEmpty = true; /// store state to analize it after display/erase is finished
241 void PartSet_CustomPrs::initPresentation(
242 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
244 AISObjectPtr anOperationPrs = AISObjectPtr(new GeomAPI_AISObject());
245 Handle(PartSet_OperationPrs) anAISPrs = new PartSet_OperationPrs(myWorkshop);
246 anOperationPrs->setImpl(new Handle(AIS_InteractiveObject)(anAISPrs));
247 if (theFlag == ModuleBase_IModule::CustomizeArguments ||
248 theFlag == ModuleBase_IModule::CustomizeResults) {
249 anOperationPrs->setPointMarker(5, 2.);
250 anOperationPrs->setWidth((theFlag == ModuleBase_IModule::CustomizeHighlightedObjects)? 2 : 1);
252 else if (theFlag == ModuleBase_IModule::CustomizeHighlightedObjects)
253 anAISPrs->useAISWidth();
255 if (anOperationPrs.get())
256 myPresentations[theFlag] = anOperationPrs;
259 Quantity_Color PartSet_CustomPrs::getShapeColor(
260 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
262 Quantity_Color aColor;
264 case ModuleBase_IModule::CustomizeArguments:
265 aColor = ModuleBase_Tools::color("Visualization", "operation_parameter_color");
267 case ModuleBase_IModule::CustomizeResults:
268 aColor = ModuleBase_Tools::color("Visualization", "operation_result_color");
270 case ModuleBase_IModule::CustomizeHighlightedObjects:
271 aColor = ModuleBase_Tools::color("Visualization", "operation_highlight_color");
279 XGUI_Workshop* PartSet_CustomPrs::workshop() const
281 XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(myWorkshop);
282 return aConnector->workshop();