IsolatedPixelTrackCandidate.h

Go to the documentation of this file.

#ifndef HcalIsolatedTrack_IsolatedPixelTrackCandidate_h
#define HcalIsolatedTrack_IsolatedPixelTrackCandidate_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/L1Trigger/interface/Jet.h"
#include "DataFormats/L1Trigger/interface/Tau.h"

#include "DataFormats/HcalIsolatedTrack/interface/IsolatedPixelTrackCandidateFwd.h"

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

namespace reco {

  class IsolatedPixelTrackCandidate : public RecoCandidate {
  public:
    // default constructor
    IsolatedPixelTrackCandidate() : RecoCandidate() {
      enIn_ = -1;
      enOut_ = -1;
      nhitIn_ = -1;
      nhitOut_ = -1;
      maxPtPxl_ = -1;
      sumPtPxl_ = -1;
      etaEcal_ = 0;
      phiEcal_ = 0;
      etaPhiEcal_ = false;
    }
    IsolatedPixelTrackCandidate(const LorentzVector& v) : RecoCandidate(0, v) {
      enIn_ = -1;
      enOut_ = -1;
      nhitIn_ = -1;
      nhitOut_ = -1;
      maxPtPxl_ = -1;
      sumPtPxl_ = -1;
      etaEcal_ = 0;
      phiEcal_ = 0;
      etaPhiEcal_ = false;
    }
    IsolatedPixelTrackCandidate(const reco::TrackRef& tr,
                                const l1extra::L1JetParticleRef& tauRef,
                                double max,
                                double sum)
        : RecoCandidate(0, LorentzVector((tr.get()->px()), (tr.get())->py(), (tr.get())->pz(), (tr.get())->p())),
          track_(tr),
          l1tauJet_(tauRef),
          maxPtPxl_(max),
          sumPtPxl_(sum) {
      enIn_ = -1;
      enOut_ = -1;
      nhitIn_ = -1;
      nhitOut_ = -1;
      etaEcal_ = 0;
      phiEcal_ = 0;
      etaPhiEcal_ = false;
    }
    // constructor from a track using l1t
    IsolatedPixelTrackCandidate(const reco::TrackRef& tr, const l1t::TauRef& tauRef, double max, double sum)
        : RecoCandidate(0, LorentzVector((tr.get()->px()), (tr.get())->py(), (tr.get())->pz(), (tr.get())->p())),
          track_(tr),
          l1ttauJet_(tauRef),
          maxPtPxl_(max),
          sumPtPxl_(sum) {
      enIn_ = -1;
      enOut_ = -1;
      nhitIn_ = -1;
      nhitOut_ = -1;
      etaEcal_ = 0;
      phiEcal_ = 0;
      etaPhiEcal_ = false;
    }

    IsolatedPixelTrackCandidate(
        const l1extra::L1JetParticleRef& tauRef, double enIn, double enOut, int nhitIn, int nhitOut)
        : RecoCandidate(0, LorentzVector(tauRef->px(), tauRef->py(), tauRef->pz(), tauRef->p())),
          l1tauJet_(tauRef),
          enIn_(enIn),
          enOut_(enOut),
          nhitIn_(nhitIn),
          nhitOut_(nhitOut) {
      maxPtPxl_ = -1;
      sumPtPxl_ = -1;
      etaEcal_ = 0;
      phiEcal_ = 0;
      etaPhiEcal_ = false;
    }
    IsolatedPixelTrackCandidate(const l1t::TauRef& tauRef, double enIn, double enOut, int nhitIn, int nhitOut)
        : RecoCandidate(0, LorentzVector(tauRef->px(), tauRef->py(), tauRef->pz(), tauRef->p())),
          l1ttauJet_(tauRef),
          enIn_(enIn),
          enOut_(enOut),
          nhitIn_(nhitIn),
          nhitOut_(nhitOut) {
      maxPtPxl_ = -1;
      sumPtPxl_ = -1;
      etaEcal_ = 0;
      phiEcal_ = 0;
      etaPhiEcal_ = false;
    }
    IsolatedPixelTrackCandidate(const IsolatedPixelTrackCandidate&);

    ~IsolatedPixelTrackCandidate() override;

    IsolatedPixelTrackCandidate* clone() const override;

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

    double maxPtPxl() const { return maxPtPxl_; }
    void setMaxPtPxl(double mptpxl) { maxPtPxl_ = mptpxl; }

    double sumPtPxl() const { return sumPtPxl_; }
    void setSumPtPxl(double sumptpxl) { sumPtPxl_ = sumptpxl; }

    virtual l1extra::L1JetParticleRef l1tau() const;
    void setL1TauJet(const l1extra::L1JetParticleRef& tauRef) { l1tauJet_ = tauRef; }

    virtual l1t::TauRef l1ttau() const;
    void setL1TTauJet(const l1t::TauRef& tauRef) { l1ttauJet_ = tauRef; }

    double energyIn() const { return enIn_; }
    void setEnergyIn(double a) { enIn_ = a; }

    double energyOut() const { return enOut_; }
    void setEnergyOut(double a) { enOut_ = a; }

    int nHitIn() const { return nhitIn_; }
    void setNHitIn(int a) { nhitIn_ = a; }

    int nHitOut() const { return nhitOut_; }
    void setNHitOut(int a) { nhitOut_ = a; }

    std::pair<int, int> towerIndex() const;

    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 etaPhiEcalValid() const { return etaPhiEcal_; }

  private:
    bool overlap(const Candidate&) const override;
    reco::TrackRef track_;
    l1extra::L1JetParticleRef l1tauJet_;
    l1t::TauRef l1ttauJet_;
    double maxPtPxl_;
    double sumPtPxl_;
    double enIn_;
    double enOut_;
    int nhitIn_;
    int nhitOut_;
    bool etaPhiEcal_;
    double etaEcal_, phiEcal_;
  };

}  // namespace reco

#endif