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

use tqdm to display progress bar for long running SVG triangulations

Browse source
This commit is contained in:
Sunkisser 2022-03-28 03:30:10 +00:00
parent cfba6c431f
commit c4ea794107
1 changed files with 12 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
manimlib/utils/space_ops.py
View file

@ -4,11 +4,13 @@ import math
import operator as op import operator as op
from functools import reduce from functools import reduce
from typing import Callable, Iterable, Sequence from typing import Callable, Iterable, Sequence
import platform
import numpy as np import numpy as np
import numpy.typing as npt import numpy.typing as npt
from mapbox_earcut import triangulate_float32 as earcut from mapbox_earcut import triangulate_float32 as earcut
from scipy.spatial.transform import Rotation from scipy.spatial.transform import Rotation
from tqdm import tqdm as ProgressDisplay
from manimlib.constants import RIGHT from manimlib.constants import RIGHT
from manimlib.constants import DOWN from manimlib.constants import DOWN
@ -414,7 +416,16 @@ def earclip_triangulation(verts: np.ndarray, ring_ends: list[int]) -> list:
)) ))
chilren = [[] for i in rings] chilren = [[] for i in rings]
for idx, i in enumerate(rings_sorted): ringenum = ProgressDisplay(
enumerate(rings_sorted),
total=len(rings),
leave=False,
ascii=True if platform.system() == 'Windows' else None,
dynamic_ncols=True,
desc="SVG Triangulation",
delay=3,
)
for idx, i in ringenum:
for j in rings_sorted[:idx][::-1]: for j in rings_sorted[:idx][::-1]:
if is_in_fast(i, j): if is_in_fast(i, j):
chilren[j].append(i) chilren[j].append(i)