public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-11-14 14:27:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Replace .points with ._points

Browse source
This commit is contained in:
Grant Sanderson 2021-01-12 12:25:12 -10:00
parent d60ec48591
commit c4f190cfc4
1 changed files with 13 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
manimlib/once_useful_constructs/graph_theory.py
View file

@ -193,10 +193,10 @@ class DiscreteGraphScene(Scene):
Scene.__init__(self, *args, **kwargs) Scene.__init__(self, *args, **kwargs)
def construct(self): def construct(self):
self.points = list(map(np.array, self.graph.vertices)) self._points = list(map(np.array, self.graph.vertices))
self.vertices = self.dots = [Dot(p) for p in self.points] self.vertices = self.dots = [Dot(p) for p in self._points]
self.edges = self.lines = [ self.edges = self.lines = [
Line(self.points[i], self.points[j]) Line(self._points[i], self._points[j])
for i, j in self.graph.edges for i, j in self.graph.edges
] ]
self.add(*self.dots + self.edges) self.add(*self.dots + self.edges)
@ -212,8 +212,8 @@ class DiscreteGraphScene(Scene):
def region_from_cycle(self, cycle): def region_from_cycle(self, cycle):
point_pairs = [ point_pairs = [
[ [
self.points[cycle[i]], self._points[cycle[i]],
self.points[cycle[(i + 1) % len(cycle)]] self._points[cycle[(i + 1) % len(cycle)]]
] ]
for i in range(len(cycle)) for i in range(len(cycle))
] ]
@ -236,7 +236,7 @@ class DiscreteGraphScene(Scene):
start = Mobject(*self.vertices) start = Mobject(*self.vertices)
end = Mobject(*[ end = Mobject(*[
Dot(point, radius=3 * Dot.DEFAULT_RADIUS, color="lightgreen") Dot(point, radius=3 * Dot.DEFAULT_RADIUS, color="lightgreen")
for point in self.points for point in self._points
]) ])
self.play(Transform( self.play(Transform(
start, end, rate_func=there_and_back, start, end, rate_func=there_and_back,
@ -281,7 +281,7 @@ class DiscreteGraphScene(Scene):
next_in_cycle = it.cycle(cycle) next_in_cycle = it.cycle(cycle)
next(next_in_cycle) # jump one ahead next(next_in_cycle) # jump one ahead
self.traced_cycle = Mobject(*[ self.traced_cycle = Mobject(*[
Line(self.points[i], self.points[j]).set_color(color) Line(self._points[i], self._points[j]).set_color(color)
for i, j in zip(cycle, next_in_cycle) for i, j in zip(cycle, next_in_cycle)
]) ])
self.play( self.play(
@ -306,8 +306,8 @@ class DiscreteGraphScene(Scene):
to_check.add(pair[1]) to_check.add(pair[1])
self.spanning_tree = Mobject(*[ self.spanning_tree = Mobject(*[
Line( Line(
self.points[pair[0]], self._points[pair[0]],
self.points[pair[1]] self._points[pair[1]]
).set_color(color) ).set_color(color)
for pair in self.spanning_tree_index_pairs for pair in self.spanning_tree_index_pairs
]) ])
@ -320,7 +320,7 @@ class DiscreteGraphScene(Scene):
self.generate_spanning_tree() self.generate_spanning_tree()
root = self.spanning_tree_root root = self.spanning_tree_root
color = self.spanning_tree.get_color() color = self.spanning_tree.get_color()
indices = list(range(len(self.points))) indices = list(range(len(self._points)))
# Build dicts # Build dicts
parent_of = dict([ parent_of = dict([
tuple(reversed(pair)) tuple(reversed(pair))
@ -376,7 +376,7 @@ class DiscreteGraphScene(Scene):
cycles = self.graph.region_cycles cycles = self.graph.region_cycles
self.dual_points = [ self.dual_points = [
center_of_mass([ center_of_mass([
self.points[index] self._points[index]
for index in cycle for index in cycle
]) ])
for cycle in cycles for cycle in cycles
@ -404,8 +404,8 @@ class DiscreteGraphScene(Scene):
if all(dual_point_pair[i] == point_at_infinity): if all(dual_point_pair[i] == point_at_infinity):
new_point = np.array(dual_point_pair[1 - i]) new_point = np.array(dual_point_pair[1 - i])
vect = center_of_mass([ vect = center_of_mass([
self.points[pair[0]], self._points[pair[0]],
self.points[pair[1]] self._points[pair[1]]
]) - new_point ]) - new_point
new_point += FRAME_X_RADIUS * vect / get_norm(vect) new_point += FRAME_X_RADIUS * vect / get_norm(vect)
dual_point_pair[i] = new_point dual_point_pair[i] = new_point