diff options
Diffstat (limited to 'klippy/extras/display/hd44780_spi.py')
| -rw-r--r-- | klippy/extras/display/hd44780_spi.py | 80 |
1 files changed, 45 insertions, 35 deletions
diff --git a/klippy/extras/display/hd44780_spi.py b/klippy/extras/display/hd44780_spi.py index f21accbb..dd46acd3 100644 --- a/klippy/extras/display/hd44780_spi.py +++ b/klippy/extras/display/hd44780_spi.py @@ -8,43 +8,43 @@ import logging from .. import bus -LINE_LENGTH_DEFAULT=20 -LINE_LENGTH_OPTIONS=[16, 20] - -TextGlyphs = { 'right_arrow': b'\x7e' } +LINE_LENGTH_DEFAULT = 20 +LINE_LENGTH_OPTIONS = [16, 20] +TextGlyphs = {"right_arrow": b"\x7e"} class hd44780_spi: def __init__(self, config): self.printer = config.get_printer() - self.hd44780_protocol_init = config.getboolean('hd44780_protocol_init', - True) + self.hd44780_protocol_init = config.getboolean("hd44780_protocol_init", True) # spi config - self.spi = bus.MCU_SPI_from_config( - config, 0x00, pin_option="latch_pin") + self.spi = bus.MCU_SPI_from_config(config, 0x00, pin_option="latch_pin") self.mcu = self.spi.get_mcu() - #self.spi.spi_send([0x01,0xa0]) - self.data_mask = (1<<1) + # self.spi.spi_send([0x01,0xa0]) + self.data_mask = 1 << 1 self.command_mask = 0 - self.enable_mask = (1<<3) + self.enable_mask = 1 << 3 self.icons = {} - self.line_length = config.getchoice('line_length', LINE_LENGTH_OPTIONS, - LINE_LENGTH_DEFAULT) + self.line_length = config.getchoice( + "line_length", LINE_LENGTH_OPTIONS, LINE_LENGTH_DEFAULT + ) # framebuffers - self.text_framebuffers = [bytearray(b' '*2*self.line_length), - bytearray(b' '*2*self.line_length)] + self.text_framebuffers = [ + bytearray(b" " * 2 * self.line_length), + bytearray(b" " * 2 * self.line_length), + ] self.glyph_framebuffer = bytearray(64) self.all_framebuffers = [ # Text framebuffers - (self.text_framebuffers[0], bytearray(b'~'*2*self.line_length), - 0x80), - (self.text_framebuffers[1], bytearray(b'~'*2*self.line_length), - 0xc0), + (self.text_framebuffers[0], bytearray(b"~" * 2 * self.line_length), 0x80), + (self.text_framebuffers[1], bytearray(b"~" * 2 * self.line_length), 0xC0), # Glyph framebuffer - (self.glyph_framebuffer, bytearray(b'~'*64), 0x40) ] + (self.glyph_framebuffer, bytearray(b"~" * 64), 0x40), + ] + def send_4_bits(self, cmd, is_data, minclock): if is_data: mask = self.data_mask @@ -53,31 +53,35 @@ class hd44780_spi: self.spi.spi_send([(cmd & 0xF0) | mask], minclock) self.spi.spi_send([(cmd & 0xF0) | mask | self.enable_mask], minclock) self.spi.spi_send([(cmd & 0xF0) | mask], minclock) + def send(self, cmds, is_data=False, minclock=0): for data in cmds: - self.send_4_bits(data,is_data,minclock) - self.send_4_bits(data<<4,is_data,minclock) + self.send_4_bits(data, is_data, minclock) + self.send_4_bits(data << 4, is_data, minclock) + def flush(self): # Find all differences in the framebuffers and send them to the chip for new_data, old_data, fb_id in self.all_framebuffers: if new_data == old_data: continue # Find the position of all changed bytes in this framebuffer - diffs = [[i, 1] for i, (n, o) in enumerate(zip(new_data, old_data)) - if n != o] + diffs = [ + [i, 1] for i, (n, o) in enumerate(zip(new_data, old_data)) if n != o + ] # Batch together changes that are close to each other - for i in range(len(diffs)-2, -1, -1): + for i in range(len(diffs) - 2, -1, -1): pos, count = diffs[i] - nextpos, nextcount = diffs[i+1] + nextpos, nextcount = diffs[i + 1] if pos + 4 >= nextpos and nextcount < 16: diffs[i][1] = nextcount + (nextpos - pos) - del diffs[i+1] + del diffs[i + 1] # Transmit changes for pos, count in diffs: chip_pos = pos self.send([fb_id + chip_pos]) - self.send(new_data[pos:pos+count], is_data=True) + self.send(new_data[pos : pos + count], is_data=True) old_data[:] = new_data + def init(self): curtime = self.printer.get_reactor().monotonic() print_time = self.mcu.estimated_print_time(curtime) @@ -87,27 +91,30 @@ class hd44780_spi: else: init = [[0x02]] # Reset (set positive direction ; enable display and hide cursor) - init.append([0x06, 0x0c]) + init.append([0x06, 0x0C]) for i, cmds in enumerate(init): - minclock = self.mcu.print_time_to_clock(print_time + i * .100) + minclock = self.mcu.print_time_to_clock(print_time + i * 0.100) self.send(cmds, minclock=minclock) self.flush() + def write_text(self, x, y, data): if x + len(data) > self.line_length: - data = data[:self.line_length - min(x, self.line_length)] + data = data[: self.line_length - min(x, self.line_length)] pos = x + ((y & 0x02) >> 1) * self.line_length - self.text_framebuffers[y & 1][pos:pos+len(data)] = data + self.text_framebuffers[y & 1][pos : pos + len(data)] = data + def set_glyphs(self, glyphs): for glyph_name, glyph_data in glyphs.items(): - data = glyph_data.get('icon5x8') + data = glyph_data.get("icon5x8") if data is not None: self.icons[glyph_name] = data + def write_glyph(self, x, y, glyph_name): data = self.icons.get(glyph_name) if data is not None: slot, bits = data self.write_text(x, y, [slot]) - self.glyph_framebuffer[slot * 8:(slot + 1) * 8] = bits + self.glyph_framebuffer[slot * 8 : (slot + 1) * 8] = bits return 1 char = TextGlyphs.get(glyph_name) if char is not None: @@ -115,11 +122,14 @@ class hd44780_spi: self.write_text(x, y, char) return 1 return 0 + def write_graphics(self, x, y, data): pass + def clear(self): - spaces = b' ' * 2*self.line_length + spaces = b" " * 2 * self.line_length self.text_framebuffers[0][:] = spaces self.text_framebuffers[1][:] = spaces + def get_dimensions(self): return (self.line_length, 4) |