mirror of
https://github.com/3b1b/manim.git
synced 2025-07-30 21:44:19 +08:00
260 lines
5.4 KiB
Python
260 lines
5.4 KiB
Python
from big_ol_pile_of_manim_imports import *
|
|
from random import *
|
|
|
|
def text_range(start,stop,step): # a range as a list of strings
|
|
numbers = np.arange(start,stop,step)
|
|
labels = []
|
|
for x in numbers:
|
|
labels.append(str(x))
|
|
return labels
|
|
|
|
|
|
class Histogram(VMobject):
|
|
|
|
CONFIG = {
|
|
"start_color" : RED,
|
|
"end_color" : BLUE,
|
|
"x_scale" : 1.0,
|
|
"y_scale" : 1.0,
|
|
}
|
|
|
|
def __init__(self, x_values, y_values, **kwargs):
|
|
|
|
digest_config(self, kwargs)
|
|
|
|
# preliminaries
|
|
self.x_values = x_values
|
|
self.y_values = y_values
|
|
|
|
self.x_steps = x_values[1:] - x_values[:-1]
|
|
self.x_min = x_values[0] - self.x_steps[0] * 0.5
|
|
self.x_posts = (x_values[1:] + x_values[:-1]) * 0.5
|
|
self.x_max = x_values[-1] + self.x_steps[-1] * 0.5
|
|
self.x_posts = np.insert(self.x_posts,0,self.x_min)
|
|
self.x_posts = np.append(self.x_posts,self.x_max)
|
|
|
|
self.x_widths = self.x_posts[1:] - self.x_posts[:-1]
|
|
|
|
self.x_values_scaled = self.x_scale * x_values
|
|
self.x_steps_scaled = self.x_scale * self.x_steps
|
|
self.x_posts_scaled = self.x_scale * self.x_posts
|
|
self.x_min_scaled = self.x_scale * self.x_min
|
|
self.x_max_scaled = self.x_scale * self.x_max
|
|
self.x_widths_scaled = self.x_scale * self.x_widths
|
|
|
|
self.y_values_scaled = self.y_scale * self.y_values
|
|
|
|
VMobject.__init__(self, **kwargs)
|
|
digest_config(self, kwargs)
|
|
|
|
|
|
def generate_points(self):
|
|
|
|
previous_bar = ORIGIN
|
|
self.bars = []
|
|
outline_points = []
|
|
self.x_labels = text_range(self.x_values[0], self.x_max, self.x_steps[0])
|
|
|
|
for (i,x) in enumerate(self.x_values):
|
|
|
|
bar = Rectangle(
|
|
width = self.x_widths_scaled[i],
|
|
height = self.y_values_scaled[i],
|
|
)
|
|
t = float(x - self.x_values[0])/(self.x_values[-1] - self.x_values[0])
|
|
bar_color = interpolate_color(
|
|
self.start_color,
|
|
self.end_color,
|
|
t
|
|
)
|
|
bar.set_fill(color = bar_color, opacity = 1)
|
|
bar.set_stroke(width = 0)
|
|
bar.next_to(previous_bar,RIGHT,buff = 0, aligned_edge = DOWN)
|
|
|
|
self.add(bar)
|
|
self.bars.append(bar)
|
|
|
|
label = TextMobject(self.x_labels[i])
|
|
label.next_to(bar,DOWN)
|
|
self.add(label)
|
|
|
|
if i == 0:
|
|
# start with the lower left
|
|
outline_points.append(bar.get_anchors()[-2])
|
|
|
|
# upper two points of each bar
|
|
outline_points.append(bar.get_anchors()[0])
|
|
outline_points.append(bar.get_anchors()[1])
|
|
|
|
previous_bar = bar
|
|
|
|
# close the outline
|
|
# lower right
|
|
outline_points.append(bar.get_anchors()[2])
|
|
# lower left
|
|
outline_points.append(outline_points[0])
|
|
|
|
self.outline = Polygon(*outline_points)
|
|
self.outline.set_stroke(color = WHITE)
|
|
self.add(self.outline)
|
|
|
|
def get_lower_left_point(self):
|
|
return self.bars[0].get_anchors()[-2]
|
|
|
|
|
|
|
|
|
|
|
|
class FlashThroughHistogram(Animation):
|
|
CONFIG = {
|
|
"cell_color" : WHITE,
|
|
"cell_opacity" : 0.8,
|
|
"hist_opacity" : 0.2
|
|
}
|
|
|
|
def __init__(self, mobject, direction = "horizontal", mode = "random", **kwargs):
|
|
|
|
digest_config(self, kwargs)
|
|
|
|
self.cell_height = mobject.y_scale
|
|
self.prototype_cell = Rectangle(
|
|
width = 1,
|
|
height = self.cell_height,
|
|
fill_color = self.cell_color,
|
|
fill_opacity = self.cell_opacity,
|
|
stroke_width = 0,
|
|
)
|
|
|
|
x_values = mobject.x_values
|
|
y_values = mobject.y_values
|
|
|
|
self.mode = mode
|
|
self.direction = direction
|
|
|
|
self.generate_cell_indices(x_values,y_values)
|
|
Animation.__init__(self,mobject,**kwargs)
|
|
|
|
|
|
|
|
def generate_cell_indices(self,x_values,y_values):
|
|
|
|
self.cell_indices = []
|
|
for (i,x) in enumerate(x_values):
|
|
|
|
nb_cells = y_values[i]
|
|
for j in range(nb_cells):
|
|
self.cell_indices.append((i, j))
|
|
|
|
self.reordered_cell_indices = self.cell_indices
|
|
if self.mode == "random":
|
|
shuffle(self.reordered_cell_indices)
|
|
|
|
|
|
def cell_for_index(self,i,j):
|
|
|
|
if self.direction == "vertical":
|
|
width = self.mobject.x_scale
|
|
height = self.mobject.y_scale
|
|
x = (i + 0.5) * self.mobject.x_scale
|
|
y = (j + 0.5) * self.mobject.y_scale
|
|
center = self.mobject.get_lower_left_point() + x * RIGHT + y * UP
|
|
|
|
elif self.direction == "horizontal":
|
|
width = self.mobject.x_scale / self.mobject.y_values[i]
|
|
height = self.mobject.y_scale * self.mobject.y_values[i]
|
|
x = i * self.mobject.x_scale + (j + 0.5) * width
|
|
y = height / 2
|
|
center = self.mobject.get_lower_left_point() + x * RIGHT + y * UP
|
|
|
|
cell = Rectangle(width = width, height = height)
|
|
cell.move_to(center)
|
|
return cell
|
|
|
|
|
|
def update_mobject(self,t):
|
|
|
|
if t == 0:
|
|
self.mobject.add(self.prototype_cell)
|
|
|
|
flash_nb = int(t * (len(self.cell_indices))) - 1
|
|
(i,j) = self.reordered_cell_indices[flash_nb]
|
|
cell = self.cell_for_index(i,j)
|
|
self.prototype_cell.width = cell.get_width()
|
|
self.prototype_cell.height = cell.get_height()
|
|
self.prototype_cell.generate_points()
|
|
self.prototype_cell.move_to(cell.get_center())
|
|
|
|
#if t == 1:
|
|
# self.mobject.remove(self.prototype_cell)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class SampleScene(Scene):
|
|
|
|
def construct(self):
|
|
|
|
x_values = np.array([1,2,3,4,5])
|
|
y_values = np.array([4,3,5,2,3])
|
|
|
|
hist1 = Histogram(
|
|
x_values = x_values,
|
|
y_values = y_values,
|
|
x_scale = 0.5,
|
|
y_scale = 0.5,
|
|
).shift(1*DOWN)
|
|
self.add(hist1)
|
|
self.wait()
|
|
|
|
y_values2 = np.array([3,8,7,15,5])
|
|
|
|
hist2 = Histogram(
|
|
x_values = x_values,
|
|
y_values = y_values2,
|
|
x_scale = 0.5,
|
|
y_scale = 0.5,
|
|
x_labels = text_range(1,6,1),
|
|
)
|
|
|
|
v1 = hist1.get_lower_left_point()
|
|
v2 = hist2.get_lower_left_point()
|
|
hist2.shift(v1 - v2)
|
|
|
|
# self.play(
|
|
# ReplacementTransform(hist1,hist2)
|
|
# )
|
|
|
|
self.play(
|
|
FlashThroughHistogram(
|
|
hist1,
|
|
direction = "horizontal",
|
|
mode = "linear",
|
|
run_time = 10,
|
|
rate_func = None,
|
|
)
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|