ActsExamples/Utilities/VertexTruthUtility.hpp

0001 // This file is part of the ACTS project.
0002 //
0003 // Copyright (C) 2016 CERN for the benefit of the ACTS project
0004 //
0005 // This Source Code Form is subject to the terms of the Mozilla Public
0006 // License, v. 2.0. If a copy of the MPL was not distributed with this
0007 // file, You can obtain one at https://mozilla.org/MPL/2.0/.
0008
0009 #pragma once
0010
0011 #include "Acts/Utilities/Logger.hpp"
0012 #include "Acts/Vertexing/Vertex.hpp"
0013 #include "ActsExamples/EventData/SimParticle.hpp"
0014 #include "ActsExamples/EventData/SimVertex.hpp"
0015 #include "ActsExamples/EventData/Track.hpp"
0016 #include "ActsExamples/EventData/TruthMatching.hpp"
0017
0018 #include <cstdint>
0019 #include <optional>
0020
0021 namespace ActsExamples {
0022
0023 std::uint32_t getNumberOfReconstructableVertices(
0024     const SimParticleContainer& collection);
0025
0026 std::uint32_t getNumberOfTruePriVertices(
0027     const SimParticleContainer& collection);
0028
0029 double calcSumPt2(const Acts::Vertex& vtx, double minTrkWeight);
0030
0031 double calculateTruthPrimaryVertexDensity(
0032     const SimVertexContainer& truthVertices, const Acts::Vertex& vtx,
0033     double vertexDensityWindow);
0034
0035 const SimParticle* findParticle(
0036     const SimParticleContainer& particles,
0037     const TrackParticleMatching& trackParticleMatching, ConstTrackProxy track,
0038     const Acts::Logger& logger);
0039
0040 std::optional<ConstTrackProxy> findTrack(const ConstTrackContainer& tracks,
0041                                          const Acts::TrackAtVertex& trkAtVtx);
0042
0043 }  // namespace ActsExamples