| Line | Branch | Exec | Source |
|---|---|---|---|
| 1 | /// @file | ||
| 2 | /// @brief Type definitions for common objects used in algorithms | ||
| 3 | #pragma once | ||
| 4 | |||
| 5 | #include <string> | ||
| 6 | #include <optional> | ||
| 7 | #include <unordered_map> | ||
| 8 | |||
| 9 | namespace iguana { | ||
| 10 | |||
| 11 | /// Vector element type | ||
| 12 | using vector_element_t = double; | ||
| 13 | |||
| 14 | /// 3-momentum type | ||
| 15 | struct Momentum3 { | ||
| 16 | /// @f$x@f$-component | ||
| 17 | vector_element_t px; | ||
| 18 | /// @f$y@f$-component | ||
| 19 | vector_element_t py; | ||
| 20 | /// @f$z@f$-component | ||
| 21 | vector_element_t pz; | ||
| 22 | }; | ||
| 23 | |||
| 24 | /// 4-momentum type | ||
| 25 | struct Momentum4 { | ||
| 26 | /// @f$x@f$-component | ||
| 27 | vector_element_t px; | ||
| 28 | /// @f$y@f$-component | ||
| 29 | vector_element_t py; | ||
| 30 | /// @f$z@f$-component | ||
| 31 | vector_element_t pz; | ||
| 32 | /// @f$E@f$-component | ||
| 33 | vector_element_t E; | ||
| 34 | }; | ||
| 35 | |||
| 36 | /// Light-weight namespace for particle constants | ||
| 37 | namespace particle { | ||
| 38 | // clang-format off | ||
| 39 | |||
| 40 | /// PDG codes | ||
| 41 | enum PDG { | ||
| 42 | electron = 11, | ||
| 43 | photon = 22, | ||
| 44 | proton = 2212, | ||
| 45 | antiproton = -2212, | ||
| 46 | neutron = 2112, | ||
| 47 | antineutron = -2112, | ||
| 48 | pi_plus = 211, | ||
| 49 | pi_minus = -211, | ||
| 50 | kaon_plus = 321, | ||
| 51 | kaon_minus = -321 | ||
| 52 | }; | ||
| 53 | |||
| 54 | /// Particle names | ||
| 55 | const std::unordered_map<PDG, std::string> name{ | ||
| 56 | { electron, "electron" }, | ||
| 57 | { photon, "photon" }, | ||
| 58 | { proton, "proton" }, | ||
| 59 | { antiproton, "antiproton" }, | ||
| 60 | { neutron, "neutron" }, | ||
| 61 | { antineutron, "antineutron" }, | ||
| 62 | { pi_plus, "pi_plus" }, | ||
| 63 | { pi_minus, "pi_minus" }, | ||
| 64 | { kaon_plus, "kaon_plus" }, | ||
| 65 | { kaon_minus, "kaon_minus" } | ||
| 66 | }; | ||
| 67 | |||
| 68 | /// Particle titles | ||
| 69 | const std::unordered_map<PDG, std::string> title{ | ||
| 70 | { electron, "e^{-}" }, | ||
| 71 | { photon, "#gamma" }, | ||
| 72 | { proton, "p" }, | ||
| 73 | { antiproton, "#bar{p}" }, | ||
| 74 | { neutron, "n" }, | ||
| 75 | { antineutron, "#bar{n}" }, | ||
| 76 | { pi_plus, "#pi^{+}" }, | ||
| 77 | { pi_minus, "#pi^{-}" }, | ||
| 78 | { kaon_plus, "K^{+}" }, | ||
| 79 | { kaon_minus, "K^{-}" } | ||
| 80 | }; | ||
| 81 | |||
| 82 | /// Particle mass in GeV | ||
| 83 | const std::unordered_map<PDG, double> mass{ | ||
| 84 | { electron, 0.000511 }, | ||
| 85 | { photon, 0.0 }, | ||
| 86 | { proton, 0.938272 }, | ||
| 87 | { antiproton, 0.938272 }, | ||
| 88 | { neutron, 0.939565 }, | ||
| 89 | { antineutron, 0.939565 }, | ||
| 90 | { pi_plus, 0.139570 }, | ||
| 91 | { pi_minus, 0.139570 }, | ||
| 92 | { kaon_plus, 0.493677 }, | ||
| 93 | { kaon_minus, 0.493677 } | ||
| 94 | }; | ||
| 95 | |||
| 96 | /// @brief get a particle property given a PDG code | ||
| 97 | /// | ||
| 98 | /// Example: | ||
| 99 | /// ```cpp | ||
| 100 | /// auto mass = particle::get(particle::mass, particle::PDG::photon); // mass => 0.0 | ||
| 101 | /// ``` | ||
| 102 | /// @param property the particle property, such as `particle::name`, `particle::title`, or `particle::mass` | ||
| 103 | /// @param pdg_code the `particle::PDG` value | ||
| 104 | /// @returns the value of the property, if defined for this `pdg_code` | ||
| 105 | template <typename VALUE_TYPE> | ||
| 106 | std::optional<VALUE_TYPE> const get(std::unordered_map<PDG,VALUE_TYPE> const& property, PDG const& pdg_code) | ||
| 107 | { | ||
| 108 |
2/2✓ Branch 0 (2→3) taken 12042 times.
✓ Branch 1 (2→4) taken 110 times.
|
12152 | if(auto const& it = property.find(pdg_code); it != property.end()) |
| 109 | return it->second; | ||
| 110 | return std::nullopt; | ||
| 111 | } | ||
| 112 | |||
| 113 | /// @brief get a particle property given a PDG code | ||
| 114 | /// | ||
| 115 | /// Example: | ||
| 116 | /// ```cpp | ||
| 117 | /// auto mass = particle::get(particle::mass, 22); // mass => 0.0 | ||
| 118 | /// ``` | ||
| 119 | /// @param property the particle property, such as `particle::name`, `particle::title`, or `particle::mass` | ||
| 120 | /// @param pdg_code the `particle::PDG` value | ||
| 121 | /// @returns the value of the property, if defined for this `pdg_code` | ||
| 122 | template <typename VALUE_TYPE> | ||
| 123 | 12152 | std::optional<VALUE_TYPE> const get(std::unordered_map<PDG,VALUE_TYPE> const& property, int const& pdg_code) | |
| 124 | { | ||
| 125 |
2/2✓ Branch 0 (2→3) taken 12042 times.
✓ Branch 1 (2→4) taken 110 times.
|
12152 | return get(property, static_cast<particle::PDG>(pdg_code)); |
| 126 | } | ||
| 127 | |||
| 128 | // clang-format on | ||
| 129 | } | ||
| 130 | |||
| 131 | } | ||
| 132 |