+bool HYDROData_Document::DumpToPython( const QString& theFileName ) const
+{
+ // Try to open the file
+ QFile aFile( theFileName );
+ if ( !aFile.open( QIODevice::WriteOnly ) )
+ return false;
+
+ MapOfTreatedObjects aTreatedObjects;
+
+ // Dump header for python script
+ QStringList aHeaderDump = DumpToPython( aTreatedObjects );
+ if ( aHeaderDump.isEmpty() )
+ return false;
+
+ HYDROData_Tool::WriteStringsToFile( aFile, aHeaderDump );
+
+ bool aRes = true;
+
+ // Dump all model objects to Python script
+ aRes = aRes && dumpPartitionToPython( aFile, aTreatedObjects, KIND_IMAGE );
+ aRes = aRes && dumpPartitionToPython( aFile, aTreatedObjects, KIND_POLYLINE );
+ aRes = aRes && dumpPartitionToPython( aFile, aTreatedObjects, KIND_BATHYMETRY );
+ aRes = aRes && dumpPartitionToPython( aFile, aTreatedObjects, KIND_IMMERSIBLE_ZONE );
+ aRes = aRes && dumpPartitionToPython( aFile, aTreatedObjects, KIND_CALCULATION );
+
+ return aRes;
+}
+
+QString HYDROData_Document::GetDocPyName() const
+{
+ QString aDocName = PYTHON_DOC_NAME;
+
+ int aDocId = 1;
+ if ( DocumentId( this, aDocId ) )
+ aDocName += "_" + QString::number( aDocId );
+
+ return aDocName;
+}
+
+QStringList HYDROData_Document::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
+{
+ QString aDocName = GetDocPyName();
+
+ // Append document in to the map of treated objects to prevent names overlaping
+ theTreatedObjects.insert( aDocName, this );
+
+ int aDocId = 1;
+ if ( !DocumentId( this, aDocId ) )
+ aDocId = 1;
+
+ QStringList aResScript;
+
+ aResScript << QString( "from HYDROData import *" );
+ aResScript << QString( "from PyQt4.QtCore import *" );
+ aResScript << QString( "from PyQt4.QtGui import *" );
+ aResScript << QString( "" );
+ aResScript << QString( "%1 = HYDROData_Document.Document( %2 );" ).arg( aDocName ).arg( aDocId );
+
+ return aResScript;
+}
+
+bool HYDROData_Document::dumpPartitionToPython( QFile& theFile,
+ MapOfTreatedObjects& theTreatedObjects,
+ const ObjectKind& theObjectKind ) const
+{
+ if ( !theFile.isOpen() )
+ return false;
+
+ QTextStream anOutStream( &theFile );
+
+ bool aRes = true;
+
+ HYDROData_Iterator anIterator( this, theObjectKind );
+ for( ; anIterator.More(); anIterator.Next() )
+ {
+ Handle(HYDROData_Entity) anObject = anIterator.Current();
+ if ( anObject.IsNull() )
+ continue;
+
+ QString anObjName = anObject->GetName();
+ if ( theTreatedObjects.contains( anObjName ) )
+ continue;
+
+ theTreatedObjects.insert( anObjName, anObject );
+
+ QStringList anObjDump = anObject->DumpToPython( theTreatedObjects );
+
+ HYDROData_Tool::WriteStringsToFile( theFile, anObjDump );
+ }
+
+ return aRes;
+}
+