From db0627b65fffd7c4e185a787f670ee130fd86e02 Mon Sep 17 00:00:00 2001
From: Brian Y <briancyiu@gmail.com>
Date: Mon, 2 Dec 2019 19:46:05 -0500
Subject: [PATCH] Finished Rock64 class

---
 .../python/lib/waveshare_epd/epdconfig.py     | 31 ++++++++++++++++++-
 1 file changed, 30 insertions(+), 1 deletion(-)

diff --git a/RaspberryPi&JetsonNano/python/lib/waveshare_epd/epdconfig.py b/RaspberryPi&JetsonNano/python/lib/waveshare_epd/epdconfig.py
index f277338..8563632 100644
--- a/RaspberryPi&JetsonNano/python/lib/waveshare_epd/epdconfig.py
+++ b/RaspberryPi&JetsonNano/python/lib/waveshare_epd/epdconfig.py
@@ -42,13 +42,42 @@ class Rock64:
 	def __init__(self):
 		import spidev
 		import R64.GPIO as GPIO
-
+		
 		self.GPIO = GPIO
 		self.SPI = spidev.SpiDev(32766, 1)
 
 	def digital_write(self, pin, value):
 		self.GPIO.output(pin, value)
 
+	def digital_read(self, pin):
+        return self.GPIO.input(pin)
+
+    def delay_ms(self, delaytime):
+        time.sleep(delaytime / 1000.0)
+
+    def spi_writebyte(self, data):
+        self.SPI.writebytes(data)
+
+    def module_init(self):
+        self.GPIO.setmode(self.GPIO.BCM)
+        self.GPIO.setwarnings(False)
+        self.GPIO.setup(self.RST_PIN, self.GPIO.OUT)
+        self.GPIO.setup(self.DC_PIN, self.GPIO.OUT)
+        self.GPIO.setup(self.CS_PIN, self.GPIO.OUT)
+        self.GPIO.setup(self.BUSY_PIN, self.GPIO.IN)
+        self.SPI.max_speed_hz = 4000000
+        self.SPI.mode = 0b00
+        return 0
+
+    def module_exit(self):
+        logging.debug("spi end")
+        self.SPI.close()
+
+        logging.debug("close 5V, Module enters 0 power consumption ...")
+        self.GPIO.output(self.RST_PIN, 0)
+        self.GPIO.output(self.DC_PIN, 0)
+        self.GPIO.cleanup()
+
 
 class RaspberryPi:
     # Pin definition