CbmLitFitNode.h

Go to the documentation of this file.

 
#ifndef CBMLITFITNODE_H_
#define CBMLITFITNODE_H_
 
#include "data/CbmLitTrackParam.h"
#include <vector>
using std::vector;
 
class CbmLitFitNode {
public:
  CbmLitFitNode()
    : fF(25, 0.)
    , fPredictedParam()
    , fUpdatedParam()
    , fSmoothedParam()
    , fChiSqFiltered(0.)
    , fChiSqSmoothed(0.) {}
 
  virtual ~CbmLitFitNode() {};
 
  /* Getters */
  const vector<litfloat>& GetF() const { return fF; }
  const CbmLitTrackParam* GetPredictedParam() const { return &fPredictedParam; }
  const CbmLitTrackParam* GetUpdatedParam() const { return &fUpdatedParam; }
  const CbmLitTrackParam* GetSmoothedParam() const { return &fSmoothedParam; }
  litfloat GetChiSqFiltered() const { return fChiSqFiltered; }
  litfloat GetChiSqSmoothed() const { return fChiSqSmoothed; }
 
  /* Setters */
  void SetF(const vector<litfloat>& F) { fF.assign(F.begin(), F.end()); }
  void SetPredictedParam(const CbmLitTrackParam* par) {
    fPredictedParam = *par;
  }
  void SetUpdatedParam(const CbmLitTrackParam* par) { fUpdatedParam = *par; }
  void SetSmoothedParam(const CbmLitTrackParam* par) { fSmoothedParam = *par; }
  void SetChiSqFiltered(litfloat chiSq) { fChiSqFiltered = chiSq; }
  void SetChiSqSmoothed(litfloat chiSq) { fChiSqSmoothed = chiSq; }
 
private:
  vector<litfloat> fF;  // Transport matrix.
 
  CbmLitTrackParam fPredictedParam;  // Predicted track parameters.
  CbmLitTrackParam fUpdatedParam;    // Updated with KF track parameters.
  CbmLitTrackParam fSmoothedParam;   // Smoothed track parameters.
 
  litfloat
    fChiSqFiltered;  // Contribution to chi-square of updated track parameters and hit.
  litfloat
    fChiSqSmoothed;  // Contribution to chi-square of smoothed track parameters and hit.
};
 
#endif /*CBMLITFITNODE_H_*/