mirror of
https://github.com/ultravideo/uvg266.git
synced 2024-11-24 10:34:05 +00:00
138 lines
4.2 KiB
C
138 lines
4.2 KiB
C
#ifndef PICTURE_H_
|
|
#define PICTURE_H_
|
|
/**
|
|
* \file
|
|
* \brief Coding Unit (CU) and picture data related functions.
|
|
*
|
|
* \author Marko Viitanen ( fador@iki.fi ),
|
|
* Tampere University of Technology,
|
|
* Department of Pervasive Computing.
|
|
* \author Ari Koivula ( ari@koivu.la ),
|
|
* Tampere University of Technology,
|
|
* Department of Pervasive Computing.
|
|
*/
|
|
|
|
#include "global.h"
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
// CONSTANTS
|
|
|
|
enum { CU_NOTSET = 0, CU_PCM, CU_SKIP, CU_SPLIT, CU_INTRA, CU_INTER };
|
|
enum { SLICE_B = 0, SLICE_P = 1, SLICE_I = 2 };
|
|
enum { REF_PIC_LIST_0 = 0, REF_PIC_LIST_1 = 1, REF_PIC_LIST_X = 100 };
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
// TYPES
|
|
|
|
/**
|
|
* \brief Struct for CU intra info
|
|
*/
|
|
typedef struct
|
|
{
|
|
int8_t mode;
|
|
uint32_t cost;
|
|
} CU_info_intra;
|
|
|
|
/**
|
|
* \brief Struct for CU inter info
|
|
*/
|
|
typedef struct
|
|
{
|
|
int8_t mode;
|
|
uint32_t cost;
|
|
int16_t mv[2];
|
|
uint8_t mv_ref; // \brief Index of the encoder_control.ref array.
|
|
uint8_t mv_dir; // \brief Probably describes if mv_ref is forward, backward or both. Might not be needed?
|
|
} CU_info_inter;
|
|
|
|
/**
|
|
* \brief Struct for CU info
|
|
*/
|
|
typedef struct
|
|
{
|
|
int8_t type;
|
|
int8_t depth;
|
|
int8_t coded;
|
|
CU_info_intra intra;
|
|
CU_info_inter inter;
|
|
int8_t split;
|
|
} CU_info;
|
|
|
|
/**
|
|
* \brief Struct which contains all picture data
|
|
*/
|
|
typedef struct
|
|
{
|
|
uint8_t* y_data; // \brief Pointer to luma pixel array.
|
|
uint8_t* u_data; // \brief Pointer to chroma U pixel array.
|
|
uint8_t* v_data; // \brief Pointer to chroma V pixel array.
|
|
|
|
uint8_t* y_recdata; // \brief Pointer to reconstructed Y-data.
|
|
uint8_t* u_recdata; // \brief Pointer to reconstructed U-data.
|
|
uint8_t* v_recdata; // \brief Pointer to reconstructed V-data.
|
|
|
|
int32_t width; // \brief Luma pixel array width.
|
|
int32_t height; // \brief Luma pixel array height.
|
|
int32_t height_in_lcu; // \brief Picture width in number of LCU's.
|
|
int32_t width_in_lcu; // \brief Picture height in number of LCU's.
|
|
uint8_t referenced; // \brief Whether this picture is referenced.
|
|
CU_info** CU; // \brief Info for each CU at each depth.
|
|
uint8_t type;
|
|
uint8_t slicetype;
|
|
} picture;
|
|
|
|
/**
|
|
* \brief Struct which contains array of picture structs
|
|
*/
|
|
typedef struct
|
|
{
|
|
picture** pics; // \brief Pointer to array of picture pointers.
|
|
unsigned int size; // \brief Array size.
|
|
unsigned int used_size;
|
|
} picture_list;
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
// FUNCTIONS
|
|
|
|
picture * picture_init(int32_t width, int32_t height,
|
|
int32_t width_in_lcu, int32_t height_in_lcu);
|
|
int picture_destroy(picture *pic);
|
|
void picture_set_block_coded(picture *pic, uint32_t x_scu, uint32_t y_scu,
|
|
uint8_t depth, int8_t coded);
|
|
void picture_set_block_split(picture *pic, uint32_t x_scu, uint32_t y_scu,
|
|
uint8_t depth, int8_t split);
|
|
|
|
picture_list * picture_list_init(int size);
|
|
int picture_list_resize(picture_list *list, int size);
|
|
int picture_list_destroy(picture_list *list);
|
|
int picture_list_add(picture_list *list, picture *pic);
|
|
int picture_list_rem(picture_list *list, int n, int8_t destroy);
|
|
|
|
uint32_t sad64x64(int16_t *block1, uint32_t stride1,
|
|
int16_t* block2, uint32_t stride2);
|
|
uint32_t sad32x32(int16_t *block1, uint32_t stride1,
|
|
int16_t* block2, uint32_t stride2);
|
|
uint32_t sad16x16(int16_t *block1, uint32_t stride1,
|
|
int16_t* block2, uint32_t stride2);
|
|
uint32_t sad8x8(int16_t *block1, uint32_t stride1,
|
|
int16_t* block2, uint32_t stride2);
|
|
uint32_t sad4x4(int16_t *block1, uint32_t stride1,
|
|
int16_t* block2, uint32_t stride2);
|
|
uint32_t sad(uint8_t *data1, uint8_t *data2,
|
|
unsigned width, unsigned height, unsigned stride);
|
|
|
|
double image_psnr(uint8_t *frame1, uint8_t *frame2, int32_t x, int32_t y);
|
|
|
|
|
|
//////////////////////////////////////////////////////////////////////////
|
|
// MACROS
|
|
|
|
#define GET_SPLITDATA(CU,curDepth) ((CU)->depth > curDepth)
|
|
#define SET_SPLITDATA(CU,flag) { (CU)->split=(flag); }
|
|
|
|
|
|
#endif
|