git.sesse.net Git - ffmpeg/blob - libavcodec/vp9_superframe_split_bsf.c

   1 /*
   2  * This file is part of FFmpeg.
   3  *
   4  * FFmpeg is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2.1 of the License, or (at your option) any later version.
   8  *
   9  * FFmpeg is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with FFmpeg; if not, write to the Free Software
  16  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17  */
  18
  19 /**
  20  * @file
  21  * This bitstream filter splits VP9 superframes into packets containing
  22  * just one frame.
  23  */
  24
  25 #include <stddef.h>
  26
  27 #include "bsf.h"
  28 #include "bsf_internal.h"
  29 #include "bytestream.h"
  30 #include "get_bits.h"
  31
  32 typedef struct VP9SFSplitContext {
  33     AVPacket *buffer_pkt;
  34
  35     int nb_frames;
  36     int next_frame;
  37     size_t next_frame_offset;
  38     int sizes[8];
  39 } VP9SFSplitContext;
  40
  41 static int vp9_superframe_split_filter(AVBSFContext *ctx, AVPacket *out)
  42 {
  43     VP9SFSplitContext *s = ctx->priv_data;
  44     AVPacket *in;
  45     int i, j, ret, marker;
  46     int is_superframe = !!s->buffer_pkt->data;
  47
  48     if (!s->buffer_pkt->data) {
  49         ret = ff_bsf_get_packet_ref(ctx, s->buffer_pkt);
  50         if (ret < 0)
  51             return ret;
  52         in = s->buffer_pkt;
  53
  54         marker = in->data[in->size - 1];
  55         if ((marker & 0xe0) == 0xc0) {
  56             int length_size = 1 + ((marker >> 3) & 0x3);
  57             int   nb_frames = 1 + (marker & 0x7);
  58             int    idx_size = 2 + nb_frames * length_size;
  59
  60             if (in->size >= idx_size && in->data[in->size - idx_size] == marker) {
  61                 GetByteContext bc;
  62                 int64_t total_size = 0;
  63
  64                 bytestream2_init(&bc, in->data + in->size + 1 - idx_size,
  65                                  nb_frames * length_size);
  66
  67                 for (i = 0; i < nb_frames; i++) {
  68                     int frame_size = 0;
  69                     for (j = 0; j < length_size; j++)
  70                         frame_size |= bytestream2_get_byte(&bc) << (j * 8);
  71
  72                     total_size += frame_size;
  73                     if (frame_size < 0 || total_size > in->size - idx_size) {
  74                         av_log(ctx, AV_LOG_ERROR,
  75                                "Invalid frame size in a superframe: %d\n", frame_size);
  76                         ret = AVERROR(EINVAL);
  77                         goto fail;
  78                     }
  79                     s->sizes[i] = frame_size;
  80                 }
  81                 s->nb_frames         = nb_frames;
  82                 s->next_frame        = 0;
  83                 s->next_frame_offset = 0;
  84                 is_superframe        = 1;
  85             }
  86         }
  87     }
  88
  89     if (is_superframe) {
  90         GetBitContext gb;
  91         int profile, invisible = 0;
  92
  93         ret = av_packet_ref(out, s->buffer_pkt);
  94         if (ret < 0)
  95             goto fail;
  96
  97         out->data += s->next_frame_offset;
  98         out->size  = s->sizes[s->next_frame];
  99
 100         s->next_frame_offset += out->size;
 101         s->next_frame++;
 102
 103         if (s->next_frame >= s->nb_frames)
 104             av_packet_unref(s->buffer_pkt);
 105
 106         ret = init_get_bits8(&gb, out->data, out->size);
 107         if (ret < 0)
 108             goto fail;
 109
 110         get_bits(&gb, 2); // frame_marker
 111         profile  = get_bits1(&gb);
 112         profile |= get_bits1(&gb) << 1;
 113         if (profile == 3)
 114             get_bits1(&gb);
 115         if (!get_bits1(&gb)) {
 116             get_bits1(&gb);
 117             invisible = !get_bits1(&gb);
 118         }
 119
 120         if (invisible)
 121             out->pts = AV_NOPTS_VALUE;
 122
 123     } else {
 124         av_packet_move_ref(out, s->buffer_pkt);
 125     }
 126
 127     return 0;
 128 fail:
 129     if (ret < 0)
 130         av_packet_unref(out);
 131     av_packet_unref(s->buffer_pkt);
 132     return ret;
 133 }
 134
 135 static int vp9_superframe_split_init(AVBSFContext *ctx)
 136 {
 137     VP9SFSplitContext *s = ctx->priv_data;
 138
 139     s->buffer_pkt = av_packet_alloc();
 140     if (!s->buffer_pkt)
 141         return AVERROR(ENOMEM);
 142
 143     return 0;
 144 }
 145
 146 static void vp9_superframe_split_flush(AVBSFContext *ctx)
 147 {
 148     VP9SFSplitContext *s = ctx->priv_data;
 149     av_packet_unref(s->buffer_pkt);
 150 }
 151
 152 static void vp9_superframe_split_uninit(AVBSFContext *ctx)
 153 {
 154     VP9SFSplitContext *s = ctx->priv_data;
 155     av_packet_free(&s->buffer_pkt);
 156 }
 157
 158 const AVBitStreamFilter ff_vp9_superframe_split_bsf = {
 159     .name = "vp9_superframe_split",
 160     .priv_data_size = sizeof(VP9SFSplitContext),
 161     .init           = vp9_superframe_split_init,
 162     .flush          = vp9_superframe_split_flush,
 163     .close          = vp9_superframe_split_uninit,
 164     .filter         = vp9_superframe_split_filter,
 165     .codec_ids      = (const enum AVCodecID []){ AV_CODEC_ID_VP9, AV_CODEC_ID_NONE },
 166 };