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     // result constructions should not be taken as a body
  37     if (aResult.get() != NULL && aResult->groupName() == ModelAPI_ResultBody::group()) {
  38       GeomShapePtr aShape = aResult->shape();
  39       if (aShape.get()) {
  40         // vertices, edges should not be taken as a body
  41         aBodyIsVisualized = aShape->shapeType() <= GeomAPI_Shape::FACE;
  42         if (aBodyIsVisualized)
  43           break;
  44       }
  45     }
  46   }
  47   return aBodyIsVisualized;
  48 }
  49
  50 bool PartSet_PreviewPlanes::hasVisualizedSketch(ModuleBase_IWorkshop* theWorkshop)
  51 {
  52   bool aSketchIsVisualized = false;
  53
  54   XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(theWorkshop);
  55   XGUI_Displayer* aDisp = aWorkshop->displayer();
  56   QObjectPtrList aDisplayed = aDisp->displayedObjects();
  57   foreach (ObjectPtr anObj, aDisplayed) {
  58     ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObj);
  59     if (aResult.get() != NULL) {
  60       FeaturePtr aFeature = ModelAPI_Feature::feature(aResult);
  61       aSketchIsVisualized = aFeature.get() && aFeature->getKind() == SketchPlugin_Sketch::ID();
  62       if (aSketchIsVisualized)
  63         break;
  64     }
  65   }
  66   return aSketchIsVisualized;
  67 }
  68
  69 void PartSet_PreviewPlanes::erasePreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
  70 {
  71   if (myPreviewDisplayed) {
  72     XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
  73     aDisp->eraseAIS(myYZPlane, false);
  74     aDisp->eraseAIS(myXZPlane, false);
  75     aDisp->eraseAIS(myXYPlane, false);
  76     myPreviewDisplayed = false;
  77   }
  78 }
  79
  80 void PartSet_PreviewPlanes::showPreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
  81 {
  82   if (myPreviewDisplayed)
  83     return;
  84
  85   if (!myYZPlane) { // If planes are not created
  86     // Create Preview
  87     std::shared_ptr<GeomAPI_Pnt> anOrigin(new GeomAPI_Pnt(0, 0, 0));
  88     std::shared_ptr<GeomAPI_Dir> aYZDir(new GeomAPI_Dir(1, 0, 0));
  89     // -1, not 1 for correct internal sketch coords (issue 898)
  90     std::shared_ptr<GeomAPI_Dir> aXZDir(new GeomAPI_Dir(0, -1, 0));
  91     std::shared_ptr<GeomAPI_Dir> aXYDir(new GeomAPI_Dir(0, 0, 1));
  92
  93     std::vector<int> aYZRGB, aXZRGB, aXYRGB;
  94     aYZRGB = Config_PropManager::color("Visualization", "yz_plane_color",
  95                                                         YZ_PLANE_COLOR);
  96     aXZRGB = Config_PropManager::color("Visualization", "xz_plane_color",
  97                                                         XZ_PLANE_COLOR);
  98     aXYRGB = Config_PropManager::color("Visualization", "xy_plane_color",
  99                                                         XY_PLANE_COLOR);
 100     int aR[] = {aYZRGB[0], aYZRGB[1], aYZRGB[2]};
 101     int aG[] = {aXZRGB[0], aXZRGB[1], aXZRGB[2]};
 102     int aB[] = {aXYRGB[0], aXYRGB[1], aXYRGB[2]};
 103
 104     myYZPlane = createPreviewPlane(anOrigin, aYZDir, aR);
 105     myXZPlane = createPreviewPlane(anOrigin, aXZDir, aG);
 106     myXYPlane = createPreviewPlane(anOrigin, aXYDir, aB);
 107   }
 108   XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
 109   aDisp->displayAIS(myYZPlane, true, false);
 110   aDisp->displayAIS(myXZPlane, true, false);
 111   aDisp->displayAIS(myXYPlane, true, false);
 112   myPreviewDisplayed = true;
 113 }
 114
 115 AISObjectPtr PartSet_PreviewPlanes::createPreviewPlane(std::shared_ptr<GeomAPI_Pnt> theOrigin,
 116                                                        std::shared_ptr<GeomAPI_Dir> theNorm,
 117                                                        const int theRGB[3])
 118 {
 119   double aSize = Config_PropManager::integer("Sketch planes", "planes_size", PLANE_SIZE);
 120   std::shared_ptr<GeomAPI_Shape> aFace = GeomAlgoAPI_FaceBuilder::square(theOrigin, theNorm, aSize);
 121   AISObjectPtr aAIS = AISObjectPtr(new GeomAPI_AISObject());
 122   aAIS->createShape(aFace);
 123   aAIS->setWidth(Config_PropManager::integer("Sketch planes", "planes_thickness", SKETCH_WIDTH));
 124   aAIS->setColor(theRGB[0], theRGB[1], theRGB[2]);
 125   return aAIS;
 126 }