blob: e9291aba192c736232107e33d21e0c433cbe719f
| 1 | /* |
| 2 | * copyright (c) 2006 Michael Niedermayer <michaelni@gmx.at> |
| 3 | * |
| 4 | * This file is part of FFmpeg. |
| 5 | * |
| 6 | * FFmpeg is free software; you can redistribute it and/or |
| 7 | * modify it under the terms of the GNU Lesser General Public |
| 8 | * License as published by the Free Software Foundation; either |
| 9 | * version 2.1 of the License, or (at your option) any later version. |
| 10 | * |
| 11 | * FFmpeg is distributed in the hope that it will be useful, |
| 12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of |
| 13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
| 14 | * Lesser General Public License for more details. |
| 15 | * |
| 16 | * You should have received a copy of the GNU Lesser General Public |
| 17 | * License along with FFmpeg; if not, write to the Free Software |
| 18 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
| 19 | */ |
| 20 | |
| 21 | #include <string.h> |
| 22 | |
| 23 | #include "avcodec.h" |
| 24 | #include "libavutil/internal.h" |
| 25 | #include "libavutil/mem.h" |
| 26 | #include "libavutil/opt.h" |
| 27 | |
| 28 | #if FF_API_OLD_BSF |
| 29 | FF_DISABLE_DEPRECATION_WARNINGS |
| 30 | |
| 31 | AVBitStreamFilter *av_bitstream_filter_next(const AVBitStreamFilter *f) |
| 32 | { |
| 33 | const AVBitStreamFilter *filter = NULL; |
| 34 | void *opaque = NULL; |
| 35 | |
| 36 | while (filter != f) |
| 37 | filter = av_bsf_next(&opaque); |
| 38 | |
| 39 | return av_bsf_next(&opaque); |
| 40 | } |
| 41 | |
| 42 | void av_register_bitstream_filter(AVBitStreamFilter *bsf) |
| 43 | { |
| 44 | } |
| 45 | |
| 46 | typedef struct BSFCompatContext { |
| 47 | AVBSFContext *ctx; |
| 48 | int extradata_updated; |
| 49 | } BSFCompatContext; |
| 50 | |
| 51 | AVBitStreamFilterContext *av_bitstream_filter_init(const char *name) |
| 52 | { |
| 53 | AVBitStreamFilterContext *ctx = NULL; |
| 54 | BSFCompatContext *priv = NULL; |
| 55 | const AVBitStreamFilter *bsf; |
| 56 | |
| 57 | bsf = av_bsf_get_by_name(name); |
| 58 | if (!bsf) |
| 59 | return NULL; |
| 60 | |
| 61 | ctx = av_mallocz(sizeof(*ctx)); |
| 62 | if (!ctx) |
| 63 | return NULL; |
| 64 | |
| 65 | priv = av_mallocz(sizeof(*priv)); |
| 66 | if (!priv) |
| 67 | goto fail; |
| 68 | |
| 69 | |
| 70 | ctx->filter = bsf; |
| 71 | ctx->priv_data = priv; |
| 72 | |
| 73 | return ctx; |
| 74 | |
| 75 | fail: |
| 76 | if (priv) |
| 77 | av_bsf_free(&priv->ctx); |
| 78 | av_freep(&priv); |
| 79 | av_freep(&ctx); |
| 80 | return NULL; |
| 81 | } |
| 82 | |
| 83 | void av_bitstream_filter_close(AVBitStreamFilterContext *bsfc) |
| 84 | { |
| 85 | BSFCompatContext *priv; |
| 86 | |
| 87 | if (!bsfc) |
| 88 | return; |
| 89 | |
| 90 | priv = bsfc->priv_data; |
| 91 | |
| 92 | av_bsf_free(&priv->ctx); |
| 93 | av_freep(&bsfc->priv_data); |
| 94 | av_free(bsfc); |
| 95 | } |
| 96 | |
| 97 | int av_bitstream_filter_filter(AVBitStreamFilterContext *bsfc, |
| 98 | AVCodecContext *avctx, const char *args, |
| 99 | uint8_t **poutbuf, int *poutbuf_size, |
| 100 | const uint8_t *buf, int buf_size, int keyframe) |
| 101 | { |
| 102 | BSFCompatContext *priv = bsfc->priv_data; |
| 103 | AVPacket pkt = { 0 }; |
| 104 | int ret; |
| 105 | |
| 106 | if (!priv->ctx) { |
| 107 | ret = av_bsf_alloc(bsfc->filter, &priv->ctx); |
| 108 | if (ret < 0) |
| 109 | return ret; |
| 110 | |
| 111 | ret = avcodec_parameters_from_context(priv->ctx->par_in, avctx); |
| 112 | if (ret < 0) |
| 113 | return ret; |
| 114 | |
| 115 | priv->ctx->time_base_in = avctx->time_base; |
| 116 | |
| 117 | if (bsfc->args && bsfc->filter->priv_class) { |
| 118 | const AVOption *opt = av_opt_next(priv->ctx->priv_data, NULL); |
| 119 | const char * shorthand[2] = {NULL}; |
| 120 | |
| 121 | if (opt) |
| 122 | shorthand[0] = opt->name; |
| 123 | |
| 124 | ret = av_opt_set_from_string(priv->ctx->priv_data, bsfc->args, shorthand, "=", ":"); |
| 125 | } |
| 126 | |
| 127 | ret = av_bsf_init(priv->ctx); |
| 128 | if (ret < 0) |
| 129 | return ret; |
| 130 | } |
| 131 | |
| 132 | pkt.data = buf; |
| 133 | pkt.size = buf_size; |
| 134 | |
| 135 | ret = av_bsf_send_packet(priv->ctx, &pkt); |
| 136 | if (ret < 0) |
| 137 | return ret; |
| 138 | |
| 139 | *poutbuf = NULL; |
| 140 | *poutbuf_size = 0; |
| 141 | |
| 142 | ret = av_bsf_receive_packet(priv->ctx, &pkt); |
| 143 | if (ret == AVERROR(EAGAIN) || ret == AVERROR_EOF) |
| 144 | return 0; |
| 145 | else if (ret < 0) |
| 146 | return ret; |
| 147 | |
| 148 | *poutbuf = av_malloc(pkt.size + AV_INPUT_BUFFER_PADDING_SIZE); |
| 149 | if (!*poutbuf) { |
| 150 | av_packet_unref(&pkt); |
| 151 | return AVERROR(ENOMEM); |
| 152 | } |
| 153 | |
| 154 | *poutbuf_size = pkt.size; |
| 155 | memcpy(*poutbuf, pkt.data, pkt.size); |
| 156 | |
| 157 | av_packet_unref(&pkt); |
| 158 | |
| 159 | /* drain all the remaining packets we cannot return */ |
| 160 | while (ret >= 0) { |
| 161 | ret = av_bsf_receive_packet(priv->ctx, &pkt); |
| 162 | av_packet_unref(&pkt); |
| 163 | } |
| 164 | |
| 165 | if (!priv->extradata_updated) { |
| 166 | /* update extradata in avctx from the output codec parameters */ |
| 167 | if (priv->ctx->par_out->extradata_size && (!args || !strstr(args, "private_spspps_buf"))) { |
| 168 | av_freep(&avctx->extradata); |
| 169 | avctx->extradata_size = 0; |
| 170 | avctx->extradata = av_mallocz(priv->ctx->par_out->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE); |
| 171 | if (!avctx->extradata) |
| 172 | return AVERROR(ENOMEM); |
| 173 | memcpy(avctx->extradata, priv->ctx->par_out->extradata, priv->ctx->par_out->extradata_size); |
| 174 | avctx->extradata_size = priv->ctx->par_out->extradata_size; |
| 175 | } |
| 176 | |
| 177 | priv->extradata_updated = 1; |
| 178 | } |
| 179 | |
| 180 | return 1; |
| 181 | } |
| 182 | FF_ENABLE_DEPRECATION_WARNINGS |
| 183 | #endif |
| 184 |