def ctapipe_subarray(self):
from ctapipe.instrument import TelescopeDescription, SubarrayDescription, \
CameraGeometry, CameraReadout, CameraDescription, OpticsDescription
import astropy.units as u
geom = CameraGeometry("sstcam", self.mapping.pixel.i,
u.Quantity(self.mapping.pixel.x, 'm'),
u.Quantity(self.mapping.pixel.y, 'm'),
u.Quantity(self.mapping.pixel.size, 'm')**2,
'square')
readout = CameraReadout(
"sstcam", u.Quantity(1 / self.waveform_sample_width, "GHz"),
self.photoelectron_pulse.amplitude[None, :],
u.Quantity(self.photoelectron_pulse.sample_width, "ns"))
camera = CameraDescription("sstcam", geom, readout)
optics = OpticsDescription.from_name('SST-ASTRI')
telescope = TelescopeDescription("SST", "SST", optics, camera)
subarray = SubarrayDescription(
'toy',
tel_positions={1: [0, 0, 0] * u.m},
tel_descriptions={1: telescope},
)
return subarray
def read_single_camera_description(filename, camera_name):
"""
Read a specific camera description from a DL1 file
Parameters
----------
filename: str
camera_name: str
Returns
-------
`ctapipe.instrument.camera.description.CameraDescription`
"""
geom = read_single_camera_geometry(filename, camera_name)
readout = read_single_camera_readout(filename, camera_name)
return CameraDescription(camera_name, geometry=geom, readout=readout)
def create_subarray(geometry_version, tel_id=1):
"""
Obtain the subarray from the EventSource
Returns
-------
ctapipe.instrument.SubarrayDescription
"""
# camera info from LSTCam-[geometry_version].camgeom.fits.gz file
camera_geom = load_camera_geometry(version=geometry_version)
# get info on the camera readout:
daq_time_per_sample, pulse_shape_time_step, pulse_shapes = read_pulse_shapes(
)
camera_readout = CameraReadout(
'LSTCam',
1 / daq_time_per_sample,
pulse_shapes,
pulse_shape_time_step,
)
camera = CameraDescription('LSTCam', camera_geom, camera_readout)
lst_tel_descr = TelescopeDescription(name='LST',
tel_type='LST',
optics=OPTICS,
camera=camera)
tel_descriptions = {tel_id: lst_tel_descr}
# LSTs telescope position taken from MC from the moment
tel_positions = {tel_id: [50., 50., 16] * u.m}
subarray = SubarrayDescription(
name=f"LST-{tel_id} subarray",
tel_descriptions=tel_descriptions,
tel_positions=tel_positions,
)
return subarray
def build_camera(cam_settings, pixel_settings, telescope):
pixel_shape = cam_settings['pixel_shape'][0]
try:
pix_type, pix_rotation = CameraGeometry.simtel_shape_to_type(
pixel_shape)
except ValueError:
warnings.warn(
f'Unkown pixel_shape {pixel_shape} for camera_type {telescope.camera_name}',
UnknownPixelShapeWarning,
)
pix_type = 'hexagon'
pix_rotation = '0d'
geometry = CameraGeometry(
telescope.camera_name,
pix_id=np.arange(cam_settings['n_pixels']),
pix_x=u.Quantity(cam_settings['pixel_x'], u.m),
pix_y=u.Quantity(cam_settings['pixel_y'], u.m),
pix_area=u.Quantity(cam_settings['pixel_area'], u.m**2),
pix_type=pix_type,
pix_rotation=pix_rotation,
cam_rotation=-Angle(cam_settings['cam_rot'], u.rad),
apply_derotation=True,
)
readout = CameraReadout(
telescope.camera_name,
sampling_rate=u.Quantity(1 / pixel_settings['time_slice'], u.GHz),
reference_pulse_shape=pixel_settings['ref_shape'].astype('float64',
copy=False),
reference_pulse_sample_width=u.Quantity(pixel_settings['ref_step'],
u.ns),
)
return CameraDescription(camera_name=telescope.camera_name,
geometry=geometry,
readout=readout)
def build_camera(cam_settings, pixel_settings, telescope):
pixel_shape = cam_settings["pixel_shape"][0]
try:
pix_type, pix_rotation = CameraGeometry.simtel_shape_to_type(
pixel_shape)
except ValueError:
warnings.warn(
f"Unkown pixel_shape {pixel_shape} for camera_type {telescope.camera_name}",
UnknownPixelShapeWarning,
)
pix_type = "hexagon"
pix_rotation = "0d"
geometry = CameraGeometry(
telescope.camera_name,
pix_id=np.arange(cam_settings["n_pixels"]),
pix_x=u.Quantity(cam_settings["pixel_x"], u.m),
pix_y=u.Quantity(cam_settings["pixel_y"], u.m),
pix_area=u.Quantity(cam_settings["pixel_area"], u.m**2),
pix_type=pix_type,
pix_rotation=pix_rotation,
cam_rotation=-Angle(cam_settings["cam_rot"], u.rad),
apply_derotation=True,
)
readout = CameraReadout(
telescope.camera_name,
sampling_rate=u.Quantity(1 / pixel_settings["time_slice"], u.GHz),
reference_pulse_shape=pixel_settings["ref_shape"].astype("float64",
copy=False),
reference_pulse_sample_width=u.Quantity(pixel_settings["ref_step"],
u.ns),
)
return CameraDescription(camera_name=telescope.camera_name,
geometry=geometry,
readout=readout)
def __init__(self, **kwargs):
"""
Constructor
Parameters
----------
kwargs: dict
Parameters to be passed.
NOTE: The file mask of the data to read can be passed with
the 'input_url' parameter.
"""
try:
import uproot
except ImportError:
raise ImportError(
"The 'uproot' package is required for the DLMAGICEventSource class."
)
self.file_list = glob.glob(kwargs['input_url'])
self.file_list.sort()
# Since EventSource can not handle file wild cards as input_url
# We substitute the input_url with first file matching
# the specified file mask.
del kwargs['input_url']
super().__init__(input_url=self.file_list[0], **kwargs)
# get run number
mask = r".*_za\d+to\d+_\d_(\d+)_([A-Z]+)_.*"
parsed_info = re.findall(mask, self.file_list[0])
self.run_number = parsed_info[0][0]
# MAGIC telescope positions in m wrt. to the center of CTA simulations
self.magic_tel_positions = {
1: [-27.24, -146.66, 50.00] * u.m,
2: [-96.44, -96.77, 51.00] * u.m
}
self.magic_tel_positions = self.magic_tel_positions
# MAGIC telescope description
optics = OpticsDescription.from_name('MAGIC')
geom = CameraGeometry.from_name('MAGICCam')
# Camera Readout for NectarCam used as a placeholder
readout = CameraReadout(
'MAGICCam',
sampling_rate=u.Quantity(1, u.GHz),
reference_pulse_shape=np.array([norm.pdf(np.arange(96), 48, 6)]),
reference_pulse_sample_width=u.Quantity(1, u.ns))
camera = CameraDescription('MAGICCam', geom, readout)
self.magic_tel_description = TelescopeDescription(name='MAGIC',
tel_type='LST',
optics=optics,
camera=camera)
self.magic_tel_descriptions = {
1: self.magic_tel_description,
2: self.magic_tel_description
}
self.magic_subarray = SubarrayDescription('MAGIC',
self.magic_tel_positions,
self.magic_tel_descriptions)
# Open ROOT files
self.calib_M1, self.calib_M2, self.star_M1, self.star_M2, self.superstar = None, None, None, None, None
for file in self.file_list:
uproot_file = uproot.open(file)
if "_Y_" in file:
if "_M1_" in file:
self.calib_M1 = uproot_file["Events"]
self.meta = uproot_file["RunHeaders"]
elif "_M2_" in file:
self.calib_M2 = uproot_file["Events"]
if "_I_" in file:
if "_M1_" in file:
self.star_M1 = uproot_file["Events"]
elif "_M2_" in file:
self.star_M2 = uproot_file["Events"]
if "_S_" in file:
self.superstar = uproot_file["Events"]
self.meta = uproot_file["RunHeaders"]
self._mc_header = self._parse_mc_header()
def __init__(self, **kwargs):
"""
Constructor
Parameters
----------
kwargs: dict
Parameters to be passed.
NOTE: The file mask of the data to read can be passed with
the 'input_url' parameter.
"""
try:
import uproot
except ImportError:
raise ImportError(
"The 'uproot' package is required for the DLMAGICEventSource class."
)
self.file_list = glob.glob(kwargs["input_url"])
self.file_list.sort()
# Since EventSource can not handle file wild cards as input_url
# We substitute the input_url with first file matching
# the specified file mask.
del kwargs["input_url"]
super().__init__(input_url=self.file_list[0], **kwargs)
# Translate MAGIC shower primary id to CTA convention
self.magic_to_cta_shower_primary_id = {
1: 0, # gamma
14: 101, # MAGIC proton
3: 1, # MAGIC electron
}
# MAGIC telescope positions in m wrt. to the center of CTA simulations
self.magic_tel_positions = {
1: [-27.24, -146.66, 50.00] * u.m,
2: [-96.44, -96.77, 51.00] * u.m,
}
self.magic_tel_positions = self.magic_tel_positions
# MAGIC telescope description
optics = OpticsDescription.from_name("MAGIC")
geom = CameraGeometry.from_name("MAGICCam")
# Camera Readout for NectarCam used as a placeholder
readout = CameraReadout(
"MAGICCam",
sampling_rate=u.Quantity(1, u.GHz),
reference_pulse_shape=np.array([norm.pdf(np.arange(96), 48, 6)]),
reference_pulse_sample_width=u.Quantity(1, u.ns),
)
camera = CameraDescription("MAGICCam", geom, readout)
self.magic_tel_description = TelescopeDescription(name="MAGIC",
tel_type="LST",
optics=optics,
camera=camera)
self.magic_tel_descriptions = {
1: self.magic_tel_description,
2: self.magic_tel_description,
}
self.magic_subarray = SubarrayDescription("MAGIC",
self.magic_tel_positions,
self.magic_tel_descriptions)
# Open ROOT files
self.calib_M1, self.calib_M2, self.star_M1, self.star_M2, self.superstar = (
None,
None,
None,
None,
None,
)
for file in self.file_list:
uproot_file = uproot.open(file)
if "_Y_" in file:
if "_M1_" in file:
self.calib_M1 = uproot_file["Events"]
self.meta = uproot_file["RunHeaders"]
elif "_M2_" in file:
self.calib_M2 = uproot_file["Events"]
if "_I_" in file:
if "_M1_" in file:
self.star_M1 = uproot_file["Events"]
elif "_M2_" in file:
self.star_M2 = uproot_file["Events"]
if "_S_" in file:
self.superstar = uproot_file["Events"]
self.meta = uproot_file["RunHeaders"]
# figure out if MC or Data run
self.mc = "MMcCorsikaRunHeader." in self.meta.keys()
# get the run number directly from the root file
if self.mc:
self.run_number = int(
uproot_file["RunHeaders"]["MMcCorsikaRunHeader."]
["MMcCorsikaRunHeader.fRunNumber"].array()[0])
else:
self.run_number = int(uproot_file["RunHeaders"]["MRawRunHeader_1."]
["MRawRunHeader_1.fRunNumber"].array()[0])
self._header = self._parse_header()
def _build_telescope_description(self, file, tel_id):
pix_x, pix_y = u.Quantity(file.get_pixel_position(tel_id), u.m)
focal_length = u.Quantity(file.get_optical_foclen(tel_id), u.m)
n_pixels = len(pix_x)
try:
telescope = guess_telescope(n_pixels, focal_length)
except ValueError:
telescope = UNKNOWN_TELESCOPE
pixel_shape = file.get_pixel_shape(tel_id)[0]
try:
pix_type, pix_rot = CameraGeometry.simtel_shape_to_type(
pixel_shape)
except ValueError:
warnings.warn(
f'Unkown pixel_shape {pixel_shape} for tel_id {tel_id}',
UnknownPixelShapeWarning,
)
pix_type = 'hexagon'
pix_rot = '0d'
pix_area = u.Quantity(file.get_pixel_area(tel_id), u.m**2)
mirror_area = u.Quantity(file.get_mirror_area(tel_id), u.m**2)
num_tiles = file.get_mirror_number(tel_id)
cam_rot = file.get_camera_rotation_angle(tel_id)
num_mirrors = file.get_mirror_number(tel_id)
sampling_rate = u.Quantity(1 / file.get_time_slice(tel_id), u.GHz)
reference_pulse_shape = file.get_ref_shapes(tel_id)
reference_pulse_sample_width = u.Quantity(file.get_ref_step(tel_id),
u.ns)
geometry = CameraGeometry(
telescope.camera_name,
pix_id=np.arange(n_pixels),
pix_x=pix_x,
pix_y=pix_y,
pix_area=pix_area,
pix_type=pix_type,
pix_rotation=pix_rot,
cam_rotation=-Angle(cam_rot, u.rad),
apply_derotation=True,
)
readout = CameraReadout(
telescope.camera_name,
sampling_rate=sampling_rate,
reference_pulse_shape=reference_pulse_shape,
reference_pulse_sample_width=reference_pulse_sample_width,
)
camera = CameraDescription(camera_name=telescope.camera_name,
geometry=geometry,
readout=readout)
optics = OpticsDescription(
name=telescope.name,
num_mirrors=num_mirrors,
equivalent_focal_length=focal_length,
mirror_area=mirror_area,
num_mirror_tiles=num_tiles,
)
return TelescopeDescription(name=telescope.name,
tel_type=telescope.type,
optics=optics,
camera=camera)
