##// 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 17 from IPython.config.configurable import SingletonConfigurable
18 18 from IPython.core.displaypub import publish_display_data
19 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 22 # Configurable for inline backend options
23 23 #-----------------------------------------------------------------------------
@@ -47,6 +47,23 b' class InlineBackendConfig(SingletonConfigurable):'
47 47 return
48 48 else:
49 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 68 shell = Instance('IPython.core.interactiveshell.InteractiveShellABC')
52 69
@@ -55,44 +72,74 b' class InlineBackendConfig(SingletonConfigurable):'
55 72 # Functions
56 73 #-----------------------------------------------------------------------------
57 74
58 def show(close=True):
59 """Show all figures as SVG payloads sent to the IPython clients.
75 def show(close=None):
76 """Show all figures as SVG/PNG payloads sent to the IPython clients.
60 77
61 78 Parameters
62 79 ----------
63 80 close : bool, optional
64 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 83 removed from the internal list of figures.
67 84 """
85 if close is None:
86 close = InlineBackendConfig.instance().close_figures
68 87 for figure_manager in Gcf.get_all_fig_managers():
69 88 send_figure(figure_manager.canvas.figure)
70 89 if close:
71 90 matplotlib.pyplot.close('all')
91 show._to_draw = []
92
72 93
73 94
74 95 # This flag will be reset by draw_if_interactive when called
75 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 101 def draw_if_interactive():
79 102 """
80 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
83 # the code execution, a separate call to show_close() will act upon this.
105 # signal that the current active figure should be sent at the end of execution.
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 121 show._draw_called = True
85 122
86
87 123 def flush_figures():
88 """Call show, close all open figures, sending all figure images.
124 """Send all figures that changed
89 125
90 126 This is meant to be called automatically and will call show() if, during
91 127 prior code execution, there had been any calls to draw_if_interactive.
92 128 """
93 if show._draw_called:
94 show()
95 show._draw_called = False
129 if not show._draw_called:
130 return
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 145 def send_figure(fig):
General Comments 0
You need to be logged in to leave comments. Login now