TrackConverter.h

Go to the documentation of this file.

#ifndef PHASE2GMT_TRACKCONVERTER
#define PHASE2GMT_TRACKCONVERTER

#include "ConvertedTTTrack.h"
#include "Constants.h"
#include "DataFormats/L1TMuonPhase2/interface/TrackerMuon.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

namespace Phase2L1GMT {

  class TrackConverter {
  public:
    TrackConverter(const edm::ParameterSet& iConfig) : verbose_(iConfig.getParameter<int>("verbose")) {}
    ~TrackConverter() {}

    std::vector<ConvertedTTTrack> convertTracks(const std::vector<edm::Ptr<l1t::TrackerMuon::L1TTTrackType> >& tracks) {
      std::vector<ConvertedTTTrack> out;
      out.reserve(tracks.size());
      for (const auto& t : tracks)
        out.push_back(convert(t));
      return out;
    }

  private:
    int verbose_;
    typedef ap_uint<96> wordtype;

    uint generateQuality(const edm::Ptr<TTTrack<Ref_Phase2TrackerDigi_> >& track) { return 1; }

    uint ptLookup(uint absCurv) {
      for (auto i : ptShifts) {
        if (absCurv >= uint(i[0]) && absCurv < uint(i[1])) {
          if (i[2] < 0)
            return i[4];
          else
            return (absCurv >> i[2]) + i[3];
        }
      }
      return 0;
    }

    uint etaLookup(uint absTanL) {
      for (auto i : etaShifts) {
        if (absTanL >= uint(i[0]) && absTanL < uint(i[1])) {
          if (i[2] < 0)
            return i[4];
          else
            return (absTanL >> i[2]) + i[3];
        }
      }
      return 0;
    }

    ConvertedTTTrack convert(const edm::Ptr<TTTrack<Ref_Phase2TrackerDigi_> >& track) {
      uint charge = (track->rInv() < 0) ? 1 : 0;
      int curvature = track->rInv() * (1 << (BITSTTCURV - 1)) / maxCurv_;
      int phi = track->phi() * (1 << (BITSPHI - 1)) / (M_PI);
      int tanLambda = track->tanL() * (1 << (BITSTTTANL - 1)) / maxTanl_;
      int z0 = track->z0() * (1 << (BITSZ0 - 1)) / maxZ0_;
      int d0 = track->d0() * (1 << (BITSD0 - 1)) / maxD0_;
      //calculate pt
      uint absCurv = curvature > 0 ? (curvature) : (-curvature);
      uint pt = ptLUT[ptLookup(absCurv)];
      uint quality = generateQuality(track);
      uint absTanL = tanLambda > 0 ? (tanLambda) : (-tanLambda);
      uint absEta = etaLUT[etaLookup(absTanL)];
      int eta = tanLambda > 0 ? (absEta) : (-absEta);

      ap_int<BITSPHI> phiSec = ap_int<BITSPHI>(phi) -
                               ap_int<BITSPHI>((track->phiSector() * 40 * M_PI / 180.) * (1 << (BITSPHI - 1)) / (M_PI));
      ap_int<BITSPHI> phiCorrected = ap_int<BITSPHI>(phiSec + track->phiSector() * 910);

      wordtype word = 0;
      int bstart = 0;
      bstart = wordconcat<wordtype>(word, bstart, curvature, BITSTTCURV);
      bstart = wordconcat<wordtype>(word, bstart, phiSec, BITSTTPHI);
      bstart = wordconcat<wordtype>(word, bstart, tanLambda, BITSTTTANL);
      bstart = wordconcat<wordtype>(word, bstart, z0, BITSZ0);
      bstart = wordconcat<wordtype>(word, bstart, d0, BITSD0);
      bstart = wordconcat<wordtype>(word, bstart, uint(track->chi2()), 4);

      ConvertedTTTrack convertedTrack(charge, curvature, absEta, pt, eta, phiCorrected.to_int(), z0, d0, quality, word);
      convertedTrack.setOfflineQuantities(track->momentum().transverse(), track->eta(), track->phi());
      if (verbose_)
        convertedTrack.print();
      convertedTrack.setTrkPtr(track);
      return convertedTrack;
    }
  };
}  // namespace Phase2L1GMT

#endif