def setup(self):
self.log_format = "%(levelname)s: %(message)s [%(name)s.%(funcName)s]"
self.reader = EventSource.from_config(parent=self)
self.seeker = EventSeeker(self.reader, parent=self)
self.extractor = ImageExtractor.from_name(self.extractor_product,
parent=self)
self.dl0 = CameraDL0Reducer(parent=self)
self.dl1 = CameraDL1Calibrator(extractor=self.extractor, parent=self)
self.viewer = BokehEventViewer(parent=self)
# Setup widgets
self.viewer.create()
self.viewer.enable_automatic_index_increment()
self.create_previous_event_widget()
self.create_next_event_widget()
self.create_event_index_widget()
self.create_goto_event_index_widget()
self.create_event_id_widget()
self.create_goto_event_id_widget()
self.create_telid_widget()
self.create_channel_widget()
self.create_dl1_widgets()
self.update_dl1_widget_values()
# Setup layout
self.layout = layout([[self.viewer.layout],
[
self.w_previous_event, self.w_next_event,
self.w_goto_event_index, self.w_goto_event_id
], [self.w_event_index, self.w_event_id],
[self.w_telid, self.w_channel],
[self.wb_extractor]])
def test_telid_incorrect(example_event, example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.event = example_event
with pytest.raises(KeyError):
viewer.telid = 148937242
def test_telid_incorrect(example_event):
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
viewer.event = example_event
with pytest.raises(KeyError):
viewer.telid = 148937242
def test_channel_incorrect(example_event, example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.event = example_event
with pytest.raises(IndexError):
viewer.channel = 148937242
def test_change_time(example_event, example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.event = example_event
t = 5
viewer.change_time(t)
for cam in viewer.cameras:
assert cam.time == t
for wf in viewer.waveforms:
assert wf.active_time == t
t = -11
viewer.change_time(t)
for cam in viewer.cameras:
assert cam.time == 0
for wf in viewer.waveforms:
assert wf.active_time == 0
tel = list(example_event.r0.tel.keys())[0]
n_samples = example_event.r0.tel[tel].waveform.shape[-1]
t = 10000
viewer.change_time(t)
for cam in viewer.cameras:
assert cam.time == n_samples - 1
for wf in viewer.waveforms:
assert wf.active_time == n_samples - 1
def test_change_time(example_event):
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
viewer.event = example_event
t = 5
viewer.change_time(t)
for cam in viewer.cameras:
assert cam.time == t
for wf in viewer.waveforms:
assert wf.active_time == t
t = -11
viewer.change_time(t)
for cam in viewer.cameras:
assert cam.time == 0
for wf in viewer.waveforms:
assert wf.active_time == 0
tel = list(example_event.r0.tels_with_data)[0]
n_samples = example_event.r0.tel[tel].waveform.shape[-1]
t = 10000
viewer.change_time(t)
for cam in viewer.cameras:
assert cam.time == n_samples - 1
for wf in viewer.waveforms:
assert wf.active_time == n_samples - 1
def test_channel_incorrect(example_event):
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
viewer.event = example_event
with pytest.raises(IndexError):
viewer.channel = 148937242
def test_on_pixel_click(example_event, example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.event = example_event
p1 = 5
viewer.cameras[0]._on_pixel_click(p1)
assert viewer.waveforms[viewer.cameras[0].active_index].pixel == p1
def test_event_setting(example_event, example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.event = example_event
for cam in viewer.cameras:
assert cam.event == example_event
for wf in viewer.waveforms:
assert wf.event == example_event
def test_event_setting(example_event):
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
viewer.event = example_event
for cam in viewer.cameras:
assert cam.event == example_event
for wf in viewer.waveforms:
assert wf.event == example_event
def test_on_pixel_click(example_event):
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
viewer.event = example_event
p1 = 5
viewer.cameras[0]._on_pixel_click(p1)
assert viewer.waveforms[viewer.cameras[0].active_index].pixel == p1
def test_channel(example_event):
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
viewer.event = example_event
assert viewer.channel == 0
for cam in viewer.cameras:
assert cam.channel == 0
for wf in viewer.waveforms:
assert wf.channel == 0
def test_channel(example_event, example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.event = example_event
assert viewer.channel == 0
for cam in viewer.cameras:
assert cam.channel == 0
for wf in viewer.waveforms:
assert wf.channel == 0
def test_on_waveform_click(example_event):
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
viewer.event = example_event
t = 5
viewer.waveforms[0]._on_waveform_click(t)
for cam in viewer.cameras:
assert cam.time == t
for wf in viewer.waveforms:
assert wf.active_time == t
def test_on_waveform_click(example_event, example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.event = example_event
t = 5
viewer.waveforms[0]._on_waveform_click(t)
for cam in viewer.cameras:
assert cam.time == t
for wf in viewer.waveforms:
assert wf.active_time == t
def test_view_wf(example_event):
viewer = BokehEventViewer()
viewer.create()
viewer.event = example_event
calibrator = CameraCalibrator()
calibrator(example_event)
t = list(example_event.r0.tels_with_data)[0]
wf = viewer.waveforms[0]
wf.view = 'r1'
assert (wf.waveform == example_event.r1.tel[t].waveform[0, :]).all()
with pytest.raises(ValueError):
wf.view = 'q'
def test_view_wf(example_event, example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.event = example_event
calibrator = CameraCalibrator(subarray=example_subarray)
calibrator(example_event)
t = list(example_event.r0.tel.keys())[0]
wf = viewer.waveforms[0]
wf.view = "r1"
assert (wf.waveform == example_event.r1.tel[t].waveform[0, :]).all()
with pytest.raises(ValueError):
wf.view = "q"
def test_view_camera(example_event, example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.event = example_event
calibrator = CameraCalibrator(subarray=example_subarray)
calibrator(example_event)
t = list(example_event.r0.tel.keys())[0]
cam = viewer.cameras[0]
cam.view = "r1"
assert (cam.image == example_event.r1.tel[t].waveform[:, 0]).all()
with pytest.raises(ValueError):
cam.view = "q"
def test_view_camera(example_event):
viewer = BokehEventViewer()
viewer.create()
viewer.event = example_event
calibrator = CameraCalibrator()
calibrator(example_event)
t = list(example_event.r0.tels_with_data)[0]
cam = viewer.cameras[0]
cam.view = 'r1'
assert (cam.image == example_event.r1.tel[t].waveform[:, 0]).all()
with pytest.raises(ValueError):
cam.view = 'q'
def test_telid(example_event):
viewer = BokehEventViewer()
viewer.create()
viewer.event = example_event
tels = list(example_event.r0.tels_with_data)
assert viewer.telid == tels[0]
for cam in viewer.cameras:
assert cam.telid == tels[0]
for wf in viewer.waveforms:
assert wf.telid == tels[0]
viewer.telid = tels[1]
assert viewer.telid == tels[1]
for cam in viewer.cameras:
assert cam.telid == tels[1]
for wf in viewer.waveforms:
assert wf.telid == tels[1]
def setup(self):
self.log_format = "%(levelname)s: %(message)s [%(name)s.%(funcName)s]"
kwargs = dict(config=self.config, tool=self)
default_url = get_dataset_path("gamma_test.simtel.gz")
EventSourceFactory.input_url.default_value = default_url
self.reader = EventSourceFactory.produce(**kwargs)
self.seeker = EventSeeker(self.reader, **kwargs)
self.extractor = ChargeExtractorFactory.produce(**kwargs)
self.cleaner = WaveformCleanerFactory.produce(**kwargs)
self.r1 = CameraR1CalibratorFactory.produce(eventsource=self.reader,
**kwargs)
self.dl0 = CameraDL0Reducer(**kwargs)
self.dl1 = CameraDL1Calibrator(extractor=self.extractor,
cleaner=self.cleaner,
**kwargs)
self.viewer = BokehEventViewer(**kwargs)
# Setup widgets
self.viewer.create()
self.viewer.enable_automatic_index_increment()
self.create_previous_event_widget()
self.create_next_event_widget()
self.create_event_index_widget()
self.create_goto_event_index_widget()
self.create_event_id_widget()
self.create_goto_event_id_widget()
self.create_telid_widget()
self.create_channel_widget()
self.create_dl1_widgets()
self.update_dl1_widget_values()
# Setup layout
self.layout = layout([[self.viewer.layout],
[
self.w_previous_event, self.w_next_event,
self.w_goto_event_index, self.w_goto_event_id
], [self.w_event_index, self.w_event_id],
[self.w_telid, self.w_channel],
[self.wb_extractor]])
def test_telid(example_event, example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.event = example_event
tels = list(example_event.r0.tel.keys())
assert viewer.telid == tels[0]
for cam in viewer.cameras:
assert cam.telid == tels[0]
for wf in viewer.waveforms:
assert wf.telid == tels[0]
viewer.telid = tels[1]
assert viewer.telid == tels[1]
for cam in viewer.cameras:
assert cam.telid == tels[1]
for wf in viewer.waveforms:
assert wf.telid == tels[1]
def test_bokeh_event_viewer_creation():
viewer = BokehEventViewer()
viewer.create()
def test_enable_automatic_index_increment():
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
viewer.enable_automatic_index_increment()
for cam in viewer.cameras:
assert cam.automatic_index_increment
def test_bokeh_event_viewer_creation(example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
def test_bokeh_event_viewer_creation():
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
def test_enable_automatic_index_increment(example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
viewer.enable_automatic_index_increment()
for cam in viewer.cameras:
assert cam.automatic_index_increment
