Salome HOME
Issue #2923: Change sketch plane
[modules/shaper.git] / src / PartSet / PartSet_PreviewSketchPlane.cpp
1 // Copyright (C) 2014-2019  CEA/DEN, EDF R&D
2 //
3 // This library is free software; you can redistribute it and/or
4 // modify it under the terms of the GNU Lesser General Public
5 // License as published by the Free Software Foundation; either
6 // version 2.1 of the License, or (at your option) any later version.
7 //
8 // This library is distributed in the hope that it will be useful,
9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
11 // Lesser General Public License for more details.
12 //
13 // You should have received a copy of the GNU Lesser General Public
14 // License along with this library; if not, write to the Free Software
15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
16 //
17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
18 //
19
20 #include "PartSet_PreviewSketchPlane.h"
21 #include "PartSet_Tools.h"
22
23 #include <ModuleBase_IWorkshop.h>
24
25 #include <ModelAPI_ResultBody.h>
26 #include <ModelAPI_ResultConstruction.h>
27 #include <ModelAPI_CompositeFeature.h>
28 #include <ModelAPI_Tools.h>
29
30 #include <GeomAPI_AISObject.h>
31 #include <GeomAPI_Pnt.h>
32
33 #include <XGUI_Tools.h>
34 #include <XGUI_Displayer.h>
35 #include <XGUI_Workshop.h>
36
37 #include <Config_PropManager.h>
38 #include <GeomAlgoAPI_FaceBuilder.h>
39
40 #include <SketchPlugin_Sketch.h>
41 #include <SketchPlugin_SketchEntity.h>
42
43 #include <BRepBndLib.hxx>
44
45 PartSet_PreviewSketchPlane::PartSet_PreviewSketchPlane()
46  : myPreviewIsDisplayed(false), mySizeOfView(0), myIsUseSizeOfView(false)
47 {
48 }
49
50 void PartSet_PreviewSketchPlane::eraseSketchPlane(ModuleBase_IWorkshop* theWorkshop,
51                                                   const bool isClearPlane)
52 {
53   if (myPreviewIsDisplayed) {
54     XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
55     aDisp->eraseAIS(myPlane, false);
56     if (isClearPlane) {
57       myPlane = std::shared_ptr<GeomAPI_AISObject>();
58       myShape = std::shared_ptr<GeomAPI_Shape>();
59     }
60     myPreviewIsDisplayed = false;
61   }
62 }
63
64 void PartSet_PreviewSketchPlane::createSketchPlane(const CompositeFeaturePtr& theSketch,
65                                                    ModuleBase_IWorkshop* theWorkshop)
66 {
67   // plane is visualized only if sketch plane is filled
68   if (!PartSet_Tools::sketchPlane(theSketch).get())
69     return;
70
71   XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
72   if (myPreviewIsDisplayed) {
73     aDisp->eraseAIS(myPlane, false);
74   }
75
76   // Create Preview
77   // selected linear face parameters
78   AttributeSelectionPtr aSelAttr = std::dynamic_pointer_cast<ModelAPI_AttributeSelection>
79     (theSketch->data()->attribute(SketchPlugin_SketchEntity::EXTERNAL_ID()));
80   if (aSelAttr) {
81     myShape = aSelAttr->value();
82     // this case is needed by constructing sketch on a plane, where result shape is equal
83     // to context result, therefore value() returns NULL and we should use shape of context.
84     if (!myShape.get() && aSelAttr->context().get())
85       myShape = aSelAttr->context()->shape();
86   }
87   if (!myShape.get()) {
88     // Create Preview for default planes
89     std::shared_ptr<GeomDataAPI_Point> anOrigin = std::dynamic_pointer_cast<GeomDataAPI_Point>(
90         theSketch->data()->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
91     std::shared_ptr<GeomDataAPI_Dir> aNormal = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
92         theSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
93
94     double aFaceSize = myIsUseSizeOfView ? mySizeOfView
95       : Config_PropManager::real(SKETCH_TAB_NAME, "planes_size");
96     if (aFaceSize <= Precision::Confusion())
97       aFaceSize = 200;   // Set default value
98
99     myShape = GeomAlgoAPI_FaceBuilder::squareFace(
100       myViewCentralPoint.get() ? myViewCentralPoint : anOrigin->pnt(), aNormal->dir(), aFaceSize);
101   }
102   myPlane = createPreviewPlane();
103
104   aDisp->displayAIS(myPlane, false/*load object in selection*/, 1/*shaded*/, false);
105   myPreviewIsDisplayed = true;
106 }
107
108 double maximumSize(double theXmin, double theYmin, double theZmin,
109                    double theXmax, double theYmax, double theZmax)
110 {
111   double aSize = fabs(theXmax - theXmin);
112   double aSizeToCompare = fabs(theYmax - theYmin);
113   if (aSizeToCompare > aSize)
114     aSize = aSizeToCompare;
115   aSizeToCompare = fabs(theZmax - theZmin);
116   if (aSizeToCompare > aSize)
117     aSize = aSizeToCompare;
118
119   return aSize;
120 }
121
122 bool PartSet_PreviewSketchPlane::getDefaultSizeOfView(
123   const CompositeFeaturePtr& theSketch, double& theSizeOfView,
124   std::shared_ptr<GeomAPI_Pnt>& theCentralPnt)
125 {
126   if (!PartSet_Tools::sketchPlane(theSketch).get())
127     return false;
128
129   AttributeSelectionPtr aSelAttr = std::dynamic_pointer_cast<ModelAPI_AttributeSelection>
130     (theSketch->data()->attribute(SketchPlugin_SketchEntity::EXTERNAL_ID()));
131   if (aSelAttr) {
132     myShape = aSelAttr->value();
133     // this case is needed by constructing sketch on a plane, where result shape is equal
134     // to context result, therefore value() returns NULL and we should use shape of context.
135     if (!myShape.get() && aSelAttr->context().get())
136       myShape = aSelAttr->context()->shape();
137   }
138
139   if (myShape.get())
140     return false;
141
142   Bnd_Box aBox;
143   int aNumberOfSubs = theSketch->numberOfSubs();
144   for (int aSubFeatureId = 0; aSubFeatureId < aNumberOfSubs; aSubFeatureId++) {
145     FeaturePtr aFeature = theSketch->subFeature(aSubFeatureId);
146     if (!aFeature.get())
147       continue;
148
149     std::list<ResultPtr> aResults = aFeature->results();
150     std::list<ResultPtr>::const_iterator aResultIt;
151     for (aResultIt = aResults.begin(); aResultIt != aResults.end(); ++aResultIt) {
152       ResultPtr aResult = *aResultIt;
153       std::shared_ptr<GeomAPI_Shape> aShapePtr = aResult->shape();
154       if (aShapePtr.get()) {
155         TopoDS_Shape aShape = aShapePtr->impl<TopoDS_Shape>();
156         if (aShape.IsNull())
157           continue;
158         BRepBndLib::Add(aShape, aBox);
159       }
160     }
161   }
162   if (aBox.IsVoid())
163     return 0;
164
165   double aXmin, aXmax, anYmin, anYmax, aZmin, aZmax;
166   aBox.Get(aXmin, anYmin, aZmin, aXmax, anYmax, aZmax);
167
168   theSizeOfView = maximumSize(aXmin, anYmin, aZmin, aXmax, anYmax, aZmax);
169   if (theSizeOfView > 0) {
170     gp_Pnt aCentre(aXmax-fabs(aXmax-aXmin)/2., anYmax-fabs(anYmax-anYmin)/2.,
171                    aZmax - fabs(aZmax-aZmin)/2.);
172     theCentralPnt = std::shared_ptr<GeomAPI_Pnt>(new GeomAPI_Pnt(aCentre.X(), aCentre.Y(),
173                                                                  aCentre.Z()));
174   }
175   return true;
176 }
177
178 void PartSet_PreviewSketchPlane::setSizeOfView(double theSizeOfView, bool isUseSizeOfView,
179   const std::shared_ptr<GeomAPI_Pnt>& theCentralPoint)
180 {
181   mySizeOfView = theSizeOfView;
182   myIsUseSizeOfView = isUseSizeOfView;
183
184   myViewCentralPoint = theCentralPoint;
185 }
186
187 AISObjectPtr PartSet_PreviewSketchPlane::createPreviewPlane()
188 {
189   if (myPlane.get()) {
190     myPlane->createShape(myShape);
191     return myPlane;
192   }
193   else {
194     AISObjectPtr aAIS = AISObjectPtr(new GeomAPI_AISObject());
195     aAIS->createShape(myShape);
196     std::vector<int> aColor = Config_PropManager::color("Visualization", "sketch_preview_plane");
197     if (aColor.size() == 3)
198       aAIS->setColor(aColor[0], aColor[1], aColor[2]);
199     aAIS->setTransparensy(0.8);
200
201     int aDispMode = 1; // shading
202     Handle(AIS_InteractiveObject) anAISIO = aAIS->impl<Handle(AIS_InteractiveObject)>();
203     if (!anAISIO.IsNull()) {
204       //anAISIO->SetInfiniteState(Standard_True);
205       anAISIO->Attributes()->SetFaceBoundaryDraw( Standard_True );
206       anAISIO->SetDisplayMode(aDispMode);
207     }
208     return aAIS;
209   }
210 }