Benchmark.cc

Go to the documentation of this file.

#include "DQMOffline/PFTau/interface/Benchmark.h"

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

#include "DQMServices/Core/interface/MonitorElement.h"
#include "DQMServices/Core/interface/DQMStore.h"

#include <TDirectory.h>


using namespace std;

DQMStore* Benchmark::DQM_ = 0;

Benchmark::~Benchmark() {

}

void Benchmark::setDirectory(TDirectory* dir) {
  dir_ = dir;
  DQM_ = 0; 
} 


TH1F* Benchmark::book1D(const char* histname, const char* title, 
            int nbins, float xmin, float xmax) {
  // DQM_ has to be initialized in a child analyzer.
  if(DQM_) {
    edm::LogInfo("Benchmark") << " Benchmark::book1D " << "booking "<<histname;
    return DQM_->book1D(histname,title,nbins,xmin, xmax)->getTH1F();
  }
  else if(dir_){
    TDirectory *oldpwd = gDirectory; 
    dir_->cd();
    TH1F *hist =  new TH1F(histname, title, nbins, xmin, xmax);
    cout<<"booking (ROOT mode) "<<histname<<" in "<<dir_->GetName()<<endl;
    oldpwd->cd();
    return hist;
  }
  else assert(0);
}

TH2F* Benchmark::book2D(const char* histname, const char* title, 
            int nbinsx, float xmin, float xmax,
            int nbinsy, float ymin, float ymax ) {
  // DQM_ has to be initialized in a child analyzer.
  if(DQM_) {
    edm::LogInfo("Benchmark") << " Benchmark::book2D "<<"booked "<<histname;
    return DQM_->book2D(histname,title,nbinsx,xmin, xmax, nbinsy, ymin, ymax)->getTH2F();
  }
  else if(dir_) {
    TDirectory *oldpwd = gDirectory; 
    dir_->cd();
    TH2F *hist = new TH2F(histname, title, nbinsx, xmin, xmax, nbinsy, ymin, ymax);
    cout<<"booked (ROOT mode) "<<histname<<" in "<<dir_->GetName()<<endl;
    oldpwd->cd();
    return hist;
  }
  else assert(0);
}

TH2F* Benchmark::book2D(const char* histname, const char* title, 
            int nbinsx, float* xbins,
            int nbinsy, float ymin, float ymax ) {
  if(DQM_) {
    edm::LogInfo("Benchmark") << " Benchmark::book2D " << " booked "<<histname;
    
    // need to build the y bin array manually, due to a missing function in DQMStore
    vector<float> ybins( nbinsy+1 );
    double binsize = (ymax - ymin) / nbinsy;
    for(int i=0; i<=nbinsy; ++i) {
      ybins[i] = ymin + i*binsize;
    } 
    
    return DQM_->book2D(histname,title,nbinsx, xbins, nbinsy, &ybins[0])->getTH2F();
  }
  else if(dir_) {
    TDirectory *oldpwd = gDirectory; 
    dir_->cd();

    // need to convert the float bin array into a double bin array, because the DQMStore functions take floats, while the ROOT functions take double. 
    vector<double> xbinsd(nbinsx+1); 
    for(int i=0; i<=nbinsx; ++i) {
      xbinsd[i] = xbins[i];
    }

    TH2F *hist = new TH2F(histname, title, nbinsx, &xbinsd[0], nbinsy, ymin, ymax);
    cout<<"booked (ROOT mode) "<<histname<<" in "<<dir_->GetName()<<endl;
    oldpwd->cd();
    return hist;
  }
  else assert(0);
}

TProfile* Benchmark::bookProfile(const char* histname, const char* title,
                 int nbinsx, float xmin, float xmax,
                 float ymin, float ymax, const char* option ) {
  // DQM_ has to be initialized in a child analyzer.
  if(DQM_) {
    edm::LogInfo("Benchmark") << " Benchmark::bookProfile "<<"booked "<<histname;
    return DQM_->bookProfile(histname, title, nbinsx, xmin, xmax, 0.0, 0.0, option )->getTProfile();
  }
  else if(dir_) {
    TDirectory *oldpwd = gDirectory;
    dir_->cd();
    TProfile *hist = new TProfile(histname, title, nbinsx, xmin, xmax, ymin, ymax, option);
    cout<<"booked (ROOT mode) "<<histname<<" in "<<dir_->GetName()<<endl;
    oldpwd->cd();
    return hist;
  }
  else assert(0);
}

TProfile* Benchmark::bookProfile(const char* histname, const char* title,
                 int nbinsx, float* xbins,
                 float ymin, float ymax, const char* option ) {

  // need to convert the float bin array into a double bin array, because the DQMStore TProfile functions take floats, while the  DQMStore TH2 functions take double. 
  vector<double> xbinsd(nbinsx+1); 
  for(int i=0; i<=nbinsx; ++i) {
    xbinsd[i] = xbins[i];
  }

  // DQM_ has to be initialized in a child analyzer.
  if(DQM_) {
    edm::LogInfo("Benchmark") << " Benchmark::bookProfile "<<"booked "<<histname;
    return DQM_->bookProfile(histname, title, nbinsx, &xbinsd[0], ymin, ymax, option)->getTProfile();
  }
  else if(dir_) {
    TDirectory *oldpwd = gDirectory;
    dir_->cd();
    TProfile *hist = new TProfile(histname, title, nbinsx, &xbinsd[0], ymin, ymax, option);
    cout<<"booked (ROOT mode) "<<histname<<" in "<<dir_->GetName()<<endl;
    oldpwd->cd();
    return hist;
  }
  else assert(0);
}


void Benchmark::write() {
  //COLIN not sure about the root mode 
  if( dir_ )
    dir_->Write();

  //COLIN remove old bullshit:
//   if ( ame.size() != 0 && file_)
//     cout<<"saving histograms in "<<fileName<<endl;
//     file_->Write(fileName.c_str());
}