public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-09-01 00:48:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add conditions for resize_with_interpolation

Browse source
This commit is contained in:
Grant Sanderson 2023-01-19 09:56:40 -08:00
parent dbeef42600
commit 1eae7c06ba
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
manimlib/utils/iterables.py
View file

@ -101,6 +101,8 @@ def resize_preserving_order(nparray: np.ndarray, length: int) -> np.ndarray:
def resize_with_interpolation(nparray: np.ndarray, length: int) -> np.ndarray:
if len(nparray) == length:
return nparray
if len(nparray) == 1 or array_is_constant(nparray):
return nparray[:1].repeat(length, axis=0)
if length == 0:
return np.zeros((0, *nparray.shape[1:]))
cont_indices = np.linspace(0, len(nparray) - 1, length)
@ -133,6 +135,7 @@ def arrays_match(arr1: np.ndarray, arr2: np.ndarray) -> bool:
def array_is_constant(arr: np.ndarray) -> bool:
return len(arr) > 0 and not (arr - arr[0]).any()
def hash_obj(obj: object) -> int:
if isinstance(obj, dict):
return hash(tuple(sorted([