import numpy as np
import matplotlib.pyplot as plt
from scipy.interpolate import griddata
import math

RawData = "Tube.DAT"
f = open(RawData, "r")
a = np.fromfile(f, dtype=np.uint32)
t = a[-1]
V = a[:-1].copy()

Fech = 100000.0*len(V)/t # in MHz

print Fech

25.6174237174

#ADC1_pins = [21,22,19,15,13,11,12,10, 8]
ADC1_pins = [21,22,19,15,13,11,12,10, 8]
ADC2_pins = [38,37,36,35,33,32,26,24,23]

#ADC1_GPIO = [9, 25,10,22,27,17,18,15,14]
ADC1_GPIO =  [9 ,25,10,22,27,17,18,15,14]

ADC2_GPIO = [20,26,16,19,13,12, 7, 8,11]

ADC1len = len(ADC1_GPIO)
ADC2len = len(ADC2_GPIO)


def GetV2(Volts):
    Signal = []
    Map = np.zeros((len(V),ADC2len), dtype=np.int)
    for i in range(len(Volts)):
        val = int(V[i])
        SignalZero = 0
        for k in range(ADC2len):
            Map[i][k] = (val & 2**k)/2**k
        for k in range(ADC2len):
            SignalZero += 2**k*((val & 2**ADC2_GPIO[k])/2**ADC2_GPIO[k])
        Signal.append(SignalZero)


    return Signal,Map



def GetV1(Volts):
    Signal = []
    Map = np.zeros((len(V),ADC1len), dtype=np.int)
    for i in range(len(Volts)):
        val = int(V[i])
        SignalZero = 0
        for k in range(ADC1len):
            Map[i][k] = (val & 2**k)/2**k
        for k in range(ADC1len):
            SignalZero += 2**k*((val & 2**ADC1_GPIO[k])/2**ADC1_GPIO[k])
        Signal.append(SignalZero)


    return Signal,Map

M = GetV2(V)[0]
M2 = GetV1(V)[0]
Ma = M - np.average(M[5000:7000])
Mb = M2 - np.average(M2[5000:7000])

rawSig = []
for k in range(len(Ma)):
    rawSig.append(Mb[k])
    rawSig.append(Ma[k])
rawSig = rawSig - np.average(rawSig)

plt.figure(figsize=(15,5))


#plt.plot(rawSig[10000:15000],"c") 

plt.plot(rawSig[80000:81000],"g") 
plt.plot(rawSig[70000:71000],"y")
plt.plot(rawSig[60000:61000],"b") 

plt.plot(rawSig[0000:1000],"c") 
plt.plot(rawSig[50000:51000],"r") 
plt.plot(rawSig[90000:91000],"y")
plt.plot(rawSig[40000:41000],"r") 
plt.plot(rawSig[30000:31000],"g") 

plt.title('9 Pulses from '+RawData.split(".")[0])
plt.savefig('Pulses_'+RawData.split(".")[0]+'.jpg', bbox_inches='tight')
plt.show()

plt.figure(figsize=(15,5))

plt.plot(rawSig[50800:51000],"r") 
plt.plot(rawSig[70800:71000],"y")
plt.plot(rawSig[40800:41000],"r") 
plt.plot(rawSig[30800:31000],"g") 

plt.title('4 Pulses from '+RawData.split(".")[0])
plt.savefig('Pulses_details_'+RawData.split(".")[0]+'.jpg', bbox_inches='tight')
plt.show()

plt.figure(figsize=(15,5))

plt.plot(rawSig[50800:51000]+rawSig[70800:71000]+rawSig[40800:41000]+rawSig[30800:31000],"r") 

plt.title('Averaged pulses from '+RawData.split(".")[0])
plt.savefig('Pulses_average_'+RawData.split(".")[0]+'.jpg', bbox_inches='tight')
plt.show()