Git confusion. Save receiver test.

[micorpython_ir.git] / ir_rx_test.py

diff --git a/ir_rx_test.py b/ir_rx_test.py
index b4f1cf95511a3636e749ce7949c0f0baee304d3e..badd2b559b76117a49f9bee3f2b9b3a87728af00 100644 (file)
--- a/ir_rx_test.py
+++ b/ir_rx_test.py
@@ -10,57 +10,50 @@ from sys import platform
 import time
 import gc
 from machine import Pin, freq
 import time
 import gc
 from machine import Pin, freq

-from ir_rx import *
-
-ESP32 = platform == 'esp32' or platform == 'esp32_LoBo'
+from ir_rx.print_error import print_error  # Optional print of error codes
+# Import all implemented classes
+from ir_rx.nec import NEC_8, NEC_16
+from ir_rx.sony import SONY_12, SONY_15, SONY_20
+from ir_rx.philips import RC5_IR, RC6_M0

 
 

+# Define pin according to platform

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

-elif ESP32:
+elif platform == 'esp32' or platform == 'esp32_LoBo':

     p = Pin(23, Pin.IN)  # was 27
 
     p = Pin(23, Pin.IN)  # was 27
 

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

 def cb(data, addr, ctrl):
 def cb(data, addr, ctrl):

-    if data == REPEAT:  # NEC protocol sends repeat codes.
+    if data < 0:  # NEC protocol sends repeat codes.

         print('Repeat code.')
         print('Repeat code.')

-    elif data >= 0:
-        print('Data {:03x} Addr {:03x} Ctrl {:01x}'.format(data, addr, ctrl))

     else:
     else:

-        print(errors[data])  # Application would ignore errors
+        print('Data {:02x} Addr {:04x} Ctrl {:02x}'.format(data, addr, ctrl))

 
 

+def test(proto=0):
+    classes = (NEC_8, NEC_16, SONY_12, SONY_15, SONY_20, RC5_IR, RC6_M0)
+    ir = classes[proto](p, cb)  # Instantiate receiver
+    ir.error_function(print_error)  # Show debug information
+    #ir.verbose = True
+    # A real application would do something here...
+    while True:
+        print('running')
+        time.sleep(5)
+        gc.collect()

 
 

+# **** DISPLAY GREETING ****

 s = '''Test for IR receiver. Run:
 from ir_rx_test import test
 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() for NEC 8 bit protocol,
+test(1) for NEC 16 bit,
+test(2) for Sony SIRC 12 bit,
+test(3) for Sony SIRC 15 bit,
+test(4) for Sony SIRC 20 bit,

 test(5) for Philips RC-5 protocol,
 test(6) for RC6 mode 0.
 
 Hit ctrl-c to stop, then ctrl-d to soft reset.'''
 
 print(s)
 test(5) for Philips RC-5 protocol,
 test(6) for RC6 mode 0.
 
 Hit ctrl-c to stop, then ctrl-d to 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)
-    # A real application would do something here...
-    while True:
-        print('running')
-        time.sleep(5)
-        gc.collect()