New files added.

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

diff --git a/src/HYDROData/HYDROData_Iterator.cxx b/src/HYDROData/HYDROData_Iterator.cxx
index 40e8322cc70b41ff12f4198feed6924bab9e6eba..401b54d40f10751a3c382aa1dbd36fb4d74df715 100644 (file)
--- a/src/HYDROData/HYDROData_Iterator.cxx
+++ b/src/HYDROData/HYDROData_Iterator.cxx
@@ -4,6 +4,8 @@
 #include <HYDROData_Polyline.h>
 #include <HYDROData_VisualState.h>
 #include <HYDROData_Bathymetry.h>
+#include <HYDROData_Calculation.h>
+#include <HYDROData_Zone.h>
 
 #include <TDataStd_Name.hxx>
 #include <NCollection_DataMap.hxx>
@@ -49,11 +51,13 @@ Handle_HYDROData_Object HYDROData_Iterator::CreateObject(
   return Object(aNewLab);
 }
 
-Handle_HYDROData_Object HYDROData_Iterator::Object(const TDF_Label theLabel)
+Handle_HYDROData_Object HYDROData_Iterator::Object( const TDF_Label theLabel )
 {
   ObjectKind aKind = theLabel.Father().Tag();
+  
   Handle(HYDROData_Object) aResult;
-  switch(aKind) {
+  switch( aKind )
+  {
   case KIND_IMAGE:
     aResult = new HYDROData_Image();
     break;
@@ -66,8 +70,16 @@ Handle_HYDROData_Object HYDROData_Iterator::Object(const TDF_Label theLabel)
   case KIND_BATHYMETRY:
     aResult = new HYDROData_Bathymetry();
     break;
+  case KIND_CALCULATION:
+    aResult = new HYDROData_Calculation();
+    break;
+  case KIND_ZONE:
+    aResult = new HYDROData_Zone();
+    break;
   }
-  if (!aResult.IsNull())
-    aResult->SetLabel(theLabel);
+
+  if ( !aResult.IsNull() )
+    aResult->SetLabel( theLabel );
+
   return aResult;
 }