from machine import Pin, ADC, Timer
from time import sleep, sleep_us
import LCD

B0 = Pin(15, Pin.IN, Pin.PULL_UP)
B1 = Pin(14, Pin.IN, Pin.PULL_UP)

a2d2 = ADC(2)
kV = 3.3 / 65535

Navy = LCD.RGB(0,0,5)
Yellow = LCD.RGB(200,200,0)
LCD.Init()
LCD.Clear(Navy)
time = 0

x = [0]*1000
y = [0]*1000
raw = [0]*1000

while(1):
    for i in range(0,1000):
        raw[i] = a2d2.read_u16()
        sleep_us(10)
    
    for i in range(0,1000):   
        Volts = raw[i] * kV
        x[i] = i    
        y[i] = Volts

    Navy = LCD.RGB(0,0,5)
    White = LCD.RGB(200,200,200)
    LCD.Clear(Navy)
    LCD.Plot(x,y)
    LCD.Title('Current',White, Navy)
    
    while(B0.value() & B1.value()):
        pass
    if(B0.value() == 0):
        print('Writing to File')
        file1 = open("Current_LED_CFL.txt", "w")
        for i in range(0,1000):
 #           file1.write(str('{: 7.3f}'.format(x[i]*T)))
            file1.write(str('{: 7.4f}'.format(y[i])))
            file1.write("\n")
        file1.close()


    sleep(1)