Salome HOME
#1857 In the Sketcher, ability to zoom the view from a given size
[modules/shaper.git] / 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 }