LASCIIGlobalWriter.cc

Go to the documentation of this file.

#include "LASCIIGlobalWriter.hh"
#include "QMatrix.hh"
#include "QBaseType.hh"
#include "QString.hh"
#include "QBool.hh"
#include <TClass.h>
#include <sstream>
 
REGISTER_GLOBAL_WRITER(LASCIIGlobalWriter, ".txt")
 
using namespace Diana;
using std::string;
using std::stringstream;
using std::vector;
 
LASCIIGlobalWriter::LASCIIGlobalWriter() : QGlobalWriter("ASCIIGlobalWriter")
{
}
 
LASCIIGlobalWriter::~LASCIIGlobalWriter()
{
    Close();
}
 
 
void LASCIIGlobalWriter::SetInt(const std::string& name, int obj)
{
        string begintag = "begin " + name;
        
        stringstream ss;
        ss << obj;
        
        fFile->Comment(begintag);
        fFile->WriteLine(ss.str());
}
 
void LASCIIGlobalWriter::SetDouble(const std::string& name, double obj)
{
        string begintag = "begin " + name;
        
        stringstream ss;
        
        ss.precision(16);
        
        ss << obj;
        
        fFile->Comment(begintag);
        fFile->WriteLine(ss.str());
}
 
void LASCIIGlobalWriter::SetBool(const std::string& name, bool obj)
{
        string begintag = "begin " + name;
        string endtag = "end " + name;
        
        stringstream ss;
        ss << obj;
        
        fFile->Comment(begintag);
        fFile->WriteLine(ss.str());
}
 
 
void LASCIIGlobalWriter::SetFloat(const std::string& name, float obj)
{
        string begintag = "begin " + name;
        
        stringstream ss;
        ss << obj;
 
        fFile->Comment(begintag);
        fFile->WriteLine(ss.str());
}
 
void LASCIIGlobalWriter::SetString(const std::string& name, const std::string& obj)
{
        string begintag = "begin " + name;
        string endtag = "end " + name;
        
        fFile->Comment(begintag);
        fFile->WriteLine(obj);
        fFile->Comment(endtag);
}
 
QError LASCIIGlobalWriter::SetQObject(const std::string& name, const std::string& className, const Diana::QObject* obj, const std::string& descr)
{
    QError err = QERR_SUCCESS;
    if(obj==NULL) {
        err = QERR_UNKNOWN_ERR;
        stringstream msg;
        msg<<"Error while writing object \""<< name <<"\": "<<" null pointer";
        err.SetDescription(__FILE__,__LINE__,msg.str());
        return err;
    }
 
    std::string type = className;
    if(type == "Diana::QVector") 
    {
        const QVector* qvec = (const QVector*)(obj);
        SetQVector(name,qvec);
    }
    else if(type == "Diana::QVectorC")
    {
        const QVectorC* qvec = (const QVectorC*)(obj);
        SetQVectorC(name,qvec);
    }
    else if(type == "Diana::QTime")
    {
        const QTime* qtime = (const QTime*)(obj);
        SetQTime(name,qtime);
    }   
    else if(type == "Diana::QMatrix")
    {
        const QMatrix* qmat = (const QMatrix*)(obj);
        SetQMatrix(name,qmat,descr);
    }
    else if(type == "Diana::QBaseType<int>")
    {
        const QInt* qint = (const QInt*)(obj);
        SetInt(name,*qint);
    }
    else if(type == "Diana::QBaseType<float>")
    {
        const QFloat* qint = (const QFloat*)(obj);
        SetFloat(name,*qint);
    }
    else if(type == "Diana::QBaseType<double>")
    {
        const QDouble* qint = (const QDouble*)(obj);
        SetDouble(name,*qint);
    }
 
    else if(type == "Diana::QBool")
    {
        const QBool* qint = (const QBool*)(obj);
        SetBool(name,*qint);
    }
    else if(type == "Diana::QString")
    {
        const QString* qint = (const QString*)(obj);
        SetString(name,*qint);
    }
    else 
    {
        string begintag = "begin " + name;
        string endtag = "end " + name;
 
        fFile->Comment(begintag);
        fFile->Comment(type);
        fFile->Comment("Automatic dump via QObject::WriteOnStream()");
        std::stringstream str;
        err = obj->WriteOnStream(str);
        fFile->WriteString(str.str(), true);
        fFile->Comment(endtag);
    }
    if(err != QERR_SUCCESS) {
        stringstream msg;
        msg<<"Error while writing object \""<< name <<"\": "<<err.GetDescription();
        err.SetDescription(__FILE__,__LINE__,msg.str());
    }
 
    return err;
}
 
void LASCIIGlobalWriter::SetQVector(const std::string&name, const QVector* obj)
{
    string begintag = "begin " + name;
    string endtag = "end " + name;
 
    fFile->Comment(begintag);
    fFile->Comment("QVector");
    fFile->WriteQVector(*obj, true);
    fFile->Comment(endtag);
 
}
 
void LASCIIGlobalWriter::SetQVectorC(const std::string&name, const QVectorC* obj)
{
    string begintag = "begin " + name;
    string endtag = "end " + name;
 
    vector<QVector> qvecs;
 
    qvecs.push_back(obj->Re());
    qvecs.push_back(obj->Im());
 
    fFile->Comment(begintag);
    fFile->Comment("QVectorC");
    fFile->WriteQVectors(qvecs, true);
    fFile->Comment(endtag);
 
}
 
void LASCIIGlobalWriter::SetQTime(const std::string&name, const QTime* obj)
{
    string begintag = "begin " + name;
    string endtag = "end " + name;
    QTime qt = *obj;
 
    fFile->Comment(begintag);
    fFile->Comment("QTime");
    fFile->WriteQTime(&qt, true);
    fFile->Comment(endtag);     
}
 
QError LASCIIGlobalWriter::Open(const std::string& filename, const std::string& opt)
{
    QError err = QERR_SUCCESS;
    fFile = new QASCII(filename); 
    fFile->WriteString(""); // Clear file.  Will overwrite existing data!!
    if(!fFile->Exists()) {
        err = QERR_CANNOT_OPEN_FILE;
        err.SetDescription(__FILE__,__LINE__,filename);
        delete fFile;
        fFile = NULL;
    }
    return err;
}
 
void LASCIIGlobalWriter::SetQMatrix(const std::string&name, const QMatrix* obj, const std::string& descr)
{
    string begintag = "begin " + name;
    string endtag = "end " + name;
 
    vector<QVector> qvecs;
 
    int ncol = obj->GetNCol();
    int nrow = obj->GetNRow();
 
    stringstream ss;    
    ss << nrow << " " << ncol;
 
    for(int i=0; i<ncol; i++)
        qvecs.push_back(obj->GetCol(i));
 
    fFile->Comment(begintag);
    fFile->Comment("QMatrix");
    fFile->Comment(ss.str());
    if(!descr.empty()) fFile->Comment(descr);
    fFile->WriteQVectors(qvecs, true);
    fFile->Comment(endtag);
 
}
 
QError LASCIIGlobalWriter::Close()
{
    QError err = QERR_SUCCESS;
    delete fFile;
    fFile = NULL;
    return err;
}