Core/EventData/FreeTrackParametersTests.cpp

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 #include <boost/test/data/test_case.hpp>
0010 #include <boost/test/unit_test.hpp>
0011
0012 #include "Acts/Definitions/Algebra.hpp"
0013 #include "Acts/Definitions/Common.hpp"
0014 #include "Acts/Definitions/TrackParametrization.hpp"
0015 #include "Acts/Definitions/Units.hpp"
0016 #include "Acts/EventData/Charge.hpp"
0017 #include "Acts/EventData/GenericFreeTrackParameters.hpp"
0018 #include "Acts/EventData/TrackParameters.hpp"
0019 #include "Acts/Utilities/UnitVectors.hpp"
0020 #include "ActsTests/CommonHelpers/FloatComparisons.hpp"
0021
0022 #include <limits>
0023 #include <optional>
0024 #include <utility>
0025 #include <vector>
0026
0027 #include "TrackParametersDatasets.hpp"
0028
0029 using namespace Acts;
0030 using namespace Acts::UnitLiterals;
0031
0032 namespace {
0033
0034 constexpr auto eps = 8 * std::numeric_limits<double>::epsilon();
0035 const FreeSquareMatrix cov = FreeSquareMatrix::Identity();
0036
0037 void checkParameters(const FreeTrackParameters& params, const Vector4& pos4,
0038                      const Vector3& unitDir, double p, double q) {
0039   const auto qOverP = (q != 0) ? (q / p) : (1 / p);
0040   const auto pos = pos4.segment<3>(ePos0);
0041
0042   // native values
0043   CHECK_CLOSE_OR_SMALL(params.template get<eFreePos0>(), pos4[ePos0], eps, eps);
0044   CHECK_CLOSE_OR_SMALL(params.template get<eFreePos1>(), pos4[ePos1], eps, eps);
0045   CHECK_CLOSE_OR_SMALL(params.template get<eFreePos2>(), pos4[ePos2], eps, eps);
0046   CHECK_CLOSE_OR_SMALL(params.template get<eFreeTime>(), pos4[eTime], eps, eps);
0047   CHECK_CLOSE_OR_SMALL(params.template get<eFreeDir0>(), unitDir[eMom0], eps,
0048                        eps);
0049   CHECK_CLOSE_OR_SMALL(params.template get<eFreeDir1>(), unitDir[eMom1], eps,
0050                        eps);
0051   CHECK_CLOSE_OR_SMALL(params.template get<eFreeDir2>(), unitDir[eMom2], eps,
0052                        eps);
0053   CHECK_CLOSE_OR_SMALL(params.template get<eFreeQOverP>(), qOverP, eps, eps);
0054   // convenience accessors
0055   CHECK_CLOSE_OR_SMALL(params.fourPosition(), pos4, eps, eps);
0056   CHECK_CLOSE_OR_SMALL(params.position(), pos, eps, eps);
0057   CHECK_CLOSE_OR_SMALL(params.time(), pos4[eFreeTime], eps, eps);
0058   CHECK_CLOSE_OR_SMALL(params.direction(), unitDir, eps, eps);
0059   CHECK_CLOSE_OR_SMALL(params.absoluteMomentum(), p, eps, eps);
0060   CHECK_CLOSE_OR_SMALL(params.transverseMomentum(),
0061                        p * unitDir.template head<2>().norm(), eps, eps);
0062   CHECK_CLOSE_OR_SMALL(params.momentum(), p * unitDir, eps, eps);
0063   BOOST_CHECK_EQUAL(params.charge(), q);
0064   // self-consistency
0065   CHECK_CLOSE_OR_SMALL(params.position(),
0066                        params.parameters().template segment<3>(eFreePos0), eps,
0067                        eps);
0068   CHECK_CLOSE_OR_SMALL(params.time(), params.template get<eFreeTime>(), eps,
0069                        eps);
0070
0071   // reflection
0072   FreeTrackParameters reflectedParams = params;
0073   reflectedParams.reflectInPlace();
0074   CHECK_CLOSE_OR_SMALL(params.reflect().parameters(),
0075                        reflectedParams.parameters(), eps, eps);
0076   CHECK_CLOSE_OR_SMALL(reflectedParams.reflect().parameters(),
0077                        params.parameters(), eps, eps);
0078 }
0079
0080 }  // namespace
0081
0082 namespace ActsTests {
0083
0084 BOOST_AUTO_TEST_SUITE(EventDataSuite)
0085
0086 BOOST_DATA_TEST_CASE(
0087     NeutralConstructFromAngles,
0088     posSymmetric* posSymmetric* posSymmetric* ts* phis* thetas* ps, x, y, z,
0089     time, phi, theta, p) {
0090   Vector4 pos4(x, y, z, time);
0091   Vector3 dir = makeDirectionFromPhiTheta(phi, theta);
0092
0093   FreeTrackParameters params(pos4, phi, theta, 1 / p, std::nullopt,
0094                              ParticleHypothesis::pion0());
0095   checkParameters(params, pos4, dir, p, 0_e);
0096   BOOST_CHECK(!params.covariance());
0097
0098   // reassign w/ covariance
0099   params = FreeTrackParameters(pos4, phi, theta, 1 / p, cov,
0100                                ParticleHypothesis::pion0());
0101   BOOST_CHECK(params.covariance());
0102   BOOST_CHECK_EQUAL(params.covariance().value(), cov);
0103 }
0104
0105 BOOST_DATA_TEST_CASE(
0106     ChargedConstructFromAngles,
0107     posSymmetric* posSymmetric* posSymmetric* ts* phis* thetas* ps* qsNonZero,
0108     x, y, z, time, phi, theta, p, q) {
0109   Vector4 pos4(x, y, z, time);
0110   Vector3 dir = makeDirectionFromPhiTheta(phi, theta);
0111
0112   FreeTrackParameters params(pos4, phi, theta, q / p, std::nullopt,
0113                              ParticleHypothesis::pionLike(std::abs(q)));
0114   checkParameters(params, pos4, dir, p, q);
0115   BOOST_CHECK(!params.covariance());
0116
0117   // reassign w/ covariance
0118   params = FreeTrackParameters(pos4, phi, theta, q / p, cov,
0119                                ParticleHypothesis::pionLike(std::abs(q)));
0120   BOOST_CHECK(params.covariance());
0121   BOOST_CHECK_EQUAL(params.covariance().value(), cov);
0122 }
0123
0124 BOOST_DATA_TEST_CASE(
0125     AnyConstructFromAngles,
0126     posSymmetric* posSymmetric* posSymmetric* ts* phis* thetas* ps* qsNonZero,
0127     x, y, z, time, phi, theta, p, q) {
0128   Vector4 pos4(x, y, z, time);
0129   Vector3 dir = makeDirectionFromPhiTheta(phi, theta);
0130
0131   FreeTrackParameters params(pos4, phi, theta, q / p, std::nullopt,
0132                              ParticleHypothesis::pionLike(std::abs(q)));
0133   checkParameters(params, pos4, dir, p, q);
0134   BOOST_CHECK(!params.covariance());
0135
0136   // reassign w/ covariance
0137   params = FreeTrackParameters(pos4, phi, theta, q / p, cov,
0138                                ParticleHypothesis::pionLike(std::abs(q)));
0139   BOOST_CHECK(params.covariance());
0140   BOOST_CHECK_EQUAL(params.covariance().value(), cov);
0141 }
0142
0143 BOOST_AUTO_TEST_SUITE_END()
0144
0145 }  // namespace ActsTests