def __init__(self, path):
self.path = path
self.reader = TIOReader(self.path, N_CELLS,
SKIP_SAMPLE, SKIP_END_SAMPLE,
SKIP_EVENT, SKIP_END_EVENT, True)
self.is_r1 = self.reader.fR1
if not self.is_r1:
raise IOError("This script is only setup to read *_r1.tio files!")
self.n_events = self.reader.fNEvents
self.run_id = self.reader.fRunID
self.n_pix = self.reader.fNPixels
self.n_modules = self.reader.fNModules
self.n_tmpix = self.n_pix // self.n_modules
self.n_samples = self.reader.fNSamples
self.n_cells = self.reader.fNCells
self.time_ns = None
self.time_sec = None
self.max_blocksinwf = self.n_samples // N_BLOCKSAMPLES + 1
self.samples = np.zeros((self.n_pix, self.n_samples), dtype=np.float32)
self.first_cell_ids = np.zeros(self.n_pix, dtype=np.uint16)
directory = dirname(path)
filename = splitext(basename(path))[0]
self.plot_directory = join(directory, filename)
def __init__(self, path):
self.path = path
self.reader = TIOReader(self.path, N_CELLS, SKIP_SAMPLE,
SKIP_END_SAMPLE, SKIP_EVENT, SKIP_END_EVENT)
self.n_events = self.reader.fNEvents
self.run_id = self.reader.fRunID
self.n_pix = self.reader.fNPixels
self.n_modules = self.reader.fNModules
self.n_tmpix = self.n_pix // self.n_modules
self.n_samples = self.reader.fNSamples
self.n_cells = self.reader.fNCells
self.max_blocksinwf = self.n_samples // N_BLOCKSAMPLES + 1
self.samples = np.zeros((self.n_pix, self.n_samples), dtype=np.uint16)
self.first_cell_ids = np.zeros(self.n_pix, dtype=np.uint16)
def __init__(self, url, max_events=None):
"""
Parameters
----------
url : string
path to the TARGET fits file
"""
self._event_index = None
self.url = url
self.max_events = max_events
self.event_id = 0
self.time_tack = None
self.time_sec = None
self.time_ns = None
self.cameraconfig = Config()
self.tio_reader = TIOReader(self.url, self.cameraconfig.n_cells,
self.cameraconfig.skip_sample,
self.cameraconfig.skip_end_sample,
self.cameraconfig.skip_event,
self.cameraconfig.skip_end_event)
self.n_events = self.tio_reader.fNEvents
first_event_id = self.tio_reader.fFirstEventID
last_event_id = self.tio_reader.fLastEventID
self.event_id_list = np.arange(first_event_id, last_event_id)
self.run_id = self.tio_reader.fRunID
self.n_pix = self.tio_reader.fNPixels
self.n_modules = self.tio_reader.fNModules
self.n_tmpix = self.n_pix // self.n_modules
self.n_samples = self.tio_reader.fNSamples
self.n_cells = self.tio_reader.fNCells
# Setup camera geom
if self.n_pix == self.n_tmpix:
self.cameraconfig.switch_to_single_module()
self.pixel_pos = self.cameraconfig.pixel_pos
self.optical_foclen = self.cameraconfig.optical_foclen
self.n_rows = self.cameraconfig.n_rows
self.n_columns = self.cameraconfig.n_columns
self.n_blocks = self.cameraconfig.n_blocks
self.refshape = self.cameraconfig.refshape
self.refstep = self.cameraconfig.refstep
self.time_slice = self.cameraconfig.time_slice
# Init arrays
self.r0_samples = None
self.r1_samples = np.zeros((self.n_pix, self.n_samples),
dtype=np.float32)[None, ...]
self.first_cell_ids = np.zeros(self.n_pix, dtype=np.uint16)
# Setup if file is already r1
self.is_r1 = self.tio_reader.fR1
if self.is_r1:
self.get_event = self.tio_reader.GetR1Event
self.samples = self.r1_samples[0]
else:
self.r0_samples = np.zeros((self.n_pix, self.n_samples),
dtype=np.uint16)[None, ...]
self.get_event = self.tio_reader.GetR0Event
self.samples = self.r0_samples[0]
self.data = None
self.init_container()