AnalysisRootpleProducer.h

Go to the documentation of this file.

#ifndef AnalysisRootpleProducer_H
#define AnalysisRootpleProducer_H

#include <iostream>

#include <FWCore/Framework/interface/Event.h>
#include <FWCore/Framework/interface/ESHandle.h>
#include <FWCore/Framework/interface/MakerMacros.h>
#include <FWCore/Framework/interface/Frameworkfwd.h>
#include <FWCore/ParameterSet/interface/ParameterSet.h>
#include <DataFormats/Common/interface/Handle.h>

#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "SimDataFormats/GeneratorProducts/interface/HepMCProduct.h"

#include <FWCore/ServiceRegistry/interface/Service.h>
#include <CommonTools/UtilAlgos/interface/TFileService.h>

#include <TROOT.h>
#include <TTree.h>
#include <TFile.h>
#include <TLorentzVector.h>
#include <TObjString.h>
#include <TClonesArray.h>

#include "DataFormats/JetReco/interface/Jet.h"
#include "DataFormats/JetReco/interface/GenJet.h"
#include "DataFormats/JetReco/interface/GenJetCollection.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"
#include "DataFormats/JetReco/interface/BasicJet.h"
#include "DataFormats/JetReco/interface/BasicJetCollection.h"
#include "DataFormats/Candidate/interface/Candidate.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"

// access trigger results
#include <DataFormats/Common/interface/TriggerResults.h>
#include <DataFormats/HLTReco/interface/TriggerEvent.h>
#include <DataFormats/HLTReco/interface/TriggerFilterObjectWithRefs.h>

class AnalysisRootpleProducer : public edm::EDAnalyzer {
public:
  //
  explicit AnalysisRootpleProducer(const edm::ParameterSet&);
  ~AnalysisRootpleProducer() override {}  // no need to delete ROOT stuff
  // as it'll be deleted upon closing TFile

  void analyze(const edm::Event&, const edm::EventSetup&) override;
  void beginJob() override;
  void endJob() override;

  void fillEventInfo(int);
  void fillMCParticles(float, float, float, float);
  void fillTracks(float, float, float, float);
  void fillInclusiveJet(float, float, float, float);
  void fillChargedJet(float, float, float, float);
  void fillTracksJet(float, float, float, float);
  void fillCaloJet(float, float, float, float);
  void store();

private:
  bool onlyRECO;

  edm::EDGetTokenT<edm::HepMCProduct> mcEventToken;          // label of MC event
  edm::EDGetTokenT<reco::GenJetCollection> genJetCollToken;  // label of Jet made with MC particles
  edm::EDGetTokenT<reco::GenJetCollection> chgJetCollToken;  // label of Jet made with only charged MC particles
  edm::EDGetTokenT<std::vector<reco::GenParticle> > chgGenPartCollToken;  // label of charged MC particles
  edm::EDGetTokenT<reco::BasicJetCollection> tracksJetCollToken;
  edm::EDGetTokenT<reco::CaloJetCollection> recoCaloJetCollToken;
  edm::EDGetTokenT<reco::CandidateCollection> tracksCollToken;
  edm::EDGetTokenT<edm::TriggerResults> triggerResultsToken;

  edm::Handle<edm::HepMCProduct> EvtHandle;
  edm::Handle<std::vector<reco::GenParticle> > CandHandleMC;
  edm::Handle<reco::GenJetCollection> GenJetsHandle;
  edm::Handle<reco::GenJetCollection> ChgGenJetsHandle;
  edm::Handle<reco::CandidateCollection> CandHandleRECO;
  edm::Handle<reco::BasicJetCollection> TracksJetsHandle;
  edm::Handle<reco::CaloJetCollection> RecoCaloJetsHandle;
  edm::Handle<edm::TriggerResults> triggerResults;
  //  edm::Handle<TriggerFilterObjectWithRefs> hltFilter; // not used at the moment: can access objects that fired the trigger

  edm::Service<TFileService> fs;

  float piG;

  TTree* AnalysisTree;

  static const int NMCPMAX = 10000;
  static const int NTKMAX = 10000;
  static const int NIJMAX = 10000;
  static const int NCJMAX = 10000;
  static const int NTJMAX = 10000;
  static const int NEHJMAX = 10000;

  int EventKind, NumberMCParticles, NumberTracks, NumberInclusiveJet, NumberChargedJet, NumberTracksJet, NumberCaloJet;

  float MomentumMC[NMCPMAX], TransverseMomentumMC[NMCPMAX], EtaMC[NMCPMAX], PhiMC[NMCPMAX];
  float MomentumTK[NTKMAX], TransverseMomentumTK[NTKMAX], EtaTK[NTKMAX], PhiTK[NTKMAX];
  float MomentumIJ[NIJMAX], TransverseMomentumIJ[NIJMAX], EtaIJ[NIJMAX], PhiIJ[NIJMAX];
  float MomentumCJ[NCJMAX], TransverseMomentumCJ[NCJMAX], EtaCJ[NCJMAX], PhiCJ[NCJMAX];
  float MomentumTJ[NTJMAX], TransverseMomentumTJ[NTJMAX], EtaTJ[NTJMAX], PhiTJ[NTJMAX];
  float MomentumEHJ[NEHJMAX], TransverseMomentumEHJ[NEHJMAX], EtaEHJ[NEHJMAX], PhiEHJ[NEHJMAX];

  TClonesArray* MonteCarlo;
  TClonesArray* InclusiveJet;
  TClonesArray* ChargedJet;
  TClonesArray* Track;
  TClonesArray* TracksJet;
  TClonesArray* CalorimeterJet;
  TClonesArray* acceptedTriggers;
};

#endif