X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FSketchPlugin%2FSketchPlugin_Feature.cpp;h=6e0f7111d1c29125622aae6885db76c5e1003b5e;hb=28d5d911e78c187ce729f155d8b2074f0d3561c3;hp=00c849fc9e1023abf5aa5c6f842c8852a6055722;hpb=a398103d11950df8cb50eeea49942237eb16edcf;p=modules%2Fshaper.git diff --git a/src/SketchPlugin/SketchPlugin_Feature.cpp b/src/SketchPlugin/SketchPlugin_Feature.cpp index 00c849fc9..6e0f7111d 100644 --- a/src/SketchPlugin/SketchPlugin_Feature.cpp +++ b/src/SketchPlugin/SketchPlugin_Feature.cpp @@ -1,41 +1,33 @@ +// Copyright (C) 2014-20xx CEA/DEN, EDF R&D --> + #include "SketchPlugin_Feature.h" #include "SketchPlugin_Sketch.h" #include #include +#include #include +#include SketchPlugin_Feature::SketchPlugin_Feature() { mySketch = 0; } -void SketchPlugin_Feature::setData(boost::shared_ptr theData) +SketchPlugin_Sketch* SketchPlugin_Feature::sketch() { - ModelAPI_Feature::setData(theData); - - // find sketch that references to this feature - int aSketches = document()->size("Construction"); - for(int a = 0; a < aSketches && !mySketch; a++) { - boost::shared_ptr aSketch = - boost::dynamic_pointer_cast(document()->feature("Construction", a)); - std::list > aList = - aSketch->data()->reflist(SKETCH_ATTR_FEATURES)->list(); - std::list >::iterator aSub = aList.begin(); - for(; aSub != aList.end(); aSub++) { - if ((*aSub)->data()->isEqual(theData)) { + if (!mySketch) { + // find sketch that references to this feature + const std::set& aBackRefs = data()->refsToMe(); + std::set::const_iterator aBackRef = aBackRefs.begin(); + for(; aBackRef != aBackRefs.end(); aBackRef++) { + std::shared_ptr aSketch = + std::dynamic_pointer_cast((*aBackRef)->owner()); + if (aSketch) { mySketch = aSketch.get(); break; } } } + return mySketch; } -void SketchPlugin_Feature::setPreview(const boost::shared_ptr& theShape) -{ - myPreview = theShape; -} - -const boost::shared_ptr& SketchPlugin_Feature::getPreview() const -{ - return myPreview; -}