clean bomb and step 1

2024-08-07 18:15:31 -05:00
parent ee178f3340
commit ed9554171b
12 changed files with 396 additions and 41 deletions
@@ -1,10 +1,310 @@
 #ifndef STEP_1_HPP
 #define STEP_1_HPP

+#include <random>
+
 static const char *STEP1_TAG = "step1";

-void step1(void) {
-    
+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},
+};
+
+std::random_device my_rd;
+std::mt19937 my_gen(my_rd());
+std::uniform_int_distribution<> zero_to_one(0, 1);
+std::uniform_int_distribution<> zero_to_two(0, 2);
+std::uniform_int_distribution<> zero_to_three(0, 3);
+
+static lv_obj_t *scr;
+static lv_obj_t *text = NULL;
+
+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
+};
+
+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);
 }

+void clean_up_step(void) {
+    lv_obj_clean(text);
+    lv_obj_clean(scr);
+}
+
+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);
+}
+
+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];
+
+    lv_label_set_text(text, text_string);
+    lv_obj_center(text);
+    lv_obj_add_style(text, color_styles[text_color], LV_STATE_DEFAULT);
+
+    generate_switch_leds();
+
+    // the correct answer is the text color
+    return text_color;
+}
+
+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];
+
+    lv_label_set_text(text, text_string);
+    lv_obj_center(text);
+    lv_obj_add_style(text, color_styles[text_color], LV_STATE_DEFAULT);
+
+    generate_switch_leds();
+
+    // the correct answer is the number
+    return text_number;
+}
+
+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";
+
+    lv_label_set_text(text, text_string);
+    lv_obj_center(text);
+    lv_obj_add_style(text, color_styles[text_color], LV_STATE_DEFAULT);
+
+    generate_switch_leds();
+
+    // the correct answer is the switch
+    return text_color + 4;
+}
+
+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));
+    }
+
+    lv_label_set_text(text, "");
+    vTaskDelay(pdMS_TO_TICKS(80));
+    play_raw(MOUNT_POINT "/correct.pcm");
+}
+
+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));
+    }
+
+    lv_label_set_text(text, "");
+    vTaskDelay(pdMS_TO_TICKS(80));
+    play_raw(MOUNT_POINT "/correct.pcm");
+}
+
+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));
+    }
+
+    lv_label_set_text(text, "");
+    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();
+}
+
+
+
 #endif /* STEP_1_HPP */
@@ -9,7 +9,6 @@ static const char *STEP6_TAG = "step6";

 static uint8_t cut_wires = 0;

-static void correct_wire_task(void *arg);

 void step6(void) {
    get_cut_wires();
@@ -28,10 +27,8 @@ void step6(void) {

        for (int i = 0; i < NUM_WIRES; i++) {
            if (just_cut_wires & (1<<i)) {
-                // ESP_LOGI(STEP6_TAG, "Wire %d cut", i);
                if (solution[i]) {
-                    xTaskCreate(correct_wire_task, "correct_wire", 4096, NULL, 5, NULL);
-                    // ESP_LOGI(STEP6_TAG, "correct wire");
+                    play_raw(MOUNT_POINT "/correct.pcm");
                } else {
                    strike("Cut wrong wire");
                }
@@ -57,10 +54,4 @@ void step6(void) {
    }
 }

-static void correct_wire_task(void *arg) {
-    ESP_LOGI(STEP6_TAG, "Correct!");
-
-    vTaskDelete(NULL);
-}
-
 #endif /* STEP_6_HPP */