Directory: | ../../../ffmpeg/ |
---|---|
File: | src/libavcodec/vp9_superframe_split_bsf.c |
Date: | 2022-07-04 00:18:54 |
Exec | Total | Coverage | |
---|---|---|---|
Lines: | 62 | 77 | 80.5% |
Branches: | 28 | 38 | 73.7% |
Line | Branch | Exec | Source |
---|---|---|---|
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 | 4536 | static int vp9_superframe_split_filter(AVBSFContext *ctx, AVPacket *out) | |
42 | { | ||
43 | 4536 | VP9SFSplitContext *s = ctx->priv_data; | |
44 | AVPacket *in; | ||
45 | int i, j, ret, marker; | ||
46 | 4536 | int is_superframe = !!s->buffer_pkt->data; | |
47 | |||
48 |
2/2✓ Branch 0 taken 4472 times.
✓ Branch 1 taken 64 times.
|
4536 | if (!s->buffer_pkt->data) { |
49 | 4472 | ret = ff_bsf_get_packet_ref(ctx, s->buffer_pkt); | |
50 |
2/2✓ Branch 0 taken 2229 times.
✓ Branch 1 taken 2243 times.
|
4472 | if (ret < 0) |
51 | 2229 | return ret; | |
52 | 2243 | in = s->buffer_pkt; | |
53 | |||
54 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 2243 times.
|
2243 | if (!in->size) |
55 | ✗ | goto passthrough; | |
56 | |||
57 | 2243 | marker = in->data[in->size - 1]; | |
58 |
2/2✓ Branch 0 taken 59 times.
✓ Branch 1 taken 2184 times.
|
2243 | if ((marker & 0xe0) == 0xc0) { |
59 | 59 | int length_size = 1 + ((marker >> 3) & 0x3); | |
60 | 59 | int nb_frames = 1 + (marker & 0x7); | |
61 | 59 | int idx_size = 2 + nb_frames * length_size; | |
62 | |||
63 |
2/4✓ Branch 0 taken 59 times.
✗ Branch 1 not taken.
✓ Branch 2 taken 59 times.
✗ Branch 3 not taken.
|
59 | if (in->size >= idx_size && in->data[in->size - idx_size] == marker) { |
64 | GetByteContext bc; | ||
65 | 59 | int64_t total_size = 0; | |
66 | |||
67 | 59 | bytestream2_init(&bc, in->data + in->size + 1 - idx_size, | |
68 | nb_frames * length_size); | ||
69 | |||
70 |
2/2✓ Branch 0 taken 123 times.
✓ Branch 1 taken 59 times.
|
182 | for (i = 0; i < nb_frames; i++) { |
71 | 123 | int frame_size = 0; | |
72 |
2/2✓ Branch 0 taken 242 times.
✓ Branch 1 taken 123 times.
|
365 | for (j = 0; j < length_size; j++) |
73 | 242 | frame_size |= bytestream2_get_byte(&bc) << (j * 8); | |
74 | |||
75 | 123 | total_size += frame_size; | |
76 |
2/4✓ Branch 0 taken 123 times.
✗ Branch 1 not taken.
✗ Branch 2 not taken.
✓ Branch 3 taken 123 times.
|
123 | if (frame_size <= 0 || total_size > in->size - idx_size) { |
77 | ✗ | av_log(ctx, AV_LOG_ERROR, | |
78 | "Invalid frame size in a superframe: %d\n", frame_size); | ||
79 | ✗ | ret = AVERROR(EINVAL); | |
80 | ✗ | goto fail; | |
81 | } | ||
82 | 123 | s->sizes[i] = frame_size; | |
83 | } | ||
84 | 59 | s->nb_frames = nb_frames; | |
85 | 59 | s->next_frame = 0; | |
86 | 59 | s->next_frame_offset = 0; | |
87 | 59 | is_superframe = 1; | |
88 | } | ||
89 | } | ||
90 | } | ||
91 | |||
92 |
2/2✓ Branch 0 taken 123 times.
✓ Branch 1 taken 2184 times.
|
2307 | if (is_superframe) { |
93 | GetBitContext gb; | ||
94 | 123 | int profile, invisible = 0; | |
95 | |||
96 | 123 | ret = av_packet_ref(out, s->buffer_pkt); | |
97 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 123 times.
|
123 | if (ret < 0) |
98 | ✗ | goto fail; | |
99 | |||
100 | 123 | out->data += s->next_frame_offset; | |
101 | 123 | out->size = s->sizes[s->next_frame]; | |
102 | |||
103 | 123 | s->next_frame_offset += out->size; | |
104 | 123 | s->next_frame++; | |
105 | |||
106 |
2/2✓ Branch 0 taken 59 times.
✓ Branch 1 taken 64 times.
|
123 | if (s->next_frame >= s->nb_frames) |
107 | 59 | av_packet_unref(s->buffer_pkt); | |
108 | |||
109 | 123 | ret = init_get_bits8(&gb, out->data, out->size); | |
110 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 123 times.
|
123 | if (ret < 0) |
111 | ✗ | goto fail; | |
112 | |||
113 | 123 | get_bits(&gb, 2); // frame_marker | |
114 | 123 | profile = get_bits1(&gb); | |
115 | 123 | profile |= get_bits1(&gb) << 1; | |
116 |
2/2✓ Branch 0 taken 28 times.
✓ Branch 1 taken 95 times.
|
123 | if (profile == 3) |
117 | 28 | get_bits1(&gb); | |
118 |
2/2✓ Branch 1 taken 121 times.
✓ Branch 2 taken 2 times.
|
123 | if (!get_bits1(&gb)) { |
119 | 121 | get_bits1(&gb); | |
120 | 121 | invisible = !get_bits1(&gb); | |
121 | } | ||
122 | |||
123 |
2/2✓ Branch 0 taken 64 times.
✓ Branch 1 taken 59 times.
|
123 | if (invisible) |
124 | 64 | out->pts = AV_NOPTS_VALUE; | |
125 | |||
126 | } else { | ||
127 | 2184 | passthrough: | |
128 | 2184 | av_packet_move_ref(out, s->buffer_pkt); | |
129 | } | ||
130 | |||
131 | 2307 | return 0; | |
132 | ✗ | fail: | |
133 | ✗ | if (ret < 0) | |
134 | ✗ | av_packet_unref(out); | |
135 | ✗ | av_packet_unref(s->buffer_pkt); | |
136 | ✗ | return ret; | |
137 | } | ||
138 | |||
139 | 469 | static int vp9_superframe_split_init(AVBSFContext *ctx) | |
140 | { | ||
141 | 469 | VP9SFSplitContext *s = ctx->priv_data; | |
142 | |||
143 | 469 | s->buffer_pkt = av_packet_alloc(); | |
144 |
1/2✗ Branch 0 not taken.
✓ Branch 1 taken 469 times.
|
469 | if (!s->buffer_pkt) |
145 | ✗ | return AVERROR(ENOMEM); | |
146 | |||
147 | 469 | return 0; | |
148 | } | ||
149 | |||
150 | ✗ | static void vp9_superframe_split_flush(AVBSFContext *ctx) | |
151 | { | ||
152 | ✗ | VP9SFSplitContext *s = ctx->priv_data; | |
153 | ✗ | av_packet_unref(s->buffer_pkt); | |
154 | } | ||
155 | |||
156 | 469 | static void vp9_superframe_split_uninit(AVBSFContext *ctx) | |
157 | { | ||
158 | 469 | VP9SFSplitContext *s = ctx->priv_data; | |
159 | 469 | av_packet_free(&s->buffer_pkt); | |
160 | 469 | } | |
161 | |||
162 | const FFBitStreamFilter ff_vp9_superframe_split_bsf = { | ||
163 | .p.name = "vp9_superframe_split", | ||
164 | .p.codec_ids = (const enum AVCodecID []){ AV_CODEC_ID_VP9, AV_CODEC_ID_NONE }, | ||
165 | .priv_data_size = sizeof(VP9SFSplitContext), | ||
166 | .init = vp9_superframe_split_init, | ||
167 | .flush = vp9_superframe_split_flush, | ||
168 | .close = vp9_superframe_split_uninit, | ||
169 | .filter = vp9_superframe_split_filter, | ||
170 | }; | ||
171 |