ThrObj.hpp

Go to the documentation of this file.

#ifndef ROOT_TOOLS_THR_OBJ_HPP
#define ROOT_TOOLS_THR_OBJ_HPP
 
#include <vector>
#include <array>
#include <string>
 
#include "TFile.h"
#include "TH1.h"
#include "TH2.h"
#include "TH3.h"
 
#include "ROOT/TThreadedObject.hxx"

namespace ROOTTools
{
   namespace ThrObjHolder
   { 
      void Write();
      void Write(const std::string& outputFileName);
 
      // other functions and variables below are not intended for the user 
      // and are called/accessed automaticaly

      std::shared_ptr<TH1F> AddHistogram(ROOT::TThreadedObject<TH1F> *hist, 
                                         const std::string& directory);
      std::shared_ptr<TH2F> AddHistogram(ROOT::TThreadedObject<TH2F> *hist, 
                                         const std::string& directory);
      std::shared_ptr<TH3F> AddHistogram(ROOT::TThreadedObject<TH3F> *hist, 
                                         const std::string& directory);
      std::shared_ptr<TH1D> AddHistogram(ROOT::TThreadedObject<TH1D> *hist, 
                                         const std::string& directory);
      std::shared_ptr<TH2D> AddHistogram(ROOT::TThreadedObject<TH2D> *hist, 
                                         const std::string& directory);
      std::shared_ptr<TH3D> AddHistogram(ROOT::TThreadedObject<TH3D> *hist, 
                                         const std::string& directory);
      std::shared_ptr<TH1L> AddHistogram(ROOT::TThreadedObject<TH1L> *hist, 
                                         const std::string& directory);
      std::shared_ptr<TH2L> AddHistogram(ROOT::TThreadedObject<TH2L> *hist, 
                                         const std::string& directory);
      std::shared_ptr<TH3L> AddHistogram(ROOT::TThreadedObject<TH3L> *hist, 
                                         const std::string& directory);
      void Clear();
      void AddTFileDirectory(const std::string& directory, const int thrObjIndex);

      std::vector<std::unique_ptr<ROOT::TThreadedObject<TH1F>>> containerTH1F;
      std::vector<std::unique_ptr<ROOT::TThreadedObject<TH2F>>> containerTH2F;
      std::vector<std::unique_ptr<ROOT::TThreadedObject<TH3F>>> containerTH3F;
      std::vector<std::unique_ptr<ROOT::TThreadedObject<TH1D>>> containerTH1D;
      std::vector<std::unique_ptr<ROOT::TThreadedObject<TH2D>>> containerTH2D;
      std::vector<std::unique_ptr<ROOT::TThreadedObject<TH3D>>> containerTH3D;
      std::vector<std::unique_ptr<ROOT::TThreadedObject<TH1L>>> containerTH1L;
      std::vector<std::unique_ptr<ROOT::TThreadedObject<TH2L>>> containerTH2L;
      std::vector<std::unique_ptr<ROOT::TThreadedObject<TH3L>>> containerTH3L;

      std::vector<std::string> containerTFileDir;
      // indices of the TFile directories to the histograms
      std::array<std::vector<int>, 9> containerTFileDirIndex;
   };

   template<typename T>
   class ThrObj
   {
      public:
      ThrObj(const std::string& name, const std::string& title,
             const int xNBins, const double xLow, const double xMax,
             const std::string& fileDirectory = "");
      ThrObj(const std::string& name, const std::string& title,
             const int xNBins, const double xLow, const double xMax, 
             const int yNBins, const double yMin, const double yMax,
             const std::string& fileDirectory = "");
      ThrObj(const std::string& name, const std::string& title,
             const int xNBins, const double xLow, const double xMax, 
             const int yNBins, const double yMin, const double yMax,
             const int zNBins, const double zMin, const double zMax,
             const std::string& fileDirectory = "");
      std::shared_ptr<T> Get();
      protected:
      std::shared_ptr<T> objPtr;
   };
};
 
#endif /* ROOT_TOOLS_THR_OBJ_HPP */