* commit '1b3439b3055b083df51d7f7838ecc6b3f708b15c': mpegvideo: move frame size dependent memory management to separate functions configure: add --toolchain option configure: Make the smoothstreaming muxer enable the ismv muxer smoothstreaming: Export the mp4 codec tags mov: check for EOF in long lasting loops avcodec: cleanup utils.c binkaudio: remove unneeded GET_BITS_SAFE macro binkaudio: use float sample format binkaudio: use a different value for the coefficient scale for the DCT codec Conflicts: configure libavcodec/mpegvideo.c libavcodec/utils.c libavformat/Makefile Merged-by: Michael Niedermayer <michaelni@gmx.at>tags/n1.0
| @@ -232,6 +232,7 @@ Advanced options (experts only): | |||||
| --target-os=OS compiler targets OS [$target_os] | --target-os=OS compiler targets OS [$target_os] | ||||
| --target-exec=CMD command to run executables on target | --target-exec=CMD command to run executables on target | ||||
| --target-path=DIR path to view of build directory on target | --target-path=DIR path to view of build directory on target | ||||
| --toolchain=NAME set tool defaults according to NAME | |||||
| --nm=NM use nm tool NM [$nm_default] | --nm=NM use nm tool NM [$nm_default] | ||||
| --ar=AR use archive tool AR [$ar_default] | --ar=AR use archive tool AR [$ar_default] | ||||
| --as=AS use assembler AS [$as_default] | --as=AS use assembler AS [$as_default] | ||||
| @@ -1485,6 +1486,7 @@ CMDLINE_SET=" | |||||
| target_exec | target_exec | ||||
| target_os | target_os | ||||
| target_path | target_path | ||||
| toolchain | |||||
| valgrind | valgrind | ||||
| yasmexe | yasmexe | ||||
| " | " | ||||
| @@ -1813,6 +1815,7 @@ rtsp_muxer_select="rtp_muxer http_protocol rtp_protocol" | |||||
| sap_demuxer_select="sdp_demuxer" | sap_demuxer_select="sdp_demuxer" | ||||
| sap_muxer_select="rtp_muxer rtp_protocol" | sap_muxer_select="rtp_muxer rtp_protocol" | ||||
| sdp_demuxer_select="rtpdec" | sdp_demuxer_select="rtpdec" | ||||
| smoothstreaming_muxer_select="ismv_muxer" | |||||
| spdif_muxer_select="aac_parser" | spdif_muxer_select="aac_parser" | ||||
| tg2_muxer_select="mov_muxer" | tg2_muxer_select="mov_muxer" | ||||
| tgp_muxer_select="mov_muxer" | tgp_muxer_select="mov_muxer" | ||||
| @@ -2308,6 +2311,17 @@ strip_default="${cross_prefix}${strip_default}" | |||||
| sysinclude_default="${sysroot}/usr/include" | sysinclude_default="${sysroot}/usr/include" | ||||
| case "$toolchain" in | |||||
| msvc) | |||||
| cc_default="c99wrap cl" | |||||
| ld_default="c99wrap link" | |||||
| nm_default="dumpbin -symbols" | |||||
| ;; | |||||
| ?*) | |||||
| die "Unknown toolchain $toolchain" | |||||
| ;; | |||||
| esac | |||||
| set_default cc cxx pkg_config strip sysinclude yasmexe | set_default cc cxx pkg_config strip sysinclude yasmexe | ||||
| enabled cross_compile || host_cc_default=$cc | enabled cross_compile || host_cc_default=$cc | ||||
| set_default host_cc | set_default host_cc | ||||
| @@ -47,8 +47,6 @@ static float quant_table[96]; | |||||
| typedef struct { | typedef struct { | ||||
| AVFrame frame; | AVFrame frame; | ||||
| GetBitContext gb; | GetBitContext gb; | ||||
| DSPContext dsp; | |||||
| FmtConvertContext fmt_conv; | |||||
| int version_b; ///< Bink version 'b' | int version_b; ///< Bink version 'b' | ||||
| int first; | int first; | ||||
| int channels; | int channels; | ||||
| @@ -59,10 +57,7 @@ typedef struct { | |||||
| unsigned int *bands; | unsigned int *bands; | ||||
| float root; | float root; | ||||
| DECLARE_ALIGNED(32, FFTSample, coeffs)[BINK_BLOCK_MAX_SIZE]; | DECLARE_ALIGNED(32, FFTSample, coeffs)[BINK_BLOCK_MAX_SIZE]; | ||||
| DECLARE_ALIGNED(16, int16_t, previous)[BINK_BLOCK_MAX_SIZE / 16]; ///< coeffs from previous audio block | |||||
| DECLARE_ALIGNED(16, int16_t, current)[BINK_BLOCK_MAX_SIZE / 16]; | |||||
| float *coeffs_ptr[MAX_CHANNELS]; ///< pointers to the coeffs arrays for float_to_int16_interleave | |||||
| float *prev_ptr[MAX_CHANNELS]; ///< pointers to the overlap points in the coeffs array | |||||
| float previous[MAX_CHANNELS][BINK_BLOCK_MAX_SIZE / 16]; ///< coeffs from previous audio block | |||||
| uint8_t *packet_buffer; | uint8_t *packet_buffer; | ||||
| union { | union { | ||||
| RDFTContext rdft; | RDFTContext rdft; | ||||
| @@ -79,9 +74,6 @@ static av_cold int decode_init(AVCodecContext *avctx) | |||||
| int i; | int i; | ||||
| int frame_len_bits; | int frame_len_bits; | ||||
| ff_dsputil_init(&s->dsp, avctx); | |||||
| ff_fmt_convert_init(&s->fmt_conv, avctx); | |||||
| /* determine frame length */ | /* determine frame length */ | ||||
| if (avctx->sample_rate < 22050) { | if (avctx->sample_rate < 22050) { | ||||
| frame_len_bits = 9; | frame_len_bits = 9; | ||||
| @@ -100,19 +92,24 @@ static av_cold int decode_init(AVCodecContext *avctx) | |||||
| if (avctx->codec->id == AV_CODEC_ID_BINKAUDIO_RDFT) { | if (avctx->codec->id == AV_CODEC_ID_BINKAUDIO_RDFT) { | ||||
| // audio is already interleaved for the RDFT format variant | // audio is already interleaved for the RDFT format variant | ||||
| avctx->sample_fmt = AV_SAMPLE_FMT_FLT; | |||||
| sample_rate *= avctx->channels; | sample_rate *= avctx->channels; | ||||
| s->channels = 1; | s->channels = 1; | ||||
| if (!s->version_b) | if (!s->version_b) | ||||
| frame_len_bits += av_log2(avctx->channels); | frame_len_bits += av_log2(avctx->channels); | ||||
| } else { | } else { | ||||
| s->channels = avctx->channels; | s->channels = avctx->channels; | ||||
| avctx->sample_fmt = AV_SAMPLE_FMT_FLTP; | |||||
| } | } | ||||
| s->frame_len = 1 << frame_len_bits; | s->frame_len = 1 << frame_len_bits; | ||||
| s->overlap_len = s->frame_len / 16; | s->overlap_len = s->frame_len / 16; | ||||
| s->block_size = (s->frame_len - s->overlap_len) * s->channels; | s->block_size = (s->frame_len - s->overlap_len) * s->channels; | ||||
| sample_rate_half = (sample_rate + 1) / 2; | sample_rate_half = (sample_rate + 1) / 2; | ||||
| s->root = 2.0 / sqrt(s->frame_len); | |||||
| if (avctx->codec->id == AV_CODEC_ID_BINKAUDIO_RDFT) | |||||
| s->root = 2.0 / (sqrt(s->frame_len) * 32768.0); | |||||
| else | |||||
| s->root = s->frame_len / (sqrt(s->frame_len) * 32768.0); | |||||
| for (i = 0; i < 96; i++) { | for (i = 0; i < 96; i++) { | ||||
| /* constant is result of 0.066399999/log10(M_E) */ | /* constant is result of 0.066399999/log10(M_E) */ | ||||
| quant_table[i] = expf(i * 0.15289164787221953823f) * s->root; | quant_table[i] = expf(i * 0.15289164787221953823f) * s->root; | ||||
| @@ -134,12 +131,6 @@ static av_cold int decode_init(AVCodecContext *avctx) | |||||
| s->bands[s->num_bands] = s->frame_len; | s->bands[s->num_bands] = s->frame_len; | ||||
| s->first = 1; | s->first = 1; | ||||
| avctx->sample_fmt = AV_SAMPLE_FMT_S16; | |||||
| for (i = 0; i < s->channels; i++) { | |||||
| s->coeffs_ptr[i] = s->coeffs + i * s->frame_len; | |||||
| s->prev_ptr[i] = s->coeffs_ptr[i] + s->frame_len - s->overlap_len; | |||||
| } | |||||
| if (CONFIG_BINKAUDIO_RDFT_DECODER && avctx->codec->id == AV_CODEC_ID_BINKAUDIO_RDFT) | if (CONFIG_BINKAUDIO_RDFT_DECODER && avctx->codec->id == AV_CODEC_ID_BINKAUDIO_RDFT) | ||||
| ff_rdft_init(&s->trans.rdft, frame_len_bits, DFT_C2R); | ff_rdft_init(&s->trans.rdft, frame_len_bits, DFT_C2R); | ||||
| @@ -167,18 +158,12 @@ static const uint8_t rle_length_tab[16] = { | |||||
| 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15, 16, 32, 64 | 2, 3, 4, 5, 6, 8, 9, 10, 11, 12, 13, 14, 15, 16, 32, 64 | ||||
| }; | }; | ||||
| #define GET_BITS_SAFE(out, nbits) do { \ | |||||
| if (get_bits_left(gb) < nbits) \ | |||||
| return AVERROR_INVALIDDATA; \ | |||||
| out = get_bits(gb, nbits); \ | |||||
| } while (0) | |||||
| /** | /** | ||||
| * Decode Bink Audio block | * Decode Bink Audio block | ||||
| * @param[out] out Output buffer (must contain s->block_size elements) | * @param[out] out Output buffer (must contain s->block_size elements) | ||||
| * @return 0 on success, negative error code on failure | * @return 0 on success, negative error code on failure | ||||
| */ | */ | ||||
| static int decode_block(BinkAudioContext *s, int16_t *out, int use_dct) | |||||
| static int decode_block(BinkAudioContext *s, float **out, int use_dct) | |||||
| { | { | ||||
| int ch, i, j, k; | int ch, i, j, k; | ||||
| float q, quant[25]; | float q, quant[25]; | ||||
| @@ -189,7 +174,8 @@ static int decode_block(BinkAudioContext *s, int16_t *out, int use_dct) | |||||
| skip_bits(gb, 2); | skip_bits(gb, 2); | ||||
| for (ch = 0; ch < s->channels; ch++) { | for (ch = 0; ch < s->channels; ch++) { | ||||
| FFTSample *coeffs = s->coeffs_ptr[ch]; | |||||
| FFTSample *coeffs = out[ch]; | |||||
| if (s->version_b) { | if (s->version_b) { | ||||
| if (get_bits_left(gb) < 64) | if (get_bits_left(gb) < 64) | ||||
| return AVERROR_INVALIDDATA; | return AVERROR_INVALIDDATA; | ||||
| @@ -218,10 +204,9 @@ static int decode_block(BinkAudioContext *s, int16_t *out, int use_dct) | |||||
| if (s->version_b) { | if (s->version_b) { | ||||
| j = i + 16; | j = i + 16; | ||||
| } else { | } else { | ||||
| int v; | |||||
| GET_BITS_SAFE(v, 1); | |||||
| int v = get_bits1(gb); | |||||
| if (v) { | if (v) { | ||||
| GET_BITS_SAFE(v, 4); | |||||
| v = get_bits(gb, 4); | |||||
| j = i + rle_length_tab[v] * 8; | j = i + rle_length_tab[v] * 8; | ||||
| } else { | } else { | ||||
| j = i + 8; | j = i + 8; | ||||
| @@ -230,7 +215,7 @@ static int decode_block(BinkAudioContext *s, int16_t *out, int use_dct) | |||||
| j = FFMIN(j, s->frame_len); | j = FFMIN(j, s->frame_len); | ||||
| GET_BITS_SAFE(width, 4); | |||||
| width = get_bits(gb, 4); | |||||
| if (width == 0) { | if (width == 0) { | ||||
| memset(coeffs + i, 0, (j - i) * sizeof(*coeffs)); | memset(coeffs + i, 0, (j - i) * sizeof(*coeffs)); | ||||
| i = j; | i = j; | ||||
| @@ -240,10 +225,10 @@ static int decode_block(BinkAudioContext *s, int16_t *out, int use_dct) | |||||
| while (i < j) { | while (i < j) { | ||||
| if (s->bands[k] == i) | if (s->bands[k] == i) | ||||
| q = quant[k++]; | q = quant[k++]; | ||||
| GET_BITS_SAFE(coeff, width); | |||||
| coeff = get_bits(gb, width); | |||||
| if (coeff) { | if (coeff) { | ||||
| int v; | int v; | ||||
| GET_BITS_SAFE(v, 1); | |||||
| v = get_bits1(gb); | |||||
| if (v) | if (v) | ||||
| coeffs[i] = -q * coeff; | coeffs[i] = -q * coeff; | ||||
| else | else | ||||
| @@ -259,30 +244,24 @@ static int decode_block(BinkAudioContext *s, int16_t *out, int use_dct) | |||||
| if (CONFIG_BINKAUDIO_DCT_DECODER && use_dct) { | if (CONFIG_BINKAUDIO_DCT_DECODER && use_dct) { | ||||
| coeffs[0] /= 0.5; | coeffs[0] /= 0.5; | ||||
| s->trans.dct.dct_calc(&s->trans.dct, coeffs); | s->trans.dct.dct_calc(&s->trans.dct, coeffs); | ||||
| s->dsp.vector_fmul_scalar(coeffs, coeffs, s->frame_len / 2, s->frame_len); | |||||
| } | } | ||||
| else if (CONFIG_BINKAUDIO_RDFT_DECODER) | else if (CONFIG_BINKAUDIO_RDFT_DECODER) | ||||
| s->trans.rdft.rdft_calc(&s->trans.rdft, coeffs); | s->trans.rdft.rdft_calc(&s->trans.rdft, coeffs); | ||||
| } | } | ||||
| s->fmt_conv.float_to_int16_interleave(s->current, | |||||
| (const float **)s->prev_ptr, | |||||
| s->overlap_len, s->channels); | |||||
| s->fmt_conv.float_to_int16_interleave(out, (const float **)s->coeffs_ptr, | |||||
| s->frame_len - s->overlap_len, | |||||
| s->channels); | |||||
| if (!s->first) { | |||||
| for (ch = 0; ch < s->channels; ch++) { | |||||
| int j; | |||||
| int count = s->overlap_len * s->channels; | int count = s->overlap_len * s->channels; | ||||
| int shift = av_log2(count); | |||||
| for (i = 0; i < count; i++) { | |||||
| out[i] = (s->previous[i] * (count - i) + out[i] * i) >> shift; | |||||
| if (!s->first) { | |||||
| j = ch; | |||||
| for (i = 0; i < s->overlap_len; i++, j += s->channels) | |||||
| out[ch][i] = (s->previous[ch][i] * (count - j) + | |||||
| out[ch][i] * j) / count; | |||||
| } | } | ||||
| memcpy(s->previous[ch], &out[ch][s->frame_len - s->overlap_len], | |||||
| s->overlap_len * sizeof(*s->previous[ch])); | |||||
| } | } | ||||
| memcpy(s->previous, s->current, | |||||
| s->overlap_len * s->channels * sizeof(*s->previous)); | |||||
| s->first = 0; | s->first = 0; | ||||
| return 0; | return 0; | ||||
| @@ -311,7 +290,6 @@ static int decode_frame(AVCodecContext *avctx, void *data, | |||||
| int *got_frame_ptr, AVPacket *avpkt) | int *got_frame_ptr, AVPacket *avpkt) | ||||
| { | { | ||||
| BinkAudioContext *s = avctx->priv_data; | BinkAudioContext *s = avctx->priv_data; | ||||
| int16_t *samples; | |||||
| GetBitContext *gb = &s->gb; | GetBitContext *gb = &s->gb; | ||||
| int ret, consumed = 0; | int ret, consumed = 0; | ||||
| @@ -339,19 +317,20 @@ static int decode_frame(AVCodecContext *avctx, void *data, | |||||
| } | } | ||||
| /* get output buffer */ | /* get output buffer */ | ||||
| s->frame.nb_samples = s->block_size / avctx->channels; | |||||
| s->frame.nb_samples = s->frame_len; | |||||
| if ((ret = avctx->get_buffer(avctx, &s->frame)) < 0) { | if ((ret = avctx->get_buffer(avctx, &s->frame)) < 0) { | ||||
| av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); | av_log(avctx, AV_LOG_ERROR, "get_buffer() failed\n"); | ||||
| return ret; | return ret; | ||||
| } | } | ||||
| samples = (int16_t *)s->frame.data[0]; | |||||
| if (decode_block(s, samples, avctx->codec->id == AV_CODEC_ID_BINKAUDIO_DCT)) { | |||||
| if (decode_block(s, (float **)s->frame.extended_data, | |||||
| avctx->codec->id == AV_CODEC_ID_BINKAUDIO_DCT)) { | |||||
| av_log(avctx, AV_LOG_ERROR, "Incomplete packet\n"); | av_log(avctx, AV_LOG_ERROR, "Incomplete packet\n"); | ||||
| return AVERROR_INVALIDDATA; | return AVERROR_INVALIDDATA; | ||||
| } | } | ||||
| get_bits_align32(gb); | get_bits_align32(gb); | ||||
| s->frame.nb_samples = s->block_size / avctx->channels; | |||||
| *got_frame_ptr = 1; | *got_frame_ptr = 1; | ||||
| *(AVFrame *)data = s->frame; | *(AVFrame *)data = s->frame; | ||||
| @@ -656,13 +656,138 @@ void ff_MPV_decode_defaults(MpegEncContext *s) | |||||
| ff_MPV_common_defaults(s); | ff_MPV_common_defaults(s); | ||||
| } | } | ||||
| /** | |||||
| * Initialize and allocates MpegEncContext fields dependent on the resolution. | |||||
| */ | |||||
| static int init_context_frame(MpegEncContext *s) | |||||
| { | |||||
| int y_size, c_size, yc_size, i, mb_array_size, mv_table_size, x, y; | |||||
| s->mb_width = (s->width + 15) / 16; | |||||
| s->mb_stride = s->mb_width + 1; | |||||
| s->b8_stride = s->mb_width * 2 + 1; | |||||
| s->b4_stride = s->mb_width * 4 + 1; | |||||
| mb_array_size = s->mb_height * s->mb_stride; | |||||
| mv_table_size = (s->mb_height + 2) * s->mb_stride + 1; | |||||
| /* set default edge pos, will be overriden | |||||
| * in decode_header if needed */ | |||||
| s->h_edge_pos = s->mb_width * 16; | |||||
| s->v_edge_pos = s->mb_height * 16; | |||||
| s->mb_num = s->mb_width * s->mb_height; | |||||
| s->block_wrap[0] = | |||||
| s->block_wrap[1] = | |||||
| s->block_wrap[2] = | |||||
| s->block_wrap[3] = s->b8_stride; | |||||
| s->block_wrap[4] = | |||||
| s->block_wrap[5] = s->mb_stride; | |||||
| y_size = s->b8_stride * (2 * s->mb_height + 1); | |||||
| c_size = s->mb_stride * (s->mb_height + 1); | |||||
| yc_size = y_size + 2 * c_size; | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->mb_index2xy, (s->mb_num + 1) * sizeof(int), fail); // error ressilience code looks cleaner with this | |||||
| for (y = 0; y < s->mb_height; y++) | |||||
| for (x = 0; x < s->mb_width; x++) | |||||
| s->mb_index2xy[x + y * s->mb_width] = x + y * s->mb_stride; | |||||
| s->mb_index2xy[s->mb_height * s->mb_width] = (s->mb_height - 1) * s->mb_stride + s->mb_width; // FIXME really needed? | |||||
| if (s->encoding) { | |||||
| /* Allocate MV tables */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->p_mv_table_base, mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_forw_mv_table_base, mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_back_mv_table_base, mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_bidir_forw_mv_table_base, mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_bidir_back_mv_table_base, mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_direct_mv_table_base, mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| s->p_mv_table = s->p_mv_table_base + s->mb_stride + 1; | |||||
| s->b_forw_mv_table = s->b_forw_mv_table_base + s->mb_stride + 1; | |||||
| s->b_back_mv_table = s->b_back_mv_table_base + s->mb_stride + 1; | |||||
| s->b_bidir_forw_mv_table = s->b_bidir_forw_mv_table_base + s->mb_stride + 1; | |||||
| s->b_bidir_back_mv_table = s->b_bidir_back_mv_table_base + s->mb_stride + 1; | |||||
| s->b_direct_mv_table = s->b_direct_mv_table_base + s->mb_stride + 1; | |||||
| /* Allocate MB type table */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->mb_type, mb_array_size * sizeof(uint16_t), fail) // needed for encoding | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->lambda_table, mb_array_size * sizeof(int), fail) | |||||
| FF_ALLOC_OR_GOTO(s->avctx, s->cplx_tab, | |||||
| mb_array_size * sizeof(float), fail); | |||||
| FF_ALLOC_OR_GOTO(s->avctx, s->bits_tab, | |||||
| mb_array_size * sizeof(float), fail); | |||||
| } | |||||
| FF_ALLOC_OR_GOTO(s->avctx, s->er_temp_buffer, | |||||
| mb_array_size * sizeof(uint8_t), fail); | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->error_status_table, | |||||
| mb_array_size * sizeof(uint8_t), fail); | |||||
| if (s->codec_id == AV_CODEC_ID_MPEG4 || (s->flags & CODEC_FLAG_INTERLACED_ME)) { | |||||
| /* interlaced direct mode decoding tables */ | |||||
| for (i = 0; i < 2; i++) { | |||||
| int j, k; | |||||
| for (j = 0; j < 2; j++) { | |||||
| for (k = 0; k < 2; k++) { | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, | |||||
| s->b_field_mv_table_base[i][j][k], | |||||
| mv_table_size * 2 * sizeof(int16_t), | |||||
| fail); | |||||
| s->b_field_mv_table[i][j][k] = s->b_field_mv_table_base[i][j][k] + | |||||
| s->mb_stride + 1; | |||||
| } | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_field_select_table [i][j], mb_array_size * 2 * sizeof(uint8_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->p_field_mv_table_base[i][j], mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| s->p_field_mv_table[i][j] = s->p_field_mv_table_base[i][j] + s->mb_stride + 1; | |||||
| } | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->p_field_select_table[i], mb_array_size * 2 * sizeof(uint8_t), fail) | |||||
| } | |||||
| } | |||||
| if (s->out_format == FMT_H263) { | |||||
| /* cbp values */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->coded_block_base, y_size, fail); | |||||
| s->coded_block = s->coded_block_base + s->b8_stride + 1; | |||||
| /* cbp, ac_pred, pred_dir */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->cbp_table , mb_array_size * sizeof(uint8_t), fail); | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->pred_dir_table, mb_array_size * sizeof(uint8_t), fail); | |||||
| } | |||||
| if (s->h263_pred || s->h263_plus || !s->encoding) { | |||||
| /* dc values */ | |||||
| // MN: we need these for error resilience of intra-frames | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->dc_val_base, yc_size * sizeof(int16_t), fail); | |||||
| s->dc_val[0] = s->dc_val_base + s->b8_stride + 1; | |||||
| s->dc_val[1] = s->dc_val_base + y_size + s->mb_stride + 1; | |||||
| s->dc_val[2] = s->dc_val[1] + c_size; | |||||
| for (i = 0; i < yc_size; i++) | |||||
| s->dc_val_base[i] = 1024; | |||||
| } | |||||
| /* which mb is a intra block */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->mbintra_table, mb_array_size, fail); | |||||
| memset(s->mbintra_table, 1, mb_array_size); | |||||
| /* init macroblock skip table */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->mbskip_table, mb_array_size + 2, fail); | |||||
| // Note the + 1 is for a quicker mpeg4 slice_end detection | |||||
| return 0; | |||||
| fail: | |||||
| return AVERROR(ENOMEM); | |||||
| } | |||||
| /** | /** | ||||
| * init common structure for both encoder and decoder. | * init common structure for both encoder and decoder. | ||||
| * this assumes that some variables like width/height are already set | * this assumes that some variables like width/height are already set | ||||
| */ | */ | ||||
| av_cold int ff_MPV_common_init(MpegEncContext *s) | av_cold int ff_MPV_common_init(MpegEncContext *s) | ||||
| { | { | ||||
| int y_size, c_size, yc_size, i, mb_array_size, mv_table_size, x, y; | |||||
| int i, err; | |||||
| int nb_slices = (HAVE_THREADS && | int nb_slices = (HAVE_THREADS && | ||||
| s->avctx->active_thread_type & FF_THREAD_SLICE) ? | s->avctx->active_thread_type & FF_THREAD_SLICE) ? | ||||
| s->avctx->thread_count : 1; | s->avctx->thread_count : 1; | ||||
| @@ -701,89 +826,36 @@ av_cold int ff_MPV_common_init(MpegEncContext *s) | |||||
| s->flags = s->avctx->flags; | s->flags = s->avctx->flags; | ||||
| s->flags2 = s->avctx->flags2; | s->flags2 = s->avctx->flags2; | ||||
| s->mb_width = (s->width + 15) / 16; | |||||
| s->mb_stride = s->mb_width + 1; | |||||
| s->b8_stride = s->mb_width * 2 + 1; | |||||
| s->b4_stride = s->mb_width * 4 + 1; | |||||
| mb_array_size = s->mb_height * s->mb_stride; | |||||
| mv_table_size = (s->mb_height + 2) * s->mb_stride + 1; | |||||
| /* set chroma shifts */ | /* set chroma shifts */ | ||||
| avcodec_get_chroma_sub_sample(s->avctx->pix_fmt, &s->chroma_x_shift, | avcodec_get_chroma_sub_sample(s->avctx->pix_fmt, &s->chroma_x_shift, | ||||
| &s->chroma_y_shift); | &s->chroma_y_shift); | ||||
| /* set default edge pos, will be overridden in decode_header if needed */ | |||||
| s->h_edge_pos = s->mb_width * 16; | |||||
| s->v_edge_pos = s->mb_height * 16; | |||||
| s->mb_num = s->mb_width * s->mb_height; | |||||
| s->block_wrap[0] = | |||||
| s->block_wrap[1] = | |||||
| s->block_wrap[2] = | |||||
| s->block_wrap[3] = s->b8_stride; | |||||
| s->block_wrap[4] = | |||||
| s->block_wrap[5] = s->mb_stride; | |||||
| y_size = s->b8_stride * (2 * s->mb_height + 1); | |||||
| c_size = s->mb_stride * (s->mb_height + 1); | |||||
| yc_size = y_size + 2 * c_size; | |||||
| /* convert fourcc to upper case */ | /* convert fourcc to upper case */ | ||||
| s->codec_tag = avpriv_toupper4(s->avctx->codec_tag); | s->codec_tag = avpriv_toupper4(s->avctx->codec_tag); | ||||
| s->stream_codec_tag = avpriv_toupper4(s->avctx->stream_codec_tag); | s->stream_codec_tag = avpriv_toupper4(s->avctx->stream_codec_tag); | ||||
| s->avctx->coded_frame = &s->current_picture.f; | s->avctx->coded_frame = &s->current_picture.f; | ||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->mb_index2xy, (s->mb_num + 1) * sizeof(int), fail); // error ressilience code looks cleaner with this | |||||
| for (y = 0; y < s->mb_height; y++) | |||||
| for (x = 0; x < s->mb_width; x++) | |||||
| s->mb_index2xy[x + y * s->mb_width] = x + y * s->mb_stride; | |||||
| s->mb_index2xy[s->mb_height * s->mb_width] = (s->mb_height - 1) * s->mb_stride + s->mb_width; // FIXME really needed? | |||||
| if (s->encoding) { | if (s->encoding) { | ||||
| /* Allocate MV tables */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->p_mv_table_base , mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_forw_mv_table_base , mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_back_mv_table_base , mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_bidir_forw_mv_table_base , mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_bidir_back_mv_table_base , mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_direct_mv_table_base , mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| s->p_mv_table = s->p_mv_table_base + s->mb_stride + 1; | |||||
| s->b_forw_mv_table = s->b_forw_mv_table_base + s->mb_stride + 1; | |||||
| s->b_back_mv_table = s->b_back_mv_table_base + s->mb_stride + 1; | |||||
| s->b_bidir_forw_mv_table= s->b_bidir_forw_mv_table_base + s->mb_stride + 1; | |||||
| s->b_bidir_back_mv_table= s->b_bidir_back_mv_table_base + s->mb_stride + 1; | |||||
| s->b_direct_mv_table = s->b_direct_mv_table_base + s->mb_stride + 1; | |||||
| if(s->msmpeg4_version){ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->ac_stats, 2*2*(MAX_LEVEL+1)*(MAX_RUN+1)*2*sizeof(int), fail); | |||||
| if (s->msmpeg4_version) { | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->ac_stats, | |||||
| 2 * 2 * (MAX_LEVEL + 1) * | |||||
| (MAX_RUN + 1) * 2 * sizeof(int), fail); | |||||
| } | } | ||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->avctx->stats_out, 256, fail); | FF_ALLOCZ_OR_GOTO(s->avctx, s->avctx->stats_out, 256, fail); | ||||
| /* Allocate MB type table */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->mb_type , mb_array_size * sizeof(uint16_t), fail) //needed for encoding | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->lambda_table, mb_array_size * sizeof(int), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_intra_matrix , 64*32 * sizeof(int), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_chroma_intra_matrix , 64*32 * sizeof(int), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_inter_matrix , 64*32 * sizeof(int), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_intra_matrix16 , 64*32*2 * sizeof(uint16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_chroma_intra_matrix16, 64*32*2 * sizeof(uint16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_inter_matrix16 , 64*32*2 * sizeof(uint16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->input_picture, MAX_PICTURE_COUNT * sizeof(Picture*), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->reordered_input_picture, MAX_PICTURE_COUNT * sizeof(Picture*), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_intra_matrix, 64 * 32 * sizeof(int), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_chroma_intra_matrix, 64 * 32 * sizeof(int), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_inter_matrix, 64 * 32 * sizeof(int), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_intra_matrix16, 64 * 32 * 2 * sizeof(uint16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_chroma_intra_matrix16, 64 * 32 * 2 * sizeof(uint16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->q_inter_matrix16, 64 * 32 * 2 * sizeof(uint16_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->input_picture, MAX_PICTURE_COUNT * sizeof(Picture *), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->reordered_input_picture, MAX_PICTURE_COUNT * sizeof(Picture *), fail) | |||||
| if(s->avctx->noise_reduction){ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->dct_offset, 2 * 64 * sizeof(uint16_t), fail) | |||||
| if (s->avctx->noise_reduction) { | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->dct_offset, 2 * 64 * sizeof(uint16_t), fail); | |||||
| } | } | ||||
| FF_ALLOC_OR_GOTO(s->avctx, s->cplx_tab, | |||||
| mb_array_size * sizeof(float), fail); | |||||
| FF_ALLOC_OR_GOTO(s->avctx, s->bits_tab, | |||||
| mb_array_size * sizeof(float), fail); | |||||
| } | } | ||||
| s->picture_count = MAX_PICTURE_COUNT * FFMAX(1, s->avctx->thread_count); | s->picture_count = MAX_PICTURE_COUNT * FFMAX(1, s->avctx->thread_count); | ||||
| @@ -793,55 +865,8 @@ av_cold int ff_MPV_common_init(MpegEncContext *s) | |||||
| avcodec_get_frame_defaults(&s->picture[i].f); | avcodec_get_frame_defaults(&s->picture[i].f); | ||||
| } | } | ||||
| FF_ALLOC_OR_GOTO(s->avctx, s->er_temp_buffer, | |||||
| mb_array_size * sizeof(uint8_t), fail); | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->error_status_table, | |||||
| mb_array_size * sizeof(uint8_t), fail); | |||||
| if(s->codec_id==AV_CODEC_ID_MPEG4 || (s->flags & CODEC_FLAG_INTERLACED_ME)){ | |||||
| /* interlaced direct mode decoding tables */ | |||||
| for (i = 0; i < 2; i++) { | |||||
| int j, k; | |||||
| for (j = 0; j < 2; j++) { | |||||
| for (k = 0; k < 2; k++) { | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_field_mv_table_base[i][j][k], mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| s->b_field_mv_table[i][j][k] = s->b_field_mv_table_base[i][j][k] + s->mb_stride + 1; | |||||
| } | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->b_field_select_table [i][j], mb_array_size * 2 * sizeof(uint8_t), fail) | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->p_field_mv_table_base[i][j], mv_table_size * 2 * sizeof(int16_t), fail) | |||||
| s->p_field_mv_table[i][j] = s->p_field_mv_table_base[i][j] + s->mb_stride + 1; | |||||
| } | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->p_field_select_table[i], mb_array_size * 2 * sizeof(uint8_t), fail) | |||||
| } | |||||
| } | |||||
| if (s->out_format == FMT_H263) { | |||||
| /* cbp values */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->coded_block_base, y_size, fail); | |||||
| s->coded_block = s->coded_block_base + s->b8_stride + 1; | |||||
| /* cbp, ac_pred, pred_dir */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->cbp_table , mb_array_size * sizeof(uint8_t), fail); | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->pred_dir_table, mb_array_size * sizeof(uint8_t), fail); | |||||
| } | |||||
| if (s->h263_pred || s->h263_plus || !s->encoding) { | |||||
| /* dc values */ | |||||
| // MN: we need these for error resilience of intra-frames | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->dc_val_base, yc_size * sizeof(int16_t), fail); | |||||
| s->dc_val[0] = s->dc_val_base + s->b8_stride + 1; | |||||
| s->dc_val[1] = s->dc_val_base + y_size + s->mb_stride + 1; | |||||
| s->dc_val[2] = s->dc_val[1] + c_size; | |||||
| for (i = 0; i < yc_size; i++) | |||||
| s->dc_val_base[i] = 1024; | |||||
| } | |||||
| /* which mb is a intra block */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->mbintra_table, mb_array_size, fail); | |||||
| memset(s->mbintra_table, 1, mb_array_size); | |||||
| /* init macroblock skip table */ | |||||
| FF_ALLOCZ_OR_GOTO(s->avctx, s->mbskip_table, mb_array_size + 2, fail); | |||||
| // Note the + 1 is for a quicker mpeg4 slice_end detection | |||||
| if ((err = init_context_frame(s))) | |||||
| goto fail; | |||||
| s->parse_context.state = -1; | s->parse_context.state = -1; | ||||
| @@ -878,24 +903,15 @@ av_cold int ff_MPV_common_init(MpegEncContext *s) | |||||
| return -1; | return -1; | ||||
| } | } | ||||
| /* init common structure for both encoder and decoder */ | |||||
| void ff_MPV_common_end(MpegEncContext *s) | |||||
| /** | |||||
| * Frees and resets MpegEncContext fields depending on the resolution. | |||||
| * Is used during resolution changes to avoid a full reinitialization of the | |||||
| * codec. | |||||
| */ | |||||
| static int free_context_frame(MpegEncContext *s) | |||||
| { | { | ||||
| int i, j, k; | int i, j, k; | ||||
| if (s->slice_context_count > 1) { | |||||
| for (i = 0; i < s->slice_context_count; i++) { | |||||
| free_duplicate_context(s->thread_context[i]); | |||||
| } | |||||
| for (i = 1; i < s->slice_context_count; i++) { | |||||
| av_freep(&s->thread_context[i]); | |||||
| } | |||||
| s->slice_context_count = 1; | |||||
| } else free_duplicate_context(s); | |||||
| av_freep(&s->parse_context.buffer); | |||||
| s->parse_context.buffer_size = 0; | |||||
| av_freep(&s->mb_type); | av_freep(&s->mb_type); | ||||
| av_freep(&s->p_mv_table_base); | av_freep(&s->p_mv_table_base); | ||||
| av_freep(&s->b_forw_mv_table_base); | av_freep(&s->b_forw_mv_table_base); | ||||
| @@ -929,15 +945,50 @@ void ff_MPV_common_end(MpegEncContext *s) | |||||
| av_freep(&s->pred_dir_table); | av_freep(&s->pred_dir_table); | ||||
| av_freep(&s->mbskip_table); | av_freep(&s->mbskip_table); | ||||
| av_freep(&s->bitstream_buffer); | |||||
| s->allocated_bitstream_buffer_size = 0; | |||||
| av_freep(&s->avctx->stats_out); | |||||
| av_freep(&s->ac_stats); | |||||
| av_freep(&s->error_status_table); | av_freep(&s->error_status_table); | ||||
| av_freep(&s->er_temp_buffer); | av_freep(&s->er_temp_buffer); | ||||
| av_freep(&s->mb_index2xy); | av_freep(&s->mb_index2xy); | ||||
| av_freep(&s->lambda_table); | av_freep(&s->lambda_table); | ||||
| av_freep(&s->cplx_tab); | |||||
| av_freep(&s->bits_tab); | |||||
| s->linesize = s->uvlinesize = 0; | |||||
| for (i = 0; i < 3; i++) | |||||
| av_freep(&s->visualization_buffer[i]); | |||||
| if (!(s->avctx->active_thread_type & FF_THREAD_FRAME)) | |||||
| avcodec_default_free_buffers(s->avctx); | |||||
| return 0; | |||||
| } | |||||
| /* init common structure for both encoder and decoder */ | |||||
| void ff_MPV_common_end(MpegEncContext *s) | |||||
| { | |||||
| int i; | |||||
| if (s->slice_context_count > 1) { | |||||
| for (i = 0; i < s->slice_context_count; i++) { | |||||
| free_duplicate_context(s->thread_context[i]); | |||||
| } | |||||
| for (i = 1; i < s->slice_context_count; i++) { | |||||
| av_freep(&s->thread_context[i]); | |||||
| } | |||||
| s->slice_context_count = 1; | |||||
| } else free_duplicate_context(s); | |||||
| av_freep(&s->parse_context.buffer); | |||||
| s->parse_context.buffer_size = 0; | |||||
| av_freep(&s->bitstream_buffer); | |||||
| s->allocated_bitstream_buffer_size = 0; | |||||
| av_freep(&s->avctx->stats_out); | |||||
| av_freep(&s->ac_stats); | |||||
| if(s->q_chroma_intra_matrix != s->q_intra_matrix ) av_freep(&s->q_chroma_intra_matrix); | if(s->q_chroma_intra_matrix != s->q_intra_matrix ) av_freep(&s->q_chroma_intra_matrix); | ||||
| if(s->q_chroma_intra_matrix16 != s->q_intra_matrix16) av_freep(&s->q_chroma_intra_matrix16); | if(s->q_chroma_intra_matrix16 != s->q_intra_matrix16) av_freep(&s->q_chroma_intra_matrix16); | ||||
| s->q_chroma_intra_matrix= NULL; | s->q_chroma_intra_matrix= NULL; | ||||
| @@ -949,8 +1000,6 @@ void ff_MPV_common_end(MpegEncContext *s) | |||||
| av_freep(&s->input_picture); | av_freep(&s->input_picture); | ||||
| av_freep(&s->reordered_input_picture); | av_freep(&s->reordered_input_picture); | ||||
| av_freep(&s->dct_offset); | av_freep(&s->dct_offset); | ||||
| av_freep(&s->cplx_tab); | |||||
| av_freep(&s->bits_tab); | |||||
| if (s->picture && !s->avctx->internal->is_copy) { | if (s->picture && !s->avctx->internal->is_copy) { | ||||
| for (i = 0; i < s->picture_count; i++) { | for (i = 0; i < s->picture_count; i++) { | ||||
| @@ -958,17 +1007,14 @@ void ff_MPV_common_end(MpegEncContext *s) | |||||
| } | } | ||||
| } | } | ||||
| av_freep(&s->picture); | av_freep(&s->picture); | ||||
| free_context_frame(s); | |||||
| s->context_initialized = 0; | s->context_initialized = 0; | ||||
| s->last_picture_ptr = | s->last_picture_ptr = | ||||
| s->next_picture_ptr = | s->next_picture_ptr = | ||||
| s->current_picture_ptr = NULL; | s->current_picture_ptr = NULL; | ||||
| s->linesize = s->uvlinesize = 0; | s->linesize = s->uvlinesize = 0; | ||||
| for (i = 0; i < 3; i++) | |||||
| av_freep(&s->visualization_buffer[i]); | |||||
| if (!(s->avctx->active_thread_type & FF_THREAD_FRAME)) | |||||
| avcodec_default_free_buffers(s->avctx); | |||||
| } | } | ||||
| void ff_init_rl(RLTable *rl, | void ff_init_rl(RLTable *rl, | ||||
| @@ -318,7 +318,7 @@ OBJS-$(CONFIG_SIFF_DEMUXER) += siff.o | |||||
| OBJS-$(CONFIG_SMACKER_DEMUXER) += smacker.o | OBJS-$(CONFIG_SMACKER_DEMUXER) += smacker.o | ||||
| OBJS-$(CONFIG_SMJPEG_DEMUXER) += smjpegdec.o smjpeg.o | OBJS-$(CONFIG_SMJPEG_DEMUXER) += smjpegdec.o smjpeg.o | ||||
| OBJS-$(CONFIG_SMJPEG_MUXER) += smjpegenc.o smjpeg.o | OBJS-$(CONFIG_SMJPEG_MUXER) += smjpegenc.o smjpeg.o | ||||
| OBJS-$(CONFIG_SMOOTHSTREAMING_MUXER) += smoothstreamingenc.o | |||||
| OBJS-$(CONFIG_SMOOTHSTREAMING_MUXER) += smoothstreamingenc.o isom.o | |||||
| OBJS-$(CONFIG_SMUSH_DEMUXER) += smush.o | OBJS-$(CONFIG_SMUSH_DEMUXER) += smush.o | ||||
| OBJS-$(CONFIG_SOL_DEMUXER) += sol.o pcm.o | OBJS-$(CONFIG_SOL_DEMUXER) += sol.o pcm.o | ||||
| OBJS-$(CONFIG_SOX_DEMUXER) += soxdec.o pcm.o | OBJS-$(CONFIG_SOX_DEMUXER) += soxdec.o pcm.o | ||||
| @@ -1141,14 +1141,19 @@ static int mov_read_stco(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| sc->chunk_count = entries; | sc->chunk_count = entries; | ||||
| if (atom.type == MKTAG('s','t','c','o')) | if (atom.type == MKTAG('s','t','c','o')) | ||||
| for (i=0; i<entries; i++) | |||||
| for (i = 0; i < entries && !pb->eof_reached; i++) | |||||
| sc->chunk_offsets[i] = avio_rb32(pb); | sc->chunk_offsets[i] = avio_rb32(pb); | ||||
| else if (atom.type == MKTAG('c','o','6','4')) | else if (atom.type == MKTAG('c','o','6','4')) | ||||
| for (i=0; i<entries; i++) | |||||
| for (i = 0; i < entries && !pb->eof_reached; i++) | |||||
| sc->chunk_offsets[i] = avio_rb64(pb); | sc->chunk_offsets[i] = avio_rb64(pb); | ||||
| else | else | ||||
| return AVERROR_INVALIDDATA; | return AVERROR_INVALIDDATA; | ||||
| sc->chunk_count = i; | |||||
| if (pb->eof_reached) | |||||
| return AVERROR_EOF; | |||||
| return 0; | return 0; | ||||
| } | } | ||||
| @@ -1198,7 +1203,9 @@ int ff_mov_read_stsd_entries(MOVContext *c, AVIOContext *pb, int entries) | |||||
| st = c->fc->streams[c->fc->nb_streams-1]; | st = c->fc->streams[c->fc->nb_streams-1]; | ||||
| sc = st->priv_data; | sc = st->priv_data; | ||||
| for (pseudo_stream_id=0; pseudo_stream_id<entries; pseudo_stream_id++) { | |||||
| for (pseudo_stream_id = 0; | |||||
| pseudo_stream_id < entries && !pb->eof_reached; | |||||
| pseudo_stream_id++) { | |||||
| //Parsing Sample description table | //Parsing Sample description table | ||||
| enum AVCodecID id; | enum AVCodecID id; | ||||
| int dref_id = 1; | int dref_id = 1; | ||||
| @@ -1483,6 +1490,9 @@ int ff_mov_read_stsd_entries(MOVContext *c, AVIOContext *pb, int entries) | |||||
| avio_skip(pb, a.size); | avio_skip(pb, a.size); | ||||
| } | } | ||||
| if (pb->eof_reached) | |||||
| return AVERROR_EOF; | |||||
| if (st->codec->codec_type==AVMEDIA_TYPE_AUDIO && st->codec->sample_rate==0 && sc->time_scale>1) | if (st->codec->codec_type==AVMEDIA_TYPE_AUDIO && st->codec->sample_rate==0 && sc->time_scale>1) | ||||
| st->codec->sample_rate= sc->time_scale; | st->codec->sample_rate= sc->time_scale; | ||||
| @@ -1585,13 +1595,18 @@ static int mov_read_stsc(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| sc->stsc_data = av_malloc(entries * sizeof(*sc->stsc_data)); | sc->stsc_data = av_malloc(entries * sizeof(*sc->stsc_data)); | ||||
| if (!sc->stsc_data) | if (!sc->stsc_data) | ||||
| return AVERROR(ENOMEM); | return AVERROR(ENOMEM); | ||||
| sc->stsc_count = entries; | |||||
| for (i=0; i<entries; i++) { | |||||
| for (i = 0; i < entries && !pb->eof_reached; i++) { | |||||
| sc->stsc_data[i].first = avio_rb32(pb); | sc->stsc_data[i].first = avio_rb32(pb); | ||||
| sc->stsc_data[i].count = avio_rb32(pb); | sc->stsc_data[i].count = avio_rb32(pb); | ||||
| sc->stsc_data[i].id = avio_rb32(pb); | sc->stsc_data[i].id = avio_rb32(pb); | ||||
| } | } | ||||
| sc->stsc_count = i; | |||||
| if (pb->eof_reached) | |||||
| return AVERROR_EOF; | |||||
| return 0; | return 0; | ||||
| } | } | ||||
| @@ -1614,13 +1629,17 @@ static int mov_read_stps(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| sc->stps_data = av_malloc(entries * sizeof(*sc->stps_data)); | sc->stps_data = av_malloc(entries * sizeof(*sc->stps_data)); | ||||
| if (!sc->stps_data) | if (!sc->stps_data) | ||||
| return AVERROR(ENOMEM); | return AVERROR(ENOMEM); | ||||
| sc->stps_count = entries; | |||||
| for (i = 0; i < entries; i++) { | |||||
| for (i = 0; i < entries && !pb->eof_reached; i++) { | |||||
| sc->stps_data[i] = avio_rb32(pb); | sc->stps_data[i] = avio_rb32(pb); | ||||
| //av_dlog(c->fc, "stps %d\n", sc->stps_data[i]); | //av_dlog(c->fc, "stps %d\n", sc->stps_data[i]); | ||||
| } | } | ||||
| sc->stps_count = i; | |||||
| if (pb->eof_reached) | |||||
| return AVERROR_EOF; | |||||
| return 0; | return 0; | ||||
| } | } | ||||
| @@ -1652,12 +1671,17 @@ static int mov_read_stss(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| sc->keyframes = av_malloc(entries * sizeof(int)); | sc->keyframes = av_malloc(entries * sizeof(int)); | ||||
| if (!sc->keyframes) | if (!sc->keyframes) | ||||
| return AVERROR(ENOMEM); | return AVERROR(ENOMEM); | ||||
| sc->keyframe_count = entries; | |||||
| for (i=0; i<entries; i++) { | |||||
| for (i = 0; i < entries && !pb->eof_reached; i++) { | |||||
| sc->keyframes[i] = avio_rb32(pb); | sc->keyframes[i] = avio_rb32(pb); | ||||
| //av_dlog(c->fc, "keyframes[]=%d\n", sc->keyframes[i]); | //av_dlog(c->fc, "keyframes[]=%d\n", sc->keyframes[i]); | ||||
| } | } | ||||
| sc->keyframe_count = i; | |||||
| if (pb->eof_reached) | |||||
| return AVERROR_EOF; | |||||
| return 0; | return 0; | ||||
| } | } | ||||
| @@ -1725,11 +1749,16 @@ static int mov_read_stsz(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| init_get_bits(&gb, buf, 8*num_bytes); | init_get_bits(&gb, buf, 8*num_bytes); | ||||
| for (i = 0; i < entries; i++) { | |||||
| for (i = 0; i < entries && !pb->eof_reached; i++) { | |||||
| sc->sample_sizes[i] = get_bits_long(&gb, field_size); | sc->sample_sizes[i] = get_bits_long(&gb, field_size); | ||||
| sc->data_size += sc->sample_sizes[i]; | sc->data_size += sc->sample_sizes[i]; | ||||
| } | } | ||||
| sc->sample_count = i; | |||||
| if (pb->eof_reached) | |||||
| return AVERROR_EOF; | |||||
| av_free(buf); | av_free(buf); | ||||
| return 0; | return 0; | ||||
| } | } | ||||
| @@ -1761,9 +1790,7 @@ static int mov_read_stts(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| if (!sc->stts_data) | if (!sc->stts_data) | ||||
| return AVERROR(ENOMEM); | return AVERROR(ENOMEM); | ||||
| sc->stts_count = entries; | |||||
| for (i=0; i<entries; i++) { | |||||
| for (i = 0; i < entries && !pb->eof_reached; i++) { | |||||
| int sample_duration; | int sample_duration; | ||||
| int sample_count; | int sample_count; | ||||
| @@ -1784,6 +1811,11 @@ static int mov_read_stts(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| total_sample_count+=sample_count; | total_sample_count+=sample_count; | ||||
| } | } | ||||
| sc->stts_count = i; | |||||
| if (pb->eof_reached) | |||||
| return AVERROR_EOF; | |||||
| st->nb_frames= total_sample_count; | st->nb_frames= total_sample_count; | ||||
| if (duration) | if (duration) | ||||
| st->duration= duration; | st->duration= duration; | ||||
| @@ -1815,9 +1847,8 @@ static int mov_read_ctts(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| sc->ctts_data = av_malloc(entries * sizeof(*sc->ctts_data)); | sc->ctts_data = av_malloc(entries * sizeof(*sc->ctts_data)); | ||||
| if (!sc->ctts_data) | if (!sc->ctts_data) | ||||
| return AVERROR(ENOMEM); | return AVERROR(ENOMEM); | ||||
| sc->ctts_count = entries; | |||||
| for (i=0; i<entries; i++) { | |||||
| for (i = 0; i < entries && !pb->eof_reached; i++) { | |||||
| int count =avio_rb32(pb); | int count =avio_rb32(pb); | ||||
| int duration =avio_rb32(pb); | int duration =avio_rb32(pb); | ||||
| @@ -1838,6 +1869,11 @@ static int mov_read_ctts(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| sc->dts_shift = FFMAX(sc->dts_shift, -duration); | sc->dts_shift = FFMAX(sc->dts_shift, -duration); | ||||
| } | } | ||||
| sc->ctts_count = i; | |||||
| if (pb->eof_reached) | |||||
| return AVERROR_EOF; | |||||
| av_dlog(c->fc, "dts shift %d\n", sc->dts_shift); | av_dlog(c->fc, "dts shift %d\n", sc->dts_shift); | ||||
| return 0; | return 0; | ||||
| @@ -2420,7 +2456,7 @@ static int mov_read_trun(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| offset = frag->base_data_offset + data_offset; | offset = frag->base_data_offset + data_offset; | ||||
| distance = 0; | distance = 0; | ||||
| av_dlog(c->fc, "first sample flags 0x%x\n", first_sample_flags); | av_dlog(c->fc, "first sample flags 0x%x\n", first_sample_flags); | ||||
| for (i = 0; i < entries; i++) { | |||||
| for (i = 0; i < entries && !pb->eof_reached; i++) { | |||||
| unsigned sample_size = frag->size; | unsigned sample_size = frag->size; | ||||
| int sample_flags = i ? frag->flags : first_sample_flags; | int sample_flags = i ? frag->flags : first_sample_flags; | ||||
| unsigned sample_duration = frag->duration; | unsigned sample_duration = frag->duration; | ||||
| @@ -2451,6 +2487,10 @@ static int mov_read_trun(MOVContext *c, AVIOContext *pb, MOVAtom atom) | |||||
| offset += sample_size; | offset += sample_size; | ||||
| sc->data_size += sample_size; | sc->data_size += sample_size; | ||||
| } | } | ||||
| if (pb->eof_reached) | |||||
| return AVERROR_EOF; | |||||
| frag->moof_offset = offset; | frag->moof_offset = offset; | ||||
| st->duration = sc->track_end = dts + sc->time_offset; | st->duration = sc->track_end = dts + sc->time_offset; | ||||
| return 0; | return 0; | ||||
| @@ -30,6 +30,7 @@ | |||||
| #include "os_support.h" | #include "os_support.h" | ||||
| #include "avc.h" | #include "avc.h" | ||||
| #include "url.h" | #include "url.h" | ||||
| #include "isom.h" | |||||
| #include "libavutil/opt.h" | #include "libavutil/opt.h" | ||||
| #include "libavutil/avstring.h" | #include "libavutil/avstring.h" | ||||
| @@ -617,5 +618,6 @@ AVOutputFormat ff_smoothstreaming_muxer = { | |||||
| .write_header = ism_write_header, | .write_header = ism_write_header, | ||||
| .write_packet = ism_write_packet, | .write_packet = ism_write_packet, | ||||
| .write_trailer = ism_write_trailer, | .write_trailer = ism_write_trailer, | ||||
| .codec_tag = (const AVCodecTag* const []){ ff_mp4_obj_type, 0 }, | |||||
| .priv_class = &ism_class, | .priv_class = &ism_class, | ||||
| }; | }; | ||||