X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_Tool.cxx;h=fee6e1105dc018dac99aff70dd9077297291c9bb;hb=deed826b2d6c39ba2ed410108cdf54d64cded321;hp=9ed8f0be5063deaae02dd4be0843c127db56ff42;hpb=a3ae70db118828f7996bab07e05d18d619dd2c41;p=modules%2Fhydro.git

diff --git a/src/HYDROData/HYDROData_Tool.cxx b/src/HYDROData/HYDROData_Tool.cxx
index 9ed8f0be..fee6e110 100644
--- a/src/HYDROData/HYDROData_Tool.cxx
+++ b/src/HYDROData/HYDROData_Tool.cxx
@@ -1,2 +1,167 @@
 
 #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>
+
+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::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) );
+}