ESP32 Forum

Code: Select all

void app_main(void)
{
    uint32_t red = 0;
    uint32_t green = 0;
    uint32_t blue = 0;
    uint16_t hue = 0;
    uint16_t value = 0;
    uint16_t start_rgb = 0;
    uint64_t cycle_time = 0;

    rmt_tx_channel_config_t tx_channel_cfg = {
        .clk_src = RMT_CLK_SRC_DEFAULT,
        .resolution_hz = 10000000, // 10MHz, 1 tick = 0.1us (led strip needs a high resolution)
        .trans_queue_depth = 4,
    };

    printf("install tx channels\r\n");
    rmt_channel_handle_t tx_channels[TEST_RMT_CHANS] = {NULL};
    int gpio_nums[TEST_RMT_CHANS] = {5, 18, 19, 16};
    size_t mem_blk_syms = 128;
    for (int i = 0; i < TEST_RMT_CHANS; i++)
    {
        tx_channel_cfg.gpio_num = gpio_nums[i];
        tx_channel_cfg.mem_block_symbols = mem_blk_syms;
        ESP_ERROR_CHECK(rmt_new_tx_channel(&tx_channel_cfg, &tx_channels[i]));
    }

    printf("install led strip encoders\r\n");
    rmt_encoder_handle_t led_strip_encoders[TEST_RMT_CHANS] = {NULL};

    led_strip_encoder_config_t encoder_config = {
        .resolution = RMT_LED_STRIP_RESOLUTION_HZ,
    };

    for (int i = 0; i < TEST_RMT_CHANS; i++)
    {
        ESP_ERROR_CHECK(rmt_new_led_strip_encoder(&encoder_config, &led_strip_encoders[i]));
    }
    printf("register tx event callback\r\n");
    rmt_tx_event_callbacks_t cbs = {
        .on_trans_done = test_rmt_tx_done_cb_record_time};
    int64_t record_stop_time[TEST_RMT_CHANS] = {};
    for (int i = 0; i < TEST_RMT_CHANS; i++)
    {
        ESP_ERROR_CHECK(rmt_tx_register_event_callbacks(tx_channels[i], &cbs, &record_stop_time[i]));
    }

    printf("enable tx channels\r\n");
    for (int i = 0; i < TEST_RMT_CHANS; i++)
    {
        ESP_ERROR_CHECK(rmt_enable(tx_channels[i]));
    }

    uint8_t leds_grb[TEST_LED_NUM * 3] = {};
    // color: Material Design Green-A200 (#69F0AE)
    for (int i = 0; i < TEST_LED_NUM * 3; i += 3)
    {
        leds_grb[i + 0] = 0xF0;
        leds_grb[i + 1] = 0x69;
        leds_grb[i + 2] = 0xAE;
    }

    printf("transmit without synchronization\r\n");
    rmt_transmit_config_t transmit_config = {
        .loop_count = 0, // no loop
    };

    while (1)
    {
        if (value == 0){
            struct timeval cycle_time_now;
            gettimeofday(&cycle_time_now, NULL);
            cycle_time = (int64_t)cycle_time_now.tv_sec * 1000000L + (int64_t)cycle_time_now.tv_usec;
        }
        if(value % 255 == 0)
        {
            struct timeval cycle_time_now;
            gettimeofday(&cycle_time_now, NULL);
            cycle_time = (int64_t)cycle_time_now.tv_sec * 1000000L + (int64_t)cycle_time_now.tv_usec - cycle_time;
            double fps = (int)cycle_time / 1000 / 255;
            fps = 1000 / fps;
            printf("\t\t\t0 - 255 cycle_time: %lld fps %.2f\n", cycle_time / 1000, fps);
            cycle_time = (int64_t)cycle_time_now.tv_sec * 1000000L + (int64_t)cycle_time_now.tv_usec;
        }
        for (int i = 0; i < 3; i++)
        {
            for (int j = i; j < TEST_LED_NUM; j += 3)
            {
                // Build RGB pixels
                hue = j * 360 / TEST_LED_NUM + start_rgb;
                // hue = start_rgb;
                led_strip_hsv2rgb(hue, 0, value % 255, &red, &green, &blue);
                leds_grb[j * 3 + 0] = green;
                leds_grb[j * 3 + 1] = blue;
                leds_grb[j * 3 + 2] = red;
            }
        }
        // printf("start_rgb: %d\thue: %d\tred: %d\tgreen: %d\tblue: %d\t\n", start_rgb, hue, (uint16_t)red, (uint16_t)green, (uint16_t)blue);
        
        struct timeval tv_now;
        uint64_t last_time[TEST_RMT_CHANS] = {NULL};
        for (int i = 0; i < TEST_RMT_CHANS; i++)
        {
            gettimeofday(&tv_now, NULL);
            last_time[i] = (int64_t)tv_now.tv_sec * 1000000L + (int64_t)tv_now.tv_usec;
            ESP_ERROR_CHECK(rmt_transmit(tx_channels[i], led_strip_encoders[i], leds_grb, TEST_LED_NUM * 3, &transmit_config));
        }
        for (int i = 0; i < TEST_RMT_CHANS; i++)
        {
            ESP_ERROR_CHECK(rmt_tx_wait_all_done(tx_channels[i], -1));
        }
        start_rgb = 1;
        value += 1;
    }
}