jan/masque
1
1
Fork 0
Code Issues Pull Requests Releases Wiki Activity

[utils.curves.bezier] be more explicit about broadcast axes

Browse Source
This commit is contained in:
Jan Petykiewicz 2025-02-25 21:23:41 -08:00
parent cae6de69c1
commit c3534beb3f
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
masque/utils/curves.py
View File

@ -32,13 +32,13 @@ def bezier(
umul[~t_half0] = 1 umul[~t_half0] = 1
udiv[t_half0] = 1 udiv[t_half0] = 1
hh = numpy.ones((tt.size, 1)) hh = numpy.ones((tt.size,))
qq = nodes[None, 0] * hh qq = nodes[None, 0] * hh[:, None]
for kk in range(1, nn): for kk in range(1, nn):
hh *= umul * (nn + 1 - kk) * weights[kk] hh *= umul * (nn + 1 - kk) * weights[kk]
hh /= kk * udiv * weights[kk - 1] + hh hh /= kk * udiv * weights[kk - 1] + hh
qq *= 1.0 - hh qq *= 1.0 - hh[:, None]
qq += hh * nodes[None, kk] qq += hh[:, None] * nodes[None, kk]
return qq return qq