Save figures in a path relative to the script

This allows running them from the `scripts` directory, _or_ the top of
the checkout.

Author: QuLogic

scripts/adjustements.py

@@ -3,12 +3,16 @@
 # Released under the BSD License
 # -----------------------------------------------------------------------------
 
+from pathlib import Path
+
 import numpy as np
 import matplotlib.pyplot as plt
 import matplotlib.patches as mpatches
 from matplotlib.collections import PatchCollection
 
 
+ROOT_DIR = Path(__file__).parent.parent
+
 fig = plt.figure(figsize=(4.25, 4.25 * 95/115))
 ax = fig.add_axes([0, 0, 1, 1], frameon=False, aspect=1,
                   xlim=(0-5, 100+10), ylim=(-10, 80+5), xticks=[], yticks=[])
@@ -125,5 +129,5 @@ def int_arrow(p0, p1):
         rotation="vertical", ha="center", va="center", size="x-small")
 
 
-plt.savefig("../figures/adjustments.pdf")
+fig.savefig(ROOT_DIR / "figures/adjustments.pdf")
 # plt.show()

scripts/advanced-plots.py

@@ -4,10 +4,15 @@
 # -----------------------------------------------------------------------------
 
 # Scripts to generate all the basic plots
+from pathlib import Path
+
 import numpy as np
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 
+
+ROOT_DIR = Path(__file__).parent.parent
+
 fig = plt.figure(figsize=(0.4, 0.4))
 mpl.rcParams['axes.linewidth'] = 0.5
 mpl.rcParams['xtick.major.size'] = 0.0
@@ -23,7 +28,7 @@
 ax.set_xlim(0, 8), ax.set_xticks(np.arange(1, 8))
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/advanced-step.pdf")
+fig.savefig(ROOT_DIR / "figures/advanced-step.pdf")
 ax.clear()
 
 # Violin plot
@@ -39,7 +44,7 @@
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/advanced-violin.pdf")
+fig.savefig(ROOT_DIR / "figures/advanced-violin.pdf")
 ax.clear()
 
 # Boxplot
@@ -61,7 +66,7 @@
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/advanced-boxplot.pdf")
+fig.savefig(ROOT_DIR / "figures/advanced-boxplot.pdf")
 ax.clear()
 
 # Barbs plot
@@ -76,7 +81,7 @@
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/advanced-barbs.pdf")
+fig.savefig(ROOT_DIR / "figures/advanced-barbs.pdf")
 ax.clear()
 
 # Event plot
@@ -90,7 +95,7 @@
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/advanced-event.pdf")
+fig.savefig(ROOT_DIR / "figures/advanced-event.pdf")
 ax.clear()
 
 # Errorbar plot
@@ -104,7 +109,7 @@
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/advanced-errorbar.pdf")
+fig.savefig(ROOT_DIR / "figures/advanced-errorbar.pdf")
 ax.clear()
 
 # Hexbin plot
@@ -119,7 +124,7 @@
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/advanced-hexbin.pdf")
+fig.savefig(ROOT_DIR / "figures/advanced-hexbin.pdf")
 ax.clear()
 
 # Hist plot
@@ -131,7 +136,7 @@
 ax.set_ylim(0, 80), ax.set_yticks(np.arange(1, 80, 10))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/advanced-hist.pdf")
+fig.savefig(ROOT_DIR / "figures/advanced-hist.pdf")
 ax.clear()
 
 # Xcorr plot
@@ -145,5 +150,5 @@
 ax.set_ylim(-.25, .25), ax.set_yticks(np.linspace(-.25, .25, 9))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/advanced-xcorr.pdf")
+fig.savefig(ROOT_DIR / "figures/advanced-xcorr.pdf")
 ax.clear()

scripts/anatomy.py

@@ -3,10 +3,16 @@
 # Author:  Nicolas P. Rougier
 # License: BSD
 # ----------------------------------------------------------------------------
+
+from pathlib import Path
+
 import numpy as np
 import matplotlib.pyplot as plt
 from matplotlib.ticker import AutoMinorLocator, MultipleLocator, FuncFormatter
 
+
+ROOT_DIR = Path(__file__).parent.parent
+
 np.random.seed(123)
 
 X = np.linspace(0.5, 3.5, 100)
@@ -136,5 +142,5 @@ def text(x, y, text):
                             connectionstyle="arc3",
                             color=color))
 
-plt.savefig("../figures/anatomy.pdf")
+fig.savefig(ROOT_DIR / "figures/anatomy.pdf")
 # plt.show()

scripts/annotate.py

@@ -2,11 +2,15 @@
 # Matplotlib cheat sheet
 # Released under the BSD License
 # -----------------------------------------------------------------------------
+from pathlib import Path
+
 import numpy as np
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 
 
+ROOT_DIR = Path(__file__).parent.parent
+
 fig = plt.figure(figsize=(6, 1))
 # ax = plt.subplot(111, frameon=False, aspect=.1)
 # b = 0.0
@@ -23,5 +27,5 @@
 plt.text( 5.5, 0.6, "xy\nxycoords", size=10, va="top", ha="center", color=".5")
 plt.text( .75, 0.6, "xytext\ntextcoords", size=10, va="top", ha="center", color=".5")
 
-plt.savefig("../figures/annotate.pdf")
+fig.savefig(ROOT_DIR / "figures/annotate.pdf")
 # plt.show()

scripts/annotation-arrow-styles.py

@@ -1,6 +1,11 @@
+from pathlib import Path
+
 import matplotlib.pyplot as plt
 import matplotlib.patches as mpatches
 
+
+ROOT_DIR = Path(__file__).parent.parent
+
 styles = mpatches.ArrowStyle.get_styles()
 
 
@@ -29,5 +34,5 @@ def demo_con_style(ax, connectionstyle):
              transform=ax.transAxes,
              family="Source Code Pro", ha="center", va="top")
 
-plt.savefig("../figures/annotation-arrow-styles.pdf")
+fig.savefig(ROOT_DIR / "figures/annotation-arrow-styles.pdf")
 # plt.show()

scripts/annotation-connection-styles.py

@@ -1,6 +1,11 @@
+from pathlib import Path
+
 import matplotlib.pyplot as plt
 
 
+ROOT_DIR = Path(__file__).parent.parent
+
+
 def demo_con_style(ax, connectionstyle):
     x1, y1 = 0.3, 0.2
     x2, y2 = 0.8, 0.6
@@ -33,5 +38,5 @@ def demo_con_style(ax, connectionstyle):
     ax.set(xlim=(0, 1), ylim=(0, 1), xticks=[], yticks=[], aspect=1)
 fig.tight_layout(pad=0.2)
 
-plt.savefig("../figures/annotation-connection-styles.pdf")
+fig.savefig(ROOT_DIR / "figures/annotation-connection-styles.pdf")
 # plt.show()

scripts/basic-plots.py

@@ -4,10 +4,15 @@
 # -----------------------------------------------------------------------------
 
 # Scripts to generate all the basic plots
+from pathlib import Path
+
 import numpy as np
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 
+
+ROOT_DIR = Path(__file__).parent.parent
+
 fig = plt.figure(figsize=(0.4, 0.4))
 mpl.rcParams['axes.linewidth'] = 0.5
 mpl.rcParams['xtick.major.size'] = 0.0
@@ -23,7 +28,7 @@
 ax.set_xlim(0, 8), ax.set_xticks(np.arange(1, 8))
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/basic-plot.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-plot.pdf")
 ax.clear()
 
 # Basic line plot (color)blaPwd
@@ -35,7 +40,7 @@
 ax.set_xlim(0, 8), ax.set_xticks(np.arange(1, 8))
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/basic-plot-color.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-plot-color.pdf")
 ax.clear()
 
 # Basic scatter plot
@@ -48,7 +53,7 @@
 ax.set_xlim(0, 8), ax.set_xticks(np.arange(1, 8))
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/basic-scatter.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-scatter.pdf")
 ax.clear()
 
 # Basic bar plot
@@ -62,7 +67,7 @@
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125)
-plt.savefig("../figures/basic-bar.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-bar.pdf")
 ax.clear()
 
 # Basic imshow plot
@@ -75,7 +80,7 @@
 ax.set_xlim(0, 8), ax.set_xticks(np.arange(1, 8))
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.grid(linewidth=0.25, color="white")
-plt.savefig("../figures/basic-imshow.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-imshow.pdf")
 ax.clear()
 
 # Basic pcolormesh plot
@@ -87,7 +92,7 @@
 plt.pcolormesh(X, Y, Z, cmap='Oranges', shading='auto')
 ax.set_xlim(-3, 3), ax.set_xticks(np.arange(-3, 4))
 ax.set_ylim(-3, 3), ax.set_yticks(np.arange(-3, 4))
-plt.savefig("../figures/basic-pcolormesh.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-pcolormesh.pdf")
 ax.clear()
 
 # Basic contour plot
@@ -100,7 +105,7 @@
             linewidths=0.125, nchunk=10)
 ax.set_xlim(0, 8), ax.set_xticks(np.arange(1, 8))
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
-plt.savefig("../figures/basic-contour.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-contour.pdf")
 ax.clear()
 
 # Basic pie plot
@@ -117,7 +122,7 @@
        wedgeprops={"linewidth": 0.25, "edgecolor": "white"}, frame=True)
 ax.pie(X, colors=colors, radius=3, center=(4, 4),
        wedgeprops={"linewidth": 0.25, "edgecolor": "white"}, frame=True)
-plt.savefig("../figures/basic-pie.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-pie.pdf")
 ax.clear()
 
 # Basic text plot
@@ -128,7 +133,7 @@
 ax.grid(linewidth=0.25, color="0.75")
 ax.text(4, 4, "TEXT", color="C1", size=8, weight="bold",
         ha="center", va="center", rotation=25)
-plt.savefig("../figures/basic-text.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-text.pdf")
 ax.clear()
 
 # Basic fill plot
@@ -143,7 +148,7 @@
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125, color="0.75")
-plt.savefig("../figures/basic-fill.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-fill.pdf")
 ax.clear()
 
 # Basic quiver plot
@@ -158,5 +163,5 @@
 ax.set_ylim(0, 8), ax.set_yticks(np.arange(1, 8))
 ax.set_axisbelow(True)
 ax.grid(linewidth=0.125, color="0.75")
-plt.savefig("../figures/basic-quiver.pdf")
+fig.savefig(ROOT_DIR / "figures/basic-quiver.pdf")
 ax.clear()

scripts/colorbar.py

@@ -2,11 +2,16 @@
 # Matplotlib cheat sheet
 # Released under the BSD License
 # -----------------------------------------------------------------------------
+
+from pathlib import Path
+
 import numpy as np
 import matplotlib as mpl
 import matplotlib.pyplot as plt
 
 
+ROOT_DIR = Path(__file__).parent.parent
+
 fig = plt.figure(figsize=(6, .65))
 # ax = plt.subplot(111, frameon=False, aspect=.1)
 b = 0.025
@@ -19,5 +24,5 @@
 plt.colorbar(sm, cax=ax, ticks=np.linspace(0, 1, 11),
              orientation="horizontal")
 
-plt.savefig("../figures/colorbar.pdf")
+fig.savefig(ROOT_DIR / "figures/colorbar.pdf")
 # plt.show()

scripts/colormaps.py

@@ -1,7 +1,11 @@
+from pathlib import Path
+
 import numpy as np
 import matplotlib.pyplot as plt
 
 
+ROOT_DIR = Path(__file__).parent.parent
+
 figsize = 4.0, 0.25
 fig = plt.figure(figsize=figsize)
 ax = fig.add_axes([0, 0, 1, 1], frameon=False, aspect=1)
@@ -62,5 +66,5 @@
                     family = "Source Pro Serif", size=10, ha="center", va="center")
     """
 
-    plt.savefig("../figures/colormap-%s.pdf" % cmap)
+    fig.savefig(ROOT_DIR / f"figures/colormap-{cmap}.pdf")
     ax.clear()

scripts/colornames.py

@@ -5,12 +5,15 @@
 
 Simple plot example with the named colors and its visual representation.
 """
-from __future__ import division
+
+from pathlib import Path
 
 import matplotlib.pyplot as plt
 from matplotlib import colors as mcolors
 
 
+ROOT_DIR = Path(__file__).parent.parent
+
 colors = dict(mcolors.BASE_COLORS, **mcolors.CSS4_COLORS)
 
 # Sort colors by hue, saturation, value and name.
@@ -53,5 +56,5 @@
                     top=1, bottom=0,
                     hspace=0, wspace=0)
 
-plt.savefig("../figures/colornames.pdf")
+fig.savefig(ROOT_DIR / "figures/colornames.pdf")
 # plt.show()