00001 
00002 
00003 
00004 
00005 
00006 
00007 
00008 
00009 
00010 
00011 
00012 
00013 
00014 
00015 
00016 
00017 
00018 
00019 
00020 
00021 
00022 
00028 #ifndef AVCODEC_MPEGVIDEO_H
00029 #define AVCODEC_MPEGVIDEO_H
00030 
00031 #include "avcodec.h"
00032 #include "dsputil.h"
00033 #include "get_bits.h"
00034 #include "put_bits.h"
00035 #include "ratecontrol.h"
00036 #include "parser.h"
00037 #include "mpeg12data.h"
00038 #include "rl.h"
00039 #include "libavutil/timecode.h"
00040 
00041 #include "libavutil/opt.h"
00042 
00043 #define FRAME_SKIPPED 100 
00044 
00045 enum OutputFormat {
00046     FMT_MPEG1,
00047     FMT_H261,
00048     FMT_H263,
00049     FMT_MJPEG,
00050     FMT_H264,
00051 };
00052 
00053 #define MPEG_BUF_SIZE (16 * 1024)
00054 
00055 #define QMAT_SHIFT_MMX 16
00056 #define QMAT_SHIFT 21
00057 
00058 #define MAX_FCODE 7
00059 #define MAX_MV 2048
00060 
00061 #define MAX_THREADS 32
00062 
00063 #define MAX_PICTURE_COUNT 32
00064 
00065 #define ME_MAP_SIZE 64
00066 #define ME_MAP_SHIFT 3
00067 #define ME_MAP_MV_BITS 11
00068 
00069 #define MAX_MB_BYTES (30*16*16*3/8 + 120)
00070 
00071 #define INPLACE_OFFSET 16
00072 
00073 
00074 #define SEQ_END_CODE            0x000001b7
00075 #define SEQ_START_CODE          0x000001b3
00076 #define GOP_START_CODE          0x000001b8
00077 #define PICTURE_START_CODE      0x00000100
00078 #define SLICE_MIN_START_CODE    0x00000101
00079 #define SLICE_MAX_START_CODE    0x000001af
00080 #define EXT_START_CODE          0x000001b5
00081 #define USER_START_CODE         0x000001b2
00082 
00083 struct MpegEncContext;
00084 
00088 typedef struct Picture{
00089     struct AVFrame f;
00090 
00094     uint8_t *interpolated[3];
00095     int8_t *qscale_table_base;
00096     int16_t (*motion_val_base[2])[2];
00097     uint32_t *mb_type_base;
00098 #define MB_TYPE_INTRA MB_TYPE_INTRA4x4 //default mb_type if there is just one type
00099 #define IS_INTRA4x4(a)   ((a)&MB_TYPE_INTRA4x4)
00100 #define IS_INTRA16x16(a) ((a)&MB_TYPE_INTRA16x16)
00101 #define IS_PCM(a)        ((a)&MB_TYPE_INTRA_PCM)
00102 #define IS_INTRA(a)      ((a)&7)
00103 #define IS_INTER(a)      ((a)&(MB_TYPE_16x16|MB_TYPE_16x8|MB_TYPE_8x16|MB_TYPE_8x8))
00104 #define IS_SKIP(a)       ((a)&MB_TYPE_SKIP)
00105 #define IS_INTRA_PCM(a)  ((a)&MB_TYPE_INTRA_PCM)
00106 #define IS_INTERLACED(a) ((a)&MB_TYPE_INTERLACED)
00107 #define IS_DIRECT(a)     ((a)&MB_TYPE_DIRECT2)
00108 #define IS_GMC(a)        ((a)&MB_TYPE_GMC)
00109 #define IS_16X16(a)      ((a)&MB_TYPE_16x16)
00110 #define IS_16X8(a)       ((a)&MB_TYPE_16x8)
00111 #define IS_8X16(a)       ((a)&MB_TYPE_8x16)
00112 #define IS_8X8(a)        ((a)&MB_TYPE_8x8)
00113 #define IS_SUB_8X8(a)    ((a)&MB_TYPE_16x16) //note reused
00114 #define IS_SUB_8X4(a)    ((a)&MB_TYPE_16x8)  //note reused
00115 #define IS_SUB_4X8(a)    ((a)&MB_TYPE_8x16)  //note reused
00116 #define IS_SUB_4X4(a)    ((a)&MB_TYPE_8x8)   //note reused
00117 #define IS_ACPRED(a)     ((a)&MB_TYPE_ACPRED)
00118 #define IS_QUANT(a)      ((a)&MB_TYPE_QUANT)
00119 #define IS_DIR(a, part, list) ((a) & (MB_TYPE_P0L0<<((part)+2*(list))))
00120 #define USES_LIST(a, list) ((a) & ((MB_TYPE_P0L0|MB_TYPE_P1L0)<<(2*(list)))) 
00121 #define HAS_CBP(a)        ((a)&MB_TYPE_CBP)
00122 
00123     int field_poc[2];           
00124     int poc;                    
00125     int frame_num;              
00126     int mmco_reset;             
00127     int pic_id;                 
00129     int long_ref;               
00130     int ref_poc[2][2][32];      
00131     int ref_count[2][2];        
00132     int mbaff;                  
00133     int field_picture;          
00134     int sync;                   
00135 
00136     int mb_var_sum;             
00137     int mc_mb_var_sum;          
00138     uint16_t *mb_var;           
00139     uint16_t *mc_mb_var;        
00140     uint8_t *mb_mean;           
00141     int32_t *mb_cmp_score;      
00142     int b_frame_score;          
00143     struct MpegEncContext *owner2; 
00144     int needs_realloc;          
00145 } Picture;
00146 
00150 typedef struct MotionEstContext{
00151     AVCodecContext *avctx;
00152     int skip;                          
00153     int co_located_mv[4][2];           
00154     int direct_basis_mv[4][2];
00155     uint8_t *scratchpad;               
00156     uint8_t *best_mb;
00157     uint8_t *temp_mb[2];
00158     uint8_t *temp;
00159     int best_bits;
00160     uint32_t *map;                     
00161     uint32_t *score_map;               
00162     unsigned map_generation;
00163     int pre_penalty_factor;
00164     int penalty_factor;                
00169     int sub_penalty_factor;
00170     int mb_penalty_factor;
00171     int flags;
00172     int sub_flags;
00173     int mb_flags;
00174     int pre_pass;                      
00175     int dia_size;
00176     int xmin;
00177     int xmax;
00178     int ymin;
00179     int ymax;
00180     int pred_x;
00181     int pred_y;
00182     uint8_t *src[4][4];
00183     uint8_t *ref[4][4];
00184     int stride;
00185     int uvstride;
00186     
00187     int mc_mb_var_sum_temp;
00188     int mb_var_sum_temp;
00189     int scene_change_score;
00190 
00191     op_pixels_func (*hpel_put)[4];
00192     op_pixels_func (*hpel_avg)[4];
00193     qpel_mc_func (*qpel_put)[16];
00194     qpel_mc_func (*qpel_avg)[16];
00195     uint8_t (*mv_penalty)[MAX_MV*2+1];  
00196     uint8_t *current_mv_penalty;
00197     int (*sub_motion_search)(struct MpegEncContext * s,
00198                                   int *mx_ptr, int *my_ptr, int dmin,
00199                                   int src_index, int ref_index,
00200                                   int size, int h);
00201 }MotionEstContext;
00202 
00206 typedef struct MpegEncContext {
00207     AVClass *class;
00208     struct AVCodecContext *avctx;
00209     
00210     int width, height;
00211     int gop_size;
00212     int intra_only;   
00213     int bit_rate;     
00214     enum OutputFormat out_format; 
00215     int h263_pred;    
00216     int pb_frame;     
00217 
00218 
00219     int h263_plus;    
00220     int h263_flv;     
00221 
00222     enum AVCodecID codec_id;     
00223     int fixed_qscale; 
00224     int encoding;     
00225     int flags;        
00226     int flags2;       
00227     int max_b_frames; 
00228     int luma_elim_threshold;
00229     int chroma_elim_threshold;
00230     int strict_std_compliance; 
00231     int workaround_bugs;       
00232     int codec_tag;             
00233     int stream_codec_tag;      
00234     
00235 
00236     
00237     int context_initialized;
00238     int input_picture_number;  
00239     int coded_picture_number;  
00240     int picture_number;       
00241     int picture_in_gop_number; 
00242     int mb_width, mb_height;   
00243     int mb_stride;             
00244     int b8_stride;             
00245     int b4_stride;             
00246     int h_edge_pos, v_edge_pos;
00247     int mb_num;                
00248     int linesize;              
00249     int uvlinesize;            
00250     Picture *picture;          
00251     Picture **input_picture;   
00252     Picture **reordered_input_picture; 
00253 
00254     int y_dc_scale, c_dc_scale;
00255     int ac_pred;
00256     int block_last_index[12];  
00257     int h263_aic;              
00258 
00259     
00260     ScanTable inter_scantable; 
00261     ScanTable intra_scantable;
00262     ScanTable intra_h_scantable;
00263     ScanTable intra_v_scantable;
00264 
00265     
00266 
00267 
00268     int64_t user_specified_pts;
00269 
00272     int64_t dts_delta;
00276     int64_t reordered_pts;
00277 
00279     PutBitContext pb;
00280 
00281     int start_mb_y;            
00282     int end_mb_y;              
00283     struct MpegEncContext *thread_context[MAX_THREADS];
00284     int slice_context_count;   
00285 
00290     Picture last_picture;
00291 
00296     Picture next_picture;
00297 
00302     Picture new_picture;
00303 
00308     Picture current_picture;    
00309 
00310     Picture *last_picture_ptr;     
00311     Picture *next_picture_ptr;     
00312     Picture *current_picture_ptr;  
00313     int picture_count;             
00314     int picture_range_start, picture_range_end; 
00315     uint8_t *visualization_buffer[3]; 
00316     int last_dc[3];                
00317     int16_t *dc_val_base;
00318     int16_t *dc_val[3];            
00319     const uint8_t *y_dc_scale_table;     
00320     const uint8_t *c_dc_scale_table;     
00321     const uint8_t *chroma_qscale_table;  
00322     uint8_t *coded_block_base;
00323     uint8_t *coded_block;          
00324     int16_t (*ac_val_base)[16];
00325     int16_t (*ac_val[3])[16];      
00326     int mb_skipped;                
00327     uint8_t *mbskip_table;        
00329     uint8_t *mbintra_table;       
00330     uint8_t *cbp_table;           
00331     uint8_t *pred_dir_table;      
00332     uint8_t *edge_emu_buffer;     
00333     uint8_t *rd_scratchpad;       
00334     uint8_t *obmc_scratchpad;
00335     uint8_t *b_scratchpad;        
00336 
00337     int qscale;                 
00338     int chroma_qscale;          
00339     unsigned int lambda;        
00340     unsigned int lambda2;       
00341     int *lambda_table;
00342     int adaptive_quant;         
00343     int dquant;                 
00344     int closed_gop;             
00345     int pict_type;              
00346     int last_pict_type; 
00347     int last_non_b_pict_type;   
00348     int dropable;
00349     int frame_rate_index;
00350     int last_lambda_for[5];     
00351     int skipdct;                
00352 
00353     
00354     int unrestricted_mv;        
00355     int h263_long_vectors;      
00356 
00357     DSPContext dsp;             
00358     int f_code;                 
00359     int b_code;                 
00360     int16_t (*p_mv_table_base)[2];
00361     int16_t (*b_forw_mv_table_base)[2];
00362     int16_t (*b_back_mv_table_base)[2];
00363     int16_t (*b_bidir_forw_mv_table_base)[2];
00364     int16_t (*b_bidir_back_mv_table_base)[2];
00365     int16_t (*b_direct_mv_table_base)[2];
00366     int16_t (*p_field_mv_table_base[2][2])[2];
00367     int16_t (*b_field_mv_table_base[2][2][2])[2];
00368     int16_t (*p_mv_table)[2];            
00369     int16_t (*b_forw_mv_table)[2];       
00370     int16_t (*b_back_mv_table)[2];       
00371     int16_t (*b_bidir_forw_mv_table)[2]; 
00372     int16_t (*b_bidir_back_mv_table)[2]; 
00373     int16_t (*b_direct_mv_table)[2];     
00374     int16_t (*p_field_mv_table[2][2])[2];   
00375     int16_t (*b_field_mv_table[2][2][2])[2];
00376     uint8_t (*p_field_select_table[2]);
00377     uint8_t (*b_field_select_table[2][2]);
00378     int me_method;                       
00379     int mv_dir;
00380 #define MV_DIR_FORWARD   1
00381 #define MV_DIR_BACKWARD  2
00382 #define MV_DIRECT        4 
00383     int mv_type;
00384 #define MV_TYPE_16X16       0   
00385 #define MV_TYPE_8X8         1   
00386 #define MV_TYPE_16X8        2   
00387 #define MV_TYPE_FIELD       3   
00388 #define MV_TYPE_DMV         4   
00389 
00394     int mv[2][4][2];
00395     int field_select[2][2];
00396     int last_mv[2][2][2];             
00397     uint8_t *fcode_tab;               
00398     int16_t direct_scale_mv[2][64];   
00399 
00400     MotionEstContext me;
00401 
00402     int no_rounding;  
00405     
00406     int mb_x, mb_y;
00407     int mb_skip_run;
00408     int mb_intra;
00409     uint16_t *mb_type;           
00410 #define CANDIDATE_MB_TYPE_INTRA    0x01
00411 #define CANDIDATE_MB_TYPE_INTER    0x02
00412 #define CANDIDATE_MB_TYPE_INTER4V  0x04
00413 #define CANDIDATE_MB_TYPE_SKIPPED   0x08
00414 
00415 
00416 #define CANDIDATE_MB_TYPE_DIRECT   0x10
00417 #define CANDIDATE_MB_TYPE_FORWARD  0x20
00418 #define CANDIDATE_MB_TYPE_BACKWARD 0x40
00419 #define CANDIDATE_MB_TYPE_BIDIR    0x80
00420 
00421 #define CANDIDATE_MB_TYPE_INTER_I    0x100
00422 #define CANDIDATE_MB_TYPE_FORWARD_I  0x200
00423 #define CANDIDATE_MB_TYPE_BACKWARD_I 0x400
00424 #define CANDIDATE_MB_TYPE_BIDIR_I    0x800
00425 
00426 #define CANDIDATE_MB_TYPE_DIRECT0    0x1000
00427 
00428     int block_index[6]; 
00429     int block_wrap[6];
00430     uint8_t *dest[3];
00431 
00432     int *mb_index2xy;        
00433 
00435     uint16_t intra_matrix[64];
00436     uint16_t chroma_intra_matrix[64];
00437     uint16_t inter_matrix[64];
00438     uint16_t chroma_inter_matrix[64];
00439 #define QUANT_BIAS_SHIFT 8
00440     int intra_quant_bias;    
00441     int inter_quant_bias;    
00442     int min_qcoeff;          
00443     int max_qcoeff;          
00444     int ac_esc_length;       
00445     uint8_t *intra_ac_vlc_length;
00446     uint8_t *intra_ac_vlc_last_length;
00447     uint8_t *inter_ac_vlc_length;
00448     uint8_t *inter_ac_vlc_last_length;
00449     uint8_t *luma_dc_vlc_length;
00450 #define UNI_AC_ENC_INDEX(run,level) ((run)*128 + (level))
00451 
00452     int coded_score[8];
00453 
00455     int (*q_intra_matrix)[64];
00456     int (*q_chroma_intra_matrix)[64];
00457     int (*q_inter_matrix)[64];
00459     uint16_t (*q_intra_matrix16)[2][64];
00460     uint16_t (*q_chroma_intra_matrix16)[2][64];
00461     uint16_t (*q_inter_matrix16)[2][64];
00462 
00463     
00464     int (*dct_error_sum)[64];
00465     int dct_count[2];
00466     uint16_t (*dct_offset)[64];
00467 
00468     void *opaque;              
00469 
00470     
00471     int64_t total_bits;
00472     int frame_bits;                
00473     int stuffing_bits;             
00474     int next_lambda;               
00475     RateControlContext rc_context; 
00476 
00477     
00478     int mv_bits;
00479     int header_bits;
00480     int i_tex_bits;
00481     int p_tex_bits;
00482     int i_count;
00483     int f_count;
00484     int b_count;
00485     int skip_count;
00486     int misc_bits; 
00487     int last_bits; 
00488 
00489     
00490     int error_count, error_occurred;
00491     uint8_t *error_status_table;       
00492 #define VP_START            1          
00493 #define ER_AC_ERROR            2
00494 #define ER_DC_ERROR            4
00495 #define ER_MV_ERROR            8
00496 #define ER_AC_END              16
00497 #define ER_DC_END              32
00498 #define ER_MV_END              64
00499 
00500 #define ER_MB_ERROR (ER_AC_ERROR|ER_DC_ERROR|ER_MV_ERROR)
00501 #define ER_MB_END   (ER_AC_END|ER_DC_END|ER_MV_END)
00502 
00503     int resync_mb_x;                 
00504     int resync_mb_y;                 
00505     GetBitContext last_resync_gb;    
00506     int mb_num_left;                 
00507     int next_p_frame_damaged;        
00508     int err_recognition;
00509 
00510     ParseContext parse_context;
00511 
00512     
00513     int gob_index;
00514     int obmc;                       
00515     int showed_packed_warning;      
00516     int mb_info;                    
00517     int prev_mb_info, last_mb_info;
00518     uint8_t *mb_info_ptr;
00519     int mb_info_size;
00520 
00521     
00522     int umvplus;                    
00523     int h263_aic_dir;               
00524     int h263_slice_structured;
00525     int alt_inter_vlc;              
00526     int modified_quant;
00527     int loop_filter;
00528     int custom_pcf;
00529 
00530     
00531     int time_increment_bits;        
00532     int last_time_base;
00533     int time_base;                  
00534     int64_t time;                   
00535     int64_t last_non_b_time;
00536     uint16_t pp_time;               
00537     uint16_t pb_time;               
00538     uint16_t pp_field_time;
00539     uint16_t pb_field_time;         
00540     int shape;
00541     int vol_sprite_usage;
00542     int sprite_width;
00543     int sprite_height;
00544     int sprite_left;
00545     int sprite_top;
00546     int sprite_brightness_change;
00547     int num_sprite_warping_points;
00548     int real_sprite_warping_points;
00549     uint16_t sprite_traj[4][2];      
00550     int sprite_offset[2][2];         
00551     int sprite_delta[2][2];          
00552     int sprite_shift[2];             
00553     int mcsel;
00554     int quant_precision;
00555     int quarter_sample;              
00556     int scalability;
00557     int hierachy_type;
00558     int enhancement_type;
00559     int new_pred;
00560     int reduced_res_vop;
00561     int aspect_ratio_info; 
00562     int sprite_warping_accuracy;
00563     int low_latency_sprite;
00564     int data_partitioning;           
00565     int partitioned_frame;           
00566     int rvlc;                        
00567     int resync_marker;               
00568     int low_delay;                   
00569     int vo_type;
00570     int vol_control_parameters;      
00571     int intra_dc_threshold;          
00572     int use_intra_dc_vlc;
00573     PutBitContext tex_pb;            
00574     PutBitContext pb2;               
00575     int mpeg_quant;
00576     int t_frame;                       
00577     int padding_bug_score;             
00578     int cplx_estimation_trash_i;
00579     int cplx_estimation_trash_p;
00580     int cplx_estimation_trash_b;
00581 
00582     
00583     int divx_version;
00584     int divx_build;
00585     int divx_packed;
00586     uint8_t *bitstream_buffer; 
00587     int bitstream_buffer_size;
00588     unsigned int allocated_bitstream_buffer_size;
00589 
00590     int xvid_build;
00591 
00592     
00593     int lavc_build;
00594 
00595     
00596     int rv10_version; 
00597     int rv10_first_dc_coded[3];
00598     int orig_width, orig_height;
00599 
00600     
00601     struct MJpegContext *mjpeg_ctx;
00602     int mjpeg_vsample[3];       
00603     int mjpeg_hsample[3];       
00604     int esc_pos;
00605 
00606     
00607     int mv_table_index;
00608     int rl_table_index;
00609     int rl_chroma_table_index;
00610     int dc_table_index;
00611     int use_skip_mb_code;
00612     int slice_height;      
00613     int first_slice_line;  
00614     int flipflop_rounding;
00615     int msmpeg4_version;   
00616     int per_mb_rl_table;
00617     int esc3_level_length;
00618     int esc3_run_length;
00620     int (*ac_stats)[2][MAX_LEVEL+1][MAX_RUN+1][2];
00621     int inter_intra_pred;
00622     int mspel;
00623 
00624     
00625     GetBitContext gb;
00626 
00627     
00628     int gop_picture_number;  
00629     int last_mv_dir;         
00630     int broken_link;         
00631     uint8_t *vbv_delay_ptr;  
00632 
00633     
00634     int progressive_sequence;
00635     int mpeg_f_code[2][2];
00636     int picture_structure;
00637 
00638 #define PICT_TOP_FIELD     1
00639 #define PICT_BOTTOM_FIELD  2
00640 #define PICT_FRAME         3
00641 
00642     int intra_dc_precision;
00643     int frame_pred_frame_dct;
00644     int top_field_first;
00645     int concealment_motion_vectors;
00646     int q_scale_type;
00647     int intra_vlc_format;
00648     int alternate_scan;
00649     int repeat_first_field;
00650     int chroma_420_type;
00651     int chroma_format;
00652 #define CHROMA_420 1
00653 #define CHROMA_422 2
00654 #define CHROMA_444 3
00655     int chroma_x_shift;
00656     int chroma_y_shift;
00657 
00658     int progressive_frame;
00659     int full_pel[2];
00660     int interlaced_dct;
00661     int first_slice;
00662     int first_field;         
00663     int drop_frame_timecode; 
00664     int scan_offset;         
00665 
00666     
00667     int rtp_mode;
00668 
00669     char *tc_opt_str;        
00670     AVTimecode tc;           
00671 
00672     uint8_t *ptr_lastgob;
00673     int swap_uv;             
00674     DCTELEM (*pblocks[12])[64];
00675 
00676     DCTELEM (*block)[64]; 
00677     DCTELEM (*blocks)[8][64]; 
00678     int (*decode_mb)(struct MpegEncContext *s, DCTELEM block[6][64]); 
00679 #define SLICE_OK         0
00680 #define SLICE_ERROR     -1
00681 #define SLICE_END       -2 
00682 #define SLICE_NOEND     -3 
00683 
00684     void (*dct_unquantize_mpeg1_intra)(struct MpegEncContext *s,
00685                            DCTELEM *block, int n, int qscale);
00686     void (*dct_unquantize_mpeg1_inter)(struct MpegEncContext *s,
00687                            DCTELEM *block, int n, int qscale);
00688     void (*dct_unquantize_mpeg2_intra)(struct MpegEncContext *s,
00689                            DCTELEM *block, int n, int qscale);
00690     void (*dct_unquantize_mpeg2_inter)(struct MpegEncContext *s,
00691                            DCTELEM *block, int n, int qscale);
00692     void (*dct_unquantize_h263_intra)(struct MpegEncContext *s,
00693                            DCTELEM *block, int n, int qscale);
00694     void (*dct_unquantize_h263_inter)(struct MpegEncContext *s,
00695                            DCTELEM *block, int n, int qscale);
00696     void (*dct_unquantize_h261_intra)(struct MpegEncContext *s,
00697                            DCTELEM *block, int n, int qscale);
00698     void (*dct_unquantize_h261_inter)(struct MpegEncContext *s,
00699                            DCTELEM *block, int n, int qscale);
00700     void (*dct_unquantize_intra)(struct MpegEncContext *s, 
00701                            DCTELEM *block, int n, int qscale);
00702     void (*dct_unquantize_inter)(struct MpegEncContext *s, 
00703                            DCTELEM *block, int n, int qscale);
00704     int (*dct_quantize)(struct MpegEncContext *s, DCTELEM *block, int n, int qscale, int *overflow);
00705     int (*fast_dct_quantize)(struct MpegEncContext *s, DCTELEM *block, int n, int qscale, int *overflow);
00706     void (*denoise_dct)(struct MpegEncContext *s, DCTELEM *block);
00707 
00708     int mpv_flags;      
00709     int quantizer_noise_shaping;
00710 
00711     
00712     uint8_t *er_temp_buffer;
00713 
00714     
00715     float *cplx_tab, *bits_tab;
00716 
00717     
00718 
00719     int context_reinit;
00720 } MpegEncContext;
00721 
00722 #define REBASE_PICTURE(pic, new_ctx, old_ctx) (pic ? \
00723     (pic >= old_ctx->picture && pic < old_ctx->picture+old_ctx->picture_count ?\
00724         &new_ctx->picture[pic - old_ctx->picture] : pic - (Picture*)old_ctx + (Picture*)new_ctx)\
00725     : NULL)
00726 
00727 
00728 #define FF_MPV_FLAG_SKIP_RD      0x0001
00729 #define FF_MPV_FLAG_STRICT_GOP   0x0002
00730 #define FF_MPV_FLAG_QP_RD        0x0004
00731 #define FF_MPV_FLAG_CBP_RD       0x0008
00732 
00733 #define FF_MPV_OFFSET(x) offsetof(MpegEncContext, x)
00734 #define FF_MPV_OPT_FLAGS (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)
00735 #define FF_MPV_COMMON_OPTS \
00736 { "mpv_flags",      "Flags common for all mpegvideo-based encoders.", FF_MPV_OFFSET(mpv_flags), AV_OPT_TYPE_FLAGS, { .i64 = 0 }, INT_MIN, INT_MAX, FF_MPV_OPT_FLAGS, "mpv_flags" },\
00737 { "skip_rd",        "RD optimal MB level residual skipping", 0, AV_OPT_TYPE_CONST, { .i64 = FF_MPV_FLAG_SKIP_RD },    0, 0, FF_MPV_OPT_FLAGS, "mpv_flags" },\
00738 { "strict_gop",     "Strictly enforce gop size",             0, AV_OPT_TYPE_CONST, { .i64 = FF_MPV_FLAG_STRICT_GOP }, 0, 0, FF_MPV_OPT_FLAGS, "mpv_flags" },\
00739 { "qp_rd",          "Use rate distortion optimization for qp selection", 0, AV_OPT_TYPE_CONST, { .i64 = FF_MPV_FLAG_QP_RD },  0, 0, FF_MPV_OPT_FLAGS, "mpv_flags" },\
00740 { "cbp_rd",         "use rate distortion optimization for CBP",          0, AV_OPT_TYPE_CONST, { .i64 = FF_MPV_FLAG_CBP_RD }, 0, 0, FF_MPV_OPT_FLAGS, "mpv_flags" },\
00741 { "luma_elim_threshold",   "single coefficient elimination threshold for luminance (negative values also consider dc coefficient)",\
00742                                                                       FF_MPV_OFFSET(luma_elim_threshold), AV_OPT_TYPE_INT, { .i64 = 0 }, INT_MIN, INT_MAX, FF_MPV_OPT_FLAGS },\
00743 { "chroma_elim_threshold", "single coefficient elimination threshold for chrominance (negative values also consider dc coefficient)",\
00744                                                                       FF_MPV_OFFSET(chroma_elim_threshold), AV_OPT_TYPE_INT, { .i64 = 0 }, INT_MIN, INT_MAX, FF_MPV_OPT_FLAGS },\
00745 { "quantizer_noise_shaping", NULL,                                  FF_MPV_OFFSET(quantizer_noise_shaping), AV_OPT_TYPE_INT, { .i64 = 0 },       0, INT_MAX, FF_MPV_OPT_FLAGS },
00746 
00747 extern const AVOption ff_mpv_generic_options[];
00748 
00749 #define FF_MPV_GENERIC_CLASS(name) \
00750 static const AVClass name ## _class = {\
00751     .class_name = #name " encoder",\
00752     .item_name  = av_default_item_name,\
00753     .option     = ff_mpv_generic_options,\
00754     .version    = LIBAVUTIL_VERSION_INT,\
00755 };
00756 
00762 void ff_MPV_common_defaults(MpegEncContext *s);
00763 
00764 void ff_MPV_decode_defaults(MpegEncContext *s);
00765 int ff_MPV_common_init(MpegEncContext *s);
00766 int ff_MPV_common_frame_size_change(MpegEncContext *s);
00767 void ff_MPV_common_end(MpegEncContext *s);
00768 void ff_MPV_decode_mb(MpegEncContext *s, DCTELEM block[12][64]);
00769 int ff_MPV_frame_start(MpegEncContext *s, AVCodecContext *avctx);
00770 void ff_MPV_frame_end(MpegEncContext *s);
00771 int ff_MPV_encode_init(AVCodecContext *avctx);
00772 int ff_MPV_encode_end(AVCodecContext *avctx);
00773 int ff_MPV_encode_picture(AVCodecContext *avctx, AVPacket *pkt,
00774                           AVFrame *frame, int *got_packet);
00775 void ff_dct_encode_init_x86(MpegEncContext *s);
00776 void ff_MPV_common_init_x86(MpegEncContext *s);
00777 void ff_MPV_common_init_axp(MpegEncContext *s);
00778 void ff_MPV_common_init_mmi(MpegEncContext *s);
00779 void ff_MPV_common_init_arm(MpegEncContext *s);
00780 void ff_MPV_common_init_altivec(MpegEncContext *s);
00781 void ff_MPV_common_init_bfin(MpegEncContext *s);
00782 void ff_clean_intra_table_entries(MpegEncContext *s);
00783 void ff_draw_horiz_band(MpegEncContext *s, int y, int h);
00784 void ff_mpeg_flush(AVCodecContext *avctx);
00785 void ff_print_debug_info(MpegEncContext *s, AVFrame *pict);
00786 void ff_write_quant_matrix(PutBitContext *pb, uint16_t *matrix);
00787 void ff_release_unused_pictures(MpegEncContext *s, int remove_current);
00788 int ff_find_unused_picture(MpegEncContext *s, int shared);
00789 void ff_denoise_dct(MpegEncContext *s, DCTELEM *block);
00790 void ff_update_duplicate_context(MpegEncContext *dst, MpegEncContext *src);
00791 int ff_MPV_lowest_referenced_row(MpegEncContext *s, int dir);
00792 void ff_MPV_report_decode_progress(MpegEncContext *s);
00793 int ff_mpeg_update_thread_context(AVCodecContext *dst, const AVCodecContext *src);
00794 const uint8_t *avpriv_mpv_find_start_code(const uint8_t *p, const uint8_t *end, uint32_t *state);
00795 void ff_set_qscale(MpegEncContext * s, int qscale);
00796 
00797 void ff_er_frame_start(MpegEncContext *s);
00798 void ff_er_frame_end(MpegEncContext *s);
00799 void ff_er_add_slice(MpegEncContext *s, int startx, int starty, int endx, int endy, int status);
00800 
00801 int ff_dct_common_init(MpegEncContext *s);
00802 int ff_dct_encode_init(MpegEncContext *s);
00803 void ff_convert_matrix(DSPContext *dsp, int (*qmat)[64], uint16_t (*qmat16)[2][64],
00804                        const uint16_t *quant_matrix, int bias, int qmin, int qmax, int intra);
00805 int ff_dct_quantize_c(MpegEncContext *s, DCTELEM *block, int n, int qscale, int *overflow);
00806 
00807 void ff_init_block_index(MpegEncContext *s);
00808 void ff_copy_picture(Picture *dst, Picture *src);
00809 
00810 void ff_MPV_motion(MpegEncContext *s,
00811                    uint8_t *dest_y, uint8_t *dest_cb,
00812                    uint8_t *dest_cr, int dir,
00813                    uint8_t **ref_picture,
00814                    op_pixels_func (*pix_op)[4],
00815                    qpel_mc_func (*qpix_op)[16]);
00816 
00821 int ff_alloc_picture(MpegEncContext *s, Picture *pic, int shared);
00822 
00823 extern const enum PixelFormat ff_pixfmt_list_420[];
00824 extern const enum PixelFormat ff_hwaccel_pixfmt_list_420[];
00825 
00826 static inline void ff_update_block_index(MpegEncContext *s){
00827     const int block_size= 8 >> s->avctx->lowres;
00828 
00829     s->block_index[0]+=2;
00830     s->block_index[1]+=2;
00831     s->block_index[2]+=2;
00832     s->block_index[3]+=2;
00833     s->block_index[4]++;
00834     s->block_index[5]++;
00835     s->dest[0]+= 2*block_size;
00836     s->dest[1]+= block_size;
00837     s->dest[2]+= block_size;
00838 }
00839 
00840 static inline int get_bits_diff(MpegEncContext *s){
00841     const int bits= put_bits_count(&s->pb);
00842     const int last= s->last_bits;
00843 
00844     s->last_bits = bits;
00845 
00846     return bits - last;
00847 }
00848 
00849 static inline int ff_h263_round_chroma(int x){
00850     static const uint8_t h263_chroma_roundtab[16] = {
00851     
00852         0, 0, 0, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 1,
00853     };
00854     return h263_chroma_roundtab[x & 0xf] + (x >> 3);
00855 }
00856 
00857 
00858 void ff_estimate_p_frame_motion(MpegEncContext * s,
00859                              int mb_x, int mb_y);
00860 void ff_estimate_b_frame_motion(MpegEncContext * s,
00861                              int mb_x, int mb_y);
00862 int ff_get_best_fcode(MpegEncContext * s, int16_t (*mv_table)[2], int type);
00863 void ff_fix_long_p_mvs(MpegEncContext * s);
00864 void ff_fix_long_mvs(MpegEncContext * s, uint8_t *field_select_table, int field_select,
00865                      int16_t (*mv_table)[2], int f_code, int type, int truncate);
00866 int ff_init_me(MpegEncContext *s);
00867 int ff_pre_estimate_p_frame_motion(MpegEncContext * s, int mb_x, int mb_y);
00868 int ff_epzs_motion_search(MpegEncContext * s, int *mx_ptr, int *my_ptr,
00869                              int P[10][2], int src_index, int ref_index, int16_t (*last_mv)[2],
00870                              int ref_mv_scale, int size, int h);
00871 int ff_get_mb_score(MpegEncContext * s, int mx, int my, int src_index,
00872                                int ref_index, int size, int h, int add_rate);
00873 
00874 
00875 extern const uint8_t ff_mpeg1_dc_scale_table[128];
00876 extern const uint8_t * const ff_mpeg2_dc_scale_table[4];
00877 
00878 void ff_mpeg1_encode_picture_header(MpegEncContext *s, int picture_number);
00879 void ff_mpeg1_encode_mb(MpegEncContext *s,
00880                         DCTELEM block[6][64],
00881                         int motion_x, int motion_y);
00882 void ff_mpeg1_encode_init(MpegEncContext *s);
00883 void ff_mpeg1_encode_slice_header(MpegEncContext *s);
00884 void ff_mpeg1_clean_buffers(MpegEncContext *s);
00885 int ff_mpeg1_find_frame_end(ParseContext *pc, const uint8_t *buf, int buf_size, AVCodecParserContext *s);
00886 
00887 extern const uint8_t ff_aic_dc_scale_table[32];
00888 extern const uint8_t ff_h263_chroma_qscale_table[32];
00889 extern const uint8_t ff_h263_loop_filter_strength[32];
00890 
00891 
00892 void ff_h261_loop_filter(MpegEncContext *s);
00893 void ff_h261_reorder_mb_index(MpegEncContext* s);
00894 void ff_h261_encode_mb(MpegEncContext *s,
00895                     DCTELEM block[6][64],
00896                     int motion_x, int motion_y);
00897 void ff_h261_encode_picture_header(MpegEncContext * s, int picture_number);
00898 void ff_h261_encode_init(MpegEncContext *s);
00899 int ff_h261_get_picture_format(int width, int height);
00900 
00901 
00902 
00903 void ff_rv10_encode_picture_header(MpegEncContext *s, int picture_number);
00904 int ff_rv_decode_dc(MpegEncContext *s, int n);
00905 void ff_rv20_encode_picture_header(MpegEncContext *s, int picture_number);
00906 
00907 
00908 
00909 void ff_msmpeg4_encode_picture_header(MpegEncContext * s, int picture_number);
00910 void ff_msmpeg4_encode_ext_header(MpegEncContext * s);
00911 void ff_msmpeg4_encode_mb(MpegEncContext * s,
00912                           DCTELEM block[6][64],
00913                           int motion_x, int motion_y);
00914 int ff_msmpeg4_decode_picture_header(MpegEncContext * s);
00915 int ff_msmpeg4_decode_ext_header(MpegEncContext * s, int buf_size);
00916 int ff_msmpeg4_decode_init(AVCodecContext *avctx);
00917 void ff_msmpeg4_encode_init(MpegEncContext *s);
00918 int ff_wmv2_decode_picture_header(MpegEncContext * s);
00919 int ff_wmv2_decode_secondary_picture_header(MpegEncContext * s);
00920 void ff_wmv2_add_mb(MpegEncContext *s, DCTELEM block[6][64], uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr);
00921 void ff_mspel_motion(MpegEncContext *s,
00922                                uint8_t *dest_y, uint8_t *dest_cb, uint8_t *dest_cr,
00923                                uint8_t **ref_picture, op_pixels_func (*pix_op)[4],
00924                                int motion_x, int motion_y, int h);
00925 int ff_wmv2_encode_picture_header(MpegEncContext * s, int picture_number);
00926 void ff_wmv2_encode_mb(MpegEncContext * s,
00927                        DCTELEM block[6][64],
00928                        int motion_x, int motion_y);
00929 
00930 #endif