]> SALOME platform Git repositories - modules/shaper.git/blob - src/FeaturesAPI/FeaturesAPI_Extrusion.cpp
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   AttributeSelectionListPtr anAttrObjects = aBase->selectionList(FeaturesPlugin_Extrusion::BASE_OBJECTS_ID());
183   AttributeSelectionPtr anAttrDirection = aBase->selection(FeaturesPlugin_Extrusion::DIRECTION_OBJECT_ID());
184
185   theDumper << aBase << " = model.addExtrusion(" << aDocName << ", " << anAttrObjects << ", " << anAttrDirection;
186
187   std::string aCreationMethod = aBase->string(FeaturesPlugin_Extrusion::CREATION_METHOD())->value();
188
189   if(aCreationMethod == FeaturesPlugin_Extrusion::CREATION_METHOD_BY_SIZES()) {
190     AttributeDoublePtr anAttrToSize = aBase->real(FeaturesPlugin_Extrusion::TO_SIZE_ID());
191     AttributeDoublePtr anAttrFromSize = aBase->real(FeaturesPlugin_Extrusion::FROM_SIZE_ID());
192
193     theDumper << ", " << anAttrToSize << ", " << anAttrFromSize;
194   } else if(aCreationMethod == FeaturesPlugin_Extrusion::CREATION_METHOD_BY_PLANES()) {
195     AttributeSelectionPtr anAttrToObject = aBase->selection(FeaturesPlugin_Extrusion::TO_OBJECT_ID());
196     AttributeDoublePtr anAttrToOffset = aBase->real(FeaturesPlugin_Extrusion::TO_OFFSET_ID());
197     AttributeSelectionPtr anAttrFromObject = aBase->selection(FeaturesPlugin_Extrusion::FROM_OBJECT_ID());
198     AttributeDoublePtr anAttrFromOffset = aBase->real(FeaturesPlugin_Extrusion::FROM_OFFSET_ID());
199
200     theDumper << ", " << anAttrToObject << ", " << anAttrToOffset << ", " << anAttrFromObject << ", " << anAttrFromOffset;
201   }
202
203   theDumper << ")" << std::endl;
204
205   AttributeReferencePtr anAttrSketch = aBase->reference(FeaturesPlugin_CompositeSketch::SKETCH_ID());
206   if(anAttrSketch->isInitialized()) {
207     theDumper << aBase << ".setNestedSketch(" << anAttrSketch << ")";
208   }
209 }
210
211 void FeaturesAPI_Extrusion::execIfBaseNotEmpty()
212 {
213   if(mybaseObjects->size() > 0) {
214     execute();
215   }
216 }
217
218 //==================================================================================================
219 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
220                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
221                           const ModelHighAPI_Double& theSize)
222 {
223   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
224   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature, theBaseObjects, theSize));
225 }
226
227 //==================================================================================================
228 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
229                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
230                           const ModelHighAPI_Selection& theDirection,
231                           const ModelHighAPI_Double& theSize)
232 {
233   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
234   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature, theBaseObjects, theDirection, theSize));
235 }
236
237 //==================================================================================================
238 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
239                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
240                           const ModelHighAPI_Double& theToSize,
241                           const ModelHighAPI_Double& theFromSize)
242 {
243   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
244   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature, theBaseObjects, theToSize, theFromSize));
245 }
246
247 //==================================================================================================
248 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
249                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
250                           const ModelHighAPI_Selection& theDirection,
251                           const ModelHighAPI_Double& theToSize,
252                           const ModelHighAPI_Double& theFromSize)
253 {
254   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
255   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature,
256                                                 theBaseObjects,
257                                                 theDirection,
258                                                 theToSize,
259                                                 theFromSize));
260 }
261
262 //==================================================================================================
263 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
264                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
265                           const ModelHighAPI_Selection& theToObject,
266                           const ModelHighAPI_Double& theToOffset,
267                           const ModelHighAPI_Selection& theFromObject,
268                           const ModelHighAPI_Double& theFromOffset)
269 {
270   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
271   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature,
272                                                 theBaseObjects,
273                                                 theToObject,
274                                                 theToOffset,
275                                                 theFromObject,
276                                                 theFromOffset));
277 }
278
279 //==================================================================================================
280 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
281                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
282                           const ModelHighAPI_Selection& theDirection,
283                           const ModelHighAPI_Selection& theToObject,
284                           const ModelHighAPI_Double& theToOffset,
285                           const ModelHighAPI_Selection& theFromObject,
286                           const ModelHighAPI_Double& theFromOffset)
287 {
288   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
289   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature,
290                                                 theBaseObjects,
291                                                 theDirection,
292                                                 theToObject,
293                                                 theToOffset,
294                                                 theFromObject,
295                                                 theFromOffset));
296 }