PreTrackMatchedMuon.h

Go to the documentation of this file.

#ifndef PHASE2GMT_PRETRACKMATCHEDMUON
#define PHASE2GMT_PRETRACKMATCHEDMUON

#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "DataFormats/L1TMuonPhase2/interface/Constants.h"
#include "DataFormats/L1TMuonPhase2/interface/MuonStub.h"
#include "DataFormats/L1TMuonPhase2/interface/SAMuon.h"
#include "DataFormats/Common/interface/Ptr.h"
#include "DataFormats/L1TrackTrigger/interface/TTTrack.h"
#include "DataFormats/L1TrackTrigger/interface/TTStub.h"
#include "DataFormats/L1TrackTrigger/interface/TTTypes.h"
#include "DataFormats/Phase2TrackerDigi/interface/Phase2TrackerDigi.h"
#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "DataFormats/L1Trigger/interface/Vertex.h"

#include <vector>

namespace Phase2L1GMT {

  class PreTrackMatchedMuon {
  public:
    PreTrackMatchedMuon(const uint& charge,
                        const uint& pt,
                        const int& eta,
                        const int& phi,
                        const int& z0,
                        const int& d0,
                        const uint& beta = 15)
        : charge_(charge),
          pt_(pt),
          eta_(eta),
          phi_(phi),
          z0_(z0),
          d0_(d0),
          beta_(beta),
          isGlobal_(false),
          quality_(0),
          stubID0_(4095),
          stubID1_(4095),
          stubID2_(4095),
          stubID3_(4095),
          stubID4_(4095),
          matchMask_(0),
          valid_(false) {}

    const uint charge() const { return charge_; }
    const uint pt() const { return pt_; }
    const int eta() const { return eta_; }
    const int phi() const { return phi_; }
    const int z0() const { return z0_; }
    const int d0() const { return d0_; }
    const uint beta() const { return beta_; }

    bool isGlobalMuon() const { return isGlobal_; }
    const int quality() const { return quality_; }
    const int offline_pt() const { return offline_pt_; }
    const float offline_eta() const { return offline_eta_; }
    const float offline_phi() const { return offline_phi_; }

    const uint stubID0() const { return stubID0_; }
    const uint stubID1() const { return stubID1_; }
    const uint stubID2() const { return stubID2_; }
    const uint stubID3() const { return stubID3_; }
    const uint stubID4() const { return stubID4_; }
    const uint matchMask() const { return matchMask_; }

    bool valid() const { return valid_; }

    void setQuality(uint quality) { quality_ = quality; }
    void setValid(bool v) { valid_ = v; }

    void setOfflineQuantities(float pt, float eta, float phi) {
      offline_pt_ = pt;
      offline_eta_ = eta;
      offline_phi_ = phi;
    }

    void addMuonRef(const l1t::SAMuonRef& ref) { muRef_.push_back(ref); }

    void resetGlobal() { isGlobal_ = false; }

    const l1t::SAMuonRefVector& muonRef() const { return muRef_; }
    void addStub(const l1t::MuonStubRef& stub, uint mask) {
      stubs_.push_back(stub);
      if (stub->tfLayer() == 0) {
        stubID0_ = stub->address();
        matchMask_ = matchMask_ | (mask);
      } else if (stub->tfLayer() == 1) {
        stubID1_ = stub->address();
        matchMask_ = matchMask_ | (mask << 2);
      } else if (stub->tfLayer() == 2) {
        stubID2_ = stub->address();
        matchMask_ = matchMask_ | (mask << 4);
      } else if (stub->tfLayer() == 3) {
        stubID3_ = stub->address();
        matchMask_ = matchMask_ | (mask << 6);
      } else if (stub->tfLayer() == 4) {
        stubID4_ = stub->address();
        matchMask_ = matchMask_ | (mask << 8);
      }
    }

    const l1t::MuonStubRefVector& stubs() const { return stubs_; }

    void setTrkPtr(const edm::Ptr<TTTrack<Ref_Phase2TrackerDigi_> >& trkPtr) { trkPtr_ = trkPtr; }

    const edm::Ptr<TTTrack<Ref_Phase2TrackerDigi_> > trkPtr() const { return trkPtr_; }

    void print() const {
      LogDebug("PreTrackMatchedMuon") << "preconstructed muon : charge=" << charge_ << " pt=" << offline_pt_ << ","
                                      << pt_ << " eta=" << offline_eta_ << "," << eta_ << " phi=" << offline_phi_ << ","
                                      << phi_ << " z0=" << z0_ << " d0=" << d0_ << " quality=" << quality_
                                      << " isGlobal=" << isGlobal_ << " valid=" << valid_ << " stubs: " << stubID0_
                                      << " " << stubID1_ << " " << stubID2_ << " " << stubID3_ << " " << stubID4_;
    }

    uint64_t lsb() const {
      wordtype w = 0;
      int bstart = 0;
      bstart = wordconcat<wordtype>(w, bstart, charge_ & 0x1, 1);
      bstart = wordconcat<wordtype>(w, bstart, pt_, BITSPT);
      bstart = wordconcat<wordtype>(w, bstart, phi_, BITSPHI);
      bstart = wordconcat<wordtype>(w, bstart, eta_, BITSETA);
      bstart = wordconcat<wordtype>(w, bstart, z0_, BITSZ0);
      bstart = wordconcat<wordtype>(w, bstart, d0_, BITSD0);
      return w.to_int();
    }

    uint64_t msb() const {
      wordtype w2 = 0;
      int bstart = 0;
      bstart = wordconcat<wordtype>(w2, bstart, stubID0_, BITSSTUBID);
      bstart = wordconcat<wordtype>(w2, bstart, stubID1_, BITSSTUBID);
      bstart = wordconcat<wordtype>(w2, bstart, stubID2_, BITSSTUBID);
      bstart = wordconcat<wordtype>(w2, bstart, stubID3_, BITSSTUBID);
      bstart = wordconcat<wordtype>(w2, bstart, stubID4_, BITSSTUBID);
      bstart = wordconcat<wordtype>(w2, bstart, isGlobal_, 1);
      bstart = wordconcat<wordtype>(w2, bstart, beta_, BITSMUONBETA);
      bstart = wordconcat<wordtype>(w2, bstart, quality_, BITSMATCHQUALITY);
      bstart = wordconcat<wordtype>(w2, bstart, valid_, 1);

      return w2.to_int();
    }

    void printWord() const {
      LogDebug("PreTrackMatchedMuon") << "PreTrackMatchedMuon : word=" << std::setfill('0') << std::setw(16) << std::hex
                                      << (long long unsigned int)(msb() >> 2) << std::setfill('0') << std::setw(16)
                                      << std::hex
                                      << (long long unsigned int)((lsb() | (msb() << 62)) & 0xffffffffffffffff);
    }

  private:
    uint charge_;
    uint pt_;
    int eta_;
    int phi_;
    int z0_;
    int d0_;
    uint beta_;
    bool isGlobal_;
    uint quality_;
    float offline_pt_;
    float offline_eta_;
    float offline_phi_;
    uint stubID0_;
    uint stubID1_;
    uint stubID2_;
    uint stubID3_;
    uint stubID4_;
    uint matchMask_;

    bool valid_;
    l1t::MuonStubRefVector stubs_;
    l1t::SAMuonRefVector muRef_;
    edm::Ptr<TTTrack<Ref_Phase2TrackerDigi_> > trkPtr_;
  };
}  // namespace Phase2L1GMT

#endif