#include <ModuleBase_IWorkshop.h>
#include <ModuleBase_IViewer.h>
+#include <ModuleBase_Tools.h>
#include <Config_PropManager.h>
//#define DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
-#define OPERATION_PARAMETER_COLOR "255, 255, 0"
-
PartSet_CustomPrs::PartSet_CustomPrs(ModuleBase_IWorkshop* theWorkshop)
- : myWorkshop(theWorkshop), myIsActive(false)
+ : myWorkshop(theWorkshop), myFeature(FeaturePtr())
{
- initPrs();
+ initPresentation(ModuleBase_IModule::CustomizeArguments);
+ initPresentation(ModuleBase_IModule::CustomizeResults);
+ initPresentation(ModuleBase_IModule::CustomizeHighlightedObjects);
+
+ myIsActive[ModuleBase_IModule::CustomizeArguments] = false;
+ myIsActive[ModuleBase_IModule::CustomizeResults] = false;
+ myIsActive[ModuleBase_IModule::CustomizeHighlightedObjects] = false;
}
-bool PartSet_CustomPrs::isActive()
+bool PartSet_CustomPrs::isActive(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
{
- return myIsActive;
- /*Handle(PartSet_OperationPrs) anOperationPrs = getPresentation();
- Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
-
- return !aContext.IsNull() && aContext->IsDisplayed(anOperationPrs);*/
+ return myIsActive[theFlag];
}
-bool PartSet_CustomPrs::activate(const FeaturePtr& theFeature, const bool theUpdateViewer)
+bool PartSet_CustomPrs::activate(const FeaturePtr& theFeature,
+ const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
+ const bool theUpdateViewer)
{
#ifdef DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
return false;
#endif
- bool isModified = false;
- Handle(PartSet_OperationPrs) anOperationPrs = getPresentation();
+ myIsActive[theFlag] = true;
+ myFeature = theFeature;
- myIsActive = true;
- anOperationPrs->setFeature(theFeature);
+ bool isModified = false;
if (theFeature.get()) {
- displayPresentation(theUpdateViewer);
+ displayPresentation(ModuleBase_IModule::CustomizeArguments, theUpdateViewer);
+ displayPresentation(ModuleBase_IModule::CustomizeResults, theUpdateViewer);
+ displayPresentation(ModuleBase_IModule::CustomizeHighlightedObjects, theUpdateViewer);
isModified = true;
}
return isModified;
}
-bool PartSet_CustomPrs::deactivate(const bool theUpdateViewer)
+bool PartSet_CustomPrs::deactivate(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
+ const bool theUpdateViewer)
{
- myIsActive = false;
+ myIsActive[theFlag] = false;
bool isModified = false;
- Handle(PartSet_OperationPrs) anOperationPrs = getPresentation();
- anOperationPrs->setFeature(FeaturePtr());
-
- Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
- if (!aContext.IsNull() && aContext->IsDisplayed(anOperationPrs)) {
- erasePresentation(theUpdateViewer);
- isModified = true;
- }
+ erasePresentation(ModuleBase_IModule::CustomizeArguments, theUpdateViewer);
+ erasePresentation(ModuleBase_IModule::CustomizeResults, theUpdateViewer);
+ erasePresentation(ModuleBase_IModule::CustomizeHighlightedObjects, theUpdateViewer);
+ isModified = true;
return isModified;
}
-bool PartSet_CustomPrs::displayPresentation(const bool theUpdateViewer)
+bool PartSet_CustomPrs::displayPresentation(
+ const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
+ const bool theUpdateViewer)
{
bool isModified = false;
- Handle(PartSet_OperationPrs) anOperationPrs = getPresentation();
- anOperationPrs->updateShapes();
+ // update the AIS objects content
+ Handle(PartSet_OperationPrs) anOperationPrs = getPresentation(theFlag);
+ // do nothing if the feature can not be displayed [is moved from presentation, to be checked]
+ if (!myFeature.get())
+ return isModified;
+
+ switch (theFlag) {
+ case ModuleBase_IModule::CustomizeArguments:
+ PartSet_OperationPrs::getFeatureShapes(myFeature, myWorkshop, anOperationPrs->featureShapes());
+ break;
+ case ModuleBase_IModule::CustomizeResults:
+ PartSet_OperationPrs::getResultShapes(myFeature, myWorkshop, anOperationPrs->featureShapes());
+ break;
+ case ModuleBase_IModule::CustomizeHighlightedObjects:
+ PartSet_OperationPrs::getHighlightedShapes(myWorkshop, anOperationPrs->featureShapes());
+ break;
+ default:
+ return isModified;
+ }
+
+ // redisplay AIS objects
Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
if (!aContext.IsNull() && !aContext->IsDisplayed(anOperationPrs)) {
- if (anOperationPrs->hasShapes()) {
+ // when the feature can not be visualized in the module, the operation preview should not
+ // be visualized also
+ if (anOperationPrs->hasShapes() && myWorkshop->module()->canDisplayObject(myFeature)) {
+ // set color here because it can be changed in preferences
+ Quantity_Color aShapeColor = getShapeColor(theFlag);
+ anOperationPrs->setShapeColor(aShapeColor);
+
PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(myWorkshop->module());
XGUI_Workshop* aWorkshop = workshop();
- aWorkshop->displayer()->displayAIS(myOperationPrs, false/*load object in selection*/, theUpdateViewer);
+ aWorkshop->displayer()->displayAIS(myPresentations[theFlag], false/*load object in selection*/,
+ theUpdateViewer);
aContext->SetZLayer(anOperationPrs, aModule->getVisualLayerId());
isModified = true;
}
}
else {
- if (!anOperationPrs->hasShapes()) {
- erasePresentation(theUpdateViewer);
+ // when the feature can not be visualized in the module, the operation preview should not
+ // be visualized also
+ if (!anOperationPrs->hasShapes() || !myWorkshop->module()->canDisplayObject(myFeature)) {
+ erasePresentation(theFlag, theUpdateViewer);
isModified = true;
}
else {
return isModified;
}
-void PartSet_CustomPrs::erasePresentation(const bool theUpdateViewer)
+void PartSet_CustomPrs::erasePresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
+ const bool theUpdateViewer)
{
XGUI_Workshop* aWorkshop = workshop();
- aWorkshop->displayer()->eraseAIS(myOperationPrs, theUpdateViewer);
+ aWorkshop->displayer()->eraseAIS(myPresentations[theFlag], theUpdateViewer);
}
-Handle(PartSet_OperationPrs) PartSet_CustomPrs::getPresentation()
+void PartSet_CustomPrs::clearPresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
{
- if (!myOperationPrs.get())
- initPrs();
- Handle(AIS_InteractiveObject) anAISIO = myOperationPrs->impl<Handle(AIS_InteractiveObject)>();
- return Handle(PartSet_OperationPrs)::DownCast(anAISIO);
+ Handle(PartSet_OperationPrs) anOperationPrs = getPresentation(theFlag);
+ if (!anOperationPrs.IsNull())
+ anOperationPrs.Nullify();
+}
+
+Handle(PartSet_OperationPrs) PartSet_CustomPrs::getPresentation(
+ const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
+{
+ Handle(PartSet_OperationPrs) aPresentation;
+
+ if (myPresentations.contains(theFlag)) {
+ AISObjectPtr anOperationPrs = myPresentations[theFlag];
+ if (!anOperationPrs.get())
+ initPresentation(theFlag);
+ Handle(AIS_InteractiveObject) anAISIO = anOperationPrs->impl<Handle(AIS_InteractiveObject)>();
+ aPresentation = Handle(PartSet_OperationPrs)::DownCast(anAISIO);
+ }
+ return aPresentation;
}
bool PartSet_CustomPrs::redisplay(const ObjectPtr& theObject,
#ifdef DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
return false;
#endif
- return displayPresentation(theUpdateViewer);
+ return displayPresentation(theFlag, theUpdateViewer);
}
void PartSet_CustomPrs::clearPrs()
{
- Handle(PartSet_OperationPrs) anOperationPrs = getPresentation();
- if (!anOperationPrs.IsNull())
- anOperationPrs.Nullify();
-
- myOperationPrs.reset();
+ clearPresentation(ModuleBase_IModule::CustomizeArguments);
+ clearPresentation(ModuleBase_IModule::CustomizeResults);
+ clearPresentation(ModuleBase_IModule::CustomizeHighlightedObjects);
}
-void PartSet_CustomPrs::initPrs()
+void PartSet_CustomPrs::initPresentation(const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
{
- myOperationPrs = AISObjectPtr(new GeomAPI_AISObject());
- myOperationPrs->setImpl(new Handle(AIS_InteractiveObject)(new PartSet_OperationPrs(myWorkshop)));
+ AISObjectPtr anOperationPrs = AISObjectPtr(new GeomAPI_AISObject());
+ Handle(PartSet_OperationPrs) anAISPrs = new PartSet_OperationPrs(myWorkshop);
+ anOperationPrs->setImpl(new Handle(AIS_InteractiveObject)(anAISPrs));
+ if (theFlag == ModuleBase_IModule::CustomizeArguments ||
+ theFlag == ModuleBase_IModule::CustomizeResults) {
+ anOperationPrs->setPointMarker(5, 2.);
+ anOperationPrs->setWidth(1);
+ }
+ else if (theFlag == ModuleBase_IModule::CustomizeHighlightedObjects)
+ anAISPrs->useAISWidth();
- std::vector<int> aColor = Config_PropManager::color("Visualization", "operation_parameter_color",
- OPERATION_PARAMETER_COLOR);
- myOperationPrs->setColor(aColor[0], aColor[1], aColor[2]);
+ if (anOperationPrs.get())
+ myPresentations[theFlag] = anOperationPrs;
+}
- myOperationPrs->setPointMarker(5, 2.);
- myOperationPrs->setWidth(1);
+Quantity_Color PartSet_CustomPrs::getShapeColor(
+ const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag)
+{
+ Quantity_Color aColor;
+ switch(theFlag) {
+ case ModuleBase_IModule::CustomizeArguments:
+ aColor = ModuleBase_Tools::color("Visualization", "operation_parameter_color",
+ OPERATION_PARAMETER_COLOR());
+ break;
+ case ModuleBase_IModule::CustomizeResults:
+ aColor = ModuleBase_Tools::color("Visualization", "operation_result_color",
+ OPERATION_RESULT_COLOR());
+ break;
+ case ModuleBase_IModule::CustomizeHighlightedObjects:
+ aColor = ModuleBase_Tools::color("Visualization", "operation_highlight_color",
+ OPERATION_HIGHLIGHT_COLOR());
+ break;
+ default:
+ break;
+ }
+ return aColor;
}
XGUI_Workshop* PartSet_CustomPrs::workshop() const