FFmpeg: libavcodec/libopenh264dec.c Source File

FFmpeg
libopenh264dec.c
Go to the documentation of this file.
1 /*
2  * OpenH264 video decoder
3  * Copyright (C) 2016 Martin Storsjo
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include <wels/codec_api.h>
23 #include <wels/codec_ver.h>
24 
25 #include "libavutil/common.h"
26 #include "libavutil/fifo.h"
27 #include "libavutil/imgutils.h"
28 #include "libavutil/intreadwrite.h"
29 #include "libavutil/mathematics.h"
30 #include "libavutil/opt.h"
31 
32 #include "avcodec.h"
33 #include "internal.h"
34 #include "libopenh264.h"
35 
36  typedef struct SVCContext {
37   ISVCDecoder *decoder;
38   AVBSFContext *bsf;
39   AVFifoBuffer *packet_fifo;
40   AVPacket pkt_filtered;
41 } SVCContext;
42 
43  static av_cold int svc_decode_close(AVCodecContext *avctx)
44 {
45  SVCContext *s = avctx->priv_data;
46  AVPacket pkt;
47 
48  if (s->decoder)
49  WelsDestroyDecoder(s->decoder);
50 
51  while (s->packet_fifo && av_fifo_size(s->packet_fifo) >= sizeof(pkt)) {
52  av_fifo_generic_read(s->packet_fifo, &pkt, sizeof(pkt), NULL);
53  av_packet_unref(&pkt);
54  }
55 
56  av_bsf_free(&s->bsf);
57  av_packet_unref(&s->pkt_filtered);
58  av_fifo_free(s->packet_fifo);
59 
60  return 0;
61 }
62 
63  static av_cold int svc_decode_init(AVCodecContext *avctx)
64 {
65  SVCContext *s = avctx->priv_data;
66  SDecodingParam param = { 0 };
67  int err;
68  int log_level;
69  WelsTraceCallback callback_function;
70 
71  if ((err = ff_libopenh264_check_version(avctx)) < 0)
72  return err;
73 
74  s->packet_fifo = av_fifo_alloc(sizeof(AVPacket));
75  if (!s->packet_fifo)
76  return AVERROR(ENOMEM);
77 
78  if (WelsCreateDecoder(&s->decoder)) {
79  av_log(avctx, AV_LOG_ERROR, "Unable to create decoder\n");
80  return AVERROR_UNKNOWN;
81  }
82 
83  // Pass all libopenh264 messages to our callback, to allow ourselves to filter them.
84  log_level = WELS_LOG_DETAIL;
85  callback_function = ff_libopenh264_trace_callback;
86  (*s->decoder)->SetOption(s->decoder, DECODER_OPTION_TRACE_LEVEL, &log_level);
87  (*s->decoder)->SetOption(s->decoder, DECODER_OPTION_TRACE_CALLBACK, (void *)&callback_function);
88  (*s->decoder)->SetOption(s->decoder, DECODER_OPTION_TRACE_CALLBACK_CONTEXT, (void *)&avctx);
89 
90 #if !OPENH264_VER_AT_LEAST(1, 6)
91  param.eOutputColorFormat = videoFormatI420;
92 #endif
93  param.eEcActiveIdc = ERROR_CON_DISABLE;
94  param.sVideoProperty.eVideoBsType = VIDEO_BITSTREAM_DEFAULT;
95 
96  if ((*s->decoder)->Initialize(s->decoder, &param) != cmResultSuccess) {
97  av_log(avctx, AV_LOG_ERROR, "Initialize failed\n");
98  return AVERROR_UNKNOWN;
99  }
100 
101  avctx->pix_fmt = AV_PIX_FMT_YUV420P;
102 
103  return 0;
104 }
105 
106  static int init_bsf(AVCodecContext *avctx)
107 {
108  SVCContext *s = avctx->priv_data;
109  const AVBitStreamFilter *filter;
110  int ret;
111 
112  if (s->bsf)
113  return 0;
114 
115  // If the input stream already is annex b, this BSF only passes the
116  // packets through unchanged.
117  filter = av_bsf_get_by_name("h264_mp4toannexb");
118  if (!filter)
119  return AVERROR_BUG;
120 
121  ret = av_bsf_alloc(filter, &s->bsf);
122  if (ret < 0)
123  return ret;
124 
125  ret = avcodec_parameters_from_context(s->bsf->par_in, avctx);
126  if (ret < 0)
127  return ret;
128 
129  s->bsf->time_base_in = avctx->time_base;
130 
131  ret = av_bsf_init(s->bsf);
132  if (ret < 0)
133  return ret;
134 
135  return ret;
136 }
137 
138  static int svc_decode_frame(AVCodecContext *avctx, void *data,
139  int *got_frame, AVPacket *avpkt)
140 {
141  SVCContext *s = avctx->priv_data;
142  SBufferInfo info = { 0 };
143  uint8_t* ptrs[3];
144  int linesize[3];
145  AVFrame *avframe = data;
146  int ret;
147  DECODING_STATE state;
148 
149  if ((ret = init_bsf(avctx)) < 0)
150  return ret;
151 
152  if (avpkt->size) {
153  AVPacket input_ref = { 0 };
154  if (av_fifo_space(s->packet_fifo) < sizeof(input_ref)) {
155  ret = av_fifo_realloc2(s->packet_fifo,
156  av_fifo_size(s->packet_fifo) + sizeof(input_ref));
157  if (ret < 0)
158  return ret;
159  }
160 
161  ret = av_packet_ref(&input_ref, avpkt);
162  if (ret < 0)
163  return ret;
164  av_fifo_generic_write(s->packet_fifo, &input_ref, sizeof(input_ref), NULL);
165  }
166 
167  while (!*got_frame) {
168  /* prepare the input data -- convert to Annex B if needed */
169  if (s->pkt_filtered.size <= 0) {
170  AVPacket input_ref;
171 
172  /* no more data */
173  if (av_fifo_size(s->packet_fifo) < sizeof(AVPacket))
174  return avpkt->size ? avpkt->size : 0;
175 
176  av_packet_unref(&s->pkt_filtered);
177 
178  av_fifo_generic_read(s->packet_fifo, &input_ref, sizeof(input_ref), NULL);
179  ret = av_bsf_send_packet(s->bsf, &input_ref);
180  if (ret < 0) {
181  av_packet_unref(&input_ref);
182  return ret;
183  }
184 
185  ret = av_bsf_receive_packet(s->bsf, &s->pkt_filtered);
186  if (ret < 0)
187  av_packet_move_ref(&s->pkt_filtered, &input_ref);
188  else
189  av_packet_unref(&input_ref);
190  }
191 
192  state = (*s->decoder)->DecodeFrame2(s->decoder, s->pkt_filtered.data, s->pkt_filtered.size, ptrs, &info);
193  s->pkt_filtered.size = 0;
194  if (state != dsErrorFree) {
195  av_log(avctx, AV_LOG_ERROR, "DecodeFrame2 failed\n");
196  return AVERROR_UNKNOWN;
197  }
198  if (info.iBufferStatus != 1) {
199  av_log(avctx, AV_LOG_DEBUG, "No frame produced\n");
200  continue;
201  }
202 
203  ret = ff_set_dimensions(avctx, info.UsrData.sSystemBuffer.iWidth, info.UsrData.sSystemBuffer.iHeight);
204  if (ret < 0)
205  return ret;
206  // The decoder doesn't (currently) support decoding into a user
207  // provided buffer, so do a copy instead.
208  if (ff_get_buffer(avctx, avframe, 0) < 0) {
209  av_log(avctx, AV_LOG_ERROR, "Unable to allocate buffer\n");
210  return AVERROR(ENOMEM);
211  }
212 
213  linesize[0] = info.UsrData.sSystemBuffer.iStride[0];
214  linesize[1] = linesize[2] = info.UsrData.sSystemBuffer.iStride[1];
215  av_image_copy(avframe->data, avframe->linesize, (const uint8_t **) ptrs, linesize, avctx->pix_fmt, avctx->width, avctx->height);
216 
217  avframe->pts = s->pkt_filtered.pts;
218  avframe->pkt_dts = s->pkt_filtered.dts;
219 #if FF_API_PKT_PTS
220 FF_DISABLE_DEPRECATION_WARNINGS
221  avframe->pkt_pts = s->pkt_filtered.pts;
222 FF_ENABLE_DEPRECATION_WARNINGS
223 #endif
224 
225  *got_frame = 1;
226  }
227  return avpkt->size;
228 }
229 
230  AVCodec ff_libopenh264_decoder = {
231  .name = "libopenh264",
232  .long_name = NULL_IF_CONFIG_SMALL("OpenH264 H.264 / AVC / MPEG-4 AVC / MPEG-4 part 10"),
233  .type = AVMEDIA_TYPE_VIDEO,
234  .id = AV_CODEC_ID_H264,
235  .priv_data_size = sizeof(SVCContext),
236  .init = svc_decode_init,
237  .decode = svc_decode_frame,
238  .close = svc_decode_close,
239  // The decoder doesn't currently support B-frames, and the decoder's API
240  // doesn't support reordering/delay, but the BSF could incur delay.
241  .capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_DR1,
242  .caps_internal = FF_CODEC_CAP_SETS_PKT_DTS | FF_CODEC_CAP_INIT_THREADSAFE |
243  FF_CODEC_CAP_INIT_CLEANUP,
244 };
FF_CODEC_CAP_INIT_CLEANUP
#define FF_CODEC_CAP_INIT_CLEANUP
The codec allows calling the close function for deallocation even if the init function returned a fai...
Definition: internal.h:48
ff_libopenh264_trace_callback
void ff_libopenh264_trace_callback(void *ctx, int level, const char *msg)
Definition: libopenh264.c:41
av_bsf_free
void av_bsf_free(AVBSFContext **ctx)
Free a bitstream filter context and everything associated with it; write NULL into the supplied point...
Definition: bsf.c:36
NULL
#define NULL
Definition: coverity.c:32
s
const char * s
Definition: avisynth_c.h:768
AVFrame
This structure describes decoded (raw) audio or video data.
Definition: frame.h:187
data
ptrdiff_t const GLvoid * data
Definition: opengl_enc.c:101
init_bsf
static int init_bsf(AVCodecContext *avctx)
Definition: libopenh264dec.c:106
SVCContext::bsf
AVBSFContext * bsf
Definition: libopenh264dec.c:38
imgutils.h
misc image utilities
ff_set_dimensions
int ff_set_dimensions(AVCodecContext *s, int width, int height)
Check that the provided frame dimensions are valid and set them on the codec context.
Definition: utils.c:210
init
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
AVBSFContext
The bitstream filter state.
Definition: avcodec.h:5830
AVPacket::size
int size
Definition: avcodec.h:1658
av_bsf_get_by_name
const AVBitStreamFilter * av_bsf_get_by_name(const char *name)
Definition: bitstream_filters.c:58
AVCodecContext::pix_fmt
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1960
pkt
static AVPacket pkt
Definition: demuxing_decoding.c:54
AVCodec
AVCodec.
Definition: avcodec.h:3681
av_bsf_init
int av_bsf_init(AVBSFContext *ctx)
Prepare the filter for use, after all the parameters and options have been set.
Definition: bsf.c:135
av_fifo_generic_write
int av_fifo_generic_write(AVFifoBuffer *f, void *src, int size, int(*func)(void *, void *, int))
Feed data from a user-supplied callback to an AVFifoBuffer.
Definition: fifo.c:122
AVCodecContext::time_base
AVRational time_base
This is the fundamental unit of time (in seconds) in terms of which frame timestamps are represented...
Definition: avcodec.h:1869
av_bsf_alloc
int av_bsf_alloc(const AVBitStreamFilter *filter, AVBSFContext **ctx)
Allocate a context for a given bitstream filter.
Definition: bsf.c:82
AV_CODEC_CAP_DELAY
#define AV_CODEC_CAP_DELAY
Encoder or decoder requires flushing with NULL input at the end in order to give the complete and cor...
Definition: avcodec.h:1019
av_bsf_receive_packet
int av_bsf_receive_packet(AVBSFContext *ctx, AVPacket *pkt)
Retrieve a filtered packet.
Definition: bsf.c:199
FF_CODEC_CAP_INIT_THREADSAFE
#define FF_CODEC_CAP_INIT_THREADSAFE
The codec does not modify any global variables in the init function, allowing to call the init functi...
Definition: internal.h:40
svc_decode_close
static av_cold int svc_decode_close(AVCodecContext *avctx)
Definition: libopenh264dec.c:43
uint8_t
uint8_t
Definition: audio_convert.c:194
av_cold
#define av_cold
Definition: attributes.h:82
svc_decode_frame
static int svc_decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
Definition: libopenh264dec.c:138
opt.h
AVOptions.
AVFrame::pts
int64_t pts
Presentation timestamp in time_base units (time when frame should be shown to user).
Definition: frame.h:271
av_fifo_space
int av_fifo_space(const AVFifoBuffer *f)
Return the amount of space in bytes in the AVFifoBuffer, that is the amount of data you can write int...
Definition: fifo.c:82
filter
static void filter(int16_t *output, ptrdiff_t out_stride, int16_t *low, ptrdiff_t low_stride, int16_t *high, ptrdiff_t high_stride, int len, uint8_t clip)
Definition: cfhd.c:80
AVPacket::data
uint8_t * data
Definition: avcodec.h:1657
av_packet_move_ref
void av_packet_move_ref(AVPacket *dst, AVPacket *src)
Move every field in src to dst and reset src.
Definition: avpacket.c:644
av_fifo_free
void av_fifo_free(AVFifoBuffer *f)
Free an AVFifoBuffer.
Definition: fifo.c:55
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:28
av_packet_ref
int av_packet_ref(AVPacket *dst, const AVPacket *src)
Setup a new reference to the data described by a given packet.
Definition: avpacket.c:598
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
AVERROR
#define AVERROR(e)
Definition: error.h:43
svc_decode_init
static av_cold int svc_decode_init(AVCodecContext *avctx)
Definition: libopenh264dec.c:63
NULL_IF_CONFIG_SMALL
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:179
state
static struct @253 state
AV_LOG_DEBUG
#define AV_LOG_DEBUG
Stuff which is only useful for libav* developers.
Definition: log.h:197
av_fifo_generic_read
int av_fifo_generic_read(AVFifoBuffer *f, void *dest, int buf_size, void(*func)(void *, void *, int))
Feed data from an AVFifoBuffer to a user-supplied callback.
Definition: fifo.c:213
AVBSFContext::time_base_in
AVRational time_base_in
The timebase used for the timestamps of the input packets.
Definition: avcodec.h:5870
AVCodec::name
const char * name
Name of the codec implementation.
Definition: avcodec.h:3688
av_image_copy
void av_image_copy(uint8_t *dst_data[4], int dst_linesizes[4], const uint8_t *src_data[4], const int src_linesizes[4], enum AVPixelFormat pix_fmt, int width, int height)
Copy image in src_data to dst_data.
Definition: imgutils.c:385
SVCContext::pkt_filtered
AVPacket pkt_filtered
Definition: libopenh264dec.c:40
AVCodecContext::width
int width
picture width / height.
Definition: avcodec.h:1919
av_bsf_send_packet
int av_bsf_send_packet(AVBSFContext *ctx, AVPacket *pkt)
Submit a packet for filtering.
Definition: bsf.c:176
AVCodecContext::height
int height
Definition: avcodec.h:1919
AVFifoBuffer
Definition: fifo.h:31
avcodec.h
Libavcodec external API header.
av_fifo_size
int av_fifo_size(const AVFifoBuffer *f)
Return the amount of data in bytes in the AVFifoBuffer, that is the amount of data you can read from ...
Definition: fifo.c:77
AVFrame::linesize
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:218
av_fifo_realloc2
int av_fifo_realloc2(AVFifoBuffer *f, unsigned int new_size)
Resize an AVFifoBuffer.
Definition: fifo.c:87
AVCodecContext
main external API structure.
Definition: avcodec.h:1732
av_packet_unref
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:589
ff_get_buffer
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
Definition: utils.c:953
fifo.h
a very simple circular buffer FIFO implementation
AVERROR_BUG
#define AVERROR_BUG
Internal bug, also see AVERROR_BUG2.
Definition: error.h:50
avcodec_parameters_from_context
int avcodec_parameters_from_context(AVCodecParameters *par, const AVCodecContext *codec)
Fill the parameters struct based on the values from the supplied codec context.
Definition: utils.c:4207
SVCContext::packet_fifo
AVFifoBuffer * packet_fifo
Definition: libopenh264dec.c:39
FF_CODEC_CAP_SETS_PKT_DTS
#define FF_CODEC_CAP_SETS_PKT_DTS
Decoders marked with FF_CODEC_CAP_SETS_PKT_DTS want to set AVFrame.pkt_dts manually.
Definition: internal.h:55
AVFrame::data
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:201
AVFrame::pkt_pts
attribute_deprecated int64_t pkt_pts
PTS copied from the AVPacket that was decoded to produce this frame.
Definition: frame.h:279
ff_libopenh264_decoder
AVCodec ff_libopenh264_decoder
Definition: libopenh264dec.c:230
AVFrame::pkt_dts
int64_t pkt_dts
DTS copied from the AVPacket that triggered returning this frame.
Definition: frame.h:287
AV_PIX_FMT_YUV420P
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:62
FF_DISABLE_DEPRECATION_WARNINGS
#define FF_DISABLE_DEPRECATION_WARNINGS
Definition: internal.h:83
internal.h
common internal api header.
common.h
common internal and external API header
AVERROR_UNKNOWN
#define AVERROR_UNKNOWN
Unknown error, typically from an external library.
Definition: error.h:71
AVCodecContext::priv_data
void * priv_data
Definition: avcodec.h:1774
FF_ENABLE_DEPRECATION_WARNINGS
#define FF_ENABLE_DEPRECATION_WARNINGS
Definition: internal.h:84
av_fifo_alloc
AVFifoBuffer * av_fifo_alloc(unsigned int size)
Initialize an AVFifoBuffer.
Definition: fifo.c:43
ff_libopenh264_check_version
int ff_libopenh264_check_version(void *logctx)
Definition: libopenh264.c:49
AVPacket::dts
int64_t dts
Decompression timestamp in AVStream->time_base units; the time at which the packet is decompressed...
Definition: avcodec.h:1656
decode
static int decode(AVCodecContext *avctx, AVFrame *frame, int *got_frame, AVPacket *pkt)
Definition: ffmpeg.c:2257
SVCContext::decoder
ISVCDecoder * decoder
Definition: libopenh264dec.c:37
AVPacket
This structure stores compressed data.
Definition: avcodec.h:1634
AVBSFContext::par_in
AVCodecParameters * par_in
Parameters of the input stream.
Definition: avcodec.h:5858
AV_CODEC_CAP_DR1
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Definition: avcodec.h:994
AVPacket::pts
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: avcodec.h:1650

Generated on Fri Jan 12 2018 01:45:40 for FFmpeg by   doxygen 1.8.6

AltStyle によって変換されたページ (->オリジナル) /