mirror of
https://github.com/3b1b/manim.git
synced 2025-08-01 17:29:06 +08:00
51 lines
1.3 KiB
Python
51 lines
1.3 KiB
Python
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)
|
|
|
|
|
|
|