QGTextFileHandler.cc

Go to the documentation of this file.

#include "QGTextFileHandler.hh"
#include <fstream>
#include <sstream>
 
ClassImp(QGTextFileHandler)
 
using std::endl;
using std::flush;
using std::ifstream;
using std::map;
using std::ofstream;
using std::pair;
using std::string;
using std::stringstream;
using std::vector;
 
QGTextFileHandler::QGTextFileHandler()
{
}
 
QGTextFileHandler::~QGTextFileHandler()
{
}
 
bool QGTextFileHandler::ReadFile(string filename)
{
    fFileMap.clear();
    fFileLines.clear();
    ifstream file(filename.c_str());
    bool success = false;
    if (file) {
        success = true;
        string line;
        while (getline(file, line)) {
            pair<string, string> parsedLine = ParseLine(line);
            fFileMap.insert(parsedLine);
            fFileLines.push_back(parsedLine);
        }
    }
    file.close();
    return success;
}
 
bool QGTextFileHandler::WriteFile(string filename)
{
    ofstream file(filename.c_str());
    bool success = false;
    if (file) {
        success = true;
        vector<pair<string, string> >::const_iterator lineIter;
        for (lineIter = fFileLines.begin(); lineIter != fFileLines.end(); ++lineIter) {
            file << BuildLine(*lineIter) << endl;
        }
    }
    file.close();
    return success;
}
 
bool QGTextFileHandler::DoesKeyExist(string key)
{
    return (fFileMap.count(key) != 0);
}
 
string QGTextFileHandler::GetValue(string key)
{
    string value = "";
    if (DoesKeyExist(key)) {
        value = fFileMap[key];
    }
    return value;
}
 
void QGTextFileHandler::AppendLine(string key, string value)
{
    fFileLines.push_back(make_pair(key, value));
    fFileMap[key] = value;
}
 
void QGTextFileHandler::AppendLine(string key, Double_t value)
{
    stringstream valueStream;
    valueStream.precision(16);
    valueStream << value << flush;
    AppendLine(key, valueStream.str());
}
 
void QGTextFileHandler::AppendLine(string key, Int_t value)
{
    stringstream valueStream;
    valueStream << value << flush;
    AppendLine(key, valueStream.str());
}
 
pair<string, string> QGTextFileHandler::ParseLine(const string& line)
{
    string key, value;
    string::size_type delimiter = line.find_first_of(" "); // could add other delimiters
    if (delimiter != string::npos) {
        key = line.substr(0, delimiter);
        value = line.substr(delimiter + 1);
    } else {
        key = line;
        value = "";
    }
    return make_pair(key, value);
}
 
string QGTextFileHandler::BuildLine(const pair<string, string>& parsedLine)
{
    string line = parsedLine.first + " " + parsedLine.second;
    return line;
}