TrackFitterBase.h

#ifndef TRACK_FITTER_BASE_H
#define TRACK_FITTER_BASE_H

#include <string>
#include <marlin/StringParameters.h>
#include <EVENT/LCRunHeader.h>
#include <EVENT/Track.h>
#include <IMPL/TrackImpl.h>
#include <EVENT/TrackerHit.h>

namespace gear
{
class TPCParameters;
class PadRowLayout2D;
}//namespace gear

namespace marlintpc
{
class TrackFitterBase
{
    public:

        // These variables are static, don't forget to instantiate them in TrackFitterBase.cc
        struct FitterTypes
        {
            //static const int BASE = 0;
            static const unsigned char LIKELIHOOD = 1; 
            static const unsigned char SIMPLECHISQUARE = 2; 
            static const unsigned char SIMPLECHISQUAREPADS = 3; 
            static const unsigned char LINEARREGRESSION = 4; 
            static const unsigned char KALMAN = 5; 
            static const unsigned char SIMPLEMINIMIZER = 6; 
            static const unsigned char CHISQUARE = 7; 
            static const unsigned char SIMPLEHELIXFIT = 8; 
        };

        static const unsigned char FITFAILEDBIT  =  30;  
        //  static const int FITFAILED = 0x1 << FITFAILEDBIT;///< The mask to test if the fit failed

        TrackFitterBase();

        virtual ~TrackFitterBase() {}

        virtual std::string getRevision() const;

        virtual unsigned char getFitterType() const = 0;

        virtual IMPL::TrackImpl *fitTrack(EVENT::Track const * seedTrack) const = 0;

        virtual EVENT::DoubleVec calculateResiduals(EVENT::Track const * testTrack,
                                                    unsigned int testHitNumber,
                                                    EVENT::Track const * referenceTrack = NULL) const;

        virtual EVENT::DoubleVec calculateResiduals(EVENT::TrackerHit const * testHit,
                                                    EVENT::Track const *trackWithoutTestHit) const;


        static void setTrackFitterType(unsigned char type, IMPL::TrackImpl *track);

        static unsigned char getFitterTypeFromTrack(int trackTypeWord);

        void setD0Start(double d0);

        void setPhiStart(double phi);

        void setOmegaStart(double omega);

        void setTanLambdaStart(double tanLambda);

        void setZ0Start(double z0);

        void setNoZFitFlag(bool noZFit = true);

        bool getNoZFitFlag() const;

        void fixD0(bool fix = true);

        void fixPhi(bool fix = true);

        void fixOmega(bool fix = true);

        void fixTanLambda(bool fix = true);

        void fixZ0(bool fix = true);

    protected:

        double distanceOnCircle(double phi1, double phi2) const;

        double _d0Start;
        bool _useD0Start;

        double _phiStart;
        bool _usePhiStart;

        double _omegaStart;
        bool _useOmegaStart;

        double _tanLambdaStart;
        bool _useTanLambdaStart;

        double _z0Start;
        bool _useZ0Start;

        bool _noZFit;

        bool _fixD0, _fixPhi, _fixOmega, _fixTanLambda, _fixZ0;
};

}// namespace marlintpc

#endif // TRACK_FITTER_BASE_H