如何使用Makie的流图为时变函数设置动画?

时间:2019-09-15 04:20:09

标签: plot julia julia-gpu

我一直在尝试绘制一个简单的函数:

v(x, y) = (y*t, 2*x*t)

(实际实现:v(x::Point2{T}, t) where T = Point2{T}(one(T) * x[2] * t, 4 * x[1])

使用Makie的2D流图功能。

当我尝试将其包装到Observable中时,可以通过创建一个匿名函数f = x -> v(x, 5e0)(例如)并绘制f来绘制每个时间步长(使用Node(f)) ,则我无法再将该Observable更新为指向另一个函数,因此我的记录循环失败。

我尝试将Node的类型强制为更抽象(Node{Function})。不幸的是,这种抽象类型似乎在Makie内部的某个地方迷失了,因此迷失了。

有什么方法可以包装我的类型,以免遇到类型冲突,但仍然是一个函数?

1 个答案:

答案 0 :(得分:6)

所以事实证明,有一个可爱的Base方法Fix2(function, argument),它基本上解决了我的问题(因为它返回了包装的类型)。

请考虑以下示例功能:

v(x::Point2{T}, t) where T = Point2{T}(one(T) * x[2] * t, 4 * x[1])

可以通过以下方式将时间设置为1.0来创建可调用函数:

fixedfunc = Base.Fix2(v, 1.0)

然后可以将其加载到Observable中,并进行更新以具有新的时间值而不会出现问题。

这是最终代码,以供参考:

v(x::Point2{T}, t) where T = Point2{T}(one(T) * x[2] * t, 4 * x[1])
sf = Node(Base.Fix2(v, 0e0))

title_str = Node("t = 0.00")

sp = streamplot(
        sf,
        -2..2, -2..2;
        linewidth = 2,
        padding = (0, 0),
        arrow_size = 0.09,
        colormap =:magma
    )

sc = title(sp, title_str)

record(sc, "test.mp4", LinRange(0, 20, 5*30)) do i
  sf[] = Base.Fix2(v, i)
  title_str[] = "t = $(round(i; sigdigits = 2))"
end

video