diff options
| -rw-r--r-- | client/commands.c | 33 | ||||
| -rw-r--r-- | client/commands.h | 8 | ||||
| -rw-r--r-- | client/ui_dirc.c | 69 | ||||
| -rw-r--r-- | robot/hypervisor.c | 2 | ||||
| -rw-r--r-- | robot/main.c | 1 | ||||
| -rw-r--r-- | robot/mode_dirc.c | 3 | ||||
| -rw-r--r-- | robot/modes.c | 10 | ||||
| -rw-r--r-- | robot/sercomm.c | 5 | ||||
| -rw-r--r-- | robot/sim.c | 2 | ||||
| -rw-r--r-- | robot/sim.h | 2 | ||||
| -rw-r--r-- | shared/util.c | 2 | ||||
| -rw-r--r-- | shared/util.h | 1 | 
12 files changed, 88 insertions, 50 deletions
diff --git a/client/commands.c b/client/commands.c index 18373f6..778e9c1 100644 --- a/client/commands.c +++ b/client/commands.c @@ -15,42 +15,41 @@ void w2_send_bin(w2_s_bin *data) {  	}  } -w2_s_bin *w2_send_mode(w2_e_mode mode) { -	W2_CREATE_MSG_BIN(w2_s_cmd_mode_rx, msg, msg_bin); - -	msg->opcode = W2_CMD_MODE | W2_CMDDIR_RX; -	msg->mode	= mode; +void w2_send_info() { +	W2_CREATE_MSG_BIN(w2_s_cmd_info_rx, msg, msg_bin); +	msg->opcode = W2_CMD_INFO | W2_CMDDIR_RX;  	w2_send_bin(msg_bin);  	free(msg_bin); -	return NULL;  } -w2_s_bin *w2_send_dirc(uint16_t left, uint16_t right) { +void w2_send_dirc(uint16_t left, uint16_t right) {  	W2_CREATE_MSG_BIN(w2_s_cmd_dirc_rx, msg, msg_bin);  	msg->opcode = W2_CMD_DIRC | W2_CMDDIR_RX;  	msg->left	= w2_bin_hton16(left);  	msg->right	= w2_bin_hton16(right); -	w2_send_bin(msg_bin); -	free(msg_bin); -	return NULL; -} -w2_s_bin *w2_send_info() { -	W2_CREATE_MSG_BIN(w2_s_cmd_info_rx, msg, msg_bin); -	msg->opcode = W2_CMD_INFO | W2_CMDDIR_RX;  	w2_send_bin(msg_bin);  	free(msg_bin); -	return NULL;  } -w2_s_bin *w2_send_ping() { +void w2_send_ping() { +	g_w2_state.ping_id = (uint8_t)rand();  	W2_CREATE_MSG_BIN(w2_s_cmd_ping_rx, msg, msg_bin);  	msg->opcode = W2_CMD_PING | W2_CMDDIR_RX;  	msg->id		= g_w2_state.ping_id; +  	w2_send_bin(msg_bin);  	free(msg_bin);  	w2_timer_start(W2_TIMER_PING); -	return NULL; +} + +void w2_send_mode(w2_e_mode mode) { +	W2_CREATE_MSG_BIN(w2_s_cmd_mode_rx, msg, msg_bin); +	msg->opcode = W2_CMD_MODE | W2_CMDDIR_RX; +	msg->mode	= mode; + +	w2_send_bin(msg_bin); +	free(msg_bin);  } diff --git a/client/commands.h b/client/commands.h index cb01189..5a12424 100644 --- a/client/commands.h +++ b/client/commands.h @@ -4,7 +4,7 @@  #include "../shared/modes.h"  #include "serial.h" -w2_s_bin *w2_send_mode(w2_e_mode mode); -w2_s_bin *w2_send_dirc(uint16_t left, uint16_t right); -w2_s_bin *w2_send_info(); -w2_s_bin *w2_send_ping(); +void w2_send_info(); +void w2_send_ping(); +void w2_send_mode(w2_e_mode mode); +void w2_send_dirc(uint16_t left, uint16_t right); diff --git a/client/ui_dirc.c b/client/ui_dirc.c index dcd4423..7720a4c 100644 --- a/client/ui_dirc.c +++ b/client/ui_dirc.c @@ -3,21 +3,16 @@  #include "commands.h"  #include "ui.h" -void w2_ui_bar_graph(unsigned int percent) { -	unsigned int width = g_w2_ui_width - 7; -	char bar[width]; -	for (unsigned int i = 0; i < width - 2; i++) { -		bar[i + 1] = i > (width - 2) * percent / 100 ? ' ' : '*'; -	} -	bar[0]		   = '|'; -	bar[width - 1] = '|'; -	mvaddnstr(4, 7, bar, width); -} - +/** decay modifier */  #define W2_DIRC_MOD ((double)0.95) -#define W2_DIRC_ADD ((double)13.0) -#define W2_DIRC_PAD ((double)1.10) -#define W2_DIRC_SPL ((unsigned int)20) +/** add value per key press */ +#define W2_DIRC_ADD ((double)17.0) +/** padding */ +#define W2_DIRC_PAD ((double)3.00) +/** average samples */ +#define W2_DIRC_SPL ((unsigned int)14) +/** steering padding */ +#define W2_DIRC_STP ((double)0.2)  int w2_avg(int *samples, unsigned int sample_count) {  	double total = 0; @@ -49,6 +44,46 @@ W2_DIRC_MOTOR_DRIVER(r);  void w2_ui_dirc_init() { w2_send_mode(W2_M_DIRC); } +void w2_ui_bar_graph(unsigned int y, unsigned int x, unsigned int width, double value) { +	char temp[width]; +	temp[0]			= '|'; +	temp[width - 1] = '|'; +	for (unsigned int i = 0; i < width - 2; i++) temp[i + 1] = i < width * value ? '*' : ' '; + +	mvaddnstr(y, x, temp, width); +} + +void w2_ui_bar_graph_pm(unsigned int y, unsigned int x, unsigned int width, double value) { +	char temp[width]; +	temp[0]			= '|'; +	temp[width - 1] = '|'; +	width -= 2; +	unsigned int hw = width / 2; +	if (value >= 0) { +		for (unsigned int i = 0; i < width; i++) +			temp[i + 1] = i < hw ? ' ' : (i - hw) < (hw * value) ? '*' : ' '; +	} else { +		for (unsigned int i = 0; i < width; i++) +			temp[i + 1] = i < hw ? ' ' : (i - hw) < (hw * value) ? '*' : ' '; +	} + +	mvaddnstr(y, x, temp, width); +} + +void w2_ui_dirc_paint(int left, int right) { +	mvaddstr(4, 0, "left drive:  "); +	w2_ui_bar_graph_pm(4, 13, g_w2_ui_width - 14, (double)left / 255); +	mvaddstr(5, 0, "right drive: "); +	w2_ui_bar_graph_pm(5, 13, g_w2_ui_width - 14, (double)right / 255); + +	mvaddstr(7, 0, +			 "             controls:\n" +			 "\n" +			 " <q>      <w>       <e>   forward\n" +			 " <a>      <s>       <d>   backward\n" +			 "left     both      right\n"); +} +  void w2_ui_dirc(bool first) {  	if (first) w2_ui_dirc_init();  	int ch			= 0; @@ -66,9 +101,9 @@ void w2_ui_dirc(bool first) {  	int drive_l = w2_dirc_motor_l(lf, lb);  	int drive_r = w2_dirc_motor_r(rf, rb); -	char temp[32] = {0}; -	sprintf(temp, "l: %04i, r: %04i", drive_l, drive_r); -	mvaddstr(4, 0, temp); +	drive_l += drive_r * W2_DIRC_STP; +	drive_r += drive_l * W2_DIRC_STP;  	w2_send_dirc(drive_l, drive_r); +	w2_ui_dirc_paint(drive_l, drive_r);  } diff --git a/robot/hypervisor.c b/robot/hypervisor.c index 4f994a8..478d3a5 100644 --- a/robot/hypervisor.c +++ b/robot/hypervisor.c @@ -27,7 +27,7 @@ void w2_hypervisor_main() {  	unsigned long sercomm_time = get_ms();  	w2_errcatch_main();  	unsigned long errcatch_time = get_ms() - sercomm_time; -	w2_io_main(); +	// w2_io_main();  	unsigned long io_time = get_ms() - errcatch_time;  	w2_modes_main();  	unsigned long mode_time = get_ms() - io_time; diff --git a/robot/main.c b/robot/main.c index b0302bd..2f15e97 100644 --- a/robot/main.c +++ b/robot/main.c @@ -5,7 +5,6 @@  #include "sim.h"  #endif -#include <unistd.h>  int main() {  #ifdef W2_SIM  	w2_sim_setup(); diff --git a/robot/mode_dirc.c b/robot/mode_dirc.c index 9ed9fef..5988816 100644 --- a/robot/mode_dirc.c +++ b/robot/mode_dirc.c @@ -1,10 +1,13 @@  #include "mode_dirc.h"  #include "io.h" +#include "orangutan_shim.h" +  int16_t g_w2_mode_dirc_motor_l = 0;  int16_t g_w2_mode_dirc_motor_r = 0;  void w2_mode_dirc() { +	set_motors(g_w2_mode_dirc_motor_l, g_w2_mode_dirc_motor_r);  	g_w2_io.motor_left.speed  = g_w2_mode_dirc_motor_l;  	g_w2_io.motor_right.speed = g_w2_mode_dirc_motor_r;  } diff --git a/robot/modes.c b/robot/modes.c index f73a809..7decf47 100644 --- a/robot/modes.c +++ b/robot/modes.c @@ -46,13 +46,11 @@ void w2_modes_switch(w2_e_mode new_mode, bool replace) {  	}  	// forward mode change to sercomm -	size_t msg_size		  = sizeof(w2_s_cmd_mode_tx); -	w2_s_cmd_mode_tx *msg = malloc(msg_size); -	msg->opcode			  = W2_CMD_MODE | W2_CMDDIR_TX; -	msg->mode			  = (uint8_t)new_mode; -	w2_s_bin *msg_bin	  = w2_bin_s_alloc(msg_size, (uint8_t *)msg); +	W2_CREATE_MSG_BIN(w2_s_cmd_mode_tx, msg, msg_bin); +	msg->opcode = W2_CMD_MODE | W2_CMDDIR_TX; +	msg->mode	= new_mode; +  	w2_sercomm_append_msg(msg_bin); -	free(msg);  	free(msg_bin);  } diff --git a/robot/sercomm.c b/robot/sercomm.c index deef39f..519568d 100644 --- a/robot/sercomm.c +++ b/robot/sercomm.c @@ -37,10 +37,11 @@ void w2_sercomm_main() {  #ifdef W2_SIM  		w2_sim_print_serial(data);  #endif -		serial_send("\xff", 1); +		serial_send_blocking("\xff", 1);  		for (uint8_t i = 0; i < data->bytes; i++) {  			uint8_t byte = data->data[i]; -			byte == 0xff ? serial_send("\xff\xff", 2) : serial_send((char *)&byte, 1); +			byte == 0xff ? serial_send_blocking("\xff\xff", 2) +						 : serial_send_blocking((char *)&byte, 1);  		}  		g_w2_sercomm_offset = (g_w2_sercomm_offset + 1) % W2_SERCOMM_BUFFER_SIZE;  	} diff --git a/robot/sim.c b/robot/sim.c index 5956fbb..9cce12f 100644 --- a/robot/sim.c +++ b/robot/sim.c @@ -74,7 +74,7 @@ void serial_set_baud_rate(unsigned int rate) {  	simprintfunc("serial_set_baud_rate", "%u", rate);  } -void serial_send(char* message, unsigned int length) { +void serial_send_blocking(char* message, unsigned int length) {  	for (unsigned int byte = 0; byte < length; byte++)  		putc(message[byte] & 0xff, stdout);  	fflush(stdout); diff --git a/robot/sim.h b/robot/sim.h index 9e250da..9d73585 100644 --- a/robot/sim.h +++ b/robot/sim.h @@ -71,7 +71,7 @@ void green_led(unsigned char on); // NOLINT  void clear(); // NOLINT  void play(const char *melody); // NOLINT  void serial_set_baud_rate(unsigned int rate); // NOLINT -void serial_send(char *message, unsigned int length); // NOLINT +void serial_send_blocking(char *message, unsigned int length); // NOLINT  void serial_receive_ring(char *buffer, unsigned char size); // NOLINT  unsigned char serial_get_received_bytes(); // NOLINT  void set_motors(int left, int right); // NOLINT diff --git a/shared/util.c b/shared/util.c index 55f3491..68503e8 100644 --- a/shared/util.c +++ b/shared/util.c @@ -4,3 +4,5 @@ unsigned long w2_util_exp_mov_avg(unsigned long current_avg, unsigned long new_m  	return (unsigned long)((((double)(current_avg)) * ((double)(1.f - W2_EMA_WEIGHT))) +  						   (((double)(new_meas)) * ((double)(W2_EMA_WEIGHT))));  } + +int w2_sign(int n) { return (n > 0) - (n < 0); } diff --git a/shared/util.h b/shared/util.h index 465e043..230c3e4 100644 --- a/shared/util.h +++ b/shared/util.h @@ -7,3 +7,4 @@  #define W2_RANGE(min, val, max) W2_MIN(max, W2_MAX(val, min))  unsigned long w2_util_exp_mov_avg(unsigned long current_avg, unsigned long new_meas); +int w2_sign(int n);  |