py-gaugette/gaugette/gpio.py at df1f485c4e79a87e6c136e2cec4866aaec387903 · guyc/py-gaugette · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
#----------------------------------------------------------------------
# Gaugette.GPIO from from https://github.com/guyc/py-gaugette
# Guy Carpenter, Clearwater Software
#
# This is an abstraction layer for GPIO calls to isolate the
# differences between the RasperryPi and BeagleBone Black implementations.
#
# On the RPi, we use wiringpi.GPIO
# On the BBB, we use Adafruit_BBIO.GPIO
#
#----------------------------------------------------------------------
import gaugette
import gaugette.platform

class GPIO:
    def __init__(self):
        if gaugette.platform.isRaspberryPi:
            import wiringpi
            self.gpio = wiringpi.GPIO(wiringpi.GPIO.WPI_MODE_PINS)
            self.setup = self.wiringpi_setup
            self.output = self.gpio.digitalWrite
            self.input = self.gpio.digitalRead
            self.trigger = self.gpio.wiringPiISR
            self.OUT = self.gpio.OUTPUT
            self.IN = self.gpio.INPUT
            self.HIGH = self.gpio.HIGH
            self.LOW = self.gpio.LOW
            self.PUD_UP = self.gpio.PUD_UP
            self.PUD_DOWN = self.gpio.PUD_DOWN
            self.PUD_OFF = self.gpio.PUD_OFF
            self.EDGE_FALLING = self.gpio.INT_EDGE_FALLING
            self.EDGE_RISING = self.gpio.INT_EDGE_RISING
            self.EDGE_BOTH = self.gpio.INT_EDGE_BOTH

        elif gaugette.platform.isBeagleBoneBlack:
            import Adafruit_BBIO.GPIO
            self.gpio = Adafruit_BBIO.GPIO
            self.setup = self.gpio.setup
            self.output = self.gpio.output
            self.input = self.gpio.input
            self.OUT = self.gpio.OUT
            self.IN = self.gpio.IN
            self.HIGH = self.gpio.HIGH
            self.LOW = self.gpio.LOW
            self.PUD_UP = self.gpio.PUD_UP
            self.PUD_DOWN = self.gpio.PUD_DOWN
            self.PUD_OFF = self.gpio.PUD_OFF

        else:
            raise NotImplementedError("Platform is not supported.")

    #----------------------------------------------------------------------
    # Implement the setup call via the wiringpi API
    def wiringpi_setup(self, channel, direction, pull_up_down=None):
        self.gpio.pinMode(channel, direction)
        if pull_up_down is None:
            pull_up_down = self.gpio.PUD_OFF
        self.gpio.pullUpDnControl(channel, pull_up_down)