parametertreeparser.hh

Go to the documentation of this file.

/*****************************************************************************
*   This program is part of the msbGrid software                            *
*                                                                           *
*   msbGrid stands for multi-structured block Grid generator                *
*                                                                           *
*   msbGrid is a free software: you can redistribute it and/or modify       *
*   it under the terms of the GNU General Public License as published by    *
*   the Free Software Foundation, either version 2 of the License, or       *
*   (at your option) any later version.                                     *
*                                                                           *
*   msbGrid is distributed in the hope that it will be useful,              *
*   but WITHOUT ANY WARRANTY; without even the implied warranty of          *
*   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the           *
*   GNU General Public License for more details.                            *
*                                                                           *
*   See the file COPYING for full copying permissions.                      *
*****************************************************************************/
#ifndef _PARAMETER_TREE_PARSER_HH_
#define _PARAMETER_TREE_PARSER_HH_

#include <set>

namespace msbGrid
{
class ParameterTreeParser;
}  /* namespace msbGrid */

class msbGrid :: ParameterTreeParser
{
  static std :: string ltrim(const std :: string &s);
  static std :: string rtrim(const std :: string &s);

public:
  static void readINITree(std :: string file,
                          ParameterTree &pt,
                          bool overwrite = true);

  static void readINITree(std :: istream &in,
                          ParameterTree &pt,
                          const std :: string srcname,
                          bool overwrite);
};

void msbGrid :: ParameterTreeParser :: readINITree(std :: string file,
                                                   ParameterTree &pt,
                                                   bool overwrite)
{
  std :: ifstream in( file.c_str() );

  if ( !in ) {
    std :: cerr << "IOError : Could not open configuration file " << file << "\n\nCalling exit(1)\n";
    exit(1);
  }

  readINITree(in, pt, "file '" + file + "'", overwrite);
}

void msbGrid :: ParameterTreeParser :: readINITree(std :: istream &in,
                                                   ParameterTree &pt,
                                                   const std :: string srcname,
                                                   bool overwrite)
{
  std :: set< std :: string > keysInFile;
  int lineIdx = 1;
  while ( !in.eof() ) {
    std :: string line;
    getline(in, line);
    line = ltrim(line);
    switch ( line [ 0 ] ) {
    case '#':
      break;

    default:
      std :: string :: size_type comment = line.find("#");
      line = line.substr(0, comment);
      std :: string :: size_type mid = line.find("=");
      if ( mid != std :: string :: npos ) {
        std :: string key = rtrim( ltrim( line.substr(0, mid) ) );
        std :: string value = ltrim( line.substr(mid + 1) );

        if ( value.length() > 0 ) {
          // handle quoted strings
          if ( ( value [ 0 ] == '\'' )or(value [ 0 ] == '"') ) {
            char quote = value [ 0 ];
            value = value.substr(1);
            while ( * ( rtrim(value).rbegin() ) != quote ) {
              if ( !in.eof() ) {
                std :: string l;
                getline(in, l);
                value = value + "\n" + l;
              } else {
                value = value + quote;
              }
            }

            value = rtrim(value);
            value = value.substr(0, value.length() - 1);
          } else {
            value = rtrim(value);
          }
        }

        if ( keysInFile.count(key) != 0 ) {
          std :: cerr << "Exception, " << "Key '" << key << "' appears twice in " << srcname << " !" << "\n\nCalling exit(1)\n";
          exit(1);
        } else {
          if ( overwrite || !pt.hasKey(key) ) {
            pt [ key ] = value;
          }

          keysInFile.insert(key);
        }
      }

      break;
    }
  }
}

std :: string msbGrid :: ParameterTreeParser :: ltrim(const std :: string &s)
{
  std :: size_t firstNonWS = s.find_first_not_of(" \t\n\r");

  if ( firstNonWS != std :: string :: npos ) {
    return s.substr(firstNonWS);
  }

  return std :: string();
}

std :: string msbGrid :: ParameterTreeParser :: rtrim(const std :: string &s)
{
  std :: size_t lastNonWS = s.find_last_not_of(" \t\n\r");

  if ( lastNonWS != std :: string :: npos ) {
    return s.substr(0, lastNonWS + 1);
  }

  return std :: string();
}

#endif  /* #ifndef _PARAMETER_TREE_PARSER_HH_ */