Fix ThreeDAxes bug

This commit is contained in:
Grant Sanderson 2019-02-04 14:12:49 -08:00
parent b70e0a0c75
commit 9619ebfd61

View file

@ -36,21 +36,23 @@ class Axes(VGroup):
def __init__(self, **kwargs): def __init__(self, **kwargs):
VGroup.__init__(self, **kwargs) VGroup.__init__(self, **kwargs)
x_axis_config = merge_config([ self.x_axis = self.get_axis(
self.x_axis_config, self.x_min, self.x_max, self.x_axis_config
{"x_min": self.x_min, "x_max": self.x_max}, )
self.number_line_config, self.y_axis = self.get_axis(
]) self.y_min, self.y_max, self.y_axis_config
y_axis_config = merge_config([ )
self.y_axis_config,
{"x_min": self.y_min, "x_max": self.y_max},
self.number_line_config,
])
self.x_axis = NumberLine(**x_axis_config)
self.y_axis = NumberLine(**y_axis_config)
self.y_axis.rotate(90 * DEGREES, about_point=ORIGIN) self.y_axis.rotate(90 * DEGREES, about_point=ORIGIN)
self.add(self.x_axis, self.y_axis) self.add(self.x_axis, self.y_axis)
def get_axis(self, min_val, max_val, axis_config):
new_config = merge_config([
axis_config,
{"x_min": min_val, "x_max": max_val},
self.number_line_config,
])
return NumberLine(**new_config)
def coords_to_point(self, *coords): def coords_to_point(self, *coords):
origin = self.x_axis.number_to_point(0) origin = self.x_axis.number_to_point(0)
result = np.array(origin) result = np.array(origin)