1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: PartSet_CustomPrs.cpp
4 // Created: 30 Jun 2015
5 // Author: Natalia ERMOLAEVA
7 #include <PartSet_CustomPrs.h>
8 #include <PartSet_Module.h>
9 #include "PartSet_OperationPrs.h"
11 #include <XGUI_ModuleConnector.h>
12 #include <XGUI_Workshop.h>
13 #include <XGUI_Displayer.h>
15 #include <ModuleBase_IWorkshop.h>
16 #include <ModuleBase_IViewer.h>
17 #include <ModuleBase_Tools.h>
19 #include <Config_PropManager.h>
21 #include <AIS_InteractiveContext.hxx>
22 #include <AIS_InteractiveObject.hxx>
23 #include <Prs3d_PointAspect.hxx>
25 //#define DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
27 PartSet_CustomPrs::PartSet_CustomPrs(ModuleBase_IWorkshop* theWorkshop)
28 : myWorkshop(theWorkshop), myFeature(FeaturePtr())
30 initPresentation(ModuleBase_IModule::CustomizeArguments);
31 initPresentation(ModuleBase_IModule::CustomizeResults);
32 initPresentation(ModuleBase_IModule::CustomizeHighlightedObjects);
34 myIsActive[ModuleBase_IModule::CustomizeArguments] = false;
35 myIsActive[ModuleBase_IModule::CustomizeResults] = false;
36 myIsActive[ModuleBase_IModule::CustomizeHighlightedObjects] = false;
39 bool PartSet_CustomPrs::isActive(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
41 return myIsActive[theFlag];
44 bool PartSet_CustomPrs::activate(const FeaturePtr& theFeature,
45 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
46 const bool theUpdateViewer)
48 #ifdef DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
52 myIsActive[theFlag] = true;
53 myFeature = theFeature;
55 bool isModified = false;
56 if (theFeature.get()) {
57 displayPresentation(ModuleBase_IModule::CustomizeArguments, theUpdateViewer);
58 displayPresentation(ModuleBase_IModule::CustomizeResults, theUpdateViewer);
59 displayPresentation(ModuleBase_IModule::CustomizeHighlightedObjects, theUpdateViewer);
65 bool PartSet_CustomPrs::deactivate(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
66 const bool theUpdateViewer)
68 myIsActive[theFlag] = false;
69 bool isModified = false;
71 erasePresentation(ModuleBase_IModule::CustomizeArguments, theUpdateViewer);
72 erasePresentation(ModuleBase_IModule::CustomizeResults, theUpdateViewer);
73 erasePresentation(ModuleBase_IModule::CustomizeHighlightedObjects, theUpdateViewer);
79 bool PartSet_CustomPrs::displayPresentation(
80 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
81 const bool theUpdateViewer)
83 bool isModified = false;
85 // update the AIS objects content
86 Handle(PartSet_OperationPrs) anOperationPrs = getPresentation(theFlag);
87 // do nothing if the feature can not be displayed [is moved from presentation, to be checked]
88 if (!myFeature.get() || !myWorkshop->module()->canDisplayObject(myFeature))
92 case ModuleBase_IModule::CustomizeArguments:
93 PartSet_OperationPrs::getFeatureShapes(myFeature, myWorkshop, anOperationPrs->featureShapes());
95 case ModuleBase_IModule::CustomizeResults:
96 PartSet_OperationPrs::getResultShapes(myFeature, myWorkshop, anOperationPrs->featureShapes());
98 case ModuleBase_IModule::CustomizeHighlightedObjects:
99 PartSet_OperationPrs::getHighlightedShapes(myWorkshop, anOperationPrs->featureShapes());
105 // redisplay AIS objects
106 Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
107 if (!aContext.IsNull() && !aContext->IsDisplayed(anOperationPrs)) {
108 if (anOperationPrs->hasShapes()) {
109 // set color here because it can be changed in preferences
110 Quantity_Color aShapeColor = getShapeColor(theFlag);
111 anOperationPrs->setShapeColor(aShapeColor);
113 PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(myWorkshop->module());
114 XGUI_Workshop* aWorkshop = workshop();
115 aWorkshop->displayer()->displayAIS(myPresentations[theFlag], false/*load object in selection*/,
117 aContext->SetZLayer(anOperationPrs, aModule->getVisualLayerId());
122 if (!anOperationPrs->hasShapes()) {
123 erasePresentation(theFlag, theUpdateViewer);
127 anOperationPrs->Redisplay();
130 workshop()->displayer()->updateViewer();
136 void PartSet_CustomPrs::erasePresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
137 const bool theUpdateViewer)
139 XGUI_Workshop* aWorkshop = workshop();
140 aWorkshop->displayer()->eraseAIS(myPresentations[theFlag], theUpdateViewer);
143 void PartSet_CustomPrs::clearPresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
145 Handle(PartSet_OperationPrs) anOperationPrs = getPresentation(theFlag);
146 if (!anOperationPrs.IsNull())
147 anOperationPrs.Nullify();
150 Handle(PartSet_OperationPrs) PartSet_CustomPrs::getPresentation(
151 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
153 Handle(PartSet_OperationPrs) aPresentation;
155 if (myPresentations.contains(theFlag)) {
156 AISObjectPtr anOperationPrs = myPresentations[theFlag];
157 if (!anOperationPrs.get())
158 initPresentation(theFlag);
159 Handle(AIS_InteractiveObject) anAISIO = anOperationPrs->impl<Handle(AIS_InteractiveObject)>();
160 aPresentation = Handle(PartSet_OperationPrs)::DownCast(anAISIO);
162 return aPresentation;
165 bool PartSet_CustomPrs::redisplay(const ObjectPtr& theObject,
166 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
167 const bool theUpdateViewer)
169 #ifdef DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
172 return displayPresentation(theFlag, theUpdateViewer);
175 void PartSet_CustomPrs::clearPrs()
177 clearPresentation(ModuleBase_IModule::CustomizeArguments);
178 clearPresentation(ModuleBase_IModule::CustomizeResults);
179 clearPresentation(ModuleBase_IModule::CustomizeHighlightedObjects);
182 void PartSet_CustomPrs::initPresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
184 AISObjectPtr anOperationPrs = AISObjectPtr(new GeomAPI_AISObject());
185 Handle(PartSet_OperationPrs) anAISPrs = new PartSet_OperationPrs(myWorkshop);
186 anOperationPrs->setImpl(new Handle(AIS_InteractiveObject)(anAISPrs));
187 if (theFlag == ModuleBase_IModule::CustomizeArguments ||
188 theFlag == ModuleBase_IModule::CustomizeResults) {
189 anOperationPrs->setPointMarker(5, 2.);
190 anOperationPrs->setWidth(1);
192 else if (theFlag == ModuleBase_IModule::CustomizeHighlightedObjects)
193 anAISPrs->useAISWidth();
195 if (anOperationPrs.get())
196 myPresentations[theFlag] = anOperationPrs;
199 Quantity_Color PartSet_CustomPrs::getShapeColor(
200 const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
202 Quantity_Color aColor;
204 case ModuleBase_IModule::CustomizeArguments:
205 aColor = ModuleBase_Tools::color("Visualization", "operation_parameter_color",
206 OPERATION_PARAMETER_COLOR());
208 case ModuleBase_IModule::CustomizeResults:
209 aColor = ModuleBase_Tools::color("Visualization", "operation_result_color",
210 OPERATION_RESULT_COLOR());
212 case ModuleBase_IModule::CustomizeHighlightedObjects:
213 aColor = ModuleBase_Tools::color("Visualization", "operation_highlight_color",
214 OPERATION_HIGHLIGHT_COLOR());
222 XGUI_Workshop* PartSet_CustomPrs::workshop() const
224 XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(myWorkshop);
225 return aConnector->workshop();