Salome HOME
525bddd1c6dd5b891b183eb76153a30355727cff
[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_CompositeSketch::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 << ")";
210   }
211 }
212
213 void FeaturesAPI_Extrusion::execIfBaseNotEmpty()
214 {
215   if(mybaseObjects->size() > 0) {
216     execute();
217   }
218 }
219
220 //==================================================================================================
221 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
222                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
223                           const ModelHighAPI_Double& theSize)
224 {
225   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
226   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature, theBaseObjects, theSize));
227 }
228
229 //==================================================================================================
230 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
231                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
232                           const ModelHighAPI_Selection& theDirection,
233                           const ModelHighAPI_Double& theSize)
234 {
235   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
236   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature, theBaseObjects, theDirection, theSize));
237 }
238
239 //==================================================================================================
240 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
241                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
242                           const ModelHighAPI_Double& theToSize,
243                           const ModelHighAPI_Double& theFromSize)
244 {
245   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
246   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature, theBaseObjects, theToSize, theFromSize));
247 }
248
249 //==================================================================================================
250 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
251                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
252                           const ModelHighAPI_Selection& theDirection,
253                           const ModelHighAPI_Double& theToSize,
254                           const ModelHighAPI_Double& theFromSize)
255 {
256   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
257   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature,
258                                                 theBaseObjects,
259                                                 theDirection,
260                                                 theToSize,
261                                                 theFromSize));
262 }
263
264 //==================================================================================================
265 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
266                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
267                           const ModelHighAPI_Selection& theToObject,
268                           const ModelHighAPI_Double& theToOffset,
269                           const ModelHighAPI_Selection& theFromObject,
270                           const ModelHighAPI_Double& theFromOffset)
271 {
272   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
273   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature,
274                                                 theBaseObjects,
275                                                 theToObject,
276                                                 theToOffset,
277                                                 theFromObject,
278                                                 theFromOffset));
279 }
280
281 //==================================================================================================
282 ExtrusionPtr addExtrusion(const std::shared_ptr<ModelAPI_Document>& thePart,
283                           const std::list<ModelHighAPI_Selection>& theBaseObjects,
284                           const ModelHighAPI_Selection& theDirection,
285                           const ModelHighAPI_Selection& theToObject,
286                           const ModelHighAPI_Double& theToOffset,
287                           const ModelHighAPI_Selection& theFromObject,
288                           const ModelHighAPI_Double& theFromOffset)
289 {
290   std::shared_ptr<ModelAPI_Feature> aFeature = thePart->addFeature(FeaturesAPI_Extrusion::ID());
291   return ExtrusionPtr(new FeaturesAPI_Extrusion(aFeature,
292                                                 theBaseObjects,
293                                                 theDirection,
294                                                 theToObject,
295                                                 theToOffset,
296                                                 theFromObject,
297                                                 theFromOffset));
298 }