Provide xkcd as a context manager

Author: mdboom

examples/showcase/xkcd.py

@@ -1,46 +1,45 @@
-from matplotlib import pyplot as plt
+import matplotlib.pyplot as plt
 import numpy as np
 
-plt.xkcd()
-
-# Based on "Stove Ownership" from XKCD by Randall Monroe
-# http://xkcd.com/418/
-
-fig = plt.figure()
-ax = fig.add_subplot(1, 1, 1)
-ax.spines['right'].set_color('none')
-ax.spines['top'].set_color('none')
-plt.xticks([])
-plt.yticks([])
-ax.set_ylim([-30, 10])
-
-data = np.ones(100)
-data[70:] -= np.arange(30)
-
-plt.annotate(
-    'THE DAY I REALIZED\nI COULD COOK BACON\nWHENEVER I WANTED',
-    xy=(70, 1), arrowprops=dict(arrowstyle='->'), xytext=(15, -10))
-
-plt.plot(data)
-
-plt.xlabel('time')
-plt.ylabel('my overall health')
-
-# Based on "The Data So Far" from XKCD by Randall Monroe
-# http://xkcd.com/373/
-
-fig = plt.figure()
-ax = fig.add_subplot(1, 1, 1)
-ax.bar([-0.125, 1.0-0.125], [0, 100], 0.25)
-ax.spines['right'].set_color('none')
-ax.spines['top'].set_color('none')
-ax.xaxis.set_ticks_position('bottom')
-ax.set_xticks([0, 1])
-ax.set_xlim([-0.5, 1.5])
-ax.set_ylim([0, 110])
-ax.set_xticklabels(['CONFIRMED BY\nEXPERIMENT', 'REFUTED BY\nEXPERIMENT'])
-plt.yticks([])
-
-plt.title("CLAIMS OF SUPERNATURAL POWERS")
+with plt.xkcd():
+    # Based on "Stove Ownership" from XKCD by Randall Monroe
+    # http://xkcd.com/418/
+
+    fig = plt.figure()
+    ax = fig.add_subplot(1, 1, 1)
+    ax.spines['right'].set_color('none')
+    ax.spines['top'].set_color('none')
+    plt.xticks([])
+    plt.yticks([])
+    ax.set_ylim([-30, 10])
+
+    data = np.ones(100)
+    data[70:] -= np.arange(30)
+
+    plt.annotate(
+        'THE DAY I REALIZED\nI COULD COOK BACON\nWHENEVER I WANTED',
+        xy=(70, 1), arrowprops=dict(arrowstyle='->'), xytext=(15, -10))
+
+    plt.plot(data)
+
+    plt.xlabel('time')
+    plt.ylabel('my overall health')
+
+    # Based on "The Data So Far" from XKCD by Randall Monroe
+    # http://xkcd.com/373/
+
+    fig = plt.figure()
+    ax = fig.add_subplot(1, 1, 1)
+    ax.bar([-0.125, 1.0-0.125], [0, 100], 0.25)
+    ax.spines['right'].set_color('none')
+    ax.spines['top'].set_color('none')
+    ax.xaxis.set_ticks_position('bottom')
+    ax.set_xticks([0, 1])
+    ax.set_xlim([-0.5, 1.5])
+    ax.set_ylim([0, 110])
+    ax.set_xticklabels(['CONFIRMED BY\nEXPERIMENT', 'REFUTED BY\nEXPERIMENT'])
+    plt.yticks([])
+
+    plt.title("CLAIMS OF SUPERNATURAL POWERS")
 
 plt.show()

lib/matplotlib/__init__.py

@@ -1050,12 +1050,15 @@ class rc_context(object):
     def __init__(self, rc=None, fname=None):
         self.rcdict = rc
         self.fname = fname
-    def __enter__(self):
         self._rcparams = rcParams.copy()
         if self.fname:
             rc_file(self.fname)
         if self.rcdict:
             rcParams.update(self.rcdict)
+
+    def __enter__(self):
+        return self
+
     def __exit__(self, type, value, tb):
         rcParams.update(self._rcparams)
 

lib/matplotlib/pyplot.py

@@ -255,37 +255,53 @@ def setp(*args, **kwargs):
     draw_if_interactive()
     return ret
 
+
 def xkcd():
     """
-    Turns on `xkcd <xkcd.com>`_ sketch-style drawing mode.  This will
-    only have effect on things drawn after this function is called.
+    Turns on `xkcd <http://xkcd.com/>`_ sketch-style drawing mode.
+    This will only have effect on things drawn after this function is
+    called.
 
     For best results, the "Humor Sans" font should be installed: it is
     not included with matplotlib.
 
-    This function works by setting a whole slew of rcParams, so it will
-    probably override others you have set before.
+    This function works by a number of rcParams, so it will probably
+    override others you have set before.
+
+    If you want the effects of this function to be temporary, it can
+    be used as a context manager, for example::
+
+        with plt.xkcd():
+            # This figure will be in XKCD-style
+            fig1 = plt.figure()
+            # ...
+
+        # This figure will be in regular style
+        fig2 = plt.figure()
     """
     from matplotlib import patheffects
-    rcParams['text.usetex'] = False
-    rcParams['font.family'] = 'fantasy'
-    rcParams['font.fantasy'] = ['Humor Sans', 'Comic Sans MS']
-    rcParams['font.size'] = 14.0
-    rcParams['path.sketch'] = (1, 100, 2)
-    rcParams['path.effects'] = [
-        patheffects.withStroke(linewidth=4, foreground="w")]
-    rcParams['axes.linewidth'] = 1.5
-    rcParams['lines.linewidth'] = 2.0
-    rcParams['figure.facecolor'] = 'white'
-    rcParams['grid.linewidth'] = 0.0
-    rcParams['axes.unicode_minus'] = False
-    rcParams['axes.color_cycle'] = ['b', 'r', 'c', 'm']
-    # rcParams['axes.clip'] = False
-    rcParams['xtick.major.size'] = 8
-    rcParams['xtick.major.width'] = 3
-    rcParams['ytick.major.size'] = 8
-    rcParams['ytick.major.width'] = 3
-
+    context = rc_context()
+    try:
+        rcParams['text.usetex'] = False
+        rcParams['font.family'] = 'Humor Sans'
+        rcParams['font.size'] = 14.0
+        rcParams['path.sketch'] = (1, 100, 2)
+        rcParams['path.effects'] = [
+            patheffects.withStroke(linewidth=4, foreground="w")]
+        rcParams['axes.linewidth'] = 1.5
+        rcParams['lines.linewidth'] = 2.0
+        rcParams['figure.facecolor'] = 'white'
+        rcParams['grid.linewidth'] = 0.0
+        rcParams['axes.unicode_minus'] = False
+        rcParams['axes.color_cycle'] = ['b', 'r', 'c', 'm']
+        rcParams['xtick.major.size'] = 8
+        rcParams['xtick.major.width'] = 3
+        rcParams['ytick.major.size'] = 8
+        rcParams['ytick.major.width'] = 3
+    except:
+        context.__exit__(*sys.exc_info())
+        raise
+    return context
 
 ## Figures ##