Files
Кобелев Андрей Андреевич a111352dc5 Прошло 2 года.
2026-03-10 22:54:23 +03:00

574 lines
20 KiB
C

// Open settings to set WiFi and other configurations for both examples:
#include <stdlib.h>
#include <sys/types.h>
#include "epd_display.h"
#include "esp_heap_caps.h"
#include "esp_log.h"
#include "esp_sleep.h"
#include "esp_system.h"
#include "esp_timer.h"
#include "esp_types.h"
#include "freertos/FreeRTOS.h"
#include "freertos/event_groups.h"
#include "freertos/task.h"
#include "settings.h"
// WiFi related
#include "esp_event.h"
#include "esp_log.h"
#include "esp_wifi.h"
#include "lwip/err.h"
#include "lwip/sys.h"
#include "nvs_flash.h"
// HTTP Client + time
#include "esp_http_client.h"
#include "esp_netif.h"
#include "esp_sntp.h"
// JPG decoder is on ESP32 rom for this version
#if ESP_IDF_VERSION_MAJOR >= 4 // IDF 4+
#include "esp32/rom/tjpgd.h"
#else // ESP32 Before IDF 4.0
#include "rom/tjpgd.h"
#endif
#include <inttypes.h>
#include <math.h> // round + pow
#include <stdio.h>
#include <string.h>
#include "esp_task_wdt.h"
#include "epd_highlevel.h"
#include "epdiy.h"
// choose the default demo board depending on the architecture
#ifdef CONFIG_IDF_TARGET_ESP32
#define DEMO_BOARD epd_board_v6
#elif defined(CONFIG_IDF_TARGET_ESP32S3)
#define DEMO_BOARD epd_board_v7
#endif
EpdiyHighlevelState hl;
// Load the EMBED_TXTFILES. Then doing (char*) server_cert_pem_start you get the SSL certificate
// Reference:
// https://docs.espressif.com/projects/esp-idf/en/latest/esp32/api-guides/build-system.html#embedding-binary-data
extern const uint8_t server_cert_pem_start[] asm("_binary_server_cert_pem_start");
// JPEG decoder
JDEC jd;
JRESULT rc;
// Buffers
uint8_t* fb; // EPD 2bpp buffer
uint8_t* source_buf = NULL; // JPG download buffer
uint8_t* decoded_image; // RAW decoded image
static uint8_t tjpgd_work[3096]; // tjpgd 3096 is the minimum size
uint32_t buffer_pos = 0;
uint32_t time_download = 0;
uint32_t time_decomp = 0;
uint32_t time_render = 0;
static const char* jd_errors[] = { "Succeeded",
"Interrupted by output function",
"Device error or wrong termination of input stream",
"Insufficient memory pool for the image",
"Insufficient stream input buffer",
"Parameter error",
"Data format error",
"Right format but not supported",
"Not supported JPEG standard" };
uint8_t gamme_curve[256];
static const char* TAG = "EPDiy";
uint16_t countDataEventCalls = 0;
uint32_t countDataBytes = 0;
uint32_t img_buf_pos = 0;
uint32_t dataLenTotal = 0;
uint64_t startTime = 0;
#if VALIDATE_SSL_CERTIFICATE == true
/* Time aware for ESP32: Important to check SSL certs validity */
void time_sync_notification_cb(struct timeval* tv) {
ESP_LOGI(TAG, "Notification of a time synchronization event");
}
static void initialize_sntp(void) {
ESP_LOGI(TAG, "Initializing SNTP");
sntp_setoperatingmode(SNTP_OPMODE_POLL);
sntp_setservername(0, "pool.ntp.org");
sntp_set_time_sync_notification_cb(time_sync_notification_cb);
#ifdef CONFIG_SNTP_TIME_SYNC_METHOD_SMOOTH
sntp_set_sync_mode(SNTP_SYNC_MODE_SMOOTH);
#endif
sntp_init();
}
static void obtain_time(void) {
initialize_sntp();
// wait for time to be set
time_t now = 0;
struct tm timeinfo = { 0 };
int retry = 0;
const int retry_count = 10;
while (sntp_get_sync_status() == SNTP_SYNC_STATUS_RESET && ++retry < retry_count) {
ESP_LOGI(TAG, "Waiting for system time to be set... (%d/%d)", retry, retry_count);
vTaskDelay(2000 / portTICK_PERIOD_MS);
}
time(&now);
localtime_r(&now, &timeinfo);
}
#endif
//====================================================================================
// This sketch contains support functions to render the Jpeg images
//
// Created by Bodmer 15th Jan 2017
// Refactored by @martinberlin for EPDiy as a Jpeg download and render example
//====================================================================================
// Return the minimum of two values a and b
#define minimum(a, b) (((a) < (b)) ? (a) : (b))
uint8_t find_closest_palette_color(uint8_t oldpixel) {
return oldpixel & 0xF0;
}
//====================================================================================
// Decode and paint onto the Epaper screen
//====================================================================================
void jpegRender(int xpos, int ypos, int width, int height) {
uint16_t ep_width = epd_width();
uint16_t ep_height = epd_height();
#if JPG_DITHERING
unsigned long pixel = 0;
for (uint16_t by = 0; by < ep_height; by++) {
for (uint16_t bx = 0; bx < ep_width; bx++) {
int oldpixel = decoded_image[pixel];
int newpixel = find_closest_palette_color(oldpixel);
int quant_error = oldpixel - newpixel;
decoded_image[pixel] = newpixel;
if (bx < (ep_width - 1))
decoded_image[pixel + 1]
= minimum(255, decoded_image[pixel + 1] + quant_error * 7 / 16);
if (by < (ep_height - 1)) {
if (bx > 0)
decoded_image[pixel + ep_width - 1]
= minimum(255, decoded_image[pixel + ep_width - 1] + quant_error * 3 / 16);
decoded_image[pixel + ep_width]
= minimum(255, decoded_image[pixel + ep_width] + quant_error * 5 / 16);
if (bx < (ep_width - 1))
decoded_image[pixel + ep_width + 1]
= minimum(255, decoded_image[pixel + ep_width + 1] + quant_error * 1 / 16);
}
pixel++;
}
}
#endif
// Write to display
uint64_t drawTime = esp_timer_get_time();
uint32_t padding_x = (epd_rotated_display_width() - width) / 2;
uint32_t padding_y = (epd_rotated_display_height() - height) / 2;
ESP_LOGI("Padding", "x:%" PRIu32 " y:%" PRIu32 "", padding_x, padding_y);
for (uint32_t by = 0; by < height - 1; by++) {
for (uint32_t bx = 0; bx < width; bx++) {
epd_draw_pixel(bx + padding_x, by + padding_y, decoded_image[by * width + bx], fb);
}
}
// calculate how long it took to draw the image
time_render = (esp_timer_get_time() - drawTime) / 1000;
ESP_LOGI("render", "%" PRIu32 " ms - jpeg draw", time_render);
}
void deepsleep() {
epd_deinit();
esp_deep_sleep(1000000LL * 60 * DEEPSLEEP_MINUTES_AFTER_RENDER);
}
static unsigned int feed_buffer(
JDEC* jd,
unsigned char* buff, // Pointer to the read buffer (NULL:skip)
unsigned int nd
) {
uint32_t count = 0;
while (count < nd) {
if (buff != NULL) {
*buff++ = source_buf[buffer_pos];
}
count++;
buffer_pos++;
}
return count;
}
/* User defined call-back function to output decoded RGB bitmap in decoded_image buffer */
static unsigned int tjd_output(
JDEC* jd, /* Decompressor object of current session */
void* bitmap, /* Bitmap data to be output */
JRECT* rect /* Rectangular region to output */
) {
vTaskDelay(0);
uint32_t w = rect->right - rect->left + 1;
uint32_t h = rect->bottom - rect->top + 1;
uint32_t image_width = jd->width;
uint8_t* bitmap_ptr = (uint8_t*)bitmap;
for (uint32_t i = 0; i < w * h; i++) {
uint8_t r = *(bitmap_ptr++);
uint8_t g = *(bitmap_ptr++);
uint8_t b = *(bitmap_ptr++);
// Calculate weighted grayscale
// uint32_t val = ((r * 30 + g * 59 + b * 11) / 100); // original formula
uint32_t val = (r * 38 + g * 75 + b * 15) >> 7; // @vroland recommended formula
int xx = rect->left + i % w;
if (xx < 0 || xx >= image_width) {
continue;
}
int yy = rect->top + i / w;
if (yy < 0 || yy >= jd->height) {
continue;
}
/* Optimization note: If we manage to apply here the epd_draw_pixel directly
then it will be no need to keep a huge raw buffer (But will loose dither) */
decoded_image[yy * image_width + xx] = gamme_curve[val];
}
return 1;
}
//====================================================================================
// This function opens source_buf Jpeg image file and primes the decoder
//====================================================================================
int drawBufJpeg(uint8_t* source_buf, int xpos, int ypos) {
rc = jd_prepare(&jd, feed_buffer, tjpgd_work, sizeof(tjpgd_work), &source_buf);
if (rc != JDR_OK) {
ESP_LOGE(TAG, "JPG jd_prepare error: %s", jd_errors[rc]);
return ESP_FAIL;
}
uint32_t decode_start = esp_timer_get_time();
// Last parameter scales v 1 will reduce the image
rc = jd_decomp(&jd, tjd_output, 0);
if (rc != JDR_OK) {
ESP_LOGE(TAG, "JPG jd_decomp error: %s", jd_errors[rc]);
return ESP_FAIL;
}
time_decomp = (esp_timer_get_time() - decode_start) / 1000;
ESP_LOGI("JPG", "width: %d height: %d\n", jd.width, jd.height);
ESP_LOGI("decode", "%" PRIu32 " ms . image decompression", time_decomp);
// Render the image onto the screen at given coordinates
jpegRender(xpos, ypos, jd.width, jd.height);
return 1;
}
// Handles Htpp events and is in charge of buffering source_buf (jpg compressed image)
esp_err_t _http_event_handler(esp_http_client_event_t* evt) {
switch (evt->event_id) {
case HTTP_EVENT_ERROR:
ESP_LOGE(TAG, "HTTP_EVENT_ERROR");
break;
case HTTP_EVENT_ON_CONNECTED:
ESP_LOGI(TAG, "HTTP_EVENT_ON_CONNECTED");
break;
case HTTP_EVENT_HEADER_SENT:
ESP_LOGI(TAG, "HTTP_EVENT_HEADER_SENT");
break;
case HTTP_EVENT_ON_HEADER:
#if DEBUG_VERBOSE
ESP_LOGI(
TAG, "HTTP_EVENT_ON_HEADER, key=%s, value=%s", evt->header_key, evt->header_value
);
#endif
if (strncmp(evt->header_key, "Content-Length", 14) == 0) {
// Should be big enough to hold the JPEG file size
size_t content_len = atol(evt->header_value);
ESP_LOGI("epdiy", "Allocating content buffer of length %X", content_len);
source_buf = (uint8_t*)heap_caps_malloc(content_len, MALLOC_CAP_SPIRAM);
if (source_buf == NULL) {
ESP_LOGE("main", "Initial alloc source_buf failed!");
}
printf("Free heap after buffers allocation: %X\n", xPortGetFreeHeapSize());
}
break;
case HTTP_EVENT_ON_DATA:
++countDataEventCalls;
#if DEBUG_VERBOSE
if (countDataEventCalls % 10 == 0) {
ESP_LOGI(TAG, "%d len:%d\n", countDataEventCalls, evt->data_len);
}
#endif
dataLenTotal += evt->data_len;
// should be allocated after the Content-Length header was received.
assert(source_buf != NULL);
if (countDataEventCalls == 1)
startTime = esp_timer_get_time();
// Append received data into source_buf
memcpy(&source_buf[img_buf_pos], evt->data, evt->data_len);
img_buf_pos += evt->data_len;
// Optional hexa dump
// ESP_LOG_BUFFER_HEX(TAG, output_buffer, evt->data_len);
break;
case HTTP_EVENT_ON_FINISH:
// Do not draw if it's a redirect (302)
if (esp_http_client_get_status_code(evt->client) == 200) {
printf("%" PRIu32 " bytes read from %s\n\n", img_buf_pos, IMG_URL);
drawBufJpeg(source_buf, 0, 0);
time_download = (esp_timer_get_time() - startTime) / 1000;
ESP_LOGI("www-dw", "%" PRIu32 " ms - download", time_download);
// Refresh display
epd_hl_update_screen(&hl, MODE_GC16, 25);
ESP_LOGI(
"total",
"%" PRIu32 " ms - total time spent\n",
time_download + time_decomp + time_render
);
}
break;
case HTTP_EVENT_DISCONNECTED:
ESP_LOGI(TAG, "HTTP_EVENT_DISCONNECTED\n");
break;
default:
ESP_LOGI(TAG, "HTTP_EVENT_ fallback caught event %d", evt->event_id);
}
return ESP_OK;
}
// Handles http request
static void http_post(void) {
/**
* NOTE: All the configuration parameters for http_client must be specified
* either in URL or as host and path parameters.
* FIX: Uncommenting cert_pem restarts even if providing the right certificate
*/
esp_http_client_config_t config = { .url = IMG_URL,
.event_handler = _http_event_handler,
.buffer_size = HTTP_RECEIVE_BUFFER_SIZE,
.disable_auto_redirect = false,
#if VALIDATE_SSL_CERTIFICATE == true
.cert_pem = (char*)server_cert_pem_start
#endif
};
esp_http_client_handle_t client = esp_http_client_init(&config);
#if DEBUG_VERBOSE
printf("Free heap before HTTP download: %X\n", xPortGetFreeHeapSize());
if (esp_http_client_get_transport_type(client) == HTTP_TRANSPORT_OVER_SSL && config.cert_pem) {
printf("SSL CERT:\n%s\n\n", (char*)server_cert_pem_start);
}
#endif
esp_err_t err = esp_http_client_perform(client);
if (err == ESP_OK) {
// esp_http_client_get_content_length returns a uint64_t in esp-idf v5, so it needs a %lld
// format specifier
ESP_LOGI(
TAG,
"\nIMAGE URL: %s\n\nHTTP GET Status = %d, content_length = %d\n",
IMG_URL,
esp_http_client_get_status_code(client),
(int)esp_http_client_get_content_length(client)
);
} else {
ESP_LOGE(TAG, "\nHTTP GET request failed: %s", esp_err_to_name(err));
}
esp_http_client_cleanup(client);
#if MILLIS_DELAY_BEFORE_SLEEP > 0
vTaskDelay(MILLIS_DELAY_BEFORE_SLEEP / portTICK_PERIOD_MS);
#endif
printf("Go to sleep %d minutes\n", DEEPSLEEP_MINUTES_AFTER_RENDER);
epd_poweroff();
deepsleep();
}
/* FreeRTOS event group to signal when we are connected*/
static EventGroupHandle_t s_wifi_event_group;
/* The event group allows multiple bits for each event, but we only care about two events:
* - we are connected to the AP with an IP
* - we failed to connect after the maximum amount of retries */
#define WIFI_CONNECTED_BIT BIT0
#define WIFI_FAIL_BIT BIT1
static int s_retry_num = 0;
static void event_handler(
void* arg, esp_event_base_t event_base, int32_t event_id, void* event_data
) {
if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_START) {
esp_wifi_connect();
} else if (event_base == WIFI_EVENT && event_id == WIFI_EVENT_STA_DISCONNECTED) {
if (s_retry_num < 5) {
esp_wifi_connect();
s_retry_num++;
ESP_LOGI(TAG, "Retry to connect to the AP");
} else {
xEventGroupSetBits(s_wifi_event_group, WIFI_FAIL_BIT);
ESP_LOGI(
TAG,
"Connect to the AP failed %d times. Going to deepsleep %d minutes",
5,
DEEPSLEEP_MINUTES_AFTER_RENDER
);
deepsleep();
}
} else if (event_base == IP_EVENT && event_id == IP_EVENT_STA_GOT_IP) {
s_retry_num = 0;
xEventGroupSetBits(s_wifi_event_group, WIFI_CONNECTED_BIT);
}
}
// Initializes WiFi the ESP-IDF way
void wifi_init_sta(void) {
s_wifi_event_group = xEventGroupCreate();
ESP_ERROR_CHECK(esp_netif_init());
ESP_ERROR_CHECK(esp_event_loop_create_default());
esp_netif_create_default_wifi_sta();
wifi_init_config_t cfg = WIFI_INIT_CONFIG_DEFAULT();
ESP_ERROR_CHECK(esp_wifi_init(&cfg));
esp_event_handler_instance_t instance_any_id;
esp_event_handler_instance_t instance_got_ip;
ESP_ERROR_CHECK(esp_event_handler_instance_register(
WIFI_EVENT, ESP_EVENT_ANY_ID, &event_handler, NULL, &instance_any_id
));
ESP_ERROR_CHECK(esp_event_handler_instance_register(
IP_EVENT, IP_EVENT_STA_GOT_IP, &event_handler, NULL, &instance_got_ip
));
wifi_config_t wifi_config = {
.sta =
{
.ssid = ESP_WIFI_SSID,
.password = ESP_WIFI_PASSWORD,
/* Setting a password implies station will connect to all security modes including
* WEP/WPA. However these modes are deprecated and not advisable to be used. Incase
* your Access point doesn't support WPA2, these mode can be enabled by commenting
* below line */
.threshold.authmode = WIFI_AUTH_WPA2_PSK,
.pmf_cfg = {.capable = true, .required = false},
},
};
ESP_ERROR_CHECK(esp_wifi_set_mode(WIFI_MODE_STA));
ESP_ERROR_CHECK(esp_wifi_set_config((wifi_interface_t)ESP_IF_WIFI_STA, &wifi_config));
ESP_ERROR_CHECK(esp_wifi_start());
ESP_LOGI(TAG, "wifi_init_sta finished.");
/* Waiting until either the connection is established (WIFI_CONNECTED_BIT) or connection failed
* for the maximum number of re-tries (WIFI_FAIL_BIT). The bits are set by event_handler() (see
* above) */
EventBits_t bits = xEventGroupWaitBits(
s_wifi_event_group, WIFI_CONNECTED_BIT | WIFI_FAIL_BIT, pdFALSE, pdFALSE, portMAX_DELAY
);
/* xEventGroupWaitBits() returns the bits before the call returned, hence we can test which
* event actually happened. */
if (bits & WIFI_CONNECTED_BIT) {
ESP_LOGI(TAG, "Connected to ap SSID:%s", ESP_WIFI_SSID);
} else if (bits & WIFI_FAIL_BIT) {
ESP_LOGI(TAG, "Failed to connect to SSID:%s", ESP_WIFI_SSID);
} else {
ESP_LOGE(TAG, "UNEXPECTED EVENT");
}
/* The event will not be processed after unregister */
ESP_ERROR_CHECK(
esp_event_handler_instance_unregister(IP_EVENT, IP_EVENT_STA_GOT_IP, instance_got_ip)
);
ESP_ERROR_CHECK(
esp_event_handler_instance_unregister(WIFI_EVENT, ESP_EVENT_ANY_ID, instance_any_id)
);
vEventGroupDelete(s_wifi_event_group);
}
void app_main() {
enum EpdInitOptions init_options = EPD_LUT_64K;
// For V6 and below, try to use less memory. V7 queue uses less anyway.
#ifdef CONFIG_IDF_TARGET_ESP32
init_options |= EPD_FEED_QUEUE_8;
#endif
epd_init(&DEMO_BOARD, &ED097TC2, init_options);
// Set VCOM for boards that allow to set this in software (in mV).
// This will print an error if unsupported. In this case,
// set VCOM using the hardware potentiometer and delete this line.
epd_set_vcom(1560);
hl = epd_hl_init(WAVEFORM);
fb = epd_hl_get_framebuffer(&hl);
epd_set_rotation(DISPLAY_ROTATION);
decoded_image = (uint8_t*)heap_caps_malloc(epd_width() * epd_height(), MALLOC_CAP_SPIRAM);
if (decoded_image == NULL) {
ESP_LOGE("main", "Initial alloc back_buf failed!");
}
memset(decoded_image, 255, epd_width() * epd_height());
double gammaCorrection = 1.0 / gamma_value;
for (int gray_value = 0; gray_value < 256; gray_value++)
gamme_curve[gray_value] = round(255 * pow(gray_value / 255.0, gammaCorrection));
// Initialize NVS
esp_err_t ret = nvs_flash_init();
if (ret == ESP_ERR_NVS_NO_FREE_PAGES || ret == ESP_ERR_NVS_NEW_VERSION_FOUND) {
ESP_ERROR_CHECK(nvs_flash_erase());
ret = nvs_flash_init();
}
ESP_ERROR_CHECK(ret);
// WiFi log level set only to Error otherwise outputs too much
esp_log_level_set("wifi", ESP_LOG_ERROR);
// Initialization: WiFi + clean screen while downloading image
wifi_init_sta();
#if VALIDATE_SSL_CERTIFICATE == true
obtain_time();
#endif
epd_poweron();
epd_fullclear(&hl, 25);
/* printf("EPD w: %d h: %d\n\n", epd_width(), epd_height());
for (uint32_t x = 0; x < epd_width(); x++) {
epd_draw_pixel(x, 0, 0, fb);
epd_draw_pixel(x, epd_height()-10, 80, fb);
epd_draw_pixel(x, epd_height()-3, 60, fb);
// This 2 lines are not written
epd_draw_pixel(x, epd_height()-2, 0, fb);
epd_draw_pixel(x, epd_height()-1, 0, fb);
}
epd_hl_update_screen(&hl, MODE_GC16, 25); */
http_post();
}