Skip to content

Commit 65db331

Browse files
authored
Add SNOWSTORM constants for data processing
1 parent 487df68 commit 65db331

File tree

1 file changed

+72
-0
lines changed

1 file changed

+72
-0
lines changed

src/graphnet/data/constants.py

Lines changed: 72 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,27 @@ class FEATURES:
2929
"sensor_pos_z",
3030
"t",
3131
]
32+
SNOWSTORM = [
33+
"dom_x",
34+
"dom_y",
35+
"dom_z",
36+
"charge",
37+
"dom_time",
38+
"width",
39+
"pmt_area",
40+
"rde",
41+
"is_bright_dom",
42+
"is_bad_dom",
43+
"is_saturated_dom",
44+
"is_errata_dom",
45+
"event_time",
46+
"hlc",
47+
"awtd",
48+
"string",
49+
"pmt_number",
50+
"dom_number",
51+
"dom_type",
52+
]
3253
KAGGLE = ["x", "y", "z", "time", "charge", "auxiliary"]
3354
LIQUIDO = ["sipm_x", "sipm_y", "sipm_z", "t"]
3455
KM3NET = [
@@ -94,6 +115,57 @@ class TRUTH:
94115
"primary_hadron_1_energy",
95116
"total_energy",
96117
]
118+
SNOWSTORM = [
119+
"energy",
120+
"position_x",
121+
"position_y",
122+
"position_z",
123+
"azimuth",
124+
"zenith",
125+
"pid",
126+
"event_time",
127+
"interaction_type",
128+
"elasticity",
129+
"RunID",
130+
"SubrunID",
131+
"EventID",
132+
"SubEventID",
133+
"dbang_decay_length",
134+
"track_length",
135+
"stopped_muon",
136+
"energy_track",
137+
"energy_cascade",
138+
"inelasticity",
139+
"DeepCoreFilter_13",
140+
"CascadeFilter_13",
141+
"MuonFilter_13",
142+
"OnlineL2Filter_17",
143+
"L3_oscNext_bool",
144+
"L4_oscNext_bool",
145+
"L5_oscNext_bool",
146+
"L6_oscNext_bool",
147+
"L7_oscNext_bool",
148+
"Homogenized_QTot",
149+
"MCLabelClassification",
150+
"MCLabelCoincidentMuons",
151+
"MCLabelBgMuonMCPE",
152+
"MCLabelBgMuonMCPECharge",
153+
"GNLabelTrackEnergyDeposited",
154+
"GNLabelTrackEnergyOnEntrance",
155+
"GNLabelTrackEnergyOnEntrancePrimary",
156+
"GNLabelTrackEnergyDepositedPrimary",
157+
"GNLabelEnergyPrimary",
158+
"GNLabelCascadeEnergyDepositedPrimary",
159+
"GNLabelCascadeEnergyDeposited",
160+
"GNLabelEnergyDepositedTotal",
161+
"GNLabelEnergyDepositedPrimary",
162+
"GNLabelHighestEInIceParticleIsChild",
163+
"GNLabelHighestEInIceParticleDistance",
164+
"GNLabelHighestEInIceParticleEFraction",
165+
"GNLabelHighestEInIceParticleEOnEntrance",
166+
"GNLabelHighestEDaughterDistance",
167+
"GNLabelHighestEDaughterEFraction",
168+
]
97169
KAGGLE = ["zenith", "azimuth"]
98170
LIQUIDO = [
99171
"vertex_x",

0 commit comments

Comments
 (0)