CbmLitFitNode.h

Go to the documentation of this file.

/* Copyright (C) 2008-2012 GSI/JINR-LIT, Darmstadt/Dubna
   SPDX-License-Identifier: GPL-3.0-only
   Authors: Andrey Lebedev [committer] */

 
#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_*/