chrissewell/pwnagotchi
1
0
Fork 0
mirror of https://github.com/jayofelony/pwnagotchi.git synced 2025-07-01 18:37:27 -04:00
Code Issues Packages Projects Releases Wiki Activity

Initial commit Waveshare OledHat

Browse Source 
Add support for Waveshare Oled Hat
Change view.py to have more variable to support more types of screen in the future
This commit is contained in:
Administrator
2019-10-14 16:38:57 +02:00
parent 82bf9b9853
commit 54a8fd81a5
6 changed files with 370 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
pwnagotchi/ui/display.py
View File

@ -130,6 +130,9 @@ class Display(View):
def is_waveshare_v2(self):
return self._display_type in ('waveshare_2', 'ws_2', 'waveshare2', 'ws2')
def is_oledhat(self):
return self._display_type in ('oledhat')
def is_waveshare_any(self):
return self.is_waveshare_v1() or self.is_waveshare_v2()
@ -176,6 +179,14 @@ class Display(View):
self._display.init(self._display.PART_UPDATE)
self._render_cb = self._waveshare_render
elif self.is_oledhat():
logging.info("initializing oledhat display")
from pwnagotchi.ui.waveshare.oledhat.epd import EPD
self._display = EPD()
self._display.init()
self._display.Clear()
self._render_cb = self._oledhat_render
else:
logging.critical("unknown display type %s" % self._display_type)
@ -192,6 +203,8 @@ class Display(View):
self._display.clear()
elif self.is_waveshare_any():
self._display.Clear(WHITE)
elif self.is_oledhat():
self._display.clear()
else:
logging.critical("unknown display type %s" % self._display_type)
@ -237,6 +250,9 @@ class Display(View):
elif self.is_waveshare_v2():
self._display.displayPartial(buf)
def _oledhat_render(self):
self._display.display(self._canvas)
def _waveshare_bc_render(self):
buf_black = self._display.getbuffer(self._canvas)
# emptyImage = Image.new('1', (self._display.height, self._display.width), 255)