week2_task.py

# Write a program that reads the state of all four protoboard buttons and uses the state
# to control protoboard’s LEDs. Use polling to read the protoboard buttons SW0 – SW2
# state and use interrupt for the rotary encoder push button.

# Program should work as follows:
# SW0 – SW2 are used to toggle LEDs D1 – D3 on and off. Each button controls one LED. 
# When a button is pressed the state of the controlled LED is toggled. 
# Each press should be counted only once; pressing and holding the button may not cause the LED to blink. 
# Rotary encoder’s button presses should be detected by using an interrupt. 
# When the rotary encoder’s button is pressed all LEDs are switched off.
# The state of the LEDs is displayed on the OLED screen. State of each LED is displayed on a separate line. 
# Note that when a pin is configured as an output you can also read the state of the pin. 
# OLED methods may not be called from an interrupt handler.

# For example, if you press SW0 and then SW2 then LEDs D1 and D3 will turn on.
# When you press rotary encoder button, they both switch off. 
# When you press SW0 again D1 switches on and switches off on the following press.

import ssd1306
from machine import Pin, I2C
from utime import sleep_ms

led1 = Pin(20, Pin.OUT)
led2 = Pin(21, Pin.OUT)
led3 = Pin(22, Pin.OUT)

button = Pin(12, mode = Pin.IN, pull = Pin.PULL_UP)
button1 = Pin(9, mode = Pin.IN, pull = Pin.PULL_UP)
button2 = Pin(8, mode = Pin.IN, pull = Pin.PULL_UP)
button3 = Pin(7, mode = Pin.IN, pull = Pin.PULL_UP)

i2c = I2C(1, sda=Pin(14), scl=Pin(15))
oled = ssd1306.SSD1306_I2C(128, 64, i2c)
        
def button_handler(pin):
    if pin.value() == 0:
        led1.value(0)
        led2.value(0)
        led3.value(0)
        
button.irq(handler = button_handler, trigger = button.IRQ_FALLING)

btn1_state = button1.value()
btn2_state = button2.value()
btn3_state = button3.value()

while True:
    sleep_ms(10)
    btn1_new_state = button1.value()
    btn2_new_state = button2.value()
    btn3_new_state = button3.value()
    
    if btn1_state != btn1_new_state:
        count1 += 1
        if count1 > 5:
            btn1_state = btn1_new_state
            count1 = 0
            if btn1_new_state == 0:
                led1.toggle()
                if led1.value() == 1:
                    oled.text('LED1 ON', 0, 0, 1)
                    oled.show()
                else:
                    oled.text('LED1 ON', 0, 0, 0)
                    oled.show()
    else:
        count1 = 0
        
    if btn2_state != btn2_new_state:
        count2 += 1
        if count2 > 5:
            btn2_state = btn2_new_state
            count2 = 0
            if btn2_new_state == 0:
                led2.toggle()
                if led2.value() == 1:
                    oled.text('LED2 ON', 0, 15, 1)
                    oled.show()
                else:
                    oled.text('LED2 ON', 0, 15, 0)
                    oled.show()
    else:
        count2 = 0
        
    if btn3_state != btn3_new_state:
        count3 += 1
        if count3 > 5:
            btn3_state = btn3_new_state
            count3 = 0
            if btn3_new_state == 0:
                led3.toggle()
                if led3.value() == 1:
                    oled.text('LED3 ON', 0, 30, 1)
                    oled.show()
                else:
                    oled.text('LED3 ON', 0, 30, 0)
                    oled.show()
    else:
        count3 = 0
        
    if button.value() == 0:
        oled.fill(0)
        oled.show()