SimpleHelixTrackModel.h

#ifndef SIMPLEHELIXTRACKMODEL_H
#define SIMPLEHELIXTRACKMODEL_H

#include <gearimpl/Vector3D.h>

#include "TVectorD.h"
#include "TMatrixD.h"
#include "TMatrixDSym.h"

#include "EVENT/Track.h"


namespace marlintpc {


class simpleHelix {
        public:
                simpleHelix(const double, const double, const double, const double, const double, const double*, const double);
                simpleHelix(const double*, const double*, const double);
                void dump() const;
                double getArcLengthXY(const double*) const;
                void getZSDirection(double*) const;
                void getPosAtArcLength(const double, double*) const;
                bool getExpectedPlanePos(const double*, const double, double&, double&, double&, double&, double&) const;
                TMatrixD analyticalHelixJacobian(const double, const double) const;
                TMatrixD simplifiedHelixJacobian(const double, const double) const;
                TMatrixD curvilinearToPerigeeJacobian() const;
                TMatrixD perigeeToILDJacobian() const;
                TMatrixD perigeeToLCIOJacobian() const;
                TMatrixD helixToLCIOJacobian() const;
                void moveTo(const double, const double, const double, double*);
                void getStateAt(const double, const double, const double, TMatrixDSym&, TVectorD&, TMatrixDSym&);

        protected:
                const double _xref;
                const double _yref;
                const double _zref;
                const double _bzc;
                const double _rinv;
                const double _phi0;
                const double _dca;
                const double _dzds;
                const double _z0;
                const double _xRelCenter;
                const double _yRelCenter;
                const double _eps; 
};


class localHelix {
        public:
                localHelix(const double*, const double*, const double);
                void dump() const;
                TMatrixD propagateTo(const double*, const double);
                simpleHelix getSimpleHelix() const;

        private:
                double _bfac;
                gear::Vector3D _bFieldc;
                TVectorD _offset;
                TMatrixD _rotation;
                TMatrixD _rotInv;
                gear::Vector3D _trackDir;
                double _cosLambda;
                double _qbyp;
                double _localPar[5];
                double _localRef[3];

                gear::Vector3D getBFieldc(const double*) const;
                void calcLocal();
                void calcGlobal();
};


class simpleFitXY {
        public:
                simpleFitXY(bool, double, double);
                void addPoint(double, double, double);
                void addTrack(EVENT::Track const *);
                int fit(double&, int&);
                TVectorD getPar() const;
                TMatrixDSym getCov() const;

        private:
                const bool _curved;
                const int _npar;
                const double _xRef;
                const double _yRef;
                int _numPoints;
                double _sx;
                double _sy;
                double _sxx;
                double _sxy;
                double _syy;
                double _sw;
                double _sr;
                double _sxr;
                double _syr;
                double _srr;
                TVectorD _parameters;
                TMatrixDSym _covariance;
};


class simpleFitZS {
        public:
                simpleFitZS();
                void addPoint(double, double, double);
                int fit(double&, int&);
                TVectorD getPar() const;
                TMatrixDSym getCov() const;

        private:
                const int _npar;
                int _numPoints;
                double _sx;
                double _sy;
                double _sxx;
                double _sxy;
                double _syy;
                double _sw;
                TVectorD _parameters;
                TMatrixDSym _covariance;
};

} // namespace marlintpc

#endif // SIMPLEHELIXTRACKMODEL_H