KalmanFitter.h

Go to the documentation of this file.

#ifndef _KALMANFITTER_H
#define _KALMANFITTER_H
 
/*
KalmanFitter.h
 
Definition of the class KalmanFitter, refine the measurement of KalmanTrack
 
Author: Kun Liu, liuk@fnal.gov
Created: 10-18-2012
*/
 
#include <list>
#include <vector>
 
#include <GlobalConsts.h>
 
#include "KalmanUtil.h"
#include "KalmanTrack.h"
#include "KalmanFilter.h"
 
class PHField;
 
class KalmanFitter
{
public:
    KalmanFitter(const PHField* field, const TGeoManager *geom);
    //~KalmanFitter();
 
    void setControlParameter(int nMaxIteration, double tolerance) { _max_iteration = nMaxIteration; _tolerance = tolerance; }
 
    int processOneTrack(KalmanTrack& _track);
    void updateTrack(KalmanTrack& _track);
 
    void init();
 
    int initNodeList(KalmanTrack& _track);
 
    void updateAlignment();
 
    void setStartingParameter(KalmanTrack& _track);
    void setStartingParameter(Node& _node);
 
    bool initSmoother(Node& _node);
 
    std::list<Node>& getNodeList() { return _nodes; }
 
    double findVertex();
 
    double getMomentumInitial(double& px, double& py, double& pz) { return _nodes.front().getSmoothed().get_mom(px, py, pz); }
    double getMomentumFinal(double& px, double& py, double& pz) { return _nodes.back().getSmoothed().get_mom(px, py, pz); }
 
    double getPositionInitial(double& x, double& y, double& z) { return _nodes.front().getSmoothed().get_pos(x, y, z); }
    double getPositionFinal(double& x, double& y, double& z) { return _nodes.back().getSmoothed().get_pos(x, y, z); }
 
    double getChisq() { return _chisq; }
 
    const TrkPar& getTrkParInitial() { return _nodes.front().getSmoothed(); }
    const TrkPar& getTrkParFinal() { return _nodes.back().getSmoothed(); }
 
    //TrkPar getTrkPar(double z) { std::cout << "Will be implemented later" << std::endl; }
 
private:
    std::list<Node> _nodes;
 
    double _chisq;
 
    KalmanFilter *_kmfit;
 
    double rM_20[nChamberPlanes];
    double rM_21[nChamberPlanes];
    double rM_22[nChamberPlanes];
    double z_planes[nChamberPlanes];
 
    int _max_iteration;
    double _tolerance;
};
 
#endif