1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
import numpy as np
import matplotlib.pyplot as plt
from basix import Lagrange, CellType, create_lattice, LatticeType
d = 3
L = Lagrange("triangle", d)
pts = create_lattice(CellType.triangle, 50, LatticeType.equispaced, True)
w0 = L.tabulate(2, pts)
np.set_printoptions(linewidth=240)
for kk in range(6):
w = w0[kk]
print(w)
ww = 1
nn = (d + 2) * (d + 1) // 2
if (nn % 4 == 0 and nn > 4):
nn //= 4
ww = 4
elif (nn % 3 == 0 and nn > 3):
nn //= 3
ww = 3
elif (nn % 2 == 0 and nn > 2):
nn //= 2
ww = 2
fig, ax = plt.subplots(ww, nn)
ax[0, 0].set_title("w[%d]" % kk)
for j, a in enumerate(ax.flatten()):
shape_fn = w[:, j]
shape_fn[np.where(abs(shape_fn)<1e-12)] = 0.0
a.tricontourf(pts[:, 0], pts[:, 1], shape_fn)
plt.show()
|
