FFmpeg
wavenc.c
Go to the documentation of this file.
1 /*
2  * WAV muxer
3  * Copyright (c) 2001, 2002 Fabrice Bellard
4  *
5  * Sony Wave64 muxer
6  * Copyright (c) 2012 Paul B Mahol
7  *
8  * WAV muxer RF64 support
9  * Copyright (c) 2013 Daniel Verkamp <daniel@drv.nu>
10  *
11  * EBU Tech 3285 - Supplement 3 - Peak Envelope Chunk encoder
12  * Copyright (c) 2014 Georg Lippitsch <georg.lippitsch@gmx.at>
13  *
14  * This file is part of FFmpeg.
15  *
16  * FFmpeg is free software; you can redistribute it and/or
17  * modify it under the terms of the GNU Lesser General Public
18  * License as published by the Free Software Foundation; either
19  * version 2.1 of the License, or (at your option) any later version.
20  *
21  * FFmpeg is distributed in the hope that it will be useful,
22  * but WITHOUT ANY WARRANTY; without even the implied warranty of
23  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
24  * Lesser General Public License for more details.
25  *
26  * You should have received a copy of the GNU Lesser General Public
27  * License along with FFmpeg; if not, write to the Free Software
28  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
29  */
30 
31 #include "config_components.h"
32 
33 #include <stdint.h>
34 #include <string.h>
35 #include <time.h>
36 
37 #include "libavutil/avstring.h"
38 #include "libavutil/dict.h"
39 #include "libavutil/common.h"
40 #include "libavutil/intreadwrite.h"
41 #include "libavutil/mathematics.h"
42 #include "libavutil/mem.h"
43 #include "libavutil/opt.h"
44 #include "libavutil/time.h"
45 #include "libavutil/time_internal.h"
46 
47 #include "avformat.h"
48 #include "avio.h"
49 #include "avio_internal.h"
50 #include "internal.h"
51 #include "mux.h"
52 #include "riff.h"
53 
54 #define RF64_AUTO (-1)
55 #define RF64_NEVER 0
56 #define RF64_ALWAYS 1
57 
58 typedef enum {
59  PEAK_OFF = 0,
60  PEAK_ON,
61  PEAK_ONLY
62 } PeakType;
63 
64 typedef enum {
65  PEAK_FORMAT_UINT8 = 1,
66  PEAK_FORMAT_UINT16
67 } PeakFormat;
68 
69 typedef struct WAVMuxContext {
70  const AVClass *class;
71  int64_t data;
72  int64_t fact_pos;
73  int64_t ds64;
74  int64_t minpts;
75  int64_t maxpts;
76  int16_t *peak_maxpos, *peak_maxneg;
77  uint32_t peak_num_frames;
78  unsigned peak_outbuf_size;
79  uint32_t peak_outbuf_bytes;
80  unsigned size_increment;
81  uint8_t *peak_output;
82  int last_duration;
83  int write_bext;
84  int write_peak;
85  int rf64;
86  int peak_block_size;
87  int peak_format;
88  int peak_block_pos;
89  int peak_ppv;
90  int peak_bps;
91 } WAVMuxContext;
92 
93 #if CONFIG_WAV_MUXER
94 static inline void bwf_write_bext_string(AVFormatContext *s, const char *key, int maxlen)
95 {
96  AVDictionaryEntry *tag;
97  size_t len = 0;
98 
99  if (tag = av_dict_get(s->metadata, key, NULL, 0)) {
100  len = strlen(tag->value);
101  len = FFMIN(len, maxlen);
102  avio_write(s->pb, tag->value, len);
103  }
104 
105  ffio_fill(s->pb, 0, maxlen - len);
106 }
107 
108 static void bwf_write_bext_chunk(AVFormatContext *s)
109 {
110  AVDictionaryEntry *tmp_tag;
111  uint64_t time_reference = 0;
112  int64_t bext = ff_start_tag(s->pb, "bext");
113 
114  bwf_write_bext_string(s, "description", 256);
115  bwf_write_bext_string(s, "originator", 32);
116  bwf_write_bext_string(s, "originator_reference", 32);
117  bwf_write_bext_string(s, "origination_date", 10);
118  bwf_write_bext_string(s, "origination_time", 8);
119 
120  if (tmp_tag = av_dict_get(s->metadata, "time_reference", NULL, 0))
121  time_reference = strtoll(tmp_tag->value, NULL, 10);
122  avio_wl64(s->pb, time_reference);
123  avio_wl16(s->pb, 1); // set version to 1
124 
125  if ((tmp_tag = av_dict_get(s->metadata, "umid", NULL, 0)) && strlen(tmp_tag->value) > 2) {
126  unsigned char umidpart_str[17] = {0};
127  int64_t i;
128  uint64_t umidpart;
129  size_t len = strlen(tmp_tag->value+2);
130 
131  for (i = 0; i < len/16; i++) {
132  memcpy(umidpart_str, tmp_tag->value + 2 + (i*16), 16);
133  umidpart = strtoull(umidpart_str, NULL, 16);
134  avio_wb64(s->pb, umidpart);
135  }
136  ffio_fill(s->pb, 0, 64 - i*8);
137  } else
138  ffio_fill(s->pb, 0, 64); // zero UMID
139 
140  ffio_fill(s->pb, 0, 190); // Reserved
141 
142  if (tmp_tag = av_dict_get(s->metadata, "coding_history", NULL, 0))
143  avio_put_str(s->pb, tmp_tag->value);
144 
145  ff_end_tag(s->pb, bext);
146 }
147 
148 static av_cold void wav_deinit(AVFormatContext *s)
149 {
150  WAVMuxContext *wav = s->priv_data;
151 
152  av_freep(&wav->peak_maxpos);
153  av_freep(&wav->peak_maxneg);
154  av_freep(&wav->peak_output);
155 }
156 
157 static av_cold int peak_init_writer(AVFormatContext *s)
158 {
159  WAVMuxContext *wav = s->priv_data;
160  AVCodecParameters *par = s->streams[0]->codecpar;
161 
162  if (par->codec_id != AV_CODEC_ID_PCM_S8 &&
163  par->codec_id != AV_CODEC_ID_PCM_S16LE &&
164  par->codec_id != AV_CODEC_ID_PCM_U8 &&
165  par->codec_id != AV_CODEC_ID_PCM_U16LE) {
166  av_log(s, AV_LOG_ERROR, "Codec %s not supported for Peak Chunk\n",
167  avcodec_get_name(par->codec_id));
168  return -1;
169  }
170 
171  wav->peak_bps = av_get_bits_per_sample(par->codec_id) / 8;
172 
173  if (wav->peak_bps == 1 && wav->peak_format == PEAK_FORMAT_UINT16) {
174  av_log(s, AV_LOG_ERROR,
175  "Writing 16 bit peak for 8 bit audio does not make sense\n");
176  return AVERROR(EINVAL);
177  }
178  if (par->ch_layout.nb_channels > INT_MAX / (wav->peak_bps * wav->peak_ppv))
179  return AVERROR(ERANGE);
180  wav->size_increment = par->ch_layout.nb_channels * wav->peak_bps * wav->peak_ppv;
181 
182  wav->peak_maxpos = av_calloc(par->ch_layout.nb_channels, sizeof(*wav->peak_maxpos));
183  wav->peak_maxneg = av_calloc(par->ch_layout.nb_channels, sizeof(*wav->peak_maxneg));
184  if (!wav->peak_maxpos || !wav->peak_maxneg)
185  goto nomem;
186 
187  return 0;
188 
189 nomem:
190  av_log(s, AV_LOG_ERROR, "Out of memory\n");
191  return AVERROR(ENOMEM);
192 }
193 
194 static int peak_write_frame(AVFormatContext *s)
195 {
196  WAVMuxContext *wav = s->priv_data;
197  AVCodecParameters *par = s->streams[0]->codecpar;
198  unsigned new_size = wav->peak_outbuf_bytes + wav->size_increment;
199  uint8_t *tmp;
200  int c;
201 
202  if (new_size > INT_MAX) {
203  wav->write_peak = PEAK_OFF;
204  return AVERROR(ERANGE);
205  }
206  tmp = av_fast_realloc(wav->peak_output, &wav->peak_outbuf_size, new_size);
207  if (!tmp) {
208  wav->write_peak = PEAK_OFF;
209  return AVERROR(ENOMEM);
210  }
211  wav->peak_output = tmp;
212 
213  for (c = 0; c < par->ch_layout.nb_channels; c++) {
214  wav->peak_maxneg[c] = -wav->peak_maxneg[c];
215 
216  if (wav->peak_bps == 2 && wav->peak_format == PEAK_FORMAT_UINT8) {
217  wav->peak_maxpos[c] = wav->peak_maxpos[c] / 256;
218  wav->peak_maxneg[c] = wav->peak_maxneg[c] / 256;
219  }
220 
221  if (wav->peak_ppv == 1)
222  wav->peak_maxpos[c] =
223  FFMAX(wav->peak_maxpos[c], wav->peak_maxneg[c]);
224 
225  if (wav->peak_format == PEAK_FORMAT_UINT8) {
226  wav->peak_output[wav->peak_outbuf_bytes++] =
227  wav->peak_maxpos[c];
228  if (wav->peak_ppv == 2) {
229  wav->peak_output[wav->peak_outbuf_bytes++] =
230  wav->peak_maxneg[c];
231  }
232  } else {
233  AV_WL16(wav->peak_output + wav->peak_outbuf_bytes,
234  wav->peak_maxpos[c]);
235  wav->peak_outbuf_bytes += 2;
236  if (wav->peak_ppv == 2) {
237  AV_WL16(wav->peak_output + wav->peak_outbuf_bytes,
238  wav->peak_maxneg[c]);
239  wav->peak_outbuf_bytes += 2;
240  }
241  }
242  wav->peak_maxpos[c] = 0;
243  wav->peak_maxneg[c] = 0;
244  }
245  wav->peak_num_frames++;
246 
247  return 0;
248 }
249 
250 static int peak_write_chunk(AVFormatContext *s)
251 {
252  WAVMuxContext *wav = s->priv_data;
253  AVIOContext *pb = s->pb;
254  AVCodecParameters *par = s->streams[0]->codecpar;
255  int64_t peak = ff_start_tag(s->pb, "levl"); // codespell:ignore
256  int64_t now0;
257  time_t now_secs;
258  char timestamp[28];
259 
260  /* Peak frame of incomplete block at end */
261  if (wav->peak_block_pos) {
262  int ret = peak_write_frame(s);
263  if (ret < 0)
264  return ret;
265  }
266 
267  memset(timestamp, 0, sizeof(timestamp));
268  if (!(s->flags & AVFMT_FLAG_BITEXACT)) {
269  struct tm tmpbuf;
270  av_log(s, AV_LOG_INFO, "Writing local time and date to Peak Envelope Chunk\n");
271  now0 = av_gettime();
272  now_secs = now0 / 1000000;
273  if (strftime(timestamp, sizeof(timestamp), "%Y:%m:%d:%H:%M:%S:", localtime_r(&now_secs, &tmpbuf))) {
274  av_strlcatf(timestamp, sizeof(timestamp), "%03d", (int)((now0 / 1000) % 1000));
275  } else {
276  av_log(s, AV_LOG_ERROR, "Failed to write timestamp\n");
277  return -1;
278  }
279  }
280 
281  avio_wl32(pb, 1); /* version */
282  avio_wl32(pb, wav->peak_format); /* 8 or 16 bit */
283  avio_wl32(pb, wav->peak_ppv); /* positive and negative */
284  avio_wl32(pb, wav->peak_block_size); /* frames per value */
285  avio_wl32(pb, par->ch_layout.nb_channels); /* number of channels */
286  avio_wl32(pb, wav->peak_num_frames); /* number of peak frames */
287  avio_wl32(pb, -1); /* audio sample frame position (not implemented) */
288  avio_wl32(pb, 128); /* equal to size of header */
289  avio_write(pb, timestamp, 28); /* ASCII time stamp */
290  ffio_fill(pb, 0, 60);
291 
292  avio_write(pb, wav->peak_output, wav->peak_outbuf_bytes);
293 
294  ff_end_tag(pb, peak);
295 
296  if (!wav->data)
297  wav->data = peak;
298 
299  return 0;
300 }
301 
302 static int wav_write_header(AVFormatContext *s)
303 {
304  WAVMuxContext *wav = s->priv_data;
305  AVIOContext *pb = s->pb;
306  int64_t fmt;
307 
308  if (wav->rf64 == RF64_ALWAYS) {
309  ffio_wfourcc(pb, "RF64");
310  avio_wl32(pb, -1); /* RF64 chunk size: use size in ds64 */
311  } else {
312  ffio_wfourcc(pb, "RIFF");
313  avio_wl32(pb, -1); /* file length */
314  }
315 
316  ffio_wfourcc(pb, "WAVE");
317 
318  if (wav->rf64 == RF64_ALWAYS) {
319  /* write empty ds64 chunk */
320  ffio_wfourcc(pb, "ds64");
321  avio_wl32(pb, 28); /* chunk size */
322  wav->ds64 = avio_tell(pb);
323  ffio_fill(pb, 0, 28);
324  }
325 
326  if (wav->write_peak != PEAK_ONLY) {
327  /* format header */
328  fmt = ff_start_tag(pb, "fmt ");
329  if (ff_put_wav_header(s, pb, s->streams[0]->codecpar, 0) < 0) {
330  av_log(s, AV_LOG_ERROR, "Codec %s not supported in WAVE format\n",
331  avcodec_get_name(s->streams[0]->codecpar->codec_id));
332  return AVERROR(ENOSYS);
333  }
334  ff_end_tag(pb, fmt);
335 
336  if (wav->rf64 == RF64_AUTO) {
337  /* reserve space for ds64 */
338  ffio_wfourcc(pb, "JUNK");
339  avio_wl32(pb, 28); /* chunk size */
340  ffio_fill(pb, 0, 28);
341 
342  /* in RF64_AUTO mode, fmt + JUNK will be overwritten by ds64 + fmt */
343  wav->ds64 = fmt;
344  }
345  }
346 
347  if (s->streams[0]->codecpar->codec_tag != 0x01 /* hence for all other than PCM */
348  && (s->pb->seekable & AVIO_SEEKABLE_NORMAL)) {
349  wav->fact_pos = ff_start_tag(pb, "fact");
350  avio_wl32(pb, 0);
351  ff_end_tag(pb, wav->fact_pos);
352  }
353 
354  if (wav->write_bext)
355  bwf_write_bext_chunk(s);
356 
357  if (wav->write_peak) {
358  int ret;
359  if ((ret = peak_init_writer(s)) < 0)
360  return ret;
361  }
362 
363  avpriv_set_pts_info(s->streams[0], 64, 1, s->streams[0]->codecpar->sample_rate);
364  wav->maxpts = wav->last_duration = 0;
365  wav->minpts = INT64_MAX;
366 
367  if (wav->write_peak != PEAK_ONLY) {
368  /* info header */
369  ff_riff_write_info(s);
370 
371  /* data header */
372  wav->data = ff_start_tag(pb, "data");
373  }
374 
375  return 0;
376 }
377 
378 static int wav_write_packet(AVFormatContext *s, AVPacket *pkt)
379 {
380  AVIOContext *pb = s->pb;
381  WAVMuxContext *wav = s->priv_data;
382 
383  if (wav->write_peak != PEAK_ONLY)
384  avio_write(pb, pkt->data, pkt->size);
385 
386  if (wav->write_peak) {
387  int c = 0;
388  int i;
389  for (i = 0; i < pkt->size; i += wav->peak_bps) {
390  if (wav->peak_bps == 1) {
391  wav->peak_maxpos[c] = FFMAX(wav->peak_maxpos[c], *(int8_t*)(pkt->data + i));
392  wav->peak_maxneg[c] = FFMIN(wav->peak_maxneg[c], *(int8_t*)(pkt->data + i));
393  } else {
394  wav->peak_maxpos[c] = FFMAX(wav->peak_maxpos[c], (int16_t)AV_RL16(pkt->data + i));
395  wav->peak_maxneg[c] = FFMIN(wav->peak_maxneg[c], (int16_t)AV_RL16(pkt->data + i));
396  }
397  if (++c == s->streams[0]->codecpar->ch_layout.nb_channels) {
398  c = 0;
399  if (++wav->peak_block_pos == wav->peak_block_size) {
400  int ret = peak_write_frame(s);
401  if (ret < 0)
402  return ret;
403  wav->peak_block_pos = 0;
404  }
405  }
406  }
407  }
408 
409  if(pkt->pts != AV_NOPTS_VALUE) {
410  wav->minpts = FFMIN(wav->minpts, pkt->pts);
411  wav->maxpts = FFMAX(wav->maxpts, pkt->pts);
412  wav->last_duration = pkt->duration;
413  } else
414  av_log(s, AV_LOG_ERROR, "wav_write_packet: NOPTS\n");
415  return 0;
416 }
417 
418 static int wav_write_trailer(AVFormatContext *s)
419 {
420  AVIOContext *pb = s->pb;
421  WAVMuxContext *wav = s->priv_data;
422  int64_t file_size, data_size;
423  int64_t number_of_samples = 0;
424  int64_t pos;
425  int rf64 = 0;
426  int ret = 0;
427 
428  if (s->pb->seekable & AVIO_SEEKABLE_NORMAL) {
429  if (wav->write_peak != PEAK_ONLY && avio_tell(pb) - wav->data < UINT32_MAX) {
430  ff_end_tag(pb, wav->data);
431  }
432 
433  if (wav->write_peak && wav->peak_output) {
434  ret = peak_write_chunk(s);
435  }
436 
437  /* update file size */
438  file_size = avio_tell(pb);
439  data_size = file_size - wav->data;
440  if (wav->rf64 == RF64_ALWAYS || (wav->rf64 == RF64_AUTO && file_size - 8 > UINT32_MAX)) {
441  rf64 = 1;
442  } else if (file_size - 8 <= UINT32_MAX) {
443  avio_seek(pb, 4, SEEK_SET);
444  avio_wl32(pb, (uint32_t)(file_size - 8));
445  avio_seek(pb, file_size, SEEK_SET);
446  } else {
447  av_log(s, AV_LOG_ERROR,
448  "Filesize %"PRId64" invalid for wav, output file will be broken\n",
449  file_size);
450  }
451  number_of_samples = av_rescale_q(wav->maxpts - wav->minpts + wav->last_duration,
452  s->streams[0]->time_base,
453  av_make_q(1, s->streams[0]->codecpar->sample_rate));
454 
455  if(s->streams[0]->codecpar->codec_tag != 0x01) {
456  /* Update num_samps in fact chunk */
457  avio_seek(pb, wav->fact_pos, SEEK_SET);
458  if (rf64 || (wav->rf64 == RF64_AUTO && number_of_samples > UINT32_MAX)) {
459  rf64 = 1;
460  avio_wl32(pb, -1);
461  } else {
462  avio_wl32(pb, number_of_samples);
463  avio_seek(pb, file_size, SEEK_SET);
464  }
465  }
466 
467  if (rf64) {
468  /* overwrite RIFF with RF64 */
469  avio_seek(pb, 0, SEEK_SET);
470  ffio_wfourcc(pb, "RF64");
471  avio_wl32(pb, -1);
472 
473  /* write ds64 chunk (overwrite fmt + JUNK if rf64 == RF64_AUTO) */
474  avio_seek(pb, wav->ds64 - 8, SEEK_SET);
475  ffio_wfourcc(pb, "ds64");
476  avio_wl32(pb, 28); /* ds64 chunk size */
477  avio_wl64(pb, file_size - 8); /* RF64 chunk size */
478  avio_wl64(pb, data_size); /* data chunk size */
479  avio_wl64(pb, number_of_samples); /* fact chunk number of samples */
480  avio_wl32(pb, 0); /* number of table entries for non-'data' chunks */
481 
482  /* rewrite fmt in its RF64 position after ds64 */
483  pos = ff_start_tag(pb, "fmt ");
484  ret = ff_put_wav_header(s, pb, s->streams[0]->codecpar, 0);
485  ff_end_tag(pb, pos);
486 
487  /* write -1 in data chunk size */
488  avio_seek(pb, wav->data - 4, SEEK_SET);
489  avio_wl32(pb, -1);
490 
491  avio_seek(pb, file_size, SEEK_SET);
492  }
493  }
494 
495  return ret;
496 }
497 
498 #define OFFSET(x) offsetof(WAVMuxContext, x)
499 #define ENC AV_OPT_FLAG_ENCODING_PARAM
500 static const AVOption options[] = {
501  { "write_bext", "Write BEXT chunk.", OFFSET(write_bext), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, ENC },
502  { "write_peak", "Write Peak Envelope chunk.", OFFSET(write_peak), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 2, ENC, .unit = "peak" },
503  { "off", "Do not write peak chunk.", 0, AV_OPT_TYPE_CONST, { .i64 = PEAK_OFF }, 0, 0, ENC, .unit = "peak" },
504  { "on", "Append peak chunk after wav data.", 0, AV_OPT_TYPE_CONST, { .i64 = PEAK_ON }, 0, 0, ENC, .unit = "peak" },
505  { "only", "Write only peak chunk, omit wav data.", 0, AV_OPT_TYPE_CONST, { .i64 = PEAK_ONLY }, 0, 0, ENC, .unit = "peak" },
506  { "rf64", "Use RF64 header rather than RIFF for large files.", OFFSET(rf64), AV_OPT_TYPE_INT, { .i64 = RF64_NEVER },-1, 1, ENC, .unit = "rf64" },
507  { "auto", "Write RF64 header if file grows large enough.", 0, AV_OPT_TYPE_CONST, { .i64 = RF64_AUTO }, 0, 0, ENC, .unit = "rf64" },
508  { "always", "Always write RF64 header regardless of file size.", 0, AV_OPT_TYPE_CONST, { .i64 = RF64_ALWAYS }, 0, 0, ENC, .unit = "rf64" },
509  { "never", "Never write RF64 header regardless of file size.", 0, AV_OPT_TYPE_CONST, { .i64 = RF64_NEVER }, 0, 0, ENC, .unit = "rf64" },
510  { "peak_block_size", "Number of audio samples used to generate each peak frame.", OFFSET(peak_block_size), AV_OPT_TYPE_INT, { .i64 = 256 }, 0, 65536, ENC },
511  { "peak_format", "The format of the peak envelope data (1: uint8, 2: uint16).", OFFSET(peak_format), AV_OPT_TYPE_INT, { .i64 = PEAK_FORMAT_UINT16 }, PEAK_FORMAT_UINT8, PEAK_FORMAT_UINT16, ENC },
512  { "peak_ppv", "Number of peak points per peak value (1 or 2).", OFFSET(peak_ppv), AV_OPT_TYPE_INT, { .i64 = 2 }, 1, 2, ENC },
513  { NULL },
514 };
515 
516 static const AVClass wav_muxer_class = {
517  .class_name = "WAV muxer",
518  .item_name = av_default_item_name,
519  .option = options,
520  .version = LIBAVUTIL_VERSION_INT,
521 };
522 
523 const FFOutputFormat ff_wav_muxer = {
524  .p.name = "wav",
525  .p.long_name = NULL_IF_CONFIG_SMALL("WAV / WAVE (Waveform Audio)"),
526  .p.mime_type = "audio/x-wav",
527  .p.extensions = "wav",
528  .priv_data_size = sizeof(WAVMuxContext),
529  .p.audio_codec = AV_CODEC_ID_PCM_S16LE,
530  .p.video_codec = AV_CODEC_ID_NONE,
531  .p.subtitle_codec = AV_CODEC_ID_NONE,
532  .write_header = wav_write_header,
533  .write_packet = wav_write_packet,
534  .write_trailer = wav_write_trailer,
535  .deinit = wav_deinit,
536  .p.flags = AVFMT_TS_NONSTRICT,
537  .p.codec_tag = ff_wav_codec_tags_list,
538  .p.priv_class = &wav_muxer_class,
539  .flags_internal = FF_OFMT_FLAG_MAX_ONE_OF_EACH,
540 };
541 #endif /* CONFIG_WAV_MUXER */
542 
543 #if CONFIG_W64_MUXER
544 #include "w64.h"
545 
546 static void start_guid(AVIOContext *pb, const uint8_t *guid, int64_t *pos)
547 {
548  *pos = avio_tell(pb);
549 
550  avio_write(pb, guid, 16);
551  avio_wl64(pb, INT64_MAX);
552 }
553 
554 static void end_guid(AVIOContext *pb, int64_t start)
555 {
556  int64_t end, pos = avio_tell(pb);
557 
558  end = FFALIGN(pos, 8);
559  ffio_fill(pb, 0, end - pos);
560  avio_seek(pb, start + 16, SEEK_SET);
561  avio_wl64(pb, end - start);
562  avio_seek(pb, end, SEEK_SET);
563 }
564 
565 static int w64_write_header(AVFormatContext *s)
566 {
567  WAVMuxContext *wav = s->priv_data;
568  AVIOContext *pb = s->pb;
569  int64_t start;
570  int ret;
571 
572  avio_write(pb, ff_w64_guid_riff, sizeof(ff_w64_guid_riff));
573  avio_wl64(pb, -1);
574  avio_write(pb, ff_w64_guid_wave, sizeof(ff_w64_guid_wave));
575  start_guid(pb, ff_w64_guid_fmt, &start);
576  if ((ret = ff_put_wav_header(s, pb, s->streams[0]->codecpar, 0)) < 0) {
577  av_log(s, AV_LOG_ERROR, "Codec %s not supported\n",
578  avcodec_get_name(s->streams[0]->codecpar->codec_id));
579  return ret;
580  }
581  end_guid(pb, start);
582 
583  if (s->streams[0]->codecpar->codec_tag != 0x01 /* hence for all other than PCM */
584  && (s->pb->seekable & AVIO_SEEKABLE_NORMAL)) {
585  start_guid(pb, ff_w64_guid_fact, &wav->fact_pos);
586  avio_wl64(pb, 0);
587  end_guid(pb, wav->fact_pos);
588  }
589 
590  start_guid(pb, ff_w64_guid_data, &wav->data);
591 
592  return 0;
593 }
594 
595 static int w64_write_trailer(AVFormatContext *s)
596 {
597  AVIOContext *pb = s->pb;
598  WAVMuxContext *wav = s->priv_data;
599  int64_t file_size;
600 
601  if (pb->seekable & AVIO_SEEKABLE_NORMAL) {
602  end_guid(pb, wav->data);
603 
604  file_size = avio_tell(pb);
605  avio_seek(pb, 16, SEEK_SET);
606  avio_wl64(pb, file_size);
607 
608  if (s->streams[0]->codecpar->codec_tag != 0x01) {
609  int64_t number_of_samples;
610 
611  number_of_samples = av_rescale(wav->maxpts - wav->minpts + wav->last_duration,
612  s->streams[0]->codecpar->sample_rate * (int64_t)s->streams[0]->time_base.num,
613  s->streams[0]->time_base.den);
614  avio_seek(pb, wav->fact_pos + 24, SEEK_SET);
615  avio_wl64(pb, number_of_samples);
616  }
617 
618  avio_seek(pb, file_size, SEEK_SET);
619  }
620 
621  return 0;
622 }
623 
624 const FFOutputFormat ff_w64_muxer = {
625  .p.name = "w64",
626  .p.long_name = NULL_IF_CONFIG_SMALL("Sony Wave64"),
627  .p.extensions = "w64",
628  .priv_data_size = sizeof(WAVMuxContext),
629  .p.audio_codec = AV_CODEC_ID_PCM_S16LE,
630  .p.video_codec = AV_CODEC_ID_NONE,
631  .p.subtitle_codec = AV_CODEC_ID_NONE,
632  .write_header = w64_write_header,
633  .write_packet = wav_write_packet,
634  .write_trailer = w64_write_trailer,
635  .p.flags = AVFMT_TS_NONSTRICT,
636  .p.codec_tag = ff_wav_codec_tags_list,
637  .flags_internal = FF_OFMT_FLAG_MAX_ONE_OF_EACH,
638 };
639 #endif /* CONFIG_W64_MUXER */
AV_CODEC_ID_PCM_S16LE
@ AV_CODEC_ID_PCM_S16LE
Definition: codec_id.h:338
WAVMuxContext::peak_output
uint8_t * peak_output
Definition: wavenc.c:81
AVOutputFormat::name
const char * name
Definition: avformat.h:506
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
WAVMuxContext::peak_maxneg
int16_t * peak_maxneg
Definition: wavenc.c:76
ffio_wfourcc
static av_always_inline void ffio_wfourcc(AVIOContext *pb, const uint8_t *s)
Definition: avio_internal.h:124
AVCodecParameters
This struct describes the properties of an encoded stream.
Definition: codec_par.h:49
WAVMuxContext::maxpts
int64_t maxpts
Definition: wavenc.c:75
int64_t
long long int64_t
Definition: coverity.c:34
AVPacket::data
uint8_t * data
Definition: packet.h:595
AVOption
AVOption.
Definition: opt.h:429
PeakFormat
PeakFormat
Definition: wavenc.c:64
avio_wl64
void avio_wl64(AVIOContext *s, uint64_t val)
Definition: aviobuf.c:428
AVPacket::duration
int64_t duration
Duration of this packet in AVStream->time_base units, 0 if unknown.
Definition: packet.h:613
mathematics.h
FFMAX
#define FFMAX(a, b)
Definition: macros.h:47
AVChannelLayout::nb_channels
int nb_channels
Number of channels in this layout.
Definition: channel_layout.h:329
WAVMuxContext::peak_outbuf_bytes
uint32_t peak_outbuf_bytes
Definition: wavenc.c:79
WAVMuxContext::fact_pos
int64_t fact_pos
Definition: wavenc.c:72
av_strlcatf
size_t av_strlcatf(char *dst, size_t size, const char *fmt,...)
Definition: avstring.c:103
ff_wav_muxer
const FFOutputFormat ff_wav_muxer
FFOutputFormat::p
AVOutputFormat p
The public AVOutputFormat.
Definition: mux.h:65
PEAK_ONLY
@ PEAK_ONLY
Definition: wavenc.c:61
avio_wl16
void avio_wl16(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:440
avpriv_set_pts_info
void avpriv_set_pts_info(AVStream *st, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: avformat.c:781
WAVMuxContext::peak_bps
int peak_bps
Definition: wavenc.c:90
avio_tell
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:494
RF64_AUTO
#define RF64_AUTO
Definition: wavenc.c:54
ENC
#define ENC
Definition: caca.c:198
PEAK_FORMAT_UINT16
@ PEAK_FORMAT_UINT16
Definition: wavenc.c:66
ff_start_tag
int64_t ff_start_tag(AVIOContext *pb, const char *tag)
Definition: riffenc.c:31
av_get_bits_per_sample
int av_get_bits_per_sample(enum AVCodecID codec_id)
Return codec bits per sample.
Definition: utils.c:549
AV_CODEC_ID_PCM_S8
@ AV_CODEC_ID_PCM_S8
Definition: codec_id.h:342
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:210
av_cold
#define av_cold
Definition: attributes.h:119
av_dict_get
AVDictionaryEntry * av_dict_get(const AVDictionary *m, const char *key, const AVDictionaryEntry *prev, int flags)
Get a dictionary entry with matching key.
Definition: dict.c:60
av_fast_realloc
void * av_fast_realloc(void *ptr, unsigned int *size, size_t min_size)
Reallocate the given buffer if it is not large enough, otherwise do nothing.
Definition: mem.c:497
intreadwrite.h
s
#define s(width, name)
Definition: cbs_vp9.c:198
RF64_NEVER
#define RF64_NEVER
Definition: wavenc.c:55
PEAK_FORMAT_UINT8
@ PEAK_FORMAT_UINT8
Definition: wavenc.c:65
PEAK_ON
@ PEAK_ON
Definition: wavenc.c:60
AV_RL16
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_RL16
Definition: bytestream.h:94
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
RF64_ALWAYS
#define RF64_ALWAYS
Definition: wavenc.c:56
key
const char * key
Definition: hwcontext_opencl.c:189
WAVMuxContext::data
int64_t data
Definition: wavenc.c:71
tmp
static uint8_t tmp[40]
Definition: aes_ctr.c:52
time_internal.h
WAVMuxContext::rf64
int rf64
Definition: wavenc.c:85
AVFormatContext
Format I/O context.
Definition: avformat.h:1263
internal.h
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
PEAK_OFF
@ PEAK_OFF
Definition: wavenc.c:59
AVClass
Describe the class of an AVClass context structure.
Definition: log.h:76
WAVMuxContext::last_duration
int last_duration
Definition: wavenc.c:82
NULL
#define NULL
Definition: coverity.c:32
ff_put_wav_header
int ff_put_wav_header(AVFormatContext *s, AVIOContext *pb, AVCodecParameters *par, int flags)
Write WAVEFORMAT header structure.
Definition: riffenc.c:54
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:242
options
Definition: swscale.c:45
FFOutputFormat
Definition: mux.h:61
time.h
WAVMuxContext::peak_block_pos
int peak_block_pos
Definition: wavenc.c:88
ffio_fill
void ffio_fill(AVIOContext *s, int b, int64_t count)
Definition: aviobuf.c:192
AVCodecParameters::ch_layout
AVChannelLayout ch_layout
The channel layout and number of channels.
Definition: codec_par.h:207
c
Undefined Behavior In the C some operations are like signed integer dereferencing freed accessing outside allocated Undefined Behavior must not occur in a C it is not safe even if the output of undefined operations is unused The unsafety may seem nit picking but Optimizing compilers have in fact optimized code on the assumption that no undefined Behavior occurs Optimizing code based on wrong assumptions can and has in some cases lead to effects beyond the output of computations The signed integer overflow problem in speed critical code Code which is highly optimized and works with signed integers sometimes has the problem that often the output of the computation does not c
Definition: undefined.txt:32
ff_w64_guid_fmt
const uint8_t ff_w64_guid_fmt[16]
Definition: w64.c:33
ff_w64_muxer
const FFOutputFormat ff_w64_muxer
options
const OptionDef options[]
AVIOContext
Bytestream IO Context.
Definition: avio.h:160
AVPacket::size
int size
Definition: packet.h:596
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:94
WAVMuxContext::write_bext
int write_bext
Definition: wavenc.c:83
PeakType
PeakType
Definition: wavenc.c:58
AVIOContext::seekable
int seekable
A combination of AVIO_SEEKABLE_ flags or 0 when the stream is not seekable.
Definition: avio.h:261
localtime_r
#define localtime_r
Definition: time_internal.h:46
i
#define i(width, name, range_min, range_max)
Definition: cbs_h264.c:63
av_make_q
static AVRational av_make_q(int num, int den)
Create an AVRational.
Definition: rational.h:71
avio.h
AV_NOPTS_VALUE
#define AV_NOPTS_VALUE
Undefined timestamp value.
Definition: avutil.h:247
WAVMuxContext::minpts
int64_t minpts
Definition: wavenc.c:74
avio_write
void avio_write(AVIOContext *s, const unsigned char *buf, int size)
Definition: aviobuf.c:206
AV_WL16
#define AV_WL16(p, v)
Definition: intreadwrite.h:408
avio_wl32
void avio_wl32(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:360
WAVMuxContext
Definition: wavenc.c:69
ff_end_tag
void ff_end_tag(AVIOContext *pb, int64_t start)
Definition: riffenc.c:38
AV_LOG_INFO
#define AV_LOG_INFO
Standard information.
Definition: log.h:221
WAVMuxContext::write_peak
int write_peak
Definition: wavenc.c:84
avcodec_get_name
const char * avcodec_get_name(enum AVCodecID id)
Get the name of a codec.
Definition: utils.c:406
WAVMuxContext::peak_block_size
int peak_block_size
Definition: wavenc.c:86
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:588
avio_internal.h
ff_w64_guid_wave
const uint8_t ff_w64_guid_wave[16]
Definition: w64.c:28
common.h
FF_OFMT_FLAG_MAX_ONE_OF_EACH
#define FF_OFMT_FLAG_MAX_ONE_OF_EACH
If this flag is set, it indicates that for each codec type whose corresponding default codec (i....
Definition: mux.h:50
FFMIN
#define FFMIN(a, b)
Definition: macros.h:49
WAVMuxContext::peak_maxpos
int16_t * peak_maxpos
Definition: wavenc.c:76
ff_w64_guid_fact
const uint8_t ff_w64_guid_fact[16]
Definition: w64.c:38
len
int len
Definition: vorbis_enc_data.h:426
av_rescale
int64_t av_rescale(int64_t a, int64_t b, int64_t c)
Rescale a 64-bit integer with rounding to nearest.
Definition: mathematics.c:129
av_calloc
void * av_calloc(size_t nmemb, size_t size)
Definition: mem.c:264
AVFMT_TS_NONSTRICT
#define AVFMT_TS_NONSTRICT
Format does not require strictly increasing timestamps, but they must still be monotonic.
Definition: avformat.h:487
WAVMuxContext::peak_ppv
int peak_ppv
Definition: wavenc.c:89
tag
uint32_t tag
Definition: movenc.c:2046
AVFMT_FLAG_BITEXACT
#define AVFMT_FLAG_BITEXACT
When muxing, try to avoid writing any random/volatile data to the output.
Definition: avformat.h:1431
ret
ret
Definition: filter_design.txt:187
avio_seek
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:236
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:81
WAVMuxContext::peak_outbuf_size
unsigned peak_outbuf_size
Definition: wavenc.c:78
pos
unsigned int pos
Definition: spdifenc.c:414
avformat.h
dict.h
OFFSET
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf default minimum maximum flags name is the option keep it simple and lowercase description are in without and describe what they for example set the foo of the bar offset is the offset of the field in your see the OFFSET() macro
w64.h
AVIO_SEEKABLE_NORMAL
#define AVIO_SEEKABLE_NORMAL
Seeking works like for a local file.
Definition: avio.h:41
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Underlying C type is int.
Definition: opt.h:259
ff_riff_write_info
void ff_riff_write_info(AVFormatContext *s)
Write all recognized RIFF tags from s->metadata.
Definition: riffenc.c:349
WAVMuxContext::peak_format
int peak_format
Definition: wavenc.c:87
Windows::Graphics::DirectX::Direct3D11::p
IDirect3DDxgiInterfaceAccess _COM_Outptr_ void ** p
Definition: vsrc_gfxcapture_winrt.hpp:53
WAVMuxContext::ds64
int64_t ds64
Definition: wavenc.c:73
avio_wb64
void avio_wb64(AVIOContext *s, uint64_t val)
Definition: aviobuf.c:434
av_gettime
int64_t av_gettime(void)
Get the current time in microseconds.
Definition: time.c:39
ff_w64_guid_data
const uint8_t ff_w64_guid_data[16]
Definition: w64.c:42
mem.h
AV_CODEC_ID_PCM_U8
@ AV_CODEC_ID_PCM_U8
Definition: codec_id.h:343
ff_w64_guid_riff
const uint8_t ff_w64_guid_riff[16]
Definition: w64.c:23
AVDictionaryEntry
Definition: dict.h:90
FFALIGN
#define FFALIGN(x, a)
Definition: macros.h:78
AVCodecParameters::codec_id
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:57
AVPacket
This structure stores compressed data.
Definition: packet.h:572
AV_OPT_TYPE_BOOL
@ AV_OPT_TYPE_BOOL
Underlying C type is int.
Definition: opt.h:327
av_freep
#define av_freep(p)
Definition: tableprint_vlc.h:35
riff.h
AV_CODEC_ID_PCM_U16LE
@ AV_CODEC_ID_PCM_U16LE
Definition: codec_id.h:340
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
AVDictionaryEntry::value
char * value
Definition: dict.h:92
pkt
static AVPacket * pkt
Definition: demux_decode.c:55
WAVMuxContext::size_increment
unsigned size_increment
Definition: wavenc.c:80
avstring.h
avio_put_str
int avio_put_str(AVIOContext *s, const char *str)
Write a NULL-terminated string.
Definition: aviobuf.c:376
AV_OPT_TYPE_CONST
@ AV_OPT_TYPE_CONST
Special option type for declaring named constants.
Definition: opt.h:299
ff_wav_codec_tags_list
const AVCodecTag *const ff_wav_codec_tags_list[]
WAVMuxContext::peak_num_frames
uint32_t peak_num_frames
Definition: wavenc.c:77
mux.h
Generated on Sun May 3 2026 19:23:22 for FFmpeg by   doxygen 1.8.17