// Copyright (C) 2022 @filterpaper
// SPDX-License-Identifier: GPL-2.0-or-later
// Inspired by 4x12 fractal from @GEIGEIGEIST

#ifdef ENABLE_RGB_MATRIX_PIXEL_FRACTAL
RGB_MATRIX_EFFECT(PIXEL_FRACTAL)
#    ifdef RGB_MATRIX_CUSTOM_EFFECT_IMPLS

static bool PIXEL_FRACTAL(effect_params_t *params) {
#        define MID_COL ((MATRIX_COLS + 1) / 2)
    static bool     led[MATRIX_ROWS][MID_COL];
    static uint32_t wait_timer = 0;

    if (params->init) {
        rgb_matrix_set_color_all(0, 0, 0);
        memset(led, 0, sizeof(led));
    }

    RGB_MATRIX_USE_LIMITS(led_min, led_max);

    if (g_rgb_timer > wait_timer) {
        rgb_t rgb = rgb_matrix_hsv_to_rgb(rgb_matrix_config.hsv);
        for (uint8_t row = 0; row < MATRIX_ROWS; ++row) {
            // Light columns outward from center, mirroring each side
            for (uint8_t col = 0; col < MID_COL; ++col) {
                rgb_t   index_rgb = led[row][col] ? rgb : (rgb_t){0, 0, 0};
                uint8_t l_idx     = g_led_config.matrix_co[row][col];
                uint8_t r_idx     = g_led_config.matrix_co[row][MATRIX_COLS - 1 - col];
                if (l_idx >= led_min && l_idx < led_max && HAS_ANY_FLAGS(g_led_config.flags[l_idx], params->flags)) {
                    rgb_matrix_set_color(l_idx, index_rgb.r, index_rgb.g, index_rgb.b);
                }
                if (r_idx >= led_min && r_idx < led_max && HAS_ANY_FLAGS(g_led_config.flags[r_idx], params->flags) && r_idx != l_idx) {
                    rgb_matrix_set_color(r_idx, index_rgb.r, index_rgb.g, index_rgb.b);
                }
            }
        }

        // Advance state and reset timer only on the final LED range iteration
        if (!rgb_matrix_check_finished_leds(led_max)) {
            for (uint8_t row = 0; row < MATRIX_ROWS; ++row) {
                memmove(led[row], led[row] + 1, (MID_COL - 1) * sizeof(bool));
                led[row][MID_COL - 1] = !(random8() & 3);
            }
            wait_timer = g_rgb_timer + 3000 / scale16by8(qadd8(rgb_matrix_config.speed, 16), 16);
        }
    }

    return rgb_matrix_check_finished_leds(led_max);
}
#    endif // RGB_MATRIX_CUSTOM_EFFECT_IMPLS
#endif     // ENABLE_RGB_MATRIX_PIXEL_FRACTAL