RootOutputFileManager.h

Go to the documentation of this file.

#ifndef ROOTOUTPUTFILEMANAGER
#define ROOTOUTPUTFILEMANAGER

#include "RootIOSvc/RootIOIncident.h"

#include <map>
#include <string>

class TFile;
class IIncidentSvc;

class RootOutputFileManager
{
public:
    RootOutputFileManager();
    virtual ~RootOutputFileManager();
    
    // singleton
    static RootOutputFileManager& get();

    // Set the IncidentSvc
    void setIncidentSvc(IIncidentSvc* incsvc);

    typedef std::map<std::string,std::pair<int,TFile*> > FileMap;

    // helpers
    TFile*& file(FileMap::iterator& it);
    int& count(FileMap::iterator& it);

    // Get TFile of given name, increase file's ref count
    TFile* get_file(const std::string& filename);

    // Decrease file's ref count, really close when it hits zero.
    // When a file is actually closed a Gaudi Incident will be emitted
    // just before and just afterwards.
    void close_file(TFile*& tfile);

private:
    FileMap m_filemap;
    IIncidentSvc* m_incsvc;

    void fire(const std::string& filename, RootIOIncident::FileState state);
};

#endif  // ROOTOUTPUTFILEMANAGER