Small correction for writing of data to file.

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

diff --git a/src/HYDROData/HYDROData_Iterator.cxx b/src/HYDROData/HYDROData_Iterator.cxx
index da22e5ebf0d41eb246e0869e667e8432d2caca77..40e8322cc70b41ff12f4198feed6924bab9e6eba 100644 (file)
--- a/src/HYDROData/HYDROData_Iterator.cxx
+++ b/src/HYDROData/HYDROData_Iterator.cxx
@@ -2,6 +2,8 @@
 
 #include <HYDROData_Image.h>
 #include <HYDROData_Polyline.h>
+#include <HYDROData_VisualState.h>
+#include <HYDROData_Bathymetry.h>
 
 #include <TDataStd_Name.hxx>
 #include <NCollection_DataMap.hxx>
@@ -58,6 +60,12 @@ Handle_HYDROData_Object HYDROData_Iterator::Object(const TDF_Label theLabel)
   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);