def test_telid_incorrect(example_event):
viewer = BokehEventViewer()
viewer.create()
viewer.event = example_event
with pytest.raises(KeyError):
viewer.telid = 148937242
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_change_time(example_event):
viewer = BokehEventViewer()
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()
viewer.create()
viewer.event = example_event
with pytest.raises(IndexError):
viewer.channel = 148937242
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_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, 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_on_pixel_click(example_event):
viewer = BokehEventViewer()
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):
viewer = BokehEventViewer()
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_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_channel(example_event):
viewer = BokehEventViewer()
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):
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_on_waveform_click(example_event):
viewer = BokehEventViewer()
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):
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_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_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_wf(example_event):
viewer = BokehEventViewer()
viewer.create()
viewer.event = example_event
c = CameraCalibrator()
c.calibrate(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, 0, :]).all()
with pytest.raises(ValueError):
wf.view = 'q'
def test_view_camera(example_event):
viewer = BokehEventViewer()
viewer.create()
viewer.event = example_event
c = CameraCalibrator()
c.calibrate(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, :, 0]).all()
with pytest.raises(ValueError):
cam.view = 'q'
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 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):
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 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_channel_incorrect(example_event):
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
viewer.event = example_event
with pytest.raises(IndexError):
viewer.channel = 148937242
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 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.r1 = CameraR1Calibrator.from_eventsource(
eventsource=self.reader,
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(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 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]
class BokehFileViewer(Tool):
name = "BokehFileViewer"
description = ("Interactively explore an event file using the bokeh "
"visualisation package")
port = Int(5006, help="Port to open bokeh server onto").tag(config=True)
disable_server = Bool(False,
help="Do not start the bokeh server "
"(useful for testing)").tag(config=True)
default_url = get_dataset_path("gamma_test_large.simtel.gz")
EventSource.input_url.default_value = default_url
extractor_product = traits.enum_trait(ImageExtractor,
default='NeighborPeakWindowSum')
aliases = Dict(
dict(
port='BokehFileViewer.port',
disable_server='BokehFileViewer.disable_server',
f='EventSource.input_url',
max_events='EventSource.max_events',
extractor='BokehFileViewer.extractor_product',
))
classes = List([
EventSource,
] + traits.classes_with_traits(ImageExtractor))
def __init__(self, **kwargs):
super().__init__(**kwargs)
self._event = None
self._event_index = None
self._event_id = None
self._telid = None
self._channel = None
self.w_next_event = None
self.w_previous_event = None
self.w_event_index = None
self.w_event_id = None
self.w_goto_event_index = None
self.w_goto_event_id = None
self.w_telid = None
self.w_channel = None
self.w_dl1_dict = None
self.wb_extractor = None
self.layout = None
self.reader = None
self.seeker = None
self.extractor = None
self.calibrator = None
self.viewer = None
self._updating_dl1 = False
# make sure, gzip files are seekable
self.config.SimTelEventSource.back_seekable = True
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,
subarray=self.reader.subarray,
)
self.calibrator = CameraCalibrator(
subarray=self.reader.subarray,
parent=self,
image_extractor=self.extractor,
)
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 start(self):
self.event_index = 0
def finish(self):
if not self.disable_server:
def modify_doc(doc):
doc.add_root(self.layout)
doc.title = self.name
directory = os.path.abspath(os.path.dirname(__file__))
theme_path = os.path.join(directory, "theme.yaml")
template_path = os.path.join(directory, "templates")
doc.theme = Theme(filename=theme_path)
env = jinja2.Environment(
loader=jinja2.FileSystemLoader(template_path))
doc.template = env.get_template('index.html')
self.log.info('Opening Bokeh application on '
'http://localhost:{}/'.format(self.port))
server = Server({'/': modify_doc}, num_procs=1, port=self.port)
server.start()
server.io_loop.add_callback(server.show, "/")
server.io_loop.start()
@property
def event_index(self):
return self._event_index
@event_index.setter
def event_index(self, val):
try:
self.event = self.seeker[val]
except IndexError:
self.log.warning(f"Event Index {val} does not exist")
@property
def event_id(self):
return self._event_id
@event_id.setter
def event_id(self, val):
try:
self.event = self.seeker[str(val)]
except IndexError:
self.log.warning(f"Event ID {val} does not exist")
@property
def telid(self):
return self._telid
@telid.setter
def telid(self, val):
self.channel = 0
tels = list(self.event.r0.tels_with_data)
if val not in tels:
val = tels[0]
self._telid = val
self.viewer.telid = val
self.update_telid_widget()
@property
def channel(self):
return self._channel
@channel.setter
def channel(self, val):
self._channel = val
self.viewer.channel = val
self.update_channel_widget()
@property
def event(self):
return self._event
@event.setter
def event(self, val):
self.calibrator(val)
self._event = val
self.viewer.event = val
self._event_index = val.count
self._event_id = val.r0.event_id
self.update_event_index_widget()
self.update_event_id_widget()
self._telid = self.viewer.telid
self.update_telid_widget()
self._channel = self.viewer.channel
self.update_channel_widget()
def update_dl1_calibrator(self, extractor=None):
"""
Recreate the dl1 calibrator with the specified extractor and cleaner
Parameters
----------
extractor : ctapipe.image.extractor.ImageExtractor
"""
if extractor is None:
extractor = self.calibrator.image_extractor
self.extractor = extractor
self.calibrator = CameraCalibrator(
subarray=self.reader.subarray,
parent=self,
image_extractor=self.extractor,
)
self.viewer.refresh()
def create_next_event_widget(self):
self.w_next_event = Button(label=">", button_type="default", width=50)
self.w_next_event.on_click(self.on_next_event_widget_click)
def on_next_event_widget_click(self):
self.event_index += 1
def create_previous_event_widget(self):
self.w_previous_event = Button(label="<",
button_type="default",
width=50)
self.w_previous_event.on_click(self.on_previous_event_widget_click)
def on_previous_event_widget_click(self):
self.event_index -= 1
def create_event_index_widget(self):
self.w_event_index = TextInput(title="Event Index:", value='')
def update_event_index_widget(self):
if self.w_event_index:
self.w_event_index.value = str(self.event_index)
def create_event_id_widget(self):
self.w_event_id = TextInput(title="Event ID:", value='')
def update_event_id_widget(self):
if self.w_event_id:
self.w_event_id.value = str(self.event_id)
def create_goto_event_index_widget(self):
self.w_goto_event_index = Button(label="GOTO Index",
button_type="default",
width=100)
self.w_goto_event_index.on_click(self.on_goto_event_index_widget_click)
def on_goto_event_index_widget_click(self):
self.event_index = int(self.w_event_index.value)
def create_goto_event_id_widget(self):
self.w_goto_event_id = Button(label="GOTO ID",
button_type="default",
width=70)
self.w_goto_event_id.on_click(self.on_goto_event_id_widget_click)
def on_goto_event_id_widget_click(self):
self.event_id = int(self.w_event_id.value)
def create_telid_widget(self):
self.w_telid = Select(title="Telescope:", value="", options=[])
self.w_telid.on_change('value', self.on_telid_widget_change)
def update_telid_widget(self):
if self.w_telid:
tels = [str(t) for t in self.event.r0.tels_with_data]
self.w_telid.options = tels
self.w_telid.value = str(self.telid)
def on_telid_widget_change(self, _, __, ___):
if self.telid != int(self.w_telid.value):
self.telid = int(self.w_telid.value)
def create_channel_widget(self):
self.w_channel = Select(title="Channel:", value="", options=[])
self.w_channel.on_change('value', self.on_channel_widget_change)
def update_channel_widget(self):
if self.w_channel:
try:
n_chan = self.event.r0.tel[self.telid].waveform.shape[0]
except AttributeError:
n_chan = 1
channels = [str(c) for c in range(n_chan)]
self.w_channel.options = channels
self.w_channel.value = str(self.channel)
def on_channel_widget_change(self, _, __, ___):
if self.channel != int(self.w_channel.value):
self.channel = int(self.w_channel.value)
def create_dl1_widgets(self):
self.w_dl1_dict = dict(
extractor=Select(title="Extractor:",
value='',
width=5,
options=BokehFileViewer.extractor_product.values),
extractor_window_start=TextInput(title="Window Start:", value=''),
extractor_window_width=TextInput(title="Window Width:", value=''),
extractor_window_shift=TextInput(title="Window Shift:", value=''),
extractor_lwt=TextInput(title="Local Pixel Weight:", value=''))
for val in self.w_dl1_dict.values():
val.on_change('value', self.on_dl1_widget_change)
self.wb_extractor = widgetbox(
PreText(text="Charge Extractor Configuration"),
self.w_dl1_dict['extractor'],
self.w_dl1_dict['extractor_window_start'],
self.w_dl1_dict['extractor_window_width'],
self.w_dl1_dict['extractor_window_shift'],
self.w_dl1_dict['extractor_lwt'])
def update_dl1_widget_values(self):
if self.w_dl1_dict:
for key, val in self.w_dl1_dict.items():
if 'extractor' in key:
if key == 'extractor':
val.value = self.extractor.__class__.__name__
else:
key = key.replace("extractor_", "")
try:
val.value = str(getattr(self.extractor, key))
except AttributeError:
val.value = ''
def on_dl1_widget_change(self, _, __, ___):
if self.event:
if not self._updating_dl1:
self._updating_dl1 = True
cmdline = []
for key, val in self.w_dl1_dict.items():
k = key.replace("extractor_", "ImageExtractor.")
if val.value:
cmdline.append(f'--{k}={val.value}')
self.parse_command_line(cmdline)
extractor = ImageExtractor.from_name(self.extractor_product,
parent=self)
self.update_dl1_calibrator(extractor)
self.update_dl1_widget_values()
self._updating_dl1 = False
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_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_bokeh_event_viewer_creation(example_subarray):
viewer = BokehEventViewer(example_subarray)
viewer.create()
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_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
def test_bokeh_event_viewer_creation():
viewer = BokehEventViewer(config=None, tool=None)
viewer.create()
def test_enable_automatic_index_increment():
viewer = BokehEventViewer()
viewer.create()
viewer.enable_automatic_index_increment()
for cam in viewer.cameras:
assert cam.automatic_index_increment
def test_bokeh_event_viewer_creation():
viewer = BokehEventViewer()
viewer.create()
class BokehFileViewer(Tool):
name = "BokehFileViewer"
description = ("Interactively explore an event file using the bokeh "
"visualisation package")
port = Int(5006, help="Port to open bokeh server onto").tag(config=True)
disable_server = Bool(False, help="Do not start the bokeh server "
"(useful for testing)").tag(config=True)
aliases = Dict(dict(
port='BokehFileViewer.port',
disable_server='BokehFileViewer.disable_server',
r='EventSourceFactory.product',
f='EventSourceFactory.input_url',
max_events='EventSourceFactory.max_events',
ped='CameraR1CalibratorFactory.pedestal_path',
tf='CameraR1CalibratorFactory.tf_path',
pe='CameraR1CalibratorFactory.pe_path',
ff='CameraR1CalibratorFactory.ff_path',
extractor='ChargeExtractorFactory.product',
extractor_t0='ChargeExtractorFactory.t0',
extractor_window_width='ChargeExtractorFactory.window_width',
extractor_window_shift='ChargeExtractorFactory.window_shift',
extractor_sig_amp_cut_HG='ChargeExtractorFactory.sig_amp_cut_HG',
extractor_sig_amp_cut_LG='ChargeExtractorFactory.sig_amp_cut_LG',
extractor_lwt='ChargeExtractorFactory.lwt',
cleaner='WaveformCleanerFactory.product',
))
classes = List([
EventSourceFactory,
ChargeExtractorFactory,
CameraR1CalibratorFactory,
CameraDL1Calibrator,
WaveformCleanerFactory
])
def __init__(self, **kwargs):
super().__init__(**kwargs)
self._event = None
self._event_index = None
self._event_id = None
self._telid = None
self._channel = None
self.w_next_event = None
self.w_previous_event = None
self.w_event_index = None
self.w_event_id = None
self.w_goto_event_index = None
self.w_goto_event_id = None
self.w_telid = None
self.w_channel = None
self.w_dl1_dict = None
self.wb_extractor = None
self.layout = None
self.reader = None
self.seeker = None
self.extractor = None
self.cleaner = None
self.r1 = None
self.dl0 = None
self.dl1 = None
self.viewer = None
self._updating_dl1 = False
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 start(self):
self.event_index = 0
def finish(self):
if not self.disable_server:
def modify_doc(doc):
doc.add_root(self.layout)
doc.title = self.name
directory = os.path.abspath(os.path.dirname(__file__))
theme_path = os.path.join(directory, "theme.yaml")
template_path = os.path.join(directory, "templates")
doc.theme = Theme(filename=theme_path)
env = jinja2.Environment(
loader=jinja2.FileSystemLoader(template_path)
)
doc.template = env.get_template('index.html')
self.log.info('Opening Bokeh application on '
'http://localhost:{}/'.format(self.port))
server = Server({'/': modify_doc}, num_procs=1, port=self.port)
server.start()
server.io_loop.add_callback(server.show, "/")
server.io_loop.start()
@property
def event_index(self):
return self._event_index
@event_index.setter
def event_index(self, val):
try:
self.event = self.seeker[val]
except IndexError:
self.log.warning("Event Index {} does not exist".format(val))
@property
def event_id(self):
return self._event_id
@event_id.setter
def event_id(self, val):
try:
self.event = self.seeker[str(val)]
except IndexError:
self.log.warning("Event ID {} does not exist".format(val))
@property
def telid(self):
return self._telid
@telid.setter
def telid(self, val):
self.channel = 0
tels = list(self.event.r0.tels_with_data)
if val not in tels:
val = tels[0]
self._telid = val
self.viewer.telid = val
self.update_telid_widget()
@property
def channel(self):
return self._channel
@channel.setter
def channel(self, val):
self._channel = val
self.viewer.channel = val
self.update_channel_widget()
@property
def event(self):
return self._event
@event.setter
def event(self, val):
# Calibrate
self.r1.calibrate(val)
self.dl0.reduce(val)
self.dl1.calibrate(val)
self._event = val
self.viewer.event = val
self._event_index = val.count
self._event_id = val.r0.event_id
self.update_event_index_widget()
self.update_event_id_widget()
self._telid = self.viewer.telid
self.update_telid_widget()
self._channel = self.viewer.channel
self.update_channel_widget()
def update_dl1_calibrator(self, extractor=None, cleaner=None):
"""
Recreate the dl1 calibrator with the specified extractor and cleaner
Parameters
----------
extractor : ctapipe.image.charge_extractors.ChargeExtractor
cleaner : ctapipe.image.waveform_cleaning.WaveformCleaner
"""
if extractor is None:
extractor = self.dl1.extractor
if cleaner is None:
cleaner = self.dl1.cleaner
self.extractor = extractor
self.cleaner = cleaner
kwargs = dict(config=self.config, tool=self)
self.dl1 = CameraDL1Calibrator(
extractor=self.extractor,
cleaner=self.cleaner,
**kwargs
)
self.dl1.calibrate(self.event)
self.viewer.refresh()
def create_next_event_widget(self):
self.w_next_event = Button(label=">", button_type="default", width=50)
self.w_next_event.on_click(self.on_next_event_widget_click)
def on_next_event_widget_click(self):
self.event_index += 1
def create_previous_event_widget(self):
self.w_previous_event = Button(
label="<",
button_type="default",
width=50
)
self.w_previous_event.on_click(self.on_previous_event_widget_click)
def on_previous_event_widget_click(self):
self.event_index -= 1
def create_event_index_widget(self):
self.w_event_index = TextInput(title="Event Index:", value='')
def update_event_index_widget(self):
if self.w_event_index:
self.w_event_index.value = str(self.event_index)
def create_event_id_widget(self):
self.w_event_id = TextInput(title="Event ID:", value='')
def update_event_id_widget(self):
if self.w_event_id:
self.w_event_id.value = str(self.event_id)
def create_goto_event_index_widget(self):
self.w_goto_event_index = Button(
label="GOTO Index",
button_type="default",
width=100
)
self.w_goto_event_index.on_click(self.on_goto_event_index_widget_click)
def on_goto_event_index_widget_click(self):
self.event_index = int(self.w_event_index.value)
def create_goto_event_id_widget(self):
self.w_goto_event_id = Button(
label="GOTO ID",
button_type="default",
width=70
)
self.w_goto_event_id.on_click(self.on_goto_event_id_widget_click)
def on_goto_event_id_widget_click(self):
self.event_id = int(self.w_event_id.value)
def create_telid_widget(self):
self.w_telid = Select(title="Telescope:", value="", options=[])
self.w_telid.on_change('value', self.on_telid_widget_change)
def update_telid_widget(self):
if self.w_telid:
tels = [str(t) for t in self.event.r0.tels_with_data]
self.w_telid.options = tels
self.w_telid.value = str(self.telid)
def on_telid_widget_change(self, _, __, ___):
if self.telid != int(self.w_telid.value):
self.telid = int(self.w_telid.value)
def create_channel_widget(self):
self.w_channel = Select(title="Channel:", value="", options=[])
self.w_channel.on_change('value', self.on_channel_widget_change)
def update_channel_widget(self):
if self.w_channel:
try:
n_chan = self.event.r0.tel[self.telid].waveform.shape[0]
except AttributeError:
n_chan = 1
channels = [str(c) for c in range(n_chan)]
self.w_channel.options = channels
self.w_channel.value = str(self.channel)
def on_channel_widget_change(self, _, __, ___):
if self.channel != int(self.w_channel.value):
self.channel = int(self.w_channel.value)
def create_dl1_widgets(self):
self.w_dl1_dict = dict(
cleaner=Select(title="Cleaner:", value='', width=5,
options=WaveformCleanerFactory.subclass_names),
extractor=Select(title="Extractor:", value='', width=5,
options=ChargeExtractorFactory.subclass_names),
extractor_t0=TextInput(title="T0:", value=''),
extractor_window_width=TextInput(title="Window Width:", value=''),
extractor_window_shift=TextInput(title="Window Shift:", value=''),
extractor_sig_amp_cut_HG=TextInput(title="Significant Amplitude "
"Cut (HG):", value=''),
extractor_sig_amp_cut_LG=TextInput(title="Significant Amplitude "
"Cut (LG):", value=''),
extractor_lwt=TextInput(title="Local Pixel Weight:", value=''))
for val in self.w_dl1_dict.values():
val.on_change('value', self.on_dl1_widget_change)
self.wb_extractor = widgetbox(
PreText(text="Charge Extractor Configuration"),
self.w_dl1_dict['cleaner'],
self.w_dl1_dict['extractor'],
self.w_dl1_dict['extractor_t0'],
self.w_dl1_dict['extractor_window_width'],
self.w_dl1_dict['extractor_window_shift'],
self.w_dl1_dict['extractor_sig_amp_cut_HG'],
self.w_dl1_dict['extractor_sig_amp_cut_LG'],
self.w_dl1_dict['extractor_lwt'])
def update_dl1_widget_values(self):
if self.w_dl1_dict:
for key, val in self.w_dl1_dict.items():
if 'extractor' in key:
if key == 'extractor':
val.value = self.extractor.__class__.__name__
else:
key = key.replace("extractor_", "")
try:
val.value = str(getattr(self.extractor, key))
except AttributeError:
val.value = ''
elif 'cleaner' in key:
if key == 'cleaner':
val.value = self.cleaner.__class__.__name__
else:
key = key.replace("cleaner_", "")
try:
val.value = str(getattr(self.cleaner, key))
except AttributeError:
val.value = ''
def on_dl1_widget_change(self, _, __, ___):
if self.event:
if not self._updating_dl1:
self._updating_dl1 = True
cmdline = []
for key, val in self.w_dl1_dict.items():
if val.value:
cmdline.append('--{}'.format(key))
cmdline.append(val.value)
self.parse_command_line(cmdline)
kwargs = dict(config=self.config, tool=self)
extractor = ChargeExtractorFactory.produce(**kwargs)
cleaner = WaveformCleanerFactory.produce(**kwargs)
self.update_dl1_calibrator(extractor, cleaner)
self.update_dl1_widget_values()
self._updating_dl1 = False
def test_bokeh_event_viewer_creation():
viewer = BokehEventViewer()
viewer.create()
