src/FeaturesPlugin/FeaturesPlugin_Revolution.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D
   2
   3 // File:        FeaturesPlugin_Revolution.cpp
   4 // Created:     12 May 2015
   5 // Author:      Dmitry Bobylev
   6
   7 #include "FeaturesPlugin_Revolution.h"
   8
   9 #include <ModelAPI_AttributeDouble.h>
  10 #include <ModelAPI_AttributeSelection.h>
  11 #include <ModelAPI_AttributeString.h>
  12 #include <ModelAPI_Session.h>
  13 #include <ModelAPI_Validator.h>
  14
  15 #include <GeomAlgoAPI_Revolution.h>
  16
  17 #include <GeomAPI_Edge.h>
  18 #include <GeomAPI_Lin.h>
  19
  20 //=================================================================================================
  21 FeaturesPlugin_Revolution::FeaturesPlugin_Revolution()
  22 {
  23 }
  24
  25 //=================================================================================================
  26 void FeaturesPlugin_Revolution::initAttributes()
  27 {
  28   initCompositeSketchAttribtues(InitBaseObjectsList);
  29
  30   data()->addAttribute(AXIS_OBJECT_ID(), ModelAPI_AttributeSelection::typeId());
  31
  32   data()->addAttribute(CREATION_METHOD(), ModelAPI_AttributeString::typeId());
  33
  34   data()->addAttribute(TO_ANGLE_ID(), ModelAPI_AttributeDouble::typeId());
  35   data()->addAttribute(FROM_ANGLE_ID(), ModelAPI_AttributeDouble::typeId());
  36
  37   data()->addAttribute(TO_OBJECT_ID(), ModelAPI_AttributeSelection::typeId());
  38   data()->addAttribute(TO_OFFSET_ID(), ModelAPI_AttributeDouble::typeId());
  39
  40   data()->addAttribute(FROM_OBJECT_ID(), ModelAPI_AttributeSelection::typeId());
  41   data()->addAttribute(FROM_OFFSET_ID(), ModelAPI_AttributeDouble::typeId());
  42
  43   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), TO_OBJECT_ID());
  44   ModelAPI_Session::get()->validators()->registerNotObligatory(getKind(), FROM_OBJECT_ID());
  45
  46   initCompositeSketchAttribtues(InitSketchLauncher);
  47 }
  48
  49 //=================================================================================================
  50 void FeaturesPlugin_Revolution::execute()
  51 {
  52   ListOfShape aBaseShapesList;
  53   ListOfMakeShape aMakeShapesList;
  54
  55   // Make revolutions.
  56   if(!makeRevolutions(aBaseShapesList, aMakeShapesList)) {
  57     return;
  58   }
  59
  60   // Store results.
  61   int aResultIndex = 0;
  62   ListOfShape::const_iterator aBaseIt = aBaseShapesList.cbegin();
  63   ListOfMakeShape::const_iterator anAlgoIt = aMakeShapesList.cbegin();
  64   for(; aBaseIt != aBaseShapesList.cend() && anAlgoIt != aMakeShapesList.cend(); ++aBaseIt, ++anAlgoIt) {
  65     storeResult(*aBaseIt, *anAlgoIt, aResultIndex++);
  66   }
  67
  68   removeResults(aResultIndex);
  69 }
  70
  71 //=================================================================================================
  72 bool FeaturesPlugin_Revolution::makeRevolutions(ListOfShape& theBaseShapes,
  73                                                 ListOfMakeShape& theMakeShapes)
  74 {
  75   theMakeShapes.clear();
  76
  77   /// Sub-feature of the composite should be set in the base list.
  78   setSketchObjectToList();
  79
  80   // Getting base shapes.
  81   getBaseShapes(theBaseShapes);
  82
  83   //Getting axis.
  84   std::shared_ptr<GeomAPI_Ax1> anAxis;
  85   std::shared_ptr<GeomAPI_Edge> anEdge;
  86   AttributeSelectionPtr aSelection = selection(AXIS_OBJECT_ID());
  87   if(aSelection.get() && aSelection->value().get() && aSelection->value()->isEdge()) {
  88     anEdge = std::shared_ptr<GeomAPI_Edge>(new GeomAPI_Edge(aSelection->value()));
  89   } else if(aSelection->context().get() &&
  90             aSelection->context()->shape().get() &&
  91             aSelection->context()->shape()->isEdge()) {
  92     anEdge = std::shared_ptr<GeomAPI_Edge>(new GeomAPI_Edge(aSelection->context()->shape()));
  93   }
  94   if(anEdge.get()) {
  95     if(anEdge->isLine()) {
  96       anAxis = std::shared_ptr<GeomAPI_Ax1>(new GeomAPI_Ax1(anEdge->line()->location(),
  97                                                             anEdge->line()->direction()));
  98     }
  99   }
 100
 101   if(!anAxis.get()) {
 102     return false;
 103   }
 104
 105   // Getting angles.
 106   double aToAngle = 0.0;
 107   double aFromAngle = 0.0;
 108
 109   if(string(CREATION_METHOD())->value() == "ByAngles") {
 110     aToAngle = real(TO_ANGLE_ID())->value();
 111     aFromAngle = real(FROM_ANGLE_ID())->value();
 112   } else {
 113     aToAngle = real(TO_OFFSET_ID())->value();
 114     aFromAngle = real(FROM_OFFSET_ID())->value();
 115   }
 116
 117   // Getting bounding planes.
 118   GeomShapePtr aToShape;
 119   GeomShapePtr aFromShape;
 120
 121   if(string(CREATION_METHOD())->value() == "ByPlanesAndOffsets") {
 122     aSelection = selection(TO_OBJECT_ID());
 123     if(aSelection.get()) {
 124       aToShape = std::dynamic_pointer_cast<GeomAPI_Shape>(aSelection->value());
 125       if(!aToShape.get() && aSelection->context().get()) {
 126         aToShape = aSelection->context()->shape();
 127       }
 128     }
 129     aSelection = selection(FROM_OBJECT_ID());
 130     if(aSelection.get()) {
 131       aFromShape = std::dynamic_pointer_cast<GeomAPI_Shape>(aSelection->value());
 132       if(!aFromShape.get() && aSelection->context().get()) {
 133         aFromShape = aSelection->context()->shape();
 134       }
 135     }
 136   }
 137
 138   // Generating result for each base shape.
 139   for(ListOfShape::const_iterator anIter = theBaseShapes.cbegin(); anIter != theBaseShapes.cend(); anIter++) {
 140     GeomShapePtr aBaseShape = *anIter;
 141
 142     std::shared_ptr<GeomAlgoAPI_Revolution> aRevolAlgo(new GeomAlgoAPI_Revolution(aBaseShape, anAxis,
 143                                                                                   aToShape, aToAngle,
 144                                                                                   aFromShape, aFromAngle));
 145     if(!isMakeShapeValid(aRevolAlgo)) {
 146       return false;
 147     }
 148
 149     theMakeShapes.push_back(aRevolAlgo);
 150   }
 151
 152   return true;
 153 }