Dump to python corrected.

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

diff --git a/src/HYDROData/HYDROData_Polyline3D.cxx b/src/HYDROData/HYDROData_Polyline3D.cxx
index 6239b431bfb24027bc578ebc2625d39a72711e63..c8549930cf3ea682cb7b7e9ca7d898210d78911a 100644 (file)
--- a/src/HYDROData/HYDROData_Polyline3D.cxx
+++ b/src/HYDROData/HYDROData_Polyline3D.cxx
@@ -34,19 +34,34 @@ HYDROData_Polyline3D::~HYDROData_Polyline3D()
 QStringList HYDROData_Polyline3D::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
 {
   QStringList aResList = dumpObjectCreation( theTreatedObjects );
-  QString aName = GetObjPyName();
+  QString aPolylineName = GetObjPyName();
 
   Handle(HYDROData_PolylineXY) aRefPolyline = GetPolylineXY();
   setPythonReferenceObject( theTreatedObjects, aResList, aRefPolyline, "SetPolylineXY" );
 
-  Handle(HYDROData_ProfileUZ) aRefProfile = GetProfileUZ();
-  setPythonReferenceObject( theTreatedObjects, aResList, aRefProfile, "SetProfileUZ" );
-
-  Handle(HYDROData_IAltitudeObject) aRefBathymetry = GetAltitudeObject();
-  setPythonReferenceObject( theTreatedObjects, aResList, aRefBathymetry, "SetAltitudeObject" );
+  Handle(HYDROData_ProfileUZ) aRefProfileUZ = GetProfileUZ();
+  if ( !aRefProfileUZ.IsNull() )
+  {
+    Handle(HYDROData_Profile) aProfile = 
+      Handle(HYDROData_Profile)::DownCast( aRefProfileUZ->GetFatherObject() );
+    if ( !aProfile.IsNull() )
+    {
+      QString aProfileName = aProfile->GetObjPyName();
+      if ( !aProfileName.isEmpty() )
+      {
+        aResList << QString( "%1.SetProfileUZ( %2.GetProfileUZ() );" )
+                     .arg( aPolylineName ).arg( aProfileName );
+      }
+    }
+  }
+  else
+  {
+    Handle(HYDROData_IAltitudeObject) aRefBathymetry = GetAltitudeObject();
+    setPythonReferenceObject( theTreatedObjects, aResList, aRefBathymetry, "SetAltitudeObject" );
+  }
 
   aResList << QString( "" );
-  aResList << QString( "%1.Update();" ).arg( aName );
+  aResList << QString( "%1.Update();" ).arg( aPolylineName );
   aResList << QString( "" );
 
   return aResList;
@@ -291,7 +306,7 @@ Handle(HYDROData_ProfileUZ) HYDROData_Polyline3D::GetChildProfileUZ( const bool
 
   aProfileUZ = aProfile->GetProfileUZ();
 
-  Handle(HYDROData_Polyline3D) me = this;
+  HYDROData_Polyline3D* me = const_cast<HYDROData_Polyline3D*>( this ); // Temporary to be revised
   me->SetReferenceObject( aProfileUZ, DataTag_ChildProfileUZ );
 
   return aProfileUZ;