--- encoder.h 2002/09/04 21:03:29 1.20 +++ encoder.h 2003/03/13 11:07:20 1.27.2.2 @@ -3,8 +3,6 @@ * XVID MPEG-4 VIDEO CODEC * - Encoder header - * - * Copyright(C) 2002 Michael Militzer - * * This program is an implementation of a part of one or more MPEG-4 * Video tools as specified in ISO/IEC 14496-2 standard. Those intending * to use this software module in hardware or software products are @@ -29,6 +27,18 @@ * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * ****************************************************************************/ +/***************************************************************************** + * + * History + * + * - 13.06.2002 Added legal header + * - 22.08.2001 Added support for EXT_MODE encoding mode + * support for EXTENDED API + * - 22.08.2001 fixed bug in iDQtab + * + * $Id: encoder.h,v 1.27.2.2 2003/03/13 11:07:20 suxen_drol Exp $ + * + ****************************************************************************/ #ifndef _ENCODER_H_ #define _ENCODER_H_ @@ -36,7 +46,6 @@ #include "xvid.h" #include "portab.h" #include "global.h" -#include "image/image.h" #include "utils/ratecontrol.h" /***************************************************************************** @@ -56,20 +65,13 @@ typedef int bool; -typedef enum -{ - I_VOP = 0, - P_VOP = 1, - B_VOP = 2 -} -VOP_TYPE; - /***************************************************************************** * Structures ****************************************************************************/ typedef struct { + /* --- constants --- */ uint32_t width; uint32_t height; @@ -82,62 +84,86 @@ uint32_t fincr; uint32_t fbase; + xvid_global_t global_flags; + int bquant_ratio; + int bquant_offset; + int frame_drop_ratio; + #ifdef _SMP int num_threads; #endif + + int iMaxKeyInterval; + int max_bframes; + +/* --- inbetween vop stuff --- */ /* rounding type; alternate 0-1 after each interframe */ /* 1 <= fixed_code <= 4 automatically adjusted using motion vector statistics inside */ /* vars that not "quite" frame independant */ - uint32_t m_quant_type; uint32_t m_rounding_type; uint32_t m_fcode; + xvid_vol_t vol_flags; - HINTINFO *hint; + int64_t m_stamp; +} +MBParam; - uint32_t m_seconds; - uint32_t m_ticks; +typedef struct +{ + int iTextBits; + int iMvSum; + int iMvCount; + int kblks; + int mblks; + int ublks; + int gblks; } -MBParam; +Statistics; +/* encoding queue */ typedef struct { - uint32_t quant; - uint32_t motion_flags; - uint32_t global_flags; + xvid_enc_frame_t frame; + unsigned char quant_intra_matrix[64]; + unsigned char quant_inter_matrix[64]; + IMAGE image; +} QUEUEINFO; + + +typedef struct +{ + xvid_vol_t vol_flags; + xvid_vop_t vop_flags; + xvid_motion_t motion_flags; - VOP_TYPE coding_type; + int coding_type; + uint32_t quant; uint32_t rounding_type; uint32_t fcode; uint32_t bcode; uint32_t seconds; uint32_t ticks; + int64_t stamp; IMAGE image; MACROBLOCK *mbs; -} -FRAMEINFO; + WARPPOINTS warp; // as in bitstream + GMC_DATA gmc_data; // common data for all MBs -typedef struct -{ - int iTextBits; - float fMvPrevSigma; - int iMvSum; - int iMvCount; - int kblks; - int mblks; - int ublks; + int length; /* the encoded size of this frame */ + + Statistics sStat; } -Statistics; - +FRAMEINFO; typedef struct @@ -145,25 +171,57 @@ MBParam mbParam; int iFrameNum; - int iMaxKeyInterval; int bitrate; + // plugins + + int plugin_flags; + int num_plugins; + xvid_enc_plugin_t * plugins; + + // dquant + + int * temp_dquants; + // images FRAMEINFO *current; FRAMEINFO *reference; -#ifdef _DEBUG_PSNR IMAGE sOriginal; -#endif IMAGE vInterH; IMAGE vInterV; IMAGE vInterVf; IMAGE vInterHV; IMAGE vInterHVf; - Statistics sStat; + IMAGE vGMC; + + /* image queue */ + int queue_head; + int queue_tail; + int queue_size; + QUEUEINFO *queue; + + /* bframe buffer */ + int bframenum_head; + int bframenum_tail; + int flush_bframes; + + FRAMEINFO **bframes; + IMAGE f_refh; + IMAGE f_refv; + IMAGE f_refhv; + + /* closed_gop fixup temporary storage */ + int closed_bframenum; /* == -1 if there is no fixup intended */ + QUEUEINFO closed_qframe; /* qFrame, only valid when >= 0 */ + + int m_framenum; /* debug frame num counter; unlike iFrameNum, does not reset at ivop */ + RateControl rate_control; + + float fMvPrevSigma; } Encoder; @@ -206,14 +264,10 @@ void init_encoder(uint32_t cpu_flags); -int encoder_create(XVID_ENC_PARAM * pParam); -int encoder_destroy(Encoder * pEnc); -int encoder_encode(Encoder * pEnc, - XVID_ENC_FRAME * pFrame, - XVID_ENC_STATS * pResult); - -int encoder_encode_bframes(Encoder * pEnc, - XVID_ENC_FRAME * pFrame, - XVID_ENC_STATS * pResult); +int enc_create(xvid_enc_create_t * create, xvid_enc_rc_t * rc); +int enc_destroy(Encoder * pEnc); +int enc_encode(Encoder * pEnc, + xvid_enc_frame_t * pFrame, + xvid_enc_stats_t * stats); #endif