docs: Export zoomed example figures

Signed-off-by: Sietze van Buuren <s.van.buuren@gmail.com>

Author: swvanbuuren

cubinterpp/main.py

@@ -1,5 +1,6 @@
 """ Compare Spline Interpolation types """
 
+from pathlib import Path
 import numpy as np
 from scipy.interpolate import RegularGridInterpolator
 import mlpyqtgraph as mpg
@@ -152,6 +153,9 @@ def main():
         ax.yticks = 8
         ax.zlim = [0, 8]
         mpg.points3(xp, yp, zp, color=(0.8, 0.1, 0.1, 1), size=3)
+        ax.distance = 115
+        filepath = Path(__file__).parents[1] / 'docs' / 'images' / f'{interp_type}_2D.png'
+        ax.export(str(filepath))
 
     x, y, f = get_test_data_2d(case='three_bumps')
     x_fine, y_fine = refine_grid(x, 46), refine_grid(y, 46)

cubinterpp/periodic_spline.py

@@ -1,5 +1,6 @@
 """ Simple test for Periodic Spline Interpolation """
 
+from pathlib import Path
 import numpy as np
 import mlpyqtgraph as mpg
 import cubinterpp.cubinterpp_py as cubinterpp  # cubinterpp_py is a pybind11 module
@@ -25,7 +26,6 @@ def test_data(case='circle'):
 
 @mpg.plotter
 def main():
-
     t, x, y = test_data('arbitrary')
 
     spline_x = cubinterpp.NaturalPeriodicSpline1D(t, x)
@@ -42,7 +42,8 @@ def main():
     mpg.plot(x_interp_scipy, y_interp_scipy, style='--')
     mpg.plot(x, y, width=0, symbol='o', symbol_color='r', symbol_size=6)
     mpg.legend('cubinterpp', 'scipy', 'data points')
-    mpg.gca().grid = True
+    ax = mpg.gca()
+    ax.grid = True
 
 
 if __name__ == '__main__':

cubinterpp/periodic_spline_2d.py

@@ -1,5 +1,6 @@
 """ Periodic spline example in 3D """
 
+from pathlib import Path
 import random
 import numpy as np
 import cubinterpp.cubinterpp_py as cubinterpp  # cubinterpp_py is a pybind11 module
@@ -60,7 +61,9 @@ def main():
     yp = y.flatten()
     zp = z.flatten()
     mpg.points3(xp, yp, zp, color=(0.8, 0.1, 0.1, 1), size=5)
-    ax.export('periodic_spline_2d.png')
+    ax.distance = 115
+    filepath = Path(__file__).parents[1] / 'docs' / 'images' / 'periodic_spline_2D.png'
+    ax.export(str(filepath))
 
 
 if __name__ == '__main__':

docs/images/akima_2D.png

@@ -20,7 +20,7 @@ classifiers = [
 ]
 urls = { Homepage = "https://github.com/swvanbuuren/cubinterpp" }
 dependencies = [
-    "mlpyqtgraph==0.12.0",
+    "mlpyqtgraph==0.13.0",
     "scipy==1.17.0",
 ]