EIC code displayed by LXR master/​acts/​Tests/​UnitTests/​Core/​Vertexing/​AdaptiveGridTrackDensityTests.cpp	Source navigation Diff markup Identifier search General search
	Version: master test Revert   Change

File indexing completed on 2025-01-18 09:13:00

 // This file is part of the ACTS project.
 //
 // Copyright (C) 2016 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 https://mozilla.org/MPL/2.0/.
 
 #include <boost/test/unit_test.hpp>
 
 #include "Acts/Definitions/Algebra.hpp"
 #include "Acts/Definitions/TrackParametrization.hpp"
 #include "Acts/EventData/ParticleHypothesis.hpp"
 #include "Acts/EventData/TrackParameters.hpp"
 #include "Acts/Surfaces/PerigeeSurface.hpp"
 #include "Acts/Surfaces/Surface.hpp"
 #include "Acts/Tests/CommonHelpers/FloatComparisons.hpp"
 #include "Acts/Utilities/Result.hpp"
 #include "Acts/Vertexing/AdaptiveGridTrackDensity.hpp"
 
 #include <memory>
 #include <numbers>
 #include <optional>
 #include <utility>
 
 using namespace Acts::UnitLiterals;
 
 namespace Acts::Test {
 
 using Covariance = BoundSquareMatrix;
 
 Covariance makeRandomCovariance(int seed = 31415) {
   std::srand(seed);
   Covariance randMat((Covariance::Random() + 1.5 * Covariance::Identity()) *
                      0.05);
 
   // symmetric covariance matrix
   Covariance covMat = 0.5 * (randMat + randMat.transpose());
 
   return covMat;
 }
 
 BOOST_AUTO_TEST_CASE(compare_to_analytical_solution_for_single_track) {
   // Using a large track grid so we can choose a small bin size
   const std::uint32_t spatialTrkGridSize = 4001;
   // Arbitrary (but small) bin size
   const double binExtent = 3.1e-4;
   // Arbitrary impact parameters
   const double d0 = 0.4;
   const double z0 = -0.2;
   Vector2 impactParameters{d0, z0};
 
   Covariance covMat = makeRandomCovariance();
   SquareMatrix2 subCovMat = covMat.block<2, 2>(0, 0);
   BoundVector paramVec;
   paramVec << d0, z0, 0, 0, 0, 0;
 
   // Create perigee surface
   std::shared_ptr<PerigeeSurface> perigeeSurface =
       Surface::makeShared<PerigeeSurface>(Vector3(0., 0., 0.));
 
   BoundTrackParameters params1(perigeeSurface, paramVec, covMat,
                                ParticleHypothesis::pion());
 
   AdaptiveGridTrackDensity::Config cfg;
   // force track to have exactly spatialTrkGridSize spatial bins for testing
   // purposes
   cfg.spatialTrkGridSizeRange = {spatialTrkGridSize, spatialTrkGridSize};
   cfg.spatialBinExtent = binExtent;
   AdaptiveGridTrackDensity grid(cfg);
 
   // Empty map
   AdaptiveGridTrackDensity::DensityMap mainDensityMap;
 
   // Add track
   auto trackDensityMap = grid.addTrack(params1, mainDensityMap);
 
   double relTol = 1e-5;
   double small = 1e-5;
 
   auto gaussian2D = [&](const Vector2& args, const Vector2& mus,
                         const SquareMatrix2& sigmas) {
     Vector2 diffs = args - mus;
     double coef = 1 / std::sqrt(sigmas.determinant());
     double expo = -0.5 * diffs.transpose().dot(sigmas.inverse() * diffs);
     return coef * std::exp(expo);
   };
 
   for (const auto& [bin, _] : mainDensityMap) {
     // Extract variables for better readability
     std::int32_t zBin = bin.first;
     float density = mainDensityMap.at(bin);
     // Argument for 2D gaussian
     Vector2 dzVec{0., grid.getBinCenter(zBin, binExtent)};
     // Compute correct density...
     float correctDensity = gaussian2D(dzVec, impactParameters, subCovMat);
     // ... and check if our result is equivalent
     CHECK_CLOSE_OR_SMALL(correctDensity, density, relTol, small);
   }
 
   // Analytical maximum of the Gaussian (can be obtained by expressing the
   // exponent as (az - b)^2 + c and noting correctMaxZ = b/a)
   double correctMaxZ =
       -0.5 * (subCovMat(0, 1) + subCovMat(1, 0)) / subCovMat(0, 0) * d0 + z0;
   // Analytical FWHM of the Gaussian (result similar to
   // https://en.wikipedia.org/wiki/Full_width_at_half_maximum#Normal_distribution
   // but the calculation needs to be slightly modified in our case)
   double correctFWHM =
       2. * std::sqrt(2 * std::numbers::ln2 * subCovMat.determinant() /
                      subCovMat(0, 0));
 
   // Estimate maximum z position and seed width
   auto res = grid.getMaxZTPositionAndWidth(mainDensityMap);
   BOOST_CHECK(res.ok());
 
   // Extract variables for better readability...
   double maxZ = res.value().first.first;
   double fwhm = res.value().second * 2.355;
 
   // ... and check if they are correct (note: the optimization is not as exact
   // as the density values).
   double relTolOptimization = 1e-3;
   CHECK_CLOSE_REL(correctMaxZ, maxZ, relTolOptimization);
   CHECK_CLOSE_REL(correctFWHM, fwhm, relTolOptimization);
 }
 
 BOOST_AUTO_TEST_CASE(
     compare_to_analytical_solution_for_single_track_with_time) {
   // Number of bins in z- and t-direction
   const std::uint32_t spatialTrkGridSize = 401;
   const std::uint32_t temporalTrkGridSize = 401;
   // Bin extents
   const double spatialBinExtent = 3.1e-3;
   const double temporalBinExtent = 3.1e-3;
   // Arbitrary impact parameters
   const double d0 = -0.1;
   const double z0 = -0.2;
   const double t0 = 0.1;
   Vector3 impactParameters{d0, z0, t0};
 
   // symmetric covariance matrix
   Covariance covMat = makeRandomCovariance();
 
   BoundVector paramVec;
   paramVec << d0, z0, 0, 0, 0, t0;
   // Create perigee surface
   std::shared_ptr<PerigeeSurface> perigeeSurface =
       Surface::makeShared<PerigeeSurface>(Vector3(0., 0., 0.));
 
   BoundTrackParameters params(perigeeSurface, paramVec, covMat,
                               ParticleHypothesis::pion());
 
   ActsSquareMatrix<3> ipCov = params.impactParameterCovariance().value();
 
   AdaptiveGridTrackDensity::Config cfg;
   // force track to have exactly spatialTrkGridSize spatial bins for testing
   // purposes
   cfg.spatialTrkGridSizeRange = {spatialTrkGridSize, spatialTrkGridSize};
   cfg.spatialBinExtent = spatialBinExtent;
   // force track to have exactly temporalTrkGridSize temporal bins for testing
   // purposes
   cfg.temporalTrkGridSizeRange = {temporalTrkGridSize, temporalTrkGridSize};
   cfg.temporalBinExtent = temporalBinExtent;
   cfg.useTime = true;
   AdaptiveGridTrackDensity grid(cfg);
 
   // Empty map
   AdaptiveGridTrackDensity::DensityMap mainDensityMap;
 
   // Add track
   auto trackDensityMap = grid.addTrack(params, mainDensityMap);
 
   double relTol = 1e-5;
   double small = 1e-5;
 
   auto gaussian3D = [&](const Vector3& args, const Vector3& mus,
                         const SquareMatrix3& sigmas) {
     Vector3 diffs = args - mus;
     double coef = 1 / std::sqrt(sigmas.determinant());
     double expo = -0.5 * diffs.transpose().dot(sigmas.inverse() * diffs);
     return coef * std::exp(expo);
   };
 
   for (const auto& [bin, density] : mainDensityMap) {
     // Extract variables for better readability
     double z = grid.getBinCenter(bin.first, spatialBinExtent);
     double t = grid.getBinCenter(bin.second, temporalBinExtent);
     // Argument for 3D gaussian
     Vector3 dztVec{0., z, t};
 
     // Compute correct density...
     float correctDensity = gaussian3D(dztVec, impactParameters, ipCov);
 
     // ... and check if our result is equivalent
     CHECK_CLOSE_OR_SMALL(correctDensity, density, relTol, small);
   }
 
   // The analytical calculations of the following can be found here:
   // https://acts.readthedocs.io/en/latest/white_papers/gaussian-track-densities.html
   // Analytical maximum of the Gaussian
   ActsSquareMatrix<3> ipWeights = ipCov.inverse();
   double denom =
       ipWeights(1, 1) * ipWeights(2, 2) - ipWeights(1, 2) * ipWeights(1, 2);
 
   double zNom =
       ipWeights(0, 1) * ipWeights(2, 2) - ipWeights(0, 2) * ipWeights(1, 2);
   double correctMaxZ = zNom / denom * d0 + z0;
 
   double tNom =
       ipWeights(0, 2) * ipWeights(1, 1) - ipWeights(0, 1) * ipWeights(1, 2);
   double correctMaxT = tNom / denom * d0 + t0;
 
   // Analytical FWHM of the Gaussian
   double correctFWHM = 2. * std::sqrt(2 * std::numbers::ln2 / ipWeights(1, 1));
 
   // Estimate maximum z position and seed width
   auto res = grid.getMaxZTPositionAndWidth(mainDensityMap);
   BOOST_CHECK(res.ok());
 
   // Extract variables for better readability...
   double maxZ = res.value().first.first;
   double maxT = res.value().first.second;
   double fwhm = res.value().second * 2.355;
 
   // ... and check if they are correct (note: the optimization is not as exact
   // as the density values).
   double relTolOptimization = 1e-1;
   CHECK_CLOSE_REL(correctMaxZ, maxZ, relTolOptimization);
   CHECK_CLOSE_REL(correctMaxT, maxT, relTolOptimization);
   CHECK_CLOSE_REL(correctFWHM, fwhm, relTolOptimization);
 }
 
 BOOST_AUTO_TEST_CASE(seed_width_estimation) {
   // Dummy track grid size (not needed for this unit test)
   const std::uint32_t spatialTrkGridSize = 1;
   double binExtent = 2.;
   AdaptiveGridTrackDensity::Config cfg;
   cfg.spatialTrkGridSizeRange = {spatialTrkGridSize, spatialTrkGridSize};
   cfg.spatialBinExtent = binExtent;
   AdaptiveGridTrackDensity grid(cfg);
 
   // Empty map
   AdaptiveGridTrackDensity::DensityMap mainDensityMap;
 
   // z-position of the maximum density
   double correctMaxZ = -2.;
 
   // Fill map with a triangular track density.
   // We use an isoscele triangle with a maximum density value of 1 and a width
   // of 20 mm. The linear approximation we use during the seed width estimation
   // should be exact in this case.
   for (std::int32_t i = -6; i <= 4; i++) {
     mainDensityMap[{i, 0}] =
         1.0 - 0.1 * std::abs(correctMaxZ - grid.getBinCenter(i, binExtent));
   }
 
   // Get maximum z position and corresponding seed width
   auto res = grid.getMaxZTPositionAndWidth(mainDensityMap);
   BOOST_CHECK(res.ok());
 
   // Check if we found the correct maximum
   double maxZ = res.value().first.first;
   BOOST_CHECK_EQUAL(correctMaxZ, maxZ);
 
   // Calculate full width at half maximum from seed width and check if it's
   // correct
   double fwhm = res.value().second * 2.355;
   CHECK_CLOSE_REL(10., fwhm, 1e-5);
 }
 
 BOOST_AUTO_TEST_CASE(track_adding) {
   const std::uint32_t spatialTrkGridSize = 15;
 
   double binExtent = 0.1;  // mm
 
   AdaptiveGridTrackDensity::Config cfg;
   cfg.spatialTrkGridSizeRange = {spatialTrkGridSize, spatialTrkGridSize};
   cfg.spatialBinExtent = binExtent;
   AdaptiveGridTrackDensity grid(cfg);
 
   // Create some test tracks in such a way that some tracks
   //  e.g. overlap and that certain tracks need to be inserted
   // between two other tracks
   Covariance covMat(Covariance::Identity());
 
   BoundVector paramVec0;
   paramVec0 << 100.0, -0.4, 0, 0, 0, 0;
   BoundVector paramVec1;
   paramVec1 << 0.01, -0.4, 0, 0, 0, 0;
   BoundVector paramVec2;
   paramVec2 << 0.01, 10.9, 0, 0, 0, 0;
   BoundVector paramVec3;
   paramVec3 << 0.01, 0.9, 0, 0, 0, 0;
 
   // Create perigee surface
   std::shared_ptr<PerigeeSurface> perigeeSurface =
       Surface::makeShared<PerigeeSurface>(Vector3(0., 0., 0.));
 
   BoundTrackParameters params0(perigeeSurface, paramVec0, covMat,
                                ParticleHypothesis::pion());
   BoundTrackParameters params1(perigeeSurface, paramVec1, covMat,
                                ParticleHypothesis::pion());
   BoundTrackParameters params2(perigeeSurface, paramVec2, covMat,
                                ParticleHypothesis::pion());
   BoundTrackParameters params3(perigeeSurface, paramVec3, covMat,
                                ParticleHypothesis::pion());
 
   // Empty map
   AdaptiveGridTrackDensity::DensityMap mainDensityMap;
 
   // Track is too far away from z axis and was not added
   auto trackDensityMap = grid.addTrack(params0, mainDensityMap);
   BOOST_CHECK(mainDensityMap.empty());
 
   // Track should have been entirely added to both grids
   trackDensityMap = grid.addTrack(params1, mainDensityMap);
   BOOST_CHECK_EQUAL(spatialTrkGridSize, mainDensityMap.size());
 
   // Track should have been entirely added to both grids
   trackDensityMap = grid.addTrack(params2, mainDensityMap);
   BOOST_CHECK_EQUAL(2 * spatialTrkGridSize, mainDensityMap.size());
 
   // Track 3 has overlap of 2 bins with track 1
   trackDensityMap = grid.addTrack(params3, mainDensityMap);
   BOOST_CHECK_EQUAL(3 * spatialTrkGridSize - 2, mainDensityMap.size());
 
   // Add first track again, should *not* introduce new z entries
   trackDensityMap = grid.addTrack(params1, mainDensityMap);
   BOOST_CHECK_EQUAL(3 * spatialTrkGridSize - 2, mainDensityMap.size());
 }
 
 BOOST_AUTO_TEST_CASE(max_z_t_and_width) {
   const std::uint32_t spatialTrkGridSize = 29;
   const std::uint32_t temporalTrkGridSize = 29;
 
   // spatial and temporal bin extent
   double binExtent = 0.05;
 
   // 1D grid of z values
   AdaptiveGridTrackDensity::Config cfg1D;
   cfg1D.spatialTrkGridSizeRange = {spatialTrkGridSize, spatialTrkGridSize};
   cfg1D.spatialBinExtent = binExtent;
   AdaptiveGridTrackDensity grid1D(cfg1D);
 
   // 2D grid of z and t values
   AdaptiveGridTrackDensity::Config cfg2D;
   cfg2D.spatialTrkGridSizeRange = {spatialTrkGridSize, spatialTrkGridSize};
   cfg2D.spatialBinExtent = binExtent;
   cfg2D.temporalTrkGridSizeRange = {temporalTrkGridSize, temporalTrkGridSize};
   cfg2D.temporalBinExtent = binExtent;
   cfg2D.useTime = true;
   AdaptiveGridTrackDensity grid2D(cfg2D);
 
   // Create some test tracks
   Covariance covMat(Covariance::Identity() * 0.005);
 
   double z0Trk1 = 0.25;
   double t0Trk1 = 0.05;
   double z0Trk2 = -10.95;
   double t0Trk2 = 0.1;
   BoundVector paramVec1;
   paramVec1 << 0.02, z0Trk1, 0, 0, 0, t0Trk1;
   BoundVector paramVec2;
   paramVec2 << 0.01, z0Trk2, 0, 0, 0, t0Trk2;
 
   // Create perigee surface
   std::shared_ptr<PerigeeSurface> perigeeSurface =
       Surface::makeShared<PerigeeSurface>(Vector3(0., 0., 0.));
 
   BoundTrackParameters params1(perigeeSurface, paramVec1, covMat,
                                ParticleHypothesis::pion());
   BoundTrackParameters params2(perigeeSurface, paramVec2, covMat,
                                ParticleHypothesis::pion());
 
   // Empty maps
   AdaptiveGridTrackDensity::DensityMap mainDensityMap1D;
   AdaptiveGridTrackDensity::DensityMap mainDensityMap2D;
 
   // Add first track to spatial grid
   auto trackDensityMap = grid1D.addTrack(params1, mainDensityMap1D);
   auto firstRes = grid1D.getMaxZTPosition(mainDensityMap1D);
   BOOST_CHECK(firstRes.ok());
   // Maximum should be at z0Trk1 position ...
   BOOST_CHECK_EQUAL(z0Trk1, (*firstRes).first);
   // ... and the corresponding time should be set to 0
   BOOST_CHECK_EQUAL(0., (*firstRes).second);
 
   // Add first track to 2D grid
   auto trackDensityMap2D = grid2D.addTrack(params1, mainDensityMap2D);
   auto firstRes2D = grid2D.getMaxZTPosition(mainDensityMap2D);
   BOOST_CHECK(firstRes2D.ok());
   // Maximum should be at z0Trk1 position ...
   BOOST_CHECK_EQUAL(z0Trk1, (*firstRes2D).first);
   // ... and the corresponding time should be at t0Trk1
   BOOST_CHECK_EQUAL(t0Trk1, (*firstRes2D).second);
 
   // Add second track to spatial grid
   trackDensityMap = grid1D.addTrack(params2, mainDensityMap1D);
   // Calculate maximum and the corresponding width
   auto secondRes = grid1D.getMaxZTPositionAndWidth(mainDensityMap1D);
   BOOST_CHECK(secondRes.ok());
   // Trk 2 is closer to z-axis and should thus yield higher density values.
   // Therefore, the new maximum is at z0Trk2 ...
   CHECK_CLOSE_OR_SMALL(z0Trk2, (*secondRes).first.first, 1e-5, 1e-5);
   // ... the corresponding time should be set to 0...
   BOOST_CHECK_EQUAL(0., (*secondRes).first.second);
   // ... and it should have a positive width
   BOOST_CHECK_LT(0, (*secondRes).second);
 
   // Add second track to 2D grid
   trackDensityMap = grid2D.addTrack(params2, mainDensityMap2D);
   // Calculate maximum and the corresponding width
   auto secondRes2D = grid2D.getMaxZTPositionAndWidth(mainDensityMap2D);
   BOOST_CHECK(secondRes2D.ok());
   // Trk 2 is closer to z-axis and should thus yield higher density values.
   // Therefore, the new maximum is at z0Trk2 ...
   CHECK_CLOSE_OR_SMALL(z0Trk2, (*secondRes2D).first.first, 1e-5, 1e-5);
   // ... the corresponding time should be at t0Trk2 ...
   BOOST_CHECK_EQUAL(t0Trk2, (*secondRes2D).first.second);
   // ... and it should have approximately the same width in z direction
   CHECK_CLOSE_OR_SMALL((*secondRes2D).second, (*secondRes).second, 1e-5, 1e-5);
 }
 
 BOOST_AUTO_TEST_CASE(highest_density_sum) {
   const std::uint32_t spatialTrkGridSize = 29;
 
   double binExtent = 0.05;  // mm
 
   AdaptiveGridTrackDensity::Config cfg;
   cfg.spatialTrkGridSizeRange = {spatialTrkGridSize, spatialTrkGridSize};
   cfg.spatialBinExtent = binExtent;
   cfg.useHighestSumZPosition = true;
 
   AdaptiveGridTrackDensity grid(cfg);
 
   // Create some test tracks
   Covariance covMat(Covariance::Identity() * 0.005);
 
   double z0Trk1 = 0.25;
   double z0Trk2 = -10.95;
   BoundVector paramVec1;
   paramVec1 << 0.01, z0Trk1, 0, 0, 0, 0;
   BoundVector paramVec2;
   paramVec2 << 0.0095, z0Trk2, 0, 0, 0, 0;
 
   // Create perigee surface
   std::shared_ptr<PerigeeSurface> perigeeSurface =
       Surface::makeShared<PerigeeSurface>(Vector3(0., 0., 0.));
 
   BoundTrackParameters params1(perigeeSurface, paramVec1, covMat,
                                ParticleHypothesis::pion());
   BoundTrackParameters params2(perigeeSurface, paramVec2, covMat,
                                ParticleHypothesis::pion());
 
   // Empty map
   AdaptiveGridTrackDensity::DensityMap mainDensityMap;
 
   // Fill grid with track densities
   auto trackDensityMap = grid.addTrack(params1, mainDensityMap);
 
   auto res1 = grid.getMaxZTPosition(mainDensityMap);
   BOOST_CHECK(res1.ok());
   // Maximum should be at z0Trk1 position
   BOOST_CHECK_EQUAL(z0Trk1, (*res1).first);
 
   // Add second track
   trackDensityMap = grid.addTrack(params2, mainDensityMap);
   auto res2 = grid.getMaxZTPosition(mainDensityMap);
   BOOST_CHECK(res2.ok());
   // Trk 2 is closer to z-axis and should yield higher density values
   // New maximum is therefore at z0Trk2
   CHECK_CLOSE_REL(z0Trk2, (*res2).first, 1e-5);
 
   // Add small density values around the maximum of track 1
   mainDensityMap[{4, 0}] += 0.5;
   mainDensityMap[{6, 0}] += 0.5;
 
   auto res3 = grid.getMaxZTPosition(mainDensityMap);
   BOOST_CHECK(res3.ok());
   // Trk 2 still has the highest peak density value, however, the small
   // added densities for track 1 around its maximum should now lead to
   // a prediction at z0Trk1 again
   BOOST_CHECK_EQUAL(z0Trk1, (*res3).first);
 }
 
 BOOST_AUTO_TEST_CASE(track_removing) {
   const std::uint32_t spatialTrkGridSize = 29;
   const std::uint32_t temporalTrkGridSize = 29;
 
   std::uint32_t trkGridSize = spatialTrkGridSize * temporalTrkGridSize;
 
   // bin extent in z and t direction
   double binExtent = 0.05;
 
   // 1D grid
   AdaptiveGridTrackDensity::Config cfg1D;
   cfg1D.spatialTrkGridSizeRange = {spatialTrkGridSize, spatialTrkGridSize};
   cfg1D.spatialBinExtent = binExtent;
   AdaptiveGridTrackDensity grid1D(cfg1D);
 
   // 2D grid
   AdaptiveGridTrackDensity::Config cfg2D;
   cfg2D.spatialTrkGridSizeRange = {spatialTrkGridSize, spatialTrkGridSize};
   cfg2D.spatialBinExtent = binExtent;
   cfg2D.temporalTrkGridSizeRange = {temporalTrkGridSize, temporalTrkGridSize};
   cfg2D.temporalBinExtent = binExtent;
   cfg2D.useTime = true;
   AdaptiveGridTrackDensity grid2D(cfg2D);
 
   // Create some test tracks
   Covariance covMat = makeRandomCovariance();
 
   // Define z0 values for test tracks
   double z0Trk1 = -0.45;
   double t0Trk1 = -0.15;
   double z0Trk2 = -0.25;
   double t0Trk2 = t0Trk1;
 
   BoundVector paramVec0;
   paramVec0 << 0.1, z0Trk1, 0, 0, 0, t0Trk1;
   BoundVector paramVec1;
   paramVec1 << 0.1, z0Trk2, 0, 0, 0, t0Trk2;
 
   // Create perigee surface
   std::shared_ptr<PerigeeSurface> perigeeSurface =
       Surface::makeShared<PerigeeSurface>(Vector3(0., 0., 0.));
 
   BoundTrackParameters params0(perigeeSurface, paramVec0, covMat,
                                ParticleHypothesis::pion());
   BoundTrackParameters params1(perigeeSurface, paramVec1, covMat,
                                ParticleHypothesis::pion());
 
   // Empty maps
   AdaptiveGridTrackDensity::DensityMap mainDensityMap1D;
   AdaptiveGridTrackDensity::DensityMap mainDensityMap2D;
 
   // Lambda for calculating total density
   auto densitySum = [](const auto& densityMap) {
     double sum = 0.;
     for (const auto& [_, density] : densityMap) {
       sum += density;
     }
     return sum;
   };
 
   // Add track 0 to 1D grid
   auto firstTrackDensityMap1D = grid1D.addTrack(params0, mainDensityMap1D);
   BOOST_CHECK(!mainDensityMap1D.empty());
   // Grid size should match spatialTrkGridSize
   BOOST_CHECK_EQUAL(spatialTrkGridSize, mainDensityMap1D.size());
   double firstDensitySum1D = densitySum(mainDensityMap1D);
 
   // Add track 0 to 2D grid
   auto firstTrackDensityMap2D = grid2D.addTrack(params0, mainDensityMap2D);
   BOOST_CHECK(!mainDensityMap2D.empty());
   // Grid size should match spatialTrkGridSize
   BOOST_CHECK_EQUAL(trkGridSize, mainDensityMap2D.size());
   double firstDensitySum2D = densitySum(mainDensityMap2D);
 
   // Add track 0 again to 1D grid
   firstTrackDensityMap1D = grid1D.addTrack(params0, mainDensityMap1D);
   BOOST_CHECK(!mainDensityMap1D.empty());
   // Grid size should still match spatialTrkGridSize
   BOOST_CHECK_EQUAL(spatialTrkGridSize, mainDensityMap1D.size());
   // Calculate new total density ...
   double secondDensitySum1D = densitySum(mainDensityMap1D);
   // ... and check that it's twice as large as before
   BOOST_CHECK_EQUAL(2 * firstDensitySum1D, secondDensitySum1D);
 
   // Add track 0 again to 2D grid
   firstTrackDensityMap2D = grid2D.addTrack(params0, mainDensityMap2D);
   BOOST_CHECK(!mainDensityMap2D.empty());
   // Grid size should still match trkGridSize
   BOOST_CHECK_EQUAL(trkGridSize, mainDensityMap2D.size());
   // Calculate new total density ...
   double secondDensitySum2D = densitySum(mainDensityMap2D);
   // ... and check that it's twice as large as before
   BOOST_CHECK_EQUAL(2 * firstDensitySum2D, secondDensitySum2D);
 
   // Remove track 0 from 1D grid
   grid1D.subtractTrack(firstTrackDensityMap1D, mainDensityMap1D);
   // Calculate new total density
   double thirdDensitySum1D = densitySum(mainDensityMap1D);
   // Density should be old one again
   BOOST_CHECK_EQUAL(firstDensitySum1D, thirdDensitySum1D);
   // Grid size should still match spatialTrkGridSize (removal does not change
   // grid size)
   BOOST_CHECK_EQUAL(spatialTrkGridSize, mainDensityMap1D.size());
 
   // Remove track 0 from 2D grid
   grid2D.subtractTrack(firstTrackDensityMap2D, mainDensityMap2D);
   // Calculate new total density
   double thirdDensitySum2D = densitySum(mainDensityMap2D);
   // Density should be old one again
   BOOST_CHECK_EQUAL(firstDensitySum2D, thirdDensitySum2D);
   // Grid size should still match trkGridSize (removal does not change grid
   // size)
   BOOST_CHECK_EQUAL(trkGridSize, mainDensityMap2D.size());
 
   // Add track 1 to 1D grid (overlaps with track 0!)
   auto secondTrackDensityMap1D = grid1D.addTrack(params1, mainDensityMap1D);
   std::uint32_t nNonOverlappingBins1D =
       static_cast<std::uint32_t>(std::abs(z0Trk1 - z0Trk2) / binExtent);
   BOOST_CHECK_EQUAL(spatialTrkGridSize + nNonOverlappingBins1D,
                     mainDensityMap1D.size());
   double fourthDensitySum1D = densitySum(mainDensityMap1D);
 
   // Add track 1 to 2D grid (overlaps with track 0!)
   auto secondTrackDensityMap2D = grid2D.addTrack(params1, mainDensityMap2D);
   std::uint32_t nNonOverlappingBins2D =
       nNonOverlappingBins1D * temporalTrkGridSize;
   BOOST_CHECK_EQUAL(trkGridSize + nNonOverlappingBins2D,
                     mainDensityMap2D.size());
   double fourthDensitySum2D = densitySum(mainDensityMap2D);
 
   // Remove second track 0 from 1D grid
   grid1D.subtractTrack(firstTrackDensityMap1D, mainDensityMap1D);
   double fifthDensitySum1D = densitySum(mainDensityMap1D);
   // Density should match differences of removed tracks
   CHECK_CLOSE_REL(fifthDensitySum1D, fourthDensitySum1D - firstDensitySum1D,
                   1e-5);
 
   // Remove second track 0 from 2D grid
   grid2D.subtractTrack(firstTrackDensityMap2D, mainDensityMap2D);
   double fifthDensitySum2D = densitySum(mainDensityMap2D);
   // Density should match differences of removed tracks
   CHECK_CLOSE_REL(fifthDensitySum2D, fourthDensitySum2D - firstDensitySum2D,
                   1e-5);
 
   // Remove track 1 from 1D grid
   grid1D.subtractTrack(secondTrackDensityMap1D, mainDensityMap1D);
   // Size should not have changed
   BOOST_CHECK_EQUAL(spatialTrkGridSize + nNonOverlappingBins1D,
                     mainDensityMap1D.size());
   double sixthDensitySum1D = densitySum(mainDensityMap1D);
   // 1D grid is now empty after all tracks were removed
   CHECK_CLOSE_ABS(0., sixthDensitySum1D, 1e-4);
 
   // Remove track 1 from 2D grid
   grid2D.subtractTrack(secondTrackDensityMap2D, mainDensityMap2D);
   // Size should not have changed
   BOOST_CHECK_EQUAL(trkGridSize + nNonOverlappingBins2D,
                     mainDensityMap2D.size());
   double sixthDensitySum2D = densitySum(mainDensityMap2D);
   // 2D grid is now empty after all tracks were removed
   CHECK_CLOSE_ABS(0., sixthDensitySum2D, 1e-4);
 }
 
 }  // namespace Acts::Test

This page was automatically generated by the 2.3.7 LXR engine. The LXR team