def prepare_tpf_datasource(tpf):
"""Prepare a bokeh DataSource object for selection glyphs
Parameters
----------
tpf : TargetPixelFile
TPF to be shown.
Returns
-------
tpf_source : bokeh.plotting.ColumnDataSource
Bokeh object to be shown.
"""
n_pixels = tpf.flux[0, :, :].size
pixel_index_array = np.arange(0, n_pixels, 1,
dtype=int).reshape(tpf.flux[0, :, :].shape)
xx = tpf.column + np.arange(tpf.shape[2])
yy = tpf.row + np.arange(tpf.shape[1])
xa, ya = np.meshgrid(xx, yy)
preselection = Selection()
preselection.indices = pixel_index_array[tpf.pipeline_mask].reshape(
-1).tolist()
tpf_source = ColumnDataSource(data=dict(xx=xa + 0.5, yy=ya + 0.5),
selected=preselection)
return tpf_source
def prepare_tpf_datasource(tpf, aperture_mask):
"""Prepare a bokeh DataSource object for selection glyphs
Parameters
----------
tpf : TargetPixelFile
TPF to be shown.
aperture_mask : boolean numpy array
The Aperture mask applied at the startup of interact
Returns
-------
tpf_source : bokeh.plotting.ColumnDataSource
Bokeh object to be shown.
"""
npix = tpf.flux[0, :, :].size
pixel_index_array = np.arange(0, npix, 1).reshape(tpf.flux[0].shape)
xx = tpf.column + np.arange(tpf.shape[2])
yy = tpf.row + np.arange(tpf.shape[1])
xa, ya = np.meshgrid(xx, yy)
preselected = Selection()
preselected.indices = pixel_index_array[aperture_mask].reshape(-1).tolist()
tpf_source = ColumnDataSource(data=dict(xx=xa + 0.5, yy=ya + 0.5),
selected=preselected)
return tpf_source
def prepare_bls_datasource(result, loc):
"""Prepare a bls result for bokeh plotting
Parameters
----------
result : BLS.model result
The BLS model result to use
loc : int
Index of the "best" period. (Usually the max power)
Returns
-------
bls_source : Bokeh.plotting.ColumnDataSource
Bokeh style source for plotting
"""
preselected = Selection()
preselected.indices = [loc]
bls_source = ColumnDataSource(data=dict(
period=result['period'],
power=result['power'],
depth=result['depth'],
duration=result['duration'],
transit_time=result['transit_time']),
selected=preselected)
return bls_source