IAP GITLAB

Skip to content
Snippets Groups Projects
Commit 829e0de6 authored by Maximilian Reininghaus's avatar Maximilian Reininghaus :vulcan:
Browse files

conversion for Sibyll

parent e4c039cf
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
Processes/CMakeLists.txt
+ 1
− 1
View file @ 829e0de6
add_subdirectory (NullModel) add_subdirectory (NullModel)
add_subdirectory (Sibyll) #add_subdirectory (Sibyll)
add_subdirectory (StackInspector) add_subdirectory (StackInspector)
Processes/Sibyll/CMakeLists.txt
+ 19
− 2
View file @ 829e0de6
add_custom_command (
OUTPUT ${PROJECT_BINARY_DIR}/Processes/Sibyll/Generated.inc
COMMAND ${PROJECT_SOURCE_DIR}/Processes/Sibyll/code_generator.py
${PROJECT_BINARY_DIR}/Framework/Particles/pythia_db.pkl
${PROJECT_SOURCE_DIR}/Processes/Sibyll/sibyll_codes.dat
DEPENDS code_generator.py
sibyll_codes.dat
${PROJECT_BINARY_DIR}/Framework/Particles/pythia_db.pkl
WORKING_DIRECTORY
${PROJECT_BINARY_DIR}/Processes/Sibyll/Particles/
COMMENT "Generate conversion tables for particle codes SIBYLL <-> CORSIKA"
VERBATIM
)
set ( set (
MODEL_SOURCES MODEL_SOURCES
ParticleConversion.cc ParticleConversion.cc
...@@ -7,6 +20,7 @@ set ( ...@@ -7,6 +20,7 @@ set (
set ( set (
MODEL_HEADERS MODEL_HEADERS
ParticleConversion.h ParticleConversion.h
${PROJECT_BINARY_DIR}/Processes/Sibyll/Generated.inc
) )
set ( set (
...@@ -48,7 +62,10 @@ install ( ...@@ -48,7 +62,10 @@ install (
# -------------------- # --------------------
# code unit testing # code unit testing
add_executable (testSibyll testSibyll.cc) add_executable (testSibyll
testSibyll.cc
${MODEL_HEADERS}
)
target_link_libraries ( target_link_libraries (
testSibyll testSibyll
......
Processes/Sibyll/ParticleConversion.h
+ 15
− 185
View file @ 829e0de6
...@@ -5,195 +5,25 @@ ...@@ -5,195 +5,25 @@
#include <map> #include <map>
namespace corsika::process { namespace corsika::process::sibyll {
enum class Code : int8_t;
using PIDIntType = std::underlying_type<Code>::type;
namespace sibyll { #include <corsika/processes/sibyll/Generated.inc>
enum class PID : int { bool handledBySibyll(corsika::particles::Code pCode) {
E_MINUS = 3, return handleable[static_cast<CodeIntType>(pCode)];
E_PLUS = 2, }
NU_E = 15,
NU_E_BAR = 16,
MU_MINUS = 5,
MU_PLUS = 4,
NU_MU = 17,
NU_MU_BAR = 18,
TAU_MINUS = 91,
TAU_PLUS = 90,
NU_TAU = 92,
NU_TAU_BAR = 93,
GAMMA = 1,
PI_0 = 6,
RHO_0 = 27,
K_L_0 = 11,
PI_PLUS = 7,
PI_MINUS = 8,
RHO_PLUS = 25,
RHO_MINUS = 26,
ETA = 23,
OMEGA = 32,
K_S_0 = 12,
K_STAR_0 = 30,
K_STAR_BAR_0 = 31,
K_PLUS = 9,
K_MINUS = 10,
K_STAR_PLUS = 28,
K_STAR_MINUS = 29,
D_PLUS = 59,
D_MINUS = 60,
D_STAR_PLUS = 78,
D_STAR_MINUS = 79,
D_0 = 71,
D_BAR_0 = 72,
D_STAR_0 = 80,
D_STAR_BAR_0 = 81,
D_S_PLUS = 74,
D_S_MINUS = 75,
D_STAR_S_PLUS = 76,
D_STAR_S_MINUS = 77,
ETA_C = 73,
N_0 = 14,
N_BAR_0 = -14,
DELTA_0 = 42,
DELTA_BAR_0 = -42,
P_PLUS = 13,
P_BAR_MINUS = -13,
DELTA_PLUS = 41,
DELTA_BAR_MINUS = -41,
DELTA_PLUS_PLUS = 40,
DELTA_BAR_MINUS_MINUS = -40,
SIGMA_MINUS = 36,
SIGMA_BAR_PLUS = -36,
LAMBDA_0 = 39,
LAMBDA_BAR_0 = -39,
SIGMA_0 = 35,
SIGMA_BAR_0 = -35,
SIGMA_PLUS = 34,
SIGMA_BAR_MINUS = -34,
XI_MINUS = 38,
XI_BAR_PLUS = -38,
XI_0 = 37,
XI_BAR_0 = -37,
OMEGA_MINUS = 49,
OMEGA_BAR_PLUS = -49,
SIGMA_C_0 = 86,
SIGMA_C_BAR_0 = -86,
SIGMA_STAR_C_0 = 96,
SIGMA_STAR_C_BAR_0 = -96,
LAMBDA_C_PLUS = 89,
LAMBDA_C_BAR_MINUS = -89,
XI_C_0 = 88,
XI_C_BAR_0 = -88,
SIGMA_C_PLUS = 85,
SIGMA_C_BAR_MINUS = -85,
SIGMA_STAR_C_PLUS = 95,
SIGMA_STAR_C_BAR_MINUS = -95,
SIGMA_C_PLUS_PLUS = 84,
SIGMA_C_BAR_MINUS_MINUS = -84,
SIGMA_STAR_C_PLUS_PLUS = 94,
SIGMA_STAR_C_BAR_MINUS_MINUS = -94,
XI_C_PLUS = 87,
XI_C_BAR_MINUS = -87,
OMEGA_C_0 = 99,
OMEGA_C_BAR_0 = -99,
J_PSI = 83,
VOID = 0,
};
static const std::map<sibyll::PID, corsika::particles::Code> Sibyll2Corsika = { Code constexpr ConvertToSibyll(corsika::particles::Code pCode) {
{PID::E_MINUS, corsika::particles::Code::Electron}, //~ assert(handledBySibyll(pCode));
{PID::E_PLUS, corsika::particles::Code::Positron}, return static_cast<Code>(corsika2sibyll[static_cast<CodeIntType>(pCode)]);
{PID::NU_E, corsika::particles::Code::NuE}, }
{PID::NU_E_BAR, corsika::particles::Code::NuEBar},
{PID::MU_MINUS, corsika::particles::Code::MuMinus},
{PID::MU_PLUS, corsika::particles::Code::MuPlus},
{PID::NU_MU, corsika::particles::Code::NuMu},
{PID::NU_MU_BAR, corsika::particles::Code::NuMuBar},
{PID::TAU_MINUS, corsika::particles::Code::TauMinus},
/*
TAU_PLUS = 90,
NU_TAU = 92,
NU_TAU_BAR = 93,
GAMMA = 1,
PI_0 = 6,
RHO_0 = 27,
K_L_0 = 11,
PI_PLUS = 7,
PI_MINUS = 8,
RHO_PLUS = 25,
RHO_MINUS = 26,
ETA = 23,
OMEGA = 32,
K_S_0 = 12,
K_STAR_0 = 30,
K_STAR_BAR_0 = 31,
K_PLUS = 9,
K_MINUS = 10,
K_STAR_PLUS = 28,
K_STAR_MINUS = 29,
D_PLUS = 59,
D_MINUS = 60,
D_STAR_PLUS = 78,
D_STAR_MINUS = 79,
D_0 = 71,
D_BAR_0 = 72,
D_STAR_0 = 80,
D_STAR_BAR_0 = 81,
D_S_PLUS = 74,
D_S_MINUS = 75,
D_STAR_S_PLUS = 76,
D_STAR_S_MINUS = 77,
ETA_C = 73,
N_0 = 14,
N_BAR_0 = -14,
DELTA_0 = 42,
DELTA_BAR_0 = -42,
P_PLUS = 13,
P_BAR_MINUS = -13,
DELTA_PLUS = 41,
DELTA_BAR_MINUS = -41,
DELTA_PLUS_PLUS = 40,
DELTA_BAR_MINUS_MINUS = -40,
SIGMA_MINUS = 36,
SIGMA_BAR_PLUS = -36,
LAMBDA_0 = 39,
LAMBDA_BAR_0 = -39,
SIGMA_0 = 35,
SIGMA_BAR_0 = -35,
SIGMA_PLUS = 34,
SIGMA_BAR_MINUS = -34,
XI_MINUS = 38,
XI_BAR_PLUS = -38,
XI_0 = 37,
XI_BAR_0 = -37,
OMEGA_MINUS = 49,
OMEGA_BAR_PLUS = -49,
SIGMA_C_0 = 86,
SIGMA_C_BAR_0 = -86,
SIGMA_STAR_C_0 = 96,
SIGMA_STAR_C_BAR_0 = -96,
LAMBDA_C_PLUS = 89,
LAMBDA_C_BAR_MINUS = -89,
XI_C_0 = 88,
XI_C_BAR_0 = -88,
SIGMA_C_PLUS = 85,
SIGMA_C_BAR_MINUS = -85,
SIGMA_STAR_C_PLUS = 95,
SIGMA_STAR_C_BAR_MINUS = -95,
SIGMA_C_PLUS_PLUS = 84,
SIGMA_C_BAR_MINUS_MINUS = -84,
SIGMA_STAR_C_PLUS_PLUS = 94,
SIGMA_STAR_C_BAR_MINUS_MINUS = -94,
XI_C_PLUS = 87,
XI_C_BAR_MINUS = -87,
OMEGA_C_0 = 99,
OMEGA_C_BAR_0 = -99,
J_PSI = 83,
VOID = 0,*/
};
} // namespace sibyll corsika::particles::Code constexpr ConvertFromSibyll(Code pCode) {
return sibyll2corsika[static_cast<PIDIntType>(pCode) - minSibyll];
}
} // namespace corsika::process } // namespace corsika::process::sibyll
#endif #endif
Processes/Sibyll/code_generator.py 0 → 100755
+ 117
− 0
View file @ 829e0de6
#!/usr/bin/env python3
# (c) Copyright 2018 CORSIKA Project, corsika-project@lists.kit.edu
#
# See file AUTHORS for a list of contributors.
#
# This software is distributed under the terms of the GNU General Public
# Licence version 3 (GPL Version 3). See file LICENSE for a full version of
# the license.
import pickle, sys, itertools
# loads the pickled pythia_db (which is an OrderedDict)
def load_pythiadb(filename):
with open(filename, "rb") as f:
pythia_db = pickle.load(f)
return pythia_db
#
def read_sibyll_codes(filename, pythia_db):
with open(filename) as f:
for line in f:
line = line.strip()
if line[0] == '#':
continue
identifier, sib_code = line.split()
try:
pythia_db[identifier]["sibyll_code"] = int(sib_code)
except KeyError as e:
raise Exception("Identifier '{:s}' not found in pythia_db".format(identifier))
# generates the enum to access sibyll particles by readable names
def generate_sibyll_enum(pythia_db):
output = "enum class PID : int16_t {\n"
for identifier, d in pythia_db.items():
if d.get('sibyll_code') != None:
output += " {:s} = {:d},\n".format(identifier, d['sibyll_code'])
output += "};\n"
return output
# generates the look-up table to convert corsika codes to sibyll codes
def generate_corsika2sibyll(pythia_db):
string = "std::array<PIDIntType, {:d}> constexpr corsika2sibyll = {{".format(len(pythia_db))
for identifier, d in pythia_db.items():
sibCode = d.get("sibyll_code", 0)
string += " {:d}, // {:s}\n".format(sibCode, identifier if sibCode else identifier + " (not implemented in SIBYLL)")
string += "};\n"
return string
# generates the look-up table to convert sibyll codes to corsika codes
def generate_sibyll2corsika(pythia_db):
d = {}
for identifier, p in pythia_db.items():
if 'sibyll_code' in p:
sib_code = p['sibyll_code']
corsika_code = p['ngc_code']
d[sib_code] = (corsika_code, identifier)
string = "std::array<corsika::particles::CodeIntType, {:d}> sibyll2corsika = {{\n".format(len(d))
for k in range(min(d.keys()), max(d.keys())+1):
if k in d:
corsika_code = d[k][0]
identifier = d[k][1]
else:
corsika_code = 0
identifier = ""
string += " {:d}, // {:s}\n".format(corsika_code, identifier)
string += "};\n"
string += "PIDIntType constexpr minSibyll = {:d};\n".format(min(d.keys()))
return string
# generates the bitset for the flag whether Sibyll knows the particle
def generate_handles_particle(pythia_db):
num_particles = len(pythia_db)
num_bytes = num_particles // 32 + 1
string = "Bitset2::bitset2<{:d}> constexpr handleable{{ std::array<uint32_t, {:d}>{{{{\n".format(num_particles, num_bytes)
numeric = 0
for identifier, d in reversed(pythia_db.items()):
handledBySibyll = int("sibyll_code" in d) & 0x1
numeric = (numeric << 1) | handledBySibyll
while numeric != 0:
low = numeric & 0xFFFFFFFF
numeric = numeric >> 32
string += " 0x{:0x},\n".format(low)
string += "}}};\n"
return string
if __name__ == "__main__":
if len(sys.argv) != 3:
print("usage: {:s} <pythia_db.pkl> <sibyll_codes.dat>".format(sys.argv[0]), file=sys.stderr)
sys.exit(1)
print("code_generator.py for SIBYLL")
pythia_db = load_pythiadb(sys.argv[1])
read_sibyll_codes(sys.argv[2], pythia_db)
with open("Generated.inc", "w") as f:
print("// this file is automatically generated\n// edit at your own risk!\n", file=f)
print(generate_sibyll_enum(pythia_db), file=f)
print(generate_corsika2sibyll(pythia_db), file=f)
print(generate_handles_particle(pythia_db), file=f)
print(generate_sibyll2corsika(pythia_db), file=f)
Processes/Sibyll/sibyll_codes.dat
+ 88
− 88
View file @ 829e0de6
E_MINUS 3 Electron 3
E_PLUS 2 Positron 2
NU_E 15 NuE 15
NU_E_BAR 16 NuEBar 16
MU_MINUS 5 MuMinus 5
MU_PLUS 4 MuPlus 4
NU_MU 17 NuMu 17
NU_MU_BAR 18 NuMuBar 18
TAU_MINUS 91 TauMinus 91
TAU_PLUS 90 TauPlus 90
NU_TAU 92 NuTau 92
NU_TAU_BAR 93 NuTauBar 93
GAMMA 1 Gamma 1
PI_0 6 Pi0 6
RHO_0 27 Rho0 27
K_L_0 11 K0Long 11
PI_PLUS 7 PiPlus 7
PI_MINUS 8 PiMinus 8
RHO_PLUS 25 RhoPlus 25
RHO_MINUS 26 RhoMinus 26
ETA 23 Eta 23
OMEGA 32 Omega 32
K_S_0 12 K0Short 12
K_STAR_0 30 KStar0 30
K_STAR_BAR_0 31 KStar0Bar 31
K_PLUS 9 KPlus 9
K_MINUS 10 KMinus 10
K_STAR_PLUS 28 KStarPlus 28
K_STAR_MINUS 29 KStarMinus 29
D_PLUS 59 DPlus 59
D_MINUS 60 DMinus 60
D_STAR_PLUS 78 DStarPlus 78
D_STAR_MINUS 79 DStarMinus 79
D_0 71 D0 71
D_BAR_0 72 D0Bar 72
D_STAR_0 80 DStar0 80
D_STAR_BAR_0 81 DStar0Bar 81
D_S_PLUS 74 DsPlus 74
D_S_MINUS 75 DsMinus 75
D_STAR_S_PLUS 76 DStarSPlus 76
D_STAR_S_MINUS 77 DStarSMinus 77
ETA_C 73 EtaC 73
N_0 14 Neutron 14
N_BAR_0 -14 AntiNeutron -14
DELTA_0 42 Delta0 42
DELTA_BAR_0 -42 Delta0Bar -42
P_PLUS 13 Proton 13
P_BAR_MINUS -13 AntiProton -13
DELTA_PLUS 41 DeltaPlus 41
DELTA_BAR_MINUS -41 DeltaMinusBar -41
DELTA_PLUS_PLUS 40 DeltaPlusPlus 40
DELTA_BAR_MINUS_MINUS -40 DeltaMinusMinusBar -40
SIGMA_MINUS 36 SigmaMinus 36
SIGMA_BAR_PLUS -36 SigmaPlusBar -36
LAMBDA_0 39 Lambda0 39
LAMBDA_BAR_0 -39 Lambda0Bar -39
SIGMA_0 35 Sigma0 35
SIGMA_BAR_0 -35 Sigma0Bar -35
SIGMA_PLUS 34 SigmaPlus 34
SIGMA_BAR_MINUS -34 SigmaMinusBar -34
XI_MINUS 38 XiMinus 38
XI_BAR_PLUS -38 XiPlusBar -38
XI_0 37 Xi0 37
XI_BAR_0 -37 Xi0Bar -37
OMEGA_MINUS 49 OmegaMinus 49
OMEGA_BAR_PLUS -49 OmegaPlusBar -49
SIGMA_C_0 86 SigmaC0 86
SIGMA_C_BAR_0 -86 SigmaC0Bar -86
SIGMA_STAR_C_0 96 SigmaStarC0 96
SIGMA_STAR_C_BAR_0 -96 SigmaStarC0Bar -96
LAMBDA_C_PLUS 89 LambdaCPlus 89
LAMBDA_C_BAR_MINUS -89 LambdaCMinusBar -89
XI_C_0 88 XiC0 88
XI_C_BAR_0 -88 XiC0Bar -88
SIGMA_C_PLUS 85 SigmaCPlus 85
SIGMA_C_BAR_MINUS -85 SigmaCMinusBar -85
SIGMA_STAR_C_PLUS 95 SigmaStarCPlus 95
SIGMA_STAR_C_BAR_MINUS -95 SigmaStarCMinusBar -95
SIGMA_C_PLUS_PLUS 84 SigmaCPlusPlus 84
SIGMA_C_BAR_MINUS_MINUS -84 SigmaCMinusMinusBar -84
SIGMA_STAR_C_PLUS_PLUS 94 SigmaStarCPlusPlus 94
SIGMA_STAR_C_BAR_MINUS_MINUS -94 SigmaStarCMinusMinusBar -94
XI_C_PLUS 87 XiCPlus 87
XI_C_BAR_MINUS -87 XiCMinusBar -87
OMEGA_C_0 99 OmegaC0 99
OMEGA_C_BAR_0 -99 OmegaC0Bar -99
J_PSI 83 Jpsi 83
VOID 0 #Unknown 0
Processes/Sibyll/testSibyll.cc
+ 10
− 7
View file @ 829e0de6
...@@ -11,17 +11,20 @@ using namespace corsika; ...@@ -11,17 +11,20 @@ using namespace corsika;
TEST_CASE("Sibyll", "[processes]") { TEST_CASE("Sibyll", "[processes]") {
SECTION("ParticleConversion") { SECTION("Sibyll -> Corsika") {
REQUIRE(corsika::particles::Electron::GetCode() == REQUIRE(corsika::particles::Electron::GetCode() ==
process::sibyll::Sibyll2Corsika.at(process::sibyll::PID::E_MINUS)); process::sibyll::ConvertFromSibyll(process::sibyll::Code::Electron));
} }
SECTION("Data") { SECTION("Corsika -> Sibyll") {
REQUIRE(corsika::particles::GetName(process::sibyll::Sibyll2Corsika.at( REQUIRE(process::sibyll::ConvertToSibyll(corsika::particles::Electron::GetCode()) ==
process::sibyll::PID::E_PLUS)) == "e+"); process::sibyll::Code::Electron);
} }
SECTION("bla") {} SECTION("handledBySibyll") {
REQUIRE(process::sibyll::handledBySibyll(corsika::particles::Electron::GetCode()));
SECTION("blubb") {} REQUIRE_FALSE(
process::sibyll::handledBySibyll(corsika::particles::XiPrimeC0::GetCode()));
}
} }
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment

data privacy - impressum