LSLTrackRep.h

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

/* 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 LSLTRACKREP_H
#define LSLTRACKREP_H

#include <ostream> // remove if you do not need streaming op

#include "TVectorT.h"
#include "TMatrixT.h"

#include "GFAbsTrackRep.h"

// Collaborating Class Declarations --
class GFAbsBField;
class AbsNystromEQM;

class LSLTrackRep : public GFAbsTrackRep {
public:

  // Constructors/Destructors ---------
  LSLTrackRep();
  LSLTrackRep(double z, double x, double y, 
        double dxdz, double dydz, double invp,
        double sigx, double sigy, 
        double sigdxdz, double sigdydz, 
        double siginvp, 
        GFAbsBField* field);
  LSLTrackRep(const LSLTrackRep& rep); // copy ctor

  virtual ~LSLTrackRep();


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

  virtual GFAbsTrackRep* prototype()const{return new LSLTrackRep();}

  // Operators
  friend std::ostream& operator<< (std::ostream& s, const LSLTrackRep& me);

  // Accessors -----------------------
  void init(const TVector3& pos,
      double dxdz, double dydz, double invp,
      double sigx, double sigy,
      double sigdxdz, double sigdydz, 
      double siginvp,
      GFAbsBField* field);

  // Modifiers
  void SetBField(GFAbsBField* b);
  void setData(const TMatrixT<double>& st, const GFDetPlane& pl, const TMatrixT<double>* cov=NULL){
    GFAbsTrackRep::setData(st,pl,cov);
    s=pl.getO().Z();
  }
  void setRungeKuttaAcc(double acc){_acc=acc;}
  void setRungeKuttaAdaptive(bool f){_adaptive=f;}
  // Operations ----------------------
  virtual double extrapolate(const GFDetPlane&, TMatrixT<double>& statePred);
  //virtual void extrapolate(const GFDetPlane&, 
  //         const TMatrixT<double>& stateFrom, 
  //         TMatrixT<double>& stateResult);

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


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


  virtual void stepalong(double h);

  virtual TVector3 getPos(const GFDetPlane&) ;
  virtual TVector3 getMom(const GFDetPlane&) ;
  virtual void getPosMom(const GFDetPlane&,TVector3& pos, TVector3& mom) ;
  virtual TVectorT<double> getGlobal(); // (x,y,z,px,py,pz)
  virtual TMatrixT<double> getGlobalCov(); // covariances

  virtual double getCharge()const {double fact= fInverted ? -1.:1.;
    return fState[4][0]<0 ? -fact : fact;}

  virtual void switchDirection(){};
  
private:

  // Private Data Members ------------

  // Private Methods -----------------
  // equation of motion in a magnetic field
  //static TVectorT<double> eqm(const TVectorT<double>& u,
  //            const TVectorT<double>& uprim,
  //            const TVectorT<double>& par);

  double s;

  double _acc; // accuracy of runge kutta; default: 1E-2cm
  bool _adaptive;

  AbsNystromEQM* _eqm; 

  // calculate jacobian of extrapolation
  void Jacobian(const GFDetPlane& pl,
    const TMatrixT<double>& statePred,
    TMatrixT<double>& jacResult);

 public:
  ClassDef(LSLTrackRep,1)

};


#endif