From 0f94c06f9a40ad232eae8cc7f72bf032da55314f Mon Sep 17 00:00:00 2001 From: Grant Sanderson Date: Wed, 26 Dec 2018 11:33:50 -0800 Subject: [PATCH] ComplexValueTracker --- mobject/value_tracker.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/mobject/value_tracker.py b/mobject/value_tracker.py index 18450868..463f95de 100644 --- a/mobject/value_tracker.py +++ b/mobject/value_tracker.py @@ -41,3 +41,13 @@ class ExponentialValueTracker(ValueTracker): def set_value(self, value): return ValueTracker.set_value(self, np.log(value)) + + +class ComplexValueTracker(ValueTracker): + def get_value(self): + return complex(*self.points[0, :2]) + + def set_value(self, z): + z = complex(z) + self.points[0, :2] = (z.real, z.imag) + return self