KinematicParameters.h

Go to the documentation of this file.

#ifndef KinematicParameters_H
#define KinematicParameters_H

#include "RecoVertex/KinematicFitPrimitives/interface/Matrices.h"
#include "DataFormats/GeometryVector/interface/GlobalPoint.h"
#include "DataFormats/GeometryVector/interface/GlobalVector.h"

class KinematicParameters {
public:
  typedef ROOT::Math::SVector<double, 7> AlgebraicVector7;

  KinematicParameters() : vl(false) {}

  template <typename... Args>
  KinematicParameters(Args... args) : par(args...), vl(true) {}

  KinematicParameters(const AlgebraicVector7& pr) : par(pr), vl(true) {}

  AlgebraicVector7 const& vector() const { return par; }

  double operator()(const int i) const { return par(i); }

  GlobalVector momentum() const { return GlobalVector(par[3], par[4], par[5]); }

  GlobalPoint position() const { return GlobalPoint(par[0], par[1], par[2]); }

  double mass() const { return par(6); }

  double energy() const { return sqrt(par(3) * par(3) + par(4) * par(4) + par(5) * par(5) + par(6) * par(6)); }

  bool isValid() const { return vl; }

private:
  AlgebraicVector7 par;
  bool vl;
};

#endif