X-Git-Url: http://git.salome-platform.org/gitweb/?a=blobdiff_plain;f=src%2FHYDROData%2FHYDROData_StricklerTable.cxx;h=8dda167363bebb7f03c64f4a6071ba8bc4ae56c2;hb=HEAD;hp=5d7623d8047cedf9f153be9b5e9d86a77f25fae4;hpb=d6a8c59ee076938a924fcfd1e3f83ef1395b7336;p=modules%2Fhydro.git

diff --git a/src/HYDROData/HYDROData_StricklerTable.cxx b/src/HYDROData/HYDROData_StricklerTable.cxx
index 5d7623d8..8dda1673 100644
--- a/src/HYDROData/HYDROData_StricklerTable.cxx
+++ b/src/HYDROData/HYDROData_StricklerTable.cxx
@@ -17,20 +17,28 @@
 //
 
 #include <HYDROData_StricklerTable.h>
+#include <HYDROData_Tool.h>
+#include <HYDROData_Iterator.h>
 
 #include <TDataStd_NamedData.hxx>
 
 #include <TDataStd_DataMapOfStringReal.hxx>
+#include <TColStd_DataMapOfStringInteger.hxx>
+#include <TDataStd_DataMapOfStringString.hxx>
 #include <TDataStd_DataMapIteratorOfDataMapOfStringReal.hxx>
+#include <TDataStd_DataMapIteratorOfDataMapOfStringString.hxx>
+#include <TDataStd_AsciiString.hxx>
 
 #include <TCollection_AsciiString.hxx>
 #include <TCollection_ExtendedString.hxx>
 
-#include <QtCore/QString>
-#include <QtCore/QStringList>
+#include <QString>
+#include <QColor>
+#include <QStringList>
+#include <QFile>
+#include <QTextStream>
 
-IMPLEMENT_STANDARD_HANDLE( HYDROData_StricklerTable, HYDROData_Entity )
-    IMPLEMENT_STANDARD_RTTIEXT( HYDROData_StricklerTable, HYDROData_Entity )
+IMPLEMENT_STANDARD_RTTIEXT( HYDROData_StricklerTable, HYDROData_Entity )
 
 HYDROData_StricklerTable::HYDROData_StricklerTable()
 : HYDROData_Entity( Geom_No )
@@ -43,93 +51,106 @@ HYDROData_StricklerTable::~HYDROData_StricklerTable()
 
 const ObjectKind HYDROData_StricklerTable::GetKind() const
 {
-    return KIND_STRICKLER_TABLE;
+  return KIND_STRICKLER_TABLE;
 }
 
-bool HYDROData_StricklerTable::Import( const TCollection_AsciiString& theFileName )
+bool HYDROData_StricklerTable::Import( const QString& theFileName )
 {
-    std::ifstream aStream( theFileName.ToCString(), std::ios::in | std::ios::binary );
-    if ( !aStream )
-        return false;
-
-    bool aRes = true;
-
-    const int aBufLen = 512;
-    char aBuf[aBufLen];
-    while( !aStream.eof() && aRes )
+  Handle(TDataStd_NamedData) aMap = Map();
+  if( aMap.IsNull() )
+    return false;
+
+  QFile aFile( theFileName );
+  if( !aFile.open( QFile::ReadOnly | QFile::Text ) )
+    return false;
+
+  aMap->ChangeReals( TDataStd_DataMapOfStringReal() );
+  aMap->ChangeStrings( TDataStd_DataMapOfStringString() );
+  aMap->ChangeIntegers( TColStd_DataMapOfStringInteger() );
+
+  QTextStream aStream( &aFile );
+  int i=0;
+  while( !aStream.atEnd() )
+  {
+    QString aLine = aStream.readLine();
+    if( i==0 )
     {
-        aStream.getline( aBuf, aBufLen );
-        TCollection_ExtendedString aStr( aBuf, Standard_True );
-
-        Standard_Integer aPos = aStr.SearchFromEnd( " " );
-        if ( aPos < 0 )
-            aRes = false;
-        else
-        {
-            TCollection_ExtendedString aValueStr = aStr.Split( aPos );
-            while ( aStr.Length() > 0 && ( aStr.Value( aStr.Length() ) == ' ' || aStr.Value( aStr.Length() ) == '\t' ) )
-                aStr.Remove( aStr.Length() );
-            while ( aStr.Length() > 0 && ( aStr.Value( 1 ) == ' ' || aStr.Value( 1 ) == '\t' ) || aStr.Value( 1 ) < 0 )
-                aStr.Remove( 1 );
-
-            if ( aStr.Length() > 0 && aStr.Value( aStr.Length() ) == '\"' )
-                aStr.Remove( aStr.Length() );
-            if ( aStr.Length() > 0 && aStr.Value( 1 ) == '\"' )
-                aStr.Remove( 1 );
-
-            if ( aValueStr.IsEmpty() || aStr.IsEmpty() )
-                aRes = false;
-            else
-            {
-                Standard_Real aValue = TCollection_AsciiString( aValueStr ).RealValue();
-                Set( QString( (QChar*)aStr.ToExtString(), aStr.Length() ), aValue );
-            }
-        }
+      SetAttrName( aLine );
     }
+    else
+    {
+      int p1 = aLine.indexOf( '"' );
+      int p2 = aLine.indexOf( '"', p1+1 );
+      QString aType = aLine.mid( p1+1, p2-p1-1 );
+      QStringList aParts = aLine.mid( p2+1 ).split( ' ', QString::SkipEmptyParts );
+      double aCoeff = aParts[0].toDouble();
+      int aColorInt = aParts[1].toInt( 0, 16 );
+      QString anAttrValue = aParts.size()>2 ? aParts[2] : QString();
+      
+      TCollection_ExtendedString aTypeExt = HYDROData_Tool::toExtString( aType );
+      aMap->SetReal( aTypeExt, aCoeff );
+      aMap->SetInteger( aTypeExt, aColorInt );
+      aMap->SetString( aTypeExt, HYDROData_Tool::toExtString( anAttrValue ) );
+    }
+    i++;
+  }
+  aFile.close();
 
-    aStream.close();
-    return aRes;
+  return true;
 }
 
-bool HYDROData_StricklerTable::Export( const TCollection_AsciiString& theFileName )
+bool HYDROData_StricklerTable::Export( const QString& theFileName )
 {
-    Handle(TDataStd_NamedData) aMap = Map();
-    if ( aMap.IsNull() )
-        return false;
+  Handle(TDataStd_NamedData) aMap = Map();
+  if( aMap.IsNull() )
+    return false;
 
-    std::ofstream aStream( theFileName.ToCString(), std::ios::out | std::ios::binary );
-    if ( !aStream )
-        return false;
+  QFile aFile( theFileName );
+  if( !aFile.open( QFile::WriteOnly | QFile::Text ) )
+    return false;
 
-    bool aRes = true;
-    for ( TDataStd_DataMapIteratorOfDataMapOfStringReal it( aMap->GetRealsContainer() ); it.More() && aRes; it.Next() )
-    {
-        TCollection_ExtendedString aLine = TCollection_ExtendedString( '\"' ) + it.Key() + TCollection_ExtendedString( '\"' ) +
-            TCollection_ExtendedString( ' ' ) + TCollection_ExtendedString( it.Value() );
-        Standard_PCharacter aBuf = (Standard_PCharacter)malloc( aLine.LengthOfCString() + 1 );
-        aStream.write( aBuf, aLine.ToUTF8CString( aBuf ) );
-        aStream.write( "\r\n", 2 );
-        free( aBuf );
-    }
+  QTextStream aStream( &aFile );
+  aStream.setCodec( "UTF-8" );
+  aStream.setGenerateByteOrderMark( true );
+
+  aStream << GetAttrName() << "\n";
+
+  bool aRes = true;
+  QStringList aTypes = GetTypes();
+  foreach( QString aType, aTypes )
+  {
+    TCollection_ExtendedString aTypeExt = HYDROData_Tool::toExtString( aType );
+
+    aStream << "\"" << aType << "\" " << Get( aType, 0.0 );
 
-    aStream.close();
-    return aRes;
+    QString aColor = QString::number( aMap->GetInteger( aTypeExt ), 16 ).toUpper();
+    aColor = QString( 6-aColor.length(), '0' ) + aColor;
+    QString anAttrValue = HYDROData_Tool::toQString( aMap->GetString( aTypeExt ) );
+
+    aStream << " " << aColor;
+    if( !anAttrValue.isEmpty() )
+      aStream << " " << anAttrValue;
+    aStream << "\n";
+  }
+
+  aFile.close();
+  return aRes;
 }
 
 double HYDROData_StricklerTable::Get( const QString& theType, double theDefault ) const
 {
-    Handle( TDataStd_NamedData ) aMap = Map();
-    TCollection_ExtendedString aType = toExtString( theType );
-    if( aMap->HasReal( aType ) )
-        return aMap->GetReal( aType );
-    else
-        return theDefault;
+  Handle( TDataStd_NamedData ) aMap = Map();
+  TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
+  if( aMap->HasReal( aType ) )
+    return aMap->GetReal( aType );
+  else
+    return theDefault;
 }
 
 void HYDROData_StricklerTable::Set( const QString& theType, double theCoefficient )
 {
-    Handle(TDataStd_NamedData) aMap = Map();
-    aMap->SetReal( toExtString( theType ), theCoefficient );
+  Handle(TDataStd_NamedData) aMap = Map();
+  aMap->SetReal( HYDROData_Tool::toExtString( theType ), theCoefficient );
 }
 
 void HYDROData_StricklerTable::GetCoefficientRange( double& theMin, double& theMax ) const
@@ -139,12 +160,15 @@ void HYDROData_StricklerTable::GetCoefficientRange( double& theMin, double& theM
   
   Handle(TDataStd_NamedData) aMap = Map();
   Standard_Boolean isFirst = Standard_True;
-  for ( TDataStd_DataMapIteratorOfDataMapOfStringReal it( aMap->GetRealsContainer() ); it.More(); it.Next() ) {
+  for ( TDataStd_DataMapIteratorOfDataMapOfStringReal it( aMap->GetRealsContainer() ); it.More(); it.Next() )
+  {
     Standard_Real aValue = it.Value();
-    if ( theMin == 0 || aValue < theMin ) {
+    if ( theMin == 0 || aValue < theMin )
+    {
       theMin = aValue;
     }
-    if ( theMax == 0 || aValue > theMax ) {
+    if ( theMax == 0 || aValue > theMax )
+    {
       theMax = aValue;
     }
   }
@@ -152,78 +176,174 @@ void HYDROData_StricklerTable::GetCoefficientRange( double& theMin, double& theM
 
 QStringList HYDROData_StricklerTable::GetTypes() const
 {
-    QStringList aSeq;
-    Handle(TDataStd_NamedData) aMap = Map();
-    if ( !aMap.IsNull() )
-    {
-        for ( TDataStd_DataMapIteratorOfDataMapOfStringReal it( aMap->GetRealsContainer() ); it.More(); it.Next() )
-            aSeq.append( toQString( it.Key() ) );
-    }
-    return aSeq;
+  QStringList aSeq;
+  Handle(TDataStd_NamedData) aMap = Map();
+  if( !aMap.IsNull() )
+  {
+    for( TDataStd_DataMapIteratorOfDataMapOfStringReal it( aMap->GetRealsContainer() ); it.More(); it.Next() )
+       aSeq.append( HYDROData_Tool::toQString( it.Key() ) );
+  }
+  aSeq.sort();
+  return aSeq;
 }
 
 bool HYDROData_StricklerTable::HasType( const QString& theType ) const
 {
   Handle(TDataStd_NamedData) aMap = Map();
   
-  TCollection_ExtendedString aType = toExtString( theType );
+  TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
   return !aMap.IsNull() && aMap->HasReal( aType );
 }
 
 void HYDROData_StricklerTable::Clear()
 {
-    Handle(TDataStd_NamedData) aMap = Map();
-    if ( !aMap.IsNull() )
-        aMap->ChangeReals( TDataStd_DataMapOfStringReal() );
+  Handle(TDataStd_NamedData) aMap = Map();
+  if( !aMap.IsNull() )
+    aMap->ChangeReals( TDataStd_DataMapOfStringReal() );
 }
 
-QStringList HYDROData_StricklerTable::DumpToPython( MapOfTreatedObjects& theTreatedObjects ) const
+QStringList HYDROData_StricklerTable::DumpToPython( const QString& thePyScriptPath,
+                                                    MapOfTreatedObjects& theTreatedObjects ) const
 {
-    QStringList aResList = dumpObjectCreation( theTreatedObjects );
-    QString aPyName = GetObjPyName();
+  QStringList aResList = dumpObjectCreation( theTreatedObjects );
+  QString aPyName = GetObjPyName();
 
-    aResList << QString( "" );
-    Handle(TDataStd_NamedData) aMap = Map();
-    if ( !aMap.IsNull() )
+  QString anAttrName = GetAttrName();
+  aResList << QString( "%1.SetAttrName( \"%2\" )" ).arg( aPyName ).arg( anAttrName );
+
+  aResList << QString( "" );
+  Handle(TDataStd_NamedData) aMap = Map();
+  if( !aMap.IsNull() )
+  {
+    for( TDataStd_DataMapIteratorOfDataMapOfStringReal it( aMap->GetRealsContainer() ); it.More(); it.Next() )
     {
-        for ( TDataStd_DataMapIteratorOfDataMapOfStringReal it( aMap->GetRealsContainer() ); it.More(); it.Next() )
-        {
-            TCollection_ExtendedString aType = it.Key();
-            Standard_Real aValue = it.Value();
-            aResList << QString( "%1.Set( \"%2\", %3 );" ).arg( aPyName ).arg( QString( (QChar*)aType.ToExtString(), aType.Length() ) ).arg( aValue );
-        }
+      QString aType = HYDROData_Tool::toQString( it.Key() );
+      Standard_Real aValue = it.Value();
+      aResList << QString( "%1.Set( u\"%2\", %3 )" ).arg( aPyName ).arg( aType ).arg( aValue );
+
+      QString anAttrValue = GetAttrValue( aType );
+      aResList << QString( "%1.SetAttrValue( u\"%2\", \"%3\" )" ).arg( aPyName ).arg( aType ).arg( anAttrValue );
+      
+      QColor aColor = GetColor( aType );
+      aResList << QString( "%1.SetColor( u\"%2\", QColor( %3, %4, %5 ) )" ).
+        arg( aPyName ).arg( aType ).arg( aColor.red() ).arg( aColor.green() ).arg( aColor.blue() );
+      aResList << QString();
     }
-    aResList << QString( "" );
-    aResList << QString( "%1.Update();" ).arg( aPyName );
+  }
+  aResList << QString( "" );
+  aResList << QString( "%1.Update()" ).arg( aPyName );
 
-    return aResList;
+  return aResList;
 }
 
 Handle(TDataStd_NamedData) HYDROData_StricklerTable::Map() const
 {
-    TDF_Label aLabel = myLab.FindChild( DataTag_Table );
-    Handle( TDataStd_NamedData ) aMap;
-    if( !aLabel.FindAttribute( TDataStd_NamedData::GetID(), aMap ) )
-        aMap = TDataStd_NamedData::Set( aLabel );
-    return aMap;
+  TDF_Label aLabel = myLab.FindChild( DataTag_Table );
+  Handle( TDataStd_NamedData ) aMap;
+  if( !aLabel.FindAttribute( TDataStd_NamedData::GetID(), aMap ) )
+    aMap = TDataStd_NamedData::Set( aLabel );
+  return aMap;
 }
 
-TCollection_ExtendedString HYDROData_StricklerTable::toExtString( const QString& theStr ) const
+QString HYDROData_StricklerTable::GetAttrName() const
 {
-    TCollection_ExtendedString aRes;
-    if( !theStr.isEmpty() )
-    {
-	    Standard_ExtString extStr = new Standard_ExtCharacter[ ( theStr.length() + 1 ) * 2 ];
-	    memcpy( (void*)extStr, theStr.unicode(), theStr.length() * 2 );
-	    ((short*)extStr)[theStr.length()] = '\0';
+  Handle(TDataStd_AsciiString) aName;
+  if( myLab.FindChild( DataTag_AttrName ).FindAttribute(TDataStd_AsciiString::GetID(), aName)) {
+    TCollection_AsciiString aStr(aName->Get());
+    return QString(aStr.ToCString());
+  }
+  return QString();}
+
+bool HYDROData_StricklerTable::SetAttrName( const QString& theAttrName ) const
+{
+  HYDROData_Iterator anIt( HYDROData_Document::Document(), KIND_STRICKLER_TABLE );
+  for( ; anIt.More(); anIt.Next() )
+  {
+    Handle( HYDROData_StricklerTable ) aTable = 
+      Handle( HYDROData_StricklerTable )::DownCast( anIt.Current() );
+    if( aTable->Label()==myLab )
+      continue;
+
+    if( theAttrName==aTable->GetAttrName() )
+      return false;
+  }
+
+  Handle(TDataStd_AsciiString) anAttr = TDataStd_AsciiString::Set( myLab.FindChild( DataTag_AttrName ), HYDROData_Tool::toExtString( theAttrName ) );
+  anAttr->SetID(TDataStd_AsciiString::GetID());
+  return true;
+}
+
+QString HYDROData_StricklerTable::GetAttrValue( const QString& theType ) const
+{
+  Handle( TDataStd_NamedData ) aMap = Map();
+  TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
+  if( aMap->HasString( aType ) )
+    return HYDROData_Tool::toQString( aMap->GetString( aType ) );
+  else
+    return "";
+}
+
+void HYDROData_StricklerTable::SetAttrValue( const QString& theType, const QString& theAttrValue ) const
+{
+  Handle( TDataStd_NamedData ) aMap = Map();
+  TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
+  aMap->SetString( aType, HYDROData_Tool::toExtString( theAttrValue ) );
+}
 
-	    aRes = TCollection_ExtendedString( extStr );
-	    delete [] extStr;
+QString HYDROData_StricklerTable::GetType( const QString& theAttrValue ) const
+{
+  if( theAttrValue.isEmpty() )
+    return "";
+
+  Handle( TDataStd_NamedData ) aMap = Map();
+  TCollection_ExtendedString anAttrValue = HYDROData_Tool::toExtString( theAttrValue );
+  for( TDataStd_DataMapIteratorOfDataMapOfStringString it( aMap->GetStringsContainer() ); it.More(); it.Next() )
+  {
+    if( it.Value() == anAttrValue )
+    {
+      QString aType = HYDROData_Tool::toQString( it.Key() );
+      return aType;
     }
-    return aRes;
+  }
+  return "";
+}
+
+QColor HYDROData_StricklerTable::GetColor( const QString& theType ) const
+{
+  Handle( TDataStd_NamedData ) aMap = Map();
+  TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
+  if( aMap->HasInteger( aType ) )
+  {
+    int aColorInt = aMap->GetInteger( aType );
+    int b = aColorInt % 256;
+    int g = (aColorInt>>8) % 256;
+    int r = (aColorInt>>16) % 256;
+    return QColor( r, g, b );
+  }
+  else
+    return QColor();
 }
 
-QString HYDROData_StricklerTable::toQString( const TCollection_ExtendedString& theStr ) const
+void HYDROData_StricklerTable::SetColor( const QString& theType, const QColor& theColor ) const
 {
-    return QString( (QChar*)theStr.ToExtString(), theStr.Length() );
+  Handle( TDataStd_NamedData ) aMap = Map();
+  TCollection_ExtendedString aType = HYDROData_Tool::toExtString( theType );
+  int r = theColor.red();
+  int g = theColor.green();
+  int b = theColor.blue();
+  int aColorInt = ( r<<16 ) + ( g<<8 ) + b;
+  aMap->SetInteger( aType, aColorInt );
+
+  // synchronize the color for the same type in other maps
+  HYDROData_Iterator anIt( HYDROData_Document::Document(), KIND_STRICKLER_TABLE );
+  for( ; anIt.More(); anIt.Next() )
+  {
+    Handle( HYDROData_StricklerTable ) aTable = 
+      Handle( HYDROData_StricklerTable )::DownCast( anIt.Current() );
+    if( aTable->Label()==myLab )
+      continue;
+
+    if( aTable->HasType( theType ) )
+      aTable->Map()->SetInteger( aType, aColorInt );
+  }
 }