Rewriting the radio in TypeScript (#2000)
* Refactoring radio into the ts * Moving the rest of the radio functionality to the TypeScript * Removing sim implementation of old radio cpp * Adding test script * Removing handler key * Rename internal functions * PR feedback * Refactoring to use event bus
This commit is contained in:
@ -2,58 +2,12 @@
|
||||
|
||||
using namespace pxt;
|
||||
|
||||
#define MAX_FIELD_NAME_LENGTH 12
|
||||
#define MAX_FIELD_DOUBLE_NAME_LENGTH 8
|
||||
#define MAX_PAYLOAD_LENGTH 20
|
||||
#define PACKET_PREFIX_LENGTH 9
|
||||
#define VALUE_PACKET_NAME_LEN_OFFSET 13
|
||||
#define DOUBLE_VALUE_PACKET_NAME_LEN_OFFSET 17
|
||||
|
||||
|
||||
// Packet Spec:
|
||||
// | 0 | 1 ... 4 | 5 ... 8 | 9 ... 28
|
||||
// ----------------------------------------------------------------
|
||||
// | packet type | system time | serial number | payload
|
||||
//
|
||||
// Serial number defaults to 0 unless enabled by user
|
||||
|
||||
// payload: number (9 ... 12)
|
||||
#define PACKET_TYPE_NUMBER 0
|
||||
|
||||
// payload: number (9 ... 12), name length (13), name (14 ... 26)
|
||||
#define PACKET_TYPE_VALUE 1
|
||||
|
||||
// payload: string length (9), string (10 ... 28)
|
||||
#define PACKET_TYPE_STRING 2
|
||||
|
||||
// payload: buffer length (9), buffer (10 ... 28)
|
||||
#define PACKET_TYPE_BUFFER 3
|
||||
|
||||
// payload: number (9 ... 16)
|
||||
#define PACKET_TYPE_DOUBLE 4
|
||||
|
||||
// payload: number (9 ... 16), name length (17), name (18 ... 26)
|
||||
#define PACKET_TYPE_DOUBLE_VALUE 5
|
||||
|
||||
//% color=#E3008C weight=96 icon="\uf012"
|
||||
namespace radio {
|
||||
|
||||
// -------------------------------------------------------------------------
|
||||
// Radio
|
||||
// -------------------------------------------------------------------------
|
||||
bool radioEnabled = false;
|
||||
bool transmitSerialNumber = false;
|
||||
|
||||
PacketBuffer packet;
|
||||
|
||||
uint8_t type;
|
||||
uint32_t time;
|
||||
uint32_t serial;
|
||||
int ivalue;
|
||||
double dvalue;
|
||||
String msg; // may be NULL before first packet
|
||||
Buffer bufMsg; // may be NULL before first packet
|
||||
|
||||
int radioEnable() {
|
||||
int r = uBit.radio.enable();
|
||||
if (r != MICROBIT_OK) {
|
||||
@ -81,303 +35,23 @@ namespace radio {
|
||||
uBit.radio.event.eventReceived(MicroBitEvent(src, value, CREATE_ONLY));
|
||||
}
|
||||
|
||||
void setPacketPrefix(uint8_t* buf, int type) {
|
||||
// prefix: type (0), time (1..4), serial (5..8)
|
||||
uint32_t t = system_timer_current_time();
|
||||
uint32_t sn = transmitSerialNumber ? microbit_serial_number() : 0;
|
||||
|
||||
buf[0] = (uint8_t) type;
|
||||
memcpy(buf + 1, &t, 4);
|
||||
memcpy(buf + 5, &sn, 4);
|
||||
}
|
||||
|
||||
uint8_t copyStringValue(uint8_t* buf, String data, uint8_t maxLength) {
|
||||
uint8_t len = min_(maxLength, data->getUTF8Size());
|
||||
|
||||
// One byte for length of the string
|
||||
buf[0] = len;
|
||||
|
||||
if (len > 0) {
|
||||
memcpy(buf + 1, data->getUTF8Data(), len);
|
||||
}
|
||||
return len + 1;
|
||||
}
|
||||
|
||||
String getStringValue(uint8_t* buf, uint8_t maxLength) {
|
||||
// First byte is the string length
|
||||
uint8_t len = min_(maxLength, buf[0]);
|
||||
return mkString((char*)buf + 1, len);
|
||||
}
|
||||
|
||||
uint8_t copyBufferValue(uint8_t* buf, Buffer data, uint8_t maxLength) {
|
||||
uint8_t len = min_(maxLength, data->length);
|
||||
|
||||
// One byte for length of the buffer
|
||||
buf[0] = len;
|
||||
if (len > 0) {
|
||||
memcpy(buf + 1, data->data, len);
|
||||
}
|
||||
return len + 1;
|
||||
}
|
||||
|
||||
Buffer getBufferValue(uint8_t* buf, uint8_t maxLength) {
|
||||
// First byte is the buffer length
|
||||
uint8_t len = min_(maxLength, buf[0]);
|
||||
// skip first byte
|
||||
return mkBuffer(buf + 1, len);
|
||||
}
|
||||
|
||||
void writePacketAsJSON(uint8_t tp, int iv, double dv, int s, int t, String m, Buffer b) {
|
||||
// Convert the packet to JSON and send over serial
|
||||
uBit.serial.send("{");
|
||||
uBit.serial.send("\"t\":");
|
||||
uBit.serial.send(t);
|
||||
uBit.serial.send(",\"s\":");
|
||||
uBit.serial.send(s);
|
||||
if ((tp == PACKET_TYPE_STRING || tp == PACKET_TYPE_VALUE) && NULL != m) {
|
||||
uBit.serial.send(",\"n\":\"");
|
||||
uBit.serial.send((uint8_t*)m->getUTF8Data(), m->getUTF8Size());
|
||||
uBit.serial.send("\"");
|
||||
}
|
||||
if (tp == PACKET_TYPE_BUFFER && NULL != b) {
|
||||
uBit.serial.send(",\"b\":\"");
|
||||
// TODO: proper base64 encoding
|
||||
uBit.serial.send(b->data, b->length);
|
||||
uBit.serial.send("\"");
|
||||
}
|
||||
if (tp == PACKET_TYPE_NUMBER || tp == PACKET_TYPE_VALUE) {
|
||||
uBit.serial.send(",\"v\":");
|
||||
uBit.serial.send(iv);
|
||||
} else if (tp == PACKET_TYPE_DOUBLE || tp == PACKET_TYPE_DOUBLE_VALUE) {
|
||||
uBit.serial.send(",\"v\":");
|
||||
TNumber td = fromDouble(dv);
|
||||
String sd = numops::toString(td);
|
||||
uBit.serial.send((uint8_t*)sd->getUTF8Data(), sd->getUTF8Size());
|
||||
decrRC(sd);
|
||||
}
|
||||
uBit.serial.send("}\r\n");
|
||||
}
|
||||
|
||||
/**
|
||||
* Takes a packet from the micro:bit radio queue.
|
||||
* @param writeToSerial if true, write the received packet to serial without updating the global packet;
|
||||
if false, update the global packet instead
|
||||
* Takes the next packet from the radio queue and returns its contents in a Buffer
|
||||
*/
|
||||
void receivePacket(bool writeToSerial) {
|
||||
PacketBuffer p = uBit.radio.datagram.recv();
|
||||
|
||||
uint8_t* buf = p.getBytes();
|
||||
uint8_t tp;
|
||||
int t;
|
||||
int s;
|
||||
int iv = 0;
|
||||
double dv = 0;
|
||||
String m = NULL;
|
||||
Buffer b = NULL;
|
||||
|
||||
memcpy(&tp, buf, 1);
|
||||
memcpy(&t, buf + 1, 4);
|
||||
memcpy(&s, buf + 5, 4);
|
||||
|
||||
switch(tp) {
|
||||
case PACKET_TYPE_STRING:
|
||||
m = getStringValue(buf + PACKET_PREFIX_LENGTH, MAX_PAYLOAD_LENGTH - 1);
|
||||
break;
|
||||
case PACKET_TYPE_BUFFER:
|
||||
b = getBufferValue(buf + PACKET_PREFIX_LENGTH, MAX_PAYLOAD_LENGTH - 1);
|
||||
break;
|
||||
case PACKET_TYPE_DOUBLE:
|
||||
case PACKET_TYPE_DOUBLE_VALUE:
|
||||
memcpy(&dv, buf + PACKET_PREFIX_LENGTH, sizeof(double));
|
||||
if (tp == PACKET_TYPE_DOUBLE_VALUE) {
|
||||
m = getStringValue(buf + DOUBLE_VALUE_PACKET_NAME_LEN_OFFSET, MAX_FIELD_DOUBLE_NAME_LENGTH);
|
||||
}
|
||||
break;
|
||||
case PACKET_TYPE_NUMBER:
|
||||
case PACKET_TYPE_VALUE:
|
||||
memcpy(&iv, buf + PACKET_PREFIX_LENGTH, sizeof(int));
|
||||
if (tp == PACKET_TYPE_VALUE) {
|
||||
m = getStringValue(buf + VALUE_PACKET_NAME_LEN_OFFSET, MAX_FIELD_NAME_LENGTH);
|
||||
}
|
||||
break;
|
||||
default: // unknown packet
|
||||
return;
|
||||
}
|
||||
|
||||
if (NULL == m)
|
||||
m = mkString("", 0);
|
||||
if (NULL == b)
|
||||
b = mkBuffer(NULL, 0);
|
||||
|
||||
if (!writeToSerial) {
|
||||
// Refresh global packet
|
||||
packet = p;
|
||||
type = tp;
|
||||
time = t;
|
||||
serial = s;
|
||||
ivalue = iv;
|
||||
dvalue = dv;
|
||||
decrRC(msg);
|
||||
decrRC(bufMsg);
|
||||
msg = m;
|
||||
bufMsg = b;
|
||||
}
|
||||
else {
|
||||
writePacketAsJSON(tp, iv, dv, s, t, m, b);
|
||||
decrRC(m);
|
||||
decrRC(b);
|
||||
}
|
||||
//% help=radio/received-packet
|
||||
Buffer readRawPacket() {
|
||||
if (radioEnable() != MICROBIT_OK) return mkBuffer(NULL, 0);
|
||||
packet = uBit.radio.datagram.recv();
|
||||
return mkBuffer(packet.getBytes(), packet.length());
|
||||
}
|
||||
|
||||
/**
|
||||
* Broadcasts a number over radio to any connected micro:bit in the group.
|
||||
* Sends a raw packet through the radio
|
||||
*/
|
||||
//% help=radio/send-number
|
||||
//% weight=60
|
||||
//% blockId=radio_datagram_send block="radio send number %value" blockGap=8
|
||||
void sendNumber(TNumber value) {
|
||||
if (radioEnable() != MICROBIT_OK) return;
|
||||
|
||||
int iv = toInt(value);
|
||||
double dv = toDouble(value);
|
||||
if (iv == dv) {
|
||||
uint8_t length = PACKET_PREFIX_LENGTH + sizeof(int);
|
||||
uint8_t buf[length];
|
||||
memset(buf, 0, length);
|
||||
setPacketPrefix(buf, PACKET_TYPE_NUMBER);
|
||||
memcpy(buf + PACKET_PREFIX_LENGTH, &iv, sizeof(int));
|
||||
uBit.radio.datagram.send(buf, length);
|
||||
} else {
|
||||
uint8_t length = PACKET_PREFIX_LENGTH + sizeof(double);
|
||||
uint8_t buf[length];
|
||||
memset(buf, 0, length);
|
||||
setPacketPrefix(buf, PACKET_TYPE_DOUBLE);
|
||||
memcpy(buf + PACKET_PREFIX_LENGTH, &dv, sizeof(double));
|
||||
uBit.radio.datagram.send(buf, length);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Broadcasts a name / value pair along with the device serial number
|
||||
* and running time to any connected micro:bit in the group. The name can
|
||||
* include no more than 8 characters.
|
||||
* @param name the field name (max 8 characters), eg: "name"
|
||||
* @param value the numeric value
|
||||
*/
|
||||
//% help=radio/send-value
|
||||
//% weight=59
|
||||
//% blockId=radio_datagram_send_value block="radio send|value %name|= %value" blockGap=8
|
||||
void sendValue(String name, TNumber value) {
|
||||
if (radioEnable() != MICROBIT_OK) return;
|
||||
|
||||
uint8_t buf[32];
|
||||
memset(buf, 0, 32);
|
||||
|
||||
int iv = toInt(value);
|
||||
double dv = toDouble(value);
|
||||
if (iv == dv) {
|
||||
setPacketPrefix(buf, PACKET_TYPE_VALUE);
|
||||
memcpy(buf + PACKET_PREFIX_LENGTH, &iv, sizeof(int));
|
||||
|
||||
int stringLen = copyStringValue(buf + VALUE_PACKET_NAME_LEN_OFFSET, name, MAX_FIELD_DOUBLE_NAME_LENGTH);
|
||||
uBit.radio.datagram.send(buf, VALUE_PACKET_NAME_LEN_OFFSET + stringLen);
|
||||
} else {
|
||||
setPacketPrefix(buf, PACKET_TYPE_DOUBLE_VALUE);
|
||||
memcpy(buf + PACKET_PREFIX_LENGTH, &dv, sizeof(double));
|
||||
|
||||
int stringLen = copyStringValue(buf + DOUBLE_VALUE_PACKET_NAME_LEN_OFFSET, name, MAX_FIELD_NAME_LENGTH);
|
||||
uBit.radio.datagram.send(buf, DOUBLE_VALUE_PACKET_NAME_LEN_OFFSET + stringLen);
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* Broadcasts a string along with the device serial number
|
||||
* and running time to any connected micro:bit in the group.
|
||||
*/
|
||||
//% help=radio/send-string
|
||||
//% weight=58
|
||||
//% blockId=radio_datagram_send_string block="radio send string %msg"
|
||||
//% msg.shadowOptions.toString=true
|
||||
void sendString(String msg) {
|
||||
if (radioEnable() != MICROBIT_OK || NULL == msg) return;
|
||||
|
||||
uint8_t buf[32];
|
||||
memset(buf, 0, 32);
|
||||
|
||||
setPacketPrefix(buf, PACKET_TYPE_STRING);
|
||||
int stringLen = copyStringValue(buf + PACKET_PREFIX_LENGTH, msg, MAX_PAYLOAD_LENGTH - 1);
|
||||
|
||||
uBit.radio.datagram.send(buf, PACKET_PREFIX_LENGTH + stringLen);
|
||||
}
|
||||
|
||||
/**
|
||||
* Broadcasts a buffer (up to 19 bytes long) along with the device serial number
|
||||
* and running time to any connected micro:bit in the group.
|
||||
*/
|
||||
//% help=radio/send-buffer
|
||||
//% weight=57
|
||||
//% advanced=true
|
||||
void sendBuffer(Buffer msg) {
|
||||
void sendRawPacket(Buffer msg) {
|
||||
if (radioEnable() != MICROBIT_OK || NULL == msg) return;
|
||||
|
||||
uint8_t buf[32];
|
||||
memset(buf, 0, 32);
|
||||
|
||||
setPacketPrefix(buf, PACKET_TYPE_BUFFER);
|
||||
int bufLen = copyBufferValue(buf + PACKET_PREFIX_LENGTH, msg, MAX_PAYLOAD_LENGTH - 1);
|
||||
|
||||
uBit.radio.datagram.send(buf, PACKET_PREFIX_LENGTH + bufLen);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Reads the next packet from the radio queue and and writes it to serial
|
||||
* as JSON.
|
||||
*/
|
||||
//% help=radio/write-value-to-serial
|
||||
//% weight=3
|
||||
//% blockId=radio_write_value_serial block="radio write value to serial"
|
||||
//% deprecated=true
|
||||
void writeValueToSerial() {
|
||||
if (radioEnable() != MICROBIT_OK) return;
|
||||
receivePacket(true);
|
||||
}
|
||||
|
||||
/**
|
||||
* Writes the last received packet to serial as JSON. This should be called
|
||||
* within an ``onDataPacketReceived`` callback.
|
||||
*/
|
||||
//% help=radio/write-received-packet-to-serial
|
||||
//% weight=3
|
||||
//% blockId=radio_write_packet_serial block="radio write received packet to serial"
|
||||
//% advanced=true
|
||||
void writeReceivedPacketToSerial() {
|
||||
if (radioEnable() != MICROBIT_OK) return;
|
||||
writePacketAsJSON(type, ivalue, dvalue, (int) serial, (int) time, msg, bufMsg);
|
||||
}
|
||||
|
||||
TNumber readNumber() {
|
||||
if (type == PACKET_TYPE_NUMBER || type == PACKET_TYPE_VALUE)
|
||||
return fromInt(ivalue);
|
||||
else if (type == PACKET_TYPE_DOUBLE || type == PACKET_TYPE_DOUBLE_VALUE)
|
||||
return fromDouble(dvalue);
|
||||
else
|
||||
return fromInt(0);
|
||||
}
|
||||
|
||||
/**
|
||||
* Reads the next packet from the radio queue and returns the packet's number
|
||||
* payload or 0 if the packet did not contain a number.
|
||||
*/
|
||||
//% help=radio/receive-number
|
||||
//% weight=46
|
||||
//% blockId=radio_datagram_receive block="radio receive number" blockGap=8
|
||||
//% deprecated=true
|
||||
TNumber receiveNumber()
|
||||
{
|
||||
if (radioEnable() != MICROBIT_OK) return 0;
|
||||
receivePacket(false);
|
||||
return readNumber();
|
||||
uBit.radio.datagram.send(msg->data, msg->length);
|
||||
}
|
||||
|
||||
/**
|
||||
@ -390,23 +64,7 @@ namespace radio {
|
||||
void onDataReceived(Action body) {
|
||||
if (radioEnable() != MICROBIT_OK) return;
|
||||
registerWithDal(MICROBIT_ID_RADIO, MICROBIT_RADIO_EVT_DATAGRAM, body);
|
||||
// make sure the receive buffer has a free spot
|
||||
receivePacket(false);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Reads the next packet from the radio queue and returns the packet's string
|
||||
* payload or the empty string if the packet did not contain a string.
|
||||
*/
|
||||
//% blockId=radio_datagram_receive_string block="radio receive string" blockGap=8
|
||||
//% weight=44
|
||||
//% help=radio/receive-string
|
||||
//% deprecated=true
|
||||
String receiveString() {
|
||||
if (radioEnable() != MICROBIT_OK) return mkString("", 0);
|
||||
receivePacket(false);
|
||||
return msg;
|
||||
readRawPacket();
|
||||
}
|
||||
|
||||
/**
|
||||
@ -419,7 +77,7 @@ namespace radio {
|
||||
//% blockId=radio_datagram_rssi block="radio received signal strength"
|
||||
//% deprecated=true
|
||||
int receivedSignalStrength() {
|
||||
if (radioEnable() != MICROBIT_OK) return 0;
|
||||
if (radioEnable() != MICROBIT_OK || packet == NULL) return 0;
|
||||
return packet.getRSSI();
|
||||
}
|
||||
|
||||
@ -449,74 +107,4 @@ namespace radio {
|
||||
if (radioEnable() != MICROBIT_OK) return;
|
||||
uBit.radio.setTransmitPower(power);
|
||||
}
|
||||
|
||||
/**
|
||||
* Set the radio to transmit the serial number in each message.
|
||||
* @param transmit value indicating if the serial number is transmitted, eg: true
|
||||
*/
|
||||
//% help=radio/set-transmit-serial-number
|
||||
//% weight=8 blockGap=8
|
||||
//% blockId=radio_set_transmit_serial_number block="radio set transmit serial number %transmit"
|
||||
//% advanced=true
|
||||
void setTransmitSerialNumber(bool transmit) {
|
||||
if (radioEnable() != MICROBIT_OK) return;
|
||||
transmitSerialNumber = transmit;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the number payload from the last packet taken from the radio queue
|
||||
* (via ``receiveNumber``, ``receiveString``, etc) or 0 if that packet did not
|
||||
* contain a number.
|
||||
*/
|
||||
//% help=radio/received-number
|
||||
TNumber receivedNumber() {
|
||||
if (radioEnable() != MICROBIT_OK) return 0;
|
||||
return readNumber();
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the serial number of the sender micro:bit from the last packet taken
|
||||
* from the radio queue (via ``receiveNumber``, ``receiveString``, etc) or 0 if
|
||||
* that packet did not send a serial number.
|
||||
*/
|
||||
//% help=radio/received-serial
|
||||
uint32_t receivedSerial() {
|
||||
if (radioEnable() != MICROBIT_OK) return 0;
|
||||
return serial;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the string payload from the last packet taken from the radio queue
|
||||
* (via ``receiveNumber``, ``receiveString``, etc) or the empty string if that
|
||||
* packet did not contain a string.
|
||||
*/
|
||||
//% help=radio/received-string
|
||||
String receivedString() {
|
||||
if (radioEnable() != MICROBIT_OK || NULL == msg) return mkString("", 0);
|
||||
incrRC(msg);
|
||||
return msg;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the buffer payload from the last packet taken from the radio queue
|
||||
* (via ``receiveNumber``, ``receiveString``, etc) or the empty string if that
|
||||
* packet did not contain a string.
|
||||
*/
|
||||
//% help=radio/received-buffer
|
||||
Buffer receivedBuffer() {
|
||||
if (radioEnable() != MICROBIT_OK || NULL == bufMsg) return mkBuffer(NULL, 0);
|
||||
incrRC(bufMsg);
|
||||
return bufMsg;
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns the system time of the sender micro:bit at the moment when it sent the
|
||||
* last packet taken from the radio queue (via ``receiveNumber``,
|
||||
* ``receiveString``, etc).
|
||||
*/
|
||||
//% help=radio/received-time
|
||||
uint32_t receivedTime() {
|
||||
if (radioEnable() != MICROBIT_OK) return 0;
|
||||
return time;
|
||||
}
|
||||
}
|
||||
|
Reference in New Issue
Block a user