1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
|
#include "../shared/protocol.h"
#include "../shared/util.h"
#include "commands.h"
#include "errcatch.h"
#include "ui.h"
unsigned int g_w2_lb = 0;
unsigned int g_w2_lf = 0;
unsigned int g_w2_rb = 0;
unsigned int g_w2_rf = 0;
/** decay modifier */
#define W2_DIRC_MOD ((double)0.95)
/** 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;
for (int i = 0; i < sample_count; i++) {
total += (double)samples[i] / (double)sample_count;
}
return (int)total;
}
#define W2_DIRC_MOTOR_DRIVER(name) \
int w2_dirc_motor_##name(unsigned int forward, unsigned int backward) { \
static unsigned int idx = 0; \
\
static double drive = 0.f; \
static int drive_avg[W2_DIRC_SPL] = {0}; \
\
drive *= W2_DIRC_MOD; \
drive += W2_DIRC_ADD * forward + -W2_DIRC_ADD * backward; \
drive = W2_RANGE(-254, drive, 255); \
\
idx = (idx + 1) % W2_DIRC_SPL; \
drive_avg[idx] = (int)W2_RANGE(-254, drive * W2_DIRC_PAD, 255); \
\
return w2_avg(drive_avg, W2_DIRC_SPL); \
}
W2_DIRC_MOTOR_DRIVER(l);
W2_DIRC_MOTOR_DRIVER(r);
void w2_ui_dirc_init() { return; }
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 ? '*' : ' ';
w2_wmvaddnstr(g_w2_ui_pad_body, 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 ? ' ' : (hw - i) < -(hw * value) ? '*' : ' ';
}
w2_wmvaddnstr(g_w2_ui_pad_body, y, x, temp, width + 2);
}
void w2_ui_dirc_paint(int left, int right) {
w2_wmvaddstr(g_w2_ui_pad_body, 0, 0, "left drive: ");
w2_ui_bar_graph_pm(0, 13, g_w2_ui_width - 13, (double)left / 255);
w2_wmvaddstr(g_w2_ui_pad_body, 1, 0, "right drive: ");
w2_ui_bar_graph_pm(1, 13, g_w2_ui_width - 13, (double)right / 255);
w2_wmvaddstr(g_w2_ui_pad_body, 3, 0,
" controls:\n"
"\n"
" <q> <w> <e> forward\n"
" <a> <s> <d> backward\n"
"left both right\n"
"\n"
"<space> send dirc mode command");
}
void w2_ui_onkey_dirc(int ch) {
if (ch == 'e' || ch == 'w' || ch == KEY_UP || ch == KEY_RIGHT) g_w2_lf++;
if (ch == 'd' || ch == 's' || ch == KEY_DOWN) g_w2_lb++;
if (ch == 'q' || ch == 'w' || ch == KEY_UP || ch == KEY_LEFT) g_w2_rf++;
if (ch == 'a' || ch == 's' || ch == KEY_DOWN) g_w2_rb++;
if (ch == ' ') w2_send_mode(W2_M_DIRC);
}
void w2_ui_tab_dirc(bool first) {
g_w2_ui_pad_body_scroll = 0;
if (first) w2_ui_dirc_init();
int drive_l = w2_dirc_motor_l(g_w2_lf, g_w2_lb);
int drive_r = w2_dirc_motor_r(g_w2_rf, g_w2_rb);
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);
g_w2_lb = 0;
g_w2_lf = 0;
g_w2_rb = 0;
g_w2_rf = 0;
}
|