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 bool PartSet_PreviewPlanes::isPreviewShape(std::shared_ptr<GeomAPI_Shape> theShape)
78 return (myYZPlane->getShape()->isEqual(theShape) ||
79 myXZPlane->getShape()->isEqual(theShape) ||
80 myXYPlane->getShape()->isEqual(theShape));
83 void PartSet_PreviewPlanes::erasePreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
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;
94 void PartSet_PreviewPlanes::showPreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
96 if (myPreviewDisplayed)
99 if (!myYZPlane) { // If planes are not created
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));
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");
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]};
121 myYZPlane = createPreviewPlane(anOrigin, aYZDir, aR);
122 myXZPlane = createPreviewPlane(anOrigin, aXZDir, aG);
123 myXYPlane = createPreviewPlane(anOrigin, aXYDir, aB);
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;
132 AISObjectPtr PartSet_PreviewPlanes::createPreviewPlane(std::shared_ptr<GeomAPI_Pnt> theOrigin,
133 std::shared_ptr<GeomAPI_Dir> theNorm,
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]);