# ir_rx_test.py Test program for IR remote control decoder arem.py
# Supports Pyboard and ESP8266

# Author: Peter Hinch
# Copyright Peter Hinch 2020 Released under the MIT license

# Run this to characterise a remote.

from sys import platform
import time
from machine import Pin, freq
from ir_rx import *

ESP32 = platform == 'esp32' or platform == 'esp32_LoBo'

if platform == 'pyboard':
    p = Pin('X3', Pin.IN)
elif platform == 'esp8266':
    freq(160000000)
    p = Pin(13, Pin.IN)
elif ESP32:
    p = Pin(23, Pin.IN)

errors = {BADSTART : 'Invalid start pulse', BADBLOCK : 'Error: bad block',
          BADREP : 'Error: repeat', OVERRUN : 'Error: overrun',
          BADDATA : 'Error: invalid data', BADADDR : 'Error: invalid address'}

def cb(data, addr, ctrl):
    if data == REPEAT:  # NEC protocol sends repeat codes.
        print('Repeat code.')
    elif data >= 0:
        print('Data {:03x} Addr {:03x} Ctrl {:01x}'.format(data, addr, ctrl))
    else:
        print(errors[data])  # Application would ignore errors


s = '''Test for IR receiver. Run:
from ir_rx_test import test
test() for NEC protocol,
test(1) for Sony SIRC 12 bit,
test(2) for Sony SIRC 15 bit,
test(3) for Sony SIRC 20 bit,
test(5) for Philips RC-5 protocol,
test(6) for RC6 mode 0.

Background processing means REPL prompt reappears.
Hit ctrl-D to stop (soft reset).'''

print(s)

def test(proto=0):
    if proto == 0:
        ir = NEC_IR(p, cb)  # Extended mode
    elif proto < 4:
        bits = (12, 15, 20)[proto - 1]
        ir = SONY_IR(p, cb, bits)
        ir.verbose = True
    elif proto == 5:
        ir = RC5_IR(p, cb)
    elif proto == 6:
        ir = RC6_M0(p, cb)
        ir.verbose = True
    # A real application would do something here...
    #while True:
        #time.sleep(5)
        #print('running')