#!/usr/bin/env python
import matplotlib.pyplot as plt
import numpy as np

amplitude = 0.8

dac_rate = 10000 #Hz
sample_rate = 1000 #Hz
freq = 5 #Hz
duration = 2 #seconds

num_samples = sample_rate * duration

sine_table = (amplitude * 32767 * np.sin(2.0 * np.pi * freq * np.arange(num_samples) / sample_rate)).astype("i2")

sine_table.tofile("sine.dat")

plt.plot(sine_table)
plt.show()

print("Now run:")
print(f"target/release/fl2k_ampliphase -D -c 2000 -s {dac_rate} -i {sample_rate} -f sine.dat -w sine")
print("And then")
print("./plot.py")