Code for animations in #431

[gatling-nrl]

@kinnala you just posted a gif at

#431 (comment)

which I assume is just a way to upload it to github for use in the docs. Is the code you used to make that available somewhere? Is that matplotlib?

[kinnala]

It's matplotlib and I have the script on my laptop. I can share it when I'm on a keyboard again.

[kinnala]

import numpy as np
import matplotlib.animation as anim
from skfem import *
from skfem.helpers import ddot, dd
from skfem.models import mass
from skfem.visuals.matplotlib import plot

@BilinearForm
def biharmonic(u, v, w):
  return ddot(dd(u), dd(v))

#m = MeshTri.init_lshaped().refined(4)
#m = MeshTri.init_sqsymmetric().refined(4)
m = MeshTri.init_tensor(
  np.linspace(0, 4, 40),
  np.linspace(0, 1, 10),
)
m = MeshTri.init_circle(nrefs=4)
m = MeshTri.init_refdom().refined(5)
basis = Basis(m, ElementTriMorley())

# matrices
A = biharmonic.assemble(basis)
M = mass.assemble(basis)

# solve eigenvalue problem
D = basis.get_dofs()
x = solve(*condense(A, M, D=D), solver=solver_eigen_scipy_sym(k=7))
nth = 4
x0 = x[1][:, nth]

# animation
import matplotlib.animation as anim

ax = plot(basis, x0, nrefs=2, shading='gouraud', figsize=(4, 4))
field = ax.get_children()[0]
fig = ax.get_figure()

ax.xaxis.set_visible(False)
ax.yaxis.set_visible(False)
ax.set_frame_on(False)
fig.patch.set_alpha(0.)

def update(t):
    u = x0 * np.sin(2. * np.pi * t)
    _, uref = basis.refinterp(u, nrefs=2)
    field.set_array(uref)

animation = anim.FuncAnimation(
    fig,
    update,
    np.linspace(0, 1, 100),
    repeat=False,
    interval=50,
)
animation.save("plate.gif", "imagemagick", dpi=80)