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 <stdexcept> // std::logic_error
#include "larreco/Genfit/GFAbsTrackRep.h"
#include "larreco/Genfit/GFDetPlane.h"
#include "larreco/Genfit/GFTrackCand.h"
#include <TMatrixT.h>

//#include "GFMaterialEffects.h"

namespace genf {

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_t>& statePred,
                           TMatrixT<Double_t>& covPred);

  double extrapolate(const GFDetPlane&, 
                           TMatrixT<Double_t>& 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);
  


  TVector3 getPos(const GFDetPlane&);

  TVector3 getMom(const GFDetPlane&);
  TVector3 getMomLast(const GFDetPlane&);

  void getPosMom(const GFDetPlane&,TVector3& pos,TVector3& mom);
  double getCharge()const {return fCharge;}
  void switchDirection(){fDirection = (!fDirection);}
  void setPDG(int);
  int getPDG();
  void rescaleCovOffDiags();


  void setData(const TMatrixT<double>& st, const GFDetPlane& pl, const TMatrixT<double>* cov=NULL, const TMatrixT<double>* aux=NULL);
  // the base class method is hidden by the one above; which does not make much
  // sense since polymorphic access will still get to it.
  // We make it official by explicitly importing the base class method as well.
  using GFAbsTrackRep::setData;

  const TMatrixT<double>* getAuxInfo(const GFDetPlane& pl);
  
  bool hasAuxInfo() { return true; }

 private:

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


  RKTrackRep& operator=(const RKTrackRep* /* rhs */){return *this;}
  RKTrackRep(const RKTrackRep& /* rhs */): GFAbsTrackRep() {}
  bool fDirection;
    
  int fPdg;
  double fMass;
  double fCharge;
  //GFMaterialEffects *fEffect;


  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_t>* state, TMatrixT<Double_t>* cov=NULL) const;

  
  //  void setData(const TMatrixT<Double_t>& /* st */, const GFDetPlane& /* pl */, const TMatrixT<Double_t>* cov=NULL, const TMatrixT<double>* aux=NULL);
    //    { throw std::logic_error(std::string(__func__) + "::setData(TMatrixT, GFDetPlane, TMatrixT) not available"); }
  
  // public:
  //ClassDef(RKTrackRep,3)

  };

} // end namespace
#endif