opensource/manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-01 17:29:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enabling curvature in Lines/arrows

Browse Source
This commit is contained in:
Grant Sanderson
2017-04-26 16:21:38 -07:00
parent f80d7e0fcd
commit fe7d1d5eaf
1 changed files with 18 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
topics/geometry.py
View File

@ -65,12 +65,26 @@ class Line(VMobject):
CONFIG = { CONFIG = {
"buff" : 0, "buff" : 0,
"considered_smooth" : False, "considered_smooth" : False,
"path_arc" : None,
"n_arc_anchors" : 10, #Only used if path_arc is not None
} }
def __init__(self, start, end, **kwargs): def __init__(self, start, end, **kwargs):
digest_config(self, kwargs) digest_config(self, kwargs)
self.set_start_and_end(start, end) self.set_start_and_end(start, end)
VMobject.__init__(self, **kwargs) VMobject.__init__(self, **kwargs)
def generate_points(self):
if self.path_arc is None:
self.set_points_as_corners([self.start, self.end])
else:
path_func = path_along_arc(self.path_arc)
self.set_points_smoothly([
path_func(self.start, self.end, alpha)
for alpha in np.linspace(0, 1, self.n_arc_anchors)
])
self.considered_smooth = True
def set_start_and_end(self, start, end): def set_start_and_end(self, start, end):
start_to_end = self.pointify(end) - self.pointify(start) start_to_end = self.pointify(end) - self.pointify(start)
vect = np.zeros(len(start_to_end)) vect = np.zeros(len(start_to_end))
@ -94,9 +108,6 @@ class Line(VMobject):
return mob_or_point.get_center() return mob_or_point.get_center()
return np.array(mob_or_point) return np.array(mob_or_point)
def generate_points(self):
self.set_points_as_corners([self.start, self.end])
def get_length(self): def get_length(self):
start, end = self.get_start_and_end() start, end = self.get_start_and_end()
return np.linalg.norm(start - end) return np.linalg.norm(start - end)
@ -111,7 +122,7 @@ class Line(VMobject):
return self.points[-1] return self.points[-1]
def get_slope(self): def get_slope(self):
end = self.get_end() start, end = self.get_start_and_end()
rise, run = [ rise, run = [
float(end[i] - start[i]) float(end[i] - start[i])
for i in [1, 0] for i in [1, 0]
@ -187,9 +198,10 @@ class Arrow(Line):
self.add_tip() self.add_tip()
def add_tip(self, add_at_end = True): def add_tip(self, add_at_end = True):
vect = self.tip_length*RIGHT
vect = rotate_vector(vect, self.get_angle()+np.pi)
start, end = self.get_start_and_end() start, end = self.get_start_and_end()
anchors = self.get_anchors()
vect = anchors[-1] - anchors[-2]
vect *= -self.tip_length / np.linalg.norm(vect)
if not add_at_end: if not add_at_end:
start, end = end, start start, end = end, start
vect = -vect vect = -vect
@ -208,9 +220,6 @@ class Arrow(Line):
[tip_points[0], end, tip_points[1]], [tip_points[0], end, tip_points[1]],
mode = "corners" mode = "corners"
) )
self.set_points_as_corners(
[start, center_of_mass(tip_points)]
)
self.add(self.tip) self.add(self.tip)
self.init_colors() self.init_colors()