IMPLEMENT_STANDARD_HANDLE(HYDROData_Document,MMgt_TShared)
IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Document,MMgt_TShared)
-#define PYTHON_DOC_NAME "doc"
+#define PYTHON_DOC_NAME "hydro_doc"
static const int UNDO_LIMIT = 10; // number of possible undo operations in the module
anError = DocError_ResourcesProblem;
break;
case PCDM_SS_WriteFailure:
- case PCDM_SS_DiskWritingFailure:
- case PCDM_SS_UserRightsFailure:
+ //case PCDM_SS_DiskWritingFailure:
+ //case PCDM_SS_UserRightsFailure:
anError = DocError_CanNotOpen;
break;
default:
HYDROData_Application::GetApplication()->RemoveDocument(this);
}
-bool HYDROData_Document::DumpToPython( const QString& theFileName ) const
+bool HYDROData_Document::DumpToPython( const QString& theFileName,
+ const bool theIsMultiFile ) const
{
// Try to open the file
QFile aFile( theFileName );
MapOfTreatedObjects aTreatedObjects;
// Dump header for python script
- QStringList aHeaderDump = DumpToPython( aTreatedObjects );
+ QStringList aHeaderDump = DumpToPython( aTreatedObjects, theIsMultiFile );
if ( aHeaderDump.isEmpty() )
return false;
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_ALTITUDE );
- aRes = aRes && dumpPartitionToPython( aFile, aTreatedObjects, KIND_IMMERSIBLE_ZONE );
- aRes = aRes && dumpPartitionToPython( aFile, aTreatedObjects, KIND_CALCULATION );
+ 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 );
+
+ // Dump code to close python fuction
+ if ( aRes && theIsMultiFile )
+ {
+ QStringList aFooterScript;
+ aFooterScript << QString( "" );
+ aFooterScript << QString( " pass" );
+ HYDROData_Tool::WriteStringsToFile( aFile, aFooterScript );
+ }
return aRes;
}
{
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
+QStringList HYDROData_Document::DumpToPython( MapOfTreatedObjects& theTreatedObjects,
+ const bool theIsMultiFile ) const
{
QString aDocName = GetDocPyName();
QStringList aResScript;
- aResScript << QString( "from HYDROData import *" );
+ aResScript << QString( "from HYDROPy 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 );
+
+ if ( theIsMultiFile )
+ {
+ aResScript << QString( "" );
+ aResScript << QString( "def RebuildData( theStudy ):" );
+ 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 );
+ }
return aResScript;
}
bool HYDROData_Document::dumpPartitionToPython( QFile& theFile,
+ const bool theIsMultiFile,
MapOfTreatedObjects& theTreatedObjects,
const ObjectKind& theObjectKind ) const
{
theTreatedObjects.insert( anObjName, anObject );
QStringList anObjDump = anObject->DumpToPython( theTreatedObjects );
+
+ if ( theIsMultiFile )
+ {
+ // For multifile dump we use the function, see the document dump header
+ QStringList::iterator anIt = anObjDump.begin();
+ for ( ; anIt != anObjDump.end(); ++anIt )
+ anIt->prepend( " " );
+ }
HYDROData_Tool::WriteStringsToFile( theFile, anObjDump );
}
myTransactionsAfterSave++;
}
-Handle(HYDROData_Entity) HYDROData_Document::CreateObject(const ObjectKind theKind)
+Handle(HYDROData_Entity) HYDROData_Document::CreateObject( const ObjectKind theKind )
+{
+ return HYDROData_Iterator::CreateObject( this, theKind );
+}
+
+Handle(HYDROData_Entity) HYDROData_Document::FindObjectByName(
+ const QString& theName,
+ const ObjectKind theObjectKind ) const
+{
+ Handle(HYDROData_Entity) anObject;
+ if ( theName.isEmpty() )
+ return anObject;
+
+ QStringList aNamesList;
+ aNamesList << theName;
+
+ HYDROData_SequenceOfObjects aSeqOfObjs = FindObjectsByNames( aNamesList, theObjectKind );
+ if( aSeqOfObjs.IsEmpty() )
+ return anObject;
+
+ anObject = aSeqOfObjs.First();
+ return anObject;
+}
+
+HYDROData_SequenceOfObjects HYDROData_Document::FindObjectsByNames(
+ const QStringList& theNames,
+ const ObjectKind theObjectKind ) const
{
- return HYDROData_Iterator::CreateObject(this, theKind);
+ HYDROData_SequenceOfObjects aResSeq;
+
+ QStringList aNamesList = theNames;
+
+ HYDROData_Iterator anIter( this, theObjectKind );
+ for( ; anIter.More(); anIter.Next() )
+ {
+ Handle(HYDROData_Entity) anObject = anIter.Current();
+ if( anObject.IsNull() )
+ continue;
+
+ QString anObjName = anObject->GetName();
+ if ( anObjName.isEmpty() || !aNamesList.contains( anObjName ) )
+ continue;
+
+ aResSeq.Append( anObject );
+
+ aNamesList.removeAll( anObjName );
+ if ( aNamesList.isEmpty() )
+ break;
+ }
+
+ return aResSeq;
}
HYDROData_Document::HYDROData_Document()