mbtk/mbtk_utils_linux/logo_partition_generate.c

#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;
}