src/PartSet/PartSet_PreviewPlanes.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        PartSet_PreviewPlanes.cpp
   4 // Created:     19 Jun 2015
   5 // Author:      Natalia ERMOLAEVA
   6
   7 #include "PartSet_PreviewPlanes.h"
   8
   9 #include <ModuleBase_IWorkshop.h>
  10
  11 #include <ModelAPI_ResultBody.h>
  12
  13 #include <XGUI_Tools.h>
  14 #include <XGUI_Displayer.h>
  15 #include <XGUI_Workshop.h>
  16
  17 #include <Config_PropManager.h>
  18 #include <GeomAlgoAPI_FaceBuilder.h>
  19
  20 #include <SketchPlugin_Sketch.h>
  21
  22 PartSet_PreviewPlanes::PartSet_PreviewPlanes()
  23  : myPreviewDisplayed(false)
  24 {
  25 }
  26
  27 bool PartSet_PreviewPlanes::hasVisualizedBodies(ModuleBase_IWorkshop* theWorkshop)
  28 {
  29   bool aBodyIsVisualized = false;
  30
  31   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(theWorkshop);
  32   XGUI_Displayer* aDisp = aWorkshop->displayer();
  33   QObjectPtrList aDisplayed = aDisp->displayedObjects();
  34   foreach (ObjectPtr anObj, aDisplayed) {
  35     ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObj);
  36     if (aResult.get() != NULL) {
  37       aBodyIsVisualized = aResult->groupName() == ModelAPI_ResultBody::group();
  38       if (aBodyIsVisualized)
  39         break;
  40     }
  41   }
  42   return aBodyIsVisualized;
  43 }
  44
  45 bool PartSet_PreviewPlanes::hasVisualizedSketch(ModuleBase_IWorkshop* theWorkshop)
  46 {
  47   bool aSketchIsVisualized = false;
  48
  49   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(theWorkshop);
  50   XGUI_Displayer* aDisp = aWorkshop->displayer();
  51   QObjectPtrList aDisplayed = aDisp->displayedObjects();
  52   foreach (ObjectPtr anObj, aDisplayed) {
  53     ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObj);
  54     if (aResult.get() != NULL) {
  55       FeaturePtr aFeature = ModelAPI_Feature::feature(aResult);
  56       aSketchIsVisualized = aFeature.get() && aFeature->getKind() == SketchPlugin_Sketch::ID();
  57       if (aSketchIsVisualized)
  58         break;
  59     }
  60   }
  61   return aSketchIsVisualized;
  62 }
  63
  64 void PartSet_PreviewPlanes::erasePreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
  65 {
  66   if (myPreviewDisplayed) {
  67     XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
  68     aDisp->eraseAIS(myYZPlane, false);
  69     aDisp->eraseAIS(myXZPlane, false);
  70     aDisp->eraseAIS(myXYPlane, false);
  71     myPreviewDisplayed = false;
  72   }
  73 }
  74
  75 void PartSet_PreviewPlanes::showPreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
  76 {
  77   if (myPreviewDisplayed)
  78     return;
  79
  80   if (!myYZPlane) { // If planes are not created
  81     // Create Preview
  82     std::shared_ptr<GeomAPI_Pnt> anOrigin(new GeomAPI_Pnt(0, 0, 0));
  83     std::shared_ptr<GeomAPI_Dir> aYZDir(new GeomAPI_Dir(1, 0, 0));
  84     // -1, not 1 for correct internal sketch coords (issue 898)
  85     std::shared_ptr<GeomAPI_Dir> aXZDir(new GeomAPI_Dir(0, -1, 0));
  86     std::shared_ptr<GeomAPI_Dir> aXYDir(new GeomAPI_Dir(0, 0, 1));
  87
  88     std::vector<int> aYZRGB, aXZRGB, aXYRGB;
  89     aYZRGB = Config_PropManager::color("Visualization", "yz_plane_color",
  90                                                         YZ_PLANE_COLOR);
  91     aXZRGB = Config_PropManager::color("Visualization", "xz_plane_color",
  92                                                         XZ_PLANE_COLOR);
  93     aXYRGB = Config_PropManager::color("Visualization", "xy_plane_color",
  94                                                         XY_PLANE_COLOR);
  95     int aR[] = {aYZRGB[0], aYZRGB[1], aYZRGB[2]};
  96     int aG[] = {aXZRGB[0], aXZRGB[1], aXZRGB[2]};
  97     int aB[] = {aXYRGB[0], aXYRGB[1], aXYRGB[2]};
  98
  99     myYZPlane = createPreviewPlane(anOrigin, aYZDir, aR);
 100     myXZPlane = createPreviewPlane(anOrigin, aXZDir, aG);
 101     myXYPlane = createPreviewPlane(anOrigin, aXYDir, aB);
 102   }
 103   XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
 104   aDisp->displayAIS(myYZPlane, true, false);
 105   aDisp->displayAIS(myXZPlane, true, false);
 106   aDisp->displayAIS(myXYPlane, true, false);
 107   myPreviewDisplayed = true;
 108 }
 109
 110 AISObjectPtr PartSet_PreviewPlanes::createPreviewPlane(std::shared_ptr<GeomAPI_Pnt> theOrigin,
 111                                                        std::shared_ptr<GeomAPI_Dir> theNorm,
 112                                                        const int theRGB[3])
 113 {
 114   double aSize = Config_PropManager::integer("Sketch planes", "planes_size", PLANE_SIZE);
 115   std::shared_ptr<GeomAPI_Shape> aFace = GeomAlgoAPI_FaceBuilder::square(theOrigin, theNorm, aSize);
 116   AISObjectPtr aAIS = AISObjectPtr(new GeomAPI_AISObject());
 117   aAIS->createShape(aFace);
 118   aAIS->setWidth(Config_PropManager::integer("Sketch planes", "planes_thickness", SKETCH_WIDTH));
 119   aAIS->setColor(theRGB[0], theRGB[1], theRGB[2]);
 120   return aAIS;
 121 }