GammaJetAnalysis.h

Go to the documentation of this file.

#ifndef Calibration_HcalCalibAlgos_GammaJetAnalysis_H_
#define Calibration_HcalCalibAlgos_GammaJetAnalysis_H_

// system include files
#include <memory>
#include <string>

// user include files
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/EDAnalyzer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DataFormats/EgammaCandidates/interface/PhotonFwd.h"
#include "DataFormats/EgammaCandidates/interface/Photon.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"
#include "DataFormats/JetReco/interface/GenJetCollection.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"
#include "DataFormats/HcalRecHit/interface/HBHERecHit.h"
#include "DataFormats/HcalRecHit/interface/HFRecHit.h"
#include "DataFormats/HcalRecHit/interface/HORecHit.h"
#include "DataFormats/METReco/interface/METCollection.h"
#include "DataFormats/METReco/interface/PFMET.h"
#include "DataFormats/METReco/interface/PFMETCollection.h"
#include "DataFormats/ParticleFlowReco/interface/PFBlockFwd.h"
#include "DataFormats/ParticleFlowReco/interface/PFBlock.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/ParticleFlowReco/interface/PFCluster.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecHit.h"
#include "DataFormats/ParticleFlowReco/interface/PFRecHitFwd.h"
#include "DataFormats/HcalDetId/interface/HcalDetId.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackExtra.h"
#include "DataFormats/Common/interface/TriggerResults.h"
#include "DataFormats/HLTReco/interface/TriggerEvent.h"
#include "DataFormats/HLTReco/interface/TriggerObject.h"

#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloSubdetectorGeometry.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "SimDataFormats/GeneratorProducts/interface/GenEventInfoProduct.h"
#include "DataFormats/VertexReco/interface/Vertex.h"
#include "DataFormats/VertexReco/interface/VertexFwd.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidate.h"
#include "DataFormats/ParticleFlowCandidate/interface/PFCandidateFwd.h"

#include "HLTrigger/HLTcore/interface/HLTPrescaleProvider.h"

// forward declarations
class TH1D;
class TH2D;
class TFile;
class TTree;

//
// class declarations
//

class PhotonPair : protected std::pair<const reco::Photon*, double> {

public:
  PhotonPair() {
    first=0;
    second=0.0;
    fIdx=-1;
  }
  PhotonPair(const reco::Photon* ph, double pt, int setIdx=-1) {
    first=ph;
    second=pt;
    fIdx=setIdx;
  }
  ~PhotonPair() {}

  inline const reco::Photon* photon(void) const { return first; }
  inline void photon(const reco::Photon* ph) { first=ph; return; }
  inline double pt(void) const { return second; }
  inline void pt(double d) { second=d; return; }
  void idx(int set_idx) { fIdx=set_idx; };
  int idx() const { return fIdx; }
  bool isValid() const { return (first!=NULL) ? true:false; }

private:
  int fIdx; // index in the photon collection
};

class PFJetCorretPair : protected std::pair<const reco::PFJet*, double> {
public:
  PFJetCorretPair() {
    first=0;
    second=1.0;
  }
  PFJetCorretPair(const reco::PFJet* j, double s) {
    first=j;
    second=s;
  }
  ~PFJetCorretPair() {}

  inline const reco::PFJet* jet(void) const { return first; }
  inline void jet(const reco::PFJet* j) { first=j; return; }
  inline double scale(void) const { return second; }
  inline void scale(double d) { second=d; return; }
  double scaledEt() const { return first->et() * second; }
  bool isValid() const { return (first!=NULL) ? true:false; }

private:
  
};

// --------------------------------------------
// Main class
// --------------------------------------------

class GammaJetAnalysis : public edm::EDAnalyzer {
public:
  explicit GammaJetAnalysis(const edm::ParameterSet&);
  ~GammaJetAnalysis();

  float pfEcalIso(const reco::Photon* localPho1, edm::Handle<reco::PFCandidateCollection> pfHandle, float dRmax, float dRVetoBarrel, float dRVetoEndcap, float etaStripBarrel, float etaStripEndcap, float energyBarrel, float energyEndcap, reco::PFCandidate::ParticleType pfToUse);

  float pfHcalIso(const reco::Photon* localPho,edm::Handle<reco::PFCandidateCollection> pfHandle,float dRmax, float dRveto,reco::PFCandidate::ParticleType pfToUse);

  std::vector<float> pfTkIsoWithVertex(const reco::Photon* localPho1, edm::Handle<reco::PFCandidateCollection> pfHandle, edm::Handle<reco::VertexCollection> vtxHandle, float dRmax, float dRvetoBarrel, float dRvetoEndcap, float ptMin, float dzMax, float dxyMax, reco::PFCandidate::ParticleType pfToUse);

private:
  virtual void beginJob();//(const edm::EventSetup&);
  virtual void analyze(const edm::Event&, const edm::EventSetup&);
  virtual void endJob();
  void beginRun(const edm::Run&, const edm::EventSetup&);

  
  // parameters
  int debug_;                      // print debug statements
  unsigned int debugEvent;
  int debugHLTTrigNames;

  edm::InputTag rhoCollection_;
  edm::InputTag pfType1METColl, pfMETColl ;

  std::string photonCollName_;      // label for the photon collection
  std::string pfJetCollName_;       // label for the PF jet collection
  std::string pfJetCorrName_;       // label for the PF jet correction service
  std::string genJetCollName_;      // label for the genjet collection
  std::string genParticleCollName_; // label for the genparticle collection
  std::string genEventInfoName_;    // label for the generator event info collection
  std::string hbheRecHitName_;      // label for HBHERecHits collection
  std::string hfRecHitName_;        // label for HFRecHit collection
  std::string hoRecHitName_;        // label for HORecHit collection
  std::string rootHistFilename_;    // name of the histogram file
  std::string pvCollName_;          // label for primary vertex collection
  std::string prodProcess_;         // the producer process for AOD=2

  bool allowNoPhoton_; // whether module is used for dijet analysis
  double photonPtMin_;   // lowest value of the leading photon pT
  double photonJetDPhiMin_; // phi angle between the leading photon and the leading jet
  double jetEtMin_;      // lowest value of the leading jet ET
  double jet2EtMax_;     // largest value of the subleading jet ET
  double jet3EtMax_;     // largest value of the third jet ET
  std::vector<std::string> photonTrigNamesV_; // photon trigger names
  std::vector<std::string> jetTrigNamesV_; // jet trigger names
  bool writeTriggerPrescale_; // whether attempt to record the prescale

  //Tokens
  edm::EDGetTokenT<reco::PhotonCollection>          tok_Photon_;
  edm::EDGetTokenT<reco::PFJetCollection>           tok_PFJet_;
  edm::EDGetTokenT<std::vector<reco::GenJet> >      tok_GenJet_;
  edm::EDGetTokenT<std::vector<reco::GenParticle> > tok_GenPart_;
  edm::EDGetTokenT<GenEventInfoProduct>             tok_GenEvInfo_; 
  edm::EDGetTokenT<edm::SortedCollection<HBHERecHit,edm::StrictWeakOrdering<HBHERecHit> > > tok_HBHE_;
  edm::EDGetTokenT<edm::SortedCollection<HFRecHit,edm::StrictWeakOrdering<HFRecHit> > >     tok_HF_;
  edm::EDGetTokenT<edm::SortedCollection<HORecHit,edm::StrictWeakOrdering<HORecHit> > >     tok_HO_;
  edm::EDGetTokenT<edm::ValueMap<Bool_t> >          tok_loosePhoton_;
  edm::EDGetTokenT<edm::ValueMap<Bool_t> >          tok_tightPhoton_;
  edm::EDGetTokenT<std::vector<Bool_t> >          tok_loosePhotonV_;
  edm::EDGetTokenT<std::vector<Bool_t> >          tok_tightPhotonV_;
  edm::EDGetTokenT<reco::PFCandidateCollection>     tok_PFCand_;
  edm::EDGetTokenT<reco::VertexCollection>          tok_Vertex_;
  edm::EDGetTokenT<reco::GsfElectronCollection>     tok_GsfElec_;
  edm::EDGetTokenT<double>                          tok_Rho_;
  edm::EDGetTokenT<reco::ConversionCollection>      tok_Conv_;
  edm::EDGetTokenT<reco::BeamSpot>                  tok_BS_;
  edm::EDGetTokenT<std::vector<reco::Vertex> >      tok_PV_;
  edm::EDGetTokenT<reco::PFMETCollection>           tok_PFMET_;
  edm::EDGetTokenT<reco::PFMETCollection>           tok_PFType1MET_;
  edm::EDGetTokenT<edm::TriggerResults>             tok_TrigRes_;
  bool doPFJets_;                   // use PFJets
  bool doGenJets_;                  // use GenJets
  int  workOnAOD_;
  bool ignoreHLT_;

  // root file/histograms
  TFile* rootfile_;

  TTree* misc_tree_; // misc.information. Will be filled only once
  TTree* calo_tree_;
  TTree* pf_tree_;

  // trigger info
  HLTPrescaleProvider hltPrescaleProvider_;
  std::vector<int> photonTrigFired_;
  std::vector<int> photonTrigPrescale_;
  std::vector<int> jetTrigFired_;
  std::vector<int> jetTrigPrescale_;

  // Event info
  int runNumber_, lumiBlock_, eventNumber_;
  float eventWeight_, eventPtHat_;
  int nPhotons_, nGenJets_;
  int nPFJets_;
  ULong64_t nProcessed_;
  int pf_NPV_;

  float met_value_, met_phi_, met_sumEt_ ;
  float metType1_value_, metType1_phi_, metType1_sumEt_ ;

  // photon info
  float rho2012_;
  float tagPho_pt_, pho_2nd_pt_, tagPho_energy_, tagPho_eta_, tagPho_phi_, tagPho_sieie_;
  float tagPho_HoE_, tagPho_r9_, tagPho_EcalIsoDR04_, tagPho_HcalIsoDR04_, tagPho_HcalIsoDR0412_, tagPho_TrkIsoHollowDR04_, tagPho_pfiso_myphoton03_;
  float tagPho_pfiso_myneutral03_;
  std::vector<std::vector<float> >  tagPho_pfiso_mycharged03 ;
  int tagPho_pixelSeed_;
  int tagPho_ConvSafeEleVeto_;
  int tagPho_idTight_, tagPho_idLoose_;
  float tagPho_genPt_, tagPho_genEnergy_, tagPho_genEta_, tagPho_genPhi_;
  float tagPho_genDeltaR_;

  // Particle-flow jets
  // leading Et jet info
  float ppfjet_pt_, ppfjet_p_, ppfjet_E_, ppfjet_eta_, ppfjet_phi_, ppfjet_scale_;
  float ppfjet_area_, ppfjet_E_NPVcorr_;
  float ppfjet_NeutralHadronFrac_, ppfjet_NeutralEMFrac_;
  int ppfjet_nConstituents_;
  float ppfjet_ChargedHadronFrac_, ppfjet_ChargedMultiplicity_, ppfjet_ChargedEMFrac_;
  float ppfjet_gendr_, ppfjet_genpt_, ppfjet_genp_, ppfjet_genE_;
  float ppfjet_unkown_E_, ppfjet_unkown_px_, ppfjet_unkown_py_, ppfjet_unkown_pz_, ppfjet_unkown_EcalE_;
  float ppfjet_electron_E_, ppfjet_electron_px_, ppfjet_electron_py_, ppfjet_electron_pz_, ppfjet_electron_EcalE_;
  float ppfjet_muon_E_, ppfjet_muon_px_, ppfjet_muon_py_, ppfjet_muon_pz_, ppfjet_muon_EcalE_;
  float ppfjet_photon_E_, ppfjet_photon_px_, ppfjet_photon_py_, ppfjet_photon_pz_, ppfjet_photon_EcalE_;
  int ppfjet_unkown_n_, ppfjet_electron_n_, ppfjet_muon_n_, ppfjet_photon_n_;
  int ppfjet_had_n_;
  std::vector<float> ppfjet_had_E_, ppfjet_had_px_, ppfjet_had_py_, ppfjet_had_pz_, ppfjet_had_EcalE_, ppfjet_had_rawHcalE_, ppfjet_had_emf_, ppfjet_had_E_mctruth_;
  std::vector<int> ppfjet_had_id_, ppfjet_had_candtrackind_, ppfjet_had_mcpdgId_, ppfjet_had_ntwrs_;
  int ppfjet_ntwrs_;
  std::vector<int> ppfjet_twr_ieta_, ppfjet_twr_iphi_, ppfjet_twr_depth_, ppfjet_twr_subdet_, ppfjet_twr_candtrackind_, ppfjet_twr_hadind_, ppfjet_twr_elmttype_, ppfjet_twr_clusterind_;
  std::vector<float> ppfjet_twr_hade_, ppfjet_twr_frac_, ppfjet_twr_dR_;
  int ppfjet_cluster_n_;
  std::vector<float> ppfjet_cluster_eta_, ppfjet_cluster_phi_, ppfjet_cluster_dR_;
  int ppfjet_ncandtracks_;
  std::vector<float> ppfjet_candtrack_px_, ppfjet_candtrack_py_, ppfjet_candtrack_pz_, ppfjet_candtrack_EcalE_;

  // subleading Et jet info
  float pfjet2_pt_, pfjet2_p_, pfjet2_E_, pfjet2_eta_, pfjet2_phi_, pfjet2_scale_;
  float pfjet2_area_, pfjet2_E_NPVcorr_;
  float pfjet2_NeutralHadronFrac_, pfjet2_NeutralEMFrac_;
  int pfjet2_nConstituents_;
  float pfjet2_ChargedHadronFrac_, pfjet2_ChargedMultiplicity_, pfjet2_ChargedEMFrac_;
  float pfjet2_gendr_, pfjet2_genpt_, pfjet2_genp_, pfjet2_genE_;
  float pfjet2_unkown_E_, pfjet2_unkown_px_, pfjet2_unkown_py_, pfjet2_unkown_pz_, pfjet2_unkown_EcalE_;
  float pfjet2_electron_E_, pfjet2_electron_px_, pfjet2_electron_py_, pfjet2_electron_pz_, pfjet2_electron_EcalE_;
  float pfjet2_muon_E_, pfjet2_muon_px_, pfjet2_muon_py_, pfjet2_muon_pz_, pfjet2_muon_EcalE_;
  float pfjet2_photon_E_, pfjet2_photon_px_, pfjet2_photon_py_, pfjet2_photon_pz_, pfjet2_photon_EcalE_;
  int pfjet2_unkown_n_, pfjet2_electron_n_, pfjet2_muon_n_, pfjet2_photon_n_;
  int pfjet2_had_n_;
  std::vector<float> pfjet2_had_E_, pfjet2_had_px_, pfjet2_had_py_, pfjet2_had_pz_, pfjet2_had_EcalE_, pfjet2_had_rawHcalE_, pfjet2_had_emf_, pfjet2_had_E_mctruth_;
  std::vector<int> pfjet2_had_id_, pfjet2_had_candtrackind_, pfjet2_had_mcpdgId_, pfjet2_had_ntwrs_;
  int pfjet2_ntwrs_;
  std::vector<int> pfjet2_twr_ieta_, pfjet2_twr_iphi_, pfjet2_twr_depth_, pfjet2_twr_subdet_, pfjet2_twr_candtrackind_, pfjet2_twr_hadind_, pfjet2_twr_elmttype_, pfjet2_twr_clusterind_;
  std::vector<float> pfjet2_twr_hade_, pfjet2_twr_frac_, pfjet2_twr_dR_;
  int pfjet2_cluster_n_;
  std::vector<float> pfjet2_cluster_eta_, pfjet2_cluster_phi_, pfjet2_cluster_dR_;
  int pfjet2_ncandtracks_;
  std::vector<float> pfjet2_candtrack_px_, pfjet2_candtrack_py_, pfjet2_candtrack_pz_, pfjet2_candtrack_EcalE_;

  float pf_thirdjet_et_;
  float pf_thirdjet_pt_, pf_thirdjet_p_, pf_thirdjet_px_, pf_thirdjet_py_;
  float pf_thirdjet_E_, pf_thirdjet_eta_, pf_thirdjet_phi_, pf_thirdjet_scale_;

  // helper functions
  template<class JetPair_type>
  float calc_dPhi(const PhotonPair &pho, const JetPair_type &jet) {
    if (!pho.isValid() || !jet.isValid()) return 9999.;
    float phi1=pho.photon()->phi();
    float phi2=jet.jet()->phi();
    float dphi=fabs(phi1-phi2);
    const float cPi= 4*atan(1);
    while (dphi>cPi) dphi = fabs(2*cPi - dphi);
    return dphi;
  }

  double deltaR(const reco::Jet* j1, const reco::Jet* j2);
  double deltaR(const double eta1, const double phi1, const double eta2, const double phi2);
  int getEtaPhi(const DetId id);
  int getEtaPhi(const HcalDetId id);

  void clear_leadingPfJetVars();
  void copy_leadingPfJetVars_to_pfJet2();

  template<class Jet_type>
  double deltaR(const PhotonPair &photon, const Jet_type *jet) {
    if (!photon.isValid()) return 9999.;
    return deltaR(photon.photon()->eta(),photon.photon()->phi(),
          jet->eta(), jet->phi());
  }

  struct PFJetCorretPairComp {
    inline bool operator() ( const PFJetCorretPair& a, const PFJetCorretPair& b) {
      return (a.jet()->pt()*a.scale()) > (b.jet()->pt()*b.scale());
    }
  };

  struct PhotonPairComp {
    inline bool operator() ( const PhotonPair& a, const PhotonPair& b) {
      return ( (a.photon()->pt()) > (b.photon()->pt()) );
    }
  };

};

#endif