HistMan.h

Go to the documentation of this file.

#ifndef HISTMAN_HISTMAN_H
#define HISTMAN_HISTMAN_H

#include <TFolder.h>
#include <TH1.h>
#include <vector>
//#include <HistMan/RootHists.h>        // needed for inside Cint.cc file

class TFile;

// Helper. Acts like "mkdir -p" to add a full path with 1 or more
// subdirectories.  Return last folder.  This should be in TFolder....
TFolder& mkdir_p(TFolder& folder, const char* path);

class HistMan {
    TFolder* fFolder;
    bool fOwn;


public:
    HistMan(const char* base_directory = "");

    HistMan(TFile& file, bool attach = true);

    HistMan(TFolder* folder, bool own=true);

    HistMan(const char** file_list, const char** hist_list);
    
    ~HistMan();

    TFolder& BaseFolder();

    void RegisterWithRoot();

    void WriteOut(TFile& opened_file);

    void WriteOut(const char* filename);

    TObject* Adopt(const char* path, TObject* hist);

    template<class THType>
    THType* Get(const char* pathname) {
        return dynamic_cast<THType*>(this->GetObject(pathname));
    }
        

    TObject* GetObject(const char* pathname);
    TFolder* GetFolder(const char* pathname);
    TObject* GetObjectOrFolder(const char* pathname);
    std::vector<TObject*> GetObjects(const char* pathname);
    std::vector<TFolder*> GetSubFolders(const char* pathname);


    template<class THType>
    THType* Book(const char* name, const char* title,
                 int nbinsx, Axis_t xmin, Axis_t xmax,
                 const char* path=".", Bool_t sumw2=kFALSE) {
        THType* h = new THType(name,title,nbinsx,xmin,xmax);
        if ( sumw2 ) {
            h->Sumw2();
        }
        TObject* o = Adopt(path, h);
        return dynamic_cast<THType*>(o);
    }

    template<class THType>
    THType* Book(const char* name, const char* title,
                 int nbinsx, Axis_t xmin, Axis_t xmax,
                 int nbinsy, Axis_t ymin, Axis_t ymax,
                 const char* path=".", Bool_t sumw2=kFALSE) {
        THType* h = new THType(name,title,nbinsx,xmin,xmax, nbinsy,ymin,ymax);
        if ( sumw2 ) {
            h->Sumw2();
        }
        TObject* o = Adopt(path, h);
        return dynamic_cast<THType*>(o);
    }

    bool Fill1d(const char* pathname, Axis_t x, Stat_t w=1.0);

    bool Fill2d(const char* pathname, Axis_t x, Axis_t y, Stat_t w=1.0);

    bool FillProfile(const char* pathname, Axis_t x, Axis_t y, Stat_t w=1.0);

};



#endif  // HISTMAN_HISTMAN_H