From 44d27ef09af5fd5063fdafa1bc2ba5c592ae4303 Mon Sep 17 00:00:00 2001 From: Cole Smith Date: Fri, 15 Apr 2022 00:06:50 -0700 Subject: [PATCH] black my bad --- kmk/extensions/peg_oledDisplay.py | 93 +++++++++++++++++++------------ 1 file changed, 58 insertions(+), 35 deletions(-) diff --git a/kmk/extensions/peg_oledDisplay.py b/kmk/extensions/peg_oledDisplay.py index d80533a..b6da9a2 100644 --- a/kmk/extensions/peg_oledDisplay.py +++ b/kmk/extensions/peg_oledDisplay.py @@ -1,10 +1,13 @@ import busio +import gc + +import adafruit_displayio_ssd1306 import displayio import terminalio from adafruit_display_text import label -import adafruit_displayio_ssd1306 + from kmk.extensions import Extension -import gc + class oled(Extension): def __init__( @@ -14,60 +17,81 @@ class oled(Extension): toDisplay: str = "TXT", oWidth: int = 128, oHeight: int = 32, - flip:bool = False, - + flip: bool = False, ): displayio.release_displays() - self._views=views + self._views = views self._toDisplay = toDisplay i2c = busio.I2C(board.SCL, board.SDA) - self.width=oWidth, - self.height=oHeight, + self.width = (oWidth,) + self.height = (oHeight,) self.rotation = 180 if flip else 0 - self._display = adafruit_displayio_ssd1306.SSD1306(displayio.I2CDisplay(i2c, device_address=0x3C), - width=oWidth, - height=oHeight,rotation=self.rotation) + self._display = adafruit_displayio_ssd1306.SSD1306( + displayio.I2CDisplay(i2c, device_address=0x3C), + width=oWidth, + height=oHeight, + rotation=self.rotation, + ) self._prevLayers = 0 gc.collect() - def returnCurrectRenderText(self,layer,singleView): + def returnCurrectRenderText(self, layer, singleView): # for now we only have static things and react to layers. But when we react to battery % and wpm we can handle the logic here - if singleView[0]=="STATIC": - return singleView[1][0] - if singleView[0]=="LAYER": - return singleView[1][layer] - - + if singleView[0] == "STATIC": + return singleView[1][0] + if singleView[0] == "LAYER": + return singleView[1][layer] + def renderOledTextLayer(self, layer): splash = displayio.Group() self._display.show(splash) splash.append( - label.Label(terminalio.FONT, - text=self.returnCurrectRenderText(layer,self._views[0]), - color=0xFFFFFF, x=0, y=10)) + label.Label( + terminalio.FONT, + text=self.returnCurrectRenderText(layer, self._views[0]), + color=0xFFFFFF, + x=0, + y=10, + ) + ) splash.append( - label.Label(terminalio.FONT, - text=self.returnCurrectRenderText(layer,self._views[1]), - color=0xFFFFFF, x=64, y=10)) + label.Label( + terminalio.FONT, + text=self.returnCurrectRenderText(layer, self._views[1]), + color=0xFFFFFF, + x=64, + y=10, + ) + ) splash.append( - label.Label(terminalio.FONT, - text=self.returnCurrectRenderText(layer,self._views[2]), - color=0xFFFFFF, x=0, y=25)) + label.Label( + terminalio.FONT, + text=self.returnCurrectRenderText(layer, self._views[2]), + color=0xFFFFFF, + x=0, + y=25, + ) + ) splash.append( - label.Label(terminalio.FONT, - text=self.returnCurrectRenderText(layer,self._views[3]), - color=0xFFFFFF, x=64, y=25)) + label.Label( + terminalio.FONT, + text=self.returnCurrectRenderText(layer, self._views[3]), + color=0xFFFFFF, + x=64, + y=25, + ) + ) gc.collect() - + def renderOledImgLayer(self, layer): splash = displayio.Group() self._display.show(splash) - odb = displayio.OnDiskBitmap('/'+self.returnCurrectRenderText(layer,self._views[0])) + odb = displayio.OnDiskBitmap( + '/' + self.returnCurrectRenderText(layer, self._views[0]) + ) image = displayio.TileGrid(odb, pixel_shader=odb.pixel_shader) splash.append(image) gc.collect() - - def updateOLED(self, sandbox): if self._toDisplay == "TXT": @@ -75,7 +99,6 @@ class oled(Extension): if self._toDisplay == "IMG": self.renderOledImgLayer(sandbox.active_layers[0]) gc.collect() - def on_runtime_enable(self, sandbox): return @@ -107,4 +130,4 @@ class oled(Extension): return def on_powersave_disable(self, sandbox): - return \ No newline at end of file + return