import matplotlib.pyplot as plt

import numpy as np

import wave

from scipy import signal

import sys

import statistics

Fs = 44100

f = 1000

ampLow = 200

ampHigh = 6000

hours = 14

minutes = 0

spf = wave.open('data/oneminch1.wav','r')

#Extract Raw Audio from Wav File

wwvbSignal = spf.readframes(-1)

wwvbSignal = np.fromstring(wwvbSignal, 'Int16')

#5th Second (BPSK 0) (Working CC)

wwvbZero = wwvbSignal[182000:226100:1]

#3rd Second (BPSK 1)

#wwvbZeroB = wwvbSignal[93800:137900:1]

#34th Second (BPSK Unsure) (Working CC)

wwvbOne = wwvbSignal[1460026:1504126:1]

#1rst Second (BPSK 0) (Working CC)

wwvbX = wwvbSignal[4800:48900:1]

def createZero(Fs, f, ampHigh, ampLow):

return second

def createOne(Fs, f, ampHigh, ampLow):

return second

def createX(Fs, f, ampHigh, ampLow):

return second

expectedZero = np.array(createZero(44100,1000, 6000, 200))

expectedOne = np.array(createOne(44100,1000, 6000, 200))

expectedX = np.array(createX(44100,1000, 6000, 200))

#Function that generatedd the expected signal

#Everything will remain Constant Exept for Hour and Minute

#Used to create One minute Expected Signals of AM during August 21, 2017

def expectedSignal(hoursS, minutesS, Zero, One, X):

return wwvbExpected

def getMinutes(minutes):

return new

def getHours(hours):

return new

Zero = createZero(Fs, f, ampHigh, ampLow)

One = createOne(Fs, f, ampHigh, ampLow)

X = createX(Fs, f, ampHigh, ampLow)

minutesS = getMinutes(minutes)

hoursS = getHours(hours)

newSignal = expectedSignal(hoursS, minutesS, Zero, One, X)

#Checks to make sure that wwvbSignal and newSignal are same length

while len(wwvbSignal) > len(newSignal):

newSignal.append(0)

nS = np.array(newSignal)

bitStream = []

templates = [Zero, One, X]

thresholds = [500000000000, 300000000000]

values = ['0', '1', 'X']

#corrects for wwvb offset

print(len(wwvbSignal))

#temp = wwvbSignal[0:4800:1]

wwvbSignal = wwvbSignal[5500::]

wwvbSignal = np.array(wwvbSignal)

#temp = np.array(temp)

#wwvbSignal = np.concatenate((wwvbSignal, temp))

print(len(wwvbSignal))

#One min wwvb

bitStream = ""

devList = []

for i in range(59):

start = i *44100

end = (i*44100) + 44100

corrZero = signal.correlate(wwvbSignal[start:end], Zero, mode='full')

corrOne = signal.correlate(wwvbSignal[start:end], One, mode='full')

corrX = signal.correlate(wwvbSignal[start:end], X, mode='full')

numZero = int(max(corrZero[44078:44123:1]) / 1000000000)

numOne = int(max(corrOne[44078:44123:1]) / 1000000000)

numX = int(max(corrX[44078:44123:1]) / 1000000000)

if numOne > (numX *2): # we have one or zero

if numZero > (numOne + (numX * 0.8)):

bitStream = bitStream + '0'

else:

bitStream = bitStream + '1'

else:

bitStream = bitStream + 'X'

devList.append(numZero)

devList.append(numOne)

devList.append(numX)

devList.append("||")

print(devList)

print(bitStream)

'''

#print(bitStream)

plt.figure(1)

plt.title('Expected Signal with AM Modulation (14:00 UTC)')

plt.ylim((-10000, 10000))

plt.xlabel("Sample Rate = 44100Hz | (60 Seconds)")

plt.plot(wwvbSignal)

plt.show()

'''