#include "HYDROData_Tool.h"
+#include "HYDROData_ArtificialObject.h"
+#include "HYDROData_Image.h"
+#include "HYDROData_Iterator.h"
+#include "HYDROData_NaturalObject.h"
+
#include <QFile>
#include <QStringList>
#include <QTextStream>
QTextStream anOutStream( &theFile );
anOutStream << aWriteStr << theSep << theSep;
}
+
+void HYDROData_Tool::SetMustBeUpdatedImages(
+ const Handle(HYDROData_Document)& theDoc )
+{
+ bool anIsChanged = true;
+
+ // iterate until there is no changes because images on all level of dependency must be updated
+ while ( anIsChanged )
+ {
+ anIsChanged = false;
+
+ HYDROData_Iterator anIter( theDoc, KIND_IMAGE );
+ for ( ; anIter.More(); anIter.Next() )
+ {
+ Handle(HYDROData_Image) anImage =
+ Handle(HYDROData_Image)::DownCast( anIter.Current() );
+ if ( anImage.IsNull() || anImage->IsMustBeUpdated() )
+ continue;
+
+ Handle(HYDROData_Image) aTrsfRefImage = anImage->GetTrsfReferenceImage();
+ if ( !aTrsfRefImage.IsNull() && aTrsfRefImage->IsMustBeUpdated() )
+ {
+ anImage->SetToUpdate( true );
+ anIsChanged = true;
+ continue;
+ }
+
+ for ( int i = 0, aNBRefs = anImage->NbReferences(); i < aNBRefs; ++i )
+ {
+ Handle(HYDROData_Image) aRefImage =
+ Handle(HYDROData_Image)::DownCast( anImage->Reference( i ) );
+ if ( !aRefImage.IsNull() && aRefImage->IsMustBeUpdated() )
+ {
+ // image references to updated => also must be updated
+ anImage->SetToUpdate( true );
+ anIsChanged = true;
+ }
+ }
+ }
+ }
+}
+
+QString HYDROData_Tool::GenerateObjectName( const Handle(HYDROData_Document)& theDoc,
+ const QString& thePrefix,
+ const QStringList& theUsedNames,
+ const bool theIsTryToUsePurePrefix )
+{
+ QStringList aNamesList( theUsedNames );
+
+ // Collect all used names in the document
+ HYDROData_Iterator anIter( theDoc );
+ for( ; anIter.More(); anIter.Next() )
+ {
+ Handle(HYDROData_Entity) anObject = anIter.Current();
+ if( anObject.IsNull() )
+ continue;
+
+ QString anObjName = anObject->GetName();
+ if ( anObjName.isEmpty() )
+ continue;
+
+ aNamesList.append( anObjName );
+ }
+
+ QString aName;
+
+ if ( theIsTryToUsePurePrefix && !aNamesList.contains( thePrefix ) ) {
+ aName = thePrefix;
+ } else {
+ int anId = 1;
+ while( anId < 1000 )
+ {
+ aName = QString( "%1_%2" ).arg( thePrefix ).arg( QString::number( anId++ ) );
+
+ // check that there are no other objects with the same name in the document
+ if ( !aNamesList.contains( aName ) )
+ break;
+ }
+ }
+
+ return aName;
+}
+
+Handle(HYDROData_Entity) HYDROData_Tool::FindObjectByName( const Handle(HYDROData_Document)& theDoc,
+ const QString& theName,
+ const ObjectKind theObjectKind )
+{
+ Handle(HYDROData_Entity) anObject;
+ if ( theName.isEmpty() || theDoc.IsNull() )
+ return anObject;
+
+ QStringList aNamesList;
+ aNamesList << theName;
+
+ HYDROData_SequenceOfObjects aSeqOfObjs = FindObjectsByNames( theDoc, aNamesList, theObjectKind );
+ if( aSeqOfObjs.IsEmpty() )
+ return anObject;
+
+ anObject = aSeqOfObjs.First();
+ return anObject;
+}
+
+HYDROData_SequenceOfObjects HYDROData_Tool::FindObjectsByNames( const Handle(HYDROData_Document)& theDoc,
+ const QStringList& theNames,
+ const ObjectKind theObjectKind )
+{
+ HYDROData_SequenceOfObjects aResSeq;
+ if( theDoc.IsNull() )
+ return aResSeq;
+
+ QStringList aNamesList = theNames;
+
+ HYDROData_Iterator anIter( theDoc, 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;
+}
+
+bool HYDROData_Tool::IsGeometryObject( const Handle(HYDROData_Entity)& theObject )
+{
+ if ( theObject.IsNull() )
+ return false;
+
+ return theObject->IsKind( STANDARD_TYPE(HYDROData_ArtificialObject) ) ||
+ theObject->IsKind( STANDARD_TYPE(HYDROData_NaturalObject) );
+}