from machine import Pin, I2C
import ssd1306
import array 
import math

# ESP32 Pin assignment 
i2c = I2C(0, scl=Pin(22), sda=Pin(21))

oled_width = 128
oled_height = 64
oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c)

class Poligono(): # Dibuja un polígono regular
    def __init__(self, centro, lados, radio, rotacion):
        self.xc = centro[0]
        self.yc = centro[1]
        self.lados = lados
        self.radio = radio
        self.rotacion = rotacion
    def calcula_vertices(self):
        vertices=[]
        for i in range (5):
            grados=((2*math.pi*i)/self.lados)-self.rotacion
            ang=grados
            xn=int(self.xc*self.radio*(math.sin(ang)))
            vertices.append(xn)
            yn=int(self.yc*self.radio*(math.cos(ang)))
            vertices.append(yn)
        return(vertices)
    def dibuja(self): # Dibuja el polígono en la pantalla
        oled.poly(0, 0, array.array('h', self.calcula_vertices()))

pol=Poligono([64,32],5,31,0)
pol.dibuja()
oled.show()

main.py

#MicroPython SSD1306 OLED driver, I2C and SPI interfaces created by Adafruit

import time
import framebuf

# register definitions
SET_CONTRAST        = const(0x81)
SET_ENTIRE_ON       = const(0xa4)
SET_NORM_INV        = const(0xa6)
SET_DISP            = const(0xae)
SET_MEM_ADDR        = const(0x20)
SET_COL_ADDR        = const(0x21)
SET_PAGE_ADDR       = const(0x22)
SET_DISP_START_LINE = const(0x40)
SET_SEG_REMAP       = const(0xa0)
SET_MUX_RATIO       = const(0xa8)
SET_COM_OUT_DIR     = const(0xc0)
SET_DISP_OFFSET     = const(0xd3)
SET_COM_PIN_CFG     = const(0xda)
SET_DISP_CLK_DIV    = const(0xd5)
SET_PRECHARGE       = const(0xd9)
SET_VCOM_DESEL      = const(0xdb)
SET_CHARGE_PUMP     = const(0x8d)


class SSD1306:
    def __init__(self, width, height, external_vcc):
        self.width = width
        self.height = height
        self.external_vcc = external_vcc
        self.pages = self.height // 8
        # Note the subclass must initialize self.framebuf to a framebuffer.
        # This is necessary because the underlying data buffer is different
        # between I2C and SPI implementations (I2C needs an extra byte).
        self.poweron()
        self.init_display()

    def init_display(self):
        for cmd in (
            SET_DISP | 0x00, # off
            # address setting
            SET_MEM_ADDR, 0x00, # horizontal
            # resolution and layout
            SET_DISP_START_LINE | 0x00,
            SET_SEG_REMAP | 0x01, # column addr 127 mapped to SEG0
            SET_MUX_RATIO, self.height - 1,
            SET_COM_OUT_DIR | 0x08, # scan from COM[N] to COM0
            SET_DISP_OFFSET, 0x00,
            SET_COM_PIN_CFG, 0x02 if self.height == 32 else 0x12,
            # timing and driving scheme
            SET_DISP_CLK_DIV, 0x80,
            SET_PRECHARGE, 0x22 if self.external_vcc else 0xf1,
            SET_VCOM_DESEL, 0x30, # 0.83*Vcc
            # display
            SET_CONTRAST, 0xff, # maximum
            SET_ENTIRE_ON, # output follows RAM contents
            SET_NORM_INV, # not inverted
            # charge pump
            SET_CHARGE_PUMP, 0x10 if self.external_vcc else 0x14,
            SET_DISP | 0x01): # on
            self.write_cmd(cmd)
        self.fill(0)
        self.show()

    def poweroff(self):
        self.write_cmd(SET_DISP | 0x00)

    def contrast(self, contrast):
        self.write_cmd(SET_CONTRAST)
        self.write_cmd(contrast)

    def invert(self, invert):
        self.write_cmd(SET_NORM_INV | (invert & 1))

    def show(self):
        x0 = 0
        x1 = self.width - 1
        if self.width == 64:
            # displays with width of 64 pixels are shifted by 32
            x0 += 32
            x1 += 32
        self.write_cmd(SET_COL_ADDR)
        self.write_cmd(x0)
        self.write_cmd(x1)
        self.write_cmd(SET_PAGE_ADDR)
        self.write_cmd(0)
        self.write_cmd(self.pages - 1)
        self.write_framebuf()

    def fill(self, col):
        self.framebuf.fill(col)

    def pixel(self, x, y, col):
        self.framebuf.pixel(x, y, col)

    def scroll(self, dx, dy):
        self.framebuf.scroll(dx, dy)

    def text(self, string, x, y, col=1):
        self.framebuf.text(string, x, y, col)

    def hline(self, x, y, w, col = 1):
        self.framebuf.hline(x, y, w, col)

    def vline(self, x, y, h, col = 1):
        self.framebuf.vline(x, y, h, col)

    def line(self, x1, y1, x2, y2, col = 1):
        self.framebuf.line(x1, y1, x2, y2, col)

    def rect(self, x, y, w, h, col = 1, fill = 0):
        self.framebuf.rect(x, y, w, h, col, fill)
    
    def poly(self, x, y, coords, col = 1, fill = 0):
        self.framebuf.poly(x, y, coords, col, fill)

class SSD1306_I2C(SSD1306):
    def __init__(self, width, height, i2c, addr=0x3c, external_vcc=False):
        self.i2c = i2c
        self.addr = addr
        self.temp = bytearray(2)
        # Add an extra byte to the data buffer to hold an I2C data/command byte
        # to use hardware-compatible I2C transactions.  A memoryview of the
        # buffer is used to mask this byte from the framebuffer operations
        # (without a major memory hit as memoryview doesn't copy to a separate
        # buffer).
        self.buffer = bytearray(((height // 8) * width) + 1)
        self.buffer[0] = 0x40  # Set first byte of data buffer to Co=0, D/C=1
        self.framebuf = framebuf.FrameBuffer1(memoryview(self.buffer)[1:], width, height)
        super().__init__(width, height, external_vcc)

    def write_cmd(self, cmd):
        self.temp[0] = 0x80 # Co=1, D/C#=0
        self.temp[1] = cmd
        self.i2c.writeto(self.addr, self.temp)

    def write_framebuf(self):
        # Blast out the frame buffer using a single I2C transaction to support
        # hardware I2C interfaces.
        self.i2c.writeto(self.addr, self.buffer)

    def poweron(self):
        pass


class SSD1306_SPI(SSD1306):
    def __init__(self, width, height, spi, dc, res, cs, external_vcc=False):
        self.rate = 10 * 1024 * 1024
        dc.init(dc.OUT, value=0)
        res.init(res.OUT, value=0)
        cs.init(cs.OUT, value=1)
        self.spi = spi
        self.dc = dc
        self.res = res
        self.cs = cs
        self.buffer = bytearray((height // 8) * width)
        self.framebuf = framebuf.FrameBuffer1(self.buffer, width, height)
        super().__init__(width, height, external_vcc)

    def write_cmd(self, cmd):
        self.spi.init(baudrate=self.rate, polarity=0, phase=0)
        self.cs.high()
        self.dc.low()
        self.cs.low()
        self.spi.write(bytearray([cmd]))
        self.cs.high()

    def write_framebuf(self):
        self.spi.init(baudrate=self.rate, polarity=0, phase=0)
        self.cs.high()
        self.dc.high()
        self.cs.low()
        self.spi.write(self.buffer)
        self.cs.high()

    def poweron(self):
        self.res.high()
        time.sleep_ms(1)
        self.res.low()
        time.sleep_ms(10)
        self.res.high()

ssd1306.py

# Clase Aplicacion ------------------------------
class Aplicacion():
    def __init__(self, oled, nombre, contenido, x, y):
        self.nombre = nombre
        self.contenido = contenido
        self.x = x
        self.y = y
        self.oled = oled
    def mostrar(self):          # Muestra el texto del argumento 'nombre' en (x,y)
        self.oled.text(self.nombre, self.x, self.y, 1)
        self.oled.show()
    def seleccionar(self):      # Encierra el texto en un rectángulo
        self.oled.rect(0, self.y - 2, 128, 12)
        self.oled.show()
    def deseleccionar(self):    # Borra el rectángulo
        self.oled.rect(0, self.y - 2, 128, 12, 0)
        self.oled.show()
    def entrar(self):           # Muestra el texto del argumento 'contenido'
        self.contenido()

# Función OLED ---------------------------------------
def OLED():
    from machine import Pin, I2C
    import ssd1306
    oled_width = 128
    oled_height = 64
    sclPin = 22
    sdaPin = 21        
    i2c = I2C(0, scl=Pin(sclPin), sda=Pin(sdaPin))
    oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c)
    return oled

# Clase Boton -------------------------------------
class Boton():
    def __init__(self, puerto):
        from machine import Pin
        self.b = Pin(puerto, Pin.IN, Pin.PULL_UP)
        self.edo_ant = 1
        self.edo_act = self.b.value()
    def bajada(self):       # Detecta flanco de bajada, devuelve False/True
        self.edo_act = self.b.value()
        if self.edo_act == 0 and self.edo_ant == 1:
            self.edo_ant = self.edo_act
            return True
        else:
            self.edo_ant = self.edo_act
            return False

clases.py

from time import sleep_ms
from clases import Aplicacion, Boton, OLED
from aplicaciones import app_puntos, app_keypad

# Creando lista de objetos de tipo Aplicacion
oled = OLED()

# Configurando botones
boton1 = Boton(18)
boton2 = Boton(19)

# Creando listas de aplicaciones
nombres = ["Puntos", "Keypad", "Generico", "Generico", "Generico", "Generico"]
contenidos = [app_puntos, app_keypad, app_puntos, app_puntos, app_puntos,
              app_puntos]
secuencia = [2, 13, 24, 35, 46, 57]

aplicaciones = []
for i in range(len(nombres)):
    aplicaciones.append(Aplicacion(oled, nombres[i], contenidos[i], 2, secuencia[i]))

# Función que muestra todas las aplicaciones y selecciona una (sel)
def menu(sel):
    oled.fill(0)
    for i in range(len(nombres)):
        aplicaciones[i].mostrar()
    aplicaciones[sel].seleccionar()
    
# --------------------------------------------------
# Programa principal

sel = 0
menu(sel)       # Muestra el menú con la primera app seleccionada
while True:
    sleep_ms(100)
    if boton1.bajada() == True:            # Si flanco de bajada (Botón 1)      
        aplicaciones[sel].deseleccionar()  # Deselecciona anterior
        if sel == len(aplicaciones) - 1:   
           sel = 0
        else:
            sel += 1
        aplicaciones[sel].seleccionar()    # Selecciona siguiente
    if boton2.bajada() == True:            # Si flanco de bajada (Botón 2)
        aplicaciones[sel].entrar()         # Entra a la app seleccionada
        menu(sel)                          # Cuando regresa, muestra el menú de nuevo
            
# --------------------------------------------------


Principal.py

from machine import Pin, I2C
import ssd1306
import array 
import math

# ESP32 Pin assignment 
i2c = I2C(0, scl=Pin(22), sda=Pin(21))

oled_width = 128
oled_height = 64
oled = ssd1306.SSD1306_I2C(oled_width, oled_height, i2c)

class Poligono(): # Dibuja un polígono regular
    def __init__(self, centro, lados, radio, rotacion):
        self.xc = centro[0]
        self.yc = centro[1]
        self.lados = lados
        self.radio = radio
        self.rotacion = rotacion
    def calcula_vertices(self):
        vertices=[]
        for i in range (5):
            grados=((2*math.pi*i)/self.lados)-self.rotacion
            ang=grados
            xn=int(self.xc*self.radio*(math.sin(ang)))
            vertices.append(xn)
            yn=int(self.yc*self.radio*(math.cos(ang)))
            vertices.append(yn)
        return(vertices)
    def dibuja(self):
        oled.poly(0,0,array.array('h',self.calcula_vertices()))

pol=Poligono([64,32],5,31,0)
pol.dibuja()
oled.show()

practica.py

class Caracter(Punto):
    # Constructor
    def __init__(self, x, y, vx, vy):
        super().__init__(x, y, vx, vy)
    # Borra el elemento anterior y muestra el siguiente
    def muestra(self):
        oled.text("E", int(self.x), int(self.y), 0)
        self.mueve() # Actualiza posición
        oled.text("E", int(self.x), int(self.y), 1)        
        
class Animacion(Punto):
    # Constructor
    def __init__(self, x, y, vx, vy):
        super().__init__(x, y, vx, vy)
        self.longitud = 0 # 
        self.direccion = 1
    # Borra el elemento anterior y muestra el siguiente
    def muestra(self):
        # Borra la cruz anterior
        oled.hline(int(self.x - self.longitud/2 + 1), int(self.y), int(self.longitud/2), 0)
        oled.hline(int(self.x), int(self.y), int(self.longitud/2), 0)
        oled.vline(int(self.x), int(self.y - self.longitud/2 + 1), int(self.longitud/2), 0)
        oled.vline(int(self.x), int(self.y), int(self.longitud/2), 0)        
        # Cambia la longitud de las líneas de la cruz      
        self.longitud = self.longitud + self.direccion
        if self.longitud > 12:
            self.longitud = 12
            self.direccion = -1
        elif self.longitud < 0:
            self.longitud = 0
            self.direccion = 1            
        # Actualiza posición
        self.mueve()
        # Dibuja la cruz actual
        oled.hline(int(self.x - self.longitud/2 + 1), int(self.y), int(self.longitud/2), 1)
        oled.hline(int(self.x), int(self.y), int(self.longitud/2), 1)
        
        oled.vline(int(self.x), int(self.y - self.longitud/2 + 1), int(self.longitud/2), 1)
        oled.vline(int(self.x), int(self.y), int(self.longitud/2), 1)

Algoritmos.py

from clases import Boton
from random import random
from clases import OLED
import array

boton2 = Boton(19)
oled = OLED()

# Clase Punto
class Punto():
    # Constructor
    def __init__(self, x, y, vx, vy):
        self.x = x
        self.y = y
        self.vx = vx
        self.vy = vy
    # Calcula la siguiente posición central
    def mueve(self):        
        if (self.x > (oled.width)) or ( self.x < 0 ):
            self.vx = -self.vx
        if (self.y > (oled.height)) or ( self.y < 0 ):
            self.vy = -self.vy        
        self.x += self.vx
        self.y += self.vy
    # Borra el elemento anterior y muestra el siguiente
    def muestra(self):
        oled.pixel(int(self.x),int(self.y),0)        
        self.mueve() # Actualiza posición
        oled.pixel(int(self.x),int(self.y),1)
        
oled.fill(0)
puntos = []
for i in range(10):
    puntos.append(Punto(0, 0, 2*random(), 2*random()))

# --- Programa principal ---
while True:
    for i in range(len(puntos)):
        puntos[i].muestra()
    oled.show()   
