X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_Document.cxx;h=653ddb8e2dd0686962afd0c40b8858d3f85cd8e1;hb=cde2ec476486fcd7c9b7b38ce814c648d73e7fd3;hp=58e4f0e7b46af33b0403ba92b0080fbfca5a2db3;hpb=637f41bd30ce380b1e9c4c100b394be08fdc5ef7;p=modules%2Fhydro.git diff --git a/src/HYDROData/HYDROData_Document.cxx b/src/HYDROData/HYDROData_Document.cxx index 58e4f0e7..653ddb8e 100644 --- a/src/HYDROData/HYDROData_Document.cxx +++ b/src/HYDROData/HYDROData_Document.cxx @@ -1,20 +1,41 @@ +// 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 #include #include #include +#include +#include +#include +#include #include #include #include -#include #include #include #include #include +#include IMPLEMENT_STANDARD_HANDLE(HYDROData_Document,MMgt_TShared) IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Document,MMgt_TShared) @@ -28,6 +49,7 @@ static const int TAG_PROPS_NEW_ID = 1; // general properties: tag for storage of static const int TAG_OBJECTS = 2; // tag of the objects sub-tree static const int TAG_HISTORY = 3; // tag of the history sub-tree (Root for History) static const int TAG_LOCAL_CS = 4; // tag of local coordinate system information +static const int TAG_DEF_STRICKLER_COEFF = 5; // tag of default strickler coefficient static const gp_Pnt2d DEFAULT_LOCAL_CS( 0, 0 ); using namespace std; @@ -184,18 +206,44 @@ void HYDROData_Document::Close() HYDROData_Application::GetApplication()->RemoveDocument(this); } -bool HYDROData_Document::DumpToPython( const QString& theFileName, +double HYDROData_Document::GetDefaultStricklerCoefficient() const +{ + double aRes = 0; + TDF_Label aLabel = myDoc->Main().FindChild(TAG_DEF_STRICKLER_COEFF, Standard_False); + if ( !aLabel.IsNull() ) + { + Handle(TDataStd_Real) anAttr; + if ( aLabel.FindAttribute( TDataStd_Real::GetID(), anAttr ) ) + aRes = anAttr->Get(); + } + + return aRes; +} + +void HYDROData_Document::SetDefaultStricklerCoefficient( double theCoeff ) const +{ + TDF_Label aLabel = myDoc->Main().FindChild(TAG_DEF_STRICKLER_COEFF); + if ( !aLabel.IsNull() ) + { + Handle(TDataStd_Real) anAttr; + if ( !aLabel.FindAttribute( TDataStd_Real::GetID(), anAttr ) ) + aLabel.AddAttribute( anAttr = new TDataStd_Real() ); + anAttr->Set( theCoeff ); + } +} + +bool HYDROData_Document::DumpToPython( const QString& thePyScriptPath, const bool theIsMultiFile ) const { // Try to open the file - QFile aFile( theFileName ); + QFile aFile( thePyScriptPath ); if ( !aFile.open( QIODevice::WriteOnly ) ) return false; MapOfTreatedObjects aTreatedObjects; // Dump header for python script - QStringList aHeaderDump = DumpToPython( aTreatedObjects, theIsMultiFile ); + QStringList aHeaderDump = DumpToPython( thePyScriptPath, aTreatedObjects, theIsMultiFile ); if ( aHeaderDump.isEmpty() ) return false; @@ -203,18 +251,27 @@ 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 ); - aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_BATHYMETRY ); - aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_PROFILE ); - aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_POLYLINE ); - aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_IMMERSIBLE_ZONE ); - aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_STREAM ); - aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_CHANNEL ); - aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_DIGUE ); - aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_OBSTACLE ); - aRes = aRes && dumpPartitionToPython( aFile, theIsMultiFile, aTreatedObjects, KIND_CALCULATION ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_IMAGE ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_STRICKLER_TABLE ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_POLYLINEXY ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_BATHYMETRY ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_PROFILE ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_POLYLINE ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_IMMERSIBLE_ZONE ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_STREAM ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_CHANNEL ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_DIGUE ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_OBSTACLE ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_LAND_COVER_MAP ); + aRes = aRes && dumpPartitionToPython( aFile, thePyScriptPath, theIsMultiFile, aTreatedObjects, KIND_CALCULATION ); // Dump code to close python fuction if ( aRes && theIsMultiFile ) @@ -241,7 +298,8 @@ QString HYDROData_Document::GetDocPyName() const return aDocName; } -QStringList HYDROData_Document::DumpToPython( MapOfTreatedObjects& theTreatedObjects, +QStringList HYDROData_Document::DumpToPython( const QString& thePyScriptPath, + MapOfTreatedObjects& theTreatedObjects, const bool theIsMultiFile ) const { QString aDocName = GetDocPyName(); @@ -261,6 +319,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 ); @@ -275,6 +334,7 @@ QStringList HYDROData_Document::DumpToPython( MapOfTreatedObjects& theTreatedObj } bool HYDROData_Document::dumpPartitionToPython( QFile& theFile, + const QString& thePyScriptPath, const bool theIsMultiFile, MapOfTreatedObjects& theTreatedObjects, const ObjectKind& theObjectKind ) const @@ -299,7 +359,7 @@ bool HYDROData_Document::dumpPartitionToPython( QFile& theFile, theTreatedObjects.insert( anObjName, anObject ); - QStringList anObjDump = anObject->DumpToPython( theTreatedObjects ); + QStringList anObjDump = anObject->DumpToPython( thePyScriptPath, theTreatedObjects ); if ( theIsMultiFile ) { @@ -640,6 +700,8 @@ HYDROData_Document::HYDROData_Document() myTransactionsAfterSave = 0; myLX = -1; myLY = -1; + + myInterpolatorsFactory = 0; } HYDROData_Document::HYDROData_Document(const Handle(TDocStd_Document)& theDoc) @@ -648,6 +710,8 @@ HYDROData_Document::HYDROData_Document(const Handle(TDocStd_Document)& theDoc) myTransactionsAfterSave = 0; myLX = -1; myLY = -1; + + myInterpolatorsFactory = 0; } HYDROData_Document::~HYDROData_Document() @@ -677,7 +741,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 +749,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( 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( this ); - aThat->myLX = aLCS.X(); - aThat->myLY = aLCS.Y(); - } + UpdateLCSFields(); if( IsToLocalCS ) { X -= myLX; @@ -751,3 +838,74 @@ void HYDROData_Document::Transform( gp_XY& thePnt, bool IsToLocalCS ) const 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( this ); + HYDROData_InterpolatorsFactory* aFactory = aThat->GetInterpolatorsFactory(); + if ( aFactory ) { + anInterpolator = aFactory->GetInterpolator( theName ); + } + + return anInterpolator; +} + +NCollection_Sequence HYDROData_Document::GetInterpolatorNames() const +{ + NCollection_Sequence aNames; + + HYDROData_Document* aThat = const_cast( this ); + HYDROData_InterpolatorsFactory* aFactory = aThat->GetInterpolatorsFactory(); + if ( aFactory ) { + aNames = aFactory->GetInterpolatorNames(); + } + + return aNames; +} + +QColor HYDROData_Document::GetAssociatedColor( const QString& theStricklerType, const Handle(HYDROData_StricklerTable)& theTable ) const +{ + if( !theTable.IsNull() && theTable->HasType( theStricklerType ) ) + return theTable->GetColor( theStricklerType ); + + HYDROData_Iterator anIt( this, KIND_STRICKLER_TABLE ); + for( ; anIt.More(); anIt.Next() ) + { + Handle(HYDROData_StricklerTable) aTable = Handle(HYDROData_StricklerTable)::DownCast( anIt.Current() ); + if( aTable->HasType( theStricklerType ) ) + return aTable->GetColor( theStricklerType ); + } + return QColor(); +} + +void HYDROData_Document::CollectQGISValues( const QString& theAttrName, + QStringList& theAttrValues, + QStringList& theStricklerTypes ) const +{ + HYDROData_Iterator It( this, KIND_STRICKLER_TABLE ); + for( ; It.More(); It.Next() ) + { + Handle(HYDROData_StricklerTable) aTable = Handle(HYDROData_StricklerTable)::DownCast( It.Current() ); + if( !aTable.IsNull() && aTable->GetAttrName()==theAttrName ) + { + theAttrValues.clear(); + theStricklerTypes = aTable->GetTypes(); + foreach( QString aType, theStricklerTypes ) + { + QString anAttrValue = aTable->GetAttrValue( aType ); + theAttrValues.append( anAttrValue ); + } + } + } +}