1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
3 // File: PartSet_PreviewPlanes.cpp
4 // Created: 19 Jun 2015
5 // Author: Natalia ERMOLAEVA
7 #include "PartSet_PreviewPlanes.h"
9 #include <ModuleBase_IWorkshop.h>
11 #include <ModelAPI_ResultBody.h>
13 #include <XGUI_Tools.h>
14 #include <XGUI_Displayer.h>
15 #include <XGUI_Workshop.h>
17 #include <Config_PropManager.h>
18 #include <GeomAlgoAPI_FaceBuilder.h>
20 #include <SketchPlugin_Sketch.h>
22 PartSet_PreviewPlanes::PartSet_PreviewPlanes()
23 : myPreviewDisplayed(false)
27 bool PartSet_PreviewPlanes::hasVisualizedBodies(ModuleBase_IWorkshop* theWorkshop)
29 bool aBodyIsVisualized = false;
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();
40 // vertices, edges should not be taken as a body
41 aBodyIsVisualized = aShape->shapeType() <= GeomAPI_Shape::FACE;
42 if (aBodyIsVisualized)
47 return aBodyIsVisualized;
50 bool PartSet_PreviewPlanes::hasVisualizedSketch(ModuleBase_IWorkshop* theWorkshop)
52 bool aSketchIsVisualized = false;
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)
66 return aSketchIsVisualized;
69 void PartSet_PreviewPlanes::erasePreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
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;
80 void PartSet_PreviewPlanes::showPreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
82 if (myPreviewDisplayed)
85 if (!myYZPlane) { // If planes are not created
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));
93 std::vector<int> aYZRGB, aXZRGB, aXYRGB;
94 aYZRGB = Config_PropManager::color("Visualization", "yz_plane_color",
96 aXZRGB = Config_PropManager::color("Visualization", "xz_plane_color",
98 aXYRGB = Config_PropManager::color("Visualization", "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]};
104 myYZPlane = createPreviewPlane(anOrigin, aYZDir, aR);
105 myXZPlane = createPreviewPlane(anOrigin, aXZDir, aG);
106 myXYPlane = createPreviewPlane(anOrigin, aXYDir, aB);
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;
115 AISObjectPtr PartSet_PreviewPlanes::createPreviewPlane(std::shared_ptr<GeomAPI_Pnt> theOrigin,
116 std::shared_ptr<GeomAPI_Dir> theNorm,
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]);