Salome HOME
FIx for the issue #2399 : optimization of the Intersection feature results (better...
[modules/shaper.git] / src / PartSet / PartSet_PreviewSketchPlane.cpp
1 // Copyright (C) 2014-2017  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
18 // email : webmaster.salome@opencascade.com<mailto:webmaster.salome@opencascade.com>
19 //
20
21 #include "PartSet_PreviewSketchPlane.h"
22 #include "PartSet_Tools.h"
23
24 #include <ModuleBase_IWorkshop.h>
25
26 #include <ModelAPI_ResultBody.h>
27 #include <ModelAPI_ResultConstruction.h>
28 #include <ModelAPI_CompositeFeature.h>
29 #include <ModelAPI_Tools.h>
30
31 #include <GeomAPI_AISObject.h>
32 #include <GeomAPI_Pnt.h>
33
34 #include <XGUI_Tools.h>
35 #include <XGUI_Displayer.h>
36 #include <XGUI_Workshop.h>
37
38 #include <Config_PropManager.h>
39 #include <GeomAlgoAPI_FaceBuilder.h>
40
41 #include <SketchPlugin_Sketch.h>
42 #include <SketchPlugin_SketchEntity.h>
43
44 #include <BRepBndLib.hxx>
45
46 PartSet_PreviewSketchPlane::PartSet_PreviewSketchPlane()
47  : myPreviewIsDisplayed(false), mySizeOfView(0), myIsUseSizeOfView(false)
48 {
49 }
50
51 void PartSet_PreviewSketchPlane::eraseSketchPlane(ModuleBase_IWorkshop* theWorkshop,
52                                                   const bool isClearPlane)
53 {
54   if (myPreviewIsDisplayed) {
55     XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
56     aDisp->eraseAIS(myPlane, false);
57     if (isClearPlane) {
58       myPlane = std::shared_ptr<GeomAPI_AISObject>();
59       myShape = std::shared_ptr<GeomAPI_Shape>();
60     }
61     myPreviewIsDisplayed = false;
62   }
63 }
64
65 void PartSet_PreviewSketchPlane::createSketchPlane(const CompositeFeaturePtr& theSketch,
66                                                    ModuleBase_IWorkshop* theWorkshop)
67 {
68   // the preview plane has been already created and displayed
69   if (myPreviewIsDisplayed)
70     return;
71
72   // plane is visualized only if sketch plane is filled
73   if (!PartSet_Tools::sketchPlane(theSketch).get())
74     return;
75
76   if (!myPlane) { // If planes are not created
77     // Create Preview
78     // selected linear face parameters
79     AttributeSelectionPtr aSelAttr = std::dynamic_pointer_cast<ModelAPI_AttributeSelection>
80       (theSketch->data()->attribute(SketchPlugin_SketchEntity::EXTERNAL_ID()));
81     if (aSelAttr) {
82       myShape = aSelAttr->value();
83       // this case is needed by constructing sketch on a plane, where result shape is equal
84       // to context result, therefore value() returns NULL and we should use shape of context.
85       if (!myShape.get() && aSelAttr->context().get())
86         myShape = aSelAttr->context()->shape();
87     }
88     if (!myShape.get()) {
89       // Create Preview for default planes
90       std::shared_ptr<GeomDataAPI_Point> anOrigin = std::dynamic_pointer_cast<GeomDataAPI_Point>(
91           theSketch->data()->attribute(SketchPlugin_Sketch::ORIGIN_ID()));
92       std::shared_ptr<GeomDataAPI_Dir> aNormal = std::dynamic_pointer_cast<GeomDataAPI_Dir>(
93           theSketch->data()->attribute(SketchPlugin_Sketch::NORM_ID()));
94
95       double aFaceSize = myIsUseSizeOfView ? mySizeOfView
96         : Config_PropManager::integer(SKETCH_TAB_NAME, "planes_size");
97
98       myShape = GeomAlgoAPI_FaceBuilder::squareFace(
99         myViewCentralPoint.get() ? myViewCentralPoint : anOrigin->pnt(), aNormal->dir(), aFaceSize);
100     }
101     myPlane = createPreviewPlane();
102   }
103
104   XGUI_Displayer* aDisp = XGUI_Tools::workshop(theWorkshop)->displayer();
105   aDisp->displayAIS(myPlane, false/*load object in selection*/, 1/*shaded*/, false);
106   myPreviewIsDisplayed = true;
107 }
108
109 double maximumSize(double theXmin, double theYmin, double theZmin,
110                    double theXmax, double theYmax, double theZmax)
111 {
112   double aSize = fabs(theXmax - theXmin);
113   double aSizeToCompare = fabs(theYmax - theYmin);
114   if (aSizeToCompare > aSize)
115     aSize = aSizeToCompare;
116   aSizeToCompare = fabs(theZmax - theZmin);
117   if (aSizeToCompare > aSize)
118     aSize = aSizeToCompare;
119
120   return aSize;
121 }
122
123 bool PartSet_PreviewSketchPlane::getDefaultSizeOfView(
124   const CompositeFeaturePtr& theSketch, double& theSizeOfView,
125   std::shared_ptr<GeomAPI_Pnt>& theCentralPnt)
126 {
127   if (!PartSet_Tools::sketchPlane(theSketch).get())
128     return false;
129
130   AttributeSelectionPtr aSelAttr = std::dynamic_pointer_cast<ModelAPI_AttributeSelection>
131     (theSketch->data()->attribute(SketchPlugin_SketchEntity::EXTERNAL_ID()));
132   if (aSelAttr) {
133     myShape = aSelAttr->value();
134     // this case is needed by constructing sketch on a plane, where result shape is equal
135     // to context result, therefore value() returns NULL and we should use shape of context.
136     if (!myShape.get() && aSelAttr->context().get())
137       myShape = aSelAttr->context()->shape();
138   }
139
140   if (myShape.get())
141     return false;
142
143   Bnd_Box aBox;
144   for (int aSubFeatureId = 0; aSubFeatureId < theSketch->numberOfSubs(); 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 }