#refs 522 (very draft) //import only

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

diff --git a/src/HYDROData/HYDROData_Polyline3D.cxx b/src/HYDROData/HYDROData_Polyline3D.cxx
index c8549930cf3ea682cb7b7e9ca7d898210d78911a..b2650b635d77f1ee52c2c5046f281dc6e5626875 100644 (file)
--- a/src/HYDROData/HYDROData_Polyline3D.cxx
+++ b/src/HYDROData/HYDROData_Polyline3D.cxx
@@ -1,3 +1,24 @@
+// Copyright (C) 2007-2015  CEA/DEN, EDF R&D, OPEN CASCADE
+//
+// Copyright (C) 2003-2007  OPEN CASCADE, EADS/CCR, LIP6, CEA/DEN,
+// CEDRAT, EDF R&D, LEG, PRINCIPIA R&D, BUREAU VERITAS
+//
+// This library is free software; you can redistribute it and/or
+// modify it under the terms of the GNU Lesser General Public
+// License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+//
+// This library is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+// Lesser General Public License for more details.
+//
+// You should have received a copy of the GNU Lesser General Public
+// License along with this library; if not, write to the Free Software
+// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
+//
+// See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
+//
 
 #include "HYDROData_Polyline3D.h"
 
@@ -44,7 +65,7 @@ QStringList HYDROData_Polyline3D::DumpToPython( MapOfTreatedObjects& theTreatedO
   {
     Handle(HYDROData_Profile) aProfile = 
       Handle(HYDROData_Profile)::DownCast( aRefProfileUZ->GetFatherObject() );
-    if ( !aProfile.IsNull() )
+    if ( checkObjectPythonDefinition( theTreatedObjects, aResList, aProfile ) )
     {
       QString aProfileName = aProfile->GetObjPyName();
       if ( !aProfileName.isEmpty() )
@@ -57,7 +78,26 @@ QStringList HYDROData_Polyline3D::DumpToPython( MapOfTreatedObjects& theTreatedO
   else
   {
     Handle(HYDROData_IAltitudeObject) aRefBathymetry = GetAltitudeObject();
-    setPythonReferenceObject( theTreatedObjects, aResList, aRefBathymetry, "SetAltitudeObject" );
+    if ( !aRefBathymetry.IsNull() )
+    {
+      Handle(HYDROData_ProfileUZ) aChildProfileUZ = GetChildProfileUZ();
+      if ( !aChildProfileUZ.IsNull() )
+      {
+        Handle(HYDROData_Profile) aProfile = 
+          Handle(HYDROData_Profile)::DownCast( aChildProfileUZ->GetFatherObject() );
+        if ( checkObjectPythonDefinition( theTreatedObjects, aResList, aProfile ) )
+        {
+          QString aProfileName = aProfile->GetObjPyName();
+          if ( !aProfileName.isEmpty() )
+          {
+            aResList << QString( "%1.SetChildProfileUZ( %2.GetProfileUZ() );" )
+                         .arg( aPolylineName ).arg( aProfileName );
+          }
+        }
+      }
+
+      setPythonReferenceObject( theTreatedObjects, aResList, aRefBathymetry, "SetAltitudeObject" );
+    }
   }
 
   aResList << QString( "" );
@@ -307,11 +347,16 @@ Handle(HYDROData_ProfileUZ) HYDROData_Polyline3D::GetChildProfileUZ( const bool
   aProfileUZ = aProfile->GetProfileUZ();
 
   HYDROData_Polyline3D* me = const_cast<HYDROData_Polyline3D*>( this ); // Temporary to be revised
-  me->SetReferenceObject( aProfileUZ, DataTag_ChildProfileUZ );
+  me->SetChildProfileUZ( aProfileUZ );
 
   return aProfileUZ;
 }
 
+void HYDROData_Polyline3D::SetChildProfileUZ( const Handle(HYDROData_ProfileUZ)& theProfile )
+{
+  SetReferenceObject( theProfile, DataTag_ChildProfileUZ );
+}
+
 HYDROData_IPolyline::PointsList generateProfileUZPoints(
   const Handle(HYDROData_PolylineXY)&      thePolyline,
   const Handle(HYDROData_IAltitudeObject)& theAltitude )