//系統定義#include<stdio.h>#include<stdlib.h>#include<string.h>#include<limits.h>#include "config.h"#include"macros.h"// This should not be changed #define MB_SIZE 16#define MBC   88#define MBR   72// 參數定義 #define PREF_NULL_VEC 100#define PREF_16_VEC 200#define PREF_PBDELTA_NULL_VEC 50//全局變量 int headerlength; int pels;int cpels;int lines;int trace;int advanced;int syntax_arith_coding;int pb_frames;int mv_outside_frame;int long_vectors;float target_framerate;FILE *tf;//**************************#define PSC        1#define PSC_LENGTH        17#define ESCAPE                          7167#define PCT_INTER                       1#define PCT_INTRA                       0#define ON                              1#define OFF                             0#define SF_SQCIF                        1  // 001 #define SF_QCIF                         2  // 010 #define SF_CIF                          3  // 011 #define SF_4CIF                         4  // 100 #define SF_16CIF                        5  // 101 #define MODE_INTER                      0#define MODE_INTER_Q                    1#define MODE_INTER4V                    2#define MODE_INTRA                      3#define MODE_INTRA_Q                    4#define PBMODE_NORMAL                   0#define PBMODE_MVDB                     1#define PBMODE_CBPB_MVDB                2#define NO_VEC                          999// 運動向量的結構 typedef struct motionvector {  int x;        // 運動向量的水平分量   int y;        // 運動向量的豎直分量   int x_half;   // 運動向量的水平半像素精度   int y_half;   // 運動向量的豎直半像素精度   int min_error;// 運動向量的最小誤差   int Mode;     // 運動向量的高級預測模式 } MotionVector;// 像素點的結構定義 typedef struct point {  int x;  int y;} Point;// 圖像數據的結構定義 typedef struct pict_image {  unsigned char *lum;  // 亮度信息平面   unsigned char *Cr;   // Cr色度信息平面   unsigned char *Cb;   // Cb色度信息平面 } PictImage;// 關于視頻幀各層的結構定義 // 視頻幀結構定義 typedef struct pict {  int prev;					//上一幀  int curr;					//當前幀  int TR;					//時間參考   int bit_rate;				//碼率  int src_frame_rate;		//源視頻格式的幀率  float target_frame_rate;  //目標視頻格式的幀率  int source_format;		//源視頻格式  int picture_coding_type;	//視頻幀編碼類型  int spare;  int unrestricted_mv_mode;	//非限制性的運動向量模式  int PB;					//PB幀模式  int QUANT;				//量化  int DQUANT;  int MB;					//編碼宏塊  int seek_dist;			//運動向量的搜索窗口  int use_gobsync;			//GOB同步的標志  int MODB;					//B幀模式  int BQUANT;				//在PB幀模式下為B幀宏塊選定的量化器  int TRB;					//為B幀準備的時間參考  float QP_mean;			//均值量化器} Pict;// 圖像分塊的結構定義typedef struct slice {  unsigned int vert_pos;	//圖像分塊的水平位置  unsigned int quant_scale;	//量化范圍} Slice;// 宏塊結構定義typedef struct macroblock {  int mb_address;			//宏塊地址  int macroblock_type;      //宏塊類型  int skipped;				//宏塊跳過表示：1表示跳過  MotionVector motion;	    //運動向量} Macroblock;// 宏塊數據的結構typedef struct mb_structure {  int lum[16][16];  int Cr[8][8];  int Cb[8][8];} MB_Structure;// 數據流統計的數據結構typedef struct bits_counted {  int Y;  int C;  int vec;  int CBPY;  int CBPCM;  int MODB;  int CBPB;  int COD;  int header;  int DQUANT;  int total;  int no_inter;  int no_inter4v;  int no_intra;} Bits;// 編碼結果的結構typedef struct results {  float SNR_l;        //亮度分量的信噪比  float SNR_Cr;       //色度分量的信噪比  float SNR_Cb;  float QP_mean;      //均值量化器} Results;void Help();void AdvancedHelp();int NextTwoPB(PictImage *p2, PictImage *bim, PictImage *p1,               int bskip, int pskip, int seek_dist);void PrintSNR(Results *res, int num);void PrintResult(Bits *bits, int num_units, int num);unsigned char *ReadImage(char *filename, int frame_no, int headerlength);PictImage *FillImage(unsigned char *in);void WriteImage(PictImage *image, char *filename);PictImage *InitImage(int size);void FreeImage(PictImage *image);char *StripName(char *s);int *MB_Encode(MB_Structure *mb_orig, int QP, int I);int MB_Decode(int *qcoeff, MB_Structure *mb_recon, int QP, int I);int Dct( int *block, int *coeff);int idct(int *coeff,int *block);void FillLumBlock( int x, int y, PictImage *image, MB_Structure *data);void FillChromBlock(int x_curr, int y_curr, PictImage *image,            MB_Structure *data);void ReconImage (int i, int j, MB_Structure *data, PictImage *recon);void CodeOneOrTwo(PictImage *curr, PictImage *B_image, PictImage *prev,           PictImage *prev_recon, int QP, int frameskip, Bits *bits,          Pict *pic, PictImage *B_recon, PictImage *recon);PictImage *CodeOneIntra(PictImage *curr, int QP, Bits *bits, Pict *pic);void Dequant(int *qcoeff, int *rcoeff, int QP, int I);void Quant(int *coeff, int *qcoeff, int QP, int I);void CountBitsCoeff(int *qcoeff, int I, int CBP, Bits *bits, int ncoeffs);int CodeCoeff(int Mode, int *qcoeff, int block, int ncoeffs);int FindCBP(int *qcoeff, int Mode, int ncoeffs);void CountBitsVectors(MotionVector *MV[5][MBR+1][MBC+2], Bits *bits,               int i, int j, int Mode, int newgob, Pict *pic);void FindPMV(MotionVector *MV[5][MBR+1][MBC+2], int x, int y,              int *p0, int *p1, int block, int newgob, int half_pel);void ZeroBits(Bits *bits);void ZeroRes(Results *res);void ZeroVec(MotionVector *MV);void MarkVec(MotionVector *MV);void CopyVec(MotionVector *MV1, MotionVector *MV2);int EqualVec(MotionVector *MV2, MotionVector *MV1);void AddBits(Bits *total, Bits *bits);void AddRes(Results *total, Results *res, Pict *pic);void AddBitsPicture(Bits *bits);void FindMB(int x, int y, unsigned char *image, int MB[16][16]);MB_Structure *MB_Recon(PictImage *prev_recon, MB_Structure *diff,               int x_curr, int y_curr, MotionVector *MV);MB_Structure *Predict(PictImage *curr, PictImage *prev_recon,              int x_curr, int y_curr, MotionVector *MV);unsigned char *InterpolateImage(unsigned char *image, int w, int h);void MotionEstimatePicture(unsigned char *curr, unsigned char *prev,            unsigned char *prev_ipol, int seek_dist,            MotionVector *MV[5][MBR+1][MBC+2], int gobsync);void MotionEstimation(unsigned char *curr, unsigned char *prev, int x_curr,              int y_curr, int xoff, int yoff, int seek_dist,               MotionVector *MV[5][MBR+1][MBC+2], int *sad_0);unsigned char *LoadArea(unsigned char *im, int x, int y,         int x_size, int y_size, int lx);int SAD_Macroblock(unsigned char *ii, unsigned char *act_block,           int h_length, int Min_FRAME);int SAD_MB_Bidir(unsigned char *ii, unsigned char *aa, unsigned char *bb,          int width, int min_sofar);int SAD_Block(unsigned char *ii, unsigned char *act_block,              int h_length, int min_sofar);int SAD_MB_integer(int *ii, int *act_block, int h_length, int min_sofar);MB_Structure *Predict_P(PictImage *curr_image, PictImage *prev_image,        unsigned char *prev_ipol,int x_curr, int y_curr,         MotionVector *fr[5][MBR+1][MBC+2], int PB);MB_Structure *Predict_B(PictImage *curr_image, PictImage *prev_image,        unsigned char *prev_ipol,int x_curr, int y_curr,        MotionVector *fr[5][MBR+1][MBC+2],        MB_Structure *recon_P, int TR,int TRB);void Clip(MB_Structure *data);void FindForwLumPredPB(unsigned char *prev_ipol, int x_curr, int y_curr,                MotionVector *fr, int *pred, int TRD, int TRB,                int bdx, int bdy, int bs, int comp);void FindBiDirLumPredPB(int *recon_P, MotionVector *fr, int *pred, int TRD,         int TRB, int bdx, int bdy, int nh, int nv);void FindBiDirChrPredPB(MB_Structure *recon_P, int dx, int dy,         MB_Structure *pred);void FindBiDirLimits(int vec, int *start, int *stop, int nhv);void FindBiDirChromaLimits(int vec, int *start, int *stop);void BiDirPredBlock(int xstart, int xstop, int ystart, int ystop,            int xvec, int yvec, int *recon, int *pred, int bl);void DoPredChrom_P(int x_curr, int y_curr, int dx, int dy,           PictImage *curr, PictImage *prev,            MB_Structure *pred_error);void FindHalfPel(int x, int y, MotionVector *MV, unsigned char *prev,          int *curr, int bs, int comp);void FindPred(int x, int y, MotionVector *fr, unsigned char *prev,               int *pred, int bs, int comp); void FindPredOBMC(int x, int y, MotionVector *MV[5][MBR+1][MBC+2],           unsigned char *prev, int *pred, int comp, int PB);MB_Structure *MB_Recon_P(PictImage *prev_image, unsigned char *prev_ipol,         MB_Structure *diff, int x_curr, int y_curr,          MotionVector *MV[5][MBR+1][MBC+2], int PB);MB_Structure *MB_Recon_B(PictImage *prev, MB_Structure *diff,         unsigned char *prev_ipol,int x_curr, int y_curr,         MotionVector *MV[5][MBR+1][MBC+2],         MB_Structure *recon_P,int TR, int TRB);void ReconLumBlock_P(int x, int y, MotionVector *fr,             unsigned char *prev, int *data,int bs,int comp);void ReconChromBlock_P(int x_curr, int y_curr, int dx, int dy,               PictImage *prev, MB_Structure *data);void FindChromBlock_P(int x_curr, int y_curr, int dx, int dy,               PictImage *prev, MB_Structure *data);void ComputeSNR(PictImage *im1, PictImage *im2, Results *res, int write);     void ZeroMBlock(MB_Structure *data);int CountBitsPicture(Pict *pic);void CountBitsMB(int Mode, int COD, int CBP, int CBPB, Pict *pic, Bits *bits);int CountBitsSlice(int slice, int quant);int ChooseMode(unsigned char *curr, int x_pos, int y_pos, int min_SAD);int ModifyMode(int Mode, int dquant);int *InterleaveCoeff(int *qp, int *qb, int bs);void MakeEdgeImage(unsigned char *src, unsigned char *dst, int width,           int height, int edge);void BitPrint(int length, int val, char *bit);#ifdef OFFLINE_RATE_CONTROLint FrameUpdateQP(int buf, int bits, int frames_left, int QP, int B,           float seconds);#elseint UpdateQuantizer(int mb, float QP_mean, int pict_type, float bit_rate,                     int mb_width, int mb_height, int bitcount) ;int InitializeQuantizer(int pict_type, float bit_rate,                                            float target_frame_rate, float QP_mean);void InitializeRateControl();void UpdateRateControl(int bits);#endif#ifndef FASTIDCT// global declarations for idctref void init_idctref (void);void idctref (int *coeff, int *block);#endif// Syntax based arithmetic coding routines void Count_sac_BitsCoeff(int *qcoeff, int I, int CBP, Bits *bits, int ncoeffs);int Code_sac_Coeff(int Mode, int *qcoeff, int block, int ncoeffs);int CodeTCoef(int mod_index, int position, int intra);void Count_sac_BitsVectors(MotionVector *MV[5][MBR+1][MBC+2], Bits *bits,                      int i, int j, int Mode, int newgob, Pict *pic);void Count_sac_BitsMB(int Mode,int COD,int CBP,int CBPB,Pict *pic,Bits *bits);int AR_Encode(int index, int cumul_freq[]);int indexfn(int value, int table[], int max);int bit_opp_bits(int);int bit_in_psc_layer(int);int encoder_flush();// bitstream-functions void initbits();void putbits (int, int);int alignbits ();int bitcount();// Fix broken header-files on suns to avoid compiler warnings // #define BROKEN_SUN_HEADERS here or in Makefile #ifdef BROKEN_SUN_HEADERSextern int printf();extern int fprintf();extern int time();extern int fclose();extern int rewind();extern int fseek();extern int fread();extern int fwrite();extern int fflush();extern int fscanf();extern int _flsbuf();extern int _filbuf();#endif