168 lines
3.7 KiB
C
168 lines
3.7 KiB
C
/**
|
||
* @file img_utils.c
|
||
* @author TsMax (QinYUN575@Foxmail.com)
|
||
* @brief
|
||
* @version 0.1
|
||
* @date 2024-01-02
|
||
*
|
||
* @copyright Copyright (c) 2021 - 2024 co., ltd.
|
||
*
|
||
* SPDX-License-Identifier: Apache-2.0
|
||
*/
|
||
#include "img_utils.h"
|
||
#include <string.h>
|
||
#ifdef __ZEPHYR__
|
||
#include <zephyr/kernel.h>
|
||
#include "lvgl.h"
|
||
#elif CONFIG_LVGL_ENV_SIMULATOR
|
||
#include "lvgl/lvgl.h"
|
||
#else
|
||
#include "lvgl.h"
|
||
#endif
|
||
|
||
#define LOG_INF LV_LOG_USER
|
||
// #include <stdio.h>
|
||
// #define LOG_INF printf
|
||
|
||
#pragma pack(push, 1)
|
||
typedef struct
|
||
{
|
||
uint32_t width;
|
||
uint32_t height;
|
||
uint8_t bit_depth;
|
||
uint8_t color_type;
|
||
uint8_t compression;
|
||
uint8_t filter;
|
||
uint8_t interlace;
|
||
} IHDRChunk;
|
||
#pragma pack(pop)
|
||
|
||
#define SWAP_UINT32(x) (((x) >> 24) | \
|
||
(((x) & 0x00FF0000) >> 8) | \
|
||
(((x) & 0x0000FF00) << 8) | \
|
||
((x) << 24))
|
||
|
||
/**
|
||
* @brief 获取图片信息
|
||
*
|
||
* @param src 图片数据
|
||
* @param size 图片数据大小
|
||
* @param header 图片信息
|
||
* @return uint8_t
|
||
*/
|
||
int img_png_get_info(const uint8_t *src, int size, img_header_t *header)
|
||
{
|
||
if (size < 8)
|
||
{
|
||
return -1;
|
||
}
|
||
|
||
const uint8_t png_signature[8] = {0x89, 'P', 'N', 'G', 0x0D, 0x0A, 0x1A, 0x0A};
|
||
if (memcmp(src, png_signature, 8) != 0)
|
||
{
|
||
// LOG_INF("img_png_get_info: signature not match");
|
||
return -2; // 签名不匹配
|
||
}
|
||
// LOG_INF("img_png_get_info: signature match");
|
||
src += 8;
|
||
size -= 8;
|
||
|
||
// 检查 IHDR 块
|
||
if (size < 25)
|
||
{ // IHDR块的最小长度：4(长度) + 4(类型) + 13(IHDR数据) + 4(CRC)
|
||
// LOG_INF("img_png_get_info: size too short");
|
||
return -1; // 剩余数据太短，不足以包含完整的 IHDR 块
|
||
}
|
||
|
||
// 读取块长度
|
||
uint32_t chunk_length = (src[0] << 24) | (src[1] << 16) | (src[2] << 8) | src[3];
|
||
if (chunk_length != 13)
|
||
{
|
||
// LOG_INF("img_png_get_info: chunk length not match");
|
||
return -1; // IHDR块长度不正确
|
||
}
|
||
|
||
// 确认块类型是 "IHDR"
|
||
if (src[4] != 'I' || src[5] != 'H' || src[6] != 'D' || src[7] != 'R')
|
||
{
|
||
// LOG_INF("img_png_get_info: chunk type not match");
|
||
return -2; // 块类型不是 IHDR
|
||
}
|
||
// LOG_INF("img_png_get_info: chunk type match");
|
||
src += 8;
|
||
IHDRChunk *ihdr;
|
||
ihdr = (IHDRChunk *)src;
|
||
// 大小端转换
|
||
header->width = SWAP_UINT32(ihdr->width);
|
||
header->height = SWAP_UINT32(ihdr->height);
|
||
// LOG_INF("img_png_get_info: w=%d, h=%d", ihdr->width, ihdr->height);
|
||
// LOG_INF("img_png_get_info: w=%d, h=%d", header->width, header->height);
|
||
return 0;
|
||
}
|
||
|
||
int img_gif_get_info(const uint8_t * src, int size, img_header_t * header)
|
||
{
|
||
LOG_INF("img_gif_get_info: src=%p, size=%d", src, size);
|
||
if (size < 6)
|
||
{
|
||
LOG_INF("img_gif_get_info: size too short");
|
||
return -1;
|
||
}
|
||
|
||
if (src[0] != 'G' || src[1] != 'I' || src[2] != 'F')
|
||
{
|
||
LOG_INF("img_gif_get_info: signature not match");
|
||
return -2;
|
||
}
|
||
|
||
header->width = src[6] | (src[7] << 8);
|
||
header->height = src[8] | (src[9] << 8);
|
||
LOG_INF("img_gif_get_info: w=%d, h=%d", header->width, header->height);
|
||
return 0;
|
||
}
|
||
|
||
int img_jpg_get_info(const uint8_t * src, int size, img_header_t * header)
|
||
{
|
||
LOG_INF("img_jpg_get_info: src=%p, size=%d\r\n", src, size);
|
||
if (size < 2)
|
||
{
|
||
LOG_INF("img_jpg_get_info: size too short\r\n");
|
||
return -1;
|
||
}
|
||
|
||
if (src[0] != 0xFF || src[1] != 0xD8)
|
||
{
|
||
LOG_INF("img_jpg_get_info: signature not match\r\n");
|
||
return -2;
|
||
}
|
||
|
||
int i = 2;
|
||
while (i < size)
|
||
{
|
||
if (src[i] != 0xFF)
|
||
{
|
||
i++;
|
||
continue;
|
||
}
|
||
|
||
if (src[i + 1] == 0xC0)
|
||
{
|
||
break;
|
||
}
|
||
|
||
i += 2;
|
||
int length = (src[i] << 8) | src[i + 1];
|
||
i += length;
|
||
}
|
||
|
||
if (i >= size)
|
||
{
|
||
LOG_INF("img_jpg_get_info: not found SOF0\r\n");
|
||
return -3;
|
||
}
|
||
|
||
header->height = (src[i + 5] << 8) | src[i + 6];
|
||
header->width = (src[i + 7] << 8) | src[i + 8];
|
||
LOG_INF("img_jpg_get_info: w=%d, h=%d\r\n", header->width, header->height);
|
||
return 0;
|
||
} |