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, aXZRGB, aXYRGB;
 108     aYZRGB = Config_PropManager::color("Visualization", "yz_plane_color",
 109                                                         YZ_PLANE_COLOR);
 110     aXZRGB = Config_PropManager::color("Visualization", "xz_plane_color",
 111                                                         XZ_PLANE_COLOR);
 112     aXYRGB = Config_PropManager::color("Visualization", "xy_plane_color",
 113                                                         XY_PLANE_COLOR);
 114     int aR[] = {aYZRGB[0], aYZRGB[1], aYZRGB[2]};
 115     int aG[] = {aXZRGB[0], aXZRGB[1], aXZRGB[2]};
 116     int aB[] = {aXYRGB[0], aXYRGB[1], aXYRGB[2]};
 117
 118     myYZPlane = createPreviewPlane(anOrigin, aYZDir, aR);
 119     myXZPlane = createPreviewPlane(anOrigin, aXZDir, aG);
 120     myXYPlane = createPreviewPlane(anOrigin, aXYDir, aB);
 121   }
 122   XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
 123   aDisp->displayAIS(myYZPlane, true, false);
 124   aDisp->displayAIS(myXZPlane, true, false);
 125   aDisp->displayAIS(myXYPlane, true, false);
 126   myPreviewDisplayed = true;
 127 }
 128
 129 AISObjectPtr PartSet_PreviewPlanes::createPreviewPlane(std::shared_ptr<GeomAPI_Pnt> theOrigin,
 130                                                        std::shared_ptr<GeomAPI_Dir> theNorm,
 131                                                        const int theRGB[3])
 132 {
 133   double aSize = Config_PropManager::integer(SKETCH_TAB_NAME, "planes_size", PLANE_SIZE);
 134   std::shared_ptr<GeomAPI_Shape> aFace =
 135     GeomAlgoAPI_FaceBuilder::squareFace(theOrigin, theNorm, aSize);
 136   AISObjectPtr aAIS = AISObjectPtr(new GeomAPI_AISObject());
 137   aAIS->createShape(aFace);
 138   aAIS->setWidth(Config_PropManager::integer(SKETCH_TAB_NAME, "planes_thickness", SKETCH_WIDTH));
 139   aAIS->setColor(theRGB[0], theRGB[1], theRGB[2]);
 140   return aAIS;
 141 }