-
Notifications
You must be signed in to change notification settings - Fork 19
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #895 from beomki-yeo/test-for-backward-propagation
Enable backward propagation
- Loading branch information
Showing
4 changed files
with
174 additions
and
18 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
152 changes: 152 additions & 0 deletions
152
tests/integration_tests/cpu/propagator/backward_propagation.cpp
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,152 @@ | ||
| /** Detray library, part of the ACTS project (R&D line) | ||
| * | ||
| * (c) 2022-2024 CERN for the benefit of the ACTS project | ||
| * | ||
| * Mozilla Public License Version 2.0 | ||
| */ | ||
|
|
||
| // Project include(s). | ||
| #include "detray/definitions/units.hpp" | ||
| #include "detray/detectors/bfield.hpp" | ||
| #include "detray/geometry/barcode.hpp" | ||
| #include "detray/geometry/shapes/rectangle2D.hpp" | ||
| #include "detray/navigation/detail/ray.hpp" | ||
| #include "detray/navigation/navigator.hpp" | ||
| #include "detray/propagator/actor_chain.hpp" | ||
| #include "detray/propagator/actors/parameter_resetter.hpp" | ||
| #include "detray/propagator/actors/parameter_transporter.hpp" | ||
| #include "detray/propagator/propagator.hpp" | ||
| #include "detray/propagator/rk_stepper.hpp" | ||
| #include "detray/tracks/tracks.hpp" | ||
|
|
||
| // Detray test include(s) | ||
| #include "detray/test/utils/detectors/build_telescope_detector.hpp" | ||
| #include "detray/test/utils/types.hpp" | ||
|
|
||
| // Vecmem include(s) | ||
| #include <vecmem/memory/host_memory_resource.hpp> | ||
|
|
||
| // google-test include(s). | ||
| #include <gtest/gtest.h> | ||
|
|
||
| using namespace detray; | ||
|
|
||
| // Algebra types | ||
| using algebra_t = test::algebra; | ||
| using point2 = test::point2; | ||
| using vector3 = test::vector3; | ||
| using matrix_operator = test::matrix_operator; | ||
|
|
||
| constexpr test::scalar tol{5e-3f}; | ||
|
|
||
| GTEST_TEST(detray_propagator, backward_propagation) { | ||
|
|
||
| vecmem::host_memory_resource host_mr; | ||
|
|
||
| // Build in x-direction from given module positions | ||
| detail::ray<algebra_t> traj{{0.f, 0.f, 0.f}, 0.f, {1.f, 0.f, 0.f}, -1.f}; | ||
| std::vector<test::scalar> positions = {0.f, 10.f, 20.f, 30.f, 40.f, 50.f, | ||
| 60.f, 70.f, 80.f, 90.f, 100.f}; | ||
|
|
||
| tel_det_config<rectangle2D> tel_cfg{200.f * unit<test::scalar>::mm, | ||
| 200.f * unit<test::scalar>::mm}; | ||
| tel_cfg.positions(positions).pilot_track(traj); | ||
|
|
||
| // Build telescope detector with rectangular planes | ||
| const auto [det, names] = build_telescope_detector(host_mr, tel_cfg); | ||
|
|
||
| // Create b field | ||
| using bfield_t = bfield::const_field_t; | ||
| vector3 B{1.f * unit<test::scalar>::T, 1.f * unit<test::scalar>::T, | ||
| 1.f * unit<test::scalar>::T}; | ||
| const bfield_t hom_bfield = bfield::create_const_field(B); | ||
|
|
||
| using navigator_t = navigator<decltype(det)>; | ||
| using rk_stepper_t = rk_stepper<bfield_t::view_t, algebra_t>; | ||
| using actor_chain_t = actor_chain<dtuple, parameter_transporter<algebra_t>, | ||
| parameter_resetter<algebra_t>>; | ||
| using propagator_t = propagator<rk_stepper_t, navigator_t, actor_chain_t>; | ||
|
|
||
| // Bound vector | ||
| bound_parameters_vector<algebra_t> bound_vector{}; | ||
| bound_vector.set_theta(constant<test::scalar>::pi_2); | ||
| bound_vector.set_qop(-1.f); | ||
|
|
||
| // Bound covariance | ||
| typename bound_track_parameters<algebra_t>::covariance_type bound_cov = | ||
| matrix_operator().template identity<e_bound_size, e_bound_size>(); | ||
|
|
||
| // Bound track parameter | ||
| const bound_track_parameters<algebra_t> bound_param0( | ||
| geometry::barcode{}.set_index(0u), bound_vector, bound_cov); | ||
|
|
||
| // Actors | ||
| parameter_transporter<algebra_t>::state bound_updater{}; | ||
| parameter_resetter<algebra_t>::state rst{}; | ||
|
|
||
| propagation::config prop_cfg{}; | ||
| prop_cfg.stepping.rk_error_tol = 1e-12f * unit<float>::mm; | ||
| prop_cfg.navigation.overstep_tolerance = -100.f * unit<float>::um; | ||
| propagator_t p{prop_cfg}; | ||
|
|
||
| // Forward state | ||
| propagator_t::state fw_state(bound_param0, hom_bfield, det, | ||
| prop_cfg.context); | ||
| fw_state.do_debug = true; | ||
|
|
||
| // Run propagator | ||
| p.propagate(fw_state, detray::tie(bound_updater, rst)); | ||
|
|
||
| // Print the debug stream | ||
| // std::cout << fw_state.debug_stream.str() << std::endl; | ||
|
|
||
| // Bound state after propagation | ||
| const auto& bound_param1 = fw_state._stepping.bound_params(); | ||
|
|
||
| // Check if the track reaches the final surface | ||
| EXPECT_EQ(bound_param0.surface_link().volume(), 4095u); | ||
| EXPECT_EQ(bound_param0.surface_link().index(), 0u); | ||
| EXPECT_EQ(bound_param1.surface_link().volume(), 0u); | ||
| EXPECT_EQ(bound_param1.surface_link().id(), surface_id::e_sensitive); | ||
| EXPECT_EQ(bound_param1.surface_link().index(), 10u); | ||
|
|
||
| // Backward state | ||
| propagator_t::state bw_state(bound_param1, hom_bfield, det, | ||
| prop_cfg.context); | ||
| bw_state.do_debug = true; | ||
| bw_state._navigation.set_direction(navigation::direction::e_backward); | ||
|
|
||
| // Run propagator | ||
| p.propagate(bw_state, detray::tie(bound_updater, rst)); | ||
|
|
||
| // Print the debug stream | ||
| // std::cout << bw_state.debug_stream.str() << std::endl; | ||
|
|
||
| // Bound state after propagation | ||
| const auto& bound_param2 = bw_state._stepping.bound_params(); | ||
|
|
||
| // Check if the track reaches the initial surface | ||
| EXPECT_EQ(bound_param2.surface_link().volume(), 0u); | ||
| EXPECT_EQ(bound_param2.surface_link().id(), surface_id::e_sensitive); | ||
| EXPECT_EQ(bound_param2.surface_link().index(), 0u); | ||
|
|
||
| const auto bound_vec0 = bound_param0.vector(); | ||
| const auto bound_vec2 = bound_param2.vector(); | ||
|
|
||
| // Check vector | ||
| for (unsigned int i = 0u; i < e_bound_size; i++) { | ||
| EXPECT_NEAR(matrix_operator().element(bound_vec0, i, 0), | ||
| matrix_operator().element(bound_vec2, i, 0), tol); | ||
| } | ||
|
|
||
| const auto bound_cov0 = bound_param0.covariance(); | ||
| const auto bound_cov2 = bound_param2.covariance(); | ||
|
|
||
| // Check covaraince | ||
| for (unsigned int i = 0u; i < e_bound_size; i++) { | ||
| for (unsigned int j = 0u; j < e_bound_size; j++) { | ||
| EXPECT_NEAR(matrix_operator().element(bound_cov0, i, j), | ||
| matrix_operator().element(bound_cov2, i, j), tol); | ||
| } | ||
| } | ||
| } |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters