#include <TrkFilteringNodes.h>

Inheritance diagram for Trk::TrkClusterNode:

Collaboration diagram for Trk::TrkClusterNode:

Public Member Functions
	TrkClusterNode (TrkPlanarSurface , double, const PrepRawData )

	TrkClusterNode (TrkPlanarSurface *, double, double, double)

virtual	~TrkClusterNode ()

virtual void	validateMeasurement (TrkTrackState *)

virtual void	updateTrackState (TrkTrackState *)

virtual double	getChi2Distance (TrkTrackState *)

virtual void	serialize (char fileName[])

virtual void	report ()

virtual void	updateWithRIO (const RIO_OnTrack *)

virtual int	getKalmanGain (double[5][2])

virtual int	getResiduals (double[2])

virtual int	getInverseResidualVariance (double[2][2])

virtual int	getMeasurementMatrix (double[2][5])

virtual void	runKalmanFilter (TrkTrackState *)

virtual double	calculateChi2 ()

virtual bool	isValidated ()

virtual void	setNodeState (int)

virtual int	getNodeState ()

virtual void	setNodeType (char)

virtual char	getNodeType ()

virtual void	updateInternal ()

TrkTrackState *	getTrackState ()

virtual TrkPlanarSurface *	getSurface ()

virtual const PrepRawData *	getPrepRawData ()

double	getChi2 () const

int	getNdof () const

Protected Attributes
double	m_B [5]

double	m_D

double	m_H [5]

double	m_K [5]

double	m_resid

double	m_m

double	m_V

int	m_nodeState

double	m_chi2Cut

char	m_nodeType

TrkPlanarSurface *	m_pSurface

const PrepRawData *	m_pPRD

double	m_dChi2

int	m_ndof

TrkTrackState *	m_pTrackState

Detailed Description

Definition at line 67 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkFilteringNodes.h.

Constructor & Destructor Documentation

◆ TrkClusterNode() [1/2]

TrkClusterNode::TrkClusterNode	(	TrkPlanarSurface *	pS,
		double	chi2Cut,
		const PrepRawData *	pPRD
	)

Definition at line 127 of file TrkFilteringNodes.cxx.

                                                                                               {
     m_pSurface = pS;
     m_chi2Cut = chi2Cut;
     m_m = pPRD->localPosition()[0];
     m_V = pPRD->localCovariance()(0, 0);
     m_pPRD = pPRD;
     m_nodeType = 2;
     m_ndof = 1;
   }

◆ TrkClusterNode() [2/2]

TrkClusterNode::TrkClusterNode	(	TrkPlanarSurface *	pS,
		double	chi2Cut,
		double	pos,
		double	cov
	)

Definition at line 137 of file TrkFilteringNodes.cxx.

                                                                                              {
     m_pSurface = pS;
     m_chi2Cut = chi2Cut;
     m_m = pos;
     m_V = cov;
     m_pPRD = nullptr;
     m_nodeType = 2;
     m_ndof = 1;
   }

◆ ~TrkClusterNode()

virtual Trk::TrkClusterNode::~TrkClusterNode ( )

inlinevirtual

Definition at line 72 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkFilteringNodes.h.

72 {};

Member Function Documentation

◆ calculateChi2()

double TrkFilteringNode1D::calculateChi2 ( )

virtualinherited

Definition at line 33 of file TrkFilteringNodes.cxx.

                                            {
     return(m_resid * m_resid * m_D);
   }

◆ getChi2()

double TrkBaseNode::getChi2 ( ) const

inherited

Definition at line 69 of file TrkBaseNode.cxx.

                                     {
     return m_dChi2;
   }

◆ getChi2Distance()

double TrkClusterNode::getChi2Distance ( TrkTrackState * pTS )

virtual

Implements Trk::TrkBaseNode.

Definition at line 161 of file TrkFilteringNodes.cxx.

                                                            {
     for (int i = 0; i < 5; i++) m_B[i] = pTS->getTrackCovariance(0, i);
     m_H[0] = 1.0;
     m_H[1] = 0.0;
     m_H[2] = 0.0;
     m_H[3] = 0.0;
     m_H[4] = 0.0;
     m_D = 1.0 / (pTS->getTrackCovariance(0, 0) + m_V);
     m_resid = m_m - pTS->getTrackState(0);
     return calculateChi2();
   }

◆ getInverseResidualVariance()

int TrkFilteringNode1D::getInverseResidualVariance ( double V[2][2] )

virtualinherited

Implements Trk::TrkBaseNode.

Definition at line 63 of file TrkFilteringNodes.cxx.

                                                                    {
     V[0][0] = m_D;
     return 1;
   }

◆ getKalmanGain()

int TrkFilteringNode1D::getKalmanGain ( double K[5][2] )

virtualinherited

Implements Trk::TrkBaseNode.

Definition at line 51 of file TrkFilteringNodes.cxx.

                                                       {
     int i;
  
     for (i = 0; i < 5; i++) K[i][0] = m_K[i];
     return 1;
   }

◆ getMeasurementMatrix()

int TrkFilteringNode1D::getMeasurementMatrix ( double H[2][5] )

virtualinherited

Implements Trk::TrkBaseNode.

Definition at line 68 of file TrkFilteringNodes.cxx.

                                                              {
     int i;
  
     for (i = 0; i < 5; i++) H[0][i] = m_H[i];
     return 1;
   }

◆ getNdof()

int TrkBaseNode::getNdof ( ) const

inherited

Definition at line 65 of file TrkBaseNode.cxx.

                                  {
     return m_ndof;
   }

◆ getNodeState()

int TrkBaseNode::getNodeState ( )

virtualinherited

Definition at line 53 of file TrkBaseNode.cxx.

                                 {
     return m_nodeState;
   }

◆ getNodeType()

char TrkBaseNode::getNodeType ( )

virtualinherited

Definition at line 61 of file TrkBaseNode.cxx.

                                 {
     return m_nodeType;
   }

◆ getPrepRawData()

const PrepRawData * TrkBaseNode::getPrepRawData ( )

virtualinherited

Definition at line 39 of file TrkBaseNode.cxx.

                                                  {
     return m_pPRD;
   }

◆ getResiduals()

int TrkFilteringNode1D::getResiduals ( double r[2] )

virtualinherited

Implements Trk::TrkBaseNode.

Definition at line 58 of file TrkFilteringNodes.cxx.

                                                   {
     r[0] = m_resid;
     return 1;
   }

◆ getSurface()

TrkPlanarSurface * TrkBaseNode::getSurface ( )

virtualinherited

Definition at line 35 of file TrkBaseNode.cxx.

                                             {
     return m_pSurface;
   }

◆ getTrackState()

TrkTrackState * TrkBaseNode::getTrackState ( )

inherited

Definition at line 43 of file TrkBaseNode.cxx.

                                             {
     return m_pTrackState;
   }

◆ isValidated()

bool TrkBaseNode::isValidated ( )

virtualinherited

Definition at line 31 of file TrkBaseNode.cxx.

                               {
   return (m_nodeState != 0);
   }

◆ report()

virtual void Trk::TrkClusterNode::report ( )

inlinevirtual

Implements Trk::TrkBaseNode.

Definition at line 77 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkFilteringNodes.h.

77 {};

◆ runKalmanFilter()

void TrkFilteringNode1D::runKalmanFilter ( TrkTrackState * pTS )

virtualinherited

Definition at line 37 of file TrkFilteringNodes.cxx.

                                                              {
     int i, j, idx = 0;
     double updState[5], updCov[15];
  
     for (i = 0; i < 5; i++) m_K[i] = m_D * m_B[i];
     for (i = 0; i < 5; i++) updState[i] = m_K[i] * m_resid;
     pTS->updateTrackState(updState);
     for (i = 0; i < 5; i++) for (j = i; j < 5; j++) {
         updCov[idx] = -m_K[i] * m_B[j];
         idx++;
       }
     pTS->updateTrackCovariance(updCov);
   }

◆ serialize()

void TrkClusterNode::serialize ( char fileName[] )

virtual

Implements Trk::TrkBaseNode.

Definition at line 147 of file TrkFilteringNodes.cxx.

                                                 {
     FILE* pFile = fopen(fileName, "a");
     if (!pFile) {
       std::cerr << "Cannot open file " << fileName << " for write.\n";
       std::abort();
     }
     fclose(pFile);
   }

◆ setNodeState()

void TrkBaseNode::setNodeState ( int s )

virtualinherited

Definition at line 49 of file TrkBaseNode.cxx.

                                       {
     m_nodeState = s;
   }

◆ setNodeType()

void TrkBaseNode::setNodeType ( char s )

virtualinherited

Definition at line 57 of file TrkBaseNode.cxx.

                                       {
     m_nodeType = s;
   }

◆ updateInternal()

void TrkBaseNode::updateInternal ( )

virtualinherited

Reimplemented in Trk::TrkTrtNode.

Definition at line 73 of file TrkBaseNode.cxx.

73 {

74 }

◆ updateTrackState()

void TrkClusterNode::updateTrackState ( TrkTrackState * pTS )

virtual

Implements Trk::TrkBaseNode.

Definition at line 173 of file TrkFilteringNodes.cxx.

                                                           {
     if (isValidated()) runKalmanFilter(pTS);
     m_pTrackState = pTS;
   }

◆ updateWithRIO()

void TrkClusterNode::updateWithRIO ( const RIO_OnTrack * pRIO )

virtual

Implements Trk::TrkBaseNode.

Definition at line 178 of file TrkFilteringNodes.cxx.

                                                             {
     /*
        printf("RIO update, SCT cluster: old m=%f new m=%f old V=%f new V=%f\n",
        m_m,pRIO->localParameters()[Trk::locX],sqrt(m_V),
        sqrt(pRIO->localErrorMatrix().localCovariance()[0][0]));
      */
     m_m = pRIO->localParameters()[Trk::locX];
     m_V = pRIO->localCovariance()(0, 0);
   }

◆ validateMeasurement()

void TrkClusterNode::validateMeasurement ( TrkTrackState * pTS )

virtual

Implements Trk::TrkBaseNode.

Definition at line 156 of file TrkFilteringNodes.cxx.

                                                              {
     m_dChi2 = getChi2Distance(pTS);
     m_nodeState = (m_dChi2 < m_chi2Cut) ? 1 : 0;
   }

Member Data Documentation

◆ m_B

double Trk::TrkFilteringNode1D::m_B[5]

protectedinherited

Definition at line 43 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkFilteringNodes.h.

◆ m_chi2Cut

double Trk::TrkBaseNode::m_chi2Cut

protectedinherited

Definition at line 52 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkBaseNode.h.

◆ m_D

double Trk::TrkFilteringNode1D::m_D

protectedinherited

Definition at line 43 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkFilteringNodes.h.

◆ m_dChi2

double Trk::TrkBaseNode::m_dChi2

protectedinherited

Definition at line 56 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkBaseNode.h.

◆ m_H

double Trk::TrkFilteringNode1D::m_H[5]

protectedinherited

Definition at line 43 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkFilteringNodes.h.

◆ m_K

double Trk::TrkFilteringNode1D::m_K[5]

protectedinherited

Definition at line 43 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkFilteringNodes.h.

◆ m_m

double Trk::TrkFilteringNode1D::m_m

protectedinherited

Definition at line 45 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkFilteringNodes.h.

◆ m_ndof

int Trk::TrkBaseNode::m_ndof

protectedinherited

Definition at line 57 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkBaseNode.h.

◆ m_nodeState

int Trk::TrkBaseNode::m_nodeState

protectedinherited

Definition at line 51 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkBaseNode.h.

◆ m_nodeType

char Trk::TrkBaseNode::m_nodeType

protectedinherited

Definition at line 53 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkBaseNode.h.

◆ m_pPRD

const PrepRawData* Trk::TrkBaseNode::m_pPRD

protectedinherited

Definition at line 55 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkBaseNode.h.

◆ m_pSurface

TrkPlanarSurface* Trk::TrkBaseNode::m_pSurface

protectedinherited

Definition at line 54 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkBaseNode.h.

◆ m_pTrackState

TrkTrackState* Trk::TrkBaseNode::m_pTrackState

protectedinherited

Definition at line 58 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkBaseNode.h.

◆ m_resid

double Trk::TrkFilteringNode1D::m_resid

protectedinherited

Definition at line 44 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkFilteringNodes.h.

◆ m_V

double Trk::TrkFilteringNode1D::m_V

protectedinherited

Definition at line 46 of file Tracking/TrkFitter/TrkDistributedKalmanFilter/TrkDistributedKalmanFilter/TrkFilteringNodes.h.

The documentation for this class was generated from the following files:

Public Member Functions

Protected Attributes

Detailed Description

Constructor & Destructor Documentation

◆ TrkClusterNode() [1/2]

◆ TrkClusterNode() [2/2]

◆ ~TrkClusterNode()

Member Function Documentation

◆ calculateChi2()

◆ getChi2()

◆ getChi2Distance()

◆ getInverseResidualVariance()

◆ getKalmanGain()

◆ getMeasurementMatrix()

◆ getNdof()

◆ getNodeState()

◆ getNodeType()

◆ getPrepRawData()

◆ getResiduals()

◆ getSurface()

◆ getTrackState()

◆ isValidated()

◆ report()

◆ runKalmanFilter()

◆ serialize()

◆ setNodeState()

◆ setNodeType()

◆ updateInternal()

◆ updateTrackState()

◆ updateWithRIO()

◆ validateMeasurement()

Member Data Documentation

◆ m_B

◆ m_chi2Cut

◆ m_D

◆ m_dChi2

◆ m_H

◆ m_K

◆ m_m

◆ m_ndof

◆ m_nodeState

◆ m_nodeType

◆ m_pPRD

◆ m_pSurface

◆ m_pTrackState

◆ m_resid

◆ m_V