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

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
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])