KalmanVertexUpdater.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file KalmanVertexUpdater.hpp

// This file is part of the Acts project.
//
// Copyright (C) 2019 CERN for the benefit of the Acts project
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.

#pragma once

#include "Acts/Utilities/Result.hpp"
#include "Acts/Vertexing/TrackAtVertex.hpp"
#include "Acts/Vertexing/Vertex.hpp"

namespace Acts {
namespace KalmanVertexUpdater {


struct MatrixCache {
  Vector3 newVertexPos = Vector3::Zero();
  SquareMatrix3 newVertexCov = SquareMatrix3::Zero();
  SquareMatrix3 newVertexWeight = SquareMatrix3::Zero();
  SquareMatrix3 oldVertexWeight = SquareMatrix3::Zero();
  SquareMatrix3 momWeightInv = SquareMatrix3::Zero();
};

template <typename input_track_t>
void updateVertexWithTrack(Vertex<input_track_t>& vtx,
                           TrackAtVertex<input_track_t>& trk);

template <typename input_track_t>
void updatePosition(const Acts::Vertex<input_track_t>& vtx,
                    const Acts::LinearizedTrack& linTrack, double trackWeight,
                    int sign, MatrixCache& matrixCache);

namespace detail {

template <typename input_track_t>
double vertexPositionChi2(const Vertex<input_track_t>& oldVtx,
                          const MatrixCache& matrixCache);

template <typename input_track_t>
double trackParametersChi2(const LinearizedTrack& linTrack,
                           const MatrixCache& matrixCache);

template <typename input_track_t>
void update(Vertex<input_track_t>& vtx, TrackAtVertex<input_track_t>& trk,
            int sign);
}  // Namespace detail

}  // Namespace KalmanVertexUpdater
}  // Namespace Acts

#include "KalmanVertexUpdater.ipp"