Plotter.h

Go to the documentation of this file.

#ifndef ConfigurableAnalysis_Plotter_H
#define ConfigurableAnalysis_Plotter_H

#include "FWCore/ParameterSet/interface/ParameterSet.h"

#include "TH1.h"
#include "TH1F.h"
#include "TH2F.h"
#include "TProfile.h"

#include "PhysicsTools/UtilAlgos/interface/VariableHelper.h"
#include "PhysicsTools/UtilAlgos/interface/CachingVariable.h"
#include "PhysicsTools/UtilAlgos/interface/ConfigurableHisto.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "CommonTools/Utils/interface/TFileDirectory.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"


class Plotter {
 public:
  Plotter(){}
  Plotter(const edm::ParameterSet& iConfig){}
  virtual ~Plotter(){}

  virtual void setDir(std::string dir) =0;
  virtual void fill(std::string subDir,const edm::Event& iEvent) =0;
  virtual void complete() =0;
};


class VariablePlotter : public Plotter {
 public:
  VariablePlotter(const edm::ParameterSet& iConfig) : currentDir_("youDidNotSetDirectoryFirst") {
    //create the master copy, never filled, just to make copies

    //    make TH1
    edm::ParameterSet th1=iConfig.getParameter<edm::ParameterSet>("TH1s");
    std::vector<std::string> th1Names;
    th1.getParameterSetNames(th1Names);
    for (unsigned int iH=0;iH!=th1Names.size();++iH){
      std::string hname = th1Names[iH];
      edm::ParameterSet hPset=th1.getParameter<edm::ParameterSet>(hname);
      if (hPset.exists("name")) hname = hPset.getParameter<std::string>("name");
      bool split=hPset.exists("splitter") || hPset.exists("splitters");
      if (split)
    master_[hname]=new SplittingConfigurableHisto(ConfigurableHisto::h1, hname, hPset);
      else
    master_[hname]=new ConfigurableHisto(ConfigurableHisto::h1, hname, hPset);
    }

    //    make profiles
    edm::ParameterSet tprof=iConfig.getParameter<edm::ParameterSet>("TProfiles");
    std::vector<std::string> tprofNames;
    tprof.getParameterSetNames(tprofNames);
    for (unsigned int iH=0;iH!=tprofNames.size();++iH){
      std::string hname = tprofNames[iH];
      edm::ParameterSet hPset=tprof.getParameter<edm::ParameterSet>(hname);
      bool split=hPset.exists("splitter") || hPset.exists("splitters");
      if (hPset.exists("name")) hname = hPset.getParameter<std::string>("name");
      if (split)
    master_[hname]=new SplittingConfigurableHisto(ConfigurableHisto::prof, hname, hPset);
      else
    master_[hname]=new ConfigurableHisto(ConfigurableHisto::prof, hname, hPset);
    }
    
    //    make TH2
    edm::ParameterSet th2=iConfig.getParameter<edm::ParameterSet>("TH2s");
    std::vector<std::string> th2Names;
    th2.getParameterSetNames(th2Names);
    for (unsigned int iH=0;iH!=th2Names.size();++iH){
      std::string hname = th2Names[iH];
      edm::ParameterSet hPset=th2.getParameter<edm::ParameterSet>(hname);
      if (hPset.exists("name")) hname = hPset.getParameter<std::string>("name");
      bool split=hPset.exists("splitter") || hPset.exists("splitters");
      if (split)
    master_[hname]=new SplittingConfigurableHisto(ConfigurableHisto::h2, hname, hPset);
      else
    master_[hname]=new ConfigurableHisto(ConfigurableHisto::h2, hname, hPset);
    }
  }

  void setDir(std::string dir) override{
    //insert a new one
    Directory & insertedDirectory = directories_[dir];

    //create the actual directory in TFile: name is <dir>
    if (!insertedDirectory.dir){
      insertedDirectory.dir=new TFileDirectory(edm::Service<TFileService>()->mkdir(dir));
      insertedDirectory.dirName=dir;
    }

    //remember which directory name this is
    currentDir_=dir;
  }
  
  void fill(std::string subDir,const edm::Event& iEvent) override{
    //what is the current directory
    Directory & currentDirectory= directories_[currentDir_];

    //what is the current set of sub directories for this
    SubDirectories & currentSetOfSubDirectories=currentDirectory.subDir;
    
    //find the subDirectory requested:
    SubDirectory * subDirectoryToUse=nullptr;
    SubDirectories::iterator subDirectoryFindIterator=currentSetOfSubDirectories.find(subDir);

    //not found? insert a new directory with this name
    if (subDirectoryFindIterator==currentSetOfSubDirectories.end()){
      edm::LogInfo("VariablePlotter")<<" gonna clone histograms for :"<<subDir<<" in "<<currentDir_;
      SubDirectory & insertedDir = currentSetOfSubDirectories[subDir];
      subDirectoryToUse = &insertedDir;
      if (!insertedDir.dir){
    insertedDir.dir=new TFileDirectory(currentDirectory.dir->mkdir(subDir));
    insertedDir.dirName=subDir;
      }

      //create a copy from the master copy
      DirectoryHistos::iterator masterHistogramIterator=master_.begin();
      DirectoryHistos::iterator masterHistogramIterator_end=master_.end();
      for (; masterHistogramIterator!=masterHistogramIterator_end;++masterHistogramIterator)
    {
      //clone does not book histogram
      insertedDir.histos[masterHistogramIterator->first]=masterHistogramIterator->second->clone();
    }
      
      //book all copies of the histos
      DirectoryHistos::iterator clonedHistogramIterator=insertedDir.histos.begin();
      DirectoryHistos::iterator clonedHistogramIterator_end=insertedDir.histos.end();
      for (; clonedHistogramIterator!=clonedHistogramIterator_end;++clonedHistogramIterator)
    {
      clonedHistogramIterator->second->book(insertedDir.dir);
    }
    }
    else{
      subDirectoryToUse=&subDirectoryFindIterator->second;
    }
    
    //now that you have the subdirectory: fill histograms for this sub directory
    DirectoryHistos::iterator histogramIterator=subDirectoryToUse->histos.begin();
    DirectoryHistos::iterator histogramIterator_end=subDirectoryToUse->histos.end();
    for(; histogramIterator!=histogramIterator_end;++histogramIterator)
      { histogramIterator->second->fill(iEvent); }
  }

  ~VariablePlotter() override{
    // CANNOT DO THAT because of TFileService holding the histograms
    /*    //loop over all subdirectories and delete all ConfigurableHistograms
      Directories::iterator dir_It = directories_.begin();
      Directories::iterator dir_It_end = directories_.end();
      // loop directories
      for (;dir_It!=dir_It_end;++dir_It){
      Directory & currentDirectory=dir_It->second;
      SubDirectories & currentSetOfSubDirectories=currentDirectory.subDir;
      SubDirectories::iterator subDir_It = currentSetOfSubDirectories.begin();
      SubDirectories::iterator subDir_It_end = currentSetOfSubDirectories.end();
      //loop subdirectories
      for (;subDir_It!=subDir_It_end;++subDir_It){
      DirectoryHistos::iterator histogramIterator=subDir_It->second.histos.begin();
      DirectoryHistos::iterator histogramIterator_end=subDir_It->second.histos.end();
      //loop configurable histograms
      for(; histogramIterator!=histogramIterator_end;++histogramIterator){
      // by doing that you are removing the histogram from the TFileService too. and this will crash
      //      delete histogramIterator->second;
      }
      }
      }
    */
  }
  void complete() override{
    
    //loop over all subdirectories and call complete() on all ConfigurableHistograms
    
    Directories::iterator dir_It = directories_.begin();
    Directories::iterator dir_It_end = directories_.end();
    // loop directories
    for (;dir_It!=dir_It_end;++dir_It){
      Directory & currentDirectory=dir_It->second;
      SubDirectories & currentSetOfSubDirectories=currentDirectory.subDir;
      SubDirectories::iterator subDir_It = currentSetOfSubDirectories.begin();
      SubDirectories::iterator subDir_It_end = currentSetOfSubDirectories.end();
      //loop subdirectories
      for (;subDir_It!=subDir_It_end;++subDir_It){
    DirectoryHistos::iterator histogramIterator=subDir_It->second.histos.begin();
    DirectoryHistos::iterator histogramIterator_end=subDir_It->second.histos.end();
    //loop configurable histograms
    for(; histogramIterator!=histogramIterator_end;++histogramIterator)
      { histogramIterator->second->complete(); }
      }
    }
  }

 private:
  typedef std::map<std::string, ConfigurableHisto *> DirectoryHistos;
  DirectoryHistos master_;

  class SubDirectory {
  public:
    SubDirectory() : dirName(""),dir(nullptr){}
    std::string dirName;
    DirectoryHistos histos;
    TFileDirectory * dir;
  };
  typedef std::map<std::string, SubDirectory> SubDirectories;

  class Directory {
  public:
    Directory() : dirName(""),dir(nullptr){}
    std::string dirName;
    SubDirectories subDir;
    TFileDirectory * dir;
  };
  typedef std::map<std::string, Directory> Directories;

  std::string currentDir_;
  Directories directories_;
};


#include "FWCore/PluginManager/interface/PluginFactory.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

typedef edmplugin::PluginFactory< Plotter* (const edm::ParameterSet&) > PlotterFactory;

#endif