Try to have working animation example

Author: AntSimi

doc/conf.py

@@ -54,7 +54,7 @@
 sphinx_gallery_conf = {
     "examples_dirs": "../examples",  # path to your example scripts
     "gallery_dirs": "python_module",
-    "capture_repr": ("_repr_html_",),
+    "capture_repr": ("_repr_html_", "__repr__"),
     "backreferences_dir": "gen_modules/backreferences",
     "doc_module": ("py_eddy_tracker",),
     "reference_url": {

examples/14_generic_tools/pet_visvalingam.py

@@ -4,6 +4,7 @@
 """
 import matplotlib.animation as animation
 from matplotlib import pyplot as plt
+from matplotlib import rc
 from numba import njit
 from numpy import array, empty
 
@@ -12,6 +13,8 @@
 from py_eddy_tracker.observations.observation import EddiesObservations
 from py_eddy_tracker.poly import vertice_overlap, visvalingam
 
+rc("animation", html="jshtml")
+
 
 @njit(cache=True)
 def visvalingam_polys(x, y, nb_pt):

notebooks/python_module/14_generic_tools/pet_visvalingam.ipynb

@@ -26,7 +26,7 @@
       },
       "outputs": [],
       "source": [
-        "import matplotlib.animation as animation\nfrom matplotlib import pyplot as plt\nfrom numba import njit\nfrom numpy import array, empty\n\nfrom py_eddy_tracker import data\nfrom py_eddy_tracker.generic import uniform_resample\nfrom py_eddy_tracker.observations.observation import EddiesObservations\nfrom py_eddy_tracker.poly import vertice_overlap, visvalingam\n\n\n@njit(cache=True)\ndef visvalingam_polys(x, y, nb_pt):\n    nb = x.shape[0]\n    x_new = empty((nb, nb_pt), dtype=x.dtype)\n    y_new = empty((nb, nb_pt), dtype=y.dtype)\n    for i in range(nb):\n        x_new[i], y_new[i] = visvalingam(x[i], y[i], nb_pt)\n    return x_new, y_new\n\n\n@njit(cache=True)\ndef uniform_resample_polys(x, y, nb_pt):\n    nb = x.shape[0]\n    x_new = empty((nb, nb_pt), dtype=x.dtype)\n    y_new = empty((nb, nb_pt), dtype=y.dtype)\n    for i in range(nb):\n        x_new[i], y_new[i] = uniform_resample(x[i], y[i], fixed_size=nb_pt)\n    return x_new, y_new\n\n\ndef update_line(num):\n    nb = 50 - num - 20\n    x_v, y_v = visvalingam_polys(a.contour_lon_e, a.contour_lat_e, nb)\n    for i, (x_, y_) in enumerate(zip(x_v, y_v)):\n        lines_v[i].set_data(x_, y_)\n    x_u, y_u = uniform_resample_polys(a.contour_lon_e, a.contour_lat_e, nb)\n    for i, (x_, y_) in enumerate(zip(x_u, y_u)):\n        lines_u[i].set_data(x_, y_)\n    scores_v = vertice_overlap(a.contour_lon_e, a.contour_lat_e, x_v, y_v) * 100.0\n    scores_u = vertice_overlap(a.contour_lon_e, a.contour_lat_e, x_u, y_u) * 100.0\n    for i, (s_v, s_u) in enumerate(zip(scores_v, scores_u)):\n        texts[i].set_text(f\"Score uniform {s_u:.1f} %\\nScore visvalingam {s_v:.1f} %\")\n    title.set_text(f\"{nb} points by contour in place of 50\")\n    return (title, *lines_u, *lines_v, *texts)"
+        "import matplotlib.animation as animation\nfrom matplotlib import pyplot as plt, rc\nfrom numba import njit\nfrom numpy import array, empty\n\nfrom py_eddy_tracker import data\nfrom py_eddy_tracker.generic import uniform_resample\nfrom py_eddy_tracker.observations.observation import EddiesObservations\nfrom py_eddy_tracker.poly import vertice_overlap, visvalingam\n\nrc('animation', html='jshtml')\n\n@njit(cache=True)\ndef visvalingam_polys(x, y, nb_pt):\n    nb = x.shape[0]\n    x_new = empty((nb, nb_pt), dtype=x.dtype)\n    y_new = empty((nb, nb_pt), dtype=y.dtype)\n    for i in range(nb):\n        x_new[i], y_new[i] = visvalingam(x[i], y[i], nb_pt)\n    return x_new, y_new\n\n\n@njit(cache=True)\ndef uniform_resample_polys(x, y, nb_pt):\n    nb = x.shape[0]\n    x_new = empty((nb, nb_pt), dtype=x.dtype)\n    y_new = empty((nb, nb_pt), dtype=y.dtype)\n    for i in range(nb):\n        x_new[i], y_new[i] = uniform_resample(x[i], y[i], fixed_size=nb_pt)\n    return x_new, y_new\n\n\ndef update_line(num):\n    nb = 50 - num - 20\n    x_v, y_v = visvalingam_polys(a.contour_lon_e, a.contour_lat_e, nb)\n    for i, (x_, y_) in enumerate(zip(x_v, y_v)):\n        lines_v[i].set_data(x_, y_)\n    x_u, y_u = uniform_resample_polys(a.contour_lon_e, a.contour_lat_e, nb)\n    for i, (x_, y_) in enumerate(zip(x_u, y_u)):\n        lines_u[i].set_data(x_, y_)\n    scores_v = vertice_overlap(a.contour_lon_e, a.contour_lat_e, x_v, y_v) * 100.0\n    scores_u = vertice_overlap(a.contour_lon_e, a.contour_lat_e, x_u, y_u) * 100.0\n    for i, (s_v, s_u) in enumerate(zip(scores_v, scores_u)):\n        texts[i].set_text(f\"Score uniform {s_u:.1f} %\\nScore visvalingam {s_v:.1f} %\")\n    title.set_text(f\"{nb} points by contour in place of 50\")\n    return (title, *lines_u, *lines_v, *texts)"
       ]
     },
     {
@@ -55,7 +55,7 @@
       },
       "outputs": [],
       "source": [
-        "fig = plt.figure()\naxs = [\n    fig.add_subplot(221),\n    fig.add_subplot(222),\n    fig.add_subplot(223),\n    fig.add_subplot(224),\n]\nlines_u, lines_v, texts, score_text = list(), list(), list(), list()\nfor i, obs in enumerate(a):\n    axs[i].set_aspect(\"equal\")\n    axs[i].grid()\n    axs[i].set_xticklabels([]), axs[i].set_yticklabels([])\n    axs[i].plot(\n        obs[\"contour_lon_e\"], obs[\"contour_lat_e\"], \"r\", lw=6, label=\"Original contour\"\n    )\n    lines_v.append(axs[i].plot([], [], color=\"limegreen\", lw=4, label=\"visvalingam\")[0])\n    lines_u.append(\n        axs[i].plot([], [], color=\"black\", lw=2, label=\"uniform resampling\")[0]\n    )\n    texts.append(axs[i].set_title(\"\", fontsize=8))\naxs[0].legend(fontsize=8)\ntitle = fig.suptitle(\"\")\nani = animation.FuncAnimation(fig, update_line, 27)"
+        "fig = plt.figure()\naxs = [\n    fig.add_subplot(221),\n    fig.add_subplot(222),\n    fig.add_subplot(223),\n    fig.add_subplot(224),\n]\nlines_u, lines_v, texts, score_text = list(), list(), list(), list()\nfor i, obs in enumerate(a):\n    axs[i].set_aspect(\"equal\")\n    axs[i].grid()\n    axs[i].set_xticklabels([]), axs[i].set_yticklabels([])\n    axs[i].plot(\n        obs[\"contour_lon_e\"], obs[\"contour_lat_e\"], \"r\", lw=6, label=\"Original contour\"\n    )\n    lines_v.append(axs[i].plot([], [], color=\"limegreen\", lw=4, label=\"visvalingam\")[0])\n    lines_u.append(\n        axs[i].plot([], [], color=\"black\", lw=2, label=\"uniform resampling\")[0]\n    )\n    texts.append(axs[i].set_title(\"\", fontsize=8))\naxs[0].legend(fontsize=8)\ntitle = fig.suptitle(\"\")\nani = animation.FuncAnimation(fig, update_line, 27)\n\n# ani"
       ]
     }
   ],