Salome HOME
Lot 2: change bathy associated to natural object propagated to all cases without...
[modules/hydro.git] / src / HYDROData / HYDROData_Stream.cxx
index 2cf711653065978c1d331327b8b2a39ffb4761a1..431e22ae26511b81ca9f270acd67f6317ca80d72 100644 (file)
@@ -28,6 +28,8 @@
 #include "HYDROData_IProfilesInterpolator.h"
 #include "HYDROData_Tool.h"
 #include "HYDROData_DTM.h"
+#include "HYDROData_LISM.h"
+
 #include <HYDROData_Bathymetry.h>
 
 #include <TDataStd_RealArray.hxx>
@@ -117,6 +119,10 @@ QStringList HYDROData_Stream::DumpToPython( const QString&       thePyScriptPath
   QStringList aResList = dumpObjectCreation( theTreatedObjects );
   QString aName = GetObjPyName();
 
+  int interpMethod = GetInterpolationMethod();
+  QString anInterpMethod = QString::number( interpMethod );
+  aResList << QString( "%1.SetInterpolationMethod( %2 )" ).arg( aName ).arg( anInterpMethod );
+
   Handle(HYDROData_PolylineXY) aHydAxis = GetHydraulicAxis();
   setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aHydAxis, "SetHydraulicAxis" );
 
@@ -133,10 +139,25 @@ QStringList HYDROData_Stream::DumpToPython( const QString&       thePyScriptPath
     setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aBottomPolyline, "SetBottomPolyline" );
   }
 
-  QString aDDZs = QString::number( GetDDZ(), 'f', 3 );
-  QString aSSteps = QString::number( GetSpatialStep(), 'f', 3 );
-  aResList << QString( "%1.SetDDZ( %2 )" ).arg( aName ).arg( aDDZs );
-  aResList << QString( "%1.SetSpatialStep( %2 )" ).arg( aName ).arg( aSSteps );
+  if (interpMethod==0)
+  {
+    QString aDDZs = QString::number( GetDDZ(), 'f', 3 );
+    QString aSSteps = QString::number( GetSpatialStep(), 'f', 3 );
+    aResList << QString( "%1.SetDDZ( %2 )" ).arg( aName ).arg( aDDZs );
+    aResList << QString( "%1.SetSpatialStep( %2 )" ).arg( aName ).arg( aSSteps );
+  }
+  else if (interpMethod==1)
+  {
+    Handle(HYDROData_PolylineXY) aLeftBank = GetLeftBank();
+    setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aLeftBank, "SetLeftBank" );
+    Handle(HYDROData_PolylineXY) aRightBank = GetRightBank();
+    setPythonReferenceObject( thePyScriptPath, theTreatedObjects, aResList, aRightBank, "SetRightBank" );
+
+    QString aHaxStep = QString::number( GetHaxStep(), 'f', 3 );
+    QString aNbProfilePoints = QString::number( GetNbProfilePoints() );
+    aResList << QString( "%1.SetHaxStep( %2 )" ).arg( aName ).arg( aHaxStep );
+    aResList << QString( "%1.SetNbProfilePoints( %2 )" ).arg( aName ).arg( aNbProfilePoints );
+  }
 
   aResList << QString( "" );
   aResList << QString( "%1.Update()" ).arg( aName );
@@ -177,27 +198,40 @@ void HYDROData_Stream::GetWarnings(NCollection_DataMap<Handle(HYDROData_Profile)
 
 void HYDROData_Stream::Update()
 {
-  if (!GetHydraulicAxis().IsNull())
-    updateProfilesOrder();
-
-  // Update bottom polyline if exists
-  const Handle(HYDROData_Polyline3D) aBottomPolyline = GetBottomPolyline();
-  if ( !aBottomPolyline.IsNull() ) {
-    if ( GenerateBottomPolyline() ) {
-      Handle(HYDROData_PolylineXY) aPolylineXY = aBottomPolyline->GetPolylineXY();
-      if ( !aPolylineXY.IsNull() ) {
-        aPolylineXY->Update();
+  if (GetInterpolationMethod() == 0)
+  {
+    if (!GetHydraulicAxis().IsNull())
+      updateProfilesOrder();
+
+    // Update bottom polyline if exists
+    const Handle(HYDROData_Polyline3D) aBottomPolyline = GetBottomPolyline();
+    if ( !aBottomPolyline.IsNull() ) {
+      if ( GenerateBottomPolyline() ) {
+        Handle(HYDROData_PolylineXY) aPolylineXY = aBottomPolyline->GetPolylineXY();
+        if ( !aPolylineXY.IsNull() ) {
+          aPolylineXY->Update();
+        }
+        aBottomPolyline->Update();
       }
-      aBottomPolyline->Update();
     }
-  }
 
-  Handle(HYDROData_DTM) dtm = DTM();
-  dtm->Update();
-  UpdatePrs( dtm );
+    Handle(HYDROData_DTM) dtm = DTM();
+    dtm->Update();
+    UpdatePrs( dtm );
 
-  myWarnings.Clear();
-  dtm->GetWarnings(myWarnings);
+    myWarnings.Clear();
+    dtm->GetWarnings(myWarnings);
+  }
+  else
+  {
+    Handle(HYDROData_LISM) lism = LISM();
+    lism->Update();
+    UpdatePrs( lism );
+
+    myWarnings.Clear();
+    //lism->GetWarnings(myWarnings);
+    //TODO warnings
+  }
 
   HYDROData_NaturalObject::Update();
 }
@@ -253,14 +287,8 @@ bool HYDROData_Stream::CreatePresentations( const Handle(HYDROData_DTM)& theDTM,
   return true;
 }
 
-void HYDROData_Stream::UpdatePrs( const Handle(HYDROData_DTM)& theDTM )
+void HYDROData_Stream::internalUpdatePrs( const PrsDefinition& aResultPrs )
 {
-  HYDROData_NaturalObject::Update();
-  
-  PrsDefinition aResultPrs;
-  if ( !CreatePresentations( theDTM, aResultPrs ) )
-    return;
-
   SetShape3D( aResultPrs.myPrs3D );
   SetTopShape( aResultPrs.myPrs2D );
 
@@ -290,6 +318,27 @@ void HYDROData_Stream::UpdatePrs( const Handle(HYDROData_DTM)& theDTM )
   anOutGroup->AddShape( aResultPrs.myOutlet );
 }
 
+
+void HYDROData_Stream::UpdatePrs( const Handle(HYDROData_DTM)& theDTM )
+{
+  HYDROData_NaturalObject::Update();
+  
+  PrsDefinition aResultPrs;
+  if ( !CreatePresentations( theDTM, aResultPrs ) )
+    return;
+
+  internalUpdatePrs(aResultPrs);
+}
+
+void HYDROData_Stream::UpdatePrs( const Handle(HYDROData_LISM)& theLISM )
+{
+  HYDROData_NaturalObject::Update();
+  PrsDefinition prsDef;
+  theLISM->GetShapePresentations(prsDef);
+
+  internalUpdatePrs(prsDef);
+}
+
 QColor HYDROData_Stream::DefaultFillingColor() const
 {
   return QColor( Qt::green );
@@ -344,6 +393,12 @@ Handle(HYDROData_DTM) HYDROData_Stream::DTM() const
   return Handle(HYDROData_DTM)::DownCast( GetAltitudeObject() );
 }
 
+Handle(HYDROData_LISM) HYDROData_Stream::LISM() const
+{
+  const_cast<HYDROData_Stream*>( this )->checkAndSetAltitudeObject();
+  return Handle(HYDROData_LISM)::DownCast( GetAltitudeObject() );
+}
+
 double HYDROData_Stream::GetDDZ() const
 {
   return DTM()->GetDDZ();
@@ -354,42 +409,103 @@ void HYDROData_Stream::SetDDZ( double theDDZ )
   DTM()->SetDDZ( theDDZ );
   Changed( Geom_3d );
 }
+
+Handle(HYDROData_PolylineXY) HYDROData_Stream::GetLeftBank() const
+{
+  return LISM()->GetLeftBank();
+}
+
+void HYDROData_Stream::SetLeftBank( const Handle(HYDROData_PolylineXY)& theBank )
+{
+  LISM()->SetLeftBank( theBank );
+  Changed( Geom_3d );
+}
+
+Handle(HYDROData_PolylineXY) HYDROData_Stream::GetRightBank() const
+{
+  return LISM()->GetRightBank();
+}
+
+void HYDROData_Stream::SetRightBank( const Handle(HYDROData_PolylineXY)& theBank )
+{
+  LISM()->SetRightBank( theBank );
+  Changed( Geom_3d );
+}
+  
+  
+double HYDROData_Stream::GetHaxStep() const
+{
+  return LISM()->GetHaxStep();
+}
+
+void HYDROData_Stream::SetHaxStep( double theHaxStep )
+{
+  LISM()->SetHaxStep( theHaxStep );
+  Changed( Geom_3d );
+}
+
+int HYDROData_Stream::GetNbProfilePoints() const
+{
+  return LISM()->GetNbProfilePoints();
+}
+
+void HYDROData_Stream::SetNbProfilePoints( int theNbPoints )
+{
+  LISM()->SetNbProfilePoints( theNbPoints );
+  Changed( Geom_3d );
+}
   
 double HYDROData_Stream::GetSpatialStep() const
 {
-  return DTM()->GetSpatialStep();
+  if (GetInterpolationMethod() == 0)
+    return DTM()->GetSpatialStep();
+  else
+    return LISM()->GetHaxStep();
 }
 
 void HYDROData_Stream::SetSpatialStep( double theSpatialStep )
 {
-  DTM()->SetSpatialStep( theSpatialStep );
+  if (GetInterpolationMethod() == 0 )  
+    DTM()->SetSpatialStep( theSpatialStep );
+  else
+    LISM()->SetHaxStep( theSpatialStep );
   Changed( Geom_3d );
 }
 
 bool HYDROData_Stream::SetHydraulicAxis( const Handle(HYDROData_PolylineXY)& theAxis )
 {
-  if ( !IsValidAsAxis( theAxis ) )
-    return false;
+  if (GetInterpolationMethod() == 0)
+  {
+    if ( !IsValidAsAxis( theAxis ) )
+      return false;
 
-  Handle(HYDROData_PolylineXY) aPrevAxis = GetHydraulicAxis();
-  if ( IsEqual( aPrevAxis, theAxis ) )
-    return true;
+    Handle(HYDROData_PolylineXY) aPrevAxis = GetHydraulicAxis();
+    if ( IsEqual( aPrevAxis, theAxis ) )
+      return true;
 
-  SetReferenceObject( theAxis, DataTag_HydraulicAxis );
+    SetReferenceObject( theAxis, DataTag_HydraulicAxis );
 
-  // Update the order of profiles
-  updateProfilesOrder();
+    // Update the order of profiles
+    updateProfilesOrder();
 
-  // Indicate model of the need to update the stream presentation
-  Changed( Geom_3d );
+    // Indicate model of the need to update the stream presentation
+    Changed( Geom_3d );
+  }
+  else
+  {
+    LISM()->SetHydraulicAxis( theAxis );
+    Changed( Geom_3d );
+  }
 
   return true;
 }
 
 Handle(HYDROData_PolylineXY) HYDROData_Stream::GetHydraulicAxis() const
 {
-  return Handle(HYDROData_PolylineXY)::DownCast( 
-           GetReferenceObject( DataTag_HydraulicAxis ) );
+  if (GetInterpolationMethod() == 0)
+    return Handle(HYDROData_PolylineXY)::DownCast( GetReferenceObject( DataTag_HydraulicAxis ) );
+  else
+    return LISM()->GetHydraulicAxis();
 }
 
 void HYDROData_Stream::RemoveHydraulicAxis()
@@ -561,7 +677,10 @@ bool HYDROData_Stream::AddProfile( const Handle(HYDROData_Profile)& theProfile )
   int aProfileIndex = insertParameter( aPar );
   insertProfileInToOrder( theProfile, aProfileIndex );
 
-  DTM()->SetProfiles( GetProfiles() );
+  if (GetInterpolationMethod()==0)
+    DTM()->SetProfiles( GetProfiles() );
+  else
+    LISM()->SetProfiles( GetProfiles() );
   
   // Indicate model of the need to update the stream presentation
   Changed( Geom_3d );
@@ -616,7 +735,10 @@ bool HYDROData_Stream::SetProfiles( const HYDROData_SequenceOfObjects& theProfil
       Changed( Geom_3d );
   }
 
-  DTM()->SetProfiles( GetProfiles() );
+  if (GetInterpolationMethod()==0)
+    DTM()->SetProfiles( GetProfiles() );
+  else
+    LISM()->SetProfiles( GetProfiles() );
   return true;
 }
 
@@ -673,6 +795,19 @@ void HYDROData_Stream::RemoveProfiles()
   Changed( Geom_3d );
 }
 
+
+int HYDROData_Stream::GetInterpolationMethod() const
+{
+  return GetInteger( DataTag_InterpMethod );
+}
+
+void HYDROData_Stream::SetInterpolationMethod( int theMethod ) //if DTM => 0 ; if LISM => 1
+{
+  SetInteger( DataTag_InterpMethod, theMethod );
+  Changed( Geom_3d );
+}
+
+
 void HYDROData_Stream::insertProfileInToOrder( const Handle(HYDROData_Profile)& theProfile,
                                                const int                        theBeforeIndex )
 {
@@ -776,7 +911,11 @@ void HYDROData_Stream::updateProfilesOrder()
 
 ObjectKind HYDROData_Stream::getAltitudeObjectType() const
 {
-  return KIND_DTM;
+  int InterpMethod = GetInterpolationMethod();
+  if (InterpMethod == 1)
+    return KIND_LISM;
+  else
+    return KIND_DTM;
 }
 
 void HYDROData_Stream::setParametersArray( const QVector<double>& theArray )
@@ -914,7 +1053,7 @@ void HYDROData_Stream::removeParameter( const int& theIndex )
 bool HYDROData_Stream::GenerateBottomPolyline()
 {
   // Get the document
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
+  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document();
   if ( aDocument.IsNull() ) {
     return false;
   }
@@ -977,7 +1116,7 @@ bool HYDROData_Stream::GenerateBottomPolyline()
   
   aProfileUZ->RemoveSection( 0 );
 
-  aProfileUZ->CalculateAndAddPoints(aBottomPoints, aPolylineXY);
+  aProfileUZ->CalculateAndAddPoints(aBottomPoints, aPolylineXY, true);
   
   return true;
 }
@@ -1002,7 +1141,7 @@ bool HYDROData_Stream::SetBottomPolyline( const Handle(HYDROData_Polyline3D)& th
 bool HYDROData_Stream::Interpolate( HYDROData_IProfilesInterpolator* theInterpolator )
 {
   // Get the document
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
+  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document();
   if ( aDocument.IsNull() ) {
     return false;
   }
@@ -1054,7 +1193,7 @@ void HYDROData_Stream::CopyTo( const Handle(HYDROData_Entity)& theDestination,
                                bool isGenerateNewName ) const
 {
   // Get the document
-  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
+  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document();
   if ( aDocument.IsNull() ) {
     return;
   }