Make generated by SWIG python script installed: refs #17

[modules/shaper.git] / src / Model / Model_Iterator.cxx

diff --git a/src/Model/Model_Iterator.cxx b/src/Model/Model_Iterator.cxx
index ae62fa928d37acc908da3989b0e1a9a163b0cb7c..001ebdff8146c3d2bab6e71cc80cbca487696fa2 100644 (file)
--- a/src/Model/Model_Iterator.cxx
+++ b/src/Model/Model_Iterator.cxx
@@ -4,6 +4,8 @@
 
 #include "Model_Iterator.h"
 #include "Model_Document.h"
+#include "ModelAPI_Feature.h"
+#include "Model_Data.h"
 #include <TDataStd_Comment.hxx>
 #include <TDataStd_Name.hxx>
 
@@ -16,7 +18,7 @@ void Model_Iterator::next()
 
 bool Model_Iterator::more()
 {
-  return myIter.More();
+  return myIter.More() == Standard_True;
 }
 
 shared_ptr<ModelAPI_Feature> Model_Iterator::current()
@@ -49,6 +51,17 @@ int Model_Iterator::numIterationsLeft()
   return aResult;
 }
 
+bool Model_Iterator::isEqual(std::shared_ptr<ModelAPI_Feature> theFeature)
+{
+  return (myIter.Value()->Label() == 
+    dynamic_pointer_cast<Model_Data>(theFeature->data())->label()) == Standard_True;
+
+}
+
+Model_Iterator::Model_Iterator()
+{
+}
+
 Model_Iterator::Model_Iterator(std::shared_ptr<Model_Document> theDoc, TDF_Label theLab)
   : myDoc(theDoc), myIter(theLab, TDataStd_Comment::GetID(), Standard_False)
 {}