src/FeaturesPlugin/FeaturesPlugin_Rotation.cpp

   1 // Copyright (C) 2014-2021  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 email : webmaster.salome@opencascade.com
  18 //
  19
  20 #include <FeaturesPlugin_Rotation.h>
  21
  22 #include <ModelAPI_AttributeDouble.h>
  23 #include <ModelAPI_AttributeSelectionList.h>
  24 #include <ModelAPI_AttributeString.h>
  25 #include <ModelAPI_ResultBody.h>
  26 #include <ModelAPI_ResultPart.h>
  27 #include <ModelAPI_Tools.h>
  28
  29 #include <GeomAlgoAPI_MakeShapeList.h>
  30 #include <GeomAlgoAPI_PointBuilder.h>
  31 #include <GeomAlgoAPI_Rotation.h>
  32 #include <GeomAlgoAPI_Tools.h>
  33
  34 #include <GeomAPI_Ax1.h>
  35 #include <GeomAPI_Edge.h>
  36 #include <GeomAPI_Lin.h>
  37 #include <GeomAPI_ShapeIterator.h>
  38 #include <GeomAPI_ShapeHierarchy.h>
  39 #include <GeomAPI_Trsf.h>
  40
  41 #include <FeaturesPlugin_Tools.h>
  42
  43 static const std::string ROTATION_VERSION_1("v9.5");
  44
  45 //=================================================================================================
  46 FeaturesPlugin_Rotation::FeaturesPlugin_Rotation()
  47 {
  48 }
  49
  50 //=================================================================================================
  51 void FeaturesPlugin_Rotation::initAttributes()
  52 {
  53   data()->addAttribute(FeaturesPlugin_Rotation::CREATION_METHOD(),
  54                        ModelAPI_AttributeString::typeId());
  55
  56   AttributeSelectionListPtr aSelection =
  57     std::dynamic_pointer_cast<ModelAPI_AttributeSelectionList>(data()->addAttribute(
  58     FeaturesPlugin_Rotation::OBJECTS_LIST_ID(), ModelAPI_AttributeSelectionList::typeId()));
  59
  60   data()->addAttribute(FeaturesPlugin_Rotation::AXIS_OBJECT_ID(),
  61                        ModelAPI_AttributeSelection::typeId());
  62   data()->addAttribute(FeaturesPlugin_Rotation::ANGLE_ID(), ModelAPI_AttributeDouble::typeId());
  63
  64   data()->addAttribute(FeaturesPlugin_Rotation::CENTER_POINT_ID(),
  65                        ModelAPI_AttributeSelection::typeId());
  66   data()->addAttribute(FeaturesPlugin_Rotation::START_POINT_ID(),
  67                        ModelAPI_AttributeSelection::typeId());
  68   data()->addAttribute(FeaturesPlugin_Rotation::END_POINT_ID(),
  69                        ModelAPI_AttributeSelection::typeId());
  70
  71   if (!aSelection->isInitialized()) {
  72     // new feature, not read from file
  73     data()->setVersion(ROTATION_VERSION_1);
  74   }
  75 }
  76
  77 //=================================================================================================
  78 void FeaturesPlugin_Rotation::execute()
  79 {
  80   AttributeStringPtr aMethodTypeAttr = string(FeaturesPlugin_Rotation::CREATION_METHOD());
  81   std::string aMethodType = aMethodTypeAttr->value();
  82
  83   GeomTrsfPtr aTrsf;
  84   if (aMethodType == CREATION_METHOD_BY_ANGLE())
  85     aTrsf = rotationByAxisAndAngle();
  86   else if (aMethodType == CREATION_METHOD_BY_THREE_POINTS())
  87     aTrsf = rotationByThreePoints();
  88
  89   performRotation(aTrsf);
  90 }
  91
  92 //=================================================================================================
  93 GeomTrsfPtr FeaturesPlugin_Rotation::rotationByAxisAndAngle()
  94 {
  95   //Getting axis.
  96   static const std::string aSelectionError = "Error: The axis shape selection is bad.";
  97   AttributeSelectionPtr anObjRef = selection(AXIS_OBJECT_ID());
  98   GeomShapePtr aShape = anObjRef->value();
  99   if (!aShape.get()) {
 100     if (anObjRef->context().get()) {
 101       aShape = anObjRef->context()->shape();
 102     }
 103   }
 104   if (!aShape.get()) {
 105     setError(aSelectionError);
 106     return GeomTrsfPtr();
 107   }
 108
 109   GeomEdgePtr anEdge;
 110   if (aShape->isEdge())
 111   {
 112     anEdge = aShape->edge();
 113   }
 114   else if (aShape->isCompound())
 115   {
 116     GeomAPI_ShapeIterator anIt(aShape);
 117     anEdge = anIt.current()->edge();
 118   }
 119
 120   if (!anEdge.get())
 121   {
 122     setError(aSelectionError);
 123     return GeomTrsfPtr();
 124   }
 125
 126   std::shared_ptr<GeomAPI_Ax1> anAxis (new GeomAPI_Ax1(anEdge->line()->location(),
 127                                                        anEdge->line()->direction()));
 128   double anAngle = real(FeaturesPlugin_Rotation::ANGLE_ID())->value();
 129
 130   GeomTrsfPtr aTrsf(new GeomAPI_Trsf());
 131   aTrsf->setRotation(anAxis, anAngle);
 132   return aTrsf;
 133 }
 134
 135 //=================================================================================================
 136 GeomTrsfPtr FeaturesPlugin_Rotation::rotationByThreePoints()
 137 {
 138   // Getting the center point and two points (start and end)
 139   std::shared_ptr<GeomAPI_Pnt> aCenterPoint;
 140   std::shared_ptr<GeomAPI_Pnt> aStartPoint;
 141   std::shared_ptr<GeomAPI_Pnt> anEndPoint;
 142   std::shared_ptr<ModelAPI_AttributeSelection> aCenterRef =
 143     selection(FeaturesPlugin_Rotation::CENTER_POINT_ID());
 144   std::shared_ptr<ModelAPI_AttributeSelection> aStartPointRef =
 145     selection(FeaturesPlugin_Rotation::START_POINT_ID());
 146   std::shared_ptr<ModelAPI_AttributeSelection> anEndPointRef =
 147     selection(FeaturesPlugin_Rotation::END_POINT_ID());
 148   if ((aCenterRef.get() != NULL) &&
 149       (aStartPointRef.get() != NULL) &&
 150       (anEndPointRef.get() != NULL)) {
 151     GeomShapePtr aCenterShape = aCenterRef->value();
 152     if (!aCenterShape.get() && aCenterRef->context().get())
 153       aCenterShape = aCenterRef->context()->shape();
 154     GeomShapePtr aStartShape = aStartPointRef->value();
 155     if (!aStartShape.get() && aStartPointRef->context().get())
 156       aStartShape = aStartPointRef->context()->shape();
 157     GeomShapePtr anEndShape = anEndPointRef->value();
 158     if (!anEndShape.get() && anEndPointRef->context().get())
 159       anEndShape = anEndPointRef->context()->shape();
 160     if (aStartShape && anEndShape && aCenterShape) {
 161       aCenterPoint = GeomAlgoAPI_PointBuilder::point(aCenterShape);
 162       aStartPoint = GeomAlgoAPI_PointBuilder::point(aStartShape);
 163       anEndPoint = GeomAlgoAPI_PointBuilder::point(anEndShape);
 164     }
 165
 166     if (!aCenterPoint || !aStartPoint || !anEndPoint)
 167       return GeomTrsfPtr();
 168   }
 169
 170   GeomTrsfPtr aTrsf(new GeomAPI_Trsf());
 171   aTrsf->setRotation(aCenterPoint, aStartPoint, anEndPoint);
 172   return aTrsf;
 173 }
 174
 175 //=================================================================================================
 176 void FeaturesPlugin_Rotation::performRotation(const GeomTrsfPtr& theTrsf)
 177 {
 178   if (!theTrsf) {
 179     setError("Invalid transformation.");
 180     return;
 181   }
 182
 183   bool isKeepSubShapes = data()->version() == ROTATION_VERSION_1;
 184
 185   // Getting objects.
 186   GeomAPI_ShapeHierarchy anObjects;
 187   std::list<ResultPtr> aParts;
 188   ResultPtr aTextureSource;
 189   AttributeSelectionListPtr anObjSelList = selectionList(OBJECTS_LIST_ID());
 190   if (!FeaturesPlugin_Tools::shapesFromSelectionList(
 191        anObjSelList, isKeepSubShapes, anObjects, aParts, aTextureSource))
 192     return;
 193
 194   std::string anError;
 195   int aResultIndex = 0;
 196   // Rotating each part.
 197   for (std::list<ResultPtr>::iterator aPRes = aParts.begin(); aPRes != aParts.end(); ++aPRes) {
 198     ResultPartPtr anOriginal = std::dynamic_pointer_cast<ModelAPI_ResultPart>(*aPRes);
 199     ResultPartPtr aResultPart = document()->copyPart(anOriginal, data(), aResultIndex);
 200     aResultPart->setTrsf(anOriginal, theTrsf);
 201     setResult(aResultPart, aResultIndex++);
 202   }
 203
 204   // Collect transformations for each object in a part.
 205   std::shared_ptr<GeomAlgoAPI_MakeShapeList> aMakeShapeList(new GeomAlgoAPI_MakeShapeList);
 206   for(GeomAPI_ShapeHierarchy::iterator anObjectsIt = anObjects.begin();
 207       anObjectsIt != anObjects.end(); ++anObjectsIt) {
 208     std::shared_ptr<GeomAPI_Shape> aBaseShape = *anObjectsIt;
 209     std::shared_ptr<GeomAlgoAPI_Transform> aRotationAlgo(
 210         new GeomAlgoAPI_Transform(aBaseShape, theTrsf));
 211
 212     // Checking that the algorithm worked properly.
 213     if (GeomAlgoAPI_Tools::AlgoError::isAlgorithmFailed(aRotationAlgo, getKind(), anError)) {
 214       setError(anError);
 215       break;
 216     }
 217
 218     anObjects.markModified(aBaseShape, aRotationAlgo->shape());
 219     aMakeShapeList->appendAlgo(aRotationAlgo);
 220   }
 221
 222   // Build results of the rotation.
 223   const ListOfShape& anOriginalShapes = anObjects.objects();
 224   ListOfShape aTopLevel;
 225   anObjects.topLevelObjects(aTopLevel);
 226   for (ListOfShape::iterator anIt = aTopLevel.begin(); anIt != aTopLevel.end(); ++anIt) {
 227     ResultBodyPtr aResultBody = document()->createBody(data(), aResultIndex);
 228     ModelAPI_Tools::loadModifiedShapes(aResultBody, anOriginalShapes, ListOfShape(),
 229                                        aMakeShapeList, *anIt, "Rotated");
 230     // Copy image data, if any
 231     ModelAPI_Tools::copyImageAttribute(aTextureSource, aResultBody);
 232     setResult(aResultBody, aResultIndex++);
 233   }
 234
 235   // Remove the rest results if there were produced in the previous pass.
 236   removeResults(aResultIndex);
 237 }