]> SALOME platform Git repositories - modules/shaper.git/blob - src/FeaturesPlugin/FeaturesPlugin_MultiTranslation.cpp
Salome HOME
Issue #2052: Modification of parameters don't work (sketch, extrusion)
[modules/shaper.git] / src / FeaturesPlugin / FeaturesPlugin_MultiTranslation.cpp
1 // Copyright (C) 2014-201x CEA/DEN, EDF R&D
2
3 // File:        FeaturesPlugin_MultiTranslation.cpp
4 // Created:     30 Jan 2017
5 // Author:      Clarisse Genrault (CEA)
6
7 #include <FeaturesPlugin_MultiTranslation.h>
8
9 #include <GeomAlgoAPI_CompoundBuilder.h>
10
11 #include <GeomAPI_Ax1.h>
12 #include <GeomAPI_Edge.h>
13 #include <GeomAPI_Lin.h>
14 #include <GeomAPI_Trsf.h>
15
16 #include <ModelAPI_AttributeDouble.h>
17 #include <ModelAPI_AttributeInteger.h>
18 #include <ModelAPI_AttributeSelectionList.h>
19 #include <ModelAPI_AttributeString.h>
20 #include <ModelAPI_ResultBody.h>
21 #include <ModelAPI_ResultPart.h>
22
23 #include <math.h>
24
25 //=================================================================================================
26 FeaturesPlugin_MultiTranslation::FeaturesPlugin_MultiTranslation()
27 {
28 }
29
30 //=================================================================================================
31 void FeaturesPlugin_MultiTranslation::initAttributes()
32 {
33   AttributeSelectionListPtr aSelection =
34     std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(data()->addAttribute(
35     FeaturesPlugin_MultiTranslation::OBJECTS_LIST_ID(),
36     ModelAPI_AttributeSelectionList::typeId()));
37
38   data()->addAttribute(FeaturesPlugin_MultiTranslation::AXIS_FIRST_DIR_ID(),
39                        ModelAPI_AttributeSelection::typeId());
40   data()->addAttribute(FeaturesPlugin_MultiTranslation::STEP_FIRST_DIR_ID(),
41                        ModelAPI_AttributeDouble::typeId());
42   data()->addAttribute(FeaturesPlugin_MultiTranslation::NB_COPIES_FIRST_DIR_ID(),
43                        ModelAPI_AttributeInteger::typeId());
44
45   data()->addAttribute(FeaturesPlugin_MultiTranslation::USE_SECOND_DIR_ID(),
46                        ModelAPI_AttributeString::typeId());
47   data()->addAttribute(FeaturesPlugin_MultiTranslation::AXIS_SECOND_DIR_ID(),
48                        ModelAPI_AttributeSelection::typeId());
49   data()->addAttribute(FeaturesPlugin_MultiTranslation::STEP_SECOND_DIR_ID(),
50                        ModelAPI_AttributeDouble::typeId());
51   data()->addAttribute(FeaturesPlugin_MultiTranslation::NB_COPIES_SECOND_DIR_ID(),
52                        ModelAPI_AttributeInteger::typeId());
53 }
54
55 //=================================================================================================
56 void FeaturesPlugin_MultiTranslation::execute()
57 {
58   std::string useSecondDir = string(FeaturesPlugin_MultiTranslation::USE_SECOND_DIR_ID())->value();
59   if(!useSecondDir.empty()) {
60     performTwoDirection();
61   } else {
62     performOneDirection();
63   }
64 }
65
66 //=================================================================================================
67 void FeaturesPlugin_MultiTranslation::performOneDirection()
68 {
69   // Getting objects.
70   ListOfShape anObjects;
71   std::list<ResultPtr> aContextes;
72   AttributeSelectionListPtr anObjectsSelList =
73     selectionList(FeaturesPlugin_MultiTranslation::OBJECTS_LIST_ID());
74   if (anObjectsSelList->size() == 0) {
75     return;
76   }
77   for(int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); anObjectsIndex++) {
78     std::shared_ptr<ModelAPI_AttributeSelection> anObjectAttr =
79       anObjectsSelList->value(anObjectsIndex);
80     std::shared_ptr<GeomAPI_Shape> anObject = anObjectAttr->value();
81     if(!anObject.get()) { // may be for not-activated parts
82       eraseResults();
83       return;
84     }
85     anObjects.push_back(anObject);
86     aContextes.push_back(anObjectAttr->context());
87   }
88
89   //Getting axis.
90   std::shared_ptr<GeomAPI_Ax1> anAxis;
91   std::shared_ptr<GeomAPI_Edge> anEdge;
92   std::shared_ptr<ModelAPI_AttributeSelection> anObjRef =
93     selection(FeaturesPlugin_MultiTranslation::AXIS_FIRST_DIR_ID());
94   if(anObjRef && anObjRef->value() && anObjRef->value()->isEdge()) {
95     anEdge = std::shared_ptr<GeomAPI_Edge>(new GeomAPI_Edge(anObjRef->value()));
96   } else if (anObjRef && !anObjRef->value() && anObjRef->context() &&
97              anObjRef->context()->shape() && anObjRef->context()->shape()->isEdge()) {
98     anEdge = std::shared_ptr<GeomAPI_Edge>(new GeomAPI_Edge(anObjRef->context()->shape()));
99   }
100   if(anEdge) {
101     anAxis = std::shared_ptr<GeomAPI_Ax1>(new GeomAPI_Ax1(anEdge->line()->location(),
102                                                           anEdge->line()->direction()));
103   }
104
105   // Getting step.
106   double aStep = real(FeaturesPlugin_MultiTranslation::STEP_FIRST_DIR_ID())->value();
107
108   // Getting number of copies.
109   int nbCopies =
110     integer(FeaturesPlugin_MultiTranslation::NB_COPIES_FIRST_DIR_ID())->value();
111
112   if (nbCopies <=0) {
113     std::string aFeatureError = "Multitranslation builder ";
114     aFeatureError+=":: the number of copies for the first direction is null or negative.";
115     setError(aFeatureError);
116     return;
117   }
118
119   // Moving each object.
120   int aResultIndex = 0;
121   std::list<ResultPtr>::iterator aContext = aContextes.begin();
122   for(ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end();
123         anObjectsIt++, aContext++) {
124     std::shared_ptr<GeomAPI_Shape> aBaseShape = *anObjectsIt;
125     bool isPart = (*aContext)->groupName() == ModelAPI_ResultPart::group();
126
127     // Setting result.
128     if (isPart) {
129       ResultPartPtr anOrigin = std::dynamic_pointer_cast<ModelAPI_ResultPart>(*aContext);
130       std::shared_ptr<GeomAPI_Trsf> aTrsf(new GeomAPI_Trsf());
131       for (int i=0; i<nbCopies; i++) {
132         aTrsf->setTranslation(anAxis, i*aStep);
133         ResultPartPtr aResultPart = document()->copyPart(anOrigin, data(), aResultIndex);
134         aResultPart->setTrsf(*aContext, aTrsf);
135         setResult(aResultPart, aResultIndex);
136         aResultIndex++;
137       }
138     } else {
139       ListOfShape aListOfShape;
140       std::list<std::shared_ptr<GeomAlgoAPI_Translation> > aListOfTranslationAlgo;
141
142       for (int i=0; i<nbCopies; i++) {
143         std::shared_ptr<GeomAlgoAPI_Translation> aTranslationAlgo(
144           new GeomAlgoAPI_Translation(aBaseShape, anAxis, i*aStep));
145
146         if (!aTranslationAlgo->check()) {
147           setError(aTranslationAlgo->getError());
148           break;
149         }
150
151         aTranslationAlgo->build();
152
153         // Checking that the algorithm worked properly.
154         if (!aTranslationAlgo->isDone()) {
155           static const std::string aFeatureError = "Error : Multitranslation algorithm failed.";
156           setError(aFeatureError);
157           break;
158         }
159         if (aTranslationAlgo->shape()->isNull()) {
160           static const std::string aShapeError = "Error : Resulting shape is null.";
161           setError(aShapeError);
162           break;
163         }
164         if (!aTranslationAlgo->isValid()) {
165           static const std::string aFeatureError = "Error : Resulting shape in not valid.";
166           setError(aFeatureError);
167           break;
168         }
169         aListOfShape.push_back(aTranslationAlgo->shape());
170         aListOfTranslationAlgo.push_back(aTranslationAlgo);
171       }
172       std::shared_ptr<GeomAPI_Shape> aCompound =
173         GeomAlgoAPI_CompoundBuilder::compound(aListOfShape);
174       ResultBodyPtr aResultBody = document()->createBody(data(), aResultIndex);
175       aResultBody->storeModified(aBaseShape, aCompound);
176       loadNamingDS(aListOfTranslationAlgo, aResultBody, aBaseShape);
177
178       setResult(aResultBody, aResultIndex);
179     }
180     aResultIndex++;
181   }
182
183   // Remove the rest results if there were produced in the previous pass.
184   removeResults(aResultIndex);
185 }
186
187 //=================================================================================================
188 void FeaturesPlugin_MultiTranslation::performTwoDirection()
189 {
190   // Getting objects.
191   ListOfShape anObjects;
192   std::list<ResultPtr> aContextes;
193   AttributeSelectionListPtr anObjectsSelList =
194     selectionList(FeaturesPlugin_MultiTranslation::OBJECTS_LIST_ID());
195   if (anObjectsSelList->size() == 0) {
196     return;
197   }
198   for(int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); anObjectsIndex++) {
199     std::shared_ptr<ModelAPI_AttributeSelection> anObjectAttr =
200       anObjectsSelList->value(anObjectsIndex);
201     std::shared_ptr<GeomAPI_Shape> anObject = anObjectAttr->value();
202     if(!anObject.get()) { // may be for not-activated parts
203       eraseResults();
204       return;
205     }
206     anObjects.push_back(anObject);
207     aContextes.push_back(anObjectAttr->context());
208   }
209
210   //Getting axis.
211   std::shared_ptr<GeomAPI_Ax1> aFirstAxis;
212   std::shared_ptr<GeomAPI_Edge> anEdge;
213   std::shared_ptr<ModelAPI_AttributeSelection> anObjRef =
214     selection(FeaturesPlugin_MultiTranslation::AXIS_FIRST_DIR_ID());
215   if(anObjRef && anObjRef->value() && anObjRef->value()->isEdge()) {
216     anEdge = std::shared_ptr<GeomAPI_Edge>(new GeomAPI_Edge(anObjRef->value()));
217   } else if (anObjRef && !anObjRef->value() && anObjRef->context() &&
218              anObjRef->context()->shape() && anObjRef->context()->shape()->isEdge()) {
219     anEdge = std::shared_ptr<GeomAPI_Edge>(new GeomAPI_Edge(anObjRef->context()->shape()));
220   }
221   if(anEdge) {
222     aFirstAxis = std::shared_ptr<GeomAPI_Ax1>(new GeomAPI_Ax1(anEdge->line()->location(),
223                                                               anEdge->line()->direction()));
224   }
225   std::shared_ptr<GeomAPI_Ax1> aSecondAxis;
226   anObjRef = selection(FeaturesPlugin_MultiTranslation::AXIS_SECOND_DIR_ID());
227   if(anObjRef && anObjRef->value() && anObjRef->value()->isEdge()) {
228     anEdge = std::shared_ptr<GeomAPI_Edge>(new GeomAPI_Edge(anObjRef->value()));
229   } else if (anObjRef && !anObjRef->value() && anObjRef->context() &&
230              anObjRef->context()->shape() && anObjRef->context()->shape()->isEdge()) {
231     anEdge = std::shared_ptr<GeomAPI_Edge>(new GeomAPI_Edge(anObjRef->context()->shape()));
232   }
233   if(anEdge) {
234     aSecondAxis = std::shared_ptr<GeomAPI_Ax1>(new GeomAPI_Ax1(anEdge->line()->location(),
235                                                                anEdge->line()->direction()));
236   }
237
238   // Getting step.
239   double aFirstStep = real(FeaturesPlugin_MultiTranslation::STEP_FIRST_DIR_ID())->value();
240   double aSecondStep = real(FeaturesPlugin_MultiTranslation::STEP_SECOND_DIR_ID())->value();
241
242   // Getting number of copies.
243   int aFirstNbCopies =
244     integer(FeaturesPlugin_MultiTranslation::NB_COPIES_FIRST_DIR_ID())->value();
245   int aSecondNbCopies =
246     integer(FeaturesPlugin_MultiTranslation::NB_COPIES_SECOND_DIR_ID())->value();
247
248   if (aFirstNbCopies <=0) {
249     std::string aFeatureError = "Multitranslation builder ";
250     aFeatureError+=":: the number of copies for the first direction is null or negative.";
251     setError(aFeatureError);
252     return;
253   }
254
255   if (aSecondNbCopies <=0) {
256     std::string aFeatureError = "Multitranslation builder ";
257     aFeatureError+=":: the number of copies for the second direction is null or negative.";
258     setError(aFeatureError);
259     return;
260   }
261
262   // Coord aFirstAxis
263   double x1 = aFirstAxis->dir()->x();
264   double y1 = aFirstAxis->dir()->y();
265   double z1 = aFirstAxis->dir()->z();
266   double norm1 = sqrt(x1*x1 + y1*y1 + z1*z1);
267
268   // Coord aSecondAxis
269   double x2 = aSecondAxis->dir()->x();
270   double y2 = aSecondAxis->dir()->y();
271   double z2 = aSecondAxis->dir()->z();
272   double norm2 = sqrt(x2*x2 + y2*y2 + z2*z2);
273
274   // Moving each object.
275   int aResultIndex = 0;
276   std::list<ResultPtr>::iterator aContext = aContextes.begin();
277   for(ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end();
278         anObjectsIt++, aContext++) {
279     std::shared_ptr<GeomAPI_Shape> aBaseShape = *anObjectsIt;
280     bool isPart = (*aContext)->groupName() == ModelAPI_ResultPart::group();
281
282     // Setting result.
283     if (isPart) {
284       ResultPartPtr anOrigin = std::dynamic_pointer_cast<ModelAPI_ResultPart>(*aContext);
285       std::shared_ptr<GeomAPI_Trsf> aTrsf(new GeomAPI_Trsf());
286       for (int j=0; j<aSecondNbCopies; j++) {
287         for (int i=0; i<aFirstNbCopies; i++) {
288           double dx = i*aFirstStep*x1/norm1+j*aSecondStep*x2/norm2;
289           double dy = i*aFirstStep*y1/norm1+j*aSecondStep*y2/norm2;
290           double dz = i*aFirstStep*z1/norm1+j*aSecondStep*z2/norm2;
291           aTrsf->setTranslation(dx, dy, dz);
292           ResultPartPtr aResultPart = document()->copyPart(anOrigin, data(), aResultIndex);
293           aResultPart->setTrsf(*aContext, aTrsf);
294           setResult(aResultPart, aResultIndex);
295           aResultIndex++;
296         }
297       }
298     } else {
299       ListOfShape aListOfShape;
300       std::list<std::shared_ptr<GeomAlgoAPI_Translation> > aListOfTranslationAlgo;
301
302       for (int j=0; j<aSecondNbCopies; j++) {
303         for (int i=0; i<aFirstNbCopies; i++) {
304           double dx = i*aFirstStep*x1/norm1+j*aSecondStep*x2/norm2;
305           double dy = i*aFirstStep*y1/norm1+j*aSecondStep*y2/norm2;
306           double dz = i*aFirstStep*z1/norm1+j*aSecondStep*z2/norm2;
307           std::shared_ptr<GeomAlgoAPI_Translation> aTranslationAlgo(
308             new GeomAlgoAPI_Translation(aBaseShape, dx, dy, dz));
309
310           if (!aTranslationAlgo->check()) {
311             setError(aTranslationAlgo->getError());
312             break;
313           }
314
315           aTranslationAlgo->build();
316
317           // Checking that the algorithm worked properly.
318           if (!aTranslationAlgo->isDone()) {
319             static const std::string aFeatureError = "Error : Multitranslation algorithm failed.";
320             setError(aFeatureError);
321             break;
322           }
323           if (aTranslationAlgo->shape()->isNull()) {
324             static const std::string aShapeError = "Error : Resulting shape is null.";
325             setError(aShapeError);
326             break;
327           }
328           if (!aTranslationAlgo->isValid()) {
329             static const std::string aFeatureError = "Error : Resulting shape in not valid.";
330             setError(aFeatureError);
331            break;
332           }
333           aListOfShape.push_back(aTranslationAlgo->shape());
334           aListOfTranslationAlgo.push_back(aTranslationAlgo);
335         }
336       }
337       std::shared_ptr<GeomAPI_Shape> aCompound =
338         GeomAlgoAPI_CompoundBuilder::compound(aListOfShape);
339       ResultBodyPtr aResultBody = document()->createBody(data(), aResultIndex);
340       aResultBody->storeModified(aBaseShape, aCompound);
341       loadNamingDS(aListOfTranslationAlgo, aResultBody, aBaseShape);
342       setResult(aResultBody, aResultIndex);
343     }
344     aResultIndex++;
345   }
346
347   // Remove the rest results if there were produced in the previous pass.
348   removeResults(aResultIndex);
349 }
350
351 //=================================================================================================
352 void FeaturesPlugin_MultiTranslation::loadNamingDS(
353     std::list<std::shared_ptr<GeomAlgoAPI_Translation> > theListOfTranslationAlgo,
354     std::shared_ptr<ModelAPI_ResultBody> theResultBody,
355     std::shared_ptr<GeomAPI_Shape> theBaseShape)
356 {
357   int aTag = 1;
358   int anIndex = 1;
359   std::string aTranslatedName;
360
361   for (std::list<std::shared_ptr<GeomAlgoAPI_Translation> >::const_iterator anIt =
362     theListOfTranslationAlgo.begin(); anIt != theListOfTranslationAlgo.cend(); ++anIt) {
363     std::shared_ptr<GeomAPI_DataMapOfShapeShape> aSubShapes = (*anIt)->mapOfSubShapes();
364
365     // naming of faces
366     aTranslatedName = "Translated_Face_" + std::to_string((long long) anIndex);
367     theResultBody->loadAndOrientModifiedShapes((*anIt).get(), theBaseShape, GeomAPI_Shape::FACE,
368                                                aTag++, aTranslatedName, *aSubShapes.get(),
369                                                false, true);
370
371     // naming of edges
372     aTranslatedName = "Translated_Edge_" + std::to_string((long long) anIndex);
373     theResultBody->loadAndOrientModifiedShapes((*anIt).get(), theBaseShape, GeomAPI_Shape::EDGE,
374                                                aTag++, aTranslatedName, *aSubShapes.get(),
375                                                false, true);
376
377     // naming of vertex
378     aTranslatedName = "Translated_Vertex_" + std::to_string((long long) anIndex);
379     theResultBody->loadAndOrientModifiedShapes((*anIt).get(), theBaseShape, GeomAPI_Shape::VERTEX,
380                                                aTag++, aTranslatedName, *aSubShapes.get(),
381                                                false, true);
382
383     ++anIndex;
384   }
385 }