import Nice
import Basda
import ArgosBasdaServiceVibrationCompensation as VibComp
import numpy as np
import matplotlib.pyplot as pt

filename = raw_input('Dump file name: ')

reader= VibComp.ArchiveReader(filename)
reader.init()
record= reader.loadNextRecord()

accell1 = list()
accell2 = list()
accell3 = list()
accell4 = list()
accell5 = list()
accell6 = list()
accell7 = list()
accell8 = list()
accell_x = list()
accell_y = list()
fsm_x = list()
fsm_y = list()

while record is not None:
    accell1.append(record.getAccelerometerSeq()[0])
    accell2.append(record.getAccelerometerSeq()[1])
    accell3.append(record.getAccelerometerSeq()[2])
    accell4.append(record.getAccelerometerSeq()[3])
    accell5.append(record.getAccelerometerSeq()[4])
    accell6.append(record.getAccelerometerSeq()[5])
    accell7.append(record.getAccelerometerSeq()[6])
    accell8.append(record.getAccelerometerSeq()[7])
    accell_x.append(record.getFSMOutputSeq()[0])
    accell_y.append(record.getFSMOutputSeq()[1])
    fsm_x.append(record.fsmReadout1)
    fsm_y.append(record.fsmReadout2)
    record= reader.loadNextRecord()


pt.figure(1,(20,10))
pt.subplot(3,4,1)
pt.plot(accell1)
pt.title('Accelerometer 1')
pt.subplot(3,4,2)
pt.plot(accell2)
pt.title('Accelerometer 2')
pt.subplot(3,4,3)
pt.plot(accell3)
pt.title('Accelerometer 3')
pt.subplot(3,4,4)
pt.plot(accell4)
pt.title('Accelerometer 4')
pt.subplot(3,4,5)
pt.plot(accell5)
pt.title('Accelerometer 5')
pt.subplot(3,4,6)
pt.plot(accell6)
pt.title('Accelerometer 6')
pt.subplot(3,4,7)
pt.plot(accell7)
pt.title('Accelerometer 7')
pt.subplot(3,4,8)
pt.plot(accell8)
pt.title('Accelerometer 8')
pt.subplot(3,4,9)
pt.plot(accell_x,'r')
pt.title('Accelerometer output X')
pt.subplot(3,4,10)
pt.plot(accell_y,'r')
pt.title('Accelerometer output Y')
pt.subplot(3,4,11)
pt.plot(fsm_x,'g')
pt.title('FSM feedback X')
pt.subplot(3,4,12)
pt.plot(fsm_y,'g')
pt.title('FSM feedback Y')
pt.show(block=False)

pt.figure(2)
pt.plot(np.fft.fft(accell1)[1:200])
pt.plot(np.fft.fft(accell2)[1:200])
pt.plot(np.fft.fft(accell3)[1:200])
pt.plot(np.fft.fft(accell4)[1:200])
pt.plot(np.fft.fft(accell5)[1:200])
pt.plot(np.fft.fft(accell6)[1:200])
pt.plot(np.fft.fft(accell7)[1:200])
pt.plot(np.fft.fft(accell8)[1:200],linestyle='--')
pt.legend(('accell1','accell2','accell3','accell4','accell5','accell6','accell7','accell8'))
pt.show(block=False)