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 if (aResult.get() != NULL) {
37 aBodyIsVisualized = aResult->groupName() == ModelAPI_ResultBody::group();
38 if (aBodyIsVisualized)
42 return aBodyIsVisualized;
45 bool PartSet_PreviewPlanes::hasVisualizedSketch(ModuleBase_IWorkshop* theWorkshop)
47 bool aSketchIsVisualized = false;
49 XGUI_Workshop* aWorkshop = XGUI_Tools::workshop(theWorkshop);
50 XGUI_Displayer* aDisp = aWorkshop->displayer();
51 QObjectPtrList aDisplayed = aDisp->displayedObjects();
52 foreach (ObjectPtr anObj, aDisplayed) {
53 ResultPtr aResult = std::dynamic_pointer_cast<ModelAPI_Result>(anObj);
54 if (aResult.get() != NULL) {
55 FeaturePtr aFeature = ModelAPI_Feature::feature(aResult);
56 aSketchIsVisualized = aFeature.get() && aFeature->getKind() == SketchPlugin_Sketch::ID();
57 if (aSketchIsVisualized)
61 return aSketchIsVisualized;
64 void PartSet_PreviewPlanes::erasePreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
66 if (myPreviewDisplayed) {
67 XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
68 aDisp->eraseAIS(myYZPlane, false);
69 aDisp->eraseAIS(myXZPlane, false);
70 aDisp->eraseAIS(myXYPlane, false);
71 myPreviewDisplayed = false;
75 void PartSet_PreviewPlanes::showPreviewPlanes(ModuleBase_IWorkshop* theWorkshop)
77 if (myPreviewDisplayed)
80 if (!myYZPlane) { // If planes are not created
82 std::shared_ptr<GeomAPI_Pnt> anOrigin(new GeomAPI_Pnt(0, 0, 0));
83 std::shared_ptr<GeomAPI_Dir> aYZDir(new GeomAPI_Dir(1, 0, 0));
84 // -1, not 1 for correct internal sketch coords (issue 898)
85 std::shared_ptr<GeomAPI_Dir> aXZDir(new GeomAPI_Dir(0, -1, 0));
86 std::shared_ptr<GeomAPI_Dir> aXYDir(new GeomAPI_Dir(0, 0, 1));
88 std::vector<int> aYZRGB, aXZRGB, aXYRGB;
89 aYZRGB = Config_PropManager::color("Visualization", "yz_plane_color",
91 aXZRGB = Config_PropManager::color("Visualization", "xz_plane_color",
93 aXYRGB = Config_PropManager::color("Visualization", "xy_plane_color",
95 int aR[] = {aYZRGB[0], aYZRGB[1], aYZRGB[2]};
96 int aG[] = {aXZRGB[0], aXZRGB[1], aXZRGB[2]};
97 int aB[] = {aXYRGB[0], aXYRGB[1], aXYRGB[2]};
99 myYZPlane = createPreviewPlane(anOrigin, aYZDir, aR);
100 myXZPlane = createPreviewPlane(anOrigin, aXZDir, aG);
101 myXYPlane = createPreviewPlane(anOrigin, aXYDir, aB);
103 XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
104 aDisp->displayAIS(myYZPlane, true, false);
105 aDisp->displayAIS(myXZPlane, true, false);
106 aDisp->displayAIS(myXYPlane, true, false);
107 myPreviewDisplayed = true;
110 AISObjectPtr PartSet_PreviewPlanes::createPreviewPlane(std::shared_ptr<GeomAPI_Pnt> theOrigin,
111 std::shared_ptr<GeomAPI_Dir> theNorm,
114 double aSize = Config_PropManager::integer("Sketch planes", "planes_size", PLANE_SIZE);
115 std::shared_ptr<GeomAPI_Shape> aFace = GeomAlgoAPI_FaceBuilder::square(theOrigin, theNorm, aSize);
116 AISObjectPtr aAIS = AISObjectPtr(new GeomAPI_AISObject());
117 aAIS->createShape(aFace);
118 aAIS->setWidth(Config_PropManager::integer("Sketch planes", "planes_thickness", SKETCH_WIDTH));
119 aAIS->setColor(theRGB[0], theRGB[1], theRGB[2]);