from time import sleep
from rp2 import PIO
from machine import ADC, Pin, PWM


sleep(0.1) # Wait for USB to become ready

# Setting up the PWM on pin 12
p12 = Pin(12)
pwm12 = PWM(p12)
pwm12.freq(50)
pwm12.duty_u16(0x7fff) #50%

# Configuring the ADC at pin 27
p27 = Pin(27)
potentiometer = ADC(p27)


@rp2.asm_pio(out_init=[PIO.OUT_LOW]*8, sideset_init=[PIO.OUT_LOW]*2)
def sevseg():
    wrap_target()
    label("0")
    pull(noblock)           .side(0)      # 0
    mov(x, osr)             .side(0)      # 1
    out(pins, 24)           .side(1)      # 2
    out(pins, 8)            .side(2)      # 3
    jmp("0")                .side(0)      # 6
    wrap()
  
sm = rp2.StateMachine(0, sevseg, freq=2000, out_base=Pin(2), sideset_base=Pin(0))
sm.active(1)

digits = [
  0b11000000, # 0
  0b11111001, # 1
  0b10100100, # 2 
  0b10110000, # 3
  0b10011001, # 4
  0b10010010, # 5
  0b10000010, # 6
  0b11111000, # 7
  0b10000000, # 8
  0b10011000, # 9
]

fulscale = 0b10001110

def segmentize(num):
  return (
    digits[num % 10] 
    | digits[num // 10 % 10] << 8 
  )


while True:
    val = potentiometer.read_u16()
    percentage = (val * 100)/0xFFFF
    print(f'Potentiometer {val} | {percentage:.2f}%')
    pwm12.duty_u16(val)

    # Display percentage on segments
    intpercentage = int(percentage)
    if(intpercentage != 100):
        sm.put(segmentize(intpercentage));
    else:
        sm.put(fulscale | fulscale << 8)
    
    sleep(0.1)