324 lines
8.0 KiB
C++
324 lines
8.0 KiB
C++
#include "step1.h"
|
|
|
|
static const char *STEP1_TAG = "step1";
|
|
|
|
static char* COLOR_NAMES[] = {
|
|
"green",
|
|
"red",
|
|
"yellow",
|
|
"purple"
|
|
};
|
|
|
|
static char* NUM_NAMES[] = {
|
|
"one",
|
|
"two",
|
|
"three",
|
|
"four"
|
|
};
|
|
|
|
static uint8_t NEOPIXEL_COLORS[4][3] = {
|
|
{0, 20, 0},
|
|
{20, 0, 0},
|
|
{20, 20, 0},
|
|
{15, 0, 20},
|
|
};
|
|
|
|
static std::random_device my_rd;
|
|
static std::mt19937 my_gen(my_rd());
|
|
static std::uniform_int_distribution<> zero_to_one(0, 1);
|
|
static std::uniform_int_distribution<> zero_to_two(0, 2);
|
|
static std::uniform_int_distribution<> zero_to_three(0, 3);
|
|
|
|
static lv_obj_t *scr;
|
|
static lv_obj_t *text = NULL;
|
|
|
|
static lv_style_t style_screen;
|
|
static lv_style_t green_text;
|
|
static lv_style_t red_text;
|
|
static lv_style_t yellow_text;
|
|
static lv_style_t purple_text;
|
|
|
|
static int switch_leds[] = {0, 0, 0, 0};
|
|
|
|
static lv_style_t* color_styles[] = {
|
|
&green_text,
|
|
&red_text,
|
|
&yellow_text,
|
|
&purple_text
|
|
};
|
|
|
|
static void init_step(void) {
|
|
scr = lv_disp_get_scr_act(NULL);
|
|
|
|
// Set the background color of the display to black.
|
|
lv_style_init(&style_screen);
|
|
lv_style_set_bg_color(&style_screen, lv_color_black());
|
|
lv_style_set_text_font(&style_screen, &lv_font_montserrat_32);
|
|
lv_obj_add_style(lv_scr_act(), &style_screen, LV_STATE_DEFAULT);
|
|
|
|
// rgb565
|
|
lv_color_t green;
|
|
green.full = 0x27e0;
|
|
lv_color_t red;
|
|
red.full = 0xf800;
|
|
lv_color_t yellow;
|
|
yellow.full = 0xffa9;
|
|
lv_color_t purple;
|
|
purple.full = 0x881f;
|
|
|
|
lv_style_init(&green_text);
|
|
lv_style_set_text_color(&green_text, green);
|
|
|
|
lv_style_init(&red_text);
|
|
lv_style_set_text_color(&red_text, red);
|
|
|
|
lv_style_init(&yellow_text);
|
|
lv_style_set_text_color(&yellow_text, yellow);
|
|
|
|
lv_style_init(&purple_text);
|
|
lv_style_set_text_color(&purple_text, purple);
|
|
|
|
text = lv_label_create(scr);
|
|
}
|
|
|
|
static void clean_up_step(void) {
|
|
lv_obj_clean(text);
|
|
lv_obj_clean(scr);
|
|
}
|
|
|
|
static void generate_switch_leds(void) {
|
|
int colors[4] = {0, 1, 2, 3};
|
|
|
|
int idx = zero_to_three(my_gen);
|
|
switch_leds[0] = colors[idx];
|
|
colors[idx] = colors[3];
|
|
|
|
idx = zero_to_two(my_gen);
|
|
switch_leds[1] = colors[idx];
|
|
colors[idx] = colors[2];
|
|
|
|
idx = zero_to_one(my_gen);
|
|
switch_leds[2] = colors[idx];
|
|
colors[idx] = colors[1];
|
|
|
|
switch_leds[3] = colors[0];
|
|
|
|
ESP_LOGI(STEP1_TAG, "%d, %d, %d, %d", switch_leds[0], switch_leds[1], switch_leds[2], switch_leds[3]);
|
|
|
|
for (int i = 0; i < 4; i++) {
|
|
uint8_t* rgb = NEOPIXEL_COLORS[switch_leds[i]];
|
|
led_strip_set_pixel(leds, Led::switch1 - i, rgb[0], rgb[1], rgb[2]);
|
|
}
|
|
led_strip_refresh(leds);
|
|
}
|
|
|
|
static int generate_part_a(void) {
|
|
int text_color = zero_to_three(my_gen);
|
|
int text_idx = zero_to_three(my_gen);
|
|
|
|
char* text_string = COLOR_NAMES[text_idx];
|
|
|
|
if (xSemaphoreTake(xGuiSemaphore, portMAX_DELAY) == pdTRUE) {
|
|
lv_label_set_text(text, text_string);
|
|
lv_obj_center(text);
|
|
lv_obj_add_style(text, color_styles[text_color], LV_STATE_DEFAULT);
|
|
xSemaphoreGive(xGuiSemaphore);
|
|
}
|
|
|
|
generate_switch_leds();
|
|
|
|
// the correct answer is the text color
|
|
return text_color;
|
|
}
|
|
|
|
static int generate_part_b(void) {
|
|
int is_color = zero_to_one(my_gen) == 0;
|
|
if (is_color) {
|
|
return generate_part_a();
|
|
}
|
|
|
|
int text_color = zero_to_three(my_gen);
|
|
int text_number = zero_to_three(my_gen);
|
|
|
|
char* text_string = NUM_NAMES[text_number];
|
|
|
|
if (xSemaphoreTake(xGuiSemaphore, portMAX_DELAY) == pdTRUE) {
|
|
lv_label_set_text(text, text_string);
|
|
lv_obj_center(text);
|
|
lv_obj_add_style(text, color_styles[text_color], LV_STATE_DEFAULT);
|
|
xSemaphoreGive(xGuiSemaphore);
|
|
}
|
|
|
|
generate_switch_leds();
|
|
|
|
// the correct answer is the number
|
|
return text_number;
|
|
}
|
|
|
|
static int generate_part_c(void) {
|
|
int type = zero_to_two(my_gen);
|
|
if (type != 0) {
|
|
return generate_part_b();
|
|
}
|
|
|
|
int text_color = zero_to_three(my_gen);
|
|
char* text_string = "switch";
|
|
|
|
if (xSemaphoreTake(xGuiSemaphore, portMAX_DELAY) == pdTRUE) {
|
|
lv_label_set_text(text, text_string);
|
|
lv_obj_center(text);
|
|
lv_obj_add_style(text, color_styles[text_color], LV_STATE_DEFAULT);
|
|
xSemaphoreGive(xGuiSemaphore);
|
|
}
|
|
generate_switch_leds();
|
|
|
|
// the correct answer is the switch
|
|
return text_color + 4;
|
|
}
|
|
|
|
static void part_a(void) {
|
|
stop_module_timer();
|
|
set_module_time(30);
|
|
|
|
lcd_clear(&lcd);
|
|
lcd_set_cursor(&lcd, 1, 1);
|
|
lcd_print(&lcd, "COLOR");
|
|
ESP_ERROR_CHECK(led_strip_set_pixel(leds, Led::char_lcd, 20, 0, 20));
|
|
ESP_ERROR_CHECK(led_strip_refresh(leds));
|
|
|
|
int times = 0;
|
|
int correct = generate_part_a();
|
|
|
|
ButtonKey button;
|
|
SwitchKey switch_;
|
|
|
|
while (times < 15) {
|
|
while (get_pressed_button(&button)) {
|
|
start_module_timer();
|
|
if ((int)(button) == correct) {
|
|
times++;
|
|
if (times >= 15) break;
|
|
correct = generate_part_a();
|
|
} else {
|
|
strike("Incorrect color action");
|
|
}
|
|
}
|
|
while (get_flipped_switch(&switch_)) {
|
|
strike("Incorrect color action");
|
|
}
|
|
|
|
vTaskDelay(pdMS_TO_TICKS(10));
|
|
}
|
|
|
|
|
|
if (xSemaphoreTake(xGuiSemaphore, portMAX_DELAY) == pdTRUE) {
|
|
lv_label_set_text(text, "");
|
|
xSemaphoreGive(xGuiSemaphore);
|
|
}
|
|
vTaskDelay(pdMS_TO_TICKS(80));
|
|
play_raw(MOUNT_POINT "/correct.pcm");
|
|
}
|
|
|
|
static void part_b(void) {
|
|
stop_module_timer();
|
|
set_module_time(25);
|
|
|
|
lcd_clear(&lcd);
|
|
lcd_set_cursor(&lcd, 1, 1);
|
|
lcd_print(&lcd, "NUMBER");
|
|
ESP_ERROR_CHECK(led_strip_set_pixel(leds, Led::char_lcd, 0, 0, 30));
|
|
ESP_ERROR_CHECK(led_strip_refresh(leds));
|
|
|
|
int times = 0;
|
|
int correct = generate_part_b();
|
|
|
|
ButtonKey button;
|
|
SwitchKey switch_;
|
|
|
|
while (times < 15) {
|
|
while (get_pressed_button(&button)) {
|
|
start_module_timer();
|
|
if ((int)(button) == correct) {
|
|
times++;
|
|
if (times >= 15) break;
|
|
correct = generate_part_b();
|
|
} else {
|
|
strike("Incorrect color action");
|
|
}
|
|
}
|
|
while (get_flipped_switch(&switch_)) {
|
|
strike("Incorrect color action");
|
|
}
|
|
|
|
vTaskDelay(pdMS_TO_TICKS(10));
|
|
}
|
|
|
|
if (xSemaphoreTake(xGuiSemaphore, portMAX_DELAY) == pdTRUE) {
|
|
lv_label_set_text(text, "");
|
|
xSemaphoreGive(xGuiSemaphore);
|
|
}
|
|
vTaskDelay(pdMS_TO_TICKS(80));
|
|
play_raw(MOUNT_POINT "/correct.pcm");
|
|
}
|
|
|
|
static void part_c(void) {
|
|
stop_module_timer();
|
|
set_module_time(20);
|
|
|
|
lcd_clear(&lcd);
|
|
lcd_set_cursor(&lcd, 1, 1);
|
|
lcd_print(&lcd, "SWITCH");
|
|
ESP_ERROR_CHECK(led_strip_set_pixel(leds, Led::char_lcd, 20, 20, 0));
|
|
ESP_ERROR_CHECK(led_strip_refresh(leds));
|
|
|
|
int times = 0;
|
|
int correct = generate_part_c();
|
|
|
|
ButtonKey button;
|
|
SwitchKey switch_;
|
|
|
|
while (times < 15) {
|
|
while (get_pressed_button(&button)) {
|
|
start_module_timer();
|
|
if ((int)(button) == correct) {
|
|
times++;
|
|
if (times >= 15) break;
|
|
correct = generate_part_c();
|
|
} else {
|
|
strike("Incorrect color action");
|
|
}
|
|
}
|
|
while (get_flipped_switch(&switch_)) {
|
|
start_module_timer();
|
|
if (switch_leds[(int)(switch_)] + 4 == correct) {
|
|
times++;
|
|
if (times >= 10) break;
|
|
correct = generate_part_c();
|
|
} else {
|
|
strike("Incorrect color action");
|
|
}
|
|
}
|
|
|
|
vTaskDelay(pdMS_TO_TICKS(10));
|
|
}
|
|
|
|
if (xSemaphoreTake(xGuiSemaphore, portMAX_DELAY) == pdTRUE) {
|
|
lv_label_set_text(text, "");
|
|
xSemaphoreGive(xGuiSemaphore);
|
|
}
|
|
vTaskDelay(pdMS_TO_TICKS(80));
|
|
play_raw(MOUNT_POINT "/correct.pcm");
|
|
}
|
|
|
|
void step1(void) {
|
|
SwitchKey switch_;
|
|
while (get_flipped_switch(&switch_));
|
|
|
|
init_step();
|
|
part_a();
|
|
part_b();
|
|
part_c();
|
|
clean_up_step();
|
|
// TODO: flash lights
|
|
}
|