vda_h264.c
Go to the documentation of this file.
1 /*
2  * VDA H.264 hardware acceleration
3  *
4  * copyright (c) 2011 Sebastien Zwickert
5  *
6  * This file is part of Libav.
7  *
8  * Libav 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  * Libav 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 Libav; if not, write to the Free Software
20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21  */
22 
23 #include "h264.h"
24 #include "h264data.h"
25 
26 #include "vda_internal.h"
27 
28 /* This structure is used to store the bitstream of the current frame. */
30  uint8_t *bitstream;
32 };
33 
34 static int start_frame(AVCodecContext *avctx,
35  av_unused const uint8_t *buffer,
36  av_unused uint32_t size)
37 {
38  const H264Context *h = avctx->priv_data;
39  struct vda_context *vda_ctx = avctx->hwaccel_context;
41 
42  if (!vda_ctx->decoder)
43  return -1;
44 
45  pic_ctx->bitstream = NULL;
46  pic_ctx->bitstream_size = 0;
47 
48  return 0;
49 }
50 
51 static int decode_slice(AVCodecContext *avctx,
52  const uint8_t *buffer,
53  uint32_t size)
54 {
55  H264Context *h = avctx->priv_data;
56  struct vda_context *vda_ctx = avctx->hwaccel_context;
58  void *tmp;
59 
60  if (!vda_ctx->decoder)
61  return -1;
62 
63  tmp = av_realloc(pic_ctx->bitstream, pic_ctx->bitstream_size+size+4);
64  if (!tmp)
65  return AVERROR(ENOMEM);
66 
67  pic_ctx->bitstream = tmp;
68 
69  AV_WB32(pic_ctx->bitstream + pic_ctx->bitstream_size, size);
70  memcpy(pic_ctx->bitstream + pic_ctx->bitstream_size + 4, buffer, size);
71 
72  pic_ctx->bitstream_size += size + 4;
73 
74  return 0;
75 }
76 
77 static int end_frame(AVCodecContext *avctx)
78 {
79  H264Context *h = avctx->priv_data;
80  struct vda_context *vda_ctx = avctx->hwaccel_context;
82  AVFrame *frame = &h->s.current_picture_ptr->f;
83  int status;
84 
85  if (!vda_ctx->decoder || !pic_ctx->bitstream)
86  return -1;
87 
88  status = ff_vda_decoder_decode(vda_ctx, pic_ctx->bitstream,
89  pic_ctx->bitstream_size,
90  frame->reordered_opaque);
91 
92  if (status)
93  av_log(avctx, AV_LOG_ERROR, "Failed to decode frame (%d)\n", status);
94 
95  av_freep(&pic_ctx->bitstream);
96 
97  return status;
98 }
99 
101  .name = "h264_vda",
102  .type = AVMEDIA_TYPE_VIDEO,
103  .id = CODEC_ID_H264,
104  .pix_fmt = PIX_FMT_VDA_VLD,
105  .start_frame = start_frame,
106  .decode_slice = decode_slice,
107  .end_frame = end_frame,
108  .priv_data_size = sizeof(struct vda_picture_context),
109 };