public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

EquationSolver2d restructure, en route to implementing BFS

Browse source
This commit is contained in:
Sridhar Ramesh 2018-03-20 17:19:29 -07:00
parent 739807cb66
commit b58b771678
1 changed files with 37 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

49
active_projects/WindingNumber.py
View file

@ -922,6 +922,20 @@ class PiWalkerCircle(PiWalker):
self.walk_coords = [r * np.array((np.cos(i * TAU/N), np.sin(i * TAU/N))) for i in range(N)] self.walk_coords = [r * np.array((np.cos(i * TAU/N), np.sin(i * TAU/N))) for i in range(N)]
PiWalker.setup(self) PiWalker.setup(self)
class EquationSolver2dNode(object):
def __init__(self, first_anim, children = []):
self.first_anim = first_anim
self.children = children
def display_in_series(self):
return Succession(self.first_anim, *map(EquationSolver2dNode.display_in_series, self.children))
def display_in_parallel(self):
return Succession(self.first_anim, AnimationGroup(*map(EquationSolver2dNode.display_in_parallel, self.children)))
def display_in_bfs(self):
print "Error! Not yet implemented bfs display!"
class EquationSolver2d(ColorMappedObjectsScene): class EquationSolver2d(ColorMappedObjectsScene):
CONFIG = { CONFIG = {
"camera_config" : {"use_z_coordinate_for_display_order": True}, "camera_config" : {"use_z_coordinate_for_display_order": True},
@ -931,7 +945,11 @@ class EquationSolver2d(ColorMappedObjectsScene):
"initial_upper_y" : 3, "initial_upper_y" : 3,
"num_iterations" : 1, "num_iterations" : 1,
"num_checkpoints" : 10, "num_checkpoints" : 10,
# Should really merge this into one enum-style variable
"display_in_parallel" : False, "display_in_parallel" : False,
"display_in_bfs" : False,
"use_fancy_lines" : True, "use_fancy_lines" : True,
# TODO: Consider adding a "find_all_roots" flag, which could be turned off # TODO: Consider adding a "find_all_roots" flag, which could be turned off
# to only explore one of the two candidate subrectangles when both are viable # to only explore one of the two candidate subrectangles when both are viable
@ -991,7 +1009,7 @@ class EquationSolver2d(ColorMappedObjectsScene):
print "Solver at depth: " + str(cur_depth) print "Solver at depth: " + str(cur_depth)
if cur_depth >= self.num_iterations: if cur_depth >= self.num_iterations:
return empty_animation return EquationSolver2dNode(empty_animation)
def draw_line_return_wind(start, end, start_wind, should_linger = False, draw_line = True): def draw_line_return_wind(start, end, start_wind, should_linger = False, draw_line = True):
alpha_winder = make_alpha_winder(clockwise_val_func, start, end, self.num_checkpoints) alpha_winder = make_alpha_winder(clockwise_val_func, start, end, self.num_checkpoints)
@ -1057,14 +1075,14 @@ class EquationSolver2d(ColorMappedObjectsScene):
# their "Nothing here" status? # their "Nothing here" status?
# Or draw a large X or something # Or draw a large X or something
fill_rect = polygonObject = Polygon(*points, fill_opacity = 0.8, color = DARK_GREY) fill_rect = polygonObject = Polygon(*points, fill_opacity = 0.8, color = DARK_GREY)
return Succession(anim, FadeIn(fill_rect)) return EquationSolver2dNode(Succession(anim, FadeIn(fill_rect)))
else: else:
(sub_rect1, sub_rect2) = rect.splits_on_dim(dim_to_split) (sub_rect1, sub_rect2) = rect.splits_on_dim(dim_to_split)
if dim_to_split == 0: if dim_to_split == 0:
sub_rect_and_sides = [(sub_rect1, 1), (sub_rect2, 3)] sub_rect_and_sides = [(sub_rect1, 1), (sub_rect2, 3)]
else: else:
sub_rect_and_sides = [(sub_rect1, 2), (sub_rect2, 0)] sub_rect_and_sides = [(sub_rect1, 2), (sub_rect2, 0)]
sub_anims = [ children = [
Animate2dSolver( Animate2dSolver(
cur_depth = cur_depth + 1, cur_depth = cur_depth + 1,
rect = sub_rect, rect = sub_rect,
@ -1077,14 +1095,8 @@ class EquationSolver2d(ColorMappedObjectsScene):
mid_line_coords = rect.split_line_on_dim(dim_to_split) mid_line_coords = rect.split_line_on_dim(dim_to_split)
mid_line_points = [num_plane.coords_to_point(x, y) + 2 * IN for (x, y) in mid_line_coords] mid_line_points = [num_plane.coords_to_point(x, y) + 2 * IN for (x, y) in mid_line_coords]
mid_line = DashedLine(*mid_line_points) mid_line = DashedLine(*mid_line_points)
if self.display_in_parallel:
recursive_anim = AnimationGroup(*sub_anims) return EquationSolver2dNode(Succession(anim, ShowCreation(mid_line)), children)
else:
recursive_anim = Succession(*sub_anims)
return Succession(anim,
ShowCreation(mid_line),
recursive_anim
)
lower_x = self.initial_lower_x lower_x = self.initial_lower_x
upper_x = self.initial_upper_x upper_x = self.initial_upper_x
@ -1098,7 +1110,7 @@ class EquationSolver2d(ColorMappedObjectsScene):
print "Starting to compute anim" print "Starting to compute anim"
anim = Animate2dSolver( node = Animate2dSolver(
cur_depth = 0, cur_depth = 0,
rect = rect, rect = rect,
dim_to_split = 0, dim_to_split = 0,
@ -1108,6 +1120,13 @@ class EquationSolver2d(ColorMappedObjectsScene):
print "Done computing anim" print "Done computing anim"
if self.display_in_parallel:
anim = node.display_in_parallel()
elif self.display_in_bfs:
anim = node.display_in_bfs()
else:
anim = node.display_in_series()
# Keep timing details here in sync with details above # Keep timing details here in sync with details above
rect_points = [ rect_points = [
rect.get_top_left(), rect.get_top_left(),
@ -1937,6 +1956,12 @@ class PreviewClip(EquationSolver2d):
"use_fancy_lines" : True, "use_fancy_lines" : True,
} }
class QuickPreview(PreviewClip):
CONFIG = {
"num_iterations" : 2,
"display_in_parallel" : False,
}
# TODO: Borsuk-Ulam visuals # TODO: Borsuk-Ulam visuals
# Note: May want to do an ordinary square scene, then MappingCamera it into a circle # Note: May want to do an ordinary square scene, then MappingCamera it into a circle
# class BorsukUlamScene(PiWalker): # class BorsukUlamScene(PiWalker):