From dcec2c1aca2539b2327aece350b4d016b9e55fec Mon Sep 17 00:00:00 2001
From: Brian Y <briancyiu@gmail.com>
Date: Mon, 2 Dec 2019 19:54:28 -0500
Subject: [PATCH] Added Rock64 to implementation

---
 RaspberryPi&JetsonNano/python/lib/waveshare_epd/epdconfig.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/RaspberryPi&JetsonNano/python/lib/waveshare_epd/epdconfig.py b/RaspberryPi&JetsonNano/python/lib/waveshare_epd/epdconfig.py
index 8563632..e58de12 100644
--- a/RaspberryPi&JetsonNano/python/lib/waveshare_epd/epdconfig.py
+++ b/RaspberryPi&JetsonNano/python/lib/waveshare_epd/epdconfig.py
@@ -31,6 +31,7 @@ import os
 import logging
 import sys
 import time
+import platform
 
 
 class Rock64:
@@ -190,6 +191,8 @@ class JetsonNano:
 
 if os.path.exists('/sys/bus/platform/drivers/gpiomem-bcm2835'):
     implementation = RaspberryPi()
+elif platform.machine() == 'aarch64':  # Rock64
+	implementation = Rock64()
 else:
     implementation = JetsonNano()