From c00c001009b68babf2b5cbd3be720e2c4163267a Mon Sep 17 00:00:00 2001 From: spirkelmann Date: Thu, 21 Nov 2019 16:39:54 +0100 Subject: [PATCH] added ssd1306 display library --- micropython/uPySensors/ssd1306.py | 162 ++++++++++++++++++++++++++ micropython/uPySensors/ssd1306_i2c.py | 95 +++++++++++++++ 2 files changed, 257 insertions(+) create mode 100644 micropython/uPySensors/ssd1306.py create mode 100644 micropython/uPySensors/ssd1306_i2c.py diff --git a/micropython/uPySensors/ssd1306.py b/micropython/uPySensors/ssd1306.py new file mode 100644 index 0000000..a666a2d --- /dev/null +++ b/micropython/uPySensors/ssd1306.py @@ -0,0 +1,162 @@ +# MicroPython SSD1306 OLED driver, I2C and SPI interfaces +from micropython import const +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 + self.buffer = bytearray(self.pages * self.width) + fb = framebuf.FrameBuffer(self.buffer, self.width, self.height, framebuf.MONO_VLSB) + self.framebuf = fb + # Provide methods for accessing FrameBuffer graphics primitives. This is a + # workround because inheritance from a native class is currently unsupported. + # http://docs.micropython.org/en/latest/pyboard/library/framebuf.html + self.fill = fb.fill + self.pixel = fb.pixel + self.hline = fb.hline + self.vline = fb.vline + self.line = fb.line + self.rect = fb.rect + self.fill_rect = fb.fill_rect + self.text = fb.text + self.scroll = fb.scroll + self.blit = fb.blit + 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_data(self.buffer) + + +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) + 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_data(self, buf): + self.temp[0] = self.addr << 1 + self.temp[1] = 0x40 # Co=0, D/C#=1 + self.i2c.start() + self.i2c.write(self.temp) + self.i2c.write(buf) + self.i2c.stop() + + 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 + super().__init__(width, height, external_vcc) + + def write_cmd(self, cmd): + self.spi.init(baudrate=self.rate, polarity=0, phase=0) + self.cs(1) + self.dc(0) + self.cs(0) + self.spi.write(bytearray([cmd])) + self.cs(1) + + def write_data(self, buf): + self.spi.init(baudrate=self.rate, polarity=0, phase=0) + self.cs(1) + self.dc(1) + self.cs(0) + self.spi.write(buf) + self.cs(1) + + def poweron(self): + self.res(1) + time.sleep_ms(1) + self.res(0) + time.sleep_ms(10) + self.res(1) diff --git a/micropython/uPySensors/ssd1306_i2c.py b/micropython/uPySensors/ssd1306_i2c.py new file mode 100644 index 0000000..ef3149a --- /dev/null +++ b/micropython/uPySensors/ssd1306_i2c.py @@ -0,0 +1,95 @@ +# https://learn.adafruit.com/micropython-hardware-ssd1306-oled-display/software +import time +from machine import I2C, Pin +from .ssd1306 import SSD1306_I2C + +class Display: + + def __init__(self, + width = 128, height = 64, + scl_pin_id = 15, sda_pin_id = 4, + freq = 400000): + + self.width = width + self.height = height + self.poweron() + self.i2c = I2C(scl = Pin(scl_pin_id, Pin.OUT), + sda = Pin(sda_pin_id), + freq = freq) + self.display = SSD1306_I2C(width, height, self.i2c) + self.show = self.display.show + + def poweron(self, pin=16): + pin_reset = Pin(pin, mode=Pin.OUT) + pin_reset.value(0) + time.sleep_ms(50) + pin_reset.value(1) + + def clear(self): + self.display.fill(0) + self.display.show() + + + def show_text(self, text, x = 0, y = 0, clear_first = True, show_now = True, hold_seconds = 0): + if clear_first: self.display.fill(0) + self.display.text(text, x, y) + if show_now: + self.display.show() + if hold_seconds > 0: time.sleep(hold_seconds) + + + def wrap(self, text, start_line = 0, + height_per_line = 8, width_per_char = 8, + start_pixel_each_line = 0): + + chars_per_line = self.width//width_per_char + max_lines = self.height//height_per_line - start_line + lines = [(text[chars_per_line*line: chars_per_line*(line+1)], start_pixel_each_line, height_per_line*(line+start_line)) + for line in range(max_lines)] + + return lines + + + def draw_logo(self, x_pos=0): + font_height = 50 + min_x = x_pos + min_y = 10 + + self.display.fill(0) + + self.display.fill_rect(min_x, min_y + font_height//2, 5, font_height//2, 1) + self.display.fill_rect(min_x, min_y + font_height//4, 5, 5, 1) + + xshift = 10 + + self.display.rect(xshift + min_x, min_y, 5, font_height, 1) + self.display.rect(xshift + min_x + 4, min_y, font_height//2, font_height//2, 1) + self.display.line(xshift + min_x + 5, min_y + font_height//2, xshift + min_x + 5 + font_height//2, min_y + font_height, 1) + self.display.show() + + def show_text_wrap(self, text, + start_line = 0, height_per_line = 8, width_per_char = 8, start_pixel_each_line = 0, + clear_first = True, show_now = True, hold_seconds = 0): + + if clear_first: self.clear() + + for line, x, y in self.wrap(text, start_line, height_per_line, width_per_char, start_pixel_each_line): + self.show_text(line, x, y, clear_first = False, show_now = False) + + if show_now: + self.display.show() + if hold_seconds > 0: time.sleep(hold_seconds) + + + def show_datetime(self, year, month, day, hour, minute, second): + datetime = [year, month, day, hour, minute, second] + datetime_str = ["{0:0>2}".format(d) for d in datetime] + + self.show_text(text = '-'.join(datetime_str[:3]), + x = 0, y = 0, clear_first = True, show_now = False) + self.show_text(text = ':'.join(datetime_str[3:6]), + x = 0, y = 10, clear_first = False, show_now = True) + + + def show_time(self, year, month, day, hour, minute, second): + self.show_datetime(year, month, day, hour, minute, second)