import sigplot as sp
import matplotlib
import matplotlib.pyplot as plt
import numpy as np


matplotlib.rcParams['toolbar'] = 'None'
plt.style.use('dark_background')

fig = plt.figure()

# seed = np.linspace(3, 7, 1000)
# a = (np.sin(2 * np.pi * seed))
# b = (np.cos(2 * np.pi * seed))
# sp.correlate(fig, b, a, 300)

t = np.linspace(0, 1, 500)
b = (np.cos(2 * np.pi * t))
# x = np.concatenate([np.zeros(500), signal.sawtooth(2 * np.pi * 5 * t), np.zeros(500), np.ones(120), np.zeros(500)])
x = np.concatenate([np.zeros(500), np.ones(500), np.zeros(500)])
sp.fourier_series(fig, x, 100, 200, 200)

plt.show()

# WriteToVideo("twoPulse.mp4", anim);

[(8, 'matplotlib.pyplot.style.use', 'plt.style.use', True, 'import matplotlib.pyplot as plt\n'), (10, 'matplotlib.pyplot.figure', 'plt.figure', True, 'import matplotlib.pyplot as plt\n'), (17, 'numpy.linspace', 'np.linspace', True, 'import numpy as np\n'), (18, 'numpy.cos', 'np.cos', True, 'import numpy as np\n'), (21, 'sigplot.fourier_series', 'sp.fourier_series', True, 'import sigplot as sp\n'), (23, 'matplotlib.pyplot.show', 'plt.show', True, 'import matplotlib.pyplot as plt\n'), (20, 'numpy.zeros', 'np.zeros', True, 'import numpy as np\n'), (20, 'numpy.ones', 'np.ones', True, 'import numpy as np\n'), (20, 'numpy.zeros', 'np.zeros', True, 'import numpy as np\n')]