TkElectron

Macros

Line Code
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62
#ifndef DataFormatsL1TCorrelator_TkElectron_h
#define DataFormatsL1TCorrelator_TkElectron_h

// -*- C++ -*-
//
// Package:     L1Trigger
// Class  :     TkEm
//

#include "DataFormats/Common/interface/Ptr.h"

#include "DataFormats/L1Trigger/interface/EGamma.h"

#include "DataFormats/L1TCorrelator/interface/TkEm.h"
#include "DataFormats/L1TCorrelator/interface/TkEmFwd.h"

#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"

#include <vector>

namespace l1t {

  class TkElectron : public TkEm {
  public:
    typedef TTTrack<Ref_Phase2TrackerDigi_> L1TTTrackType;
    typedef std::vector<L1TTTrackType> L1TTTrackCollection;

    TkElectron();

    TkElectron(const LorentzVector& p4,
               const edm::Ptr<L1Candidate>& egCaloPtr,
               const edm::Ptr<L1TTTrackType>& trkPtr,
               float tkisol = -999.);

    TkElectron(const LorentzVector& p4, float tkisol = -999.);

    // ---------- const member functions ---------------------

    const edm::Ptr<L1TTTrackType>& trkPtr() const { return trkPtr_; }

    float trkzVtx() const { return trkzVtx_; }
    float idScore() const { return idScore_; }
    // ---------- member functions ---------------------------

    void setTrkPtr(const edm::Ptr<L1TTTrackType>& tkPtr) { trkPtr_ = tkPtr; }
    void setTrkzVtx(float TrkzVtx) { trkzVtx_ = TrkzVtx; }
    void setIdScore(float score) { idScore_ = score; }

    l1gt::Electron hwObj() const {
      if (encoding() != HWEncoding::GT) {
        throw cms::Exception("RuntimeError") << "TkElectron::hwObj : encoding is not in GT format!" << std::endl;
      }
      return l1gt::Electron::unpack_ap(egBinaryWord<l1gt::Electron::BITWIDTH>());
    }

  private:
    edm::Ptr<L1TTTrackType> trkPtr_;
    float trkzVtx_;
    float idScore_;
  };
}  // namespace l1t
#endif