src/SketchPlugin/SketchPlugin_Feature.cpp

   1 // Copyright (C) 2014-20xx CEA/DEN, EDF R&D -->
   2
   3 #include "SketchPlugin_Feature.h"
   4 #include "SketchPlugin_Sketch.h"
   5 #include <ModelAPI_Document.h>
   6 #include <ModelAPI_Data.h>
   7 #include <ModelAPI_Object.h>
   8 #include <ModelAPI_AttributeRefList.h>
   9 #include <ModelAPI_ResultConstruction.h>
  10
  11 SketchPlugin_Feature::SketchPlugin_Feature()
  12 {
  13   mySketch = 0;
  14 }
  15
  16 void SketchPlugin_Feature::erase()
  17 {
  18   SketchPlugin_Sketch* aSketch = sketch();
  19   if (aSketch)
  20     aSketch->removeFeature(this);
  21
  22   ModelAPI_Feature::erase();
  23 }
  24
  25 SketchPlugin_Sketch* SketchPlugin_Feature::sketch()
  26 {
  27   if (!mySketch) {
  28     // find sketch that references to this feature
  29     const std::set<AttributePtr>& aBackRefs = data()->refsToMe();
  30     std::set<AttributePtr>::const_iterator aBackRef = aBackRefs.begin();
  31     for(; aBackRef != aBackRefs.end(); aBackRef++) {
  32       std::shared_ptr<SketchPlugin_Sketch> aSketch =
  33         std::dynamic_pointer_cast<SketchPlugin_Sketch>((*aBackRef)->owner());
  34       if (aSketch) {
  35         mySketch = aSketch.get();
  36         break;
  37       }
  38     }
  39   }
  40   return mySketch;
  41 }
  42