In This Package:

RootOutputFileManager.cc

Go to the documentation of this file.

#include "RootOutputFileManager.h"
#include "RootIOSvc/RootIOIncident.h"
#include "GaudiKernel/IIncidentSvc.h"

#include "TFile.h"

using namespace std;

RootOutputFileManager::RootOutputFileManager()
    : m_incsvc(0)
{
}

RootOutputFileManager::~RootOutputFileManager()
{
    if (m_incsvc) {
        m_incsvc->release();
    }
    m_incsvc = 0;
}

RootOutputFileManager& RootOutputFileManager::get()
{
    static RootOutputFileManager* me = 0;
    if (!me) me = new RootOutputFileManager;
    return *me;
}


void RootOutputFileManager::setIncidentSvc(IIncidentSvc* incsvc)
{
    if (m_incsvc) {
        m_incsvc->release();
        m_incsvc = 0;
    }
    m_incsvc = incsvc;
    if (m_incsvc) {
        m_incsvc->addRef();
    }
}

TFile*& RootOutputFileManager::file(FileMap::iterator& it) 
{
    return it->second.second; 
}

int& RootOutputFileManager::count(FileMap::iterator& it) 
{
    return it->second.first; 
}

void RootOutputFileManager::fire(const std::string& filename, RootIOIncident::FileState state)
{
    if (!m_incsvc) {
        return;                 // error message?
    }

    m_incsvc->fireIncident(RootIOIncident("RootOutputFileManager","RootIOIncident",filename,state));    
}

TFile* RootOutputFileManager::get_file(const string& filename) {
    FileMap::iterator it = m_filemap.find(filename);
    if (it == m_filemap.end()) {  // miss
 
        TFile* file = TFile::Open(filename.c_str(),"recreate","RootOutputStream file");
        m_filemap[filename] = pair<int,TFile*>(1,file);
        this->fire(filename,RootIOIncident::opened_output);
        return file;
    }
    // hit
    ++count(it);
    return file(it);
}

void RootOutputFileManager::close_file(TFile*& tfile) {
    string filename = tfile->GetName();
    tfile = 0;
    FileMap::iterator it = m_filemap.find(filename);
    if (it == m_filemap.end()) {  // miss, already closed?
        return;
    }

    --count(it);
    if (count(it)) return;  // still more references

    this->fire(filename,RootIOIncident::closing_output);
    file(it)->Close();
    this->fire(filename,RootIOIncident::closed_output);
    delete file(it);
    file(it) = 0;
    m_filemap.erase(it);
}

---