ExpressionHisto.h

Go to the documentation of this file.

#ifndef CommonTools_Utils_ExpressionHisto_h
#define CommonTools_Utils_ExpressionHisto_h
// -*- C++ -*-
//
// Package:     UtilAlgos
// Class  :     ExpressionHisto
// 
//
// Original Author: Benedikt HEGNER
//         Created:  Fri Jun  1 14:35:22 CEST 2007
// $Id: ExpressionHisto.h,v 1.1 2010/05/04 10:36:06 hegner Exp $
//

// system include files

// user include files
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "CommonTools/Utils/interface/TFileDirectory.h"
#include "CommonTools/Utils/interface/StringObjectFunction.h"

#include "TFile.h"
#include "TH1F.h"
#include "TH1.h"

template<typename T>
class ExpressionHisto {
public:
  ExpressionHisto(const edm::ParameterSet& iConfig);
  ~ExpressionHisto();
  
  void initialize(TFileDirectory& fs);
  bool fill(const T& element, double weight=1.0, uint32_t i=0);  
  
private:
  double min, max;
  int nbins;
  std::string name, description;
  uint32_t nhistos;
  bool     separatePlots;
  TH1F ** hist;
  StringObjectFunction<T> function;      
};

template<typename T>
ExpressionHisto<T>::ExpressionHisto(const edm::ParameterSet& iConfig):
  min(iConfig.template getUntrackedParameter<double>("min")),
  max(iConfig.template getUntrackedParameter<double>("max")),
  nbins(iConfig.template getUntrackedParameter<int>("nbins")),
  name(iConfig.template getUntrackedParameter<std::string>("name")),
  description(iConfig.template getUntrackedParameter<std::string>("description")),
  function(iConfig.template getUntrackedParameter<std::string>("plotquantity"), 
           iConfig.template getUntrackedParameter<bool>("lazyParsing", false)) {
  int32_t itemsToPlot = iConfig.template getUntrackedParameter<int32_t>("itemsToPlot", -1);
  if (itemsToPlot <= 0) {
      nhistos = 1; separatePlots = false;
  } else {
      nhistos = itemsToPlot; separatePlots = true;
  }
}

template<typename T>
ExpressionHisto<T>::~ExpressionHisto() {
}

template<typename T>
void ExpressionHisto<T>::initialize(TFileDirectory& fs) 
{
   hist = new TH1F*[nhistos];
   char buff[1024],baff[1024];
   if (separatePlots) {
       for (uint32_t i = 0; i < nhistos; i++) {
               if (strstr(name.c_str(), "%d") != 0) {
                       snprintf(buff, 1024, name.c_str(), i+1);
               } else {
                       snprintf(buff, 1024, "%s [#%d]", name.c_str(), i+1);
               }
               if (strstr(description.c_str(), "%d") != 0) {
                       snprintf(baff, 1024, description.c_str(), i+1);
               } else {
                       snprintf(baff, 1024, "%s [#%d]", description.c_str(), i+1);
               }
               hist[i] = fs.make<TH1F>(buff,baff,nbins,min,max);
       }
    } else {
       hist[0] = fs.make<TH1F>(name.c_str(),description.c_str(),nbins,min,max);
    }
}

template<typename T>
bool ExpressionHisto<T>::fill(const T& element, double weight, uint32_t i) 
{
  if (!separatePlots) hist[0]->Fill( function(element), weight );
  else if (i < nhistos)  hist[i]->Fill( function(element), weight );
  else return false;
  return true;
}

#endif