QGlobalWriterDispatcher.cc

Go to the documentation of this file.

#include "QGlobalWriterDispatcher.hh"
#include "QGlobalRWFactory.hh"
#include "QGlobalWriter.hh"
#include "QMessage.hh"
#include "QError.hh"
 
using namespace Diana;
 
QGlobalWriterDispatcher& QGlobalWriterDispatcher::GetInstance()
{
    static QGlobalWriterDispatcher dispatcher;
    return dispatcher;
}
 
 
QGlobalWriterDispatcher::QGlobalWriterDispatcher() : QNamed("QGlobalWriterDispatcher")
{
    fLastWriter = NULL;
    fCurrentWriterFilename = "";
}
 
QGlobalWriter* QGlobalWriterDispatcher::GetWriter(const std::string& filename, QError& err)
{
    std::string fname = filename;
 
    // check for current writer
    bool isWriter = false;
    if(fname.find(Q_CURRENT_WRITER,0) != std::string::npos) {
        if(!fCurrentWriterFilename.empty())
            fname = fCurrentWriterFilename;
        isWriter = true;
    }
    // check for already open files
    /* deprecated
    if(fname == "same") {
        if(!fLastWriter) QMessageHandler::Error(GetName(), std::string("Cannot write to the \"same\" file, as none has been previously  well defined!"));
        return fLastWriter;
    } else */
    if(fWriters.find(fname) != fWriters.end()) {
        fLastWriter = fWriters[fname];
        return fLastWriter;
    }
 
    // Create the GlobalWriter for filename checking its extension
    fLastWriter = NULL;
    size_t dotpos =  fname.find_last_of(".");
    std::string fileExtension = "";
    if(dotpos != std::string::npos)
        fileExtension = fname.substr( dotpos );
 
 
    std::list<const QGlobalWriterFactory*> writerFactories = QGlobalRWFactory::GetInstance().GetListOfGlobalWriters();
    std::list<const QGlobalWriterFactory*>::iterator iter;
    for(iter = writerFactories.begin(); iter!= writerFactories.end(); iter++)
        if((*iter)->GetExt() == fileExtension) 
            fLastWriter = QGlobalRWFactory::GetInstance().CreateGlobalWriter((*iter)->GetName());
    // defaults
    if(fLastWriter == 0) {
        if(fileExtension == ".root") {
            QGlobalRWFactory::GetInstance().LoadPlugin("LRootGlobalWriter");
            QMessageHandler::Info(GetName(), std::string("loaded default g-writer LRootGlobalWriter"));
        }
        else if(fileExtension == ".txt") {
            QGlobalRWFactory::GetInstance().LoadPlugin("LASCIIGlobalWriter");
            QMessageHandler::Info(GetName(), std::string("loaded default g-writer LASCIIGlobalWriter"));
        }
        else if(fileExtension == ".gui") {
            QGlobalRWFactory::GetInstance().LoadPlugin("LGuiSessionWriter");
            QMessageHandler::Info(GetName(), std::string("loaded default g-writer LGUISessionWriter"));
        }
        std::list<const QGlobalWriterFactory*> writerFactories = QGlobalRWFactory::GetInstance().GetListOfGlobalWriters();
        for(iter = writerFactories.begin(); iter!= writerFactories.end(); iter++)
            if((*iter)->GetExt() == fileExtension) 
                fLastWriter = QGlobalRWFactory::GetInstance().CreateGlobalWriter((*iter)->GetName());
    }
 
 
    // no writer found
    if(fLastWriter == NULL){ 
        err = QERR_CANNOT_OPEN_FILE;
        err.SetDescription(GetName(),__LINE__,std::string("No writer found for extension: \"") + fileExtension + "\"" + " (file: "+ fname  +")");
        return NULL;
    }
 
    // open file and register it
    if(isWriter)
        err = fLastWriter->Open(fname,"CURRENT_WRITER");
    else
        err = fLastWriter->Open(fname);
        
    if(err != QERR_SUCCESS) {
        delete fLastWriter;
        return NULL;
    }
    fWriters[fname] = fLastWriter;
    return fLastWriter; 
}
 
QGlobalWriterDispatcher::~QGlobalWriterDispatcher()
{
    Clear();
}
 
 
void QGlobalWriterDispatcher::Clear()
{
    std::map<std::string, QGlobalWriter*>::iterator iter;
    for(iter=fWriters.begin(); iter!=fWriters.end(); iter++) {
        QGlobalWriter* writer = iter->second;
        if(writer) {
            writer->Close();
            delete writer;
            iter->second = 0;
        }
    }    
    fWriters.clear();
    fLastWriter = NULL;
}