import matplotlib.pyplot as plt
import matplotlib.animation as animation
import numpy as np

class MathAnimation:
    def __init__(self, xlim=(-10, 10), ylim=(-10, 10)):
        self.fig, self.ax = plt.subplots()
        self.ax.set_xlim(xlim)
        self.ax.set_ylim(ylim)
        self.lines = []
        self.anim = None

    def add_function(self, func, color='b', label=None):
        x = np.linspace(self.ax.get_xlim()[0], self.ax.get_xlim()[1], 1000)
        y = func(x)
        line, = self.ax.plot(x, y, color=color, label=label)
        self.lines.append((line, func))

    def update(self, frame):
        for line, func in self.lines:
            x = np.linspace(self.ax.get_xlim()[0], self.ax.get_xlim()[1], 1000)
            y = func(x + frame / 10.0)  # Simple shift animation
            line.set_ydata(y)
        return [line for line, _ in self.lines]

    def animate(self, interval=50, frames=200, save_as_video=False):
        self.anim = animation.FuncAnimation(self.fig, self.update, frames=frames, interval=interval, blit=True)
        plt.legend()

        if save_as_video:
            self.anim.save('sinandcos.mp4', writer='ffmpeg', fps=30)
        plt.show()

if __name__ == "__main__":
    anim = MathAnimation()
    anim.add_function(lambda x: np.sin(x), color='r', label='sin(x)')
    anim.add_function(lambda x: np.cos(x), color='g', label='cos(x)')
    anim.animate(save_as_video=True)

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.animation as animation
from mpl_toolkits.mplot3d import Axes3D

# Generate a parametric curve
t = np.linspace(0, 4 * np.pi, 100)
x = np.sin(t)
y = np.cos(t)
z = t / (4 * np.pi)

fig = plt.figure()
ax = fig.add_subplot(111, projection='3d')
line, = ax.plot([], [], [], 'r', lw=2)

ax.set_xlim([-1, 1])
ax.set_ylim([-1, 1])
ax.set_zlim([0, 1])
ax.set_xlabel("X-axis")
ax.set_ylabel("Y-axis")
ax.set_zlabel("Z-axis")
ax.set_title("3D Animated Parametric Curve")

# Animation function
def update(num):
    line.set_data(x[:num], y[:num])
    line.set_3d_properties(z[:num])
    return line,
ani = animation.FuncAnimation(fig, update, frames=len(t), interval=50, blit=False)
writer = animation.FFMpegWriter(fps=30)
ani.save('animatedcurve.mp4', writer=writer)
plt.show()