Salome HOME
26.12.2013. Small update.
[modules/hydro.git] / src / HYDROData / HYDROData_Tool.cxx
1
2 #include "HYDROData_Tool.h"
3
4 #include "HYDROData_ArtificialObject.h"
5 #include "HYDROData_Image.h"
6 #include "HYDROData_Iterator.h"
7 #include "HYDROData_NaturalObject.h"
8
9 #include <TopoDS_Shape.hxx>
10
11 #include <TopTools_SequenceOfShape.hxx>
12
13 #include <TopExp_Explorer.hxx>
14
15 #include <QFile>
16 #include <QStringList>
17 #include <QTextStream>
18
19 void HYDROData_Tool::WriteStringsToFile( QFile&             theFile,
20                                          const QStringList& theStrings,
21                                          const QString&     theSep )
22 {
23   if ( !theFile.isOpen() || theStrings.isEmpty() )
24     return;
25   
26   QString aWriteStr = theStrings.join( theSep );
27   if ( aWriteStr.isEmpty() )
28     return;
29
30   QTextStream anOutStream( &theFile );
31   anOutStream << aWriteStr << theSep << theSep;
32 }
33
34 void HYDROData_Tool::SetMustBeUpdatedObjects(
35   const Handle(HYDROData_Document)& theDoc  )
36 {
37   bool anIsChanged = true;
38
39   // iterate until there is no changes because objects on all level of dependency must be updated
40   while ( anIsChanged )
41   {
42     anIsChanged = false;
43
44     HYDROData_Iterator anIter( theDoc );
45     for ( ; anIter.More(); anIter.Next() )
46     {
47       Handle(HYDROData_Entity) anObject = anIter.Current();
48       if ( anObject.IsNull() || anObject->IsMustBeUpdated() )
49         continue;
50
51       HYDROData_SequenceOfObjects aRefSeq = anObject->GetAllReferenceObjects();
52       for ( int i = 1, n = aRefSeq.Length(); i <= n; ++i )
53       {
54         Handle(HYDROData_Entity) aRefObject = aRefSeq.Value( i );
55         if ( aRefObject.IsNull() || !aRefObject->IsMustBeUpdated() )
56           continue;
57
58         anObject->SetToUpdate( true );
59         anIsChanged = true;
60         break;
61       }
62     }
63   }
64 }
65
66 QString HYDROData_Tool::GenerateObjectName( const Handle(HYDROData_Document)& theDoc,
67                                             const QString&                    thePrefix,
68                                             const QStringList&                theUsedNames,
69                                             const bool                        theIsTryToUsePurePrefix )
70 {
71   QStringList aNamesList( theUsedNames );
72
73   // Collect all used names in the document
74   HYDROData_Iterator anIter( theDoc );
75   for( ; anIter.More(); anIter.Next() )
76   {
77     Handle(HYDROData_Entity) anObject = anIter.Current();
78     if( anObject.IsNull() )
79       continue;
80
81     QString anObjName = anObject->GetName();
82     if ( anObjName.isEmpty() )
83       continue;
84
85     aNamesList.append( anObjName );
86   }
87
88   QString aName;
89
90   if ( theIsTryToUsePurePrefix && !aNamesList.contains( thePrefix ) ) {
91     aName = thePrefix;
92   } else {
93     int anId = 1;
94     int aMaxId = std::numeric_limits<int>::max();
95     while( anId < aMaxId )
96     {
97       aName = QString( "%1_%2" ).arg( thePrefix ).arg( QString::number( anId++ ) );
98
99       // check that there are no other objects with the same name in the document
100       if ( !aNamesList.contains( aName ) )
101         break;
102     }
103   }
104
105   return aName;
106 }
107
108 Handle(HYDROData_Entity) HYDROData_Tool::FindObjectByName( const Handle(HYDROData_Document)& theDoc,
109                                                            const QString&                    theName,
110                                                            const ObjectKind                  theObjectKind )
111 {
112   Handle(HYDROData_Entity) anObject;
113   if ( theName.isEmpty() || theDoc.IsNull() )
114     return anObject;
115
116   QStringList aNamesList;
117   aNamesList << theName;
118
119   HYDROData_SequenceOfObjects aSeqOfObjs = FindObjectsByNames( theDoc, aNamesList, theObjectKind );
120   if( aSeqOfObjs.IsEmpty() )
121     return anObject;
122   
123   anObject = aSeqOfObjs.First();
124   return anObject;
125 }
126
127 HYDROData_SequenceOfObjects HYDROData_Tool::FindObjectsByNames( const Handle(HYDROData_Document)& theDoc,
128                                                                 const QStringList&                theNames,
129                                                                 const ObjectKind                  theObjectKind )
130 {
131   HYDROData_SequenceOfObjects aResSeq;
132   if( theDoc.IsNull() )
133     return aResSeq;
134
135   QStringList aNamesList = theNames;
136
137   HYDROData_Iterator anIter( theDoc, theObjectKind );
138   for( ; anIter.More(); anIter.Next() )
139   {
140     Handle(HYDROData_Entity) anObject = anIter.Current();
141     if( anObject.IsNull() )
142       continue;
143
144     QString anObjName = anObject->GetName();
145     if ( anObjName.isEmpty() || !aNamesList.contains( anObjName ) )
146       continue;
147
148     aResSeq.Append( anObject );
149
150     aNamesList.removeAll( anObjName );
151     if ( aNamesList.isEmpty() )
152       break;
153   }
154
155   return aResSeq;
156 }
157
158 bool HYDROData_Tool::IsGeometryObject( const Handle(HYDROData_Entity)& theObject )
159 {
160   if ( theObject.IsNull() )
161     return false;
162   
163   return theObject->IsKind( STANDARD_TYPE(HYDROData_ArtificialObject) ) ||
164          theObject->IsKind( STANDARD_TYPE(HYDROData_NaturalObject) );
165 }
166
167 void HYDROData_Tool::UpdateChildObjectName( const QString&                  theOldStr,
168                                             const QString&                  theNewStr,
169                                             const Handle(HYDROData_Entity)& theObject )
170 {
171   if ( theObject.IsNull() )
172     return;
173
174   QString anObjName = theObject->GetName();
175   if ( theOldStr.isEmpty() )
176   {
177     while ( anObjName.startsWith( '_' ) )
178       anObjName.remove( 0, 1 );
179
180     anObjName.prepend( theNewStr + "_" );
181   }
182   else if ( anObjName.startsWith( theOldStr ) )
183   {
184     anObjName.replace( 0, theOldStr.length(), theNewStr );
185   }
186   else
187     return;
188
189   theObject->SetName( anObjName );
190 }
191