main.py -- put your code here!

from pyb import OLED, delay, Pin

Create OLED Object

oled = OLED()

Clear OLED Cache

oled.fill(0) oled.show()

Create Coordinate Variable

x = 0 y = 16

Create function to draw sine wave

def draw_sine(): oled.fill(0) for i in range(128): y = int(16 + 8 * math.sin(2 * math.pi * i / 128)) oled.pixel(i, y, 1) oled.show()

Create function to draw cosine wave

def draw_cosine(): oled.fill(0) for i in range(128): y = int(16 + 8 * math.cos(2 * math.pi * i / 128)) oled.pixel(i, y, 1) oled.show()

Create K1 and K2 buttons

k1 = Pin('K1', Pin.IN, Pin.PULL_UP) k2 = Pin('K2', Pin.IN, Pin.PULL_UP)

Initialize variables

draw_sine_flag = True

while True: # Check if K1 button is pressed if not k1.value(): draw_sine_flag = True draw_sine() while not k1.value(): pass

# Check if K2 button is pressed
if not k2.value():
    draw_sine_flag = False
    draw_cosine()
    while not k2.value():
        pass

# If drawing sine wave, update x coordinate and draw
if draw_sine_flag:
    x -= 1
    if x < -128:
        x = 0
    draw_sine()
    delay(500