"""
Created on Tue Dec 24 20:25 2019
@author: anne marie delaney
         eoin brophy
         
Script to generate a training and test data set of sine waves
"""


import pandas as pd
import numpy as np
import random

"""Create a training set of sine waves with 10000 records"""
a = np.arange(0.1,0.9,0.02)
x = np.arange(0,20,0.5)
r = np.arange(2,6.1,0.1)
count = 0
fs = len(x)
y = np.zeros((1,len(x)))

for n in range(10000):
  amp = a[random.randint(0,len(a)-1)]
  rad = r[random.randint(0,len(r)-1)]
  phase = random.uniform(-1,1)*np.pi
  y = np.append(y,amp*np.sin(((2*np.pi*rad*x)+phase)/fs).reshape((1,len(x))),axis = 0)
     
data = pd.DataFrame(y[1:][:])  
data.to_csv('./sinedata_v2.csv', header = False, index = False)

"""Creating a test set of sine waves with 3000 records"""
a = np.arange(0.1,0.9,0.02)
x = np.arange(0,20,0.5)
r = np.arange(2,6.1,0.1)
count = 0
fs = len(x)
y = np.zeros((1,len(x)))

for n in range(3000):
  amp = a[random.randint(0,len(a)-1)]
  rad = r[random.randint(0,len(r)-1)]
  phase = random.uniform(-1,1)*np.pi

  y = np.append(y,amp*np.sin(((2*np.pi*rad*x)+phase)/fs).reshape((1,len(x))),axis = 0)
  
data = pd.DataFrame(y[1:][:])  
data.to_csv('sinedata_test_v2.csv', header = False, index = False)