src/PartSet/PartSet_CustomPrs.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        PartSet_CustomPrs.cpp
   4 // Created:     30 Jun 2015
   5 // Author:      Natalia ERMOLAEVA
   6
   7 #include <PartSet_CustomPrs.h>
   8 #include <PartSet_Module.h>
   9 #include "PartSet_OperationPrs.h"
  10
  11 #include <XGUI_ModuleConnector.h>
  12 #include <XGUI_Workshop.h>
  13 #include <XGUI_Displayer.h>
  14
  15 #include <ModuleBase_IWorkshop.h>
  16 #include <ModuleBase_IViewer.h>
  17
  18 #include <Config_PropManager.h>
  19
  20 #include <AIS_InteractiveContext.hxx>
  21 #include <AIS_InteractiveObject.hxx>
  22 #include <Prs3d_PointAspect.hxx>
  23
  24 //#define DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
  25
  26 #define OPERATION_PARAMETER_COLOR "255, 255, 0"
  27
  28 PartSet_CustomPrs::PartSet_CustomPrs(ModuleBase_IWorkshop* theWorkshop)
  29   : myWorkshop(theWorkshop), myIsActive(false)
  30 {
  31   initPrs();
  32 }
  33
  34 bool PartSet_CustomPrs::isActive()
  35 {
  36   return myIsActive;
  37   /*Handle(PartSet_OperationPrs) anOperationPrs = getPresentation();
  38   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
  39
  40   return !aContext.IsNull() && aContext->IsDisplayed(anOperationPrs);*/
  41 }
  42
  43 bool PartSet_CustomPrs::activate(const FeaturePtr& theFeature, const bool theUpdateViewer)
  44 {
  45 #ifdef DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
  46   return false;
  47 #endif
  48
  49   bool isModified = false;
  50   Handle(PartSet_OperationPrs) anOperationPrs = getPresentation();
  51
  52   myIsActive = true;
  53   anOperationPrs->setFeature(theFeature);
  54   if (theFeature.get()) {
  55     displayPresentation(theUpdateViewer);
  56     isModified = true;
  57   }
  58   return isModified;
  59 }
  60
  61 bool PartSet_CustomPrs::deactivate(const bool theUpdateViewer)
  62 {
  63   myIsActive = false;
  64   bool isModified = false;
  65
  66   Handle(PartSet_OperationPrs) anOperationPrs = getPresentation();
  67   anOperationPrs->setFeature(FeaturePtr());
  68
  69   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
  70   if (!aContext.IsNull() && aContext->IsDisplayed(anOperationPrs)) {
  71     erasePresentation(theUpdateViewer);
  72     isModified = true;
  73   }
  74
  75   return isModified;
  76 }
  77
  78 bool PartSet_CustomPrs::displayPresentation(const bool theUpdateViewer)
  79 {
  80   bool isModified = false;
  81   Handle(PartSet_OperationPrs) anOperationPrs = getPresentation();
  82   anOperationPrs->updateShapes();
  83
  84   Handle(AIS_InteractiveContext) aContext = myWorkshop->viewer()->AISContext();
  85   if (!aContext.IsNull() && !aContext->IsDisplayed(anOperationPrs)) {
  86     if (anOperationPrs->hasShapes()) {
  87       PartSet_Module* aModule = dynamic_cast<PartSet_Module*>(myWorkshop->module());
  88       XGUI_Workshop* aWorkshop = workshop();
  89       aWorkshop->displayer()->displayAIS(myOperationPrs, false/*load object in selection*/, theUpdateViewer);
  90       aContext->SetZLayer(anOperationPrs, aModule->getVisualLayerId());
  91       isModified = true;
  92     }
  93   }
  94   else {
  95     if (!anOperationPrs->hasShapes()) {
  96       erasePresentation(theUpdateViewer);
  97       isModified = true;
  98     }
  99     else {
 100       anOperationPrs->Redisplay();
 101       isModified = true;
 102       if (theUpdateViewer)
 103         workshop()->displayer()->updateViewer();
 104     }
 105   }
 106   return isModified;
 107 }
 108
 109 void PartSet_CustomPrs::erasePresentation(const bool theUpdateViewer)
 110 {
 111   XGUI_Workshop* aWorkshop = workshop();
 112   aWorkshop->displayer()->eraseAIS(myOperationPrs, theUpdateViewer);
 113 }
 114
 115 Handle(PartSet_OperationPrs) PartSet_CustomPrs::getPresentation()
 116 {
 117   if (!myOperationPrs.get())
 118     initPrs();
 119   Handle(AIS_InteractiveObject) anAISIO = myOperationPrs->impl<Handle(AIS_InteractiveObject)>();
 120   return Handle(PartSet_OperationPrs)::DownCast(anAISIO);
 121 }
 122
 123 bool PartSet_CustomPrs::redisplay(const ObjectPtr& theObject,
 124                                   const ModuleBase_IModule::ModuleBase_CustomizeFlag& theFlag,
 125                                   const bool theUpdateViewer)
 126 {
 127 #ifdef DO_NOT_VISUALIZE_CUSTOM_PRESENTATION
 128   return false;
 129 #endif
 130   return displayPresentation(theUpdateViewer);
 131 }
 132
 133 void PartSet_CustomPrs::clearPrs()
 134 {
 135   Handle(PartSet_OperationPrs) anOperationPrs = getPresentation();
 136   if (!anOperationPrs.IsNull())
 137     anOperationPrs.Nullify();
 138
 139   myOperationPrs.reset();
 140 }
 141
 142 void PartSet_CustomPrs::initPrs()
 143 {
 144   myOperationPrs = AISObjectPtr(new GeomAPI_AISObject());
 145   myOperationPrs->setImpl(new Handle(AIS_InteractiveObject)(new PartSet_OperationPrs(myWorkshop)));
 146
 147   std::vector<int> aColor = Config_PropManager::color("Visualization", "operation_parameter_color",
 148                                                       OPERATION_PARAMETER_COLOR);
 149   myOperationPrs->setColor(aColor[0], aColor[1], aColor[2]);
 150
 151   myOperationPrs->setPointMarker(5, 2.);
 152   myOperationPrs->setWidth(1);
 153 }
 154
 155 XGUI_Workshop* PartSet_CustomPrs::workshop() const
 156 {
 157   XGUI_ModuleConnector* aConnector = dynamic_cast<XGUI_ModuleConnector*>(myWorkshop);
 158   return aConnector->workshop();
 159 }