WenuPlots.h

Go to the documentation of this file.

#ifndef WenuPlots_H
#define WenuPlots_H

#include <memory>

#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"
#include "FWCore/Utilities/interface/InputTag.h"
#include "DataFormats/PatCandidates/interface/Electron.h"
#include "DataFormats/PatCandidates/interface/MET.h"
#include "DataFormats/PatCandidates/interface/CompositeCandidate.h"
#include "DataFormats/EgammaCandidates/interface/Electron.h"
#include "DataFormats/JetReco/interface/PFJet.h"
#include "DataFormats/JetReco/interface/CaloJet.h"
#include "DataFormats/JetReco/interface/PFJetCollection.h"
#include "DataFormats/JetReco/interface/CaloJetCollection.h"
#include "DataFormats/VertexReco/interface/Vertex.h"

//#include "DataFormats/BeamSpot/interface/BeamSpot.h"

#include <vector>
#include <iostream>
#include "TFile.h"
#include "TTree.h"
#include "TString.h"
#include "TH1F.h"
#include "TMath.h"

//
// class decleration
//

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


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

      // ----------member data ---------------------------
      Bool_t CheckCuts( const pat::Electron * ele);
      Bool_t CheckCut( const pat::Electron *wenu, Int_t i);
      Bool_t CheckCutsInverse(const pat::Electron *ele);
      Bool_t CheckCutInv( const pat::Electron *wenu, Int_t i);
      Bool_t CheckCutsNminusOne(const pat::Electron *ele, Int_t jj);
      Double_t ReturnCandVar(const pat::Electron *ele, Int_t i);
      Bool_t   PassPreselectionCriteria(const pat::Electron *ele);
  // for the extra identifications and selections
  Bool_t   usePrecalcID_;
  std::string usePrecalcIDSign_;
  std::string usePrecalcIDType_;
  Double_t usePrecalcIDValue_;
  // for extra preselection criteria:
  Bool_t useValidFirstPXBHit_;
  Bool_t useConversionRejection_;
  Bool_t useExpectedMissingHits_;
  Int_t  maxNumberOfExpectedMissingHits_;
  Bool_t usePreselection_;
  std::string outputFile_;
  edm::EDGetTokenT<pat::CompositeCandidateCollection> wenuCollectionToken_;
  edm::InputTag caloJetCollectionTag_;
  edm::EDGetTokenT< reco::CaloJetCollection > caloJetCollectionToken_;
  edm::InputTag pfJetCollectionTag_;
  edm::EDGetTokenT< reco::PFJetCollection > pfJetCollectionToken_;
  edm::EDGetTokenT< std::vector<reco::Vertex> >   PrimaryVerticesCollectionToken_;
  edm::EDGetTokenT< std::vector<reco::Vertex> >   PrimaryVerticesCollectionBSToken_;
  TFile *histofile;
  //
  //  math::XYZPoint bspotPosition_; // comment out only if you don't use pat
  //
  // the histograms ********************

  TH1F *h_met;
  TH1F *h_met_inverse;
  TH1F *h_mt;
  TH1F *h_mt_inverse;

  TH1F *h_met_EB;
  TH1F *h_met_inverse_EB;
  TH1F *h_mt_EB;
  TH1F *h_mt_inverse_EB;

  TH1F *h_met_EE;
  TH1F *h_met_inverse_EE;
  TH1F *h_mt_EE;
  TH1F *h_mt_inverse_EE;


  TH1F *h_scEt;
  TH1F *h_scEta;
  TH1F *h_scPhi;

  TH1F *h_EB_trkiso;
  TH1F *h_EB_ecaliso;
  TH1F *h_EB_hcaliso;
  TH1F *h_EB_sIetaIeta;
  TH1F *h_EB_dphi;
  TH1F *h_EB_deta;
  TH1F *h_EB_HoE;

  TH1F *h_EE_trkiso;
  TH1F *h_EE_ecaliso;
  TH1F *h_EE_hcaliso;
  TH1F *h_EE_sIetaIeta;
  TH1F *h_EE_dphi;
  TH1F *h_EE_deta;
  TH1F *h_EE_HoE;

  //
  TH1F *h_trackIso_eb_NmOne;
  TH1F *h_trackIso_ee_NmOne;
  // ***********************************
  //
  // the selection cuts
  Double_t trackIso_EB_;
  Double_t ecalIso_EB_;
  Double_t hcalIso_EB_;
  //
  Double_t trackIso_EE_;
  Double_t ecalIso_EE_;
  Double_t hcalIso_EE_;
  //
  Double_t sihih_EB_;
  Double_t deta_EB_;
  Double_t dphi_EB_;
  Double_t hoe_EB_;
  Double_t cIso_EB_;
  Double_t tip_bspot_EB_;
  Double_t eop_EB_;
  //
  Double_t sihih_EE_;
  Double_t deta_EE_;
  Double_t dphi_EE_;
  Double_t hoe_EE_;
  Double_t cIso_EE_;
  Double_t tip_bspot_EE_;
  Double_t eop_EE_;
  //
  Double_t trackIsoUser_EB_;
  Double_t ecalIsoUser_EB_;
  Double_t hcalIsoUser_EB_;
  Double_t trackIsoUser_EE_;
  Double_t ecalIsoUser_EE_;
  Double_t hcalIsoUser_EE_;
  //
  Bool_t trackIso_EB_inv;
  Bool_t ecalIso_EB_inv;
  Bool_t hcalIso_EB_inv;
  //
  Bool_t trackIso_EE_inv;
  Bool_t ecalIso_EE_inv;
  Bool_t hcalIso_EE_inv;
  //
  Bool_t sihih_EB_inv;
  Bool_t deta_EB_inv;
  Bool_t dphi_EB_inv;
  Bool_t hoe_EB_inv;
  Bool_t cIso_EB_inv;
  Bool_t tip_bspot_EB_inv;
  Bool_t eop_EB_inv;
  //
  Bool_t sihih_EE_inv;
  Bool_t deta_EE_inv;
  Bool_t dphi_EE_inv;
  Bool_t hoe_EE_inv;
  Bool_t cIso_EE_inv;
  Bool_t tip_bspot_EE_inv;
  Bool_t eop_EE_inv;
  //
  Bool_t trackIsoUser_EB_inv;
  Bool_t ecalIsoUser_EB_inv;
  Bool_t hcalIsoUser_EB_inv;
  Bool_t trackIsoUser_EE_inv;
  Bool_t ecalIsoUser_EE_inv;
  Bool_t hcalIsoUser_EE_inv;
  //
  //
  Int_t nBarrelVars_;
  //
  std::vector<Double_t> CutVars_;
  std::vector<Bool_t> InvVars_;
  //
  // variables related to the VBTF root tuples:
  //
  Int_t runNumber, lumiSection;
  Long64_t eventNumber;
  Float_t ele_sc_energy, ele_sc_eta, ele_sc_phi, ele_sc_rho;
  Float_t ele_sc_gsf_et;
  Float_t ele_cand_et, ele_cand_eta, ele_cand_phi;
  Float_t ele_iso_track, ele_iso_ecal, ele_iso_hcal;
  Float_t ele_id_sihih, ele_id_dphi, ele_id_deta, ele_id_hoe;
  Float_t ele_cr_dcot, ele_cr_dist;
  Int_t   ele_cr_mhitsinner;
  Float_t ele_vx, ele_vy, ele_vz;
  Float_t ele_pin, ele_pout;
  Float_t pv_x, pv_y, pv_z;
  Int_t   ele_gsfCharge, ele_ctfCharge, ele_scPixCharge;
  Float_t ele_eop, ele_tip_bs, ele_tip_pv;
  Float_t event_caloMET, event_pfMET, event_tcMET;
  Float_t event_caloSumEt, event_pfSumEt, event_tcSumEt;
  Float_t event_caloMET_phi, event_pfMET_phi, event_tcMET_phi;
  Float_t event_caloMT, event_pfMT, event_tcMT;
  Float_t calojet_et[5];
  Float_t calojet_eta[5];
  Float_t calojet_phi[5];
  Float_t pfjet_et[5];
  Float_t pfjet_eta[5];
  Float_t pfjet_phi[5];
  Float_t ele2nd_sc_gsf_et;
  Float_t ele2nd_sc_eta;
  Float_t ele2nd_sc_phi;
  Float_t ele2nd_sc_rho;
  Float_t ele2nd_cand_eta;
  Float_t ele2nd_cand_phi;
  Float_t ele2nd_pin;
  Float_t ele2nd_pout;
  Int_t   ele2nd_passes_selection;
  Int_t   ele2nd_ecalDriven;
  Float_t ele_hltmatched_dr;
  Int_t   event_triggerDecision;
  Int_t event_datasetTag;

  TFile *WENU_VBTFpreseleFile_;
  TFile *WENU_VBTFselectionFile_;
  TTree *vbtfSele_tree;
  TTree *vbtfPresele_tree;
  std::string WENU_VBTFselectionFileName_;
  std::string WENU_VBTFpreseleFileName_;
  Bool_t includeJetInformationInNtuples_;
  Bool_t storeExtraInformation_;
  Double_t DRJetFromElectron_;
  Int_t DatasetTag_;
  // for the 2nd electron storage
  Bool_t storeAllSecondElectronVariables_;
  Float_t ele2nd_cand_et;
  Float_t ele2nd_iso_track, ele2nd_iso_ecal, ele2nd_iso_hcal;
  Float_t ele2nd_id_sihih, ele2nd_id_deta, ele2nd_id_dphi, ele2nd_id_hoe;
  Float_t ele2nd_cr_dcot, ele2nd_cr_dist;
  Float_t ele2nd_vx, ele2nd_vy, ele2nd_vz;
  Int_t   ele2nd_cr_mhitsinner, ele2nd_gsfCharge, ele2nd_ctfCharge, ele2nd_scPixCharge;
  Float_t ele2nd_eop, ele2nd_tip_bs, ele2nd_tip_pv;
  Float_t ele2nd_hltmatched_dr;
  std::vector<Int_t> VtxTracksSize;
  std::vector<Float_t> VtxNormalizedChi2;
  std::vector<Int_t> VtxTracksSizeBS;
  std::vector<Float_t> VtxNormalizedChi2BS;
  Float_t pvbs_x, pvbs_y, pvbs_z;
  Float_t ele_tip_pvbs, ele2nd_tip_pvbs;
};

#endif