public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

slight improvements to NumberPlane

Browse source
This commit is contained in:
Grant Sanderson 2015-10-06 18:41:26 -07:00
parent 54de0881f5
commit 72f0af0fbf
1 changed files with 16 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
mobject/function_graphs.py
View file

@ -200,23 +200,33 @@ class NumberPlane(Mobject1D):
def generate_points(self): def generate_points(self):
color = self.color color = self.color
faded = Color(rgb = self.fade_factor*np.array(color.get_rgb())) faded = Color(rgb = self.fade_factor*np.array(color.get_rgb()))
colors = iter([faded, color]) freq_color_pairs = [
for freq in self.x_faded_line_frequency, self.x_line_frequency: (self.x_faded_line_frequency, faded),
(self.x_line_frequency, color)
]
for freq, color in freq_color_pairs:
if not freq:
continue
self.add_points([ self.add_points([
(sgns[0]*self.x_unit_to_spatial_width*x, sgns[1]*y, 0) (sgns[0]*self.x_unit_to_spatial_width*x, sgns[1]*y, 0)
for x in np.arange(0, self.x_radius, freq) for x in np.arange(0, self.x_radius, freq)
for y in np.arange(0, self.y_radius, self.epsilon) for y in np.arange(0, self.y_radius, self.epsilon)
for sgns in it.product([-1, 1], [-1, 1]) for sgns in it.product([-1, 1], [-1, 1])
], color = colors.next()) ], color = color)
#Horizontal lines #Horizontal lines
colors = iter([faded, color]) freq_color_pairs = [
for freq in self.y_faded_line_frequency, self.y_line_frequency: (self.y_faded_line_frequency, faded),
(self.y_line_frequency, color)
]
for freq, color in freq_color_pairs:
if not freq:
continue
self.add_points([ self.add_points([
(sgns[0]*x, sgns[1]*self.y_uint_to_spatial_height*y, 0) (sgns[0]*x, sgns[1]*self.y_uint_to_spatial_height*y, 0)
for x in np.arange(0, self.x_radius, self.epsilon) for x in np.arange(0, self.x_radius, self.epsilon)
for y in np.arange(0, self.y_radius, freq) for y in np.arange(0, self.y_radius, freq)
for sgns in it.product([-1, 1], [-1, 1]) for sgns in it.product([-1, 1], [-1, 1])
], color = colors.next()) ], color = color)
self.shift(self.get_center_point()) self.shift(self.get_center_point())
def get_center_point(self): def get_center_point(self):