From 877da1614a13de80d12f291c6200f3d36f46426a Mon Sep 17 00:00:00 2001
From: Ace <saadnaseer63@yahoo.de>
Date: Tue, 26 May 2020 19:11:52 +0200
Subject: [PATCH] Added rendering support (work in progress)

---
 inkycal/display/__init__.py |  1 +
 inkycal/display/epaper.py   | 91 +++++++++++++++++++++++++++++++++++++
 2 files changed, 92 insertions(+)
 create mode 100644 inkycal/display/epaper.py

diff --git a/inkycal/display/__init__.py b/inkycal/display/__init__.py
index e69de29..488b9a3 100644
--- a/inkycal/display/__init__.py
+++ b/inkycal/display/__init__.py
@@ -0,0 +1 @@
+from .epaper import Display
diff --git a/inkycal/display/epaper.py b/inkycal/display/epaper.py
new file mode 100644
index 0000000..b415847
--- /dev/null
+++ b/inkycal/display/epaper.py
@@ -0,0 +1,91 @@
+#!/usr/bin/python3
+# -*- coding: utf-8 -*-
+"""
+Inky-Calendar epaper functions
+Copyright by aceisace
+"""
+class Display:
+  """Display class for inkycal
+  Handles rendering on display"""
+
+  _allowed_models = []
+
+  def __init__(self, epaper_model, supports_colour = False):
+    """Load the drivers for this epaper model"""
+
+    self.supports_colour = supports_colour
+    
+    try:
+      importer = 'from inkycal.display.drivers import {} as driver'.format(
+        epaper_model)
+      exec(importer)
+      
+    except ModuleNotFoundError:
+      raise Exception('This module is not supported. Check your spellings?')
+    
+    except FileNotFoundError:
+      raise Exception('SPI could not be found. Please check if SPI is enabled')
+
+    self._epaper = driver.EPD()
+  
+    def render(self, im_black, im_colour = None):
+      """Render an image on the epaper
+      im_colour is required for three-colour epapers"""
+
+      epaper = self._epaper
+      
+      if self.supports_colour == False:
+        print('Initialising..', end = '')
+        epaper.init()
+        print('Updating display......', end = '')
+        epaper.display(epaper.getbuffer(im_black))
+        print('Done')
+        
+      elif self.supports_colour == True:
+        if not im_colour:
+          raise Exception('im_colour is required for coloured epaper displays')
+        print('Initialising..', end = '')
+        epaper.init()
+        print('Updating display......', end = '')
+        epaper.display(epaper.getbuffer(im_black))
+        print('Done')
+
+      print('Sending E-Paper to deep sleep...', end = '')
+      epaper.sleep()
+      print('Done')
+
+    def calibrate(self, cycles=3):
+      """Flush display with single colour to prevent burn-ins (ghosting)
+      cycles -> int. How many times should each colour be flushed?
+      recommended cycles = 3"""
+
+      epaper = self._epaper
+      epaper.init()
+
+      white = Image.new('1', (display_width, display_height), 'white')
+      black = Image.new('1', (display_width, display_height), 'black')
+
+      print('----------Started calibration of ePaper display----------')
+      if self.supports_colour == True:
+        for _ in range(cycles):
+          print('Calibrating...', end= ' ')
+          print('black...', end= ' ')
+          epaper.display(epaper.getbuffer(black), epaper.getbuffer(white))
+          print('colour...', end = ' ')
+          epaper.display(epaper.getbuffer(white), epaper.getbuffer(black))
+          print('white...')
+          epaper.display(epaper.getbuffer(white), epaper.getbuffer(white))
+          print('Cycle {0} of {1} complete'.format(_+1, no_of_cycles))
+      if self.supports_colour == False:
+        for _ in range(no_of_cycles):
+          print('Calibrating...', end= ' ')
+          print('black...', end = ' ')
+          epaper.display(epaper.getbuffer(black))
+          print('white...')
+          epaper.display(epaper.getbuffer(white)),
+          print('Cycle {0} of {1} complete'.format(_+1, no_of_cycles))
+
+        print('-----------Calibration complete----------')
+        epaper.sleep()
+
+