X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_Document.cxx;h=56c6590f3c34cdaaf8d73e40155d52ae452cad72;hb=05d82ada59110252be57aaa774095a18ebdca455;hp=cf515aa8d8c22d73caa6bc82d893f99019bfba52;hpb=2cf9e838e652ae811d9b3931914771ad1836c6b8;p=modules%2Fhydro.git diff --git a/src/HYDROData/HYDROData_Document.cxx b/src/HYDROData/HYDROData_Document.cxx index cf515aa8..56c6590f 100644 --- a/src/HYDROData/HYDROData_Document.cxx +++ b/src/HYDROData/HYDROData_Document.cxx @@ -21,7 +21,10 @@ #include #include #include +#include +#include +#include #include #include @@ -32,8 +35,8 @@ #include #include #include +#include -IMPLEMENT_STANDARD_HANDLE(HYDROData_Document,MMgt_TShared) IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Document,MMgt_TShared) #define PYTHON_DOC_NAME "hydro_doc" @@ -45,12 +48,13 @@ 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; -typedef QMap MapOfOrdered; -typedef QMap MapOfUnordered; +typedef QMap MapOfOrdered; +typedef QMap MapOfUnordered; Handle(HYDROData_Document) HYDROData_Document::Document(const int theStudyID) { @@ -201,18 +205,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; @@ -222,23 +252,25 @@ bool HYDROData_Document::DumpToPython( const QString& theFileName, // Dump the local CS data to Python UpdateLCSFields(); - QString aLCS = QString( "%1.SetLocalCS( %2, %3 )" ).arg( GetDocPyName() ).arg( myLX ).arg( myLY ); + QString aLCS = QString( "%1.SetLocalCS( %2, %3 )" ).arg( GetDocPyName() ).arg( myLX, 0, 'f', 3 ).arg( myLY, 0, 'f', 3 ); 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 ) @@ -265,7 +297,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(); @@ -288,18 +321,19 @@ QStringList HYDROData_Document::DumpToPython( MapOfTreatedObjects& theTreatedObj aResScript << QString( "import salome" ); aResScript << QString( "" ); aResScript << QString( "def RebuildData( theStudy ):" ); - aResScript << QString( " %1 = HYDROData_Document.Document( theStudy._get_StudyId() );" ).arg( aDocName ); + aResScript << QString( " %1 = HYDROData_Document.Document( theStudy._get_StudyId() )" ).arg( aDocName ); } else { aResScript << QString( "" ); - aResScript << QString( "%1 = HYDROData_Document.Document( theStudy._get_StudyId() );" ).arg( aDocName ); + aResScript << QString( "%1 = HYDROData_Document.Document( theStudy._get_StudyId() )" ).arg( aDocName ); } return aResScript; } bool HYDROData_Document::dumpPartitionToPython( QFile& theFile, + const QString& thePyScriptPath, const bool theIsMultiFile, MapOfTreatedObjects& theTreatedObjects, const ObjectKind& theObjectKind ) const @@ -324,7 +358,7 @@ bool HYDROData_Document::dumpPartitionToPython( QFile& theFile, theTreatedObjects.insert( anObjName, anObject ); - QStringList anObjDump = anObject->DumpToPython( theTreatedObjects ); + QStringList anObjDump = anObject->DumpToPython( thePyScriptPath, theTreatedObjects ); if ( theIsMultiFile ) { @@ -448,7 +482,7 @@ void HYDROData_Document::SetObjectsLayerOrder( const HYDROData_SequenceOfObjects } } -void HYDROData_Document::Show( const Handle_HYDROData_Entity& theObject ) +void HYDROData_Document::Show( const Handle(HYDROData_Entity)& theObject ) { HYDROData_SequenceOfObjects anOrder; anOrder.Append( theObject ); @@ -837,4 +871,40 @@ NCollection_Sequence HYDROData_Document::GetInterpolato } return aNames; -} \ No newline at end of file +} + +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 ); + } + } + } +}