mirror of
https://github.com/TREX-CoE/Sherman-Morrison.git
synced 2024-12-27 06:43:55 +01:00
143 lines
4.3 KiB
Python
143 lines
4.3 KiB
Python
# General functions for filling plots with data in all report's views
|
|
|
|
from bokeh.plotting import figure
|
|
from bokeh.models import HoverTool, TapTool, CustomJS
|
|
|
|
from math import pi
|
|
|
|
|
|
def fill_dotplot(
|
|
plot, source, data_field,
|
|
tooltips=None, tooltips_formatters=None,
|
|
js_tap_callback=None, server_tap_callback=None,
|
|
lines=False,
|
|
lower_bound=False
|
|
):
|
|
|
|
# (Optional) Tooltip and tooltip formatters
|
|
if tooltips is not None:
|
|
hover = HoverTool(tooltips=tooltips, mode="vline", names=["circle"])
|
|
|
|
if tooltips_formatters is not None:
|
|
hover.formatters = tooltips_formatters
|
|
|
|
plot.add_tools(hover)
|
|
|
|
# (Optional) Add TapTool (for JS tap callback)
|
|
if js_tap_callback is not None:
|
|
tap = TapTool(callback=CustomJS(code=js_tap_callback))
|
|
plot.add_tools(tap)
|
|
|
|
# (Optional) Add segment to represent a lower bound
|
|
if lower_bound:
|
|
lower_segment = plot.segment(
|
|
x0="%s_x" % data_field, y0=data_field,
|
|
x1="%s_x" % data_field, y1="%s_lower_bound" % data_field,
|
|
source=source, line_color="black"
|
|
)
|
|
|
|
# Draw dots (actually Bokeh circles)
|
|
circle = plot.circle(
|
|
name="circle",
|
|
x="%s_x" % data_field, y=data_field, source=source, size=12
|
|
)
|
|
|
|
# (Optional) Draw lines between dots
|
|
if lines:
|
|
line = plot.line(x="%s_x" % data_field, y=data_field, source=source)
|
|
|
|
# (Optional) Add server tap callback
|
|
if server_tap_callback is not None:
|
|
circle.data_source.selected.on_change("indices", server_tap_callback)
|
|
|
|
# Plot appearance
|
|
plot.xgrid.grid_line_color = None
|
|
plot.ygrid.grid_line_color = None
|
|
|
|
plot.yaxis[0].formatter.power_limit_high = 0
|
|
plot.yaxis[0].formatter.power_limit_low = 0
|
|
plot.yaxis[0].formatter.precision = 3
|
|
|
|
plot.xaxis[0].major_label_orientation = pi / 8
|
|
|
|
|
|
def fill_boxplot(
|
|
plot, source,
|
|
prefix="",
|
|
tooltips=None, tooltips_formatters=None,
|
|
js_tap_callback=None, server_tap_callback=None
|
|
):
|
|
|
|
# (Optional) Tooltip and tooltip formatters
|
|
if tooltips is not None:
|
|
hover = HoverTool(tooltips=tooltips, mode="vline", names=["full_box"])
|
|
|
|
if tooltips_formatters is not None:
|
|
hover.formatters = tooltips_formatters
|
|
|
|
plot.add_tools(hover)
|
|
|
|
# (Optional) Add TapTool (for JS tap callback)
|
|
if js_tap_callback is not None:
|
|
tap = TapTool(callback=CustomJS(code=js_tap_callback))
|
|
plot.add_tools(tap)
|
|
|
|
# Draw boxes (the prefix argument modifies the fields of ColumnDataSource
|
|
# that are used)
|
|
|
|
if prefix != "":
|
|
prefix = "%s_" % prefix
|
|
|
|
# Stems
|
|
top_stem = plot.segment(
|
|
x0="%sx" % prefix, y0="%smax" % prefix,
|
|
x1="%sx" % prefix, y1="%squantile75" % prefix,
|
|
source=source, line_color="black"
|
|
)
|
|
bottom_stem = plot.segment(
|
|
x0="%sx" % prefix, y0="%smin" % prefix,
|
|
x1="%sx" % prefix, y1="%squantile25" % prefix,
|
|
source=source, line_color="black"
|
|
)
|
|
|
|
# Boxes
|
|
full_box = plot.vbar(
|
|
name="full_box",
|
|
x="%sx" % prefix, width=0.5,
|
|
top="%squantile75" % prefix, bottom="%squantile25" % prefix,
|
|
source=source, line_color="black"
|
|
)
|
|
bottom_box = plot.vbar(
|
|
x="%sx" % prefix, width=0.5,
|
|
top="%squantile50" % prefix, bottom="%squantile25" % prefix,
|
|
source=source, line_color="black"
|
|
)
|
|
|
|
# Mu dot
|
|
mu_dot = plot.dot(
|
|
x="%sx" % prefix, y="%smu" % prefix, size=30, source=source,
|
|
color="black"
|
|
)
|
|
|
|
# (Optional) Add server tap callback
|
|
if server_tap_callback is not None:
|
|
top_stem.data_source.selected.on_change("indices", server_tap_callback)
|
|
bottom_stem.data_source.selected.on_change(
|
|
"indices", server_tap_callback)
|
|
|
|
full_box.data_source.selected.on_change("indices", server_tap_callback)
|
|
bottom_box.data_source.selected.on_change(
|
|
"indices", server_tap_callback)
|
|
|
|
mu_dot.data_source.selected.on_change("indices", server_tap_callback)
|
|
|
|
# Plot appearance
|
|
plot.xgrid.grid_line_color = None
|
|
plot.ygrid.grid_line_color = None
|
|
|
|
plot.yaxis[0].formatter.power_limit_high = 0
|
|
plot.yaxis[0].formatter.power_limit_low = 0
|
|
plot.yaxis[0].formatter.precision = 3
|
|
|
|
plot.xaxis[0].major_label_orientation = pi / 8
|