GFAbsTrackRep.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file GFAbsTrackRep.h

/* Copyright 2008-2010, 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 GFABSTRACKREP_H
#define GFABSTRACKREP_H

#include<vector>
#include<list>
#include<iostream>

#include "TMatrixT.h"
#include "TVector3.h"
#include "TMath.h"

#include "GFDetPlane.h"

class GFAbsRecoHit;    

class GFAbsTrackRep : public TObject{

  /*----- Data members -----*/
 protected:
  unsigned int fDimension;

  TMatrixT<double> fState;
  
  TMatrixT<double> fCov;   

  double           fChiSqu;
  double           fForwardChiSqu;
  unsigned int     fNdf;

  int fStatusFlag; 
  bool fInverted;   

  TMatrixT<double> fFirstState; 
  TMatrixT<double> fFirstCov;

  TMatrixT<double> fLastState; 
  TMatrixT<double> fLastCov;
  GFDetPlane fFirstPlane;  
  GFDetPlane fLastPlane;  

  // detector plane where the track parameters are given
  GFDetPlane         fRefPlane;

  double fXX0;


 public:
  virtual GFAbsTrackRep* clone() const = 0;

  virtual GFAbsTrackRep* prototype() const = 0;
  

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


 public:

  GFAbsTrackRep();
  GFAbsTrackRep(int);
  virtual ~GFAbsTrackRep();



  virtual void extrapolateToPoint(const TVector3& point,
         TVector3& poca,
         TVector3& normVec);


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

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


  virtual double extrapolate(const GFDetPlane& plane,
           TMatrixT<double>& statePred,
           TMatrixT<double>& covPred)=0;
  

  double extrapolate(const GFDetPlane& plane);

  unsigned int getDim() const {return fDimension;}  
  
  virtual void Print(const Option_t* = "") const;

  inline TMatrixT<double> getState() const {
    return fState;
  }
  inline TMatrixT<double> getCov() const {
    return fCov;
  }

  double getStateElem(int i) const {return fState(i,0);}
  double getCovElem(int i, int j) const {return fCov(i,j);}

  

  virtual TVector3 getPos(const GFDetPlane& pl)=0; 
  virtual TVector3 getMom(const GFDetPlane& pl)=0; 
  virtual void getPosMom(const GFDetPlane& pl,TVector3& pos,TVector3& mom)=0;


  virtual void getPosMomCov(const GFDetPlane& pl,TVector3& pos,TVector3& mom,TMatrixT<double>& cov);

  virtual double getCharge()const =0;


  virtual int getPDG();

  TVector3 getPos() {return getPos(fRefPlane);}
  TVector3 getMom() {return getMom(fRefPlane);}
  void getPosMomCov(TVector3& pos,TVector3& mom,TMatrixT<double>& c){
    getPosMomCov(fRefPlane,pos,mom,c);
  }

  inline TMatrixT<double> getFirstState() const {
    return fFirstState;
  }
  inline TMatrixT<double> getFirstCov() const {
    return fFirstCov;
  }
  inline GFDetPlane getFirstPlane() const {
    return fFirstPlane;
  }
  inline TMatrixT<double> getLastState() const {
    return fLastState;
  }
  inline TMatrixT<double> getLastCov() const {
    return fLastCov;
  }
  inline GFDetPlane getLastPlane() const {
    return fLastPlane;
  }
  inline double getChiSqu() const {
    return fChiSqu;
  }
  inline double getForwardChiSqu() const {
    return fForwardChiSqu;
  }
  inline double getRedChiSqu() const {
    if(getNDF()>0) return getChiSqu()/getNDF();
    return 0;
  }
  // returns p-value
  inline double getPVal() const {
    return TMath::Prob(getChiSqu(), getNDF());
  }
  inline unsigned int getNDF() const {
    if(fNdf>getDim())  return fNdf-getDim();
    return 0;
  }
  inline double getXX0() const {
    return fXX0;
  }
  virtual void setData(const TMatrixT<double>& st, const GFDetPlane& pl, const TMatrixT<double>* cov=NULL, const TMatrixT<double>* aux=NULL){
    fState=st;
    fRefPlane=pl;
    if(cov!=NULL) fCov=*cov;
    static_cast<void>(aux);
  }
  inline void setCov(const TMatrixT<double>& aCov) {
    fCov=aCov;
  }
  inline void setFirstState(const TMatrixT<double>& aState) {
    fFirstState = aState;
  }
  inline void setFirstCov(const TMatrixT<double>& aCov) {
    fFirstCov = aCov;
  }
  inline void setFirstPlane(const GFDetPlane& aPlane) {
    fFirstPlane = aPlane;;
  }
  inline void setLastState(const TMatrixT<double>& aState) {
    fLastState = aState;
  }
  inline void setLastCov(const TMatrixT<double>& aCov) {
    fLastCov = aCov;
  }
  inline void setLastPlane(const GFDetPlane& aPlane) {
    fLastPlane = aPlane;;
  }


  virtual void setPosMomCov(const TVector3& pos, const TVector3& mom, const TMatrixT<double>& cov);

  const GFDetPlane& getReferencePlane() const {return fRefPlane;}

  inline void setChiSqu(double aChiSqu) {
    fChiSqu = aChiSqu;
  }
  inline void setForwardChiSqu(double aChiSqu) {
    fForwardChiSqu = aChiSqu;
  }
  inline void setNDF(unsigned int n) {
    fNdf = n;
  }
  inline void addChiSqu(double aChiSqu) {
    fChiSqu += aChiSqu;
  }
  inline void addForwardChiSqu(double aChiSqu) {
    fForwardChiSqu += aChiSqu;
  }
  inline void addNDF(unsigned int n) {
    fNdf += n;
  }
  inline void setStatusFlag(int _val) {
    fStatusFlag = _val;
  }
  
  virtual void switchDirection() = 0;

  bool setInverted(bool f=true){fInverted=f; return true;}
  
  inline bool getStatusFlag() {
    return fStatusFlag;
  }

  virtual void reset();

  virtual bool hasAuxInfo() {
    return false;
  }

  virtual const TMatrixT<double>* getAuxInfo(const GFDetPlane&) {
    return NULL;
  }


  void resetXX0() {
    fXX0 = 0.;
  }

 private:
  void Abort(std::string method);


  ClassDef(GFAbsTrackRep,4)

};

#endif