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

Added Line.get_unit_vector

Browse source
This commit is contained in:
Grant Sanderson 2018-07-17 12:44:18 -07:00
parent a443ddc8f2
commit 0b290a19b9
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
mobject/geometry.py
View file

@ -368,6 +368,14 @@ class Line(VMobject):
def get_vector(self):
return self.get_end() - self.get_start()
def get_unit_vector(self):
vect = self.get_vector()
norm = np.linalg.norm(vect)
if norm == 0:
# TODO, is this the behavior I want?
return np.array(ORIGIN)
return vect / norm
def get_start(self):
return np.array(self.points[0])