0545e9dc6d
commit d1edce71135cc6d98c0a4b5729774542b676e769 Author: sophgo-forum-service <forum_service@sophgo.com> Date: Fri Mar 15 16:07:33 2024 +0800 [fix] recommend using ssh method to clone repo. [fix] fix sensor driver repo branch name.
141 lines
3.7 KiB
C++
141 lines
3.7 KiB
C++
#ifndef __OSD_CMPR_H__
|
|
#define __OSD_CMPR_H__
|
|
#include <cstdlib>
|
|
#include <stdbool.h>
|
|
#include <stdint.h>
|
|
#include <vector>
|
|
using namespace std;
|
|
|
|
// DataType-free color field copy
|
|
#define CPY_C(in, out) \
|
|
{ \
|
|
out.r = in.r; \
|
|
out.g = in.g; \
|
|
out.b = in.b; \
|
|
out.a = in.a; \
|
|
}
|
|
|
|
#define HDR_SZ (8)
|
|
|
|
enum OSD_FORMAT {
|
|
OSD_ARGB8888 = 0,
|
|
OSD_ARGB4444 = 4,
|
|
OSD_ARGB1555 = 5,
|
|
OSD_LUT8 = 8,
|
|
OSD_LUT4 = 10,
|
|
NUM_OF_FORMAT
|
|
};
|
|
|
|
typedef struct {
|
|
int img_width;
|
|
int img_height;
|
|
bool palette_mode_en;
|
|
bool zeroize_by_alpha;
|
|
int rgb_trunc_bit;
|
|
int alpha_trunc_bit;
|
|
int run_len_bd;
|
|
int palette_idx_bd;
|
|
OSD_FORMAT osd_format;
|
|
bool hdr_en;
|
|
} OSDCmpr_Cfg;
|
|
|
|
typedef struct {
|
|
uint8_t *stream; // stream buffer pointer
|
|
int bit_pos; // current pointer (in bit)
|
|
int buf_size; // in byte
|
|
int status;
|
|
} StreamBuffer;
|
|
|
|
void init_stream(StreamBuffer *bs, const uint8_t *buf, int buf_size,
|
|
bool read_only);
|
|
void write_stream(StreamBuffer *bs, uint8_t *src, int bit_len);
|
|
void parse_stream(StreamBuffer *bs, uint8_t *dest, int bit_len, bool read_only);
|
|
void move_stream_ptr(StreamBuffer *bs, int bit_len);
|
|
|
|
uint8_t get_bit_val(uint8_t *buf, int byte_idx, int bit_idx);
|
|
|
|
typedef union {
|
|
struct {
|
|
uint8_t g;
|
|
uint8_t b;
|
|
uint8_t r;
|
|
uint8_t a;
|
|
};
|
|
uint32_t code;
|
|
} RGBA;
|
|
|
|
typedef union {
|
|
struct {
|
|
uint16_t g : 4;
|
|
uint16_t b : 4;
|
|
uint16_t r : 4;
|
|
uint16_t a : 4;
|
|
};
|
|
uint16_t code;
|
|
} ARGB4444;
|
|
|
|
typedef union {
|
|
struct {
|
|
uint16_t g : 5;
|
|
uint16_t b : 5;
|
|
uint16_t r : 5;
|
|
uint16_t a : 1;
|
|
};
|
|
uint16_t code;
|
|
} ARGB1555;
|
|
|
|
union CODE {
|
|
RGBA color;
|
|
int palette_idx;
|
|
};
|
|
|
|
enum MODE_TYPE { Literal = 0, Palette, Literal_RL, Palette_RL, NUM_OF_MODE };
|
|
|
|
typedef struct {
|
|
int reg_image_width;
|
|
int reg_image_height;
|
|
bool reg_zeroize_by_alpha;
|
|
int reg_rgb_trunc_bit;
|
|
int reg_alpha_trunc_bit;
|
|
bool reg_palette_mode_en;
|
|
int reg_run_len_bd;
|
|
int reg_palette_idx_bd;
|
|
OSD_FORMAT reg_osd_format;
|
|
int pel_sz;
|
|
vector<RGBA> palette_cache;
|
|
int bs_buf_size;
|
|
uint8_t *bsbuf; // intermediate bitstream buffer
|
|
} OSDCmpr_Ctrl;
|
|
|
|
RGBA get_color(uint8_t *ptr, OSD_FORMAT format = OSD_ARGB8888);
|
|
void set_color(uint8_t *ptr, RGBA color, OSD_FORMAT format = OSD_ARGB8888);
|
|
|
|
void osd_cmpr_frame_init(OSDCmpr_Ctrl *p_ctrl);
|
|
int osd_cmpr_enc_one_frame(uint8_t *ibuf, uint8_t *obs, OSDCmpr_Ctrl *p_ctrl);
|
|
void osd_cmpr_dec_one_frame(uint8_t *bsbuf, size_t bs_size, uint8_t *obuf,
|
|
OSDCmpr_Ctrl *p_ctrl);
|
|
void osd_cmpr_enc_header(uint8_t *hdrbuf, OSDCmpr_Ctrl *p_ctrl);
|
|
void osd_cmpr_dec_header(uint8_t *hdrbuf, OSDCmpr_Ctrl *p_ctrl);
|
|
|
|
void osd_cmpr_setup(OSDCmpr_Ctrl *p_ctrl, OSDCmpr_Cfg *p_cfg);
|
|
void osd_cmpr_enc_const_pixel(RGBA cur_c, RGBA &last_c, int &rl_cnt,
|
|
MODE_TYPE &md, CODE &code, int &length,
|
|
bool is_force_new_run, int max_run_len,
|
|
OSDCmpr_Ctrl *p_ctrl, StreamBuffer *bitstream);
|
|
void osd_cmpr_enc_followed_run(RGBA cur_c, int &rl_cnt, MODE_TYPE &md,
|
|
CODE &code, int &length, int max_run_len,
|
|
OSDCmpr_Ctrl *p_ctrl, StreamBuffer *bitstream);
|
|
|
|
size_t osd_cmpr_get_pixel_sz(OSD_FORMAT format);
|
|
size_t osd_cmpr_get_bs_buf_max_sz(int pel_num, int pel_sz);
|
|
size_t osd_cmpr_get_header_sz(void);
|
|
|
|
void osd_cmpr_debug_frame_compare(OSDCmpr_Ctrl *p_ctrl, uint8_t *buf0,
|
|
uint8_t *buf1);
|
|
|
|
void palette_cache_init(vector<RGBA> &cache);
|
|
int palette_cache_lookup_color(vector<RGBA> &cache, RGBA color);
|
|
void palette_cache_lru_update(vector<RGBA> &cache, int index);
|
|
void palette_cache_push_color(vector<RGBA> &cache, RGBA color);
|
|
#endif /* __OSD_CMPR_H__ */
|