RZLine.h

Go to the documentation of this file.

#ifndef PixelTrackFitting_RZLine_H
#define PixelTrackFitting_RZLine_H

#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryCommonDetAlgo/interface/GlobalError.h"
#include "CommonTools/Utils/interface/DynArray.h"

#include "CommonTools/Statistics/interface/LinearFit.h"

#include <vector>

class RZLine {
public:
  struct ErrZ2_tag {};

  template <typename P, typename E, typename B>
  RZLine(const P& points, const E& errors, const B& isBarrel) {
    const size_t n = points.size();
    declareDynArray(float, n, r);
    declareDynArray(float, n, z);
    declareDynArray(float, n, errZ2);
    for (size_t i = 0; i < n; ++i) {
      const GlobalPoint& p = points[i];
      r[i] = p.perp();
      z[i] = p.z();
    }

    float simpleCot2 = sqr((z[n - 1] - z[0]) / (r[n - 1] - r[0]));
    for (size_t i = 0; i < n; ++i) {
      errZ2[i] = (isBarrel[i]) ? errors[i].czz() : errors[i].rerr(points[i]) * simpleCot2;
    }

    calculate(r, z, errZ2);
  }

  RZLine(const std::vector<float>& r, const std::vector<float>& z, const std::vector<float>& errZ) {
    const size_t n = errZ.size();
    declareDynArray(float, n, errZ2);
    for (size_t i = 0; i < n; ++i)
      errZ2[i] = sqr(errZ[i]);
    calculate(r, z, errZ2);
  }

  template <size_t N>
  RZLine(const std::array<float, N>& r, const std::array<float, N>& z, const std::array<float, N>& errZ) {
    std::array<float, N> errZ2;
    for (size_t i = 0; i < N; ++i)
      errZ2[i] = sqr(errZ[i]);
    calculate(r, z, errZ2);
  }

  template <typename T>
  RZLine(const T& r, const T& z, const T& errZ2, ErrZ2_tag) {
    calculate(r, z, errZ2);
  }

  float cotTheta() const { return cotTheta_; }
  float intercept() const { return intercept_; }
  float covss() const { return covss_; }
  float covii() const { return covii_; }
  float covsi() const { return covsi_; }

  float chi2() const { return chi2_; }

private:
  template <typename R, typename Z, typename E>
  void calculate(const R& r, const Z& z, const E& errZ2) {
    const size_t n = r.size();
    linearFit(r.data(), z.data(), n, errZ2.data(), cotTheta_, intercept_, covss_, covii_, covsi_);
    chi2_ = 0.f;
    for (size_t i = 0; i < n; ++i) {
      chi2_ += sqr(((z[i] - intercept_) - cotTheta_ * r[i])) / errZ2[i];
    }
  }

  template <typename T>
  T sqr(T t) {
    return t * t;
  }

  float cotTheta_;
  float intercept_;
  float covss_;
  float covii_;
  float covsi_;
  float chi2_;
};
#endif