Upload figures

.github/workflows/examples.yml

@@ -59,6 +59,9 @@ jobs:
           EXAMPLE_DIR=$(dirname "${EXAMPLE}")
           EXAMPLE_FILE=$(basename "${EXAMPLE}")
 
+          mkdir -p figs
+          python rhodium/test/plot_to_savefig.py "${EXAMPLE}" "$(realpath figs/)"
+
           pushd "${EXAMPLE_DIR}"
 
           if [ -f "Makefile" ]; then
@@ -67,3 +70,9 @@ jobs:
 
           LD_LIBRARY_PATH="$(pwd):${LD_LIBRARY_PATH}"
           python "${EXAMPLE_FILE}"
+
+      - name: Upload figures
+        uses: actions/upload-artifact@v4
+        with:
+          name: figures
+          path: figs/

rhodium/test/plot_to_savefig.py

@@ -0,0 +1,30 @@
+import sys
+import re
+import pathlib
+
+if __name__ == "__main__":    
+    if len(sys.argv) != 3:
+        print("Usage: " + __file__ + " [file] [fig_path]")
+        sys.exit(-1)
+
+    fig_index = 1
+    fig_basename = pathlib.Path(sys.argv[2]) / pathlib.Path(__file__).stem
+    pyplot_alias = "matplotlib.pyplot"
+    result = []
+
+    with open(sys.argv[1], "r") as f:
+        for line in f:
+            m = re.match(r"import matplotlib.pyplot as ([a-zA-Z_]+)", line)
+            if m:
+                pyplot_alias = m.group(1)
+
+            m = re.match(pyplot_alias + r".show\(\)", line)
+            if m:
+                line = line[:m.start()] + pyplot_alias + ".savefig('" + str(fig_basename) + "." + str(fig_index) + ".png')" + line[m.end():]
+                fig_index += 1
+
+            result.append(line)
+
+    with open(sys.argv[1], "w") as f:
+        for line in result:
+            f.write(line)