]> SALOME platform Git repositories - modules/shaper.git/blob - src/FeaturesPlugin/FeaturesPlugin_Rotation.cpp
Salome HOME
Reverted some changes in salome build/run scripts.
[modules/shaper.git] / src / FeaturesPlugin / FeaturesPlugin_Rotation.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 <FeaturesPlugin_Rotation.h>
22
23 #include <ModelAPI_AttributeDouble.h>
24 #include <ModelAPI_AttributeSelectionList.h>
25 #include <ModelAPI_AttributeString.h>
26 #include <ModelAPI_ResultBody.h>
27 #include <ModelAPI_ResultPart.h>
28
29 #include <GeomAlgoAPI_PointBuilder.h>
30
31 #include <GeomAPI_Edge.h>
32 #include <GeomAPI_Lin.h>
33 #include <GeomAPI_Trsf.h>
34
35 #include <FeaturesPlugin_Tools.h>
36
37 //=================================================================================================
38 FeaturesPlugin_Rotation::FeaturesPlugin_Rotation()
39 {
40 }
41
42 //=================================================================================================
43 void FeaturesPlugin_Rotation::initAttributes()
44 {
45   data()->addAttribute(FeaturesPlugin_Rotation::CREATION_METHOD(),
46                        ModelAPI_AttributeString::typeId());
47
48   AttributeSelectionListPtr aSelection =
49     std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(data()->addAttribute(
50     FeaturesPlugin_Rotation::OBJECTS_LIST_ID(), ModelAPI_AttributeSelectionList::typeId()));
51
52   data()->addAttribute(FeaturesPlugin_Rotation::AXIS_OBJECT_ID(),
53                        ModelAPI_AttributeSelection::typeId());
54   data()->addAttribute(FeaturesPlugin_Rotation::ANGLE_ID(), ModelAPI_AttributeDouble::typeId());
55
56   data()->addAttribute(FeaturesPlugin_Rotation::CENTER_POINT_ID(),
57                        ModelAPI_AttributeSelection::typeId());
58   data()->addAttribute(FeaturesPlugin_Rotation::START_POINT_ID(),
59                        ModelAPI_AttributeSelection::typeId());
60   data()->addAttribute(FeaturesPlugin_Rotation::END_POINT_ID(),
61                        ModelAPI_AttributeSelection::typeId());
62 }
63
64 //=================================================================================================
65 void FeaturesPlugin_Rotation::execute()
66 {
67   AttributeStringPtr aMethodTypeAttr = string(FeaturesPlugin_Rotation::CREATION_METHOD());
68   std::string aMethodType = aMethodTypeAttr->value();
69
70   if (aMethodType == CREATION_METHOD_BY_ANGLE()) {
71     performTranslationByAxisAndAngle();
72   }
73
74   if (aMethodType == CREATION_METHOD_BY_THREE_POINTS()) {
75     performTranslationByThreePoints();
76   }
77 }
78
79 //=================================================================================================
80 void FeaturesPlugin_Rotation::performTranslationByAxisAndAngle()
81 {
82   // Getting objects.
83   ListOfShape anObjects;
84   std::list<ResultPtr> aContextes;
85   AttributeSelectionListPtr anObjectsSelList =
86     selectionList(FeaturesPlugin_Rotation::OBJECTS_LIST_ID());
87   if (anObjectsSelList->size() == 0) {
88     return;
89   }
90   for(int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); anObjectsIndex++) {
91     std::shared_ptr<ModelAPI_AttributeSelection> anObjectAttr =
92       anObjectsSelList->value(anObjectsIndex);
93     std::shared_ptr<GeomAPI_Shape> anObject = anObjectAttr->value();
94     if(!anObject.get()) {
95       return;
96     }
97     anObjects.push_back(anObject);
98     aContextes.push_back(anObjectAttr->context());
99   }
100
101   //Getting axis.
102   std::shared_ptr<GeomAPI_Ax1> anAxis;
103   std::shared_ptr<GeomAPI_Edge> anEdge;
104   std::shared_ptr<ModelAPI_AttributeSelection> anObjRef =
105     selection(FeaturesPlugin_Rotation::AXIS_OBJECT_ID());
106   if(anObjRef && anObjRef->value() && anObjRef->value()->isEdge()) {
107     anEdge = std::shared_ptr<GeomAPI_Edge>(new GeomAPI_Edge(anObjRef->value()));
108   } else if (anObjRef && !anObjRef->value() && anObjRef->context() &&
109              anObjRef->context()->shape() && anObjRef->context()->shape()->isEdge()) {
110     anEdge = std::shared_ptr<GeomAPI_Edge>(new GeomAPI_Edge(anObjRef->context()->shape()));
111   }
112   if(anEdge) {
113     anAxis = std::shared_ptr<GeomAPI_Ax1>(new GeomAPI_Ax1(anEdge->line()->location(),
114                                                           anEdge->line()->direction()));
115   }
116
117   // Getting angle.
118   double anAngle = real(FeaturesPlugin_Rotation::ANGLE_ID())->value();
119
120   // Rotating each object.
121   int aResultIndex = 0;
122   std::list<ResultPtr>::iterator aContext = aContextes.begin();
123   for(ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end();
124         anObjectsIt++, aContext++) {
125     std::shared_ptr<GeomAPI_Shape> aBaseShape = *anObjectsIt;
126     bool isPart = aContext->get() && (*aContext)->groupName() == ModelAPI_ResultPart::group();
127
128     // Setting result.
129     if (isPart) {
130       std::shared_ptr<GeomAPI_Trsf> aTrsf(new GeomAPI_Trsf());
131       aTrsf->setRotation(anAxis, anAngle);
132       ResultPartPtr anOrigin = std::dynamic_pointer_cast<ModelAPI_ResultPart>(*aContext);
133       ResultPartPtr aResultPart = document()->copyPart(anOrigin, data(), aResultIndex);
134       aResultPart->setTrsf(*aContext, aTrsf);
135       setResult(aResultPart, aResultIndex);
136     } else {
137       GeomAlgoAPI_Rotation aRotationAlgo(aBaseShape, anAxis, anAngle);
138
139       if (!aRotationAlgo.check()) {
140         setError(aRotationAlgo.getError());
141         return;
142       }
143
144       aRotationAlgo.build();
145
146       // Checking that the algorithm worked properly.
147       if(!aRotationAlgo.isDone()) {
148         static const std::string aFeatureError = "Error: Rotation algorithm failed.";
149         setError(aFeatureError);
150         break;
151       }
152       if(aRotationAlgo.shape()->isNull()) {
153         static const std::string aShapeError = "Error: Resulting shape is Null.";
154         setError(aShapeError);
155         break;
156       }
157       if(!aRotationAlgo.isValid()) {
158         std::string aFeatureError = "Error: Resulting shape is not valid.";
159         setError(aFeatureError);
160         break;
161       }
162
163       ResultBodyPtr aResultBody = document()->createBody(data(), aResultIndex);
164       loadNamingDS(aRotationAlgo, aResultBody, aBaseShape);
165       setResult(aResultBody, aResultIndex);
166     }
167     aResultIndex++;
168   }
169
170   // Remove the rest results if there were produced in the previous pass.
171   removeResults(aResultIndex);
172 }
173
174 //=================================================================================================
175 void FeaturesPlugin_Rotation::performTranslationByThreePoints()
176 {
177   // Getting objects.
178   ListOfShape anObjects;
179   std::list<ResultPtr> aContextes;
180   AttributeSelectionListPtr anObjectsSelList =
181     selectionList(FeaturesPlugin_Rotation::OBJECTS_LIST_ID());
182   if (anObjectsSelList->size() == 0) {
183     return;
184   }
185   for(int anObjectsIndex = 0; anObjectsIndex < anObjectsSelList->size(); anObjectsIndex++) {
186     std::shared_ptr<ModelAPI_AttributeSelection> anObjectAttr =
187       anObjectsSelList->value(anObjectsIndex);
188     std::shared_ptr<GeomAPI_Shape> anObject = anObjectAttr->value();
189     if(!anObject.get()) {
190       return;
191     }
192     anObjects.push_back(anObject);
193     aContextes.push_back(anObjectAttr->context());
194   }
195
196   // Getting the center point and two points (start and end)
197   std::shared_ptr<GeomAPI_Pnt> aCenterPoint;
198   std::shared_ptr<GeomAPI_Pnt> aStartPoint;
199   std::shared_ptr<GeomAPI_Pnt> anEndPoint;
200   std::shared_ptr<ModelAPI_AttributeSelection> aCenterRef =
201     selection(FeaturesPlugin_Rotation::CENTER_POINT_ID());
202   std::shared_ptr<ModelAPI_AttributeSelection> aStartPointRef =
203     selection(FeaturesPlugin_Rotation::START_POINT_ID());
204   std::shared_ptr<ModelAPI_AttributeSelection> anEndPointRef =
205     selection(FeaturesPlugin_Rotation::END_POINT_ID());
206   if ((aCenterRef.get() != NULL) && (aStartPointRef.get() != NULL)
207       && (anEndPointRef.get() != NULL)) {
208     GeomShapePtr aCenterShape = aCenterRef->value();
209     if (!aCenterShape.get())
210       aCenterShape = aCenterRef->context()->shape();
211     GeomShapePtr aStartShape = aStartPointRef->value();
212     if (!aStartShape.get())
213       aStartShape = aStartPointRef->context()->shape();
214       GeomShapePtr anEndShape = anEndPointRef->value();
215     if (!anEndShape.get())
216       anEndShape = anEndPointRef->context()->shape();
217     if (aStartShape && anEndShape && aCenterShape) {
218       aCenterPoint = GeomAlgoAPI_PointBuilder::point(aCenterShape);
219       aStartPoint = GeomAlgoAPI_PointBuilder::point(aStartShape);
220       anEndPoint = GeomAlgoAPI_PointBuilder::point(anEndShape);
221     }
222   }
223
224   // Rotating each object.
225   int aResultIndex = 0;
226   std::list<ResultPtr>::iterator aContext = aContextes.begin();
227   for(ListOfShape::iterator anObjectsIt = anObjects.begin(); anObjectsIt != anObjects.end();
228         anObjectsIt++, aContext++) {
229     std::shared_ptr<GeomAPI_Shape> aBaseShape = *anObjectsIt;
230     bool isPart = (*aContext)->groupName() == ModelAPI_ResultPart::group();
231
232     // Setting result.
233     if (isPart) {
234        std::shared_ptr<GeomAPI_Trsf> aTrsf(new GeomAPI_Trsf());
235        aTrsf->setRotation(aCenterPoint, aStartPoint, anEndPoint);
236        ResultPartPtr anOrigin = std::dynamic_pointer_cast<ModelAPI_ResultPart>(*aContext);
237        ResultPartPtr aResultPart = document()->copyPart(anOrigin, data(), aResultIndex);
238        aResultPart->setTrsf(*aContext, aTrsf);
239        setResult(aResultPart, aResultIndex);
240     } else {
241       GeomAlgoAPI_Rotation aRotationAlgo(aBaseShape, aCenterPoint, aStartPoint, anEndPoint);
242
243       if (!aRotationAlgo.check()) {
244         setError(aRotationAlgo.getError());
245         return;
246       }
247
248       aRotationAlgo.build();
249
250       // Checking that the algorithm worked properly.
251       if(!aRotationAlgo.isDone()) {
252         static const std::string aFeatureError = "Error: Rotation algorithm failed.";
253         setError(aFeatureError);
254         break;
255       }
256       if(aRotationAlgo.shape()->isNull()) {
257         static const std::string aShapeError = "Error : Resulting shape is Null.";
258         setError(aShapeError);
259         break;
260       }
261       if(!aRotationAlgo.isValid()) {
262         std::string aFeatureError = "Error: Resulting shape is not valid.";
263         setError(aFeatureError);
264         break;
265       }
266
267       ResultBodyPtr aResultBody = document()->createBody(data(), aResultIndex);
268       loadNamingDS(aRotationAlgo, aResultBody, aBaseShape);
269       setResult(aResultBody, aResultIndex);
270     }
271     aResultIndex++;
272   }
273 }
274
275 //=================================================================================================
276 void FeaturesPlugin_Rotation::loadNamingDS(GeomAlgoAPI_Rotation& theRotaionAlgo,
277                                            std::shared_ptr<ModelAPI_ResultBody> theResultBody,
278                                            std::shared_ptr<GeomAPI_Shape> theBaseShape)
279 {
280   // Store result.
281   theResultBody->storeModified(theBaseShape, theRotaionAlgo.shape());
282
283   std::string aRotatedName = "Rotated";
284   std::shared_ptr<GeomAPI_DataMapOfShapeShape> aSubShapes = theRotaionAlgo.mapOfSubShapes();
285
286   FeaturesPlugin_Tools::storeModifiedShapes(theRotaionAlgo, theResultBody,
287                                             theBaseShape, 1, 2, 3, aRotatedName,
288                                             *aSubShapes.get());
289 }