CaloExtractor.h

Go to the documentation of this file.

#ifndef MuonIsolation_CaloExtractor_H
#define MuonIsolation_CaloExtractor_H
 
#include <string>
 
#include "PhysicsTools/IsolationAlgos/interface/IsoDepositExtractor.h"
 
#include "FWCore/Framework/interface/ConsumesCollector.h"
 
#include "DataFormats/RecoCandidate/interface/IsoDeposit.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
 
#include "DataFormats/CaloTowers/interface/CaloTowerCollection.h"
 
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
 
#include "MagneticField/Engine/interface/MagneticField.h"
#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
 
namespace muonisolation {
 
  class CaloExtractor : public reco::isodeposit::IsoDepositExtractor {
  public:
    CaloExtractor(){};
    CaloExtractor(const edm::ParameterSet& par, edm::ConsumesCollector&& iC);
 
    ~CaloExtractor() override {}
 
    void fillVetos(const edm::Event& ev, const edm::EventSetup& evSetup, const reco::TrackCollection& tracks) override;
    reco::IsoDeposit deposit(const edm::Event& ev,
                             const edm::EventSetup& evSetup,
                             const reco::Track& track) const override;
 
    static GlobalPoint MuonAtCaloPosition(
        const reco::Track& muon, const double bz, const GlobalPoint& endpos, bool fixVxy = false, bool fixVz = false);
 
  private:
    // CaloTower Collection Label
    edm::EDGetTokenT<CaloTowerCollection> theCaloTowerCollectionToken;
 
    // Label of deposit
    std::string theDepositLabel;
 
    const edm::ESGetToken<CaloGeometry, CaloGeometryRecord> theCaloGeomToken;
    const edm::ESGetToken<MagneticField, IdealMagneticFieldRecord> theFieldToken;
 
    // Cone cuts and thresholds
    double theWeight_E;
    double theWeight_H;
    double theThreshold_E;
    double theThreshold_H;
    double theDR_Veto_E;
    double theDR_Veto_H;
    double theDR_Max;
    bool vertexConstraintFlag_XY;
    bool vertexConstraintFlag_Z;
 
    // Vector of calo Ids to veto
    std::vector<DetId> theVetoCollection;
 
    // Determine noise for HCAL and ECAL (take some defaults for the time being)
    double noiseEcal(const CaloTower& tower) const;
    double noiseHcal(const CaloTower& tower) const;
  };
 
}  // namespace muonisolation
 
#endif