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(EXTRUSION_FACE, ModelAPI_AttributeReference::type());
  26   data()->addAttribute(EXTRUSION_SIZE, ModelAPI_AttributeDouble::type());
  27   data()->addAttribute(EXTRUSION_REVERSE, ModelAPI_AttributeBoolean::type());
  28 }
  29
  30 void FeaturesPlugin_Extrusion::execute()
  31 {
  32   boost::shared_ptr<ModelAPI_AttributeReference> aFaceRef =
  33     boost::dynamic_pointer_cast<ModelAPI_AttributeReference>(data()->attribute(EXTRUSION_FACE));
  34   if (!aFaceRef)
  35     return;
  36   boost::shared_ptr<ModelAPI_ResultConstruction> aConstr =
  37     boost::dynamic_pointer_cast<ModelAPI_ResultConstruction>(aFaceRef->value());
  38   if (!aConstr)
  39     return;
  40   boost::shared_ptr<GeomAPI_Shape> aFace = aConstr->shape();
  41   if (!aFace)
  42     return;
  43
  44   double aSize = data()->real(EXTRUSION_SIZE)->value();
  45   if (data()->boolean(EXTRUSION_REVERSE)->value())
  46     aSize = -aSize;
  47   boost::shared_ptr<ModelAPI_ResultBody> aResult = document()->createBody();
  48   aResult->store(GeomAlgoAPI_Extrusion::makeExtrusion(aFace, aSize));
  49   document()->storeResult(data(), aResult);
  50   setResult(aResult);
  51 }