3b1b-manim/topics/functions.py

from scipy import integrate

from mobject.vectorized_mobject import VMobject

from helpers import *

class FunctionGraph(VMobject):
    CONFIG = {
        "color" : BLUE_D,
        "x_min" : -SPACE_WIDTH,
        "x_max" : SPACE_WIDTH,
        "space_unit_to_num" : 1,
        "epsilon" : 0.5,
    }
    def __init__(self, function, **kwargs):
        self.function = function
        VMobject.__init__(self, **kwargs)

    def generate_points(self):
        self.set_anchor_points([
            x*RIGHT + self.function(x)*UP
            for pre_x in np.arange(self.x_min, self.x_max, self.epsilon)
            for x in [self.space_unit_to_num*pre_x]
        ], mode = "smooth")


class ParametricFunction(VMobject):
    CONFIG = {
        "t_min" : 0,
        "t_max" : 1,
        "epsilon" : 0.1,
    }
    def __init__(self, function, **kwargs):
        self.function = function
        VMobject.__init__(self, **kwargs)

    def generate_points(self):
        self.set_anchor_points([
            self.function(t)
            for t in np.arange(self.t_min, self.t_max, self.epsilon)
        ], mode = "smooth")


class Axes(VMobject):
    def generate_points(self):
        self.x_axis = NumberLine(**kwargs)
        self.y_axis = NumberLine(**kwargs).rotate(np.pi/2)
        self.add(self.x_axis, self.y_axis)