139 lines
4.5 KiB
Python
139 lines
4.5 KiB
Python
"""Test for nrf24l01 module. Portable between MicroPython targets."""
|
|
|
|
import sys
|
|
import ustruct as struct
|
|
import utime
|
|
from machine import Pin, SPI
|
|
from nrf24l01 import NRF24L01
|
|
from micropython import const
|
|
|
|
# Slave pause between receiving data and checking for further packets.
|
|
_RX_POLL_DELAY = const(15)
|
|
# Slave pauses an additional _SLAVE_SEND_DELAY ms after receiving data and before
|
|
# transmitting to allow the (remote) master time to get into receive mode. The
|
|
# master may be a slow device. Value tested with Pyboard, ESP32 and ESP8266.
|
|
_SLAVE_SEND_DELAY = const(10)
|
|
|
|
if sys.platform == 'pyboard':
|
|
cfg = {'spi': 2, 'miso': 'Y7', 'mosi': 'Y8', 'sck': 'Y6', 'csn': 'Y5', 'ce': 'Y4'}
|
|
elif sys.platform == 'esp8266': # Hardware SPI
|
|
cfg = {'spi': 1, 'miso': 12, 'mosi': 13, 'sck': 14, 'csn': 4, 'ce': 5}
|
|
elif sys.platform == 'esp32': # Software SPI
|
|
cfg = {'spi': -1, 'miso': 32, 'mosi': 33, 'sck': 25, 'csn': 26, 'ce': 27}
|
|
else:
|
|
raise ValueError('Unsupported platform {}'.format(sys.platform))
|
|
|
|
pipes = (b'\xf0\xf0\xf0\xf0\xe1', b'\xf0\xf0\xf0\xf0\xd2')
|
|
|
|
def master():
|
|
csn = Pin(cfg['csn'], mode=Pin.OUT, value=1)
|
|
ce = Pin(cfg['ce'], mode=Pin.OUT, value=0)
|
|
if cfg['spi'] == -1:
|
|
spi = SPI(-1, sck=Pin(cfg['sck']), mosi=Pin(cfg['mosi']), miso=Pin(cfg['miso']))
|
|
nrf = NRF24L01(spi, csn, ce, payload_size=8)
|
|
else:
|
|
nrf = NRF24L01(SPI(cfg['spi']), csn, ce, payload_size=8)
|
|
|
|
nrf.open_tx_pipe(pipes[0])
|
|
nrf.open_rx_pipe(1, pipes[1])
|
|
nrf.start_listening()
|
|
|
|
num_needed = 16
|
|
num_successes = 0
|
|
num_failures = 0
|
|
led_state = 0
|
|
|
|
print('NRF24L01 master mode, sending %d packets...' % num_needed)
|
|
|
|
while num_successes < num_needed and num_failures < num_needed:
|
|
# stop listening and send packet
|
|
nrf.stop_listening()
|
|
millis = utime.ticks_ms()
|
|
led_state = max(1, (led_state << 1) & 0x0f)
|
|
print('sending:', millis, led_state)
|
|
try:
|
|
nrf.send(struct.pack('ii', millis, led_state))
|
|
except OSError:
|
|
pass
|
|
|
|
# start listening again
|
|
nrf.start_listening()
|
|
|
|
# wait for response, with 250ms timeout
|
|
start_time = utime.ticks_ms()
|
|
timeout = False
|
|
while not nrf.any() and not timeout:
|
|
if utime.ticks_diff(utime.ticks_ms(), start_time) > 250:
|
|
timeout = True
|
|
|
|
if timeout:
|
|
print('failed, response timed out')
|
|
num_failures += 1
|
|
|
|
else:
|
|
# recv packet
|
|
got_millis, = struct.unpack('i', nrf.recv())
|
|
|
|
# print response and round-trip delay
|
|
print('got response:', got_millis, '(delay', utime.ticks_diff(utime.ticks_ms(), got_millis), 'ms)')
|
|
num_successes += 1
|
|
|
|
# delay then loop
|
|
utime.sleep_ms(250)
|
|
|
|
print('master finished sending; successes=%d, failures=%d' % (num_successes, num_failures))
|
|
|
|
def slave():
|
|
csn = Pin(cfg['csn'], mode=Pin.OUT, value=1)
|
|
ce = Pin(cfg['ce'], mode=Pin.OUT, value=0)
|
|
if cfg['spi'] == -1:
|
|
spi = SPI(-1, sck=Pin(cfg['sck']), mosi=Pin(cfg['mosi']), miso=Pin(cfg['miso']))
|
|
nrf = NRF24L01(spi, csn, ce, payload_size=8)
|
|
else:
|
|
nrf = NRF24L01(SPI(cfg['spi']), csn, ce, payload_size=8)
|
|
|
|
nrf.open_tx_pipe(pipes[1])
|
|
nrf.open_rx_pipe(1, pipes[0])
|
|
nrf.start_listening()
|
|
|
|
print('NRF24L01 slave mode, waiting for packets... (ctrl-C to stop)')
|
|
|
|
while True:
|
|
if nrf.any():
|
|
while nrf.any():
|
|
buf = nrf.recv()
|
|
millis, led_state = struct.unpack('ii', buf)
|
|
print('received:', millis, led_state)
|
|
for led in leds:
|
|
if led_state & 1:
|
|
led.on()
|
|
else:
|
|
led.off()
|
|
led_state >>= 1
|
|
utime.sleep_ms(_RX_POLL_DELAY)
|
|
|
|
# Give master time to get into receive mode.
|
|
utime.sleep_ms(_SLAVE_SEND_DELAY)
|
|
nrf.stop_listening()
|
|
try:
|
|
nrf.send(struct.pack('i', millis))
|
|
except OSError:
|
|
pass
|
|
print('sent response')
|
|
nrf.start_listening()
|
|
|
|
try:
|
|
import pyb
|
|
leds = [pyb.LED(i + 1) for i in range(4)]
|
|
except:
|
|
leds = []
|
|
|
|
print('NRF24L01 test module loaded')
|
|
print('NRF24L01 pinout for test:')
|
|
print(' CE on', cfg['ce'])
|
|
print(' CSN on', cfg['csn'])
|
|
print(' SCK on', cfg['sck'])
|
|
print(' MISO on', cfg['miso'])
|
|
print(' MOSI on', cfg['mosi'])
|
|
print('run nrf24l01test.slave() on slave, then nrf24l01test.master() on master')
|