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 bool PartSet_PreviewPlanes::isPreviewShape(std::shared_ptr<GeomAPI_Shape> theShape)
  77 {
  78   return (myYZPlane->getShape()->isEqual(theShape) ||
  79           myXZPlane->getShape()->isEqual(theShape) ||
  80           myXYPlane->getShape()->isEqual(theShape));
  81 }
  82
  83 void PartSet_PreviewPlanes::erasePreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
  84 {
  85   if (myPreviewDisplayed) {
  86     XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
  87     aDisp->eraseAIS(myYZPlane, false);
  88     aDisp->eraseAIS(myXZPlane, false);
  89     aDisp->eraseAIS(myXYPlane, false);
  90     myPreviewDisplayed = false;
  91   }
  92 }
  93
  94 void PartSet_PreviewPlanes::showPreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
  95 {
  96   if (myPreviewDisplayed)
  97     return;
  98
  99   if (!myYZPlane) { // If planes are not created
 100     // Create Preview
 101     std::shared_ptr<GeomAPI_Pnt> anOrigin(new GeomAPI_Pnt(0, 0, 0));
 102     std::shared_ptr<GeomAPI_Dir> aYZDir(new GeomAPI_Dir(1, 0, 0));
 103     // -1, not 1 for correct internal sketch coords (issue 898)
 104     std::shared_ptr<GeomAPI_Dir> aXZDir(new GeomAPI_Dir(0, -1, 0));
 105     std::shared_ptr<GeomAPI_Dir> aXYDir(new GeomAPI_Dir(0, 0, 1));
 106
 107     std::vector<int> aYZRGB(3, 0), aXZRGB(3, 0), aXYRGB(3, 0);
 108 #ifdef SET_PLANES_COLOR_IN_PREFERENCES
 109     aYZRGB = Config_PropManager::color("Visualization", "yz_plane_color");
 110     aXZRGB = Config_PropManager::color("Visualization", "xz_plane_color");
 111     aXYRGB = Config_PropManager::color("Visualization", "xy_plane_color");
 112 #else
 113     aYZRGB[0] = 225;
 114     aXZRGB[1] = 225;
 115     aXYRGB[2] = 225;
 116 #endif
 117     int aR[] = {aYZRGB[0], aYZRGB[1], aYZRGB[2]};
 118     int aG[] = {aXZRGB[0], aXZRGB[1], aXZRGB[2]};
 119     int aB[] = {aXYRGB[0], aXYRGB[1], aXYRGB[2]};
 120
 121     myYZPlane = createPreviewPlane(anOrigin, aYZDir, aR);
 122     myXZPlane = createPreviewPlane(anOrigin, aXZDir, aG);
 123     myXYPlane = createPreviewPlane(anOrigin, aXYDir, aB);
 124   }
 125   XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
 126   aDisp->displayAIS(myYZPlane, true, false);
 127   aDisp->displayAIS(myXZPlane, true, false);
 128   aDisp->displayAIS(myXYPlane, true, false);
 129   myPreviewDisplayed = true;
 130 }
 131
 132 AISObjectPtr PartSet_PreviewPlanes::createPreviewPlane(std::shared_ptr<GeomAPI_Pnt> theOrigin,
 133                                                        std::shared_ptr<GeomAPI_Dir> theNorm,
 134                                                        const int theRGB[3])
 135 {
 136   double aSize = Config_PropManager::integer(SKETCH_TAB_NAME, "planes_size");
 137   std::shared_ptr<GeomAPI_Shape> aFace =
 138     GeomAlgoAPI_FaceBuilder::squareFace(theOrigin, theNorm, aSize);
 139   AISObjectPtr aAIS = AISObjectPtr(new GeomAPI_AISObject());
 140   aAIS->createShape(aFace);
 141   aAIS->setWidth(Config_PropManager::integer(SKETCH_TAB_NAME, "planes_thickness"));
 142   aAIS->setColor(theRGB[0], theRGB[1], theRGB[2]);
 143   return aAIS;
 144 }