X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_Stream.cxx;h=6ffdd0ee43ec1d1110e89b351d5c97bcb11b6e30;hb=c7cf59e092fe050003d1e556715e3ac97acf6bd4;hp=128239eac56a836a0e07c9f6d1a6c71ddda2c72c;hpb=8e79913fc37eacbd1edf9dc08632b6f8bc7250d8;p=modules%2Fhydro.git

diff --git a/src/HYDROData/HYDROData_Stream.cxx b/src/HYDROData/HYDROData_Stream.cxx
index 128239ea..6ffdd0ee 100644
--- a/src/HYDROData/HYDROData_Stream.cxx
+++ b/src/HYDROData/HYDROData_Stream.cxx
@@ -1,12 +1,36 @@
+// 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_Stream.h"
 
 #include "HYDROData_Document.h"
 #include "HYDROData_PolylineXY.h"
+#include "HYDROData_Polyline3D.h"
 #include "HYDROData_Profile.h"
 #include "HYDROData_ShapesGroup.h"
 #include "HYDROData_ShapesTool.h"
 #include "HYDROData_IAltitudeObject.h"
+#include "HYDROData_IProfilesInterpolator.h"
+#include "HYDROData_Tool.h"
 
 #include <TDataStd_RealArray.hxx>
 
@@ -98,6 +122,12 @@ QStringList HYDROData_Stream::DumpToPython( MapOfTreatedObjects& theTreatedObjec
     setPythonReferenceObject( theTreatedObjects, aResList, aProfile, "AddProfile" );
   }
 
+  // Set bottom polyline if exists
+  const Handle(HYDROData_Polyline3D) aBottomPolyline = GetBottomPolyline();
+  if ( !aBottomPolyline.IsNull() ) {
+    setPythonReferenceObject( theTreatedObjects, aResList, aBottomPolyline, "SetBottomPolyline" );
+  }
+
   aResList << QString( "" );
   aResList << QString( "%1.Update();" ).arg( aName );
   aResList << QString( "" );
@@ -143,6 +173,19 @@ Handle(Geom_BSplineCurve) HYDROData_Stream::buildInterpolationCurve(
 void HYDROData_Stream::Update()
 {
   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();
+    }
+  }
+
   UpdatePrs();
 }
 
@@ -152,8 +195,8 @@ bool HYDROData_Stream::IsHas2dPrs() const
 }
 
 bool HYDROData_Stream::CreatePresentations( const Handle(HYDROData_PolylineXY)& theHydAxis,
-                                           const HYDROData_SequenceOfObjects&  theProfiles,
-                                           PrsDefinition&                      thePrs )
+                                            const HYDROData_SequenceOfObjects&  theProfiles,
+                                            PrsDefinition&                      thePrs )
 {
   if ( theHydAxis.IsNull() || theProfiles.Length() < 2 )
     return false;
@@ -175,7 +218,7 @@ bool HYDROData_Stream::CreatePresentations( const Handle(HYDROData_PolylineXY)&
 
     const TopoDS_Shape& aProf3d = aProfile->GetShape3D();
     gp_XY aPnt1, aPnt2;
-    if ( !aProfile->GetLeftPoint( aPnt1 ) || !aProfile->GetRightPoint( aPnt2 ) )
+    if ( !aProfile->GetLeftPoint( aPnt1, false ) || !aProfile->GetRightPoint( aPnt2, false ) )
       continue;
 
     anArrOfProfiles.SetValue(i,aProfile->GetShape3D());//aProfile->GetTopShape();
@@ -361,7 +404,7 @@ void HYDROData_Stream::UpdatePrs()
   anInGroup->AddShape( aResultPrs.myInlet );
 
   QString anOutGroupName = GetName() + "_Outlet";
-
+  
   Handle(HYDROData_ShapesGroup) anOutGroup = createGroupObject();
   anOutGroup->SetName( anOutGroupName );
   anOutGroup->AddShape( aResultPrs.myOutlet );
@@ -976,3 +1019,197 @@ void HYDROData_Stream::removeParameter( const int& theIndex )
 
   setParametersArray( aNewArr );
 }
+
+bool HYDROData_Stream::GenerateBottomPolyline()
+{
+  // Get the document
+  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
+  if ( aDocument.IsNull() ) {
+    return false;
+  }
+
+  // Collect bottom points ( one bottom point from each profile of the stream )
+  HYDROData_Profile::ProfilePoints aBottomPoints;
+  
+  HYDROData_SequenceOfObjects aSeqOfProfiles = GetProfiles();
+  for ( int i = 1, aNb = aSeqOfProfiles.Size(); i <= aNb; i++ ) {
+    const Handle(HYDROData_Profile) aProfile = 
+      Handle(HYDROData_Profile)::DownCast( aSeqOfProfiles.Value( i ) );
+    if ( aProfile.IsNull() ) {
+      continue;
+    }
+    
+    aBottomPoints.Append( aProfile->GetBottomPoint() );
+  }
+
+  int aNbBottomPoints = aBottomPoints.Size();
+
+  if ( aNbBottomPoints < 2 ) {
+    return false;
+  }
+
+  // Create bottom polyline object if the stream doesn't contain it yet
+  Handle(HYDROData_Polyline3D) aBottom = GetBottomPolyline();
+  if ( aBottom.IsNull() ) {
+    aBottom = Handle(HYDROData_Polyline3D)::DownCast( aDocument->CreateObject( KIND_POLYLINE ) );  
+    QString aBaseName = GetName() + "_bottom";
+    QString aName = HYDROData_Tool::GenerateObjectName( aDocument, aBaseName, QStringList(), true );
+    aBottom->SetName( aName );
+
+    SetReferenceObject( aBottom, DataTag_BottomPolyline );
+  }
+  
+  // Create 2D polyline if the bottom polyline doesn't contain it yet
+  Handle(HYDROData_PolylineXY) aPolylineXY = aBottom->GetPolylineXY();
+  if ( aPolylineXY.IsNull() ) {
+    aPolylineXY = Handle(HYDROData_PolylineXY)::DownCast( aDocument->CreateObject( KIND_POLYLINEXY ) );
+    QString aBaseName = GetName() + "_bottom_2d";
+    QString aName = HYDROData_Tool::GenerateObjectName( aDocument, aBaseName, QStringList(), true );
+    aPolylineXY->SetName( aName );
+    aBottom->SetPolylineXY( aPolylineXY, false );
+  }
+
+  aPolylineXY->RemoveSections();
+  aPolylineXY->AddSection( "", HYDROData_PolylineXY::SECTION_SPLINE, false );
+  
+  // Create profile if the bottom polyline doesn't contain it yet
+  Handle(HYDROData_ProfileUZ) aProfileUZ = aBottom->GetProfileUZ();
+  if ( aProfileUZ.IsNull() ) {
+    Handle(HYDROData_Profile) aProfile = 
+      Handle(HYDROData_Profile)::DownCast( aDocument->CreateObject( KIND_PROFILE ) );
+    QString aBaseName = GetName() + "_bottom_profile";
+    QString aName = HYDROData_Tool::GenerateObjectName( aDocument, aBaseName, QStringList(), true );
+    aProfile->SetName( aName );
+    aProfileUZ = aProfile->GetProfileUZ( true );
+    aBottom->SetProfileUZ( aProfileUZ );
+  }
+  
+  aProfileUZ->RemoveSection( 0 );
+
+  // Fill 2D polyline
+  for ( int i = 1; i <= aNbBottomPoints; i++ ) {
+    const HYDROData_Profile::ProfilePoint& aBottomPoint = aBottomPoints.Value( i );
+    aPolylineXY->AddPoint( 0, HYDROData_PolylineXY::Point( aBottomPoint.X(), aBottomPoint.Y() ) );
+  }
+  
+  // Calculate profile UZ points
+
+  // First point
+  const HYDROData_Profile::ProfilePoint& aFirstBottomPoint = aBottomPoints.First();
+  aProfileUZ->AddPoint( 0, HYDROData_ProfileUZ::Point( 0, aFirstBottomPoint.Z() ) );
+
+  // Intermediate points
+  double aPolylineCommonDist = aPolylineXY->GetDistance( 0, aPolylineXY->NbPoints( 0 ) - 1 );
+
+  for ( int i = 2, aNbPoints = aBottomPoints.Size(); i < aNbPoints; i++ ) {
+    const HYDROData_Profile::ProfilePoint& aBottomPoint = aBottomPoints.Value( i );
+    
+    double aDistance = aPolylineXY->GetDistance( 0, i - 1 );
+    
+    Standard_Real anU = aDistance; // = ( aDistance / aPolylineCommonDist ) * aPolylineCommonDist;
+    aProfileUZ->AddPoint( 0, HYDROData_ProfileUZ::Point( anU, aBottomPoint.Z() ) );
+  }
+  
+  // Last point
+  const HYDROData_Profile::ProfilePoint& aLastBottomPoint = aBottomPoints.Last();
+  aProfileUZ->AddPoint( 0, HYDROData_ProfileUZ::Point( aPolylineCommonDist, aLastBottomPoint.Z() ) );
+  
+  return true;
+}
+
+Handle(HYDROData_Polyline3D) HYDROData_Stream::GetBottomPolyline() const
+{
+  return Handle(HYDROData_Polyline3D)::DownCast( 
+           GetReferenceObject( DataTag_BottomPolyline ) );
+}
+
+bool HYDROData_Stream::SetBottomPolyline( const Handle(HYDROData_Polyline3D)& theBottom )
+{
+  if ( theBottom.IsNull() ) {
+    return false;
+  }
+
+  SetReferenceObject( theBottom, DataTag_BottomPolyline );
+
+  return true;
+}
+
+bool HYDROData_Stream::Interpolate( HYDROData_IProfilesInterpolator* theInterpolator )
+{
+  // Get the document
+  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
+  if ( aDocument.IsNull() ) {
+    return false;
+  }
+  
+  if ( theInterpolator->GetCalculatedProfilesNumber() < 1 ) {
+    theInterpolator->Calculate();
+  }
+
+  if ( theInterpolator->GetErrorCode() != OK ) {
+    return false;
+  }
+
+  bool isOK = true;
+
+  for ( int aProfileInd = 0; aProfileInd < theInterpolator->GetCalculatedProfilesNumber(); aProfileInd++ ) {
+    // Get calculated point coordinates
+    HYDROData_Profile::ProfilePoints aResultPoints = theInterpolator->GetResultProfilePoints( aProfileInd );
+    if ( aResultPoints.IsEmpty() ) {
+      isOK = false;
+      continue;
+    }
+        
+    // Create profile object
+    Handle(HYDROData_Profile) aProfile = 
+      Handle(HYDROData_Profile)::DownCast( aDocument->CreateObject( KIND_PROFILE ) );
+    QString aBaseName = GetName() + "_interp_profile";
+    QString aName = HYDROData_Tool::GenerateObjectName( aDocument, aBaseName );
+    aProfile->SetName( aName );
+
+    // Fill the profile with points
+    aProfile->SetProfilePoints( aResultPoints );
+
+    // Add profile to the stream
+    bool isAdded = AddProfile( aProfile );
+    if ( !isAdded ) {
+      aProfile->Remove();
+    }
+    else
+      aProfile->Update();
+  }
+
+  if ( isOK )
+    Update();
+
+  return isOK;
+}
+
+void HYDROData_Stream::CopyTo( const Handle(HYDROData_Entity)& theDestination ) const
+{
+  // Get the document
+  Handle(HYDROData_Document) aDocument = HYDROData_Document::Document( myLab );
+  if ( aDocument.IsNull() ) {
+    return;
+  }
+
+  // Call base method
+  HYDROData_Entity::CopyTo( theDestination );
+
+  Handle(HYDROData_Stream) aStreamCopy = 
+    Handle(HYDROData_Stream)::DownCast( theDestination );
+
+  // Copy bottom polyline if exists
+  if ( !aStreamCopy.IsNull() ) {
+    const Handle(HYDROData_Polyline3D) aBottom = GetBottomPolyline();
+    if ( !aBottom.IsNull() ) {
+      aStreamCopy->ClearReferenceObjects( DataTag_BottomPolyline );
+      aStreamCopy->GenerateBottomPolyline();
+      const Handle(HYDROData_Polyline3D) aBottomCopy = aStreamCopy->GetBottomPolyline();
+      if ( !aBottomCopy.IsNull() && !aBottomCopy->GetPolylineXY().IsNull() ) {
+        aBottomCopy->GetPolylineXY()->Update();
+        aBottomCopy->Update();
+      }
+    }
+  }
+}
\ No newline at end of file