1 /*
2 * Copyright (c) 2019 Jan Van Winkel <jan.van_winkel@dxplore.eu>
3 * Copyright (c) 2024 STMicroelectronics
4 *
5 * SPDX-License-Identifier: Apache-2.0
6 */
7
8 #include "img.h"
9
10 #include <zephyr/kernel.h>
11 #include <zephyr/device.h>
12 #include <zephyr/drivers/display.h>
13 #include <zephyr/fs/fs.h>
14 #include <zephyr/fs/littlefs.h>
15 #include <zephyr/ztest.h>
16 #include <lvgl_zephyr.h>
17
18 #include <lvgl.h>
19
20 #ifdef CONFIG_FS_LITTLEFS_BLK_DEV
21
22 #ifdef CONFIG_DISK_DRIVER_SDMMC
23 #define DISK_NAME "SD"
24 #elif defined(CONFIG_DISK_DRIVER_MMC)
25 #define DISK_NAME "SD2"
26 #else
27 #error "No disk device defined, is your board supported?"
28 #endif /* CONFIG_DISK_DRIVER_SDMMC */
29
30 #define IMG_FILE_PATH "/"DISK_NAME":/img.bin"
31
32 struct fs_littlefs lfsfs;
33
34 static struct fs_mount_t mnt = {
35 .type = FS_LITTLEFS,
36 .fs_data = &lfsfs,
37 .storage_dev = (void *)DISK_NAME,
38 .mnt_point = "/"DISK_NAME":",
39 .flags = FS_MOUNT_FLAG_USE_DISK_ACCESS,
40 };
41
42 #else /* CONFIG_FS_LITTLEFS_BLK_DEV */
43
44 #define IMG_FILE_PATH "/mnt/img.bin"
45
46 #define LVGL_PARTITION storage_partition
47 #define LVGL_PARTITION_ID FIXED_PARTITION_ID(LVGL_PARTITION)
48
49 FS_LITTLEFS_DECLARE_DEFAULT_CONFIG(cstorage);
50
51 static struct fs_mount_t mnt = {
52 .type = FS_LITTLEFS,
53 .fs_data = &cstorage,
54 .storage_dev = (void *)LVGL_PARTITION_ID,
55 .mnt_point = "/mnt"
56 };
57 #endif /* CONFIG_FS_LITTLEFS_BLK_DEV */
58
59 static const struct device *display_dev = DEVICE_DT_GET(DT_CHOSEN(zephyr_display));
60
ZTEST(lvgl_screen,test_get_default_screen)61 ZTEST(lvgl_screen, test_get_default_screen)
62 {
63 zassert_not_null(lv_scr_act(), "No default screen");
64 }
65
ZTEST(lvgl_screen,test_add_delete_screen)66 ZTEST(lvgl_screen, test_add_delete_screen)
67 {
68 lv_obj_t *default_screen = lv_scr_act();
69
70 zassert_not_null(default_screen, "No default screen");
71
72 lv_obj_t *new_screen = lv_obj_create(NULL);
73
74 zassert_not_null(new_screen, "Failed to create new screen");
75
76 lv_scr_load(new_screen);
77
78 lv_obj_t *act_screen = lv_scr_act();
79
80 zassert_equal_ptr(act_screen, new_screen, "New screen not active");
81
82 lv_scr_load(default_screen);
83
84 lv_obj_del(new_screen);
85
86 act_screen = lv_scr_act();
87 zassert_equal_ptr(act_screen, default_screen,
88 "Default screen not active");
89
90 }
ZTEST_USER(lvgl_fs,test_add_img)91 ZTEST_USER(lvgl_fs, test_add_img)
92 {
93 lv_obj_t *img = lv_img_create(lv_scr_act());
94
95 zassert_not_null(img, "Failed to create image");
96
97 lv_img_set_src(img, IMG_FILE_PATH);
98 lv_obj_align(img, LV_ALIGN_CENTER, 0, 0);
99 }
100
setup_lvgl(void)101 void *setup_lvgl(void)
102 {
103 int ret;
104
105 #if CONFIG_LV_COLOR_DEPTH_1 == 1
106 display_set_pixel_format(display_dev, PIXEL_FORMAT_MONO10);
107 #elif CONFIG_LV_COLOR_DEPTH_8 == 1 || CONFIG_LV_COLOR_DEPTH_24 == 1
108 /* No 8bit display pixel format not supported */
109 display_set_pixel_format(display_dev, PIXEL_FORMAT_RGB_888);
110 #elif CONFIG_LV_COLOR_DEPTH_16 == 1
111 display_set_pixel_format(display_dev, PIXEL_FORMAT_RGB_565);
112 #elif CONFIG_LV_COLOR_DEPTH_32 == 1
113 display_set_pixel_format(display_dev, PIXEL_FORMAT_ARGB_8888);
114 #else
115 #error "No display pixel format defined, is your board supported?"
116 #endif
117
118 ret = lvgl_init();
119 zassert_equal(ret, 0, "Failed to initialize lvgl");
120
121 return NULL;
122 }
123
setup_fs(void)124 void *setup_fs(void)
125 {
126 struct fs_file_t img;
127 struct fs_dirent info;
128 int ret;
129 const lv_image_dsc_t *c_img = get_lvgl_img();
130
131 setup_lvgl();
132
133 ret = fs_mount(&mnt);
134 if (ret < 0) {
135 TC_PRINT("Failed to mount file system: %d\n", ret);
136 ztest_test_fail();
137 return NULL;
138 }
139
140 ret = fs_stat(IMG_FILE_PATH, &info);
141 if ((ret == 0) && (info.type == FS_DIR_ENTRY_FILE)) {
142 return NULL;
143 }
144
145 fs_file_t_init(&img);
146 ret = fs_open(&img, IMG_FILE_PATH, FS_O_CREATE | FS_O_WRITE);
147 if (ret < 0) {
148 TC_PRINT("Failed to open image file: %d\n", ret);
149 ztest_test_fail();
150 return NULL;
151 }
152
153 ret = fs_write(&img, &c_img->header, sizeof(lv_image_header_t));
154 if (ret < 0) {
155 TC_PRINT("Failed to write image file header: %d\n", ret);
156 ztest_test_fail();
157 return NULL;
158 }
159
160 ret = fs_write(&img, c_img->data, c_img->data_size);
161 if (ret < 0) {
162 TC_PRINT("Failed to write image file data: %d\n", ret);
163 ztest_test_fail();
164 return NULL;
165 }
166
167 ret = fs_close(&img);
168 if (ret < 0) {
169 TC_PRINT("Failed to close image file: %d\n", ret);
170 ztest_test_fail();
171 return NULL;
172 }
173 return NULL;
174 }
175
teardown_lvgl(void * data)176 void teardown_lvgl(void *data)
177 {
178 lv_deinit();
179 }
180
181 ZTEST_SUITE(lvgl_screen, NULL, setup_lvgl, NULL, NULL, teardown_lvgl);
182 ZTEST_SUITE(lvgl_fs, NULL, setup_fs, NULL, NULL, teardown_lvgl);
183
