RecoTracker/MkFitCore/interface/Hit.h - master - githubmirror/cmssw

RecoTracker/MkFitCore/interface/Hit.h
278 строк10 KB
Mario Masciovecchio
Speedup of mkFit hit converters, and update of HLT customization
15 янв 2025, 14:38
15 янв 2025, 14:3882b554f
#ifndef RecoTracker_MkFitCore_interface_Hit_h
#define RecoTracker_MkFitCore_interface_Hit_h

#include "RecoTracker/MkFitCore/interface/Config.h"
#include "RecoTracker/MkFitCore/interface/MatrixSTypes.h"

#include <cmath>
#include <vdt/atan2.h>
#include <vdt/cos.h>
#include <vdt/log.h>
#include <vdt/sin.h>
#include <vdt/sincos.h>
#include <vdt/tan.h>
#include <vector>
#include <string_view>

namespace mkfit {

  template <typename T>
  inline T sqr(T x) {
    return x * x;
  }
  template <typename T>
  inline T cube(T x) {
    return x * x * x;
  }

  inline float squashPhiGeneral(float phi) {
    return phi - floor(0.5 * Const::InvPI * (phi + Const::PI)) * Const::TwoPI;
  }

  inline float squashPhiMinimal(float phi) {
    return phi >= Const::PI ? phi - Const::TwoPI : (phi < -Const::PI ? phi + Const::TwoPI : phi);
  }

  inline float getRad2(float x, float y) { return x * x + y * y; }

  inline float getInvRad2(float x, float y) { return 1.0f / (x * x + y * y); }

  inline float getPhi(float x, float y) { return vdt::fast_atan2f(y, x); }

  inline float getTheta(float r, float z) { return vdt::fast_atan2f(r, z); }

  inline float getEta(float r, float z) { return -1.0f * vdt::fast_logf(vdt::fast_tanf(getTheta(r, z) / 2.0f)); }

  inline float getEta(float theta) { return -1.0f * vdt::fast_logf(vdt::fast_tanf(theta / 2.0f)); }

  inline float getEta(float x, float y, float z) {
    const float theta = vdt::fast_atan2f(std::sqrt(x * x + y * y), z);
    return -1.0f * vdt::fast_logf(vdt::fast_tanf(theta / 2.0f));
  }

  inline float getHypot(float x, float y) { return std::sqrt(x * x + y * y); }

  inline float getRadErr2(float x, float y, float exx, float eyy, float exy) {
    return (x * x * exx + y * y * eyy + 2.0f * x * y * exy) / getRad2(x, y);
  }

  inline float getInvRadErr2(float x, float y, float exx, float eyy, float exy) {
    return (x * x * exx + y * y * eyy + 2.0f * x * y * exy) / cube(getRad2(x, y));
  }

  inline float getPhiErr2(float x, float y, float exx, float eyy, float exy) {
    const float rad2 = getRad2(x, y);
    return (y * y * exx + x * x * eyy - 2.0f * x * y * exy) / (rad2 * rad2);
  }

  inline float getThetaErr2(
      float x, float y, float z, float exx, float eyy, float ezz, float exy, float exz, float eyz) {
    const float rad2 = getRad2(x, y);
    const float rad = std::sqrt(rad2);
    const float hypot2 = rad2 + z * z;
    const float dthetadx = x * z / (rad * hypot2);
    const float dthetady = y * z / (rad * hypot2);
    const float dthetadz = -rad / hypot2;
    return dthetadx * dthetadx * exx + dthetady * dthetady * eyy + dthetadz * dthetadz * ezz +
           2.0f * dthetadx * dthetady * exy + 2.0f * dthetadx * dthetadz * exz + 2.0f * dthetady * dthetadz * eyz;
  }

  inline float getEtaErr2(float x, float y, float z, float exx, float eyy, float ezz, float exy, float exz, float eyz) {
    const float rad2 = getRad2(x, y);
    const float detadx = -x / (rad2 * std::sqrt(1 + rad2 / (z * z)));
    const float detady = -y / (rad2 * std::sqrt(1 + rad2 / (z * z)));
    const float detadz = 1.0f / (z * std::sqrt(1 + rad2 / (z * z)));
    return detadx * detadx * exx + detady * detady * eyy + detadz * detadz * ezz + 2.0f * detadx * detady * exy +
           2.0f * detadx * detadz * exz + 2.0f * detady * detadz * eyz;
  }

  inline float getPxPxErr2(float ipt, float phi, float vipt, float vphi) {  // ipt = 1/pT, v = variance
    const float iipt2 = 1.0f / (ipt * ipt);                                 //iipt = 1/(1/pT) = pT
    float cosP;
    float sinP;
    vdt::fast_sincosf(phi, sinP, cosP);
    return iipt2 * (iipt2 * cosP * cosP * vipt + sinP * sinP * vphi);
  }

  inline float getPyPyErr2(float ipt, float phi, float vipt, float vphi) {  // ipt = 1/pT, v = variance
    const float iipt2 = 1.0f / (ipt * ipt);                                 //iipt = 1/(1/pT) = pT
    float cosP;
    float sinP;
    vdt::fast_sincosf(phi, sinP, cosP);
    return iipt2 * (iipt2 * sinP * sinP * vipt + cosP * cosP * vphi);
  }

  inline float getPzPzErr2(float ipt, float theta, float vipt, float vtheta) {  // ipt = 1/pT, v = variance
    const float iipt2 = 1.0f / (ipt * ipt);                                     //iipt = 1/(1/pT) = pT
    const float cotT = 1.0f / vdt::fast_tanf(theta);
    const float cscT = 1.0f / vdt::fast_sinf(theta);
    return iipt2 * (iipt2 * cotT * cotT * vipt + cscT * cscT * cscT * cscT * vtheta);
  }

  struct MCHitInfo {
    MCHitInfo() {}
    MCHitInfo(int track, int layer, int ithlayerhit, int mcHitID)
        : mcTrackID_(track), layer_(layer), ithLayerHit_(ithlayerhit), mcHitID_(mcHitID) {}

    int mcTrackID_;
    int layer_;
    int ithLayerHit_;
    int mcHitID_;

    int mcTrackID() const { return mcTrackID_; }
    int layer() const { return layer_; }
    int mcHitID() const { return mcHitID_; }
    static void reset();
  };
  typedef std::vector<MCHitInfo> MCHitInfoVec;

  struct MeasurementState {
  public:
    MeasurementState() {}
    MeasurementState(const SVector3& p, const SMatrixSym33& e) : pos_(p), err_(e) {}

    const SVector3& parameters() const { return pos_; }
    const SMatrixSym33& errors() const { return err_; }

    SVector3 pos_;
    SMatrixSym33 err_;
  };

  class Hit {
  public:
    Hit() : mcHitID_(-1) {}

    Hit(const SVector3& position, const SMatrixSym33& error, int mcHitID = -1)
        : state_(position, error), mcHitID_(mcHitID) {}

    const SVector3& position() const { return state_.parameters(); }
    const SVector3& parameters() const { return state_.parameters(); }
    const SMatrixSym33& error() const { return state_.err_; }

    const float* posArray() const { return state_.pos_.Array(); }
    const float* errArray() const { return state_.err_.Array(); }

    // Non-const versions needed for CopyOut of Matriplex.
    SVector3& parameters_nc() { return state_.pos_; }
    SMatrixSym33& error_nc() { return state_.err_; }

    float r() const {
      return sqrtf(state_.parameters().At(0) * state_.parameters().At(0) +
                   state_.parameters().At(1) * state_.parameters().At(1));
    }
    float x() const { return state_.parameters().At(0); }
    float y() const { return state_.parameters().At(1); }
    float z() const { return state_.parameters().At(2); }
    float exx() const { return state_.errors().At(0, 0); }
    float eyy() const { return state_.errors().At(1, 1); }
    float ezz() const { return state_.errors().At(2, 2); }
    float phi() const { return getPhi(state_.parameters().At(0), state_.parameters().At(1)); }
    float eta() const {
      return getEta(state_.parameters().At(0), state_.parameters().At(1), state_.parameters().At(2));
    }
    float ephi() const { return getPhiErr2(x(), y(), exx(), eyy(), state_.errors().At(0, 1)); }
    float eeta() const {
      return getEtaErr2(x(),
                        y(),
                        z(),
                        exx(),
                        eyy(),
                        ezz(),
                        state_.errors().At(0, 1),
                        state_.errors().At(0, 2),
                        state_.errors().At(1, 2));
    }

    const MeasurementState& measurementState() const { return state_; }

    int mcHitID() const { return mcHitID_; }
    int layer(const MCHitInfoVec& globalMCHitInfo) const { return globalMCHitInfo[mcHitID_].layer(); }
    int mcTrackID(const MCHitInfoVec& globalMCHitInfo) const { return globalMCHitInfo[mcHitID_].mcTrackID(); }

    // Indices for "fake" hit addition
    // Only if fake_hit_idx==-1, then count as missing hit for candidate score
    static constexpr int kHitMissIdx = -1;        //  hit is missed
    static constexpr int kHitStopIdx = -2;        //  track is stopped
    static constexpr int kHitEdgeIdx = -3;        //  track not in sensitive region of detector
    static constexpr int kHitMaxClusterIdx = -5;  //  hit cluster size > maxClusterSize
    static constexpr int kHitInGapIdx = -7;       //  track passing through inactive module
    static constexpr int kHitCCCFilterIdx = -9;   //  hit filtered via CCC (unused)

    static constexpr int kMinChargePerCM = 1620;
    static constexpr int kChargePerCMBits = 8;
    static constexpr int kDetIdInLayerBits = 14;
    static constexpr int kClusterSizeBits = 5;
    static constexpr int kMaxClusterSize = (1 << kClusterSizeBits) - 1;

    struct PackedData {
      unsigned int detid_in_layer : kDetIdInLayerBits;
      unsigned int charge_pcm : kChargePerCMBits;  // MIMI see set/get funcs; applicable for phase-0/1
      unsigned int span_rows : kClusterSizeBits;
      unsigned int span_cols : kClusterSizeBits;

      PackedData() : detid_in_layer(0), charge_pcm(0), span_rows(0), span_cols(0) {}

      void set_charge_pcm(int cpcm) {
        if (cpcm < kMinChargePerCM)
          charge_pcm = 0;
        else
          charge_pcm = std::min((1 << kChargePerCMBits) - 1, ((cpcm - kMinChargePerCM) >> 3) + 1);
      }
      unsigned int get_charge_pcm() const {
        if (charge_pcm == 0)
          return 0;
        else
          return ((charge_pcm - 1) << 3) + kMinChargePerCM;
      }
    };

    unsigned int detIDinLayer() const { return pdata_.detid_in_layer; }
    unsigned int chargePerCM() const { return pdata_.get_charge_pcm(); }
    unsigned int spanRows() const { return pdata_.span_rows + 1; }
    unsigned int spanCols() const { return pdata_.span_cols + 1; }

    static unsigned int minChargePerCM() { return kMinChargePerCM; }
    static unsigned int maxChargePerCM() { return kMinChargePerCM + (((1 << kChargePerCMBits) - 2) << 3); }
    static unsigned int maxSpan() { return kMaxClusterSize; }

    void setupAsPixel(unsigned int id, int rows, int cols) {
      pdata_.detid_in_layer = id;
      pdata_.charge_pcm = (1 << kChargePerCMBits) - 1;
      pdata_.span_rows = std::min(kMaxClusterSize, rows - 1);
      pdata_.span_cols = std::min(kMaxClusterSize, cols - 1);
    }

    void setupAsStrip(unsigned int id, int cpcm, int rows) {
      pdata_.detid_in_layer = id;
      pdata_.set_charge_pcm(cpcm);
      pdata_.span_rows = std::min(kMaxClusterSize, rows - 1);
    }

  private:
    MeasurementState state_;
    int mcHitID_;
    PackedData pdata_;
  };

  typedef std::vector<Hit> HitVec;

  struct HitOnTrack {
    int index : 24;
    int layer : 8;

    HitOnTrack() : index(-1), layer(-1) {}
    HitOnTrack(int i, int l) : index(i), layer(l) {}

    bool operator<(const HitOnTrack o) const {
      if (layer != o.layer)
        return layer < o.layer;
      return index < o.index;
    }
  };

  typedef std::vector<HitOnTrack> HoTVec;

  void print(std::string_view label, const MeasurementState& s);

}  // end namespace mkfit
#endif