def dl1_muon_file(dl1_tmp_path):
"""
DL1 file containing only images from a muon simulation set.
"""
from ctapipe.core import run_tool
from ctapipe.tools.process import ProcessorTool
output = dl1_tmp_path / "muons.dl1.h5"
# prevent running process multiple times in case of parallel tests
with FileLock(output.with_suffix(output.suffix + ".lock")):
if output.is_file():
return output
infile = get_dataset_path("lst_muons.simtel.zst")
argv = [
f"--input={infile}",
f"--output={output}",
"--write-images",
"--DataWriter.write_parameters=False",
"--DataWriter.Contact.name=αℓℓ the äüöß",
"--SimTelEventSource.focal_length_choice=nominal",
]
assert run_tool(ProcessorTool(), argv=argv, cwd=dl1_tmp_path) == 0
return output
def dl1_proton_file(dl1_tmp_path, prod5_proton_simtel_path):
"""
DL1 file containing images and parameters for a prod5 proton run
"""
from ctapipe.tools.process import ProcessorTool
from ctapipe.core import run_tool
output = dl1_tmp_path / "proton.dl1.h5"
with FileLock(output.with_suffix(output.suffix + ".lock")):
if output.is_file():
return output
argv = [
f"--input={prod5_proton_simtel_path}",
f"--output={output}",
"--write-images",
]
assert run_tool(ProcessorTool(), argv=argv, cwd=dl1_tmp_path) == 0
return output
def dl1_parameters_file(dl1_tmp_path, prod5_gamma_simtel_path):
"""
DL1 File containing only parameters (DL1B) from a gamma simulation set.
"""
from ctapipe.core import run_tool
from ctapipe.tools.process import ProcessorTool
output = dl1_tmp_path / "gamma_parameters.dl1.h5"
# prevent running process multiple times in case of parallel tests
with FileLock(output.with_suffix(output.suffix + ".lock")):
if output.is_file():
return output
argv = [
f"--input={prod5_gamma_simtel_path}",
f"--output={output}",
"--write-parameters",
"--max-events=20",
]
assert run_tool(ProcessorTool(), argv=argv, cwd=dl1_tmp_path) == 0
return output
def dl1_by_type_file(dl1_tmp_path, prod5_gamma_simtel_path):
"""
DL1 file containing both images and parameters from a gamma simulation set.
"""
from ctapipe.tools.process import ProcessorTool
from ctapipe.core import run_tool
output = dl1_tmp_path / "gamma_by_type.dl1.h5"
# prevent running stage1 multiple times in case of parallel tests
with FileLock(output.with_suffix(output.suffix + ".lock")):
if output.is_file():
return output
argv = [
f"--input={prod5_gamma_simtel_path}",
f"--output={output}",
"--write-images",
"--max-events=20",
"--DataWriter.split_datasets_by=tel_type",
]
assert run_tool(ProcessorTool(), argv=argv, cwd=dl1_tmp_path) == 0
return output
def dl2_proton_geometry_file(dl2_tmp_path, prod5_proton_simtel_path):
"""
File containing both parameters and shower geometry from a gamma simulation set.
"""
from ctapipe.core import run_tool
from ctapipe.tools.process import ProcessorTool
output = dl2_tmp_path / "proton.training.h5"
# prevent running process multiple times in case of parallel tests
with FileLock(output.with_suffix(output.suffix + ".lock")):
if output.is_file():
return output
argv = [
f"--input={prod5_proton_simtel_path}",
f"--output={output}",
"--write-images",
"--write-stereo-shower",
"--max-events=20",
]
assert run_tool(ProcessorTool(), argv=argv, cwd=dl2_tmp_path) == 0
return output