import Adafruit_BBIO.GPIO as GPIO

# Pins dont exist in CPython so...lets make our own!
class Pin:
    IN = 0
    OUT = 1
    LOW = 0
    HIGH = 1
    PULL_NONE = 0
    PULL_UP = 1
    PULL_DOWN = 2

    id = None
    _value = LOW
    _mode = IN

    def __init__(self, pin_name):
        self.id = pin_name

    def __repr__(self):
        return str(self.id)

    def __eq__(self, other):
        return self.id == other

    def init(self, mode=IN, pull=None):
        if mode != None:
            if mode == self.IN:
                self._mode = self.IN
                GPIO.setup(self.id, GPIO.IN)
            elif mode == self.OUT:
                self._mode = self.OUT
                GPIO.setup(self.id, GPIO.OUT)
            else:
                raise RuntimeError("Invalid mode for pin: %s" % self.id)
        if pull != None:
            if self._mode != self.IN:
                raise RuntimeError("Cannot set pull resistor on output")
            if pull == self.PULL_UP:
                GPIO.setup(self.id, GPIO.IN, pull_up_down=GPIO.PUD_UP)
            elif pull == self.PULL_DOWN:
                GPIO.setup(self.id, GPIO.IN, pull_up_down=GPIO.PUD_DOWN)
            else:
                raise RuntimeError("Invalid pull for pin: %s" % self.id)

    def value(self, val=None):
        if val != None:
            if val == self.LOW:
                self._value = val
                GPIO.output(self.id, val)
            elif val == self.HIGH:
                self._value = val
                GPIO.output(self.id, val)
            else:
                raise RuntimeError("Invalid value for pin")
        else:
            return GPIO.input(self.id)


# names in comments copied from
# https://github.com/adafruit/adafruit-beaglebone-io-python/blob/master/source/common.c#L73

# PocketBeagle
# P1_1 = SYS VIN        # VIN_AC
P1_2 = Pin('P1_2')      # GPIO2_23 - GPIO_87
P1_3 = Pin('P1_3')      # USB1_VBUS_OUT - (silkscreen: USB1 V_EN)
P1_4 = Pin('P1_4')      # GPIO2_25 - GPIO_89
# P1_5 = USB VBUS       # USB1_VBUS_IN
P1_6 = Pin('P1_6')      # SPI0_CS0 - GPIO_5
# P1_7 = USB VIN        # VIN-USB
P1_8 = Pin('P1_8')      # SPI0_SCLK - GPIO_2
# P1_9 = USB1 DN        # USB1-DN
P1_10 = Pin('P1_10')    # SPI0_D0 - GPIO_3
# P1_11 = USB1 DP       # USB1-DP
P1_12 = Pin('P1_12')    # SPI0_D1 - GPIO_4
# P1_13 = USB1 ID       # USB1-ID
# P1_14 = SYS 3.3V      # VOUT-3.3V
# P1_15 = SYS GND       # GND
# P1_16 = SYS GND       # GND
# P1_17 = AIN 1.8V REF- # VREFN
# P1_18 = AIN 1.8V REF+ # VREFP
P1_19 = Pin('P1_19')    # AIN0
P1_20 = Pin('P1_20')    # GPIO0_20 - GPIO_20
P1_21 = Pin('P1_21')    # AIN1
# P1_22 = SYS GND       # GND
P1_23 = Pin('P1_23')    # AIN2
# P1_22 = SYS VOUT      # VOUT-5V
P1_25 = Pin('P1_25')    # AIN3
P1_26 = Pin('P1_26')    # I2C2_SDA - GPIO_12
P1_27 = Pin('P1_27')    # AIN4
P1_28 = Pin('P1_28')    # I2C2_SCL - GPIO_13
P1_29 = Pin('P1_29')    # GPIO3_21 - GPIO_117
P1_30 = Pin('P1_30')    # UART0_TXD - GPIO_43
P1_31 = Pin('P1_31')    # GPIO3_18 - GPIO_114
P1_32 = Pin('P1_32')    # UART0_RXD - GPIO_42
P1_33 = Pin('P1_33')    # GPIO3_15 - GPIO_111
P1_34 = Pin('P1_34')    # GPIO0_26 - GPIO_26
P1_35 = Pin('P1_35')    # GPIO2_24 - GPIO_88
P1_36 = Pin('P1_36')    # EHRPWM0A - GPIO_110


P2_1 = Pin('P2_1')      # EHRPWM1A - GPIO_50
P2_2 = Pin('P2_2')      # GPIO1_27 - GPIO_59
P2_3 = Pin('P2_3')      # GPIO0_23 - GPIO_23
P2_4 = Pin('P2_4')      # GPIO1_26 - GPIO_58
P2_5 = Pin('P2_5')      # UART4_RXD - GPIO_30
P2_6 = Pin('P2_6')      # GPIO1_25 - GPIO_57
P2_7 = Pin('P2_7')      # UART4_TXD - GPIO_31
P2_8 = Pin('P2_8')      # GPIO1_28 - GPIO_60
P2_9 = Pin('P2_9')      # I2C1_SCL - GPIO_15
P2_10 = Pin('P2_10')    # GPIO1_20 - GPIO_52
P2_11 = Pin('P2_11')    # I2C1_SDA - GPIO_14
# P2_12 = SYS  PWR BTN  # POWER_BUTTON
# P2_13 = SYS VOUT      # VOUT-5V
# P2_14 = BAT VIN       # BAT-VIN
# P2_15 = SYS GND       # GND
# P2_16 = BAT TEMP      # BAT-TEMP
P2_17 = Pin('P2_17')    # GPIO2_1 - GPIO_65
P2_18 = Pin('P2_18')    # GPIO1_15 - GPIO_47
P2_19 = Pin('P2_19')    # GPIO0_27 - GPIO_27
P2_20 = Pin('P2_20')    # GPIO2_0 - GPIO_64
# P2_21 = SYS GND       # GND
P2_22 = Pin('P2_22')    # GPIO1_14 - GPIO_46
# P2_23 = SYS 3.3V      # VOUT-3.3V
P2_24 = Pin('P2_24')    # GPIO1_12 - GPIO_44
P2_25 = Pin('P2_25')    # SPI1_D1 - GPIO_41
# P2_26 = SYS NRST      # RESET#
P2_27 = Pin('P2_27')    # SPI1_D0 - GPIO_40
P2_28 = Pin('P2_28')    # GPIO3_20 - GPIO_116
P2_29 = Pin('P2_29')    # SPI1_SCLK - GPIO_7
P2_30 = Pin('P2_30')    # GPIO3_17 - GPIO_113
P2_31 = Pin('P2_31')    # SPI1_CS1 - GPIO_19
P2_32 = Pin('P2_32')    # GPIO3_16 - GPIO_112
P2_33 = Pin('P2_33')    # GPIO1_13 - GPIO_45
P2_34 = Pin('P2_34')    # GPIO3_19 - GPIO_115
P2_35 = Pin('P2_35')    # GPIO2_22 - GPIO_86
P2_36 = Pin('P2_36')    # AIN7


# BeagleBone Black
# P8_1 = DGND           # DGND - GPIO_0
# P8_2 = DGND           # DGND - GPIO_0
P8_3 = Pin('P8_3')      # GPIO1_6 - GPIO_38
P8_4 = Pin('P8_4')      # GPIO1_7 - GPIO_39
P8_5 = Pin('P8_5')      # GPIO1_2 - GPIO_34
P8_6 = Pin('P8_6')      # GPIO1_3 - GPIO_35
P8_7 = Pin('P8_7')      # TIMER4 - GPIO_66
P8_8 = Pin('P8_8')      # TIMER7 - GPIO_67
P8_9 = Pin('P8_9')      # TIMER5 - GPIO_69
P8_10 = Pin('P8_10')    # TIMER6 - GPIO_68
P8_11 = Pin('P8_11')    # GPIO1_13 - GPIO_45
P8_12 = Pin('P8_12')    # GPIO1_12 - GPIO_44
P8_13 = Pin('P8_13')    # EHRPWM2B - GPIO_23
P8_14 = Pin('P8_14')    # GPIO0_26 - GPIO_26
P8_15 = Pin('P8_15')    # GPIO1_15 - GPIO_47
P8_16 = Pin('P8_16')    # GPIO1_14 - GPIO_46
P8_17 = Pin('P8_17')    # GPIO0_27 - GPIO_27
P8_18 = Pin('P8_18')    # GPIO2_1 - GPIO_65
P8_19 = Pin('P8_19')    # EHRPWM2A - GPIO_22
P8_20 = Pin('P8_20')    # GPIO1_31 - GPIO_63
P8_21 = Pin('P8_21')    # GPIO1_30 - GPIO_62
P8_22 = Pin('P8_22')    # GPIO1_5 - GPIO_37
P8_23 = Pin('P8_23')    # GPIO1_4 - GPIO_36
P8_24 = Pin('P8_24')    # GPIO1_1 - GPIO_33
P8_25 = Pin('P8_25')    # GPIO1_0 - GPIO_32
P8_26 = Pin('P8_26')    # GPIO1_29 - GPIO_61
P8_27 = Pin('P8_27')    # GPIO2_22 - GPIO_86
P8_28 = Pin('P8_28')    # GPIO2_24 - GPIO_88
P8_29 = Pin('P8_29')    # GPIO2_23 - GPIO_87
P8_30 = Pin('P8_30')    # GPIO2_25 - GPIO_89
P8_31 = Pin('P8_31')    # UART5_CTSN - GPIO_10
P8_32 = Pin('P8_32')    # UART5_RTSN - GPIO_11
P8_33 = Pin('P8_33')    # UART4_RTSN - GPIO_9
P8_34 = Pin('P8_34')    # UART3_RTSN - GPIO_81
P8_35 = Pin('P8_35')    # UART4_CTSN - GPIO_8
P8_36 = Pin('P8_36')    # UART3_CTSN - GPIO_80
P8_37 = Pin('P8_37')    # UART5_TXD - GPIO_78
P8_38 = Pin('P8_38')    # UART5_RXD - GPIO_79
P8_39 = Pin('P8_39')    # GPIO2_12 - GPIO_76
P8_40 = Pin('P8_40')    # GPIO2_13 - GPIO_77
P8_41 = Pin('P8_41')    # GPIO2_10 - GPIO_74
P8_42 = Pin('P8_42')    # GPIO2_11 - GPIO_75
P8_43 = Pin('P8_43')    # GPIO2_8 - GPIO_72
P8_44 = Pin('P8_44')    # GPIO2_9 - GPIO_73
P8_45 = Pin('P8_45')    # GPIO2_6 - GPIO_70
P8_46 = Pin('P8_46')    # GPIO2_7 - GPIO_71

# P9_1 = DGND           # DGND - GPIO_0
# P9_2 = DGND           # DGND - GPIO_0
# P9_3 = VDD_3V3        # VDD_3V3 - GPIO_0
# P9_4 = VDD_3V3        # VDD_3V3 - GPIO_0
# P9_5 = VDD_5V         # VDD_5V - GPIO_0
# P9_6 = VDD_5V         # VDD_5V - GPIO_0
# P9_7 = SYS_5V         # SYS_5V - GPIO_0
# P9_8 = SYS_5V         # SYS_5V - GPIO_0
# P9_9 = PWR_BUT        # PWR_BUT - GPIO_0
# P9_10 = SYS_RESETN    # SYS_RESETn - GPIO_0
P9_11 = Pin('P9_11')    # UART4_RXD - GPIO_30
P9_12 = Pin('P9_12')    # GPIO1_28 - GPIO_60
P9_13 = Pin('P9_13')    # UART4_TXD - GPIO_31
P9_14 = Pin('P9_14')    # EHRPWM1A - GPIO_50
P9_15 = Pin('P9_15')    # GPIO1_16 - GPIO_48
P9_16 = Pin('P9_16')    # EHRPWM1B - GPIO_51
P9_17 = Pin('P9_17')    # I2C1_SCL - GPIO_5
P9_18 = Pin('P9_18')    # I2C1_SDA - GPIO_4
P9_19 = Pin('P9_19')    # I2C2_SCL - GPIO_13
P9_20 = Pin('P9_20')    # I2C2_SDA - GPIO_12
P9_21 = Pin('P9_21')    # UART2_TXD - GPIO_3
P9_22 = Pin('P9_22')    # UART2_RXD - GPIO_2
P9_23 = Pin('P9_23')    # GPIO1_17 - GPIO_49
P9_24 = Pin('P9_24')    # UART1_TXD - GPIO_15
P9_25 = Pin('P9_25')    # GPIO3_21 - GPIO_117
P9_26 = Pin('P9_26')    # UART1_RXD - GPIO_14
P9_27 = Pin('P9_27')    # GPIO3_19 - GPIO_115
P9_28 = Pin('P9_28')    # SPI1_CS0 - GPIO_113
P9_29 = Pin('P9_29')    # SPI1_D0 - GPIO_111
P9_30 = Pin('P9_30')    # SPI1_D1 - GPIO_112
P9_31 = Pin('P9_31')    # SPI1_SCLK - GPIO_110
# P9_32 = VDD_ADC       # VDD_ADC - GPIO_0
# P9_33 = AIN4          # AIN4 - GPIO_0
# P9_34 = GNDA_ADC      # GNDA_ADC - GPIO_0
# P9_35 = AIN6          # AIN6 - GPIO_0
# P9_36 = AIN5          # AIN5 - GPIO_0
# P9_37 = AIN2          # AIN2 - GPIO_0
# P9_38 = AIN3          # AIN3 - GPIO_0
# P9_39 = AIN0          # AIN0 - GPIO_0
# P9_40 = AIN1          # AIN1 - GPIO_0
P9_41 = Pin('P9_41')    # CLKOUT2 - GPIO_20
P9_42 = Pin('P9_42')    # GPIO0_7 - GPIO_7
# P9_43 = DGND          # DGND - GPIO_0
# P9_44 = DGND          # DGND - GPIO_0
# P9_45 = DGND          # DGND - GPIO_0
# P9_46 = DGND          # DGND - GPIO_0


##########################################
# common to all beagles
USR0 = Pin('USR0')      # USR0 - GPIO_53
USR1 = Pin('USR1')      # USR1 - GPIO_54
USR2 = Pin('USR2')      # USR2 - GPIO_55
USR3 = Pin('USR3')      # USR3 - GPIO_56


##########################################
# specials

# analog input
AIN0 = Pin('AIN0')
AIN1 = Pin('AIN1')
AIN2 = Pin('AIN2')
AIN3 = Pin('AIN3')
AIN4 = Pin('AIN4')
AIN5 = Pin('AIN5')
AIN6 = Pin('AIN6')
AIN7 = Pin('AIN7')

# PWM
EHRPWM0A = Pin('EHRPWM0A')
EHRPWM0B = Pin('EHRPWM0B')
EHRPWM1A = Pin('EHRPWM1A')
EHRPWM1B = Pin('EHRPWM1B')
EHRPWM2A = Pin('EHRPWM2A')
EHRPWM2B = Pin('EHRPWM2B')
ECAPPWM0 = Pin('ECAPPWM0')
ECAPPWM2 = Pin('ECAPPWM2')
TIMER4 = Pin('TIMER4')
TIMER5 = Pin('TIMER5')
TIMER6 = Pin('TIMER6')
TIMER7 = Pin('TIMER7')


# I2C1
I2C1_SDA = Pin('I2C1_SDA')
I2C1_SCL = Pin('I2C1_SCL')

# I2C2
I2C2_SDA = Pin('I2C2_SDA')
I2C2_SCL = Pin('I2C2_SCL')

# SPI0
SPI0_CS0 = Pin('SPI0_CS0')
SPI0_SCLK = Pin('SPI0_SCLK')
SPI0_D1 = Pin('SPI0_D1')
SPI0_D0 = Pin('SPI0_D0')

# SPI1
SPI1_CS0 = Pin('SPI1_CS0')
SPI1_CS1 = Pin('SPI1_CS1')
SPI1_SCLK = Pin('SPI1_SCLK')
SPI1_D1 = Pin('SPI1_D1')
SPI1_D0 = Pin('SPI1_D0')

# UART0
UART0_TXD = Pin('UART0_TXD')
UART0_RXD = Pin('UART0_RXD')

# UART1
UART1_TXD = Pin('UART1_TXD')
UART1_RXD = Pin('UART1_RXD')
UART1_RTSn = Pin('UART1_RTSn')
UART1_CTSn = Pin('UART1_CTSn')

# UART2
UART2_TXD = Pin('UART2_TXD')
UART2_RXD = Pin('UART2_RXD')

# UART3
UART3_TXD = Pin('UART3_TXD')
UART3_RXD = Pin('UART3_RXD')
UART3_RTSn = Pin('UART3_RTSn')
UART3_CTSn = Pin('UART3_CTSn')

# UART4
UART4_TXD = Pin('UART4_TXD')
UART4_RXD = Pin('UART4_RXD')
UART4_RTSn = Pin('UART4_RTSn')
UART4_CTSn = Pin('UART4_CTSn')

# UART5
UART5_TXD = Pin('UART5_TXD')
UART5_RXD = Pin('UART5_RXD')
UART5_RTSn = Pin('UART5_RTSn')
UART5_CTSn = Pin('UART5_CTSn')


# ordered as spiId, sckId, mosiId, misoId
spiPorts = (
    (1, SPI0_SCLK, SPI0_D1, SPI0_D0),
    (2, SPI1_SCLK, SPI1_D1, SPI1_D0),
)

# ordered as uartId, txId, rxId
uartPorts = (
    # (0, UART0_TXD, UART0_RXD),
    # (1, UART1_TXD, UART1_RXD),
    # (2, UART2_TXD, UART2_RXD),
    # (4, UART4_TXD, UART4_RXD),
    # (5, UART5_TXD, UART5_RXD),
)

# ordered as i2cId, SCL, SDA
i2cPorts = (
    (1, I2C1_SCL, I2C1_SDA),
    (2, I2C2_SCL, I2C2_SDA),
)
