public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-08-05 16:49:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Faster VMobject.get_arc_length

Browse source
This commit is contained in:
Grant Sanderson 2023-02-16 15:02:30 -08:00
parent 3a05352f73
commit c372ef4aaa
1 changed files with 12 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
manimlib/mobject/types/vectorized_mobject.py
View file

@ -17,6 +17,7 @@ from manimlib.utils.bezier import bezier
from manimlib.utils.bezier import get_quadratic_approximation_of_cubic from manimlib.utils.bezier import get_quadratic_approximation_of_cubic
from manimlib.utils.bezier import approx_smooth_quadratic_bezier_handles from manimlib.utils.bezier import approx_smooth_quadratic_bezier_handles
from manimlib.utils.bezier import smooth_quadratic_path from manimlib.utils.bezier import smooth_quadratic_path
from manimlib.utils.bezier import interpolate
from manimlib.utils.bezier import integer_interpolate from manimlib.utils.bezier import integer_interpolate
from manimlib.utils.bezier import inverse_interpolate from manimlib.utils.bezier import inverse_interpolate
from manimlib.utils.bezier import find_intersection from manimlib.utils.bezier import find_intersection
@ -38,6 +39,7 @@ from manimlib.utils.space_ops import get_unit_normal
from manimlib.utils.space_ops import line_intersects_path from manimlib.utils.space_ops import line_intersects_path
from manimlib.utils.space_ops import midpoint from manimlib.utils.space_ops import midpoint
from manimlib.utils.space_ops import normalize_along_axis from manimlib.utils.space_ops import normalize_along_axis
from manimlib.utils.space_ops import poly_line_length
from manimlib.utils.space_ops import z_to_vector from manimlib.utils.space_ops import z_to_vector
from manimlib.shader_wrapper import ShaderWrapper from manimlib.shader_wrapper import ShaderWrapper
from manimlib.shader_wrapper import FillShaderWrapper from manimlib.shader_wrapper import FillShaderWrapper
@ -814,14 +816,16 @@ class VMobject(Mobject):
return np.vstack(new_points) return np.vstack(new_points)
def get_arc_length(self, n_sample_points: int | None = None) -> float: def get_arc_length(self, n_sample_points: int | None = None) -> float:
if n_sample_points is None: if n_sample_points is not None:
n_sample_points = 4 * self.get_num_curves() + 1 points = np.array([
points = np.array([ self.quick_point_from_proportion(a)
self.point_from_proportion(a) for a in np.linspace(0, 1, n_sample_points)
for a in np.linspace(0, 1, n_sample_points) ])
]) return poly_line_length(points)
diffs = points[1:] - points[:-1] points = self.get_points()
return sum(map(get_norm, diffs)) inner_len = poly_line_length(points[::2])
outer_len = poly_line_length(points)
return interpolate(inner_len, outer_len, 1 / 3)
def get_area_vector(self) -> Vect3: def get_area_vector(self) -> Vect3:
# Returns a vector whose length is the area bound by # Returns a vector whose length is the area bound by