Refactored helpers.py into a folder of various util files, take 2

animation/compositions.py

@@ -7,7 +7,7 @@ import warnings
 from mobject.mobject import Mobject, Group
 from mobject.vectorized_mobject import VMobject
 from mobject.tex_mobject import TextMobject
-from animation import Animation
+from .animation import Animation
 from transform import Transform
 from utils.bezier import inverse_interpolate
 from utils.config_ops import digest_config

animation/playground.py

@@ -1,9 +1,10 @@
 import numpy as np
 import operator as op
 
-from animation import Animation
+from .animation import Animation
 from transform import Transform
-from mobject.mobject import Mobject1D, Mobject
+from mobject.mobject import Mobject
+from mobject.point_cloud_mobject import Mobject1D
 from topics.geometry import Line
 from utils.paths import path_along_arc
 

animation/simple_animations.py

@@ -7,7 +7,7 @@ import warnings
 from mobject.mobject import Mobject, Group
 from mobject.vectorized_mobject import VMobject
 from mobject.tex_mobject import TextMobject
-from animation import Animation
+from .animation import Animation
 from transform import Transform
 from utils.bezier import interpolate
 from utils.config_ops import digest_config

animation/transform.py

@@ -6,8 +6,9 @@ import warnings
 
 from constants import *
 
-from animation import Animation
-from mobject.mobject import Mobject, Point, VMobject, Group
+from .animation import Animation
+from mobject.mobject import Mobject, Group
+from mobject.vectorized_mobject import VMobject, VectorizedPoint
 from topics.geometry import Dot, Circle
 from utils.config_ops import digest_config
 from utils.iterables import adjacent_pairs
@@ -112,7 +113,7 @@ class GrowFromPoint(Transform):
     def __init__(self, mobject, point, **kwargs):
         digest_config(self, kwargs)
         target = mobject.copy()
-        point_mob = Point(point)
+        point_mob = VectorizedPoint(point)
         if self.point_color:
             point_mob.set_color(self.point_color)
         mobject.replace(point_mob)

big_ol_pile_of_manim_imports.py

@@ -63,7 +63,6 @@ from utils.color import *
 from utils.config_ops import *
 from utils.images import *
 from utils.iterables import *
-from utils.linear_algebra import *
 from utils.paths import *
 from utils.rate_functions import *
 from utils.simple_functions import *

extract_scene.py

@@ -14,7 +14,7 @@ import subprocess as sp
 from constants import *
 from scene.scene import Scene
 from camera.camera import Camera
-from utils.sound import play_error_sound, play_finish_sound
+from utils.sounds import play_error_sound, play_finish_sound
 
 HELP_MESSAGE = """
    Usage:

mobject/__init__.py

@@ -1,5 +0,0 @@
-__all__ = [
-    "mobject",
-    "image_mobject",
-    "tex_mobject",
-]

mobject/image_mobject.py

@@ -5,7 +5,7 @@ from PIL import Image
 from random import random
 
 from constants import *
-from mobject.mobject import Mobject
+from .mobject import Mobject
 from point_cloud_mobject import PMobject
 from utils.bezier import interpolate
 from utils.color import color_to_int_rgb

mobject/mobject.py

@@ -1,5 +1,6 @@
 import numpy as np
 import operator as op
+import itertools as it
 import os
 import copy
 from PIL import Image

mobject/point_cloud_mobject.py

@@ -1,7 +1,7 @@
 from constants import *
-from mobject.mobject import Mobject
+from .mobject import Mobject
 from utils.bezier import interpolate
-from utils.color import color_to_rgb, color_to_rgba, rgba_to_color, 
+from utils.color import color_to_rgb, color_to_rgba, rgba_to_color
 from utils.color import color_gradient
 from utils.color import interpolate_color
 from utils.config_ops import digest_config

mobject/svg_mobject.py

@@ -1,4 +1,6 @@
 from xml.dom import minidom
+import itertools as it
+import re
 import warnings
 
 from constants import *

mobject/tex_mobject.py

@@ -7,6 +7,7 @@ from utils.config_ops import digest_config
 
 import collections
 import sys
+import operator as op
 
 TEX_MOB_SCALE_FACTOR = 0.05
 

mobject/vectorized_mobject.py

@@ -1,8 +1,9 @@
 import re
+from colour import Color
 
 from constants import *
-from mobject.mobject import Mobject
-from utils.bezier import bezier, partial_bezier_points, 
+from .mobject import Mobject
+from utils.bezier import bezier, partial_bezier_points
 from utils.bezier import interpolate, get_smooth_handle_points, is_closed
 from utils.color import color_to_rgb
 from utils.color import interpolate_color

old_projects/uncertainty.py

@@ -2,7 +2,7 @@
 
 import scipy
 from big_ol_pile_of_manim_imports import *
-
+from old_projects.fourier import *
 
 FREQUENCY_COLOR = RED
 USE_ALMOST_FOURIER_BY_DEFAULT = False

scene/moving_camera_scene.py

@@ -1,7 +1,7 @@
 from constants import *
 
-from camera import MovingCamera
-from scene.scene import Scene
+from camera.camera import MovingCamera
+from .scene import Scene
 from topics.geometry import ScreenRectangle
 
 class MovingCameraScene(Scene):

scene/reconfigurable_scene.py

@@ -1,6 +1,6 @@
 import numpy as np
 
-from scene.scene import Scene
+from .scene import Scene
 from animation.transform import Transform
 from mobject.mobject import Mobject
 

scene/scene.py

@@ -10,13 +10,15 @@ import copy
 from tqdm import tqdm as ProgressDisplay
 import inspect
 import subprocess as sp
+import random
 
 from constants import *
 
 from camera.camera import Camera
 from tk_scene import TkSceneRoot
-from mobject.mobject import Mobject, VMobject
-from animation import Animation
+from mobject.mobject import Mobject
+from mobject.vectorized_mobject import VMobject
+from animation.animation import Animation
 from animation.transform import MoveToTarget
 from animation.continual_animation import ContinualAnimation
 from utils.iterables import list_update

scene/scene_from_video.py

@@ -3,7 +3,7 @@ import cv2
 import itertools as it
 from tqdm import tqdm as show_progress
 
-from scene.scene import Scene
+from .scene import Scene
 
 
 class SceneFromVideo(Scene):

scene/zoomed_scene.py

@@ -1,10 +1,11 @@
 import numpy as np
 
-from scene.scene import Scene
+from .scene import Scene
 from animation.transform import FadeIn
 from mobject.mobject import Mobject
 from topics.geometry import Rectangle
-from camera import MovingCamera, Camera
+from camera.camera import Camera
+from camera.camera import MovingCamera
 
 from constants import *
 

topics/arithmetic.py

@@ -3,7 +3,7 @@ import itertools as it
 
 from constants import *
 from scene.scene import Scene
-from animation import Animation
+from animation.animation import Animation
 from mobject.tex_mobject import TexMobject
 
 class RearrangeEquation(Scene):

topics/characters.py

@@ -1,6 +1,10 @@
+import random
+import numpy as np
+import itertools as it
+
 from constants import *
 
-from mobject.mobject import Mobject
+from mobject.mobject import Mobject, Group
 from mobject.svg_mobject import SVGMobject
 from mobject.vectorized_mobject import VMobject, VGroup
 from mobject.tex_mobject import TextMobject, TexMobject
@@ -8,8 +12,9 @@ from mobject.tex_mobject import TextMobject, TexMobject
 from topics.objects import Bubble, ThoughtBubble, SpeechBubble
 from topics.geometry import ScreenRectangle
 
-from animation import Animation
-from animation.transform import *
+from animation.animation import Animation
+from animation.transform import Transform, ApplyMethod, MoveToTarget
+from animation.transform import ReplacementTransform, FadeOut, FadeIn
 from animation.simple_animations import Write, ShowCreation
 from animation.compositions import AnimationGroup
 from scene.scene import Scene

topics/common_scenes.py

@@ -2,7 +2,7 @@
 from constants import *
 
 from scene.scene import Scene
-from animation import Animation
+from animation.animation import Animation
 from animation.simple_animations import Write, DrawBorderThenFill
 from animation.compositions import LaggedStart
 from animation.transform import FadeIn, FadeOut, ApplyMethod

topics/complex_numbers.py

@@ -1,10 +1,10 @@
 from constants import *
 
 
-from mobject import VGroup
+from mobject.vectorized_mobject import VGroup
 from mobject.tex_mobject import TexMobject, TextMobject
 from number_line import NumberPlane
-from animation import Animation
+from animation.animation import Animation
 from animation.transform import ApplyPointwiseFunction, MoveToTarget
 from animation.simple_animations import Homotopy, ShowCreation, \
     SmoothedVectorizedHomotopy

topics/geometry.py

@@ -1,5 +1,8 @@
 from constants import *
 
+import itertools as it
+import numpy as np
+
 from mobject.mobject import Mobject
 from mobject.vectorized_mobject import VMobject, VGroup
 from utils.bezier import interpolate

topics/number_line.py

@@ -1,6 +1,5 @@
 from constants import *
 
-from mobject.mobject import Mobject1D
 from mobject.vectorized_mobject import VMobject, VGroup
 from mobject.tex_mobject import TexMobject
 from topics.geometry import Line, Arrow

topics/numerals.py

@@ -1,7 +1,7 @@
 
 from mobject.vectorized_mobject import VMobject, VGroup, VectorizedPoint
 from mobject.tex_mobject import TexMobject
-from animation import Animation
+from animation.animation import Animation
 from animation.continual_animation import ContinualAnimation
 from topics.geometry import BackgroundRectangle
 from scene.scene import Scene

topics/objects.py

@@ -5,7 +5,7 @@ from mobject.vectorized_mobject import VGroup, VMobject, VectorizedPoint
 from mobject.svg_mobject import SVGMobject
 from mobject.tex_mobject import TextMobject, TexMobject, Brace
 
-from animation import Animation
+from animation.animation import Animation
 from animation.simple_animations import Rotating
 from animation.compositions import LaggedStart, AnimationGroup
 from animation.transform import ApplyMethod, FadeIn, GrowFromCenter

topics/vector_space_scene.py

@@ -4,7 +4,7 @@ from scene.scene import Scene
 from mobject.mobject import Mobject
 from mobject.vectorized_mobject import VMobject, VGroup
 from mobject.tex_mobject import TexMobject, TextMobject
-from animation import Animation    
+from animation.animation import Animation    
 from animation.transform import ApplyPointwiseFunction, Transform, \
     ApplyMethod, FadeOut, ApplyFunction
 from animation.simple_animations import ShowCreation, Write

utils/bezier.py

@@ -2,7 +2,7 @@ import numpy as np
 from scipy import linalg
 from utils.simple_functions import choose
 
-CLOSED_THRESHOLD = 0.01
+CLOSED_THRESHOLD = 0.0
 
 def bezier(points):
     n = len(points) - 1
@@ -123,4 +123,4 @@ def diag_to_matrix(l_and_u, diag):
     return matrix
 
 def is_closed(points):
-    return np.linalg.norm(points[0] - points[-1]) < CLOSED_THRESHOLDp
+    return np.linalg.norm(points[0] - points[-1]) < CLOSED_THRESHOLD

utils/color.py

@@ -1,5 +1,5 @@
 from colour import Color
-from numpy import np
+import numpy as np
 import random
 
 from utils.bezier import interpolate

utils/paths.py

@@ -1,7 +1,9 @@
 import numpy as np
 from utils.bezier import interpolate
 from utils.space_ops import rotation_matrix
+from constants import OUT
 
+STRAIGHT_PATH_THRESHOLD = 0.01
 
 def straight_path(start_points, end_points, alpha):
     """

utils/rate_functions.py

@@ -1,5 +1,6 @@
 import numpy as np
 from utils.simple_functions import sigmoid
+from utils.bezier import bezier
 
 def smooth(t, inflection = 10.0):
     error = sigmoid(-inflection / 2)