2015-04-11 01:07:36 -07:00
|
|
|
/*
|
|
|
|
Copyright 2012-2015 Benjamin Vedder benjamin@vedder.se
|
|
|
|
|
|
|
|
This program is free software: you can redistribute it and/or modify
|
|
|
|
it under the terms of the GNU General Public License as published by
|
|
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
|
|
(at your option) any later version.
|
|
|
|
|
|
|
|
This program is distributed in the hope that it will be useful,
|
|
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
|
|
GNU General Public License for more details.
|
|
|
|
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
|
|
along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
*/
|
|
|
|
|
|
|
|
/*
|
|
|
|
* nrf_driver.c
|
|
|
|
*
|
|
|
|
* Created on: 29 mar 2015
|
|
|
|
* Author: benjamin
|
|
|
|
*/
|
|
|
|
|
2015-12-27 16:50:23 -08:00
|
|
|
#include <string.h>
|
2015-04-11 01:07:36 -07:00
|
|
|
#include "nrf_driver.h"
|
|
|
|
#include "rf.h"
|
|
|
|
#include "rfhelp.h"
|
|
|
|
#include "conf_general.h"
|
|
|
|
#include "app.h"
|
|
|
|
#include "buffer.h"
|
|
|
|
#include "commands.h"
|
2015-12-27 16:50:23 -08:00
|
|
|
#include "crc.h"
|
|
|
|
#include "packet.h"
|
|
|
|
|
|
|
|
// Settings
|
2016-01-24 04:44:51 -08:00
|
|
|
#define MAX_PL_LEN 25
|
|
|
|
#define RX_BUFFER_SIZE PACKET_MAX_PL_LEN
|
|
|
|
|
|
|
|
#define ALIVE_INTERVAL 50 // Send alive packets at this rate
|
|
|
|
#define NRF_RESTART_TIMEOUT 500 // Restart the NRF if nothing has been received or acked for this time
|
2015-04-11 01:07:36 -07:00
|
|
|
|
|
|
|
// Variables
|
2015-12-27 16:50:23 -08:00
|
|
|
static THD_WORKING_AREA(rx_thread_wa, 2048);
|
2015-10-08 14:09:39 -07:00
|
|
|
static THD_WORKING_AREA(tx_thread_wa, 512);
|
2015-04-11 01:07:36 -07:00
|
|
|
static mote_state mstate;
|
2015-12-27 16:50:23 -08:00
|
|
|
static uint8_t rx_buffer[RX_BUFFER_SIZE];
|
2016-01-24 04:44:51 -08:00
|
|
|
static int nosend_cnt;
|
|
|
|
static int nrf_restart_rx_time;
|
|
|
|
static int nrf_restart_tx_time;
|
2015-04-11 01:07:36 -07:00
|
|
|
|
|
|
|
// Functions
|
2015-10-08 14:09:39 -07:00
|
|
|
static THD_FUNCTION(rx_thread, arg);
|
|
|
|
static THD_FUNCTION(tx_thread, arg);
|
2016-01-24 04:44:51 -08:00
|
|
|
static int rf_tx_wrapper(char *data, int len);
|
2015-04-11 01:07:36 -07:00
|
|
|
|
|
|
|
void nrf_driver_init(void) {
|
|
|
|
rfhelp_init();
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
nosend_cnt = 0;
|
|
|
|
nrf_restart_rx_time = 0;
|
|
|
|
nrf_restart_tx_time = 0;
|
|
|
|
|
2015-04-11 01:07:36 -07:00
|
|
|
chThdCreateStatic(rx_thread_wa, sizeof(rx_thread_wa), NORMALPRIO - 1, rx_thread, NULL);
|
|
|
|
chThdCreateStatic(tx_thread_wa, sizeof(tx_thread_wa), NORMALPRIO - 1, tx_thread, NULL);
|
|
|
|
}
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
static int rf_tx_wrapper(char *data, int len) {
|
|
|
|
int res = rfhelp_send_data_crc(data, len);
|
|
|
|
|
|
|
|
if (res == 0) {
|
|
|
|
nrf_restart_tx_time = NRF_RESTART_TIMEOUT;
|
|
|
|
}
|
|
|
|
|
|
|
|
return res;
|
|
|
|
}
|
|
|
|
|
2015-10-08 14:09:39 -07:00
|
|
|
static THD_FUNCTION(tx_thread, arg) {
|
2015-04-11 01:07:36 -07:00
|
|
|
(void)arg;
|
|
|
|
|
|
|
|
chRegSetThreadName("Nrf TX");
|
|
|
|
|
|
|
|
for(;;) {
|
2016-01-24 04:44:51 -08:00
|
|
|
nosend_cnt++;
|
|
|
|
|
|
|
|
if (nosend_cnt >= ALIVE_INTERVAL) {
|
|
|
|
uint8_t pl[2];
|
|
|
|
int32_t index = 0;
|
|
|
|
pl[index++] = MOTE_PACKET_ALIVE;
|
|
|
|
rf_tx_wrapper((char*)pl, index);
|
|
|
|
nosend_cnt = 0;
|
|
|
|
}
|
2015-04-11 01:07:36 -07:00
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
chThdSleepMilliseconds(1);
|
2015-04-11 01:07:36 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
}
|
|
|
|
|
2015-10-08 14:09:39 -07:00
|
|
|
static THD_FUNCTION(rx_thread, arg) {
|
2015-04-11 01:07:36 -07:00
|
|
|
(void)arg;
|
|
|
|
|
2015-12-27 16:50:23 -08:00
|
|
|
chRegSetThreadName("Nrf RX");
|
2015-04-11 01:07:36 -07:00
|
|
|
|
|
|
|
for(;;) {
|
|
|
|
uint8_t buf[32];
|
|
|
|
int len;
|
|
|
|
int pipe;
|
|
|
|
|
|
|
|
for(;;) {
|
2015-07-31 14:26:50 -07:00
|
|
|
int res = rfhelp_read_rx_data_crc((char*)buf, &len, &pipe);
|
2016-01-24 04:44:51 -08:00
|
|
|
|
2015-04-11 01:07:36 -07:00
|
|
|
chuck_data cdata;
|
2015-12-27 16:50:23 -08:00
|
|
|
int32_t ind = 0;
|
2015-04-11 01:07:36 -07:00
|
|
|
int buttons;
|
|
|
|
|
|
|
|
// If something was read
|
|
|
|
if (res >= 0) {
|
|
|
|
MOTE_PACKET packet = buf[0];
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
nrf_restart_rx_time = NRF_RESTART_TIMEOUT;
|
|
|
|
|
2015-04-11 01:07:36 -07:00
|
|
|
switch (packet) {
|
|
|
|
case MOTE_PACKET_BATT_LEVEL:
|
|
|
|
// TODO!
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MOTE_PACKET_BUTTONS:
|
2015-12-27 16:50:23 -08:00
|
|
|
ind = 1;
|
|
|
|
mstate.js_x = buf[ind++];
|
|
|
|
mstate.js_y = buf[ind++];
|
|
|
|
buttons = buf[ind++];
|
2015-04-11 01:07:36 -07:00
|
|
|
mstate.bt_c = buttons & (1 << 0);
|
|
|
|
mstate.bt_z = buttons & (1 << 1);
|
|
|
|
mstate.bt_push = buttons & (1 << 2);
|
2015-12-27 16:50:23 -08:00
|
|
|
mstate.vbat = (float)buffer_get_int16(buf, &ind) / 1000.0;
|
2015-04-11 01:07:36 -07:00
|
|
|
|
2015-05-20 07:13:48 -07:00
|
|
|
cdata.js_x = 255 - mstate.js_x;
|
2015-04-11 01:07:36 -07:00
|
|
|
cdata.js_y = mstate.js_y;
|
|
|
|
cdata.bt_c = mstate.bt_c;
|
|
|
|
cdata.bt_z = mstate.bt_z;
|
|
|
|
|
|
|
|
app_nunchuk_update_output(&cdata);
|
|
|
|
break;
|
|
|
|
|
2015-12-27 16:50:23 -08:00
|
|
|
case MOTE_PACKET_FILL_RX_BUFFER:
|
|
|
|
memcpy(rx_buffer + buf[1], buf + 2, len - 2);
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MOTE_PACKET_FILL_RX_BUFFER_LONG: {
|
|
|
|
int rxbuf_ind = (unsigned int)buf[1] << 8;
|
|
|
|
rxbuf_ind |= buf[2];
|
|
|
|
if (rxbuf_ind < RX_BUFFER_SIZE) {
|
|
|
|
memcpy(rx_buffer + rxbuf_ind, buf + 3, len - 3);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MOTE_PACKET_PROCESS_RX_BUFFER: {
|
|
|
|
ind = 1;
|
|
|
|
int rxbuf_len = (unsigned int)buf[ind++] << 8;
|
|
|
|
rxbuf_len |= (unsigned int)buf[ind++];
|
|
|
|
|
|
|
|
if (rxbuf_len > RX_BUFFER_SIZE) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t crc_high = buf[ind++];
|
|
|
|
uint8_t crc_low = buf[ind++];
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
memcpy(rx_buffer + rxbuf_len - (len - ind), buf + ind, len - ind);
|
|
|
|
|
2015-12-27 16:50:23 -08:00
|
|
|
if (crc16(rx_buffer, rxbuf_len)
|
|
|
|
== ((unsigned short) crc_high << 8
|
|
|
|
| (unsigned short) crc_low)) {
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
// Wait a bit in case retries are still made
|
|
|
|
chThdSleepMilliseconds(2);
|
|
|
|
|
2015-12-27 16:50:23 -08:00
|
|
|
commands_set_send_func(nrf_driver_send_buffer);
|
|
|
|
commands_process_packet(rx_buffer, rxbuf_len);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
|
|
case MOTE_PACKET_PROCESS_SHORT_BUFFER:
|
2016-01-24 04:44:51 -08:00
|
|
|
// Wait a bit in case retries are still made
|
|
|
|
chThdSleepMilliseconds(2);
|
|
|
|
|
2015-12-27 16:50:23 -08:00
|
|
|
commands_set_send_func(nrf_driver_send_buffer);
|
|
|
|
commands_process_packet(buf + 1, len - 1);
|
|
|
|
break;
|
|
|
|
|
2015-04-11 01:07:36 -07:00
|
|
|
default:
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// Stop when there is no more data to read.
|
|
|
|
if (res <= 0) {
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
// Sleep a bit to prevent locking the other threads.
|
2015-12-27 16:50:23 -08:00
|
|
|
chThdSleepMilliseconds(1);
|
2015-04-11 01:07:36 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
chThdSleepMilliseconds(5);
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
// Restart the nrf if nothing has been received for a while
|
|
|
|
if (nrf_restart_rx_time > 0 && nrf_restart_tx_time > 0) {
|
|
|
|
nrf_restart_rx_time -= 5;
|
|
|
|
nrf_restart_tx_time -= 5;
|
|
|
|
} else {
|
|
|
|
rfhelp_power_up();
|
2015-04-11 01:07:36 -07:00
|
|
|
rfhelp_restart();
|
2016-01-24 04:44:51 -08:00
|
|
|
nrf_restart_rx_time = NRF_RESTART_TIMEOUT;
|
|
|
|
nrf_restart_tx_time = NRF_RESTART_TIMEOUT;
|
2015-04-11 01:07:36 -07:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
2015-12-27 16:50:23 -08:00
|
|
|
|
|
|
|
void nrf_driver_send_buffer(unsigned char *data, unsigned int len) {
|
|
|
|
uint8_t send_buffer[MAX_PL_LEN];
|
|
|
|
|
|
|
|
if (len <= (MAX_PL_LEN - 1)) {
|
|
|
|
uint32_t ind = 0;
|
|
|
|
send_buffer[ind++] = MOTE_PACKET_PROCESS_SHORT_BUFFER;
|
|
|
|
memcpy(send_buffer + ind, data, len);
|
|
|
|
ind += len;
|
2016-01-24 04:44:51 -08:00
|
|
|
rf_tx_wrapper((char*)send_buffer, ind);
|
|
|
|
nosend_cnt = 0;
|
2015-12-27 16:50:23 -08:00
|
|
|
} else {
|
|
|
|
unsigned int end_a = 0;
|
2016-01-24 04:44:51 -08:00
|
|
|
unsigned int len2 = len - (MAX_PL_LEN - 5);
|
2015-12-27 16:50:23 -08:00
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
for (unsigned int i = 0;i < len2;i += (MAX_PL_LEN - 2)) {
|
2015-12-27 16:50:23 -08:00
|
|
|
if (i > 255) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
end_a = i + (MAX_PL_LEN - 2);
|
|
|
|
|
2015-12-27 16:50:23 -08:00
|
|
|
uint8_t send_len = (MAX_PL_LEN - 2);
|
|
|
|
send_buffer[0] = MOTE_PACKET_FILL_RX_BUFFER;
|
|
|
|
send_buffer[1] = i;
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
if ((i + (MAX_PL_LEN - 2)) <= len2) {
|
2015-12-27 16:50:23 -08:00
|
|
|
memcpy(send_buffer + 2, data + i, send_len);
|
|
|
|
} else {
|
2016-01-24 04:44:51 -08:00
|
|
|
send_len = len2 - i;
|
2015-12-27 16:50:23 -08:00
|
|
|
memcpy(send_buffer + 2, data + i, send_len);
|
|
|
|
}
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
rf_tx_wrapper((char*)send_buffer, send_len + 2);
|
|
|
|
nosend_cnt = 0;
|
2015-12-27 16:50:23 -08:00
|
|
|
}
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
for (unsigned int i = end_a;i < len2;i += (MAX_PL_LEN - 3)) {
|
2015-12-27 16:50:23 -08:00
|
|
|
uint8_t send_len = (MAX_PL_LEN - 3);
|
|
|
|
send_buffer[0] = MOTE_PACKET_FILL_RX_BUFFER_LONG;
|
|
|
|
send_buffer[1] = i >> 8;
|
|
|
|
send_buffer[2] = i & 0xFF;
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
if ((i + (MAX_PL_LEN - 3)) <= len2) {
|
2015-12-27 16:50:23 -08:00
|
|
|
memcpy(send_buffer + 3, data + i, send_len);
|
|
|
|
} else {
|
2016-01-24 04:44:51 -08:00
|
|
|
send_len = len2 - i;
|
2015-12-27 16:50:23 -08:00
|
|
|
memcpy(send_buffer + 3, data + i, send_len);
|
|
|
|
}
|
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
rf_tx_wrapper((char*)send_buffer, send_len + 3);
|
|
|
|
nosend_cnt = 0;
|
2015-12-27 16:50:23 -08:00
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t ind = 0;
|
|
|
|
send_buffer[ind++] = MOTE_PACKET_PROCESS_RX_BUFFER;
|
|
|
|
send_buffer[ind++] = len >> 8;
|
|
|
|
send_buffer[ind++] = len & 0xFF;
|
|
|
|
unsigned short crc = crc16(data, len);
|
|
|
|
send_buffer[ind++] = (uint8_t)(crc >> 8);
|
|
|
|
send_buffer[ind++] = (uint8_t)(crc & 0xFF);
|
2016-01-24 04:44:51 -08:00
|
|
|
memcpy(send_buffer + 5, data + len2, len - len2);
|
|
|
|
ind += len - len2;
|
2015-12-27 16:50:23 -08:00
|
|
|
|
2016-01-24 04:44:51 -08:00
|
|
|
rf_tx_wrapper((char*)send_buffer, ind);
|
|
|
|
nosend_cnt = 0;
|
2015-12-27 16:50:23 -08:00
|
|
|
}
|
|
|
|
}
|