LCOV - code coverage report
Current view: top level - libavcodec - mpeg4video.h (source / functions) Hit Total Coverage
Test: coverage.info Lines: 31 42 73.8 %
Date: 2018-05-20 11:54:08 Functions: 1 1 100.0 %

          Line data    Source code
       1             : /*
       2             :  * MPEG-4 encoder/decoder internal header.
       3             :  * Copyright (c) 2000,2001 Fabrice Bellard
       4             :  * Copyright (c) 2002-2010 Michael Niedermayer <michaelni@gmx.at>
       5             :  *
       6             :  * This file is part of FFmpeg.
       7             :  *
       8             :  * FFmpeg is free software; you can redistribute it and/or
       9             :  * modify it under the terms of the GNU Lesser General Public
      10             :  * License as published by the Free Software Foundation; either
      11             :  * version 2.1 of the License, or (at your option) any later version.
      12             :  *
      13             :  * FFmpeg is distributed in the hope that it will be useful,
      14             :  * but WITHOUT ANY WARRANTY; without even the implied warranty of
      15             :  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      16             :  * Lesser General Public License for more details.
      17             :  *
      18             :  * You should have received a copy of the GNU Lesser General Public
      19             :  * License along with FFmpeg; if not, write to the Free Software
      20             :  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
      21             :  */
      22             : 
      23             : #ifndef AVCODEC_MPEG4VIDEO_H
      24             : #define AVCODEC_MPEG4VIDEO_H
      25             : 
      26             : #include <stdint.h>
      27             : 
      28             : #include "get_bits.h"
      29             : #include "mpegvideo.h"
      30             : #include "rl.h"
      31             : 
      32             : // shapes
      33             : #define RECT_SHAPE       0
      34             : #define BIN_SHAPE        1
      35             : #define BIN_ONLY_SHAPE   2
      36             : #define GRAY_SHAPE       3
      37             : 
      38             : #define SIMPLE_VO_TYPE           1
      39             : #define CORE_VO_TYPE             3
      40             : #define MAIN_VO_TYPE             4
      41             : #define NBIT_VO_TYPE             5
      42             : #define ARTS_VO_TYPE            10
      43             : #define ACE_VO_TYPE             12
      44             : #define SIMPLE_STUDIO_VO_TYPE   14
      45             : #define CORE_STUDIO_VO_TYPE     15
      46             : #define ADV_SIMPLE_VO_TYPE      17
      47             : 
      48             : #define VOT_VIDEO_ID 1
      49             : #define VOT_STILL_TEXTURE_ID 2
      50             : 
      51             : // aspect_ratio_info
      52             : #define EXTENDED_PAR 15
      53             : 
      54             : //vol_sprite_usage / sprite_enable
      55             : #define STATIC_SPRITE 1
      56             : #define GMC_SPRITE 2
      57             : 
      58             : #define MOTION_MARKER 0x1F001
      59             : #define DC_MARKER     0x6B001
      60             : 
      61             : #define VOS_STARTCODE        0x1B0
      62             : #define USER_DATA_STARTCODE  0x1B2
      63             : #define GOP_STARTCODE        0x1B3
      64             : #define VISUAL_OBJ_STARTCODE 0x1B5
      65             : #define VOP_STARTCODE        0x1B6
      66             : #define SLICE_STARTCODE      0x1B7
      67             : #define EXT_STARTCODE        0x1B8
      68             : 
      69             : #define QUANT_MATRIX_EXT_ID  0x3
      70             : 
      71             : /* smaller packets likely don't contain a real frame */
      72             : #define MAX_NVOP_SIZE 19
      73             : 
      74             : typedef struct Mpeg4DecContext {
      75             :     MpegEncContext m;
      76             : 
      77             :     /// number of bits to represent the fractional part of time
      78             :     int time_increment_bits;
      79             :     int shape;
      80             :     int vol_sprite_usage;
      81             :     int sprite_brightness_change;
      82             :     int num_sprite_warping_points;
      83             :     /// sprite trajectory points
      84             :     uint16_t sprite_traj[4][2];
      85             :     /// sprite shift [isChroma]
      86             :     int sprite_shift[2];
      87             : 
      88             :     // reversible vlc
      89             :     int rvlc;
      90             :     /// could this stream contain resync markers
      91             :     int resync_marker;
      92             :     /// time distance of first I -> B, used for interlaced B-frames
      93             :     int t_frame;
      94             : 
      95             :     int new_pred;
      96             :     int enhancement_type;
      97             :     int scalability;
      98             :     int use_intra_dc_vlc;
      99             : 
     100             :     /// QP above which the ac VLC should be used for intra dc
     101             :     int intra_dc_threshold;
     102             : 
     103             :     /* bug workarounds */
     104             :     int divx_version;
     105             :     int divx_build;
     106             :     int xvid_build;
     107             :     int lavc_build;
     108             : 
     109             :     /// flag for having shown the warning about invalid Divx B-frames
     110             :     int showed_packed_warning;
     111             :     /** does the stream contain the low_delay flag,
     112             :      *  used to work around buggy encoders. */
     113             :     int vol_control_parameters;
     114             :     int cplx_estimation_trash_i;
     115             :     int cplx_estimation_trash_p;
     116             :     int cplx_estimation_trash_b;
     117             : 
     118             :     VLC studio_intra_tab[12];
     119             :     VLC studio_luma_dc;
     120             :     VLC studio_chroma_dc;
     121             : 
     122             :     int rgb;
     123             : } Mpeg4DecContext;
     124             : 
     125             : static const uint8_t mpeg4_block_count[4] = {0, 6, 8, 12};
     126             : 
     127             : /* dc encoding for MPEG-4 */
     128             : extern const uint8_t ff_mpeg4_DCtab_lum[13][2];
     129             : extern const uint8_t ff_mpeg4_DCtab_chrom[13][2];
     130             : 
     131             : extern const uint16_t ff_mpeg4_intra_vlc[103][2];
     132             : extern const int8_t ff_mpeg4_intra_level[102];
     133             : extern const int8_t ff_mpeg4_intra_run[102];
     134             : 
     135             : extern RLTable ff_mpeg4_rl_intra;
     136             : 
     137             : /* Note this is identical to the intra rvlc except that it is reordered. */
     138             : extern RLTable ff_rvlc_rl_inter;
     139             : extern RLTable ff_rvlc_rl_intra;
     140             : 
     141             : extern const uint16_t ff_sprite_trajectory_tab[15][2];
     142             : extern const uint8_t ff_mb_type_b_tab[4][2];
     143             : 
     144             : /* these matrixes will be permuted for the idct */
     145             : extern const int16_t ff_mpeg4_default_intra_matrix[64];
     146             : extern const int16_t ff_mpeg4_default_non_intra_matrix[64];
     147             : 
     148             : extern const uint8_t ff_mpeg4_y_dc_scale_table[32];
     149             : extern const uint8_t ff_mpeg4_c_dc_scale_table[32];
     150             : extern const uint16_t ff_mpeg4_resync_prefix[8];
     151             : 
     152             : extern const uint8_t ff_mpeg4_dc_threshold[8];
     153             : 
     154             : extern const uint16_t ff_mpeg4_studio_dc_luma[19][2];
     155             : extern const uint16_t ff_mpeg4_studio_dc_chroma[19][2];
     156             : extern const uint16_t ff_mpeg4_studio_intra[12][22][2];
     157             : 
     158             : void ff_mpeg4_encode_mb(MpegEncContext *s,
     159             :                         int16_t block[6][64],
     160             :                         int motion_x, int motion_y);
     161             : void ff_mpeg4_pred_ac(MpegEncContext *s, int16_t *block, int n,
     162             :                       int dir);
     163             : void ff_set_mpeg4_time(MpegEncContext *s);
     164             : int ff_mpeg4_encode_picture_header(MpegEncContext *s, int picture_number);
     165             : 
     166             : int ff_mpeg4_decode_picture_header(Mpeg4DecContext *ctx, GetBitContext *gb);
     167             : void ff_mpeg4_encode_video_packet_header(MpegEncContext *s);
     168             : void ff_mpeg4_clean_buffers(MpegEncContext *s);
     169             : void ff_mpeg4_stuffing(PutBitContext *pbc);
     170             : void ff_mpeg4_init_partitions(MpegEncContext *s);
     171             : void ff_mpeg4_merge_partitions(MpegEncContext *s);
     172             : void ff_clean_mpeg4_qscales(MpegEncContext *s);
     173             : int ff_mpeg4_decode_partitions(Mpeg4DecContext *ctx);
     174             : int ff_mpeg4_get_video_packet_prefix_length(MpegEncContext *s);
     175             : int ff_mpeg4_decode_video_packet_header(Mpeg4DecContext *ctx);
     176             : int ff_mpeg4_decode_studio_slice_header(Mpeg4DecContext *ctx);
     177             : void ff_mpeg4_init_direct_mv(MpegEncContext *s);
     178             : void ff_mpeg4videodec_static_init(void);
     179             : int ff_mpeg4_workaround_bugs(AVCodecContext *avctx);
     180             : int ff_mpeg4_frame_end(AVCodecContext *avctx, const uint8_t *buf, int buf_size);
     181             : 
     182             : /**
     183             :  * @return the mb_type
     184             :  */
     185             : int ff_mpeg4_set_direct_mv(MpegEncContext *s, int mx, int my);
     186             : 
     187             : extern uint8_t ff_mpeg4_static_rl_table_store[3][2][2 * MAX_RUN + MAX_LEVEL + 3];
     188             : 
     189             : #if 0 //3IV1 is quite rare and it slows things down a tiny bit
     190             : #define IS_3IV1 s->codec_tag == AV_RL32("3IV1")
     191             : #else
     192             : #define IS_3IV1 0
     193             : #endif
     194             : 
     195             : /**
     196             :  * Predict the dc.
     197             :  * encoding quantized level -> quantized diff
     198             :  * decoding quantized diff -> quantized level
     199             :  * @param n block index (0-3 are luma, 4-5 are chroma)
     200             :  * @param dir_ptr pointer to an integer where the prediction direction will be stored
     201             :  */
     202     2262612 : static inline int ff_mpeg4_pred_dc(MpegEncContext *s, int n, int level,
     203             :                                    int *dir_ptr, int encoding)
     204             : {
     205             :     int a, b, c, wrap, pred, scale, ret;
     206             :     int16_t *dc_val;
     207             : 
     208             :     /* find prediction */
     209     2262612 :     if (n < 4)
     210     1508408 :         scale = s->y_dc_scale;
     211             :     else
     212      754204 :         scale = s->c_dc_scale;
     213             :     if (IS_3IV1)
     214             :         scale = 8;
     215             : 
     216     2262612 :     wrap   = s->block_wrap[n];
     217     2262612 :     dc_val = s->dc_val[0] + s->block_index[n];
     218             : 
     219             :     /* B C
     220             :      * A X
     221             :      */
     222     2262612 :     a = dc_val[-1];
     223     2262612 :     b = dc_val[-1 - wrap];
     224     2262612 :     c = dc_val[-wrap];
     225             : 
     226             :     /* outside slice handling (we can't do that by memset as we need the
     227             :      * dc for error resilience) */
     228     2262612 :     if (s->first_slice_line && n != 3) {
     229      231180 :         if (n != 2)
     230      184944 :             b = c = 1024;
     231      231180 :         if (n != 1 && s->mb_x == s->resync_mb_x)
     232        9840 :             b = a = 1024;
     233             :     }
     234     2262612 :     if (s->mb_x == s->resync_mb_x && s->mb_y == s->resync_mb_y + 1) {
     235       14814 :         if (n == 0 || n == 4 || n == 5)
     236        7407 :             b = 1024;
     237             :     }
     238             : 
     239     2262612 :     if (abs(a - b) < abs(b - c)) {
     240      835154 :         pred     = c;
     241      835154 :         *dir_ptr = 1; /* top */
     242             :     } else {
     243     1427458 :         pred     = a;
     244     1427458 :         *dir_ptr = 0; /* left */
     245             :     }
     246             :     /* we assume pred is positive */
     247     2262612 :     pred = FASTDIV((pred + (scale >> 1)), scale);
     248             : 
     249     2262612 :     if (encoding) {
     250     1205160 :         ret = level - pred;
     251             :     } else {
     252     1057452 :         level += pred;
     253     1057452 :         ret    = level;
     254             :     }
     255     2262612 :     level *= scale;
     256     2262612 :     if (level & (~2047)) {
     257           0 :         if (!s->encoding && (s->avctx->err_recognition & (AV_EF_BITSTREAM | AV_EF_AGGRESSIVE))) {
     258           0 :             if (level < 0) {
     259           0 :                 av_log(s->avctx, AV_LOG_ERROR,
     260             :                        "dc<0 at %dx%d\n", s->mb_x, s->mb_y);
     261           0 :                 return AVERROR_INVALIDDATA;
     262             :             }
     263           0 :             if (level > 2048 + scale) {
     264           0 :                 av_log(s->avctx, AV_LOG_ERROR,
     265             :                        "dc overflow at %dx%d\n", s->mb_x, s->mb_y);
     266           0 :                 return AVERROR_INVALIDDATA;
     267             :             }
     268             :         }
     269           0 :         if (level < 0)
     270           0 :             level = 0;
     271           0 :         else if (!(s->workaround_bugs & FF_BUG_DC_CLIP))
     272           0 :             level = 2047;
     273             :     }
     274     2262612 :     dc_val[0] = level;
     275             : 
     276     2262612 :     return ret;
     277             : }
     278             : 
     279             : #endif /* AVCODEC_MPEG4VIDEO_H */

Generated by: LCOV version 1.13