1 // Copyright (C) 2014-2015 EDF-R&D
2 // This library is free software; you can redistribute it and/or
3 // modify it under the terms of the GNU Lesser General Public
4 // License as published by the Free Software Foundation; either
5 // version 2.1 of the License, or (at your option) any later version.
7 // This library is distributed in the hope that it will be useful,
8 // but WITHOUT ANY WARRANTY; without even the implied warranty of
9 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
10 // Lesser General Public License for more details.
12 // You should have received a copy of the GNU Lesser General Public
13 // License along with this library; if not, write to the Free Software
14 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
16 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
19 #include <HYDROData_StricklerTable.h>
20 #include <HYDROData_Tool.h>
21 #include <HYDROData_Iterator.h>
23 #include <TDataStd_NamedData.hxx>
25 #include <TDataStd_DataMapOfStringReal.hxx>
26 #include <TColStd_DataMapOfStringInteger.hxx>
27 #include <TDataStd_DataMapOfStringString.hxx>
28 #include <TDataStd_DataMapIteratorOfDataMapOfStringReal.hxx>
29 #include <TDataStd_DataMapIteratorOfDataMapOfStringString.hxx>
30 #include <TDataStd_AsciiString.hxx>
32 #include <TCollection_AsciiString.hxx>
33 #include <TCollection_ExtendedString.hxx>
37 #include <QStringList>
39 #include <QTextStream>
41 IMPLEMENT_STANDARD_HANDLE( HYDROData_StricklerTable, HYDROData_Entity )
42 IMPLEMENT_STANDARD_RTTIEXT( HYDROData_StricklerTable, HYDROData_Entity )
44 HYDROData_StricklerTable::HYDROData_StricklerTable()
45 : HYDROData_Entity( Geom_No )
49 HYDROData_StricklerTable::~HYDROData_StricklerTable()
53 const ObjectKind HYDROData_StricklerTable::GetKind() const
55 return KIND_STRICKLER_TABLE;
58 bool HYDROData_StricklerTable::Import( const QString& theFileName )
60 Handle(TDataStd_NamedData) aMap = Map();
64 QFile aFile( theFileName );
65 if( !aFile.open( QFile::ReadOnly | QFile::Text ) )
68 aMap->ChangeReals( TDataStd_DataMapOfStringReal() );
69 aMap->ChangeStrings( TDataStd_DataMapOfStringString() );
70 aMap->ChangeIntegers( TColStd_DataMapOfStringInteger() );
72 QTextStream aStream( &aFile );
74 while( !aStream.atEnd() )
76 QString aLine = aStream.readLine();
83 int p1 = aLine.indexOf( '"' );
84 int p2 = aLine.indexOf( '"', p1+1 );
85 QString aType = aLine.mid( p1+1, p2-p1-1 );
86 QStringList aParts = aLine.mid( p2+1 ).split( ' ', QString::SkipEmptyParts );
87 double aCoeff = aParts[0].toDouble();
88 int aColorInt = aParts[1].toInt( 0, 16 );
89 QString anAttrValue = aParts.size()>2 ? aParts[2] : QString();
91 TCollection_ExtendedString aTypeExt = HYDROData_Tool::toExtString( aType );
92 aMap->SetReal( aTypeExt, aCoeff );
93 aMap->SetInteger( aTypeExt, aColorInt );
94 aMap->SetString( aTypeExt, HYDROData_Tool::toExtString( anAttrValue ) );
103 bool HYDROData_StricklerTable::Export( const QString& theFileName )
105 Handle(TDataStd_NamedData) aMap = Map();
109 QFile aFile( theFileName );
110 if( !aFile.open( QFile::WriteOnly | QFile::Text ) )
113 QTextStream aStream( &aFile );
114 aStream.setCodec( "UTF-8" );
115 aStream.setGenerateByteOrderMark( true );
117 aStream << GetAttrName() << "\n";
120 QStringList aTypes = GetTypes();
121 foreach( QString aType, aTypes )
123 TCollection_ExtendedString aTypeExt = HYDROData_Tool::toExtString( aType );
125 aStream << "\"" << aType << "\" " << Get( aType, 0.0 );
127 QString aColor = QString::number( aMap->GetInteger( aTypeExt ), 16 ).toUpper();
128 aColor = QString( 6-aColor.length(), '0' ) + aColor;
129 QString anAttrValue = HYDROData_Tool::toQString( aMap->GetString( aTypeExt ) );
131 aStream << " " << aColor;
132 if( !anAttrValue.isEmpty() )
133 aStream << " " << anAttrValue;
141 double HYDROData_StricklerTable::Get( const QString& theType, double theDefault ) const
143 Handle( TDataStd_NamedData ) aMap = Map();
144 TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
145 if( aMap->HasReal( aType ) )
146 return aMap->GetReal( aType );
151 void HYDROData_StricklerTable::Set( const QString& theType, double theCoefficient )
153 Handle(TDataStd_NamedData) aMap = Map();
154 aMap->SetReal( HYDROData_Tool::toExtString( theType ), theCoefficient );
157 void HYDROData_StricklerTable::GetCoefficientRange( double& theMin, double& theMax ) const
162 Handle(TDataStd_NamedData) aMap = Map();
163 Standard_Boolean isFirst = Standard_True;
164 for ( TDataStd_DataMapIteratorOfDataMapOfStringReal it( aMap->GetRealsContainer() ); it.More(); it.Next() )
166 Standard_Real aValue = it.Value();
167 if ( theMin == 0 || aValue < theMin )
171 if ( theMax == 0 || aValue > theMax )
178 QStringList HYDROData_StricklerTable::GetTypes() const
181 Handle(TDataStd_NamedData) aMap = Map();
184 for( TDataStd_DataMapIteratorOfDataMapOfStringReal it( aMap->GetRealsContainer() ); it.More(); it.Next() )
185 aSeq.append( HYDROData_Tool::toQString( it.Key() ) );
191 bool HYDROData_StricklerTable::HasType( const QString& theType ) const
193 Handle(TDataStd_NamedData) aMap = Map();
195 TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
196 return !aMap.IsNull() && aMap->HasReal( aType );
199 void HYDROData_StricklerTable::Clear()
201 Handle(TDataStd_NamedData) aMap = Map();
203 aMap->ChangeReals( TDataStd_DataMapOfStringReal() );
206 QStringList HYDROData_StricklerTable::DumpToPython( const QString& thePyScriptPath,
207 MapOfTreatedObjects& theTreatedObjects ) const
209 QStringList aResList = dumpObjectCreation( theTreatedObjects );
210 QString aPyName = GetObjPyName();
212 QString anAttrName = GetAttrName();
213 aResList << QString( "%1.SetAttrName( \"%2\" )" ).arg( aPyName ).arg( anAttrName );
215 aResList << QString( "" );
216 Handle(TDataStd_NamedData) aMap = Map();
219 for( TDataStd_DataMapIteratorOfDataMapOfStringReal it( aMap->GetRealsContainer() ); it.More(); it.Next() )
221 QString aType = HYDROData_Tool::toQString( it.Key() );
222 Standard_Real aValue = it.Value();
223 aResList << QString( "%1.Set( u\"%2\", %3 )" ).arg( aPyName ).arg( aType ).arg( aValue );
225 QString anAttrValue = GetAttrValue( aType );
226 aResList << QString( "%1.SetAttrValue( u\"%2\", \"%3\" )" ).arg( aPyName ).arg( aType ).arg( anAttrValue );
228 QColor aColor = GetColor( aType );
229 aResList << QString( "%1.SetColor( u\"%2\", QColor( %3, %4, %5 ) )" ).
230 arg( aPyName ).arg( aType ).arg( aColor.red() ).arg( aColor.green() ).arg( aColor.blue() );
231 aResList << QString();
234 aResList << QString( "" );
235 aResList << QString( "%1.Update()" ).arg( aPyName );
240 Handle(TDataStd_NamedData) HYDROData_StricklerTable::Map() const
242 TDF_Label aLabel = myLab.FindChild( DataTag_Table );
243 Handle( TDataStd_NamedData ) aMap;
244 if( !aLabel.FindAttribute( TDataStd_NamedData::GetID(), aMap ) )
245 aMap = TDataStd_NamedData::Set( aLabel );
249 QString HYDROData_StricklerTable::GetAttrName() const
251 Handle(TDataStd_AsciiString) aName;
252 if( myLab.FindChild( DataTag_AttrName ).FindAttribute(TDataStd_AsciiString::GetID(), aName)) {
253 TCollection_AsciiString aStr(aName->Get());
254 return QString(aStr.ToCString());
258 bool HYDROData_StricklerTable::SetAttrName( const QString& theAttrName ) const
260 HYDROData_Iterator anIt( HYDROData_Document::Document( myLab ), KIND_STRICKLER_TABLE );
261 for( ; anIt.More(); anIt.Next() )
263 Handle( HYDROData_StricklerTable ) aTable =
264 Handle( HYDROData_StricklerTable )::DownCast( anIt.Current() );
265 if( aTable->Label()==myLab )
268 if( theAttrName==aTable->GetAttrName() )
272 TDataStd_AsciiString::Set( myLab.FindChild( DataTag_AttrName ), HYDROData_Tool::toExtString( theAttrName ) );
276 QString HYDROData_StricklerTable::GetAttrValue( const QString& theType ) const
278 Handle( TDataStd_NamedData ) aMap = Map();
279 TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
280 if( aMap->HasString( aType ) )
281 return HYDROData_Tool::toQString( aMap->GetString( aType ) );
286 void HYDROData_StricklerTable::SetAttrValue( const QString& theType, const QString& theAttrValue ) const
288 Handle( TDataStd_NamedData ) aMap = Map();
289 TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
290 aMap->SetString( aType, HYDROData_Tool::toExtString( theAttrValue ) );
293 QString HYDROData_StricklerTable::GetType( const QString& theAttrValue ) const
295 if( theAttrValue.isEmpty() )
298 Handle( TDataStd_NamedData ) aMap = Map();
299 TCollection_ExtendedString anAttrValue = HYDROData_Tool::toExtString( theAttrValue );
300 for( TDataStd_DataMapIteratorOfDataMapOfStringString it( aMap->GetStringsContainer() ); it.More(); it.Next() )
302 if( it.Value() == anAttrValue )
304 QString aType = HYDROData_Tool::toQString( it.Key() );
311 QColor HYDROData_StricklerTable::GetColor( const QString& theType ) const
313 Handle( TDataStd_NamedData ) aMap = Map();
314 TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
315 if( aMap->HasInteger( aType ) )
317 int aColorInt = aMap->GetInteger( aType );
318 int b = aColorInt % 256;
319 int g = (aColorInt>>8) % 256;
320 int r = (aColorInt>>16) % 256;
321 return QColor( r, g, b );
327 void HYDROData_StricklerTable::SetColor( const QString& theType, const QColor& theColor ) const
329 Handle( TDataStd_NamedData ) aMap = Map();
330 TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
331 int r = theColor.red();
332 int g = theColor.green();
333 int b = theColor.blue();
334 int aColorInt = ( r<<16 ) + ( g<<8 ) + b;
335 aMap->SetInteger( aType, aColorInt );
337 // synchronize the color for the same type in other maps
338 HYDROData_Iterator anIt( HYDROData_Document::Document( myLab ), KIND_STRICKLER_TABLE );
339 for( ; anIt.More(); anIt.Next() )
341 Handle( HYDROData_StricklerTable ) aTable =
342 Handle( HYDROData_StricklerTable )::DownCast( anIt.Current() );
343 if( aTable->Label()==myLab )
346 if( aTable->HasType( theType ) )
347 aTable->Map()->SetInteger( aType, aColorInt );