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>
12 #include <ModelAPI_ResultConstruction.h>
14 #include <XGUI_Tools.h>
15 #include <XGUI_Displayer.h>
16 #include <XGUI_Workshop.h>
18 #include <Config_PropManager.h>
19 #include <GeomAlgoAPI_FaceBuilder.h>
21 #include <SketchPlugin_Sketch.h>
23 PartSet_PreviewPlanes::PartSet_PreviewPlanes()
24 : myPreviewDisplayed(false)
28 bool PartSet_PreviewPlanes::hasVisualizedBodies(ModuleBase_IWorkshop* theWorkshop)
30 bool aBodyIsVisualized = false;
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();
41 // vertices, edges should not be taken as a body
42 aBodyIsVisualized = aShape->shapeType() <= GeomAPI_Shape::FACE;
43 if (aBodyIsVisualized)
48 return aBodyIsVisualized;
51 bool PartSet_PreviewPlanes::hasVisualizedSketch(ModuleBase_IWorkshop* theWorkshop)
53 bool aSketchIsVisualized = false;
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>
65 if (aCResult.get() && aCResult->facesNum() > 0) {
66 aSketchIsVisualized = true;
73 return aSketchIsVisualized;
76 void PartSet_PreviewPlanes::erasePreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
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;
87 void PartSet_PreviewPlanes::showPreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
89 if (myPreviewDisplayed)
92 if (!myYZPlane) { // If planes are not created
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));
100 std::vector<int> aYZRGB, aXZRGB, aXYRGB;
101 aYZRGB = Config_PropManager::color("Visualization", "yz_plane_color",
103 aXZRGB = Config_PropManager::color("Visualization", "xz_plane_color",
105 aXYRGB = Config_PropManager::color("Visualization", "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]};
111 myYZPlane = createPreviewPlane(anOrigin, aYZDir, aR);
112 myXZPlane = createPreviewPlane(anOrigin, aXZDir, aG);
113 myXYPlane = createPreviewPlane(anOrigin, aXYDir, aB);
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;
122 AISObjectPtr PartSet_PreviewPlanes::createPreviewPlane(std::shared_ptr<GeomAPI_Pnt> theOrigin,
123 std::shared_ptr<GeomAPI_Dir> theNorm,
126 double aSize = Config_PropManager::integer("Sketch planes", "planes_size", PLANE_SIZE);
127 std::shared_ptr<GeomAPI_Shape> aFace = GeomAlgoAPI_FaceBuilder::square(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]);