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

namespace genf {
  class GFTrackCand;
}

//#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