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