+
#include <HYDROData_Document.h>
#include <HYDROData_Application.h>
#include <HYDROData_Iterator.h>
+#include <HYDROData_Tool.h>
#include <TDataStd_Integer.hxx>
#include <TDF_Delta.hxx>
+#include <QFile>
+#include <QStringList>
+#include <QTextStream>
+
IMPLEMENT_STANDARD_HANDLE(HYDROData_Document,MMgt_TShared)
IMPLEMENT_STANDARD_RTTIEXT(HYDROData_Document,MMgt_TShared)
+#define PYTHON_DOC_NAME "hydro_doc"
+
static const int UNDO_LIMIT = 10; // number of possible undo operations in the module
static const int TAG_PROPS = 1; // general properties tag
return aResult;
}
+Handle(HYDROData_Document) HYDROData_Document::Document(
+ const TDF_Label& theObjectLabel )
+{
+ Handle(HYDROData_Document) aResDoc;
+ if ( theObjectLabel.IsNull() )
+ return aResDoc;
+
+ Handle(TDocStd_Document) anObjDoc;
+ try
+ {
+ anObjDoc = TDocStd_Document::Get( theObjectLabel );
+ }
+ catch( ... )
+ {
+ }
+
+ if ( anObjDoc.IsNull() )
+ return aResDoc;
+
+ HYDROData_Application* anApp = HYDROData_Application::GetApplication();
+
+ DataMapOfStudyIDDocument::Iterator aMapIt( anApp->myDocuments );
+ for ( ; aMapIt.More(); aMapIt.Next() )
+ {
+ Handle(HYDROData_Document) anAppDoc = aMapIt.Value();
+ if ( anAppDoc.IsNull() || anAppDoc->myDoc != anObjDoc )
+ continue;
+
+ aResDoc = anAppDoc;
+ break;
+ }
+
+ return aResDoc;
+}
+
bool HYDROData_Document::HasDocument(const int theStudyID)
{
Handle(HYDROData_Document) aResult =
return !aResult.IsNull();
}
+bool HYDROData_Document::DocumentId(const Handle(HYDROData_Document)& theDocument,
+ int& theDocId )
+{
+ return HYDROData_Application::GetApplication()->GetDocumentId(theDocument, theDocId);
+}
+
Data_DocError HYDROData_Document::Load(const char* theFileName, const int theStudyID)
{
Handle(TDocStd_Document) aResult;
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 theIsMultiFile ) 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, theIsMultiFile );
+ if ( aHeaderDump.isEmpty() )
+ return false;
+
+ HYDROData_Tool::WriteStringsToFile( aFile, aHeaderDump );
+
+ bool aRes = true;
+
+ // 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 );
+
+ // Dump code to close python fuction
+ if ( aRes && theIsMultiFile )
+ {
+ QStringList aFooterScript;
+ aFooterScript << QString( "" );
+ aFooterScript << QString( " pass" );
+ HYDROData_Tool::WriteStringsToFile( aFile, aFooterScript );
+ }
+
+ 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 bool theIsMultiFile ) 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 HYDROPy import *" );
+ aResScript << QString( "from PyQt4.QtCore import *" );
+ aResScript << QString( "from PyQt4.QtGui import *" );
+
+ 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
+{
+ 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 );
+
+ 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 );
+ }
+
+ return aRes;
+}
+
void HYDROData_Document::StartOperation()
{
myDoc->NewCommand();
myTransactionsAfterSave++;
}
-Handle_HYDROData_Object 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()