mdolab
diff --git a/‎.github/workflows/niceplots.yml
+44-8 b/‎.github/workflows/niceplots.yml
+44-8
diff --git a/‎.gitignore
+8-4 b/‎.gitignore
+8-4
diff --git a/‎.readthedocs.yaml
+11-1 b/‎.readthedocs.yaml
+11-1
diff --git a/‎README.md
+40-19 b/‎README.md
+40-19
diff --git a/‎doc/conf.py
+43 b/‎doc/conf.py
+43
diff --git a/‎doc/index.rst
+1-1 b/‎doc/index.rst
+1-1
diff --git a/‎doc/requirements.txt
+1-1 b/‎doc/requirements.txt
+1-1
diff --git a/‎examples/ImageComparisonTest.sh
+48 b/‎examples/ImageComparisonTest.sh
+48
diff --git a/‎examples/ParulaContours.png
-1010 KB b/‎examples/ParulaContours.png
-1010 KB
diff --git a/‎examples/ParulaContours_dark.png
-1010 KB b/‎examples/ParulaContours_dark.png
-1010 KB
diff --git a/‎examples/ParulaContours_grey.png
-958 KB b/‎examples/ParulaContours_grey.png
-958 KB
diff --git a/‎examples/ParulaContours_navy.png
-979 KB b/‎examples/ParulaContours_navy.png
-979 KB
diff --git a/‎examples/bar_chart.png
-385 KB b/‎examples/bar_chart.png
-385 KB
diff --git a/‎examples/coloredLine.png
-362 KB b/‎examples/coloredLine.png
-362 KB
diff --git a/‎examples/coloredLine_dark.png
-299 KB b/‎examples/coloredLine_dark.png
-299 KB
diff --git a/‎examples/defaultPulseResponse.png
-685 KB b/‎examples/defaultPulseResponse.png
-685 KB
diff --git a/‎examples/niceplotsPulseResponse.png
-846 KB b/‎examples/niceplotsPulseResponse.png
-846 KB
diff --git a/‎examples/optProb-hashed.png
-1.82 MB b/‎examples/optProb-hashed.png
-1.82 MB
diff --git a/‎examples/optProb-shaded.png
-1.55 MB b/‎examples/optProb-shaded.png
-1.55 MB
diff --git a/‎examples/opt_stacks.png
-308 KB b/‎examples/opt_stacks.png
-308 KB
diff --git a/‎examples/opt_stacks_more_data.png
-637 KB b/‎examples/opt_stacks_more_data.png
-637 KB
diff --git a/‎examples/plot_bar_chart.py
+14-32 b/‎examples/plot_bar_chart.py
+14-32
@@ -8,7 +8,6 @@ on:
     tags:
       - v*.*.*
   pull_request:
-    branches: [main]
 
 jobs:
   black:
@@ -21,27 +20,64 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.7, 3.8, 3.9]
-        numpy-version: ["1.16.6", "1.18.5"]
-        mpl-version: ["3.2.*", "3.3.*", "3.4.*"]
+        python-version: ["3.8", "3.9", "3.10"]
+        numpy-version: ["1.19.*", "1.21.*", "1.24.*"]
+        mpl-version: ["3.4.*", "3.6.*"]
+        exclude:
+          - python-version: "3.9"
+            numpy-version: "1.19.*"
+          - python-version: "3.10"
+            numpy-version: "1.19.*"
     steps:
-      - uses: actions/checkout@v2
+      - uses: actions/checkout@v3
       - name: Set up Python ${{ matrix.python-version }}
         uses: actions/setup-python@v2
         with:
           python-version: ${{ matrix.python-version }}
-      - name: Install dependencies
+      - name: Install apt dependencies
+        uses: awalsh128/cache-apt-pkgs-action@latest
+        with:
+          packages: fonts-cmu nodejs npm
+          version: 1.0
+      - name: Install node dependencies
+        run: |
+          sudo npm install -g odiff-bin
+      - name: Install Prompt font
+        run: |
+          mkdir Prompt
+          cd Prompt
+          wget "https://fonts.google.com/download?family=Prompt" -O Prompt.zip
+          unzip Prompt.zip
+          mkdir ~/.fonts
+          cp *.ttf ~/.fonts
+      - name: Install python dependencies
         run: |
-          sudo apt-get install fonts-cmu
           pip install --upgrade pip wheel
           pip install numpy==${{ matrix.numpy-version }}
           pip install matplotlib==${{ matrix.mpl-version }}
           pip install .
       - name: Test examples
         run: |
-          # this is very janky for now
           cd examples
           bash testExamples.sh
+      - name: Compare against reference images
+        if: ${{ success() && matrix.python-version == '3.10' && matrix.numpy-version == '1.24.*' && matrix.mpl-version == '3.6.*' }}
+        run: |
+          cd examples
+          bash ImageComparisonTest.sh
+      - name: Upload examples if failed
+        uses: actions/upload-artifact@v3
+        if: ${{ failure() && matrix.python-version == '3.10' && matrix.numpy-version == '1.24.*' && matrix.mpl-version == '3.6.*' }}
+        with:
+          name: Examples
+          path: examples/
+      # - name: Upload new reference images
+      #   if: ${{ github.event_name == 'push' && success() && matrix.python-version == '3.10' && matrix.numpy-version == '1.24.*' && matrix.mpl-version == '3.6.*' }}
+      #   uses: stefanzweifel/git-auto-commit-action@v4
+      #   with:
+      #     file_pattern: 'examples/ref/*.png'
+      #     commit_message: Update reference images
+
 
   # --- publish to PyPI
   pypi:
 
@@ -1,11 +1,15 @@
-# Ignore any pdf's or png's except in the examples folder
+# Ignore any figures, except the reference images used for testing
 *.pdf
-!examples/*.pdf
 *.png
-!examples/*.png
+!examples/ref/*.png
+*.svg
 
 .vscode/
 
+# Ignore generated doc files
+doc/_build/
+doc/auto_examples/
+
 ### Python ###
 # Byte-compiled / optimized / DLL files
 __pycache__/
@@ -152,4 +156,4 @@ dmypy.json
 !.vscode/tasks.json
 !.vscode/launch.json
 !.vscode/extensions.json
-*.code-workspace
+*.code-workspace
@@ -9,9 +9,19 @@ version: 2
 build:
   os: ubuntu-20.04
   tools:
-    python: "3.9"
+    python: "3.10"
   apt_packages:
     - fonts-cmu
+  # Install the fonts for the james style
+  jobs:
+    pre_install:
+      - mkdir Prompt
+      - cd Prompt
+      - wget "https://fonts.google.com/download?family=Prompt" -O Prompt.zip
+      - unzip Prompt.zip
+      - mkdir ~/.fonts
+      - cp *.ttf ~/.fonts
+      - cd ..
 
 # Build documentation in the docs/ directory with Sphinx
 sphinx:
 
@@ -7,53 +7,74 @@
 [![PyPI - Downloads](https://img.shields.io/pypi/dm/niceplots)](https://pypi.org/project/niceplots/)
 [![Code style: black](https://img.shields.io/badge/code%20style-black-000000.svg)](https://github.com/psf/black)
 
-<img src="https://raw.githubusercontent.com/mdolab/niceplots/main/examples/ParulaContours.png" width="62.22%" /> <img src="https://raw.githubusercontent.com/mdolab/niceplots/main/examples/optProb-shaded.png" width="35%" />
-<img src="https://raw.githubusercontent.com/mdolab/niceplots/main/examples/bar_chart.png" width="47.65%" /> <img src="https://raw.githubusercontent.com/mdolab/niceplots/main/examples/niceplotsPulseResponse.png" width="49.5%" />
+<img src="https://mdolab-niceplots.readthedocs-hosted.com/en/latest/_images/sphx_glr_plot_parula_contours_001.svg" width="61%" /> <img src="https://mdolab-niceplots.readthedocs-hosted.com/en/latest/_images/sphx_glr_plot_opt_prob_002.svg" width="38.2%" />
+<img src="https://mdolab-niceplots.readthedocs-hosted.com/en/latest/_images/sphx_glr_plot_bar_chart_001.svg" width="61%" /> <img src="https://mdolab-niceplots.readthedocs-hosted.com/en/latest/_images/sphx_glr_plot_style_demo_004.svg" width="38.2%" />
 
+<!-- https://mdolab-niceplots.readthedocs-hosted.com/en/latest/_images/ -->
 
-### How do I install?
+## How do I install?
 
-Niceplots can be pip installed directly from PyPI
+NicePlots can be pip installed directly from PyPI
 
 ```shell
 pip install niceplots
 ```
 
-#### If you want to make changes
+### If you want to make changes
 
 * Clone this repository, then enter the folder in the command line terminal.
 * Enter `pip install -e .` within the `niceplots` folder.
 
-#### Font installation (optional)
+### Font installation (optional)
 
-Niceplots will try and use the [computer modern bright](https://tug.org/FontCatalogue/computermodernbright/) font for the best looking plots so be sure to install it as a system font if you want to recreate the style of the plots above.
-Otherwise, niceplots will still work but revert back to the matplotlib default sans-serif font, DejaVu Sans.
+NicePlots styles use fonts that do not ship with most operating systems, so you'll need to install them separately.
+If they are not installed, matplotlib will revert back to its default sans-serif font, DejaVu Sans.
+
+The font used by each style is as follows:
+- doumont-light (default niceplots): CMU Bright
+- doumont-dark: CMU Bright
+- james-dark: Prompt
+- james-light: Prompt
+
+Install the fonts on your system and then delete Matplotlib's font cache, which is located in `~/.cache/matplotlib` by default on most operating systems.
+Matplotlib will rebuild the font cache next time it is run and (hopefully) find the new fonts.
+
+#### CMU Bright (doumont-light and doumont-dark)
+
+The computer modern bright font can be downloaded from [this link](https://tug.org/FontCatalogue/computermodernbright/).
+Alternatively, on Ubuntu, the font can be installed with the following commands:
 
-To install the font on Ubuntu, run the following commands:
 ```
 sudo apt-get update
 sudo apt-get install fonts-cmu
 ```
 Arch linux users can get the font by installing the `otf-cm-unicode` package from AUR.
 
-If niceplots doesn't recognize the font, it might be necessary to delete Matplotlib's font cache file from its location on your computer, likely in `~/.cache/matplotlib`
+#### Prompt (james-dark and james-light)
+
+The Prompt font can be download from [Google Fonts](https://fonts.google.com/specimen/Prompt).
+
+## How do I get set up?
+
+* `import matplotlib.pyplot as plt` and `import niceplots` at the top of a file where you would like to use any function defined in this package.
+* Use `plt.style.use(niceplots.get_style())` to set some defaults for nice-looking plots. You can also try passing different styles to `get_style()`, such as NicePlots' `"james-dark"` or any of matplotlib's styles (see the function's documentation for a full list of available NicePlots styles).
+* Take advantage of NicePlots' helper functions, including (but not limited to) `adjust_spines`, `horiz_bar`, and `plot_nested_pie`, which are all documented in the [examples gallery](https://mdolab-niceplots.readthedocs-hosted.com/en/latest/auto_examples/index.html).
+* Admire your beautiful data.
 
-### How do I get set up?
+## Do you have docs?
 
-* Use `import niceplots` at the top of a file where you would like to use any function defined in this package.
-* Use `niceplots.setRCParams()` to set some matplotlib defaults for nice looking plots. Set `dark_mode=True` and `set_background_color=True` to make plots with a dark background.
-* Use `niceplots.All()` after all the plot commands to apply the niceplot standards on the figure.
-* To use the Matlab colormap "parula", execute `from niceplots import parula` then use `parula.parula_map` as your colormap within your plotting script. See the contour plot example code for an example of this.
+Sort of, you can find our examples gallery and API documentation [here](https://mdolab-niceplots.readthedocs-hosted.com/en/latest)
 
-### Do you have docs?
+## Help, my old NicePlots code doesn't work anymore!
 
-Sort of, you can find our examples gallery and api documentation [here](https://mdolab-niceplots.readthedocs-hosted.com/en/latest)
+We made a couple of changes to the API in version 2.0.0, most of them can be fixed with a simple find and replace.
+Check the [release notes](https://github.com/mdolab/niceplots/releases/tag/v2.0.0) for more details.
 
-### Contribution guidelines
+## Contribution guidelines
 
 * Make any changes you see fit. Please fork your own version and submit a pull request.
 
-### Who do I talk to?
+## Who do I talk to?
 
 * Alasdair Gray, alachris@umich.edu
 * Eytan Adler, eytana@umich.edu
 
@@ -1,4 +1,46 @@
 from sphinx_mdolab_theme.config import *
+from glob import glob
+import shutil
+import os
+from sphinx_gallery.scrapers import figure_rst
+from pathlib import PurePosixPath
+
+
+class svgScraper(object):
+    """This is a custom scraper for sphinx-gallery that allows us to use the svg files written by our examples. It is
+    almost entirely copied from the PNGScraper shown at:
+    https://sphinx-gallery.github.io/dev/advanced.html#example-2-detecting-image-files-on-disk
+    """
+
+    def __init__(self):
+        self.seen = set()
+
+    def __repr__(self):
+        return "svgScraper"
+
+    def __call__(self, block, block_vars, gallery_conf):
+        # Find all svg files in the directory of this example.
+        path_current_example = os.path.dirname(block_vars["src_file"])
+        svgs = sorted(glob(os.path.join(path_current_example, "*.svg")))
+
+        # Get the name of the current example, e.g if the file is called "plot_nested_pie_chart.py",
+        # then example_name = "nested_pie_chart"
+        example_name = os.path.splitext(os.path.basename(block_vars["src_file"]))[0].split("plot_")[0]
+
+        # Iterate through svgs, copy them to the sphinx-gallery output directory
+        image_names = list()
+        image_path_iterator = block_vars["image_path_iterator"]
+        for svg in svgs:
+            if svg not in self.seen and example_name.lower() in svg.lower():
+                self.seen |= set(svg)
+                this_image_path = image_path_iterator.next()
+                image_path = PurePosixPath(this_image_path)
+                image_path = image_path.with_suffix(".svg")
+                image_names.append(image_path)
+                shutil.move(svg, image_path)
+        # Use the `figure_rst` helper function to generate rST for image files
+        return figure_rst(image_names, gallery_conf["src_dir"])
+
 
 # -- Project information -----------------------------------------------------
 project = "niceplots"
@@ -8,4 +50,5 @@
 sphinx_gallery_conf = {
     "examples_dirs": "../examples",  # path to your example scripts
     "gallery_dirs": "auto_examples",  # path to where to save gallery generated output
+    "image_scrapers": (svgScraper(),),
 }
@@ -3,7 +3,7 @@
    You can adapt this file completely to your liking, but it should at least
    contain the root `toctree` directive.
 
-Welcome to niceplots's documentation!
+Welcome to NicePlots's documentation!
 =====================================
 
 
 
@@ -1,3 +1,3 @@
 sphinx_mdolab_theme
 sphinx-gallery
-matplotlib>=3.4
+matplotlib>=3.6
@@ -0,0 +1,48 @@
+#!/bin/bash
+
+# This script compares the images generated by the example scripts to a set of reference images
+
+test_passed=true # Kepp track of whether any tests fail
+
+# First check that there aren't any images produced by the examples that don't have a reference image
+for f in *.png
+do
+    if ! test -f "ref/$f"; then
+        echo ""
+        echo "$f doesn't have a reference image to compare to"
+        test_passed=false
+    fi
+done
+
+mkdir -p diffs
+
+# For each image in the ref directory, check that there is a corresponding image in the current directory and then compare them
+for f in ref/*.png
+do
+    echo ""
+    base_name=$(basename ${f})
+
+    if test -f "$base_name"; then
+        filename_no_ext="${base_name%.*}"
+        echo "Comparing $base_name and $f"
+        odiff --antialiasing --threshold=0.1 --diff-mask $base_name $f diffs/${filename_no_ext}-diff.png
+        if [ $? -ne 0 ]; then
+            test_passed=false
+        fi
+    else
+        echo "Couldn't find image to compare against $f"
+        test_passed=false
+    fi
+done
+
+echo ""
+echo "==========================="
+if [ "$test_passed" = true ] ; then
+    echo "All comparisons passed"
+    echo "==========================="
+    exit 0
+else
+    echo "Some comparisons failed"
+    echo "==========================="
+    exit 0 # In future (if we can make the image comparison more robust) we should exit with a non-zero exit code (e.g. 1)
+fi
@@ -3,41 +3,23 @@
 =========
 An example of a bar chart.
 """
-import random
 import matplotlib.pyplot as plt
 import niceplots
 
-niceplots.setRCParams()
 
-try:
-    # use random words for the example
-    word_file = "/usr/share/dict/words"
-    words = open(word_file).read().splitlines()
-    wl = len(words)
-    random.seed(100)
+header = ["Method", "Time (sec)"]
+labels = [
+    "Analytic Forward",
+    "Analytic Adjoint",
+    "FD Forward Diff.",
+    "FD Central Diff.",
+    "FD Backward Diff.",
+]
+times = [0.00456, 0.00847, 0.0110, 0.0213, 0.011]
+nd = 4
 
-    def rw():
-        return random.choice(words)
+with plt.style.context(niceplots.get_style()):
+    niceplots.horiz_bar(labels, times, header, nd=nd, size=[7, 0.65])
 
-    header = [rw(), rw()]
-    n = 15  # number of bars to create
-    labels = [rw() for i in range(n)]
-    times = [random.random() * random.randint(0, 1000) for i in range(n)]
-    nd = 1
-
-except FileNotFoundError:  # noqa: E722 if user is not on a *nix system
-    header = ["Method", "Time (sec)"]
-    labels = [
-        "Analytic Forward",
-        "Analytic Adjoint",
-        "FD Forward Diff.",
-        "FD Central Diff.",
-        "FD Backward Diff.",
-    ]
-    times = [0.00456, 0.00847, 0.0110, 0.0213, 0.011]
-    nd = 4
-
-niceplots.horiz_bar(labels, times, header, nd=nd, size=[7, 0.65])
-
-plt.savefig("bar_chart.pdf", bbox_inches="tight")
-plt.savefig("bar_chart.png", dpi=400, bbox_inches="tight")
+    plt.savefig("bar_chart.png")
+    plt.savefig("bar_chart.svg")