JetIDHelper.h

Go to the documentation of this file.

#ifndef RecoJets_JetProducers_interface_JetIDHelper_h
#define RecoJets_JetProducers_interface_JetIDHelper_h

#include <atomic>

#include "DataFormats/HcalRecHit/interface/HcalRecHitCollections.h"
#include "DataFormats/HcalRecHit/interface/HcalRecHitFwd.h"
#include "DataFormats/EcalRecHit/interface/EcalRecHitCollections.h"
#include "DataFormats/EcalDetId/interface/EcalDetIdCollections.h"

#include "FWCore/Framework/interface/ConsumesCollector.h"
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/Event.h"
#include "DataFormats/JetReco/interface/CaloJet.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/ParameterSet/interface/ParameterSetDescription.h"

namespace reco {

  namespace helper {

    class JetIDHelper {

    public : 
      // construction
      JetIDHelper() {}
      JetIDHelper( edm::ParameterSet const & pset, edm::ConsumesCollector&& iC );
      ~JetIDHelper() {} 

      void fillDescription(edm::ParameterSetDescription& iDesc);

      void initValues ();

      // interface
      void calculate( const edm::Event& event, const reco::CaloJet &jet, const int iDbg = 0 );

      // member access
      
      // these require RecHits, so can not be evaluated from AOD
      double fHPD()          const    { return    fHPD_;}           
      double fRBX()          const    { return    fRBX_;}           
      int    n90Hits()       const    { return    n90Hits_;}
      double fSubDetector1() const    { return    fSubDetector1_;}  
      double fSubDetector2() const    { return    fSubDetector2_;}  
      double fSubDetector3() const    { return    fSubDetector3_;}  
      double fSubDetector4() const    { return    fSubDetector4_;}  
      double fEB()           const    { return    fEB_;}
      double fEE()           const    { return    fEE_;}
      double fHB()           const    { return    fHB_;}
      double fHE()           const    { return    fHE_;}
      double fHO()           const    { return    fHO_;}
      double fLong()         const    { return    fLong_;}
      double fShort()        const    { return    fShort_;}
      double fLSbad()        const    { return    fLS_;}  
      double fHFOOT()        const    { return    fHFOOT_;}  
      // these are tower based
      double restrictedEMF() const    { return    restrictedEMF_;}
      int    nHCALTowers()   const    { return    nHCALTowers_;}    
      int    nECALTowers()   const    { return    nECALTowers_;}    
      // tower based approximations / inferior options
      double approximatefHPD() const { return    approximatefHPD_;}           
      double approximatefRBX() const { return    approximatefRBX_;}           
      int    hitsInN90()        const { return    hitsInN90_;}        
 
      struct subtower { // contents of a sub-detector's tower
    double E;
    int Nhit;
    
    subtower( double xE, int xN ) { E = xE; Nhit = xN; }
      };
      
  
    private:

     
      // helper functions
      void classifyJetComponents( const edm::Event& event, const reco::CaloJet &jet, 
                  std::vector< double > &energies,
                  std::vector< double > &subdet_energies,
                  std::vector< double > &Ecal_energies, std::vector< double > &Hcal_energies, 
                  std::vector< double > &HO_energies,
                  std::vector< double > &HPD_energies,  std::vector< double > &RBX_energies,
                  double& LS_bad_energy, double& HF_OOT_energy, const int iDbg = 0);

      void classifyJetTowers( const edm::Event& event, const reco::CaloJet &jet, 
                  std::vector< subtower > &subtowers,      
                  std::vector< subtower > &Ecal_subtowers, 
                  std::vector< subtower > &Hcal_subtowers, 
                  std::vector< subtower > &HO_subtowers,
                  std::vector< double > &HPD_energies,  
                  std::vector< double > &RBX_energies,
                  const int iDbg = 0);

      unsigned int nCarrying( double fraction, const std::vector< double >& descending_energies );
      unsigned int hitsInNCarrying( double fraction, const std::vector< subtower >& descending_towers );
      
      enum Region{
    unknown_region = -1,
    HFneg, HEneg, HBneg, HBpos, HEpos, HFpos };
      
      int HBHE_oddness( int iEta, int depth );
      Region HBHE_region( int iEta, int depth );
      // tower-based. -1 means can't figure it out
      int HBHE_oddness( int iEta );
      Region region( int iEta );
      

      double fHPD_;
      double fRBX_;
      int    n90Hits_;
      double fSubDetector1_;
      double fSubDetector2_;
      double fSubDetector3_;
      double fSubDetector4_;
      double restrictedEMF_;
      int    nHCALTowers_;
      int    nECALTowers_;
      double approximatefHPD_;
      double approximatefRBX_;
      int    hitsInN90_;
      double approximatefSubDetector1_;
      double approximatefSubDetector2_;
      double approximatefSubDetector3_;
      double approximatefSubDetector4_;
      
      double fEB_, fEE_, fHB_, fHE_, fHO_, fLong_, fShort_;
      double fLS_, fHFOOT_;
      
      bool useRecHits_;
      edm::InputTag hbheRecHitsColl_;
      edm::InputTag hoRecHitsColl_;
      edm::InputTag hfRecHitsColl_;
      edm::InputTag ebRecHitsColl_;
      edm::InputTag eeRecHitsColl_;

      static std::atomic<int> sanity_checks_left_;

      edm::EDGetTokenT<HBHERecHitCollection> input_HBHERecHits_token_;
      edm::EDGetTokenT<HORecHitCollection> input_HORecHits_token_;
      edm::EDGetTokenT<HFRecHitCollection> input_HFRecHits_token_;
      edm::EDGetTokenT<EBRecHitCollection> input_EBRecHits_token_;
      edm::EDGetTokenT<EERecHitCollection> input_EERecHits_token_;


    };
  }
}
#endif