Salome HOME
Issue #1648: Dump Python in the High Level Parameterized Geometry API
[modules/shaper.git] / src / FeaturesAPI / FeaturesAPI_Extrusion.cpp
1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
2
3 // File:        FeaturesAPI_Extrusion.cpp
4 // Created:     09 June 2016
5 // Author:      Dmitry Bobylev
6
7 #include "FeaturesAPI_Extrusion.h"
8
9 #include <ModelHighAPI_Double.h>
10 #include <ModelHighAPI_Dumper.h>
11 #include <ModelHighAPI_Reference.h>
12 #include <ModelHighAPI_Tools.h>
13
14 //==================================================================================================
15 FeaturesAPI_Extrusion::FeaturesAPI_Extrusion(const std::shared_ptr<ModelAPI_Feature>& theFeature)
16 : ModelHighAPI_Interface(theFeature)
17 {
18   initialize();
19 }
20
21 //==================================================================================================
22 FeaturesAPI_Extrusion::FeaturesAPI_Extrusion(const std::shared_ptr<ModelAPI_Feature>& theFeature,
23                                              const std::list<ModelHighAPI_Selection>& theBaseObjects,
24                                              const ModelHighAPI_Double& theSize)
25 : ModelHighAPI_Interface(theFeature)
26 {
27   if(initialize()) {
28     fillAttribute(theBaseObjects, mybaseObjects);
29     setSizes(theSize, ModelHighAPI_Double());
30   }
31 }
32
33 //==================================================================================================
34 FeaturesAPI_Extrusion::FeaturesAPI_Extrusion(const std::shared_ptr<ModelAPI_Feature>& theFeature,
35                                              const std::list<ModelHighAPI_Selection>& theBaseObjects,
36                                              const ModelHighAPI_Selection& theDirection,
37                                              const ModelHighAPI_Double& theSize)
38 : ModelHighAPI_Interface(theFeature)
39 {
40   if(initialize()) {
41     fillAttribute(theBaseObjects, mybaseObjects);
42     fillAttribute(theDirection, mydirection);
43     setSizes(theSize, ModelHighAPI_Double());
44   }
45 }
46
47 //==================================================================================================
48 FeaturesAPI_Extrusion::FeaturesAPI_Extrusion(const std::shared_ptr<ModelAPI_Feature>& theFeature,
49                                              const std::list<ModelHighAPI_Selection>& theBaseObjects,
50                                              const ModelHighAPI_Double& theToSize,
51                                              const ModelHighAPI_Double& theFromSize)
52 : ModelHighAPI_Interface(theFeature)
53 {
54   if(initialize()) {
55     fillAttribute(theBaseObjects, mybaseObjects);
56     setSizes(theToSize, theFromSize);
57   }
58 }
59
60 //==================================================================================================
61 FeaturesAPI_Extrusion::FeaturesAPI_Extrusion(const std::shared_ptr<ModelAPI_Feature>& theFeature,
62                                              const std::list<ModelHighAPI_Selection>& theBaseObjects,
63                                              const ModelHighAPI_Selection& theDirection,
64                                              const ModelHighAPI_Double& theToSize,
65                                              const ModelHighAPI_Double& theFromSize)
66 : ModelHighAPI_Interface(theFeature)
67 {
68   if(initialize()) {
69     fillAttribute(theBaseObjects, mybaseObjects);
70     fillAttribute(theDirection, mydirection);
71     setSizes(theToSize, theFromSize);
72   }
73 }
74
75 //==================================================================================================
76 FeaturesAPI_Extrusion::FeaturesAPI_Extrusion(const std::shared_ptr<ModelAPI_Feature>& theFeature,
77                                              const std::list<ModelHighAPI_Selection>& theBaseObjects,
78                                              const ModelHighAPI_Selection& theToObject,
79                                              const ModelHighAPI_Double& theToOffset,
80                                              const ModelHighAPI_Selection& theFromObject,
81                                              const ModelHighAPI_Double& theFromOffset)
82 : ModelHighAPI_Interface(theFeature)
83 {
84   if(initialize()) {
85     fillAttribute(theBaseObjects, mybaseObjects);
86     setPlanesAndOffsets(theToObject, theToOffset, theFromObject, theFromOffset);
87   }
88 }
89
90 //==================================================================================================
91 FeaturesAPI_Extrusion::FeaturesAPI_Extrusion(const std::shared_ptr<ModelAPI_Feature>& theFeature,
92                                              const std::list<ModelHighAPI_Selection>& theBaseObjects,
93                                              const ModelHighAPI_Selection& theDirection,
94                                              const ModelHighAPI_Selection& theToObject,
95                                              const ModelHighAPI_Double& theToOffset,
96                                              const ModelHighAPI_Selection& theFromObject,
97                                              const ModelHighAPI_Double& theFromOffset)
98 : ModelHighAPI_Interface(theFeature)
99 {
100   if(initialize()) {
101     fillAttribute(theBaseObjects, mybaseObjects);
102     fillAttribute(theDirection, mydirection);
103     setPlanesAndOffsets(theToObject, theToOffset, theFromObject, theFromOffset);
104   }
105 }
106
107 //==================================================================================================
108 FeaturesAPI_Extrusion::~FeaturesAPI_Extrusion()
109 {
110 }
111
112 //==================================================================================================
113 void FeaturesAPI_Extrusion::setNestedSketch(const ModelHighAPI_Reference& theSketch)
114 {
115   mysketch->setValue(theSketch.feature());
116   mybaseObjects->clear();
117   mybaseObjects->append(theSketch.feature()->firstResult(), GeomShapePtr());
118
119   execIfBaseNotEmpty();
120 }
121
122 //==================================================================================================
123 void FeaturesAPI_Extrusion::setBase(const std::list<ModelHighAPI_Selection>& theBaseObjects)
124 {
125   mysketch->setValue(ObjectPtr());
126   mybaseObjects->clear();
127   fillAttribute(theBaseObjects, mybaseObjects);
128
129   execIfBaseNotEmpty();
130 }
131
132 //==================================================================================================
133 void FeaturesAPI_Extrusion::setDirection(const ModelHighAPI_Selection& theDirection)
134 {
135   fillAttribute(theDirection, mydirection);
136
137   execIfBaseNotEmpty();
138 }
139
140 //==================================================================================================
141 void FeaturesAPI_Extrusion::setSizes(const ModelHighAPI_Double& theToSize,
142                                      const ModelHighAPI_Double& theFromSize)
143 {
144   fillAttribute(FeaturesPlugin_Extrusion::CREATION_METHOD_BY_SIZES(), mycreationMethod);
145   fillAttribute(theToSize, mytoSize);
146   fillAttribute(theFromSize, myfromSize);
147
148   execIfBaseNotEmpty();
149 }
150
151 //==================================================================================================
152 void FeaturesAPI_Extrusion::setSize(const ModelHighAPI_Double& theSize)
153 {
154   fillAttribute(FeaturesPlugin_Extrusion::CREATION_METHOD_BY_SIZES(), mycreationMethod);
155   fillAttribute(theSize, mytoSize);
156   fillAttribute(ModelHighAPI_Double(), myfromSize);
157
158   execIfBaseNotEmpty();
159 }
160
161 //==================================================================================================
162 void FeaturesAPI_Extrusion::setPlanesAndOffsets(const ModelHighAPI_Selection& theToObject,
163                                                 const ModelHighAPI_Double& theToOffset,
164                                                 const ModelHighAPI_Selection& theFromObject,
165                                                 const ModelHighAPI_Double& theFromOffset)
166 {
167   fillAttribute(FeaturesPlugin_Extrusion::CREATION_METHOD_BY_PLANES(), mycreationMethod);
168   fillAttribute(theToObject, mytoObject);
169   fillAttribute(theToOffset, mytoOffset);
170   fillAttribute(theFromObject, myfromObject);
171   fillAttribute(theFromOffset, myfromOffset);
172
173   execIfBaseNotEmpty();
174 }
175
176 //==================================================================================================
177 void FeaturesAPI_Extrusion::dump(ModelHighAPI_Dumper& theDumper) const
178 {
179   FeaturePtr aBase = feature();
180   const std::string& aDocName = theDumper.name(aBase->document());
181
182   AttributeReferencePtr anAttrSketch = aBase->reference(FeaturesPlugin_Extrusion::SKETCH_ID());
183   AttributeSelectionListPtr anAttrObjects = aBase->selectionList(FeaturesPlugin_Extrusion::BASE_OBJECTS_ID());
184   AttributeSelectionPtr anAttrDirection = aBase->selection(FeaturesPlugin_Extrusion::DIRECTION_OBJECT_ID());
185
186   theDumper << aBase << " = model.addExtrusion(" << aDocName << ", ";
187   anAttrSketch->isInitialized() ? theDumper << "[]" : theDumper << anAttrObjects;
188   theDumper << ", " << anAttrDirection;
189
190   std::string aCreationMethod = aBase->string(FeaturesPlugin_Extrusion::CREATION_METHOD())->value();
191
192   if(aCreationMethod == FeaturesPlugin_Extrusion::CREATION_METHOD_BY_SIZES()) {
193     AttributeDoublePtr anAttrToSize = aBase->real(FeaturesPlugin_Extrusion::TO_SIZE_ID());
194     AttributeDoublePtr anAttrFromSize = aBase->real(FeaturesPlugin_Extrusion::FROM_SIZE_ID());
195
196     theDumper << ", " << anAttrToSize << ", " << anAttrFromSize;
197   } else if(aCreationMethod == FeaturesPlugin_Extrusion::CREATION_METHOD_BY_PLANES()) {
198     AttributeSelectionPtr anAttrToObject = aBase->selection(FeaturesPlugin_Extrusion::TO_OBJECT_ID());
199     AttributeDoublePtr anAttrToOffset = aBase->real(FeaturesPlugin_Extrusion::TO_OFFSET_ID());
200     AttributeSelectionPtr anAttrFromObject = aBase->selection(FeaturesPlugin_Extrusion::FROM_OBJECT_ID());
201     AttributeDoublePtr anAttrFromOffset = aBase->real(FeaturesPlugin_Extrusion::FROM_OFFSET_ID());
202
203     theDumper << ", " << anAttrToObject << ", " << anAttrToOffset << ", " << anAttrFromObject << ", " << anAttrFromOffset;
204   }
205
206   theDumper << ")" << std::endl;
207
208   if(anAttrSketch->isInitialized()) {
209     theDumper << aBase << ".setNestedSketch(" << anAttrSketch << ")" << std::endl;
210   }
211 }
212
213 //==================================================================================================
214 void FeaturesAPI_Extrusion::execIfBaseNotEmpty()
215 {
216   if(mybaseObjects->size() > 0) {
217     execute();
218   }
219 }
220
221 //==================================================================================================
222 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
223                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
224                           const ModelHighAPI_Double& theSize)
225 {
226   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
227   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature, theBaseObjects, theSize));
228 }
229
230 //==================================================================================================
231 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
232                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
233                           const ModelHighAPI_Selection& theDirection,
234                           const ModelHighAPI_Double& theSize)
235 {
236   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
237   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature, theBaseObjects, theDirection, theSize));
238 }
239
240 //==================================================================================================
241 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
242                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
243                           const ModelHighAPI_Double& theToSize,
244                           const ModelHighAPI_Double& theFromSize)
245 {
246   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
247   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature, theBaseObjects, theToSize, theFromSize));
248 }
249
250 //==================================================================================================
251 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
252                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
253                           const ModelHighAPI_Selection& theDirection,
254                           const ModelHighAPI_Double& theToSize,
255                           const ModelHighAPI_Double& theFromSize)
256 {
257   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
258   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature,
259                                                 theBaseObjects,
260                                                 theDirection,
261                                                 theToSize,
262                                                 theFromSize));
263 }
264
265 //==================================================================================================
266 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
267                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
268                           const ModelHighAPI_Selection& theToObject,
269                           const ModelHighAPI_Double& theToOffset,
270                           const ModelHighAPI_Selection& theFromObject,
271                           const ModelHighAPI_Double& theFromOffset)
272 {
273   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
274   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature,
275                                                 theBaseObjects,
276                                                 theToObject,
277                                                 theToOffset,
278                                                 theFromObject,
279                                                 theFromOffset));
280 }
281
282 //==================================================================================================
283 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
284                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
285                           const ModelHighAPI_Selection& theDirection,
286                           const ModelHighAPI_Selection& theToObject,
287                           const ModelHighAPI_Double& theToOffset,
288                           const ModelHighAPI_Selection& theFromObject,
289                           const ModelHighAPI_Double& theFromOffset)
290 {
291   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
292   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature,
293                                                 theBaseObjects,
294                                                 theDirection,
295                                                 theToObject,
296                                                 theToOffset,
297                                                 theFromObject,
298                                                 theFromOffset));
299 }