PlotCombiner.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DQMServices/Core/interface/DQMStore.h"
#include "DQMServices/Core/interface/MonitorElement.h"

#include "TH1F.h"
#include "TH2F.h"

using namespace edm;
using namespace std;

class PlotCombiner : public edm::EDAnalyzer{
  public:
    PlotCombiner(const edm::ParameterSet& pset);
    virtual ~PlotCombiner();
    virtual void analyze(const edm::Event& event, const edm::EventSetup& eventSetup) {};
    virtual void endRun(const edm::Run &, const edm::EventSetup &);
  private:
    void makePlot(const ParameterSet& pset);
    DQMStore * dqmStore;
    string myDQMrootFolder;
    const VParameterSet plots;
};

PlotCombiner::PlotCombiner(const edm::ParameterSet& pset):
  myDQMrootFolder( pset.getUntrackedParameter<string>("MyDQMrootFolder") ),
  plots( pset.getUntrackedParameter<VParameterSet>("Plots") )
{
}

void PlotCombiner::endRun(const edm::Run &, const edm::EventSetup &){
  dqmStore = Service<DQMStore>().operator->();
  if( !dqmStore ){
    LogError("HLTriggerOfflineHeavyFlavor") << "Could not find DQMStore service\n";
    return;
  }
  for(VParameterSet::const_iterator pset = plots.begin(); pset!=plots.end(); pset++){
    makePlot(*pset);
  }
}
  
void PlotCombiner::makePlot(const ParameterSet& pset){
//get hold of MEs
  vector<string> inputMEnames = pset.getUntrackedParameter<vector<string> >("InputMEnames");
  vector<string> inputLabels = pset.getUntrackedParameter<vector<string> >("InputLabels");
  if( inputMEnames.size() != inputLabels.size() ){
    LogDebug("HLTriggerOfflineHeavyFlavor") << "Number of labels must match the histos[0]ber of InputMEnames"<<endl;
    return;
  }
  vector<TH1*> histos;
  vector<TString> labels;
  for(size_t i=0; i<inputMEnames.size(); i++){
    string MEname = myDQMrootFolder+"/"+inputMEnames[i];
    MonitorElement *ME = dqmStore->get(MEname);
    if(ME==0){
      LogDebug("HLTriggerOfflineHeavyFlavor") << "Could not find ME: "<<MEname<<endl;
      continue;
    }
    histos.push_back( ME->getTH1() );
    labels.push_back( inputLabels[i] );
  }
  if(histos.size()==0){
    return;
  }
  //figure out the output directory name
  string outputMEname = pset.getUntrackedParameter<string>("OutputMEname");;
  string outputDir = myDQMrootFolder;
  string::size_type slashPos = outputMEname.rfind('/');
  if ( string::npos != slashPos ) {
    outputDir += "/"+outputMEname.substr(0, slashPos);
    outputMEname.erase(0, slashPos+1);
  }
  dqmStore->setCurrentFolder(outputDir);
  //create output ME
  TH2F * output;
  if(histos[0]->GetXaxis()->GetXbins()->GetSize()==0){
    output = new TH2F(outputMEname.c_str(),outputMEname.c_str(),histos[0]->GetXaxis()->GetNbins(),histos[0]->GetXaxis()->GetXmin(),histos[0]->GetXaxis()->GetXmax(),histos.size(),0,histos.size());
  }else{
    output = new TH2F(outputMEname.c_str(),outputMEname.c_str(),histos[0]->GetXaxis()->GetNbins(),histos[0]->GetXaxis()->GetXbins()->GetArray(),histos.size(),0,histos.size());
  }
  output->SetTitle(outputMEname.c_str());
  output->SetXTitle( histos[0]->GetXaxis()->GetTitle() );
  output->SetStats(kFALSE);
  output->SetOption("colztexte");
  for(size_t i=0; i<histos.size(); i++){
    for(int j=1; j<=histos[0]->GetNbinsX(); j++){
      output->SetBinContent(j,i+1,histos[i]->GetBinContent(j));
      output->SetBinError(j,i+1,histos[i]->GetBinError(j));
    }
    output->GetYaxis()->SetBinLabel(i+1,labels[i]);
  }
  dqmStore->book2D(outputMEname,output);
  delete output;
}

PlotCombiner::~PlotCombiner(){
}

//define this as a plug-in
DEFINE_FWK_MODULE(PlotCombiner);