Skip to content

Commit 6d72893

Browse files
committed
Let ValueTracker track vectors
1 parent 40290ad commit 6d72893

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

manimlib/mobject/value_tracker.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
import numpy as np
22

33
from manimlib.mobject.mobject import Mobject
4+
from manimlib.utils.iterables import listify
45

56

67
class ValueTracker(Mobject):
@@ -15,18 +16,22 @@ class ValueTracker(Mobject):
1516
}
1617

1718
def __init__(self, value=0, **kwargs):
19+
self.value = value
1820
super().__init__(**kwargs)
19-
self.set_value(value)
2021

2122
def init_data(self):
2223
super().init_data()
23-
self.data["value"] = np.zeros((1, 1), dtype=self.value_type)
24+
self.data["value"] = np.array(
25+
listify(self.value),
26+
ndmin=2,
27+
dtype=self.value_type,
28+
)
2429

2530
def get_value(self):
26-
return self.data["value"][0, 0]
31+
return self.data["value"][0, :]
2732

2833
def set_value(self, value):
29-
self.data["value"][0, 0] = value
34+
self.data["value"][0, :] = value
3035
return self
3136

3237
def increment_value(self, d_value):

0 commit comments

Comments
 (0)