Прошло 2 года.
This commit is contained in:
Кобелев Андрей Андреевич
283
lib/epdiy/test/test_lut.c
Normal file
283
lib/epdiy/test/test_lut.c
Normal file
@@ -0,0 +1,283 @@
|
||||
#include <esp_heap_caps.h>
|
||||
#include <stdbool.h>
|
||||
#include <stdint.h>
|
||||
#include <stdio.h>
|
||||
#include <string.h>
|
||||
#include <sys/types.h>
|
||||
#include <unity.h>
|
||||
#include "epd_internals.h"
|
||||
#include "epdiy.h"
|
||||
#include "esp_attr.h"
|
||||
#include "esp_timer.h"
|
||||
|
||||
#include "output_common/lut.h"
|
||||
#include "output_common/render_method.h"
|
||||
|
||||
#define DEFAULT_EXAMPLE_LEN 1408
|
||||
|
||||
static const uint8_t input_data_pattern[16] = { 0xFF, 0xFF, 0xF0, 0xFF, 0xFF, 0x00, 0x01, 0x10,
|
||||
0xA5, 0xA5, 0x5A, 0x5A, 0xFF, 0xFF, 0x00, 0x08 };
|
||||
|
||||
static const uint8_t result_pattern_1ppB[4] = { 0x20, 0x90, 0x5A, 0x40 };
|
||||
static const uint8_t result_pattern_2ppB_white[8]
|
||||
= { 0x00, 0x01, 0x50, 0x55, 0x55, 0x55, 0x00, 0x55 };
|
||||
static const uint8_t result_pattern_2ppB_black[8]
|
||||
= { 0xAA, 0xA8, 0x0A, 0x82, 0xAA, 0xAA, 0xAA, 0x20 };
|
||||
static const uint8_t result_pattern_8ppB_on_white[32]
|
||||
= { 0x00, 0x00, 0x00, 0x00, 0x55, 0x00, 0x00, 0x00, 0x00, 0x00, 0x55,
|
||||
0x55, 0x54, 0x55, 0x55, 0x54, 0x44, 0x11, 0x44, 0x11, 0x11, 0x44,
|
||||
0x11, 0x44, 0x00, 0x00, 0x00, 0x00, 0x55, 0x55, 0x15, 0x55 };
|
||||
static const uint8_t result_pattern_8ppB_on_black[32]
|
||||
= { 0xAA, 0xAA, 0xAA, 0xAA, 0x00, 0xAA, 0xAA, 0xAA, 0xAA, 0xAA, 0x00,
|
||||
0x00, 0x02, 0x00, 0x00, 0x02, 0x22, 0x88, 0x22, 0x88, 0x88, 0x22,
|
||||
0x88, 0x22, 0xAA, 0xAA, 0xAA, 0xAA, 0x00, 0x00, 0x80, 0x00 };
|
||||
|
||||
typedef void (*lut_func_t)(const uint32_t*, uint8_t*, const uint8_t*, uint32_t);
|
||||
static uint8_t waveform_phases[16][4];
|
||||
|
||||
void calc_epd_input_1ppB_1k_S3_VE(
|
||||
const uint32_t* ld, uint8_t* epd_input, const uint8_t* conversion_lut, uint32_t epd_width
|
||||
);
|
||||
|
||||
static EpdWaveformPhases test_waveform = {
|
||||
.phase_times = NULL,
|
||||
.phases = 1,
|
||||
.luts = (uint8_t*)waveform_phases,
|
||||
};
|
||||
|
||||
typedef struct {
|
||||
uint8_t* line_data;
|
||||
uint8_t* result_line;
|
||||
uint8_t* expected_line;
|
||||
uint8_t* lut;
|
||||
/// Ratio of input bytes to output bytes
|
||||
float in_out_ratio;
|
||||
int example_len_px;
|
||||
} LutTestBuffers;
|
||||
|
||||
static void fill_test_waveform() {
|
||||
for (int to = 0; to < 16; to++) {
|
||||
memset(waveform_phases[to], 0, 4);
|
||||
|
||||
for (int from = 0; from < 16; from++) {
|
||||
uint8_t val = 0x00;
|
||||
if (to < from)
|
||||
val = 0x01;
|
||||
if (to > from)
|
||||
val = 0x02;
|
||||
waveform_phases[to][from >> 2] |= val << (3 - (from & 0x3)) * 2;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* (Re-)fill buffers with example data, clear result buffers.
|
||||
*/
|
||||
static void lut_test_buffers_fill(LutTestBuffers* bufs, const uint8_t* result_pattern) {
|
||||
int result_pattern_len = sizeof(input_data_pattern) / bufs->in_out_ratio;
|
||||
|
||||
// initialize test and check patterns
|
||||
for (int i = 0; i < bufs->example_len_px; i++) {
|
||||
bufs->line_data[i] = input_data_pattern[i % sizeof(input_data_pattern)];
|
||||
}
|
||||
|
||||
for (int i = 0; i < bufs->example_len_px / bufs->in_out_ratio; i++) {
|
||||
bufs->expected_line[i] = result_pattern[i % result_pattern_len];
|
||||
}
|
||||
|
||||
memset(bufs->lut, 0, 1 << 16);
|
||||
memset(bufs->result_line, 0, bufs->example_len_px / bufs->in_out_ratio);
|
||||
|
||||
fill_test_waveform();
|
||||
heap_caps_check_integrity_all(true);
|
||||
}
|
||||
|
||||
/*
|
||||
* Allocates and populates buffers for LUT tests.
|
||||
*/
|
||||
static void lut_test_buffers_init(
|
||||
LutTestBuffers* bufs, int example_len_px, const uint8_t* result_pattern, float in_out_ratio
|
||||
) {
|
||||
uint32_t caps = MALLOC_CAP_INTERNAL | MALLOC_CAP_8BIT;
|
||||
bufs->line_data = heap_caps_aligned_alloc(16, example_len_px, caps);
|
||||
bufs->result_line = heap_caps_aligned_alloc(16, (int)(example_len_px / in_out_ratio), caps);
|
||||
bufs->expected_line = heap_caps_aligned_alloc(16, (int)(example_len_px / in_out_ratio), caps);
|
||||
bufs->lut = heap_caps_malloc(1 << 16, caps);
|
||||
bufs->example_len_px = example_len_px;
|
||||
bufs->in_out_ratio = in_out_ratio;
|
||||
|
||||
assert(bufs->line_data != NULL);
|
||||
assert(bufs->result_line != NULL);
|
||||
assert(bufs->expected_line != NULL);
|
||||
assert(bufs->lut != NULL);
|
||||
|
||||
lut_test_buffers_fill(bufs, result_pattern);
|
||||
}
|
||||
|
||||
/**
|
||||
* Free buffers used for LUT testing.
|
||||
*/
|
||||
static void diff_test_buffers_free(LutTestBuffers* bufs) {
|
||||
heap_caps_free(bufs->line_data);
|
||||
heap_caps_free(bufs->expected_line);
|
||||
heap_caps_free(bufs->result_line);
|
||||
heap_caps_free(bufs->lut);
|
||||
}
|
||||
|
||||
static void IRAM_ATTR test_with_alignments(LutTestBuffers* bufs, lut_func_t lut_func) {
|
||||
int len = bufs->example_len_px;
|
||||
int out_len = bufs->example_len_px / bufs->in_out_ratio;
|
||||
|
||||
uint8_t* expectation_backup = heap_caps_aligned_alloc(16, out_len, MALLOC_CAP_DEFAULT);
|
||||
memcpy(expectation_backup, bufs->expected_line, out_len);
|
||||
|
||||
// test combinations of start / end missalignment in four byte steps
|
||||
for (int start_offset = 0; start_offset <= 16; start_offset += 4) {
|
||||
for (int end_offset = 0; end_offset <= 16; end_offset += 4) {
|
||||
/// for 8ppB buffers, we skip 4 byte start offset since an input byte encodes 2 output
|
||||
/// bytes, there is no way to shift the output by just one byte.
|
||||
if (bufs->in_out_ratio < 1.0 && (start_offset % 8) == 4) {
|
||||
continue;
|
||||
}
|
||||
|
||||
int unaligned_len = len - end_offset - start_offset;
|
||||
|
||||
memset(bufs->result_line, 0, out_len);
|
||||
memcpy(bufs->expected_line, expectation_backup, out_len);
|
||||
|
||||
// before and after the designated range the buffer shoulld be clear
|
||||
memset(bufs->expected_line, 0, start_offset / 4);
|
||||
memset(bufs->expected_line + (start_offset + unaligned_len) / 4, 0, end_offset / 4);
|
||||
|
||||
printf(
|
||||
"benchmarking and checking with alignment (in px): (%d, %d)... ",
|
||||
start_offset,
|
||||
unaligned_len
|
||||
);
|
||||
uint64_t start = esp_timer_get_time();
|
||||
for (int i = 0; i < 100; i++) {
|
||||
lut_func(
|
||||
// We shift the alignment of the input data by four pixels
|
||||
(uint32_t*)(bufs->line_data + (int)(start_offset * bufs->in_out_ratio / 4)),
|
||||
// we shift the alignment of the result line by one byte
|
||||
bufs->result_line + start_offset / 4,
|
||||
bufs->lut,
|
||||
unaligned_len
|
||||
);
|
||||
}
|
||||
|
||||
heap_caps_check_integrity_all(true);
|
||||
uint64_t end = esp_timer_get_time();
|
||||
printf("took %.2fus per iter.\n", (end - start) / 100.0);
|
||||
|
||||
// Compare computed outputs to the expectation. We limit the comparison to len / 4,
|
||||
// since the LUT functions only compute a full display line, not more, even though our
|
||||
// test buffer may be larger.
|
||||
TEST_ASSERT_EQUAL_UINT8_ARRAY(bufs->expected_line, bufs->result_line, len / 4);
|
||||
}
|
||||
}
|
||||
heap_caps_check_integrity_all(true);
|
||||
|
||||
heap_caps_free(expectation_backup);
|
||||
}
|
||||
|
||||
TEST_CASE("1ppB lookup LCD, 64k LUT", "[epdiy,unit,lut]") {
|
||||
LutTestBuffers bufs;
|
||||
lut_test_buffers_init(&bufs, DEFAULT_EXAMPLE_LEN, result_pattern_1ppB, 4);
|
||||
|
||||
enum EpdDrawMode mode = MODE_GL16 | MODE_PACKING_1PPB_DIFFERENCE | MODE_FORCE_NO_PIE;
|
||||
LutFunctionPair func_pair = find_lut_functions(mode, 1 << 16);
|
||||
func_pair.build_func(bufs.lut, &test_waveform, 0);
|
||||
test_with_alignments(&bufs, func_pair.lookup_func);
|
||||
|
||||
diff_test_buffers_free(&bufs);
|
||||
}
|
||||
|
||||
#if !DISABLED_FOR_TARGETS(ESP32)
|
||||
TEST_CASE("1ppB lookup LCD, 1k LUT, PIE", "[epdiy,unit,lut]") {
|
||||
LutTestBuffers bufs;
|
||||
lut_test_buffers_init(&bufs, DEFAULT_EXAMPLE_LEN, result_pattern_1ppB, 4);
|
||||
|
||||
enum EpdDrawMode mode = MODE_GL16 | MODE_PACKING_1PPB_DIFFERENCE;
|
||||
LutFunctionPair func_pair = find_lut_functions(mode, 1 << 10);
|
||||
func_pair.build_func(bufs.lut, &test_waveform, 0);
|
||||
test_with_alignments(&bufs, calc_epd_input_1ppB_1k_S3_VE);
|
||||
diff_test_buffers_free(&bufs);
|
||||
}
|
||||
#endif
|
||||
|
||||
TEST_CASE("2ppB lookup LCD, 64k LUT, previously white", "[epdiy,unit,lut]") {
|
||||
LutTestBuffers bufs;
|
||||
lut_test_buffers_init(&bufs, DEFAULT_EXAMPLE_LEN, result_pattern_2ppB_white, 2);
|
||||
|
||||
enum EpdDrawMode mode = MODE_GL16 | MODE_PACKING_2PPB | PREVIOUSLY_WHITE;
|
||||
LutFunctionPair func_pair = find_lut_functions(mode, 1 << 16);
|
||||
func_pair.build_func(bufs.lut, &test_waveform, 0);
|
||||
test_with_alignments(&bufs, func_pair.lookup_func);
|
||||
|
||||
diff_test_buffers_free(&bufs);
|
||||
}
|
||||
|
||||
TEST_CASE("2ppB lookup LCD, 64k LUT, previously black", "[epdiy,unit,lut]") {
|
||||
LutTestBuffers bufs;
|
||||
lut_test_buffers_init(&bufs, DEFAULT_EXAMPLE_LEN, result_pattern_2ppB_black, 2);
|
||||
|
||||
enum EpdDrawMode mode = MODE_GL16 | MODE_PACKING_2PPB | PREVIOUSLY_BLACK;
|
||||
LutFunctionPair func_pair = find_lut_functions(mode, 1 << 16);
|
||||
func_pair.build_func(bufs.lut, &test_waveform, 0);
|
||||
test_with_alignments(&bufs, func_pair.lookup_func);
|
||||
|
||||
diff_test_buffers_free(&bufs);
|
||||
}
|
||||
|
||||
TEST_CASE("2ppB lookup LCD, 1k LUT, previously white", "[epdiy,unit,lut]") {
|
||||
LutTestBuffers bufs;
|
||||
lut_test_buffers_init(&bufs, DEFAULT_EXAMPLE_LEN, result_pattern_2ppB_white, 2);
|
||||
|
||||
enum EpdDrawMode mode = MODE_GL16 | MODE_PACKING_2PPB | PREVIOUSLY_WHITE;
|
||||
LutFunctionPair func_pair = find_lut_functions(mode, 1 << 10);
|
||||
func_pair.build_func(bufs.lut, &test_waveform, 0);
|
||||
test_with_alignments(&bufs, func_pair.lookup_func);
|
||||
|
||||
diff_test_buffers_free(&bufs);
|
||||
}
|
||||
|
||||
TEST_CASE("2ppB lookup LCD, 1k LUT, previously black", "[epdiy,unit,lut]") {
|
||||
LutTestBuffers bufs;
|
||||
lut_test_buffers_init(&bufs, DEFAULT_EXAMPLE_LEN, result_pattern_2ppB_black, 2);
|
||||
|
||||
enum EpdDrawMode mode = MODE_GL16 | MODE_PACKING_2PPB | PREVIOUSLY_BLACK;
|
||||
LutFunctionPair func_pair = find_lut_functions(mode, 1 << 10);
|
||||
func_pair.build_func(bufs.lut, &test_waveform, 0);
|
||||
test_with_alignments(&bufs, func_pair.lookup_func);
|
||||
|
||||
diff_test_buffers_free(&bufs);
|
||||
}
|
||||
|
||||
TEST_CASE("8ppB lookup LCD, 1k LUT, previously white", "[epdiy,unit,lut]") {
|
||||
LutTestBuffers bufs;
|
||||
lut_test_buffers_init(&bufs, DEFAULT_EXAMPLE_LEN / 2, result_pattern_8ppB_on_white, 0.5);
|
||||
|
||||
enum EpdDrawMode mode = MODE_DU | MODE_PACKING_8PPB | PREVIOUSLY_WHITE;
|
||||
LutFunctionPair func_pair = find_lut_functions(mode, 1 << 10);
|
||||
TEST_ASSERT_NOT_NULL(func_pair.build_func);
|
||||
TEST_ASSERT_NOT_NULL(func_pair.lookup_func);
|
||||
func_pair.build_func(bufs.lut, &test_waveform, 0);
|
||||
test_with_alignments(&bufs, func_pair.lookup_func);
|
||||
|
||||
diff_test_buffers_free(&bufs);
|
||||
}
|
||||
|
||||
TEST_CASE("8ppB lookup LCD, 1k LUT, previously black", "[epdiy,unit,lut]") {
|
||||
LutTestBuffers bufs;
|
||||
lut_test_buffers_init(&bufs, DEFAULT_EXAMPLE_LEN / 2, result_pattern_8ppB_on_black, 0.5);
|
||||
|
||||
enum EpdDrawMode mode = MODE_DU | MODE_PACKING_8PPB | PREVIOUSLY_BLACK;
|
||||
LutFunctionPair func_pair = find_lut_functions(mode, 1 << 10);
|
||||
TEST_ASSERT_NOT_NULL(func_pair.build_func);
|
||||
TEST_ASSERT_NOT_NULL(func_pair.lookup_func);
|
||||
func_pair.build_func(bufs.lut, &test_waveform, 0);
|
||||
test_with_alignments(&bufs, func_pair.lookup_func);
|
||||
|
||||
diff_test_buffers_free(&bufs);
|
||||
}
|
||||
Reference in New Issue
Block a user