mirror of
https://github.com/3b1b/manim.git
synced 2025-04-13 09:47:07 +00:00
61 lines
1.5 KiB
Python
61 lines
1.5 KiB
Python
import numpy as np
|
|
|
|
from manimlib.constants import ORIGIN
|
|
from manimlib.constants import UP
|
|
from manimlib.utils.space_ops import get_norm
|
|
from manimlib.utils.space_ops import get_unit_normal
|
|
|
|
|
|
def get_3d_vmob_gradient_start_and_end_points(vmob):
|
|
return (
|
|
get_3d_vmob_start_corner(vmob),
|
|
get_3d_vmob_end_corner(vmob),
|
|
)
|
|
|
|
|
|
def get_3d_vmob_start_corner_index(vmob):
|
|
return 0
|
|
|
|
|
|
def get_3d_vmob_end_corner_index(vmob):
|
|
return ((len(vmob.points) - 1) // 6) * 3
|
|
|
|
|
|
def get_3d_vmob_start_corner(vmob):
|
|
if vmob.get_num_points() == 0:
|
|
return np.array(ORIGIN)
|
|
return vmob.points[get_3d_vmob_start_corner_index(vmob)]
|
|
|
|
|
|
def get_3d_vmob_end_corner(vmob):
|
|
if vmob.get_num_points() == 0:
|
|
return np.array(ORIGIN)
|
|
return vmob.points[get_3d_vmob_end_corner_index(vmob)]
|
|
|
|
|
|
def get_3d_vmob_unit_normal(vmob, point_index):
|
|
n_points = vmob.get_num_points()
|
|
if len(vmob.get_anchors()) <= 2:
|
|
return np.array(UP)
|
|
i = point_index
|
|
im3 = i - 3 if i > 2 else (n_points - 4)
|
|
ip3 = i + 3 if i < (n_points - 3) else 3
|
|
unit_normal = get_unit_normal(
|
|
vmob.points[ip3] - vmob.points[i],
|
|
vmob.points[im3] - vmob.points[i],
|
|
)
|
|
if get_norm(unit_normal) == 0:
|
|
return np.array(UP)
|
|
return unit_normal
|
|
|
|
|
|
def get_3d_vmob_start_corner_unit_normal(vmob):
|
|
return get_3d_vmob_unit_normal(
|
|
vmob, get_3d_vmob_start_corner_index(vmob)
|
|
)
|
|
|
|
|
|
def get_3d_vmob_end_corner_unit_normal(vmob):
|
|
return get_3d_vmob_unit_normal(
|
|
vmob, get_3d_vmob_end_corner_index(vmob)
|
|
)
|