nemunaire/waveshareteam-e-Paper
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Remove jetson specific functions

Browse source
This commit is contained in:
missionfloyd 2021-04-01 21:22:07 -06:00 • committed by GitHub
commit d3aedbc96d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 39 additions and 110 deletions
Download patch file Download diff file Expand all files Collapse all files

97
RaspberryPi_JetsonNano/python/lib/waveshare_epd/epdconfig.py
View file

@ -32,37 +32,35 @@ import logging
import sys import sys
import time import time
# Pin definition
RST_PIN = 17
DC_PIN = 25
CS_PIN = 8
BUSY_PIN = 24
class RaspberryPi: def __init__(self):
# Pin definition
RST_PIN = 17
DC_PIN = 25
CS_PIN = 8
BUSY_PIN = 24
def __init__(self):
import spidev import spidev
import RPi.GPIO import RPi.GPIO
self.GPIO = RPi.GPIO self.GPIO = RPi.GPIO
self.SPI = spidev.SpiDev() self.SPI = spidev.SpiDev()
def digital_write(self, pin, value): def digital_write(self, pin, value):
self.GPIO.output(pin, value) self.GPIO.output(pin, value)
def digital_read(self, pin): def digital_read(self, pin):
return self.GPIO.input(pin) return self.GPIO.input(pin)
def delay_ms(self, delaytime): def delay_ms(self, delaytime):
time.sleep(delaytime / 1000.0) time.sleep(delaytime / 1000.0)
def spi_writebyte(self, data): def spi_writebyte(self, data):
self.SPI.writebytes(data) self.SPI.writebytes(data)
def spi_writebyte2(self, data): def spi_writebyte2(self, data):
self.SPI.writebytes2(data) self.SPI.writebytes2(data)
def module_init(self): def module_init(self):
self.GPIO.setmode(self.GPIO.BCM) self.GPIO.setmode(self.GPIO.BCM)
self.GPIO.setwarnings(False) self.GPIO.setwarnings(False)
self.GPIO.setup(self.RST_PIN, self.GPIO.OUT) self.GPIO.setup(self.RST_PIN, self.GPIO.OUT)
@ -76,7 +74,7 @@ class RaspberryPi:
self.SPI.mode = 0b00 self.SPI.mode = 0b00
return 0 return 0
def module_exit(self): def module_exit(self):
logging.debug("spi end") logging.debug("spi end")
self.SPI.close() self.SPI.close()
@ -86,73 +84,4 @@ class RaspberryPi:
self.GPIO.cleanup() self.GPIO.cleanup()
class JetsonNano:
# Pin definition
RST_PIN = 17
DC_PIN = 25
CS_PIN = 8
BUSY_PIN = 24
def __init__(self):
import ctypes
find_dirs = [
os.path.dirname(os.path.realpath(__file__)),
'/usr/local/lib',
'/usr/lib',
]
self.SPI = None
for find_dir in find_dirs:
so_filename = os.path.join(find_dir, 'sysfs_software_spi.so')
if os.path.exists(so_filename):
self.SPI = ctypes.cdll.LoadLibrary(so_filename)
break
if self.SPI is None:
raise RuntimeError('Cannot find sysfs_software_spi.so')
import Jetson.GPIO
self.GPIO = Jetson.GPIO
def digital_write(self, pin, value):
self.GPIO.output(pin, value)
def digital_read(self, pin):
return self.GPIO.input(self.BUSY_PIN)
def delay_ms(self, delaytime):
time.sleep(delaytime / 1000.0)
def spi_writebyte(self, data):
self.SPI.SYSFS_software_spi_transfer(data[0])
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.SYSFS_software_spi_begin()
return 0
def module_exit(self):
logging.debug("spi end")
self.SPI.SYSFS_software_spi_end()
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()
if os.path.exists('/sys/bus/platform/drivers/gpiomem-bcm2835'):
implementation = RaspberryPi()
else:
implementation = JetsonNano()
for func in [x for x in dir(implementation) if not x.startswith('_')]:
setattr(sys.modules[__name__], func, getattr(implementation, func))
### END OF FILE ### ### END OF FILE ###