tracking/GenfitTools/trackrep/RKTrackRep/RKTrackRep.h

Go to the documentation of this file.

/* Copyright 2008-2009, Technische Universitaet Muenchen,
   Authors: Christian Hoeppner & Sebastian Neubert

   This file is part of GENFIT.

   GENFIT is free software: you can redistribute it and/or modify
   it under the terms of the GNU Lesser General Public License as published
   by the Free Software Foundation, either version 3 of the License, or
   (at your option) any later version.

   GENFIT is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
   GNU Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public License
   along with GENFIT.  If not, see <http://www.gnu.org/licenses/>.
*/

/*
 */

#ifndef RKTRACKREP_H
#define RKTRACKREP_H

#include "GFAbsTrackRep.h"
#include "GFDetPlane.h"
#include "GFTrackCand.h"
#include <TMatrixT.h>

class RKTrackRep : public GFAbsTrackRep {

 public:
  // Constructors/Destructors ---------
  RKTrackRep();
  RKTrackRep(const TVector3 &pos, const TVector3 &mom, const TVector3 &poserr, const TVector3 &momerr, const int &PDGCode);

  RKTrackRep(const GFTrackCand *aGFTrackCandPtr);

  RKTrackRep(const TVector3 &pos, const TVector3 &mom, const int &PDGCode);

  RKTrackRep(const GFDetPlane &pl, const TVector3 &mom, const int &PDGCode);

  virtual ~RKTrackRep();

  virtual GFAbsTrackRep *clone() const { return new RKTrackRep(*this); }
  virtual GFAbsTrackRep *prototype() const { return new RKTrackRep(); }


  double extrapolate(const GFDetPlane &, TMatrixT<double> &statePred, TMatrixT<double> &covPred);

  double extrapolate(const GFDetPlane &, TMatrixT<double> &statePred);

  void extrapolateToPoint(const TVector3 &pos, TVector3 &poca, TVector3 &dirInPoca);

  void extrapolateToLine(const TVector3 &point1, const TVector3 &point2, TVector3 &poca, TVector3 &dirInPoca, TVector3 &poca_onwire);


  double stepalong(double h, TVector3 &point, TVector3 &dir);


  TVector3 getPos(const GFDetPlane &);

  TVector3 getMom(const GFDetPlane &);

  void getPosMom(const GFDetPlane &, TVector3 &pos, TVector3 &mom);
  double getCharge() const { return fCharge; }

  int getPDG() { return fPdg; };

  void switchDirection() { fDirection = (!fDirection); }
  void setPDG(int);


  void setData(const TMatrixT<double> &st, const GFDetPlane &pl, const TMatrixT<double> *cov = nullptr, const TMatrixT<double> *aux = nullptr);

  const TMatrixT<double> *getAuxInfo(const GFDetPlane &pl);

  bool hasAuxInfo() { return true; }

  void getPosMomCov(const GFDetPlane &pl, TVector3 &pos, TVector3 &mom, TMatrixT<double> &cov); // to be implemented soon

 private:
  RKTrackRep &operator=(const RKTrackRep *) { return *this; }; // rhs // [R.K.03/2017] unused variable(s)


  bool RKutta(const GFDetPlane &plane, double *P, double &coveredDistance, std::vector<TVector3> &points, std::vector<double> &pointLengths, const double &maxLen = -1,
              bool calcCov = true) const;

  TVector3 poca2Line(const TVector3 &extr1, const TVector3 &extr2, const TVector3 &point) const;


  double Extrap(const GFDetPlane &plane, TMatrixT<double> *state, TMatrixT<double> *cov = nullptr) const;

  // RKTrackRep(const RKTrackRep& rhs){};

  // data members

  bool fDirection;

  int fPdg;
  double fMass;
  double fCharge;

  GFDetPlane fCachePlane;
  double fCacheSpu;
  double fSpu;
  TMatrixT<double> fAuxInfo;

 public:
  ClassDef(RKTrackRep, 4)
};

#endif