global.h

JM 11.0 KTA 2.1 Source Code

/*!
************************************************************************
*  \file
*     global.h
*
*  \brief
*     global definitions for for H.264 encoder.
*
*  \author
*     Copyright (C) 1999  Telenor Satellite Services,Norway
*                         Ericsson Radio Systems, Sweden
*
*     Inge Lille-Langoy               <[e-mail]>
*
*     Telenor Satellite Services
*     Keysers gt.13                       tel.:   [phone]
*     N-0130 Oslo,Norway                  fax.:   [phone]
*
*     Rickard Sjoberg                 <[e-mail]>
*
*     Ericsson Radio Systems
*     KI/ERA/T/VV
*     164 80 Stockholm, Sweden
*
************************************************************************
*/
#ifndef _GLOBAL_H_
#define _GLOBAL_H_

#include <stdio.h>

#include "defines.h"
#include "parsetcommon.h"
#include "q_matrix.h"
#include "q_offsets.h"
#include "minmax.h"

// Tool defines (AIF, 1/8pel, MVC) is moved to define.h by AT 2006/11/01

#ifdef MV_COMPETITION
#define MAX_MV_PREDICTOR  8
#endif

#ifdef WIN32
#define  snprintf _snprintf
#endif

#if defined(WIN32) && !defined(__GNUC__)
typedef __int64   int64;
# define FORMAT_OFF_T "I64d"
#ifndef INT64_MIN
# define INT64_MIN        (-9223372036854775807i64 - 1i64)
#endif
#else
typedef long long int64;
# define FORMAT_OFF_T "lld"
#ifndef INT64_MIN
# define INT64_MIN        (-9223372036854775807LL - 1LL)
#endif
#endif

#ifdef WIN32
#define  snprintf _snprintf
#define  open     _open
#define  close    _close
#define  read     _read
#define  write    _write
#define  lseek    _lseeki64
#define  fsync    _commit
#define  OPENFLAGS_WRITE _O_WRONLY|_O_CREAT|_O_BINARY|_O_TRUNC
#define  OPEN_PERMISSIONS _S_IREAD | _S_IWRITE
#define  OPENFLAGS_READ  _O_RDONLY|_O_BINARY
#else
#define  OPENFLAGS_WRITE O_WRONLY|O_CREAT|O_TRUNC
#define  OPENFLAGS_READ  O_RDONLY
#define  OPEN_PERMISSIONS S_IRUSR | S_IWUSR
#endif


/***********************************************************************
* T y p e    d e f i n i t i o n s    f o r    T M L
***********************************************************************
*/

//#define pel_t byte

#define imgpel unsigned short
#define distpel int
//#define imgpel byte
//#define distpel unsigned short

#define pel_t imgpel

//! Data Partitioning Modes
typedef enum
{
  PAR_DP_1,   //!< no data partitioning is supported
    PAR_DP_3    //!< data partitioning with 3 partitions
} PAR_DP_TYPE;


//! Output File Types
typedef enum
{
  PAR_OF_ANNEXB,    //!< Annex B byte stream format
    PAR_OF_RTP       //!< RTP packets in outfile
} PAR_OF_TYPE;

//! Field Coding Types
typedef enum 
{
  FRAME_CODING,
    FIELD_CODING,
    ADAPTIVE_CODING
} CodingType;

//! definition of H.264 syntax elements
typedef enum 
{
  SE_HEADER,
    SE_PTYPE,
    SE_MBTYPE,
    SE_REFFRAME,
    SE_INTRAPREDMODE,
    SE_MVD,
#ifdef MV_COMPETITION
    SE_MV_PREDICTOR,
#endif
    SE_CBP_INTRA,
    SE_LUM_DC_INTRA,
    SE_CHR_DC_INTRA,
    SE_LUM_AC_INTRA,
    SE_CHR_AC_INTRA,
    SE_CBP_INTER,
    SE_LUM_DC_INTER,
    SE_CHR_DC_INTER,
    SE_LUM_AC_INTER,
    SE_CHR_AC_INTER,
    SE_DELTA_QUANT_INTER,
    SE_DELTA_QUANT_INTRA,
    SE_BFRAME,
    SE_EOS,
    SE_MAX_ELEMENTS  //!< number of maximum syntax elements
} SE_type;         // substituting the definitions in elements.h


typedef enum 
{
  INTER_MB,
    INTRA_MB_4x4,
    INTRA_MB_16x16
} IntraInterDecision;


typedef enum 
{
  BITS_HEADER,
    BITS_TOTAL_MB,
    BITS_MB_MODE,
    BITS_INTER_MB,
    BITS_CBP_MB,
    BITS_COEFF_Y_MB,
    BITS_COEFF_UV_MB,
    BITS_DELTA_QUANT_MB,
#ifdef MV_COMPETITION
    BITS_SKIP_PRED,
    BITS_MV_PRED,
#endif
    MAX_BITCOUNTER_MB
} BitCountType;


typedef enum 
{
  NO_SLICES,
    FIXED_MB,
    FIXED_RATE,
    CALLBACK,
    FMO
} SliceMode;


typedef enum 
{
  UVLC,
    CABAC
} SymbolMode;


typedef enum 
{
  FRAME,
    TOP_FIELD,
    BOTTOM_FIELD
} PictureStructure;           //!< New enum for field processing

typedef enum 
{
  P_SLICE = 0,
    B_SLICE,
    I_SLICE,
    SP_SLICE,
    SI_SLICE
} SliceType;

/***********************************************************************
* D a t a    t y p e s   f o r  C A B A C
***********************************************************************
*/

//! struct to characterize the state of the arithmetic coding engine
typedef struct
{
  unsigned int  Elow, Erange;
  unsigned int  Ebuffer;
  unsigned int  Ebits_to_go;
  unsigned int  Ebits_to_follow;
  byte          *Ecodestrm;
  int           *Ecodestrm_len;
  int           C;
  int           E;
  
  // storage in case of recode MB
  unsigned int  ElowS, ErangeS;
  unsigned int  EbufferS;
  unsigned int  Ebits_to_goS;
  unsigned int  Ebits_to_followS;
  byte          *EcodestrmS;
  int           *Ecodestrm_lenS;
  int           CS;
  int           ES;
} EncodingEnvironment;

typedef EncodingEnvironment *EncodingEnvironmentPtr;

//! struct for context management
typedef struct
{
  unsigned short state;         // index into state-table CP  
  unsigned char  MPS;           // Least Probable Symbol 0/1 CP
  
  unsigned long  count;
  
} BiContextType;

typedef BiContextType *BiContextTypePtr;


/**********************************************************************
* C O N T E X T S   F O R   T M L   S Y N T A X   E L E M E N T S
**********************************************************************
*/


#define NUM_MB_TYPE_CTX  11
#define NUM_B8_TYPE_CTX  9
#define NUM_MV_RES_CTX   10
#define NUM_REF_NO_CTX   6
#define NUM_DELTA_QP_CTX 4
#define NUM_MB_AFF_CTX 4

#define NUM_TRANSFORM_SIZE_CTX 3
#ifdef ADAPTIVE_QUANTIZATION
#define NUM_MODULATED_QUANTIZATION_CTX 2
#endif

typedef struct
{
  BiContextType mb_type_contexts [3][NUM_MB_TYPE_CTX];
  BiContextType b8_type_contexts [2][NUM_B8_TYPE_CTX];
  BiContextType mv_res_contexts  [2][NUM_MV_RES_CTX];
#ifdef MV_COMPETITION
  BiContextType mv_predictor_skip_contexts  [MAX_MV_PREDICTOR];
  BiContextType mv_predictor_mvp_contexts   [MAX_MV_PREDICTOR];
  BiContextType mv_predictor_mvb_contexts   [MAX_MV_PREDICTOR];
#endif
  BiContextType ref_no_contexts  [2][NUM_REF_NO_CTX];
  BiContextType delta_qp_contexts   [NUM_DELTA_QP_CTX];
  BiContextType mb_aff_contexts     [NUM_MB_AFF_CTX];
  BiContextType transform_size_contexts   [NUM_TRANSFORM_SIZE_CTX];
#ifdef ADAPTIVE_QUANTIZATION
  BiContextType modulated_quantization_contexts [NUM_MODULATED_QUANTIZATION_CTX];
#endif
} MotionInfoContexts;


#define NUM_IPR_CTX    2
#define NUM_CIPR_CTX   4
#define NUM_CBP_CTX    4
#define NUM_BCBP_CTX   4
#define NUM_MAP_CTX   15
#define NUM_LAST_CTX  15
#define NUM_ONE_CTX    5
#define NUM_ABS_CTX    5

#ifdef ADAPTIVE_FD_SD_CODING
#define NUM_BCBP_FD_SD_CTX   4
#define NUM_MAP8_CTX        63
#define NUM_LAST8_CTX       63
#define NUM_MB_adap_CTX      4
#endif


typedef struct
{
  BiContextType  ipr_contexts [NUM_IPR_CTX]; 
  BiContextType  cipr_contexts[NUM_CIPR_CTX]; 
  BiContextType  cbp_contexts [3][NUM_CBP_CTX];
  BiContextType  bcbp_contexts[NUM_BLOCK_TYPES][NUM_BCBP_CTX];
  BiContextType  map_contexts [NUM_BLOCK_TYPES][NUM_MAP_CTX];
  BiContextType  last_contexts[NUM_BLOCK_TYPES][NUM_LAST_CTX];
  BiContextType  one_contexts [NUM_BLOCK_TYPES][NUM_ONE_CTX];
  BiContextType  abs_contexts [NUM_BLOCK_TYPES][NUM_ABS_CTX];
  BiContextType  fld_map_contexts [NUM_BLOCK_TYPES][NUM_MAP_CTX];
  BiContextType  fld_last_contexts[NUM_BLOCK_TYPES][NUM_LAST_CTX];
#ifdef ADAPTIVE_FD_SD_CODING
  BiContextType  MB_adaptive_SD_context   [NUM_MB_adap_CTX];
  BiContextType  bcbp_contexts_FD_SD      [NUM_BCBP_CTX];
  BiContextType  bcbp8_contexts_FD_SD     [NUM_BCBP_CTX];
  BiContextType  map_contexts_SD          [NUM_MAP_CTX];
  BiContextType  map8_contexts_SD         [NUM_MAP8_CTX];
  BiContextType  last_contexts_SD         [NUM_LAST_CTX];
  BiContextType  last8_contexts_SD        [NUM_LAST8_CTX];
  BiContextType  one_contexts_SD          [NUM_ONE_CTX];
  BiContextType  one8_contexts_SD         [NUM_ONE_CTX];
  BiContextType  abs_contexts_SD          [NUM_ABS_CTX];
  BiContextType  abs8_contexts_SD         [NUM_ABS_CTX];
#endif
} TextureInfoContexts;

//*********************** end of data type definition for CABAC *******************

typedef struct pix_pos
{
  int available;
  int mb_addr;
  int x;
  int y;
  int pos_x;
  int pos_y;
} PixelPos;

/*! Buffer structure for RMPNI commands */
typedef struct RMPNIbuffer_s
{
  int RMPNI;
  int Data;
  struct RMPNIbuffer_s *Next;
} RMPNIbuffer_t;

/*! Buffer structure for decoded referenc picture marking commands */
typedef struct DecRefPicMarking_s
{
  int memory_management_control_operation;
  int difference_of_pic_nums_minus1;
  int long_term_pic_num;
  int long_term_frame_idx;
  int max_long_term_frame_idx_plus1;
  struct DecRefPicMarking_s *Next;
} DecRefPicMarking_t;

//! Syntaxelement
typedef struct syntaxelement
{
  int                 type;           //!< type of syntax element for data part.
  int                 value1;         //!< numerical value of syntax element
  int                 value2;         //!< for blocked symbols, e.g. run/level
  int                 len;            //!< length of code