ErsatzMEt.h

Go to the documentation of this file.

#ifndef ElectroWeakAnalysis_ErsatzMEt_h
#define ElectroWeakAnalysis_ErsatzMEt_h
// -*- C++ -*-
//
// Package:    ErsatzMEt
// Class:      ErsatzMEt
//
//
// Original Author:  David Wardrope
//         Created:  Tue Nov 11 16:47:29 GMT 2008
//
//
 
// 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/ESHandle.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CommonTools/UtilAlgos/interface/TFileService.h"
//Random Number Generator
#include "FWCore/Utilities/interface/RandomNumberGenerator.h"
#include "CLHEP/Random/RandFlat.h"
//CMSSW Containers
#include "DataFormats/Common/interface/ValueMap.h"
//Egamma Objects
#include "DataFormats/EgammaCandidates/interface/GsfElectron.h"
#include "DataFormats/EgammaCandidates/interface/GsfElectronFwd.h"
#include "DataFormats/EgammaReco/interface/SuperCluster.h"
//ECAL
#include "RecoEcal/EgammaCoreTools/interface/EcalClusterLazyTools.h"
#include "RecoEcal/EgammaClusterAlgos/interface/EgammaSCEnergyCorrectionAlgo.h"
//Geometry
#include "RecoCaloTools/Navigation/interface/CaloNavigator.h"
#include "Geometry/CaloTopology/interface/CaloTopology.h"
#include "Geometry/EcalAlgo/interface/EcalBarrelGeometry.h"
#include "Geometry/EcalAlgo/interface/EcalEndcapGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloSubdetectorGeometry.h"
#include "Geometry/Records/interface/CaloTopologyRecord.h"
//DetIds
#include "DataFormats/CaloTowers/interface/CaloTowerDetId.h"
//Other Objects
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/HepMCCandidate/interface/GenParticle.h"
#include "DataFormats/HLTReco/interface/TriggerEvent.h"
#include "DataFormats/Common/interface/TriggerResults.h"
#include "HLTrigger/HLTcore/interface/HLTConfigProvider.h"
#include "DataFormats/METReco/interface/CaloMET.h"
#include "DataFormats/METReco/interface/CaloMETFwd.h"
#include "DataFormats/METReco/interface/GenMET.h"
#include "DataFormats/METReco/interface/GenMETFwd.h"
#include "DataFormats/METReco/interface/PFMET.h"
#include "DataFormats/METReco/interface/PFMETFwd.h"
#include "DataFormats/METReco/interface/MET.h"
#include "DataFormats/METReco/interface/METFwd.h"
//Physics Tools
#include "DataFormats/Math/interface/deltaR.h"
 
//Maths
#include "Math/GenVector/VectorUtil.h"
//ROOT
#include "TTree.h"
#include "TH1.h"
#include "TH2.h"
//Helper Functions
#include "ElectroWeakAnalysis/ZEE/interface/UniqueElectrons.h"
#include "ElectroWeakAnalysis/ZEE/interface/ElectronSelector.h"
#include "ElectroWeakAnalysis/ZEE/interface/CaloVectors.h"
#include "ElectroWeakAnalysis/ZEE/interface/SCEnergyCorrections.h"
 
#define nEntries_arr_ 4
//#include "DataFormats/EgammaReco/interface/SuperCluster.h"
//
// class declaration
//
 
class ErsatzMEt : public edm::EDAnalyzer {
public:
  explicit ErsatzMEt(const edm::ParameterSet&);
  ~ErsatzMEt() override;
 
private:
  void beginJob() override;
  void analyze(const edm::Event&, const edm::EventSetup&) override;
  //    std::map<reco::GsfElectronRef, reco::SuperClusterRef> probeFinder(const std::vector<reco::GsfElectronRef>&,
  //                            const edm::Handle<reco::SuperClusterCollection>&,
  //                            const edm::Handle<reco::SuperClusterCollection>&);
  std::map<reco::GsfElectronRef, reco::GsfElectronRef> probeFinder(const std::vector<reco::GsfElectronRef>&,
                                                                   const edm::Handle<reco::GsfElectronCollection>);
  reco::MET ersatzFabrik(const reco::GsfElectronRef&, const reco::SuperCluster&, const reco::MET&, const int);
  reco::MET ersatzFabrik(const reco::GsfElectronRef&, const reco::GsfElectronRef&, const reco::MET&);
  bool isInBarrel(double);
  bool isInEndCap(double);
  bool isInFiducial(double);
 
  void endJob() override;
 
  // ----------member data ---------------------------
  edm::EDGetTokenT<reco::GenParticleCollection> MCTruthCollection_;
  edm::EDGetTokenT<reco::GsfElectronCollection> ElectronCollection_;
  edm::EDGetTokenT<reco::SuperClusterCollection> HybridScCollection_;
  edm::EDGetTokenT<reco::SuperClusterCollection> M5x5ScCollection_;
  edm::EDGetTokenT<reco::GenMETCollection> GenMEtCollection_;
  edm::EDGetTokenT<reco::CaloMETCollection> CaloMEtCollection_;
  edm::EDGetTokenT<reco::METCollection> T1MEtCollection_;
  edm::EDGetTokenT<reco::PFMETCollection> PfMEtCollection_;
  edm::EDGetTokenT<reco::METCollection> TcMEtCollection_;
  edm::EDGetTokenT<trigger::TriggerEvent> TriggerEvent_;
  edm::EDGetTokenT<edm::TriggerResults> TriggerResults_;
  edm::InputTag TriggerPath_;
  std::string TriggerName_, ProcessName_;
  edm::ParameterSet hyb_fCorrPSet_, m5x5_fCorrPSet_;
  double mW_, mZ_, mTPmin_, mTPmax_;
  double BarrelEtaMax_, EndCapEtaMin_, EndCapEtaMax_;
 
  enum cut_index_t {
    EtCut_,
    EB_sIhIh_,
    EB_dEtaIn_,
    EB_dPhiIn_,
    EB_TrckIso_,
    EB_EcalIso_,
    EB_HcalIso_,
    EE_sIhIh_,
    EE_dEtaIn_,
    EE_dPhiIn_,
    EE_TrckIso_,
    EE_EcalIso_,
    EE_HcalIso_
  };
 
  std::vector<double> CutVector_;
 
  int etaWidth_, phiWidth_;
  bool Zevent_, HLTPathCheck_;
  //    std::vector<double> EtaWeights_;
  HLTConfigProvider hltConfig_;
 
  edm::ESHandle<CaloGeometry> geoHandle_;
  edm::ESHandle<CaloTopology> pTopology_;
  //Output variables
  TTree* t_;
  int nTags_, nProbes_;
  double CaloMEt_, T1MEt_, PfMEt_, TcMEt_;
  double CaloMEtphi_, T1MEtphi_, PfMEtphi_, TcMEtphi_;
  int McElec_nZmum_, McElec_nFinal_;
  double McZ_m_, McZ_pt_, McZ_phi_, McZ_eta_, McZ_y_;
  double McZ_rescM_, McZ_rescPt_, McZ_rescEta_, McZ_rescPhi_, McZ_rescY_;
 
  int probe_nClus_[nEntries_arr_];
  int tag_q_[nEntries_arr_];
  double tag_pt_[nEntries_arr_], tag_eta_[nEntries_arr_], tag_phi_[nEntries_arr_];
  //double tag_caloV1_rescPt_[nEntries_arr_], tag_caloV1_rescEta_[nEntries_arr_], tag_caloV1_rescPhi_[nEntries_arr_];
  //double tag_caloV2_rescPt_[nEntries_arr_], tag_caloV2_rescEta_[nEntries_arr_], tag_caloV2_rescPhi_[nEntries_arr_];
  //double tag_caloV3_rescPt_[nEntries_arr_], tag_caloV3_rescEta_[nEntries_arr_], tag_caloV3_rescPhi_[nEntries_arr_];
  //double tag_caloV4_rescPt_[nEntries_arr_], tag_caloV4_rescEta_[nEntries_arr_], tag_caloV4_rescPhi_[nEntries_arr_];
  double tag_rescPt_[nEntries_arr_], tag_rescEta_[nEntries_arr_], tag_rescPhi_[nEntries_arr_];
  double tag_sIhIh_[nEntries_arr_], tag_dPhiIn_[nEntries_arr_], tag_dEtaIn_[nEntries_arr_];
  double tag_trckIso_[nEntries_arr_], tag_ecalIso_[nEntries_arr_], tag_hcalIso_[nEntries_arr_];
  double tag_e2x5Max_[nEntries_arr_], tag_e1x5Max_[nEntries_arr_], tag_e5x5_[nEntries_arr_];
  double tag_hoe_[nEntries_arr_], tag_eop_[nEntries_arr_], tag_pin_[nEntries_arr_], tag_pout_[nEntries_arr_];
  int probe_q_[nEntries_arr_];
  double probe_pt_[nEntries_arr_], probe_eta_[nEntries_arr_], probe_phi_[nEntries_arr_];
  double probe_rescPt_[nEntries_arr_], probe_rescEta_[nEntries_arr_], probe_rescPhi_[nEntries_arr_];
  double probe_sIhIh_[nEntries_arr_], probe_dPhiIn_[nEntries_arr_], probe_dEtaIn_[nEntries_arr_];
  double probe_trckIso_[nEntries_arr_], probe_ecalIso_[nEntries_arr_], probe_hcalIso_[nEntries_arr_];
  double probe_e2x5Max_[nEntries_arr_], probe_e1x5Max_[nEntries_arr_], probe_e5x5_[nEntries_arr_];
  double probe_hoe_[nEntries_arr_], probe_eop_[nEntries_arr_], probe_pin_[nEntries_arr_], probe_pout_[nEntries_arr_];
  double Z_pt_[nEntries_arr_], Z_eta_[nEntries_arr_], Z_phi_[nEntries_arr_], Z_m_[nEntries_arr_], Z_y_[nEntries_arr_];
  double Z_rescPt_[nEntries_arr_], Z_rescEta_[nEntries_arr_], Z_rescPhi_[nEntries_arr_], Z_rescM_[nEntries_arr_],
      Z_rescY_[nEntries_arr_];
  double Z_probe_dPhi_[nEntries_arr_];
  double ErsatzV1CaloMEt_[nEntries_arr_], ErsatzV1CaloMt_[nEntries_arr_], ErsatzV1CaloMEtPhi_[nEntries_arr_];
  double ErsatzV2CaloMEt_[nEntries_arr_], ErsatzV2CaloMEtPhi_[nEntries_arr_], ErsatzV2CaloMt_[nEntries_arr_];
  double ErsatzV3CaloMEt_[nEntries_arr_], ErsatzV3CaloMEtPhi_[nEntries_arr_], ErsatzV3CaloMt_[nEntries_arr_];
  double ErsatzV4CaloMEt_[nEntries_arr_], ErsatzV4CaloMEtPhi_[nEntries_arr_], ErsatzV4CaloMt_[nEntries_arr_];
  double ErsatzV1T1MEt_[nEntries_arr_], ErsatzV1T1Mt_[nEntries_arr_], ErsatzV1T1MEtPhi_[nEntries_arr_];
  double ErsatzV1PfMEt_[nEntries_arr_], ErsatzV1PfMt_[nEntries_arr_], ErsatzV1PfMEtPhi_[nEntries_arr_];
  double ErsatzV1TcMEt_[nEntries_arr_], ErsatzV1TcMt_[nEntries_arr_], ErsatzV1TcMEtPhi_[nEntries_arr_];
  double ErsatzV1_Mesc_[nEntries_arr_], ErsatzV1_rescMesc_[nEntries_arr_];
  double ErsatzV2_Mesc_[nEntries_arr_], ErsatzV2_rescMesc_[nEntries_arr_];
  double ErsatzV3_Mesc_[nEntries_arr_], ErsatzV3_rescMesc_[nEntries_arr_];
  double ErsatzV4_Mesc_[nEntries_arr_], ErsatzV4_rescMesc_[nEntries_arr_];
 
  double McElec_pt_[nEntries_arr_], McElec_eta_[nEntries_arr_], McElec_phi_[nEntries_arr_];
  double McElec_rescPt_[nEntries_arr_], McElec_rescEta_[nEntries_arr_], McElec_rescPhi_[nEntries_arr_];
  double McProbe_pt_[nEntries_arr_], McProbe_eta_[nEntries_arr_], McProbe_phi_[nEntries_arr_];
  double McProbe_rescPt_[nEntries_arr_], McProbe_rescEta_[nEntries_arr_], McProbe_rescPhi_[nEntries_arr_];
  double McElecProbe_dPhi_[nEntries_arr_], McElecProbe_dEta_[nEntries_arr_], McElecProbe_dR_[nEntries_arr_];
 
  double probe_d_MCE_SCE_[nEntries_arr_];
  double probe_sc_pt_[nEntries_arr_], probe_sc_eta_[nEntries_arr_], probe_sc_phi_[nEntries_arr_];
  double probe_sc_E_[nEntries_arr_], probe_sc_rawE_[nEntries_arr_], probe_sc_nClus_[nEntries_arr_];
  double probe_scV2_E_[nEntries_arr_];
  double probe_scV3_E_[nEntries_arr_];
  double probe_scV4_E_[nEntries_arr_];
 
  int iComb_;
};
#endif