sporq/kmk_firmware
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

black my bad

Browse Source
This commit is contained in:
Cole Smith 2022-04-15 00:06:50 -07:00 committed by xs5871
parent 86e3b138b2
commit 44d27ef09a
1 changed files with 58 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

93
kmk/extensions/peg_oledDisplay.py
View File

@ -1,10 +1,13 @@
import busio import busio
import gc
import adafruit_displayio_ssd1306
import displayio import displayio
import terminalio import terminalio
from adafruit_display_text import label from adafruit_display_text import label
import adafruit_displayio_ssd1306
from kmk.extensions import Extension from kmk.extensions import Extension
import gc
class oled(Extension): class oled(Extension):
def __init__( def __init__(
@ -14,60 +17,81 @@ class oled(Extension):
toDisplay: str = "TXT", toDisplay: str = "TXT",
oWidth: int = 128, oWidth: int = 128,
oHeight: int = 32, oHeight: int = 32,
flip:bool = False, flip: bool = False,
): ):
displayio.release_displays() displayio.release_displays()
self._views=views self._views = views
self._toDisplay = toDisplay self._toDisplay = toDisplay
i2c = busio.I2C(board.SCL, board.SDA) i2c = busio.I2C(board.SCL, board.SDA)
self.width=oWidth, self.width = (oWidth,)
self.height=oHeight, self.height = (oHeight,)
self.rotation = 180 if flip else 0 self.rotation = 180 if flip else 0
self._display = adafruit_displayio_ssd1306.SSD1306(displayio.I2CDisplay(i2c, device_address=0x3C), self._display = adafruit_displayio_ssd1306.SSD1306(
width=oWidth, displayio.I2CDisplay(i2c, device_address=0x3C),
height=oHeight,rotation=self.rotation) width=oWidth,
height=oHeight,
rotation=self.rotation,
)
self._prevLayers = 0 self._prevLayers = 0
gc.collect() 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 # 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": if singleView[0] == "STATIC":
return singleView[1][0] return singleView[1][0]
if singleView[0]=="LAYER": if singleView[0] == "LAYER":
return singleView[1][layer] return singleView[1][layer]
def renderOledTextLayer(self, layer): def renderOledTextLayer(self, layer):
splash = displayio.Group() splash = displayio.Group()
self._display.show(splash) self._display.show(splash)
splash.append( splash.append(
label.Label(terminalio.FONT, label.Label(
text=self.returnCurrectRenderText(layer,self._views[0]), terminalio.FONT,
color=0xFFFFFF, x=0, y=10)) text=self.returnCurrectRenderText(layer, self._views[0]),
color=0xFFFFFF,
x=0,
y=10,
)
)
splash.append( splash.append(
label.Label(terminalio.FONT, label.Label(
text=self.returnCurrectRenderText(layer,self._views[1]), terminalio.FONT,
color=0xFFFFFF, x=64, y=10)) text=self.returnCurrectRenderText(layer, self._views[1]),
color=0xFFFFFF,
x=64,
y=10,
)
)
splash.append( splash.append(
label.Label(terminalio.FONT, label.Label(
text=self.returnCurrectRenderText(layer,self._views[2]), terminalio.FONT,
color=0xFFFFFF, x=0, y=25)) text=self.returnCurrectRenderText(layer, self._views[2]),
color=0xFFFFFF,
x=0,
y=25,
)
)
splash.append( splash.append(
label.Label(terminalio.FONT, label.Label(
text=self.returnCurrectRenderText(layer,self._views[3]), terminalio.FONT,
color=0xFFFFFF, x=64, y=25)) text=self.returnCurrectRenderText(layer, self._views[3]),
color=0xFFFFFF,
x=64,
y=25,
)
)
gc.collect() gc.collect()
def renderOledImgLayer(self, layer): def renderOledImgLayer(self, layer):
splash = displayio.Group() splash = displayio.Group()
self._display.show(splash) 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) image = displayio.TileGrid(odb, pixel_shader=odb.pixel_shader)
splash.append(image) splash.append(image)
gc.collect() gc.collect()
def updateOLED(self, sandbox): def updateOLED(self, sandbox):
if self._toDisplay == "TXT": if self._toDisplay == "TXT":
@ -75,7 +99,6 @@ class oled(Extension):
if self._toDisplay == "IMG": if self._toDisplay == "IMG":
self.renderOledImgLayer(sandbox.active_layers[0]) self.renderOledImgLayer(sandbox.active_layers[0])
gc.collect() gc.collect()
def on_runtime_enable(self, sandbox): def on_runtime_enable(self, sandbox):
return return
@ -107,4 +130,4 @@ class oled(Extension):
return return
def on_powersave_disable(self, sandbox): def on_powersave_disable(self, sandbox):
return return