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