Propagator.h

Go to the documentation of this file.

#ifndef PROPAGATOR_H
#define PROPAGATOR_H

#include <memory>
#include "lardata/RecoObjects/KalmanLinearAlgebra.h"
#include "lardata/RecoObjects/KETrack.h"
#include "lardata/RecoObjects/Interactor.h"
#include "boost/optional.hpp"

namespace trkf {

  class Propagator
  {
  public:

    enum PropDirection {FORWARD, BACKWARD, UNKNOWN};

    Propagator(double tcut, bool doDedx, const std::shared_ptr<const Interactor>& interactor);

    virtual ~Propagator();

    // Accessors.

    double getTcut() const {return fTcut;}
    bool getDoDedx() const {return fDoDedx;}
    const std::shared_ptr<const Interactor>& getInteractor() const {return fInteractor;}

    // Virtual methods.

    virtual Propagator* clone() const = 0;

    virtual boost::optional<double> short_vec_prop(KTrack& trk,
                                                   const std::shared_ptr<const Surface>& psurf, 
                                                   PropDirection dir,
                                                   bool doDedx,
                                                   TrackMatrix* prop_matrix = 0,
                                                   TrackError* noise_matrix = 0) const = 0;

    virtual boost::optional<double> origin_vec_prop(KTrack& trk,
                                                    const std::shared_ptr<const Surface>& porient,
                                                    TrackMatrix* prop_matrix = 0) const = 0;

    boost::optional<double> vec_prop(KTrack& trk,
                                     const std::shared_ptr<const Surface>& psurf, 
                                     PropDirection dir,
                                     bool doDedx,
                                     TrackMatrix* prop_matrix = 0,
                                     TrackError* noise_matrix = 0) const;

    boost::optional<double> lin_prop(KTrack& trk,
                                     const std::shared_ptr<const Surface>& psurf, 
                                     PropDirection dir,
                                     bool doDedx,
                                     KTrack* ref = 0,
                                     TrackMatrix* prop_matrix = 0,
                                     TrackError* noise_matrix = 0) const;
    boost::optional<double> err_prop(KETrack& tre,
                                     const std::shared_ptr<const Surface>& psurf, 
                                     PropDirection dir,
                                     bool doDedx,
                                     KTrack* ref = 0,
                                     TrackMatrix* prop_matrix = 0) const;

    boost::optional<double> noise_prop(KETrack& tre,
                                       const std::shared_ptr<const Surface>& psurf, 
                                       PropDirection dir,
                                       bool doDedx,
                                       KTrack* ref = 0) const;

    boost::optional<double> dedx_prop(double pinv, double mass,
                                      double s, double* deriv=0) const;

  private:

    // Attributes.

    double fTcut;                                   
    bool fDoDedx;                                   
    std::shared_ptr<const Interactor> fInteractor;  
  };
}

#endif