##// END OF EJS Templates
Merge pull request #892 from minrk/show...
Fernando Perez -
r5123:d2b3a0c6 merge
parent child Browse files
Show More
@@ -17,7 +17,7 b' from matplotlib._pylab_helpers import Gcf'
17 from IPython.config.configurable import SingletonConfigurable
17 from IPython.config.configurable import SingletonConfigurable
18 from IPython.core.displaypub import publish_display_data
18 from IPython.core.displaypub import publish_display_data
19 from IPython.lib.pylabtools import print_figure, select_figure_format
19 from IPython.lib.pylabtools import print_figure, select_figure_format
20 from IPython.utils.traitlets import Dict, Instance, CaselessStrEnum
20 from IPython.utils.traitlets import Dict, Instance, CaselessStrEnum, CBool
21 #-----------------------------------------------------------------------------
21 #-----------------------------------------------------------------------------
22 # Configurable for inline backend options
22 # Configurable for inline backend options
23 #-----------------------------------------------------------------------------
23 #-----------------------------------------------------------------------------
@@ -47,6 +47,23 b' class InlineBackendConfig(SingletonConfigurable):'
47 return
47 return
48 else:
48 else:
49 select_figure_format(self.shell, new)
49 select_figure_format(self.shell, new)
50
51 close_figures = CBool(True, config=True,
52 help="""Close all figures at the end of each cell.
53
54 When True, ensures that each cell starts with no active figures, but it
55 also means that one must keep track of references in order to edit or
56 redraw figures in subsequent cells. This mode is ideal for the notebook,
57 where residual plots from other cells might be surprising.
58
59 When False, one must call figure() to create new figures. This means
60 that gcf() and getfigs() can reference figures created in other cells,
61 and the active figure can continue to be edited with pylab/pyplot
62 methods that reference the current active figure. This mode facilitates
63 iterative editing of figures, and behaves most consistently with
64 other matplotlib backends, but figure barriers between cells must
65 be explicit.
66 """)
50
67
51 shell = Instance('IPython.core.interactiveshell.InteractiveShellABC')
68 shell = Instance('IPython.core.interactiveshell.InteractiveShellABC')
52
69
@@ -55,44 +72,74 b' class InlineBackendConfig(SingletonConfigurable):'
55 # Functions
72 # Functions
56 #-----------------------------------------------------------------------------
73 #-----------------------------------------------------------------------------
57
74
58 def show(close=True):
75 def show(close=None):
59 """Show all figures as SVG payloads sent to the IPython clients.
76 """Show all figures as SVG/PNG payloads sent to the IPython clients.
60
77
61 Parameters
78 Parameters
62 ----------
79 ----------
63 close : bool, optional
80 close : bool, optional
64 If true, a ``plt.close('all')`` call is automatically issued after
81 If true, a ``plt.close('all')`` call is automatically issued after
65 sending all the SVG figures. If this is set, the figures will entirely
82 sending all the figures. If this is set, the figures will entirely
66 removed from the internal list of figures.
83 removed from the internal list of figures.
67 """
84 """
85 if close is None:
86 close = InlineBackendConfig.instance().close_figures
68 for figure_manager in Gcf.get_all_fig_managers():
87 for figure_manager in Gcf.get_all_fig_managers():
69 send_figure(figure_manager.canvas.figure)
88 send_figure(figure_manager.canvas.figure)
70 if close:
89 if close:
71 matplotlib.pyplot.close('all')
90 matplotlib.pyplot.close('all')
91 show._to_draw = []
92
72
93
73
94
74 # This flag will be reset by draw_if_interactive when called
95 # This flag will be reset by draw_if_interactive when called
75 show._draw_called = False
96 show._draw_called = False
97 # list of figures to draw when flush_figures is called
98 show._to_draw = []
76
99
77
100
78 def draw_if_interactive():
101 def draw_if_interactive():
79 """
102 """
80 Is called after every pylab drawing command
103 Is called after every pylab drawing command
81 """
104 """
82 # We simply flag we were called and otherwise do nothing. At the end of
105 # signal that the current active figure should be sent at the end of execution.
83 # the code execution, a separate call to show_close() will act upon this.
106 # Also sets the _draw_called flag, signaling that there will be something to send.
107 # At the end of the code execution, a separate call to flush_figures()
108 # will act upon these values
109
110 fig = Gcf.get_active().canvas.figure
111
112 # ensure current figure will be drawn, and each subsequent call
113 # of draw_if_interactive() moves the active figure to ensure it is
114 # drawn last
115 try:
116 show._to_draw.remove(fig)
117 except ValueError:
118 # ensure it only appears in the draw list once
119 pass
120 show._to_draw.append(fig)
84 show._draw_called = True
121 show._draw_called = True
85
122
86
87 def flush_figures():
123 def flush_figures():
88 """Call show, close all open figures, sending all figure images.
124 """Send all figures that changed
89
125
90 This is meant to be called automatically and will call show() if, during
126 This is meant to be called automatically and will call show() if, during
91 prior code execution, there had been any calls to draw_if_interactive.
127 prior code execution, there had been any calls to draw_if_interactive.
92 """
128 """
93 if show._draw_called:
129 if not show._draw_called:
94 show()
130 return
95 show._draw_called = False
131
132 if InlineBackendConfig.instance().close_figures:
133 # ignore the tracking, just draw and close all figures
134 return show(True)
135
136 # exclude any figures that were closed:
137 active = set([fm.canvas.figure for fm in Gcf.get_all_fig_managers()])
138 for fig in [ fig for fig in show._to_draw if fig in active ]:
139 send_figure(fig)
140 # clear flags for next round
141 show._to_draw = []
142 show._draw_called = False
96
143
97
144
98 def send_figure(fig):
145 def send_figure(fig):
General Comments 0
You need to be logged in to leave comments. Login now