Bug #334: Different behavior in Edit calculation case input panel.

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

diff --git a/src/HYDROData/HYDROData_Stream.cxx b/src/HYDROData/HYDROData_Stream.cxx
index fbe9212b3b94d89cb156c23d547c3acd52d87759..df6ad67b532cbdb31289a6330ec2929853b49af7 100644 (file)
--- a/src/HYDROData/HYDROData_Stream.cxx
+++ b/src/HYDROData/HYDROData_Stream.cxx
@@ -88,7 +88,15 @@ QStringList HYDROData_Stream::DumpToPython( MapOfTreatedObjects& theTreatedObjec
   QStringList aResList = dumpObjectCreation( theTreatedObjects );
   QString aName = GetObjPyName();
 
-  // TODO
+  Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
+  setPythonReferenceObject( theTreatedObjects, aResList, aHydAxis, "SetHydraulicAxis" );
+
+  HYDROData_SequenceOfObjects aSeqOfProfiles = GetProfiles();
+  for ( int i = 1, aNb = aSeqOfProfiles.Size(); i <= aNb; ++i )
+  {
+    const Handle(HYDROData_Entity) aProfile = aSeqOfProfiles.Value( i );
+    setPythonReferenceObject( theTreatedObjects, aResList, aProfile, "AddProfile" );
+  }
 
   aResList << QString( "" );
   aResList << QString( "%1.Update();" ).arg( aName );
@@ -321,14 +329,6 @@ void HYDROData_Stream::UpdatePrs()
   Handle(HYDROData_ShapesGroup) anOutGroup = createGroupObject();
   anOutGroup->SetName( anOutGroupName );
   anOutGroup->AddShape( aTopEdge );
-
-  // This peace of code is for testing of functionality of altitude,
-  // will be removed by adv when testing will be finished
-  Handle(HYDROData_IAltitudeObject) anAltObj = GetAltitudeObject();
-  gp_XY aTestPnt( 5, 0 );
-  anAltObj->GetAltitudeForPoint( aTestPnt );
-  // End of test code
-
 }
 
 QColor HYDROData_Stream::DefaultFillingColor()