forked from microsoft/tensorwatch
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
added save functionality for mpl, test array line plot for mpl
- Loading branch information
Showing
4 changed files
with
456 additions
and
437 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,169 +1,173 @@ | ||
| # Copyright (c) Microsoft Corporation. | ||
| # Licensed under the MIT license. | ||
|
|
||
| #from IPython import get_ipython, display | ||
| #if get_ipython(): | ||
| # get_ipython().magic('matplotlib notebook') | ||
|
|
||
| #import matplotlib | ||
| #if os.name == 'posix' and "DISPLAY" not in os.environ: | ||
| # matplotlib.use('Agg') # Must be before importing matplotlib.pyplot or pylab! | ||
|
|
||
| #from ipywidgets.widgets.interaction import show_inline_matplotlib_plots | ||
| #from ipykernel.pylab.backend_inline import flush_figures | ||
|
|
||
| from ..vis_base import VisBase | ||
|
|
||
| import sys, logging | ||
| from abc import abstractmethod | ||
| from .. import utils | ||
|
|
||
|
|
||
| class BaseMplPlot(VisBase): | ||
| def __init__(self, cell:VisBase.widgets.Box=None, title:str=None, show_legend:bool=None, is_3d:bool=False, | ||
| stream_name:str=None, console_debug:bool=False, **vis_args): | ||
| super(BaseMplPlot, self).__init__(VisBase.widgets.Output(), cell, title, show_legend, | ||
| stream_name=stream_name, console_debug=console_debug, **vis_args) | ||
|
|
||
| self._fig_init_done = False | ||
| self.show_legend = show_legend | ||
| self.is_3d = is_3d | ||
| if is_3d: | ||
| # this is needed for some reason | ||
| from mpl_toolkits.mplot3d import Axes3D | ||
| # graph objects | ||
| self.figure = None | ||
| self._ax_main = None | ||
| # matplotlib animation | ||
| self.animation = None | ||
| self.anim_interval = None | ||
| #print(matplotlib.get_backend()) | ||
| #display.display(self.cell) | ||
|
|
||
| # anim_interval in seconds | ||
| def init_fig(self, anim_interval:float=1.0): | ||
| import matplotlib.pyplot as plt # delayed import due to matplotlib threading issue | ||
|
|
||
| """(for derived class) Initializes matplotlib figure""" | ||
| if self._fig_init_done: | ||
| return False | ||
|
|
||
| # create figure and animation | ||
| self.figure = plt.figure(figsize=(8, 3)) | ||
| self.anim_interval = anim_interval | ||
|
|
||
| # default color pallet | ||
| import matplotlib.pyplot as plt # delayed import due to matplotlib threading issue | ||
|
|
||
| plt.set_cmap('Dark2') | ||
| plt.rcParams['image.cmap']='Dark2' | ||
|
|
||
| self._fig_init_done = True | ||
| return True | ||
|
|
||
| def get_main_axis(self): | ||
| # if we don't yet have main axis, create one | ||
| if not self._ax_main: | ||
| # by default assign one subplot to whole graph | ||
| self._ax_main = self.figure.add_subplot(111, | ||
| projection=None if not self.is_3d else '3d') | ||
| self._ax_main.grid(self.is_show_grid()) | ||
| # change the color of the top and right spines to opaque gray | ||
| self._ax_main.spines['right'].set_color((.8,.8,.8)) | ||
| self._ax_main.spines['top'].set_color((.8,.8,.8)) | ||
| if self.title is not None: | ||
| title = self._ax_main.set_title(self.title) | ||
| title.set_weight('bold') | ||
| return self._ax_main | ||
|
|
||
| # overridable | ||
| def is_show_grid(self): | ||
| return True | ||
|
|
||
| def _on_update(self, frame): # pylint: disable=unused-argument | ||
| try: | ||
| self._update_stream_plots() | ||
| except Exception as ex: | ||
| # when exception occurs here, animation will stop and there | ||
| # will be no further plot updates | ||
| # TODO: may be we don't need all of below but none of them | ||
| # are popping up exception in Jupyter Notebook because these | ||
| # exceptions occur in background? | ||
| self.last_ex = ex | ||
| logging.exception('Exception in matplotlib update loop') | ||
|
|
||
|
|
||
| def show(self, blocking=False): | ||
| if not self.is_shown and self.anim_interval: | ||
| from matplotlib.animation import FuncAnimation # function-level import as this one is expensive | ||
| self.animation = FuncAnimation(self.figure, self._on_update, interval=self.anim_interval*1000.0) | ||
| super(BaseMplPlot, self).show(blocking) | ||
|
|
||
| def _post_update_stream_plot(self, stream_vis): | ||
| import matplotlib.pyplot as plt # delayed import due to matplotlib threading issue | ||
|
|
||
| utils.debug_log("Plot updated", stream_vis.stream.stream_name, verbosity=5) | ||
|
|
||
| if self.layout_dirty: | ||
| # do not do tight_layout() call on every update | ||
| # that would jumble up the graphs! it should only called | ||
| # once each time there is change in layout | ||
| self.figure.tight_layout() | ||
| self.layout_dirty = False | ||
|
|
||
| # below forces redraw and it was helpful to | ||
| # repaint even if there was error in interval loop | ||
| # but it does work in native UX and not in Jupyter Notebook | ||
| #self.figure.canvas.draw() | ||
| #self.figure.canvas.flush_events() | ||
|
|
||
| if self._use_hbox and VisBase.get_ipython(): | ||
| self.widget.clear_output(wait=True) | ||
| with self.widget: | ||
| plt.show(self.figure) | ||
|
|
||
| # everything else that doesn't work | ||
| #self.figure.show() | ||
| #display.clear_output(wait=True) | ||
| #display.display(self.figure) | ||
| #flush_figures() | ||
| #plt.show() | ||
| #show_inline_matplotlib_plots() | ||
| #elif not get_ipython(): | ||
| # self.figure.canvas.draw() | ||
|
|
||
| def _post_add_subscription(self, stream_vis, **stream_vis_args): | ||
| import matplotlib.pyplot as plt # delayed import due to matplotlib threading issue | ||
|
|
||
| # make sure figure is initialized | ||
| self.init_fig() | ||
| self.init_stream_plot(stream_vis, **stream_vis_args) | ||
|
|
||
| # redo the legend | ||
| #self.figure.legend(loc='center right', bbox_to_anchor=(1.5, 0.5)) | ||
| if self.show_legend: | ||
| self.figure.legend(loc='lower right') | ||
| plt.subplots_adjust(hspace=0.6) | ||
|
|
||
| def _show_widget_native(self, blocking:bool): | ||
| import matplotlib.pyplot as plt # delayed import due to matplotlib threading issue | ||
|
|
||
| #plt.ion() | ||
| #plt.show() | ||
| return plt.show(block=blocking) | ||
|
|
||
| def _show_widget_notebook(self): | ||
| # no need to return anything because %matplotlib notebook will | ||
| # detect spawning of figure and paint it | ||
| # if self.figure is returned then you will see two of them | ||
| return None | ||
| #plt.show() | ||
| #return self.figure | ||
|
|
||
| def _can_update_stream_plots(self): | ||
| return False # we run interval timer which will flush the key | ||
|
|
||
| @abstractmethod | ||
| def init_stream_plot(self, stream_vis, **stream_vis_args): | ||
| """(for derived class) Create new plot info for this stream""" | ||
| pass | ||
| # Copyright (c) Microsoft Corporation. | ||
| # Licensed under the MIT license. | ||
|
|
||
| #from IPython import get_ipython, display | ||
| #if get_ipython(): | ||
| # get_ipython().magic('matplotlib notebook') | ||
|
|
||
| #import matplotlib | ||
| #if os.name == 'posix' and "DISPLAY" not in os.environ: | ||
| # matplotlib.use('Agg') # Must be before importing matplotlib.pyplot or pylab! | ||
|
|
||
| #from ipywidgets.widgets.interaction import show_inline_matplotlib_plots | ||
| #from ipykernel.pylab.backend_inline import flush_figures | ||
|
|
||
| from ..vis_base import VisBase | ||
|
|
||
| import sys, logging | ||
| from abc import abstractmethod | ||
| from .. import utils | ||
|
|
||
|
|
||
| class BaseMplPlot(VisBase): | ||
| def __init__(self, cell:VisBase.widgets.Box=None, title:str=None, show_legend:bool=None, is_3d:bool=False, | ||
| stream_name:str=None, console_debug:bool=False, **vis_args): | ||
| super(BaseMplPlot, self).__init__(VisBase.widgets.Output(), cell, title, show_legend, | ||
| stream_name=stream_name, console_debug=console_debug, **vis_args) | ||
|
|
||
| self._fig_init_done = False | ||
| self.show_legend = show_legend | ||
| self.is_3d = is_3d | ||
| if is_3d: | ||
| # this is needed for some reason | ||
| from mpl_toolkits.mplot3d import Axes3D | ||
| # graph objects | ||
| self.figure = None | ||
| self._ax_main = None | ||
| # matplotlib animation | ||
| self.animation = None | ||
| self.anim_interval = None | ||
| #print(matplotlib.get_backend()) | ||
| #display.display(self.cell) | ||
|
|
||
| # anim_interval in seconds | ||
| def init_fig(self, anim_interval:float=1.0): | ||
| import matplotlib.pyplot as plt # delayed import due to matplotlib threading issue | ||
|
|
||
| """(for derived class) Initializes matplotlib figure""" | ||
| if self._fig_init_done: | ||
| return False | ||
|
|
||
| # create figure and animation | ||
| self.figure = plt.figure(figsize=(8, 3)) | ||
| self.anim_interval = anim_interval | ||
|
|
||
| # default color pallet | ||
| import matplotlib.pyplot as plt # delayed import due to matplotlib threading issue | ||
|
|
||
| plt.set_cmap('Dark2') | ||
| plt.rcParams['image.cmap']='Dark2' | ||
|
|
||
| self._fig_init_done = True | ||
| return True | ||
|
|
||
| def get_main_axis(self): | ||
| # if we don't yet have main axis, create one | ||
| if not self._ax_main: | ||
| # by default assign one subplot to whole graph | ||
| self._ax_main = self.figure.add_subplot(111, | ||
| projection=None if not self.is_3d else '3d') | ||
| self._ax_main.grid(self.is_show_grid()) | ||
| # change the color of the top and right spines to opaque gray | ||
| self._ax_main.spines['right'].set_color((.8,.8,.8)) | ||
| self._ax_main.spines['top'].set_color((.8,.8,.8)) | ||
| if self.title is not None: | ||
| title = self._ax_main.set_title(self.title) | ||
| title.set_weight('bold') | ||
| return self._ax_main | ||
|
|
||
| # overridable | ||
| def is_show_grid(self): | ||
| return True | ||
|
|
||
| def _on_update(self, frame): # pylint: disable=unused-argument | ||
| try: | ||
| self._update_stream_plots() | ||
| except Exception as ex: | ||
| # when exception occurs here, animation will stop and there | ||
| # will be no further plot updates | ||
| # TODO: may be we don't need all of below but none of them | ||
| # are popping up exception in Jupyter Notebook because these | ||
| # exceptions occur in background? | ||
| self.last_ex = ex | ||
| logging.exception('Exception in matplotlib update loop') | ||
|
|
||
|
|
||
| def show(self, blocking=False): | ||
| if not self.is_shown and self.anim_interval: | ||
| from matplotlib.animation import FuncAnimation # function-level import as this one is expensive | ||
| self.animation = FuncAnimation(self.figure, self._on_update, interval=self.anim_interval*1000.0) | ||
| super(BaseMplPlot, self).show(blocking) | ||
|
|
||
| def _post_update_stream_plot(self, stream_vis): | ||
| import matplotlib.pyplot as plt # delayed import due to matplotlib threading issue | ||
|
|
||
| utils.debug_log("Plot updated", stream_vis.stream.stream_name, verbosity=5) | ||
|
|
||
| if self.layout_dirty: | ||
| # do not do tight_layout() call on every update | ||
| # that would jumble up the graphs! it should only called | ||
| # once each time there is change in layout | ||
| self.figure.tight_layout() | ||
| self.layout_dirty = False | ||
|
|
||
| # below forces redraw and it was helpful to | ||
| # repaint even if there was error in interval loop | ||
| # but it does work in native UX and not in Jupyter Notebook | ||
| #self.figure.canvas.draw() | ||
| #self.figure.canvas.flush_events() | ||
|
|
||
| if self._use_hbox and VisBase.get_ipython(): | ||
| self.widget.clear_output(wait=True) | ||
| with self.widget: | ||
| plt.show(self.figure) | ||
|
|
||
| # everything else that doesn't work | ||
| #self.figure.show() | ||
| #display.clear_output(wait=True) | ||
| #display.display(self.figure) | ||
| #flush_figures() | ||
| #plt.show() | ||
| #show_inline_matplotlib_plots() | ||
| #elif not get_ipython(): | ||
| # self.figure.canvas.draw() | ||
|
|
||
| def _post_add_subscription(self, stream_vis, **stream_vis_args): | ||
| import matplotlib.pyplot as plt # delayed import due to matplotlib threading issue | ||
|
|
||
| # make sure figure is initialized | ||
| self.init_fig() | ||
| self.init_stream_plot(stream_vis, **stream_vis_args) | ||
|
|
||
| # redo the legend | ||
| #self.figure.legend(loc='center right', bbox_to_anchor=(1.5, 0.5)) | ||
| if self.show_legend: | ||
| self.figure.legend(loc='lower right') | ||
| plt.subplots_adjust(hspace=0.6) | ||
|
|
||
| def _show_widget_native(self, blocking:bool): | ||
| import matplotlib.pyplot as plt # delayed import due to matplotlib threading issue | ||
|
|
||
| #plt.ion() | ||
| #plt.show() | ||
| return plt.show(block=blocking) | ||
|
|
||
| def _show_widget_notebook(self): | ||
| # no need to return anything because %matplotlib notebook will | ||
| # detect spawning of figure and paint it | ||
| # if self.figure is returned then you will see two of them | ||
| return None | ||
| #plt.show() | ||
| #return self.figure | ||
|
|
||
| def _can_update_stream_plots(self): | ||
| return False # we run interval timer which will flush the key | ||
|
|
||
| @abstractmethod | ||
| def init_stream_plot(self, stream_vis, **stream_vis_args): | ||
| """(for derived class) Create new plot info for this stream""" | ||
| pass | ||
|
|
||
| def _save_widget(self, filepath:str)->None: | ||
| self._update_stream_plots() | ||
| self.figure.savefig(filepath) |
Oops, something went wrong.