Add logo_partition_generate
Change-Id: Ib8b5adef63049228bbe66a01bb61722d13239fda
diff --git a/mbtk/mbtk_utils_linux/logo_partition_generate.c b/mbtk/mbtk_utils_linux/logo_partition_generate.c
new file mode 100755
index 0000000..3468558
--- /dev/null
+++ b/mbtk/mbtk_utils_linux/logo_partition_generate.c
@@ -0,0 +1,342 @@
+#include <stdio.h>
+#include <unistd.h>
+#include <stdlib.h>
+#include <errno.h>
+#include <string.h>
+#include <sys/types.h>
+#include <sys/stat.h>
+#include <fcntl.h>
+#include <stdint.h>
+
+#include "logo_partition.h"
+
+#pragma pack(push, 1) // 取消结构体对齐
+typedef struct {
+ uint16_t bfType; // 文件类型，应为"BM"（0x4D42）
+ uint32_t bfSize; // 文件总大小
+ uint16_t bfReserved1; // 保留字段
+ uint16_t bfReserved2;
+ uint32_t bfOffBits; // 像素数据偏移量
+} BITMAPFILEHEADER;
+
+typedef struct {
+ uint32_t biSize; // 信息头大小（通常为40）
+ int32_t biWidth; // 图像宽度（像素）
+ int32_t biHeight; // 图像高度（像素）
+ uint16_t biPlanes; // 颜色平面数（必须为1）
+ uint16_t biBitCount; // 每像素位数（如24表示RGB）
+ uint32_t biCompression; // 压缩类型（0表示无压缩）
+ uint32_t biSizeImage; // 像素数据大小（含填充）
+ int32_t biXPelsPerMeter;// 水平分辨率（像素/米）
+ int32_t biYPelsPerMeter;// 垂直分辨率
+ uint32_t biClrUsed; // 实际使用的颜色数
+ uint32_t biClrImportant; // 重要颜色数
+} BITMAPINFOHEADER;
+#pragma pack(pop)
+
+#define ANIM_PIC_MAX 30 // 开机动画的最大图片数量 (1.bmp - 30.bmp)
+
+static mbtk_logo_header_t logo_header = {
+ .addr = LOGO_ADDR_START,
+ .pic_width = 240,
+ .pic_height = 320,
+ .bg_rgb = 0x00000000 // black
+};
+
+static mbtk_animation_header_t animation_header = {
+ .addr = ANIMATION_ADDR_START,
+ .pic_num = 0,
+ .pic_width = 240,
+ .pic_height = 320,
+ .frame_rate = 10,
+ .bg_rgb = 0x00000000 // black
+};
+
+static int is_directory_exists(const char *path) {
+ struct stat stats;
+ if (stat(path, &stats) == 0 && S_ISDIR(stats.st_mode)) {
+ return 1; // 目录存在
+ }
+ printf("%s not exists.\n", path);
+ return 0; // 目录不存在或路径是文件
+}
+
+static void help()
+{
+ //printf("logo_partition_generate [pic_dir] -a [logo_w] -b [logo_h] -c [logo_bg] -d [anim_num] -e [anim_w] -f [anim_h] -g [anim_rate] -i [anim_bg]\n");
+ printf("logo_partition_generate [pic_dir] -a [logo_bg] -b [anim_rate] -c [anim_bg]\n");
+}
+
+static int bmp_read(const char* file_name, uint32 *width, uint32 *height, char *buff, int buff_size)
+{
+ FILE* file = fopen(file_name, "rb");
+ if (!file) return 0; // 文件不存在
+
+ // 读取文件头和信息头
+ BITMAPFILEHEADER file_header;
+ BITMAPINFOHEADER info_header;
+ fread(&file_header, sizeof(file_header), 1, file);
+ fread(&info_header, sizeof(info_header), 1, file);
+
+ // 验证是否为BMP文件
+ if (file_header.bfType != 0x4D42) {
+ fclose(file);
+ return -1;
+ }
+
+ // 仅支持24位无压缩格式
+ if (info_header.biBitCount != 24 || info_header.biCompression != 0) {
+ fclose(file);
+ return -1;
+ }
+
+ *width = info_header.biWidth;
+ *height = abs(info_header.biHeight); // 高度可能为负（表示从上到下存储）
+
+ // 计算每行实际字节数（含4字节对齐填充）
+ uint32_t row_size = ((info_header.biWidth * 3 + 3) / 4) * 4; // [[9, 18]]
+ uint32_t data_size = row_size * abs(info_header.biHeight); // [[9, 17]]
+ if(buff_size < data_size) {
+ printf("buff_size too small...");
+ return -1;
+ }
+
+ // 分配内存并读取像素数据
+ fseek(file, file_header.bfOffBits, SEEK_SET); // 定位到数据区 [[2, 8, 15]]
+ fread(buff, 1, data_size, file);
+ fclose(file);
+
+ return data_size;
+}
+
+static int write_header(int fd, mbtk_logo_header_t *logo_h, mbtk_animation_header_t *anim_h)
+{
+ if(-1 == lseek(fd, 0, SEEK_SET)) {
+ printf("lseek() fail:%d\n", errno);
+ return -1;
+ }
+
+ if(sizeof(mbtk_logo_header_t) != write(fd, logo_h, sizeof(mbtk_logo_header_t))) {
+ printf("Write logo header fail:%d\n", errno);
+ return -1;
+ }
+
+ if(sizeof(mbtk_animation_header_t) != write(fd, anim_h, sizeof(mbtk_animation_header_t))) {
+ printf("Write anim header fail:%d\n", errno);
+ return -1;
+ }
+
+ return 0;
+}
+
+static int write_item_logo(int fd, const char *dir, mbtk_logo_header_t *logo_h)
+{
+ if(-1 == lseek(fd, logo_h->addr, SEEK_SET)) {
+ printf("lseek() fail:%d\n", errno);
+ return -1;
+ }
+
+ char pic_name[256] = {0};
+ char buff[4 * 1024 * 1024] = {0}; // 4MB
+ int len = 0;
+ snprintf(pic_name, sizeof(pic_name), "%s/%s", dir, "logo.bmp");
+ if((len = bmp_read(pic_name, &(logo_h->pic_width), &(logo_h->pic_height), buff, sizeof(buff))) <= 0) {
+ printf("bmp_read() fail.\n");
+ return -1;
+ }
+
+ if(len != write(fd, buff, len)) {
+ printf("Write logo buffer fail:%d\n", errno);
+ return -1;
+ }
+
+ return 0;
+}
+
+static int write_item_anim(int fd, const char *dir, mbtk_animation_header_t *anim_h)
+{
+ if(-1 == lseek(fd, anim_h->addr, SEEK_SET)) {
+ printf("lseek() fail:%d\n", errno);
+ return -1;
+ }
+
+ char pic_name[256] = {0};
+ char buff[4 * 1024 * 1024] = {0}; // 4MB
+ int len, i = 1; // 1.bmp - ANIM_PIC_MAX.bmp
+
+ for(; i <= ANIM_PIC_MAX; i++) {
+ memset(pic_name, 0, sizeof(pic_name));
+ snprintf(pic_name, sizeof(pic_name), "%s/%d.bmp", dir, i);
+
+/*
+ if(!is_directory_exists(pic_name)) {
+ anim_h->pic_num = i - 1;
+ printf("Anim pic num is : %d\n", i - 1);
+ return 0;
+ }
+*/
+
+ len = bmp_read(pic_name, &(anim_h->pic_width), &(anim_h->pic_height), buff, sizeof(buff));
+ if(len == 0) { // 文件不存在
+ anim_h->pic_num = i - 1;
+ printf("Anim pic num is : %d\n", i - 1);
+ return 0;
+ } else if(len < 0) {
+ printf("bmp_read() fail.\n");
+ return -1;
+ }
+
+ if(len != write(fd, buff, len)) {
+ printf("Write logo buffer fail:%d\n", errno);
+ return -1;
+ }
+ }
+
+ anim_h->pic_num = ANIM_PIC_MAX;
+ return 0;
+}
+
+
+/*
+*
+* logo_partition_generate [pic_dir] -a [logo_w] -b [logo_h] -c [logo_bg] -d [anim_num] -e [anim_w] -f [anim_h] -g [anim_rate] -i [anim_bg]
+* logo_partition_generate [pic_dir] -a [logo_bg] -b [anim_rate] -c [anim_bg]
+*/
+int main(int argc, char *argv[])
+{
+ int ch;
+ char *out_bin = "logo.img";
+ uint32 temp_int;
+ if(argc < 2) {
+ help();
+ return -1;
+ }
+
+ char *dir = argv[1];
+ if(!is_directory_exists(dir)) {
+ printf("Dir %s not exists.\n", dir);
+ return -1;
+ }
+
+ while((ch = getopt(argc, argv, "a:b:c:"))!= -1){
+ switch(ch)
+ {
+#if 0
+ case 'a':
+ temp_int = (uint32)atoi(optarg);
+ if(temp_int > 0) {
+ logo_header.pic_width = temp_int;
+ } else {
+ printf("logo width error.");
+ }
+ break;
+ case 'b':
+ temp_int = (uint32)atoi(optarg);
+ if(temp_int > 0) {
+ logo_header.pic_height = temp_int;
+ } else {
+ printf("logo height error.");
+ }
+ break;
+ case 'c':
+ temp_int = (uint32)strtoul(optarg, NULL, 0); // 自动识别
+ logo_header.bg_rgb = temp_int & 0x00FFFFFF;
+ break;
+ case 'd':
+ temp_int = (uint32)atoi(optarg);
+ if(temp_int > 0) {
+ animation_header.pic_num = temp_int;
+ } else {
+ printf("animation number error.");
+ }
+ break;
+ case 'e':
+ temp_int = (uint32)atoi(optarg);
+ if(temp_int > 0) {
+ animation_header.pic_width = temp_int;
+ } else {
+ printf("animation width error.");
+ }
+ break;
+ case 'f':
+ temp_int = (uint32)atoi(optarg);
+ if(temp_int > 0) {
+ animation_header.pic_height = temp_int;
+ } else {
+ printf("animation height error.");
+ }
+ break;
+ case 'g':
+ temp_int = (uint32)atoi(optarg);
+ if(temp_int > 0) {
+ animation_header.frame_rate = temp_int;
+ } else {
+ printf("animation frame_rate error.");
+ }
+ break;
+ case 'i':
+ temp_int = (uint32)strtoul(optarg, NULL, 0); // 自动识别
+ animation_header.bg_rgb = temp_int & 0x00FFFFFF;
+ break;
+#else
+ case 'a':
+ temp_int = (uint32)strtoul(optarg, NULL, 0); // 自动识别
+ logo_header.bg_rgb = temp_int & 0x00FFFFFF;
+ break;
+ case 'b':
+ temp_int = (uint32)atoi(optarg);
+ if(temp_int > 0) {
+ animation_header.frame_rate = temp_int;
+ } else {
+ printf("animation frame_rate error.");
+ }
+ break;
+ case 'c':
+ temp_int = (uint32)strtoul(optarg, NULL, 0); // 自动识别
+ animation_header.bg_rgb = temp_int & 0x00FFFFFF;
+ break;
+#endif
+ default:
+ help();
+ return -1;
+ }
+ }
+
+ printf("Before Logo : addr - 0x%x, w - %d, h - %d, bg - 0x%06x\n", logo_header.addr ,logo_header.pic_width, logo_header.pic_height, logo_header.bg_rgb);
+ printf("Before Anim : addr - 0x%x, w - %d, h - %d, bg - 0x%06x, num - %d, frame_rate - %d\n", animation_header.addr ,animation_header.pic_width, animation_header.pic_height, animation_header.bg_rgb,
+ animation_header.pic_num, animation_header.frame_rate);
+
+ int fd = open(out_bin, O_WRONLY | O_TRUNC | O_CREAT, 0644);
+ if(fd < 0) {
+ printf("Open(%s) fail:%d\n", out_bin, errno);
+ return -1;
+ }
+
+ if(write_item_logo(fd, dir, &logo_header)) {
+ printf("write_item_logo() fail.");
+ goto fail;
+ }
+
+ if(write_item_anim(fd, dir, &animation_header)) {
+ printf("write_item_anim() fail.");
+ goto fail;
+ }
+
+ // Write header in the last.
+ if(write_header(fd, &logo_header, &animation_header)) {
+ printf("write_header() fail.");
+ goto fail;
+ }
+
+ printf("After Logo : addr - 0x%x, w - %d, h - %d, bg - 0x%06x\n", logo_header.addr ,logo_header.pic_width, logo_header.pic_height, logo_header.bg_rgb);
+ printf("After Anim : addr - 0x%x, w - %d, h - %d, bg - 0x%06x, num - %d, frame_rate - %d\n", animation_header.addr ,animation_header.pic_width, animation_header.pic_height, animation_header.bg_rgb,
+ animation_header.pic_num, animation_header.frame_rate);
+
+ printf("Success generate logo_partition bin:%s\n", out_bin);
+ close(fd);
+ return 0;
+fail:
+ close(fd);
+ return -1;
+}
+