Vertex.hpp

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file Vertex.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/Definitions/Algebra.hpp"
#include "Acts/Vertexing/TrackAtVertex.hpp"

namespace Acts {

template <typename input_track_t>
class Vertex {
 public:
  Vertex() = default;

  Vertex(const Vector3& position);

  Vertex(const Vector4& position);

  Vertex(const Vector3& position, const SquareMatrix3& covariance,
         const std::vector<TrackAtVertex<input_track_t>>& tracks);

  Vertex(const Vector4& position, const SquareMatrix4& covariance,
         const std::vector<TrackAtVertex<input_track_t>>& tracks);

  Vector3 position() const;

  ActsScalar time() const;

  const Vector4& fullPosition() const;

  SquareMatrix3 covariance() const;

  const SquareMatrix4& fullCovariance() const;

  const std::vector<TrackAtVertex<input_track_t>>& tracks() const;

  std::pair<double, double> fitQuality() const;

  void setPosition(const Vector3& position, ActsScalar time = 0);

  void setFullPosition(const Vector4& fullPosition);

  void setTime(ActsScalar time);

  void setCovariance(const SquareMatrix3& covariance);

  void setFullCovariance(const SquareMatrix4& covariance);

  void setTracksAtVertex(
      const std::vector<TrackAtVertex<input_track_t>>& tracks);

  void setFitQuality(double chiSquared, double numberDoF);

  void setFitQuality(std::pair<double, double> fitQuality);

 private:
  Vector4 m_position = Vector4::Zero();
  SquareMatrix4 m_covariance = SquareMatrix4::Zero();
  std::vector<TrackAtVertex<input_track_t>> m_tracksAtVertex;
  double m_chiSquared = 0.;  // chi2 of the fit
  double m_numberDoF = 0.;   // number of degrees of freedom
};

}  // namespace Acts

#include "Vertex.ipp"