diff --git a/km3buu/output.py b/km3buu/output.py index 095b157fcbdccbaa531f5068ef97458a2e090e1b..c495ff6dd4b4accddc17894fbd0d77e5818731d7 100644 --- a/km3buu/output.py +++ b/km3buu/output.py @@ -201,6 +201,7 @@ W2LIST_LOOKUP = { "TARGETZ": 16, "VERINCAN": 17, "LEPINCAN": 18, + "GIBUU_WEIGHT": 19 } W2LIST_LENGTH = len(W2LIST_LOOKUP) @@ -646,6 +647,7 @@ def write_detector_file(gibuu_output, evt.w2list[W2LIST_LOOKUP["ICHAN"]] = event.evType evt.w2list[W2LIST_LOOKUP["VERINCAN"]] = 1 evt.w2list[W2LIST_LOOKUP["LEPINCAN"]] = 1 + evt.w2list[W2LIST_LOOKUP["GIBUU_WEIGHT"]] = event.weight # Vertex Position vtx_pos = np.array(geometry.random_pos()) diff --git a/km3buu/tests/test_output.py b/km3buu/tests/test_output.py index ece49b3716fd19197a763329b1bf7e611cad0334..3ea01d93876bff4fdce04442240a278aac7c6d08 100644 --- a/km3buu/tests/test_output.py +++ b/km3buu/tests/test_output.py @@ -130,3 +130,5 @@ class TestAANET(unittest.TestCase): np.testing.assert_equal(evt.w2list[9], 34) # CC/NC np.testing.assert_equal(evt.w2list[10], 2) + # GiBUU weight + np.testing.assert_almost_equal(evt.w2list[19], 0.8167222969153614)