Salome HOME
debug of tests for DTM
[modules/hydro.git] / src / HYDROData / HYDROData_ProfileUZ.cxx
old mode 100755 (executable)
new mode 100644 (file)
index 014288d..bfe75e3
@@ -1,15 +1,39 @@
+// Copyright (C) 2014-2015  EDF-R&D
+// 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_ProfileUZ.h"
 
 #include "HYDROData_Tool.h"
 
 #include <gp_XY.hxx>
+#include <gp_Pnt2d.hxx>
 
 #include <TColStd_ListIteratorOfListOfReal.hxx>
 
+#include <TDataStd_BooleanList.hxx>
+#include <TDataStd_ExtStringList.hxx>
+#include <TDataStd_IntegerList.hxx>
 #include <TDataStd_RealList.hxx>
 
-#include <TopoDS_Wire.hxx>
+#include <TopoDS_Shape.hxx>
+
+#include <HYDROData_Profile.h>
+#include <HYDROData_PolylineXY.h>
 
 
 IMPLEMENT_STANDARD_HANDLE(HYDROData_ProfileUZ, HYDROData_IPolyline)
@@ -24,116 +48,122 @@ HYDROData_ProfileUZ::~HYDROData_ProfileUZ()
 {
 }
 
-CurveCreator_ICurve::ListAISObjects HYDROData_ProfileUZ::constructWire() const
+TopoDS_Shape HYDROData_ProfileUZ::GetShape() const
 {
-  // TODO
-  ListAISObjects aProfileObjects;
-  return aProfileObjects;
+  return TopoDS_Shape();
 }
 
-bool HYDROData_ProfileUZ::clear()
+double HYDROData_ProfileUZ::GetDepthFromDistance( const PointsList& thePoints,
+                                                  const double&     theDistance )
 {
-  removePointsLists();
-  return true;
-}
+  double aResDepth = 0.0;
 
-bool HYDROData_ProfileUZ::join( const int theISectionTo, 
-                                const int theISectionFrom )
-{
-  return false;
+  int aNbPoints = thePoints.Size();
+  if ( aNbPoints < 2 )
+    return aResDepth;
+
+  double aCompDist = 0.0;
+  HYDROData_IPolyline::Point aPrevPoint = thePoints.First();
+  for ( int i = 2; i <= aNbPoints; ++i )
+  {
+    const Point& aCurPoint = thePoints.Value( i );
+
+    double aPntDist = gp_Pnt2d( aPrevPoint.X(), 0 ).Distance( gp_Pnt2d( aCurPoint.X(), 0 ) );
+
+    aCompDist += aPntDist;
+
+    if ( theDistance < aCompDist )
+    {
+      double aComPntDist = gp_Pnt2d( thePoints.First().X(), 0 ).Distance( gp_Pnt2d( aPrevPoint.X(), 0 ) );
+
+      double aFindDist = theDistance - aComPntDist;
+      double aRatio = aFindDist / ( aPntDist - aFindDist );
+
+      aResDepth = ( aPrevPoint.Y() + aRatio * aCurPoint.Y() ) / ( 1 + aRatio );
+      break;
+    }
+    else aResDepth = aCurPoint.Y();  // TODO: workaround for normalized flat altitudes
+
+    aPrevPoint = aCurPoint;
+  }
+
+  return aResDepth;
 }
 
-int HYDROData_ProfileUZ::getNbSections() const
+int HYDROData_ProfileUZ::NbSections() const
 {
   return 1;
 }
 
-int HYDROData_ProfileUZ::addSection( const std::string&              theName, 
-                                     const CurveCreator::SectionType theType,
-                                     const bool                      theIsClosed )
+void HYDROData_ProfileUZ::AddSection( const TCollection_AsciiString& /*theSectName*/,
+                                      const SectionType              /*theSectionType*/,
+                                      const bool                     /*theIsClosed*/ )
 {
-  return 0;
 }
 
-bool HYDROData_ProfileUZ::removeSection( const int theISection )
+TCollection_AsciiString HYDROData_ProfileUZ::GetSectionName( const int /*theSectionIndex*/ ) const
 {
-  return clear();
+  return "Section_1";
 }
 
-bool HYDROData_ProfileUZ::isClosed( const int theISection ) const
+void HYDROData_ProfileUZ::SetSectionName( const int                      /*theSectionIndex*/, 
+                                          const TCollection_AsciiString& /*theSectionName*/ )
 {
-  return false;
 }
 
-bool HYDROData_ProfileUZ::setClosed( const int  theISection, 
-                                     const bool theIsClosed )
+HYDROData_ProfileUZ::SectionType HYDROData_ProfileUZ::GetSectionType( const int /*theSectionIndex*/ ) const
 {
-  return false;
+  Handle(TDataStd_ExtStringList) aNamesList;
+  Handle(TDataStd_IntegerList) aTypesList;
+  Handle(TDataStd_BooleanList) aClosuresList;
+  getSectionsLists( aNamesList, aTypesList, aClosuresList, false );
+  if ( aTypesList.IsNull() || aTypesList->IsEmpty() )
+    return SECTION_POLYLINE;
+
+  return (SectionType)aTypesList->First();
 }
 
-std::string HYDROData_ProfileUZ::getSectionName( const int theISection ) const
+void HYDROData_ProfileUZ::SetSectionType( const int         /*theSectionIndex*/, 
+                                          const SectionType theSectionType )
 {
-  return "Section_1";
+  Handle(TDataStd_ExtStringList) aNamesList;
+  Handle(TDataStd_IntegerList) aTypesList;
+  Handle(TDataStd_BooleanList) aClosuresList;
+  getSectionsLists( aNamesList, aTypesList, aClosuresList );
+  if ( aTypesList.IsNull()  )
+    return;
+
+  // Refill the existing list
+  aTypesList->Clear();
+  aTypesList->Append( theSectionType );
 }
 
-bool HYDROData_ProfileUZ::setSectionName( const int          theISection, 
-                                          const std::string& theName )
+bool HYDROData_ProfileUZ::IsClosedSection( const int /*theSectionIndex*/ ) const
 {
   return false;
 }
 
-CurveCreator::SectionType HYDROData_ProfileUZ::getSectionType( const int theISection ) const
+void HYDROData_ProfileUZ::SetSectionClosed( const int  /*theSectionIndex*/, 
+                                            const bool /*theIsClosed*/ )
 {
-  return CurveCreator::Polyline;
 }
 
-bool HYDROData_ProfileUZ::setSectionType( const int                       theISection, 
-                                          const CurveCreator::SectionType theType )
+void HYDROData_ProfileUZ::RemoveSection( const int /*theSectionIndex*/ )
 {
-  return false;
+  RemoveSections();
 }
 
-bool HYDROData_ProfileUZ::addPoints( const CurveCreator::Coordinates& theCoords,
-                                     const int                        theISection,
-                                     const int                        theIPnt )
+void HYDROData_ProfileUZ::RemoveSections()
 {
-  bool anIsOperation = myIsOperation;
-  if ( !anIsOperation )
-    startOperation();
-
-  bool aRes = true;
-
-  int anAfterPnt = theIPnt;
-
-  CurveCreator::Coordinates::const_iterator aBegIter = theCoords.begin();
-  CurveCreator::Coordinates::const_iterator anEndIter = theCoords.end();
-  while ( aBegIter != anEndIter )
-  {
-    const CurveCreator::TypeCoord& aCoordX = *aBegIter++;
-    if ( aBegIter == anEndIter )
-      break;
-
-    const CurveCreator::TypeCoord& aCoordY = *aBegIter++;
-
-    aRes = addPoint( aCoordX, aCoordY, theISection, anAfterPnt ) && aRes;
-    
-    if ( anAfterPnt != -1 )
-      ++anAfterPnt;
-  }
-
-  if ( !anIsOperation )
-    commitOperation();
-
-  return aRes;
+  removePointsLists( 0 );
 }
 
-bool HYDROData_ProfileUZ::addPoint( const CurveCreator::TypeCoord& theCoordX,
-                                    const CurveCreator::TypeCoord& theCoordY,
-                                    const int                      theISection,
-                                    const int                      theIPnt )
+void HYDROData_ProfileUZ::AddPoint( const int    /*theSectionIndex*/,
+                                    const Point& thePoint,
+                                    const int    thePointIndex )
 {
-  CurveCreator::TypeCoord aNewCoordU = theCoordX;
-  CurveCreator::TypeCoord aNewCoordZ = theCoordY;
+  double aNewCoordU = thePoint.X();
+  double aNewCoordZ = thePoint.Y();
 
   Handle(TDataStd_RealList) aListU, aListZ;
   getPointsLists( 0, aListU, aListZ );
@@ -142,69 +172,69 @@ bool HYDROData_ProfileUZ::addPoint( const CurveCreator::TypeCoord& theCoordX,
   {
     aListU->Append( aNewCoordU );
     aListZ->Append( aNewCoordZ );
+    return;
   }
   else if ( aNewCoordU < aListU->First() )
   {
     aListU->Prepend( aNewCoordU );
     aListZ->Prepend( aNewCoordZ );
+    return;
   }
-  else 
-  {
-    TColStd_ListOfReal anOldListU;
-    anOldListU = aListU->List();
 
-    TColStd_ListOfReal anOldListZ;
-    anOldListZ = aListZ->List();
+  TColStd_ListOfReal anOldListU;
+  anOldListU = aListU->List();
+
+  TColStd_ListOfReal anOldListZ;
+  anOldListZ = aListZ->List();
 
-    // Refill the existing lists
-    aListU->Clear();
-    aListZ->Clear();
+  // Crsat new lists
+  removePointsLists( 0 );
+  getPointsLists( 0, aListU, aListZ );
 
-    bool anIsInserted = false;
-    TColStd_ListIteratorOfListOfReal anIterU( anOldListU );
-    TColStd_ListIteratorOfListOfReal anIterZ( anOldListZ );
-    for ( ; anIterU.More() && anIterZ.More(); anIterU.Next(), anIterZ.Next() )
-    {
-      double aCoordU = anIterU.Value();
-      double aCoordZ = anIterZ.Value();
+  bool anIsInserted = false;
+  TColStd_ListIteratorOfListOfReal anIterU( anOldListU );
+  TColStd_ListIteratorOfListOfReal anIterZ( anOldListZ );
+  for ( ; anIterU.More() && anIterZ.More(); anIterU.Next(), anIterZ.Next() )
+  {
+    double aCoordU = anIterU.Value();
+    double aCoordZ = anIterZ.Value();
 
-      if ( !anIsInserted )
+    if ( !anIsInserted )
+    {
+      if ( ValuesEquals( aNewCoordU, aCoordU ) )
       {
-        if ( ValuesEquals( aNewCoordU, aCoordU ) )
-        {
-          // Just update Z value
-          aCoordZ = aNewCoordZ;
-          anIsInserted = true;
-        }
-        else if ( aNewCoordU < aCoordU )
-        {
-          // Insert new point
-          aListU->Append( aNewCoordU );
-          aListZ->Append( aNewCoordZ );
-          anIsInserted = true;
-        }
+        // Just update Z value
+        aCoordZ = aNewCoordZ;
+        anIsInserted = true;
+      }
+      else if ( aNewCoordU < aCoordU )
+      {
+        // Insert new point
+        aListU->Append( aNewCoordU );
+        aListZ->Append( aNewCoordZ );
+        anIsInserted = true;
       }
-
-      aListU->Append( aCoordU );
-      aListZ->Append( aCoordZ );
     }
+
+    aListU->Append( aCoordU );
+    aListZ->Append( aCoordZ );
   }
+}
 
-  return true;
+void HYDROData_ProfileUZ::SetPoint( const int    theSectionIndex,
+                                    const Point& thePoint,
+                                    const int    /*thePointIndex*/ )
+{
+  AddPoint( theSectionIndex, thePoint );
 }
 
-bool HYDROData_ProfileUZ::removePoint( const int /*theISection*/,
-                                       const int theIPnt )
+void HYDROData_ProfileUZ::RemovePoint( const int /*theSectionIndex*/,
+                                       const int thePointIndex )
 {
   Handle(TDataStd_RealList) aListU, aListZ;
   getPointsLists( 0, aListU, aListZ, false );
-  if ( aListU.IsNull() || aListZ.IsNull() || 
-       theIPnt < 0 || theIPnt >= aListU->Extent() )
-    return false;
-
-  bool anIsOperation = myIsOperation;
-  if ( !anIsOperation )
-    startOperation();
+  if ( aListU.IsNull() || aListZ.IsNull() || aListU->IsEmpty() )
+    return;
 
   TColStd_ListOfReal anOldListU;
   anOldListU = aListU->List();
@@ -212,62 +242,26 @@ bool HYDROData_ProfileUZ::removePoint( const int /*theISection*/,
   TColStd_ListOfReal anOldListZ;
   anOldListZ = aListZ->List();
 
-  // Refill the existing lists
-  aListU->Clear();
-  aListZ->Clear();
+  // Creat new lists
+  removePointsLists( 0 );
+  getPointsLists( 0, aListU, aListZ );
 
+  bool anIsInserted = false;
   TColStd_ListIteratorOfListOfReal anIterU( anOldListU );
   TColStd_ListIteratorOfListOfReal anIterZ( anOldListZ );
   for ( int i = 0; anIterU.More() && anIterZ.More(); anIterU.Next(), anIterZ.Next(), ++i )
   {
-    if ( i == theIPnt )
+    if ( i == thePointIndex )
       continue; // skip index to remove
 
     aListU->Append( anIterU.Value() );
     aListZ->Append( anIterZ.Value() );
   }
-
-  if ( !anIsOperation )
-    commitOperation();
-
-  return true;
-}
-
-//! Remove several points from different sections with given ids
-bool HYDROData_ProfileUZ::removeSeveralPoints( const SectionToPointList &theSectionToPntIDs)
-{
-  return false;
 }
 
-bool HYDROData_ProfileUZ::setPoint( const int                        /*theISection*/,
-                                    const int                        theIPoint,
-                                    const CurveCreator::Coordinates& theNewCoords )
+HYDROData_ProfileUZ::PointsList HYDROData_ProfileUZ::GetPoints( const int /*theSectionIndex*/, bool /*IsConvertToGlobal*/ ) const
 {
-  bool anIsOperation = myIsOperation;
-  if ( !anIsOperation )
-    startOperation();
-
-  // At first we remove point
-  removePoint( 0, theIPoint );
-
-  // And then we insert it to correct place
-  bool aRes = addPoints( theNewCoords, 0 );
-
-  if ( !anIsOperation )
-    commitOperation();
-
-  return aRes;
-}
-
-//! Set coordinates of specified points from different sections
-bool HYDROData_ProfileUZ::setSeveralPoints( const SectionToPointCoordsList &theSectionToPntCoords)
-{
-  return false;
-}
-
-CurveCreator::Coordinates HYDROData_ProfileUZ::getPoints( const int /*theISection*/ ) const
-{
-  CurveCreator::Coordinates aResList;
+  PointsList aResList;
 
   Handle(TDataStd_RealList) aListU, aListZ;
   getPointsLists( 0, aListU, aListZ, false );
@@ -278,14 +272,43 @@ CurveCreator::Coordinates HYDROData_ProfileUZ::getPoints( const int /*theISectio
   TColStd_ListIteratorOfListOfReal anIterZ( aListZ->List() );
   for ( ; anIterU.More() && anIterZ.More(); anIterU.Next(), anIterZ.Next() )
   {
-    const double& aCoordU = anIterU.Value();
-    const double& aCoordZ = anIterZ.Value();
-
-    aResList.push_back( aCoordU );
-    aResList.push_back( aCoordZ );
+    Point aPoint( anIterU.Value(), anIterZ.Value() );
+    aResList.Append( aPoint );
   }
 
   return aResList;
 }
 
+void HYDROData_ProfileUZ::CalculateAndAddPoints(const NCollection_Sequence<gp_XYZ>& theXYZPoints, Handle_HYDROData_PolylineXY& thePolylineXY)
+{
+   // Fill 2D polyline
+  for ( int i = 1; i <= theXYZPoints.Size(); i++ ) {
+    const HYDROData_Profile::ProfilePoint& aBottomPoint = theXYZPoints.Value( i );
+    thePolylineXY->AddPoint( 0, HYDROData_PolylineXY::Point( aBottomPoint.X(), aBottomPoint.Y() ) );
+  }
+  
+  // Calculate profile UZ points
+
+  // First point
+  const HYDROData_Profile::ProfilePoint& aFirstBottomPoint = theXYZPoints.First();
+  AddPoint( 0, HYDROData_ProfileUZ::Point( 0, aFirstBottomPoint.Z() ) );
+
+  // Intermediate points
+  double aPolylineCommonDist = thePolylineXY->GetDistance( 0, thePolylineXY->NbPoints( 0 ) - 1 );
+
+  for ( int i = 2, aNbPoints = theXYZPoints.Size(); i < aNbPoints; i++ ) {
+    const HYDROData_Profile::ProfilePoint& aBottomPoint = theXYZPoints.Value( i );
+    
+    double aDistance = thePolylineXY->GetDistance( 0, i - 1 );
+    
+    Standard_Real anU = aDistance; // = ( aDistance / aPolylineCommonDist ) * aPolylineCommonDist;
+    AddPoint( 0, HYDROData_ProfileUZ::Point( anU, aBottomPoint.Z() ) );
+  }
+  
+  // Last point
+  const HYDROData_Profile::ProfilePoint& aLastBottomPoint = theXYZPoints.Last();
+  AddPoint( 0, HYDROData_ProfileUZ::Point( aPolylineCommonDist, aLastBottomPoint.Z() ) );
+}
+