diff --git a/flarestack/data/icecube/__init__.py b/flarestack/data/icecube/__init__.py index 9f50587e..c6054f7b 100644 --- a/flarestack/data/icecube/__init__.py +++ b/flarestack/data/icecube/__init__.py @@ -5,6 +5,7 @@ from flarestack.data.icecube.northern_tracks.nt_v002_p05 import nt_v002_p05 from flarestack.data.icecube.northern_tracks.nt_v005_p00 import nt_v005_p00 from flarestack.data.icecube.northern_tracks.nt_v005_p01 import nt_v005_p01 +from flarestack.data.icecube.northern_tracks.nt_v005_p04 import nt_v005_p04 from flarestack.data.icecube.ps_tracks.ps_v002_p01 import ps_v002_p01 from flarestack.data.icecube.ps_tracks.ps_v002_p03 import ps_v002_p03 from flarestack.data.icecube.ps_tracks.ps_v003_p01 import ps_v003_p01 diff --git a/flarestack/data/icecube/northern_tracks/nt_v005_p04.py b/flarestack/data/icecube/northern_tracks/nt_v005_p04.py new file mode 100644 index 00000000..c686aa47 --- /dev/null +++ b/flarestack/data/icecube/northern_tracks/nt_v005_p04.py @@ -0,0 +1,60 @@ +from flarestack.data.icecube.ic_season import IceCubeDataset, icecube_dataset_dir +from flarestack.data.icecube.northern_tracks import ( + NTSeasonNewStyle, + get_diffuse_binning, +) + +nt_data_dir = icecube_dataset_dir / "northern_tracks/version-005-p04" + +sample_name = "northern_tracks_v005_p04" + +dataset_name = "icecube." + sample_name + +nt_v005_p04 = IceCubeDataset(name=dataset_name) + +IC86_start_year = 2011 +IC86_stop_year = 2022 +IC86_timerange = range(IC86_start_year, IC86_stop_year + 1) + +seasons = [f"IC86_{yr}" for yr in IC86_timerange] + +# ================================== +# Add individual years as subseasons +# ================================== + + +def generate_diffuse_season(name): + season = NTSeasonNewStyle( + season_name=name, + sample_name=sample_name, + exp_path=nt_data_dir / f"{name}_exp.npy", + mc_path=nt_data_dir / "IC86_pass2_MC.npy", + grl_path=nt_data_dir / f"GRL/{name}_exp.npy", + sin_dec_bins=get_diffuse_binning(name)[0], + log_e_bins=get_diffuse_binning(name)[1], + ) + return season + + +for season in seasons: + subseason = generate_diffuse_season(season) + nt_v005_p04.add_subseason(subseason) + + +# ================================== +# Add combo season +# ================================== + +name = "IC86_1-12" + +combo_season = NTSeasonNewStyle( + season_name=name, + sample_name=sample_name, + exp_path=[nt_data_dir / f"IC86_{yr}_exp.npy" for yr in IC86_timerange], + mc_path=nt_data_dir / "IC86_pass2_MC.npy", + grl_path=[nt_data_dir / f"GRL/IC86_{yr}_exp.npy" for yr in IC86_timerange], + sin_dec_bins=get_diffuse_binning(name)[0], + log_e_bins=get_diffuse_binning(name)[1], +) + +nt_v005_p04.add_season(combo_season)