File size: 818 Bytes
cc9dfd7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 |
from .core import *
libsixel = try_import('libsixel')
def _sixel_encode(data, width, height):
s = io.BytesIO()
output = libsixel.sixel_output_new(lambda data, s: s.write(data), s)
dither = libsixel.sixel_dither_new(256)
w,h = int(width),int(height)
libsixel.sixel_dither_initialize(dither, data, w, h, libsixel.SIXEL_PIXELFORMAT_RGBA8888)
libsixel.sixel_encode(data, w, h, 1, dither, output)
return s.getvalue().decode('ascii')
def plot_sixel(fig=None):
if not libsixel:
warn("You could see this plot with `libsixel`. See https://github.com/saitoha/libsixel")
return
if fig is None: fig = plt.gcf()
fig.canvas.draw()
dpi = fig.get_dpi()
res = _sixel_encode(fig.canvas.buffer_rgba(), fig.get_figwidth()* dpi, fig.get_figheight() * dpi)
print(res)
|