Salome HOME
#refs 522 (very draft) //import only
[modules/hydro.git] / src / HYDROData / HYDROData_Document.cxx
index 57cb05992d47bdb70a19ca83417f4d9ff00cbec9..7c9773ba065bbda794f090dc6bd689c2ba954a04 100644 (file)
@@ -1,15 +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_Document.h>
 #include <HYDROData_Application.h>
 #include <HYDROData_Iterator.h>
 #include <HYDROData_Tool.h>
+#include <HYDROData_InterpolatorsFactory.h>
 
 #include <TDataStd_Integer.hxx>
 #include <TDataXtd_Position.hxx>
 
 #include <TDF_Delta.hxx>
 
-#include <gp_Pnt2d.hxx>
 #include <gp_Pnt.hxx>
 
 #include <QFile>
@@ -203,6 +224,13 @@ bool HYDROData_Document::DumpToPython( const QString& theFileName,
 
   bool aRes = true;
 
+  // Dump the local CS data to Python 
+  UpdateLCSFields();
+  QString aLCS = QString( "%1.SetLocalCS( %2, %3 )" ).arg( GetDocPyName() ).arg( myLX ).arg( myLY );
+  if( theIsMultiFile )
+    aLCS.prepend( "  " );
+  HYDROData_Tool::WriteStringsToFile( aFile, QStringList() << aLCS );
+
   // Dump all model objects to Python script
   aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_IMAGE );
   aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_POLYLINEXY );
@@ -261,6 +289,7 @@ QStringList HYDROData_Document::DumpToPython( MapOfTreatedObjects& theTreatedObj
 
   if ( theIsMultiFile )
   {
+    aResScript << QString( "import salome" );
     aResScript << QString( "" );
     aResScript << QString( "def RebuildData( theStudy ):" );
     aResScript << QString( "  %1 = HYDROData_Document.Document( theStudy._get_StudyId() );" ).arg( aDocName );
@@ -640,6 +669,8 @@ HYDROData_Document::HYDROData_Document()
   myTransactionsAfterSave = 0;
   myLX = -1;
   myLY = -1;
+
+  myInterpolatorsFactory = 0;
 }
 
 HYDROData_Document::HYDROData_Document(const Handle(TDocStd_Document)& theDoc)
@@ -648,6 +679,8 @@ HYDROData_Document::HYDROData_Document(const Handle(TDocStd_Document)& theDoc)
   myTransactionsAfterSave = 0;
   myLX = -1;
   myLY = -1;
+
+  myInterpolatorsFactory = 0;
 }
 
 HYDROData_Document::~HYDROData_Document()
@@ -677,7 +710,7 @@ TDF_Label HYDROData_Document::LabelOfLocalCS() const
   return myDoc->Main().FindChild(TAG_LOCAL_CS);
 }
 
-gp_Pnt2d HYDROData_Document::GetLocalCS() const
+void HYDROData_Document::GetLocalCS( double& theLX, double& theLY ) const
 {
   TDF_Label aLocalCSLab = LabelOfLocalCS();
 
@@ -685,35 +718,58 @@ gp_Pnt2d HYDROData_Document::GetLocalCS() const
   if( aLocalCSLab.FindAttribute( TDataXtd_Position::GetID(), aLocalCS ) )
   {
     gp_Pnt aLocalCS3d = aLocalCS->GetPosition();
-    return gp_Pnt2d( aLocalCS3d.X(), aLocalCS3d.Y() );
+    theLX = aLocalCS3d.X();
+    theLY = aLocalCS3d.Y();
   }
   else
-    return DEFAULT_LOCAL_CS;
+  {
+    theLX = DEFAULT_LOCAL_CS.X();
+    theLY = DEFAULT_LOCAL_CS.Y();
+  }
 }
 
-void HYDROData_Document::SetLocalCS( const gp_Pnt2d& theLocalCS )
+void HYDROData_Document::SetLocalCS( double theLX, double theLY )
 {
-  TDF_Label aLocalCSLab = LabelOfLocalCS();
+  UpdateLCSFields();
 
+  // update the local CS data in attribute
+  TDF_Label aLocalCSLab = LabelOfLocalCS();
   Handle( TDataXtd_Position ) aLocalCS;
   if( !aLocalCSLab.FindAttribute( TDataXtd_Position::GetID(), aLocalCS ) )
     aLocalCS = TDataXtd_Position::Set( aLocalCSLab );
 
-  gp_Pnt aLocalCS3d( theLocalCS.X(), theLocalCS.Y(), 0 );
+  gp_Pnt aLocalCS3d( theLX, theLY, 0 );
   aLocalCS->SetPosition( aLocalCS3d );
-  myLX = theLocalCS.X();
-  myLY = theLocalCS.Y();
+
+  // calculate delta for coordinates
+  double aDX = myLX - theLX;
+  double aDY = myLY - theLY;
+
+  // update the local CS data in internal fields
+  myLX = theLX;
+  myLY = theLY;
+
+  //update all objects in the document
+  HYDROData_Iterator anIterator( this, KIND_UNKNOWN );
+  for( ; anIterator.More(); anIterator.Next() )
+    anIterator.Current()->UpdateLocalCS( aDX, aDY );
+}
+
+void HYDROData_Document::UpdateLCSFields() const
+{
+  if( myLX >= 0 && myLY >= 0 )
+    return;
+
+  double aLX, aLY;
+  GetLocalCS( aLX, aLY );
+  HYDROData_Document* aThat = const_cast<HYDROData_Document*>( this );
+  aThat->myLX = aLX;
+  aThat->myLY = aLY;
 }
 
 void HYDROData_Document::Transform( double& X, double& Y, bool IsToLocalCS ) const
 {
-  if( myLX < 0 || myLY < 0 )
-  {
-    gp_Pnt2d aLCS = GetLocalCS();
-    HYDROData_Document* aThat = const_cast<HYDROData_Document*>( this );
-    aThat->myLX = aLCS.X();
-    aThat->myLY = aLCS.Y();
-  }
+  UpdateLCSFields();
   if( IsToLocalCS )
   {
     X -= myLX;
@@ -743,3 +799,46 @@ void HYDROData_Document::Transform( gp_XYZ& thePnt, bool IsToLocalCS ) const
   Transform( X, Y, IsToLocalCS );
   thePnt = gp_XYZ( X, Y, Z ); 
 }
+
+void HYDROData_Document::Transform( gp_XY& thePnt, bool IsToLocalCS ) const
+{
+  double X = thePnt.X();
+  double Y = thePnt.Y();
+  Transform( X, Y, IsToLocalCS );
+  thePnt = gp_XY( X, Y ); 
+}
+
+HYDROData_InterpolatorsFactory* HYDROData_Document::GetInterpolatorsFactory()
+{
+  if ( !myInterpolatorsFactory ) {
+    myInterpolatorsFactory = new HYDROData_InterpolatorsFactory();
+  }
+
+  return myInterpolatorsFactory;
+}
+
+HYDROData_IProfilesInterpolator* HYDROData_Document::GetInterpolator( const TCollection_AsciiString& theName ) const
+{
+  HYDROData_IProfilesInterpolator* anInterpolator = NULL;
+
+  HYDROData_Document* aThat = const_cast<HYDROData_Document*>( this );
+  HYDROData_InterpolatorsFactory* aFactory = aThat->GetInterpolatorsFactory();
+  if ( aFactory ) {
+    anInterpolator = aFactory->GetInterpolator( theName );
+  }
+
+  return anInterpolator;
+}
+NCollection_Sequence<TCollection_AsciiString> HYDROData_Document::GetInterpolatorNames() const
+{
+  NCollection_Sequence<TCollection_AsciiString> aNames;
+
+  HYDROData_Document* aThat = const_cast<HYDROData_Document*>( this );
+  HYDROData_InterpolatorsFactory* aFactory = aThat->GetInterpolatorsFactory();
+  if ( aFactory ) {
+    aNames = aFactory->GetInterpolatorNames();
+  }
+
+  return aNames;
+}
\ No newline at end of file