TH1Store.h

Go to the documentation of this file.

// -*- C++ -*-

#if !defined(TH1Store_H)
#define TH1Store_H

#include <map>
#include <string>
#include <set>

#include "TH1.h"
#include "TFile.h"
#include "TString.h"
#include "TDirectory.h"

class TH1Store
{
   public:

      // Public Constants //

      typedef std::vector< std::string >            SVec;
      typedef std::map< std::string, std::string >  SSMap;
      typedef std::map< std::string, TH1* >         STH1PtrMap;
      typedef SSMap::const_iterator                 SSMapConstIter;
      typedef STH1PtrMap::iterator                  STH1PtrMapIter;
      typedef STH1PtrMap::const_iterator            STH1PtrMapConstIter;

      static const SVec kEmptyVec;

      // friends //
      // tells particle data how to print itself out
      friend std::ostream& operator<< (std::ostream& o_stream, 
                                       const TH1Store &rhs);

      //            _         //
      // |\/|      |_         //
      // |  |EMBER | UNCTIONS //
      //                      //

      // Constructors and Destructor //
      TH1Store();
      ~TH1Store();

      // One Liners //
      // Whether or not to delete histogram pointers on destruction
      void setDeleteOnDestruction (bool deleteOnDestruction = true) 
      { m_deleteOnDestruction = deleteOnDestruction; }

      // Regular Member Functions //

      // adds a histogram pointer to the map
      void add (TH1 *histPtr, const std::string &directory = "");

      // given a string, returns corresponding histogram pointer
      TH1* hist (const std::string &name);
      TH1* hist (const char* name)    
      { return hist( (const std::string) name ); }
      TH1* hist (const TString &name) 
      { return hist( (const char*) name ); }

      // write all histograms to a root file
      void write (const std::string &filename, 
                  const SVec &argsVec = kEmptyVec, 
                  const SVec &inputFilesVec = kEmptyVec) const;
      void write (TFile *filePtr, 
                  const SVec &argsVec = kEmptyVec, 
                  const SVec &inputFilesVec = kEmptyVec) const;

      // Static Member Functions //

      // turn on verbose messages (e.g., printing out histogram names
      // when being made)
      static void setVerbose (bool verbose = true)
      { sm_verbose = verbose; }

  private:

      // Private Member Functions //

      // creates directory and all parent directories as needed
      // (equivalent to unix 'mkdir -p') and then changes (cd's) to
      // that directory.  Returns TDirectory of pointing to dirname.
      TDirectory* _createDir (const std::string &dirname, TFile *filePtr) const;

      // Private Member Data //

      bool       m_deleteOnDestruction;
      STH1PtrMap m_ptrMap;
      SSMap      m_nameDirMap;
      
      // Private Static Member Data //

      static bool sm_verbose;      

};


#endif // TH1Store_H