#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