Dump Image data to python script (Feature #13).

[modules/hydro.git] / src / HYDROData / HYDROData_Iterator.cxx

diff --git a/src/HYDROData/HYDROData_Iterator.cxx b/src/HYDROData/HYDROData_Iterator.cxx
index 534cbd13c62e6bdb60621dd497bee362a2cc7558..40e8322cc70b41ff12f4198feed6924bab9e6eba 100644 (file)
--- a/src/HYDROData/HYDROData_Iterator.cxx
+++ b/src/HYDROData/HYDROData_Iterator.cxx
@@ -1,6 +1,9 @@
 #include <HYDROData_Iterator.h>
 
 #include <HYDROData_Image.h>
+#include <HYDROData_Polyline.h>
+#include <HYDROData_VisualState.h>
+#include <HYDROData_Bathymetry.h>
 
 #include <TDataStd_Name.hxx>
 #include <NCollection_DataMap.hxx>
@@ -54,6 +57,15 @@ Handle_HYDROData_Object HYDROData_Iterator::Object(const TDF_Label theLabel)
   case KIND_IMAGE:
     aResult = new HYDROData_Image();
     break;
+  case KIND_POLYLINE:
+    aResult = new HYDROData_Polyline();
+    break;
+  case KIND_VISUAL_STATE:
+    aResult = new HYDROData_VisualState();
+    break;
+  case KIND_BATHYMETRY:
+    aResult = new HYDROData_Bathymetry();
+    break;
   }
   if (!aResult.IsNull())
     aResult->SetLabel(theLabel);