FFmpeg
pcm_rechunk_bsf.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2020 Marton Balint
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 "bsf.h"
22 #include "bsf_internal.h"
23 #include "libavutil/avassert.h"
24 #include "libavutil/opt.h"
25 
26 typedef struct PCMContext {
27  const AVClass *class;
28 
30  int pad;
32 
36  int64_t n;
37 } PCMContext;
38 
39 static int init(AVBSFContext *ctx)
40 {
42  AVRational sr = av_make_q(ctx->par_in->sample_rate, 1);
43  int64_t min_samples;
44 
45  if (ctx->par_in->ch_layout.nb_channels <= 0 || ctx->par_in->sample_rate <= 0)
46  return AVERROR(EINVAL);
47 
48  ctx->time_base_out = av_inv_q(sr);
49  s->sample_size = ctx->par_in->ch_layout.nb_channels *
50  av_get_bits_per_sample(ctx->par_in->codec_id) / 8;
51 
52  if (s->frame_rate.num) {
53  min_samples = av_rescale_q_rnd(1, sr, s->frame_rate, AV_ROUND_DOWN);
54  } else {
55  min_samples = s->nb_out_samples;
56  }
57  if (min_samples <= 0 || min_samples > INT_MAX / s->sample_size - 1)
58  return AVERROR(EINVAL);
59 
60  s->in_pkt = av_packet_alloc();
61  s->out_pkt = av_packet_alloc();
62  if (!s->in_pkt || !s->out_pkt)
63  return AVERROR(ENOMEM);
64 
65  return 0;
66 }
67 
68 static void uninit(AVBSFContext *ctx)
69 {
71  av_packet_free(&s->in_pkt);
72  av_packet_free(&s->out_pkt);
73 }
74 
75 static void flush(AVBSFContext *ctx)
76 {
78  av_packet_unref(s->in_pkt);
79  av_packet_unref(s->out_pkt);
80  s->n = 0;
81 }
82 
83 static int send_packet(PCMContext *s, int nb_samples, AVPacket *pkt)
84 {
85  pkt->duration = nb_samples;
86  s->n++;
87  return 0;
88 }
89 
90 static void drain_packet(AVPacket *pkt, int drain_data, int drain_samples)
91 {
92  pkt->size -= drain_data;
93  pkt->data += drain_data;
94  if (pkt->dts != AV_NOPTS_VALUE)
95  pkt->dts += drain_samples;
96  if (pkt->pts != AV_NOPTS_VALUE)
97  pkt->pts += drain_samples;
98 }
99 
101 {
103  if (s->frame_rate.num) {
104  AVRational sr = av_make_q(ctx->par_in->sample_rate, 1);
105  return av_rescale_q(s->n + 1, sr, s->frame_rate) - av_rescale_q(s->n, sr, s->frame_rate);
106  } else {
107  return s->nb_out_samples;
108  }
109 }
110 
112 {
114  int nb_samples = get_next_nb_samples(ctx);
115  int data_size = nb_samples * s->sample_size;
116  int ret;
117 
118  do {
119  if (s->in_pkt->size) {
120  if (s->out_pkt->size || s->in_pkt->size < data_size) {
121  int drain = FFMIN(s->in_pkt->size, data_size - s->out_pkt->size);
122  if (!s->out_pkt->size) {
123  ret = av_new_packet(s->out_pkt, data_size);
124  if (ret < 0)
125  return ret;
126  ret = av_packet_copy_props(s->out_pkt, s->in_pkt);
127  if (ret < 0) {
128  av_packet_unref(s->out_pkt);
129  return ret;
130  }
131  s->out_pkt->size = 0;
132  }
133  memcpy(s->out_pkt->data + s->out_pkt->size, s->in_pkt->data, drain);
134  s->out_pkt->size += drain;
135  drain_packet(s->in_pkt, drain, drain / s->sample_size);
136  if (!s->in_pkt->size)
137  av_packet_unref(s->in_pkt);
138  if (s->out_pkt->size == data_size) {
139  av_packet_move_ref(pkt, s->out_pkt);
140  return send_packet(s, nb_samples, pkt);
141  }
142  av_assert0(!s->in_pkt->size);
143  } else if (s->in_pkt->size > data_size) {
144  ret = av_packet_ref(pkt, s->in_pkt);
145  if (ret < 0)
146  return ret;
147  pkt->size = data_size;
148  drain_packet(s->in_pkt, data_size, nb_samples);
149  return send_packet(s, nb_samples, pkt);
150  } else {
151  av_assert0(s->in_pkt->size == data_size);
152  av_packet_move_ref(pkt, s->in_pkt);
153  return send_packet(s, nb_samples, pkt);
154  }
155  } else
156  av_packet_unref(s->in_pkt);
157 
158  ret = ff_bsf_get_packet_ref(ctx, s->in_pkt);
159  if (ret == AVERROR_EOF && s->out_pkt->size) {
160  if (s->pad) {
161  memset(s->out_pkt->data + s->out_pkt->size, 0, data_size - s->out_pkt->size);
162  s->out_pkt->size = data_size;
163  } else {
164  nb_samples = s->out_pkt->size / s->sample_size;
165  }
166  av_packet_move_ref(pkt, s->out_pkt);
167  return send_packet(s, nb_samples, pkt);
168  }
169  if (ret >= 0)
170  av_packet_rescale_ts(s->in_pkt, ctx->time_base_in, ctx->time_base_out);
171  } while (ret >= 0);
172 
173  return ret;
174 }
175 
176 #define OFFSET(x) offsetof(PCMContext, x)
177 #define FLAGS (AV_OPT_FLAG_AUDIO_PARAM | AV_OPT_FLAG_BSF_PARAM)
178 static const AVOption options[] = {
179  { "nb_out_samples", "set the number of per-packet output samples", OFFSET(nb_out_samples), AV_OPT_TYPE_INT, {.i64=1024}, 1, INT_MAX, FLAGS },
180  { "n", "set the number of per-packet output samples", OFFSET(nb_out_samples), AV_OPT_TYPE_INT, {.i64=1024}, 1, INT_MAX, FLAGS },
181  { "pad", "pad last packet with zeros", OFFSET(pad), AV_OPT_TYPE_BOOL, {.i64=1} , 0, 1, FLAGS },
182  { "p", "pad last packet with zeros", OFFSET(pad), AV_OPT_TYPE_BOOL, {.i64=1} , 0, 1, FLAGS },
183  { "frame_rate", "set number of packets per second", OFFSET(frame_rate), AV_OPT_TYPE_RATIONAL, {.dbl=0}, 0, INT_MAX, FLAGS },
184  { "r", "set number of packets per second", OFFSET(frame_rate), AV_OPT_TYPE_RATIONAL, {.dbl=0}, 0, INT_MAX, FLAGS },
185  { NULL },
186 };
187 
188 static const AVClass pcm_rechunk_class = {
189  .class_name = "pcm_rechunk_bsf",
190  .item_name = av_default_item_name,
191  .option = options,
192  .version = LIBAVUTIL_VERSION_INT,
193 };
194 
195 static const enum AVCodecID codec_ids[] = {
212 };
213 
215  .p.name = "pcm_rechunk",
216  .p.codec_ids = codec_ids,
217  .p.priv_class = &pcm_rechunk_class,
218  .priv_data_size = sizeof(PCMContext),
220  .init = init,
221  .flush = flush,
222  .close = uninit,
223 };
AV_CODEC_ID_PCM_S16LE
@ AV_CODEC_ID_PCM_S16LE
Definition: codec_id.h:331
av_packet_unref
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:426
PCMContext::nb_out_samples
int nb_out_samples
Definition: pcm_rechunk_bsf.c:29
AV_CODEC_ID_PCM_F32BE
@ AV_CODEC_ID_PCM_F32BE
Definition: codec_id.h:351
AVERROR
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later. That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another. Frame references ownership and permissions
opt.h
bsf_internal.h
AVERROR_EOF
#define AVERROR_EOF
End of file.
Definition: error.h:57
AVBitStreamFilter::name
const char * name
Definition: bsf.h:112
AVPacket::data
uint8_t * data
Definition: packet.h:515
uninit
static void uninit(AVBSFContext *ctx)
Definition: pcm_rechunk_bsf.c:68
AVOption
AVOption.
Definition: opt.h:251
AVPacket::duration
int64_t duration
Duration of this packet in AVStream->time_base units, 0 if unknown.
Definition: packet.h:533
PCMContext::sample_size
int sample_size
Definition: pcm_rechunk_bsf.c:35
filter
filter_frame For filters that do not use the this method is called when a frame is pushed to the filter s input It can be called at any time except in a reentrant way If the input frame is enough to produce then the filter should push the output frames on the output link immediately As an exception to the previous rule if the input frame is enough to produce several output frames then the filter needs output only at least one per link The additional frames can be left buffered in the filter
Definition: filter_design.txt:228
AV_OPT_TYPE_RATIONAL
@ AV_OPT_TYPE_RATIONAL
Definition: opt.h:230
av_packet_free
void av_packet_free(AVPacket **pkt)
Free the packet, if the packet is reference counted, it will be unreferenced first.
Definition: avpacket.c:74
AVBSFContext
The bitstream filter state.
Definition: bsf.h:68
AV_CODEC_ID_PCM_S64LE
@ AV_CODEC_ID_PCM_S64LE
Definition: codec_id.h:362
FLAGS
#define FLAGS
Definition: pcm_rechunk_bsf.c:177
bsf.h
AV_CODEC_ID_PCM_S16BE
@ AV_CODEC_ID_PCM_S16BE
Definition: codec_id.h:332
rechunk_filter
static int rechunk_filter(AVBSFContext *ctx, AVPacket *pkt)
Definition: pcm_rechunk_bsf.c:111
OFFSET
#define OFFSET(x)
Definition: pcm_rechunk_bsf.c:176
av_get_bits_per_sample
int av_get_bits_per_sample(enum AVCodecID codec_id)
Return codec bits per sample.
Definition: utils.c:558
flush
static void flush(AVBSFContext *ctx)
Definition: pcm_rechunk_bsf.c:75
AV_CODEC_ID_PCM_S8
@ AV_CODEC_ID_PCM_S8
Definition: codec_id.h:335
avassert.h
pkt
AVPacket * pkt
Definition: movenc.c:59
s
#define s(width, name)
Definition: cbs_vp9.c:198
av_new_packet
int av_new_packet(AVPacket *pkt, int size)
Allocate the payload of a packet and initialize its fields with default values.
Definition: avpacket.c:98
av_assert0
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:40
ctx
AVFormatContext * ctx
Definition: movenc.c:48
AV_CODEC_ID_PCM_F24LE
@ AV_CODEC_ID_PCM_F24LE
Definition: codec_id.h:365
av_rescale_q
int64_t av_rescale_q(int64_t a, AVRational bq, AVRational cq)
Rescale a 64-bit integer by 2 rational numbers.
Definition: mathematics.c:142
init
static int init(AVBSFContext *ctx)
Definition: pcm_rechunk_bsf.c:39
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:66
NULL
#define NULL
Definition: coverity.c:32
FFBitStreamFilter
Definition: bsf_internal.h:27
AVRational
Rational number (pair of numerator and denominator).
Definition: rational.h:58
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:237
AV_CODEC_ID_PCM_S64BE
@ AV_CODEC_ID_PCM_S64BE
Definition: codec_id.h:363
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:434
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:483
AVCodecID
AVCodecID
Identify the syntax and semantics of the bitstream.
Definition: codec_id.h:49
FFBitStreamFilter::p
AVBitStreamFilter p
The public AVBitStreamFilter.
Definition: bsf_internal.h:31
AV_CODEC_ID_PCM_S24LE
@ AV_CODEC_ID_PCM_S24LE
Definition: codec_id.h:343
PCMContext::in_pkt
AVPacket * in_pkt
Definition: pcm_rechunk_bsf.c:33
AVPacket::size
int size
Definition: packet.h:516
PCMContext::n
int64_t n
Definition: pcm_rechunk_bsf.c:36
PCMContext
Definition: pcm_rechunk_bsf.c:26
av_make_q
static AVRational av_make_q(int num, int den)
Create an AVRational.
Definition: rational.h:71
AV_NOPTS_VALUE
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:248
AVPacket::dts
int64_t dts
Decompression timestamp in AVStream->time_base units; the time at which the packet is decompressed.
Definition: packet.h:514
av_packet_alloc
AVPacket * av_packet_alloc(void)
Allocate an AVPacket and set its fields to default values.
Definition: avpacket.c:63
av_packet_rescale_ts
void av_packet_rescale_ts(AVPacket *pkt, AVRational src_tb, AVRational dst_tb)
Convert valid timing fields (timestamps / durations) in a packet from one timebase to another.
Definition: avpacket.c:530
av_packet_copy_props
int av_packet_copy_props(AVPacket *dst, const AVPacket *src)
Copy only "properties" fields from src to dst.
Definition: avpacket.c:389
AV_CODEC_ID_NONE
@ AV_CODEC_ID_NONE
Definition: codec_id.h:50
AVPacket::pts
int64_t pts
Presentation timestamp in AVStream->time_base units; the time at which the decompressed packet will b...
Definition: packet.h:508
pcm_rechunk_class
static const AVClass pcm_rechunk_class
Definition: pcm_rechunk_bsf.c:188
AV_ROUND_DOWN
@ AV_ROUND_DOWN
Round toward -infinity.
Definition: mathematics.h:133
AV_CODEC_ID_PCM_F64BE
@ AV_CODEC_ID_PCM_F64BE
Definition: codec_id.h:353
FFMIN
#define FFMIN(a, b)
Definition: macros.h:49
AV_CODEC_ID_PCM_S32BE
@ AV_CODEC_ID_PCM_S32BE
Definition: codec_id.h:340
av_inv_q
static av_always_inline AVRational av_inv_q(AVRational q)
Invert a rational.
Definition: rational.h:159
AV_CODEC_ID_PCM_F16LE
@ AV_CODEC_ID_PCM_F16LE
Definition: codec_id.h:364
PCMContext::pad
int pad
Definition: pcm_rechunk_bsf.c:30
ret
ret
Definition: filter_design.txt:187
AVClass::class_name
const char * class_name
The name of the class; usually it is the same name as the context structure type to which the AVClass...
Definition: log.h:71
codec_ids
static enum AVCodecID codec_ids[]
Definition: pcm_rechunk_bsf.c:195
ff_pcm_rechunk_bsf
const FFBitStreamFilter ff_pcm_rechunk_bsf
Definition: pcm_rechunk_bsf.c:214
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:225
send_packet
static int send_packet(PCMContext *s, int nb_samples, AVPacket *pkt)
Definition: pcm_rechunk_bsf.c:83
AV_CODEC_ID_PCM_S32LE
@ AV_CODEC_ID_PCM_S32LE
Definition: codec_id.h:339
options
static const AVOption options[]
Definition: pcm_rechunk_bsf.c:178
AV_CODEC_ID_PCM_F64LE
@ AV_CODEC_ID_PCM_F64LE
Definition: codec_id.h:354
AVPacket
This structure stores compressed data.
Definition: packet.h:492
AV_OPT_TYPE_BOOL
@ AV_OPT_TYPE_BOOL
Definition: opt.h:244
AV_CODEC_ID_PCM_F32LE
@ AV_CODEC_ID_PCM_F32LE
Definition: codec_id.h:352
ff_bsf_get_packet_ref
int ff_bsf_get_packet_ref(AVBSFContext *ctx, AVPacket *pkt)
Called by bitstream filters to get packet for filtering.
Definition: bsf.c:256
get_next_nb_samples
static int get_next_nb_samples(AVBSFContext *ctx)
Definition: pcm_rechunk_bsf.c:100
AVFormatContext::priv_data
void * priv_data
Format private data.
Definition: avformat.h:1220
AV_CODEC_ID_PCM_S24BE
@ AV_CODEC_ID_PCM_S24BE
Definition: codec_id.h:344
av_rescale_q_rnd
int64_t av_rescale_q_rnd(int64_t a, AVRational bq, AVRational cq, enum AVRounding rnd)
Rescale a 64-bit integer by 2 rational numbers with specified rounding.
Definition: mathematics.c:134
PCMContext::out_pkt
AVPacket * out_pkt
Definition: pcm_rechunk_bsf.c:34
PCMContext::frame_rate
AVRational frame_rate
Definition: pcm_rechunk_bsf.c:31
drain_packet
static void drain_packet(AVPacket *pkt, int drain_data, int drain_samples)
Definition: pcm_rechunk_bsf.c:90