src/FeaturesPlugin/FeaturesPlugin_Extrusion.cpp

   1 // File:        FeaturesPlugin_Extrusion.cpp
   2 // Created:     30 May 2014
   3 // Author:      Vitaly SMETANNIKOV
   4
   5 #include "FeaturesPlugin_Extrusion.h"
   6 #include <ModelAPI_PluginManager.h>
   7 #include <ModelAPI_Document.h>
   8 #include <ModelAPI_Data.h>
   9 #include <ModelAPI_ResultConstruction.h>
  10 #include <ModelAPI_ResultBody.h>
  11 #include <ModelAPI_AttributeDouble.h>
  12 #include <ModelAPI_AttributeReference.h>
  13 #include <ModelAPI_AttributeBoolean.h>
  14
  15 #include <GeomAlgoAPI_Extrusion.h>
  16
  17 using namespace std;
  18
  19 FeaturesPlugin_Extrusion::FeaturesPlugin_Extrusion()
  20 {
  21 }
  22
  23 void FeaturesPlugin_Extrusion::initAttributes()
  24 {
  25   data()->addAttribute(FeaturesPlugin_Extrusion::FACE_ID(), ModelAPI_AttributeReference::type());
  26   data()->addAttribute(FeaturesPlugin_Extrusion::SIZE_ID(), ModelAPI_AttributeDouble::type());
  27   data()->addAttribute(FeaturesPlugin_Extrusion::REVERSE_ID(), ModelAPI_AttributeBoolean::type());
  28 }
  29
  30 void FeaturesPlugin_Extrusion::execute()
  31 {
  32   boost::shared_ptr<ModelAPI_AttributeReference> aFaceRef = boost::dynamic_pointer_cast<
  33       ModelAPI_AttributeReference>(data()->attribute(FeaturesPlugin_Extrusion::FACE_ID()));
  34   if (!aFaceRef)
  35     return;
  36   boost::shared_ptr<GeomAPI_Shape> aFace;
  37   boost::shared_ptr<ModelAPI_ResultConstruction> aConstr = boost::dynamic_pointer_cast<
  38       ModelAPI_ResultConstruction>(aFaceRef->value());
  39   if (aConstr) {
  40     aFace = aConstr->shape();
  41   }
  42   if (!aFace) {
  43     // Check for body
  44     boost::shared_ptr<ModelAPI_ResultBody> aBody = boost::dynamic_pointer_cast<
  45         ModelAPI_ResultBody>(aFaceRef->value());
  46     if (aBody)
  47       aFace = aBody->shape();
  48   }
  49   if (!aFace)
  50     return;
  51
  52   double aSize = data()->real(FeaturesPlugin_Extrusion::SIZE_ID())->value();
  53   if (data()->boolean(FeaturesPlugin_Extrusion::REVERSE_ID())->value())
  54     aSize = -aSize;
  55   boost::shared_ptr<ModelAPI_ResultBody> aResult = document()->createBody(data());
  56   aResult->store(GeomAlgoAPI_Extrusion::makeExtrusion(aFace, aSize));
  57   setResult(aResult);
  58 }