28 lines
523 B
Python
28 lines
523 B
Python
import numpy as np
|
|
|
|
from bokeh.plotting import figure
|
|
from bokeh.models import ColumnDataSource
|
|
|
|
from buki import Object
|
|
|
|
|
|
class Plot2D(Object):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
data = {
|
|
"image": [np.zeros((1, 1))]
|
|
}
|
|
|
|
self.source = source = ColumnDataSource(data=data)
|
|
|
|
fig = figure()
|
|
fig.image(source=source, x=0, y=0, dw=1, dh=1, palette="Spectral11")
|
|
super().__init__(fig)
|
|
|
|
|
|
def set(self, image):
|
|
self.source.data.update(image=[image])
|
|
|
|
|
|
|