# /*****************************************************************************
# * | File        :	  epdconfig.py
# * | Author      :   Waveshare team
# * | Function    :   Hardware underlying interface
# * | Info        :
# *----------------
# * | This version:   V1.0
# * | Date        :   2020-12-21
# * | Info        :   
# ******************************************************************************
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documnetation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to  whom the Software is
# furished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS OR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
#

import gpiozero
import time
from smbus import SMBus
import spidev
import ctypes
import logging

# e-Paper
EPD_RST_PIN     = 17
EPD_DC_PIN      = 25
EPD_CS_PIN      = 8
EPD_BUSY_PIN    = 24

# TP
TRST    = 22
INT     = 27

spi     = spidev.SpiDev(0, 0)
address = 0x0
# address = 0x14
# address = 0x48
bus     = SMBus(1)


GPIO_RST_PIN    = gpiozero.LED(EPD_RST_PIN)
GPIO_DC_PIN     = gpiozero.LED(EPD_DC_PIN)
# GPIO_CS_PIN     = gpiozero.LED(EPD_CS_PIN)
GPIO_TRST       = gpiozero.LED(TRST)

GPIO_BUSY_PIN   = gpiozero.Button(EPD_BUSY_PIN, pull_up = False)
GPIO_INT        = gpiozero.Button(INT, pull_up = False)


def digital_write(pin, value):
    if pin == EPD_RST_PIN:
        if value:
            GPIO_RST_PIN.on()
        else:
            GPIO_RST_PIN.off()
    elif pin == EPD_DC_PIN:
        if value:
            GPIO_DC_PIN.on()
        else:
            GPIO_DC_PIN.off()
    # elif pin == EPD_CS_PIN:
    #     if value:
    #         GPIO_CS_PIN.on()
    #     else:
    #         GPIO_CS_PIN.off()
    elif pin == TRST:
        if value:
            GPIO_TRST.on()
        else:
            GPIO_TRST.off()

def digital_read(pin):
    if pin == EPD_BUSY_PIN:
        return GPIO_BUSY_PIN.value
    elif pin == INT:
        return GPIO_INT.value

def delay_ms(delaytime):
    time.sleep(delaytime / 1000.0)

def spi_writebyte(data):
    spi.writebytes(data)

def spi_writebyte2(data):
    spi.writebytes2(data)

def i2c_writebyte(reg, value):
    bus.write_word_data(address, (reg>>8) & 0xff, (reg & 0xff) | ((value & 0xff) << 8))

def i2c_write(reg):
    bus.write_byte_data(address, (reg>>8) & 0xff, reg & 0xff)

def i2c_readbyte(reg, len):
    i2c_write(reg)
    rbuf = []
    for i in range(len):
        rbuf.append(int(bus.read_byte(address)))
    return rbuf

def module_init():
   
    spi.max_speed_hz = 10000000
    spi.mode = 0b00
    
    return 0

def module_exit():
    logging.debug("spi end")
    spi.close()
    bus.close()
        
    logging.debug("close 5V, Module enters 0 power consumption ...")
    GPIO_RST_PIN.off()
    GPIO_DC_PIN.off()
    # GPIO_CS_PIN.off()
    GPIO_TRST.off()

    GPIO_RST_PIN.close()
    GPIO_DC_PIN.close()
    # GPIO_CS_PIN.close()
    GPIO_TRST.close()

    GPIO_BUSY_PIN.close()
    GPIO_INT.close()


### END OF FILE ###