L1EventTreeProducer.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    L1TriggerDPG/L1Ntuples
// Class:      L1EventTreeProducer
// 
//
// Original Author:  
//         Created:  
// $Id: L1EventTreeProducer.cc,v 1.8 2012/08/29 12:44:03 jbrooke Exp $
//
//


// system include files
#include <memory>

// framework
#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/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

// data formats
#include "DataFormats/L1Trigger/interface/EGamma.h"
#include "DataFormats/L1Trigger/interface/Tau.h"
#include "DataFormats/L1Trigger/interface/Jet.h"
#include "DataFormats/L1Trigger/interface/Muon.h"
#include "DataFormats/L1Trigger/interface/EtSum.h"

// ROOT output stuff
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
#include "TTree.h"

#include "L1Trigger/L1TNtuples/interface/L1AnalysisEvent.h"

//
// class declaration
//

class L1EventTreeProducer : public edm::EDAnalyzer {
public:
  explicit L1EventTreeProducer(const edm::ParameterSet&);
  ~L1EventTreeProducer();
  
  
private:
  virtual void beginJob(void) ;
  virtual void analyze(const edm::Event&, const edm::EventSetup&);
  virtual void endJob();

public:
  
  L1Analysis::L1AnalysisEvent* l1Event;
  L1Analysis::L1AnalysisEventDataFormat * l1EventData;

private:

  // output file
  edm::Service<TFileService> fs_;
  
  // tree
  TTree * tree_;

  edm::EDGetTokenT<edm::TriggerResults> hltSource_;
  
  // EDM input tags
  //edm::EDGetTokenT<l1t::EGammaBxCollection> egToken_;

};



L1EventTreeProducer::L1EventTreeProducer(const edm::ParameterSet& iConfig)
{
  hltSource_           = consumes<edm::TriggerResults>(iConfig.getParameter<edm::InputTag>("hltSource"));
  
  std::string puMCFile   = iConfig.getUntrackedParameter<std::string>("puMCFile", "");
  std::string puMCHist   = iConfig.getUntrackedParameter<std::string>("puMCHist", "pileup");
  std::string puDataFile = iConfig.getUntrackedParameter<std::string>("puDataFile", "");
  std::string puDataHist = iConfig.getUntrackedParameter<std::string>("puDataHist", "pileup");

  bool useAvgVtx          = iConfig.getUntrackedParameter<bool>("useAvgVtx", true);
  double maxAllowedWeight = iConfig.getUntrackedParameter<double>("maxAllowedWeight", -1);
 
  l1Event     = new L1Analysis::L1AnalysisEvent(puMCFile, puMCHist, 
                              puDataFile, puDataHist,
                              useAvgVtx, maxAllowedWeight,consumesCollector());
  l1EventData = l1Event->getData();
  
  // set up output
  tree_=fs_->make<TTree>("L1EventTree", "L1EventTree");
  tree_->Branch("Event", "L1Analysis::L1AnalysisEventDataFormat", &l1EventData, 32000, 3);
}


L1EventTreeProducer::~L1EventTreeProducer()
{
 
  // do anything here that needs to be done at desctruction time
  // (e.g. close files, deallocate resources etc.)
  delete l1Event;
  
}


//
// member functions
//

// ------------ method called to for each event  ------------
void
L1EventTreeProducer::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
{
  if(!hltSource_.isUninitialized()) {
    l1Event->Reset();
    l1Event->Set(iEvent,hltSource_);
  }
  tree_->Fill();
}

// ------------ method called once each job just before starting event loop  ------------
void 
L1EventTreeProducer::beginJob(void)
{
}

// ------------ method called once each job just after ending the event loop  ------------
void 
L1EventTreeProducer::endJob() {
}

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