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 }