#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