#include "HYDROData_Tool.h"
+
+#include "HYDROData_ArtificialObject.h"
+#include "HYDROData_Image.h"
+#include "HYDROData_Iterator.h"
+#include "HYDROData_NaturalObject.h"
+
+#include <TopoDS_Shape.hxx>
+
+#include <TopTools_SequenceOfShape.hxx>
+
+#include <TopExp_Explorer.hxx>
+
+#include <QFile>
+#include <QStringList>
+#include <QTextStream>
+
+#include <limits>
+
+static int aMaxNameId = std::numeric_limits<int>::max();
+
+void HYDROData_Tool::WriteStringsToFile( QFile& theFile,
+ const QStringList& theStrings,
+ const QString& theSep )
+{
+ if ( !theFile.isOpen() || theStrings.isEmpty() )
+ return;
+
+ QString aWriteStr = theStrings.join( theSep );
+ if ( aWriteStr.isEmpty() )
+ return;
+
+ QTextStream anOutStream( &theFile );
+ anOutStream << aWriteStr << theSep << theSep;
+}
+
+void HYDROData_Tool::SetMustBeUpdatedObjects(
+ const Handle(HYDROData_Document)& theDoc )
+{
+ bool anIsChanged = true;
+
+ // iterate until there is no changes because objects on all level of dependency must be updated
+ while ( anIsChanged )
+ {
+ anIsChanged = false;
+
+ HYDROData_Iterator anIter( theDoc );
+ for ( ; anIter.More(); anIter.Next() )
+ {
+ Handle(HYDROData_Entity) anObject = anIter.Current();
+ if ( anObject.IsNull() || anObject->IsMustBeUpdated() )
+ continue;
+
+ HYDROData_SequenceOfObjects aRefSeq = anObject->GetAllReferenceObjects();
+ for ( int i = 1, n = aRefSeq.Length(); i <= n; ++i )
+ {
+ Handle(HYDROData_Entity) aRefObject = aRefSeq.Value( i );
+ if ( aRefObject.IsNull() || !aRefObject->IsMustBeUpdated() )
+ continue;
+
+ anObject->SetToUpdate( true );
+ anIsChanged = true;
+ break;
+ }
+ }
+ }
+}
+
+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 < aMaxNameId )
+ {
+ 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) );
+}
+
+void HYDROData_Tool::UpdateChildObjectName( const QString& theOldStr,
+ const QString& theNewStr,
+ const Handle(HYDROData_Entity)& theObject )
+{
+ if ( theObject.IsNull() )
+ return;
+
+ QString anObjName = theObject->GetName();
+ if ( theOldStr.isEmpty() )
+ {
+ while ( anObjName.startsWith( '_' ) )
+ anObjName.remove( 0, 1 );
+
+ anObjName.prepend( theNewStr + "_" );
+ }
+ else if ( anObjName.startsWith( theOldStr ) )
+ {
+ anObjName.replace( 0, theOldStr.length(), theNewStr );
+ }
+ else
+ return;
+
+ theObject->SetName( anObjName );
+}
+
+QString HYDROData_Tool::GenerateNameForPython( const MapOfTreatedObjects& theTreatedObjects,
+ const QString& thePrefix )
+{
+ QString aName = thePrefix;
+ if ( !theTreatedObjects.contains( aName ) )
+ return aName;
+
+ int anId = 1;
+ while( anId < aMaxNameId )
+ {
+ aName = QString( "%1_%2" ).arg( thePrefix ).arg( QString::number( anId++ ) );
+
+ // check that there are no other objects with the same name
+ if ( !theTreatedObjects.contains( aName ) )
+ break;
+ }
+
+ return aName;
+}
\ No newline at end of file