HcalIsolatedTrackCandidate.h

Go to the documentation of this file.

#ifndef HcalIsolatedTrack_HcalIsolatedTrackCandidate_h
#define HcalIsolatedTrack_HcalIsolatedTrackCandidate_h

#include "DataFormats/RecoCandidate/interface/RecoCandidate.h"
#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/L1Trigger/interface/L1JetParticle.h"
#include "DataFormats/L1Trigger/interface/L1JetParticleFwd.h"
#include "DataFormats/Math/interface/LorentzVector.h"
#include "DataFormats/HcalIsolatedTrack/interface/HcalIsolatedTrackCandidateFwd.h"

#include <vector>
#include <map>
#include <utility>

namespace reco {

  class HcalIsolatedTrackCandidate : public RecoCandidate {
  public:
    // default constructor
    HcalIsolatedTrackCandidate() : RecoCandidate() {
      maxP_ = -1;
      enEcal_ = -1;
      ptL1_ = etaL1_ = phiL1_ = 0;
      etaEcal_ = phiEcal_ = 0;
      etaHcal_ = phiHcal_ = ietaHcal_ = iphiHcal_ = 0;
      etaPhiEcal_ = etaPhiHcal_ = false;
    }
    HcalIsolatedTrackCandidate(const LorentzVector& v) : RecoCandidate(0, v) {
      maxP_ = -1;
      enEcal_ = -1;
      ptL1_ = etaL1_ = phiL1_ = 0;
      etaEcal_ = phiEcal_ = 0;
      etaHcal_ = phiHcal_ = ietaHcal_ = iphiHcal_ = 0;
      etaPhiEcal_ = etaPhiHcal_ = false;
    }
    HcalIsolatedTrackCandidate(const reco::TrackRef& tr, double max, double ene)
        : RecoCandidate(0, LorentzVector((tr.get()->px()), (tr.get())->py(), (tr.get())->pz(), (tr.get())->p())),
          track_(tr),
          maxP_(max),
          enEcal_(ene) {
      ptL1_ = etaL1_ = phiL1_ = 0;
      etaEcal_ = phiEcal_ = 0;
      etaHcal_ = phiHcal_ = ietaHcal_ = iphiHcal_ = 0;
      etaPhiEcal_ = etaPhiHcal_ = false;
    }
    HcalIsolatedTrackCandidate(const HcalIsolatedTrackCandidate&);

    ~HcalIsolatedTrackCandidate() override;

    HcalIsolatedTrackCandidate* clone() const override;

    reco::TrackRef track() const override;
    void setTrack(const reco::TrackRef& tr) { track_ = tr; }

    double maxP() const { return maxP_; }
    void SetMaxP(double mp) { maxP_ = mp; }

    virtual l1extra::L1JetParticleRef l1jet() const;
    void setL1Jet(const l1extra::L1JetParticleRef& jetRef) { l1Jet_ = jetRef; }
    std::pair<double, double> EtaPhiL1() const { return std::pair<double, double>(etaL1_, phiL1_); }
    math::XYZTLorentzVector l1jetp() const;
    void setL1(double pt, double eta, double phi) {
      ptL1_ = pt;
      etaL1_ = eta;
      phiL1_ = phi;
    }

    double energyEcal() const { return enEcal_; }
    void SetEnergyEcal(double a) { enEcal_ = a; }

    void SetEtaPhiEcal(double eta, double phi) {
      etaEcal_ = eta;
      phiEcal_ = phi;
      etaPhiEcal_ = true;
    }
    std::pair<double, double> EtaPhiEcal() const {
      return ((etaPhiEcal_) ? std::pair<double, double>(etaEcal_, phiEcal_) : std::pair<double, double>(0, 0));
    }
    bool etaPhiEcal() const { return etaPhiEcal_; }

    void SetEtaPhiHcal(double eta, double phi, int ieta, int iphi) {
      etaHcal_ = eta;
      phiHcal_ = phi;
      ietaHcal_ = ieta;
      iphiHcal_ = iphi;
      etaPhiHcal_ = true;
    }
    std::pair<double, double> EtaPhiHcal() const {
      return ((etaPhiHcal_) ? std::pair<double, double>(etaHcal_, phiHcal_) : std::pair<double, double>(0, 0));
    }
    std::pair<int, int> towerIndex() const {
      return ((etaPhiHcal_) ? std::pair<int, int>(ietaHcal_, iphiHcal_) : std::pair<int, int>(0, 0));
    }
    bool etaPhiHcal() const { return etaPhiHcal_; }

  private:
    bool overlap(const Candidate&) const override;
    reco::TrackRef track_;
    l1extra::L1JetParticleRef l1Jet_;
    double maxP_;
    double enEcal_;
    double ptL1_, etaL1_, phiL1_;
    bool etaPhiEcal_;
    double etaEcal_, phiEcal_;
    bool etaPhiHcal_;
    double etaHcal_, phiHcal_;
    int ietaHcal_, iphiHcal_;
  };

}  // namespace reco

#endif