Source code for abce.gui.bokehwidget

"""
Simple example:

.. UIExample:: 300

    import numpy as np
    from bokeh.plotting import figure
    from flexx import app, ui, event

    x = np.linspace(0, 6, 50)

    p1 = figure()
    p1.line(x, np.sin(x))

    p2 = figure()
    p2.line(x, np.cos(x))

    class Example(ui.Widget):

        def init(self):
            with ui.BoxPanel():
                ui.BokehWidget(plot=p1)
                ui.BokehWidget(plot=p2)

"""


import os

from flexx import event
from flexx.pyscript import window
from flexx.ui import Widget

Bokeh = None  # fool flakes


class BokehWidget(Widget):
    """ A widget that shows a Bokeh plot object.

    For Bokeh 0.12 and up. The plot's ``sizing_mode`` property is set to
    ``stretch_both`` unless it was set to something other than ``fixed``. Other
    responsive modes are 'scale_width', 'scale_height' and 'scale_both`, which
    all keep aspect ratio while being responsive in a certain direction.
    """

    CSS = """
    .flx-BokehWidget > .plotdiv {
        overflow: hidden;
    }
    """

    def init(self):

        # Handle client dependencies
        import bokeh
        dev = os.environ.get('BOKEH_RESOURCES', '') == 'relative-dev'
        modname = 'bokeh.' if dev else 'bokeh.min.'
        if not (modname + 'js') in self.session.get_used_asset_names():
            res = bokeh.resources.bokehjsdir()
            if dev:
                res = os.path.abspath(
                    os.path.join(bokeh.__file__,
                                 '..', '..', 'bokehjs', 'build'))
            for x in ('css', 'js'):
                filename = os.path.join(res, x, modname + x)
                self.session.add_global_asset(modname + x, filename)

    @event.prop
    def plot(self, plot=None):
        """ The Bokeh plot object to display. In JS, this prop
        provides the corresponding backbone model.
        """

        # The sizing_mode is fixed by default, but that's silly
        try:
            if plot.sizing_mode == 'fixed':
                plot.sizing_mode = 'stretch_both'
        except AttributeError:
            pass
        if plot is not None:
            self._plot_components(plot)
        return plot

    @event.emitter
    def _plot_components(self, plot):
        from bokeh.embed import components
        script, div = components(plot)
        script = '\n'.join(script.strip().split('\n')[1:-1])
        return dict(script=script, div=div, id=plot.ref['id'])

    class JS:

        @event.prop
        def plot(self, plot=None):
            return plot

        @event.connect('_plot_components')
        def __set_plot_components(self, *events):
            ev = events[-1]
            self.node.innerHTML = ev.div
            el = window.document.createElement('script')
            el.innerHTML = ev.script
            self.node.appendChild(el)

            def getplot():
                # Get plot from id in next event-loop iter
                self.plot = Bokeh.index[ev.id]
                canvas = self.plot.plot_canvas_view
                canvas.reset_dimensions()

            window.setTimeout(getplot, 100)

        @event.connect('size')
        def __resize_plot(self, *events):
            if self.plot and self.parent:
                    self.plot.resize()