FFmpeg
ttaenc.c
Go to the documentation of this file.
1 /*
2  * True Audio (TTA) muxer
3  * Copyright (c) 2016 James Almer
4  *
5  * This file is part of FFmpeg.
6  *
7  * FFmpeg is free software; you can redistribute it and/or
8  * modify it under the terms of the GNU Lesser General Public
9  * License as published by the Free Software Foundation; either
10  * version 2.1 of the License, or (at your option) any later version.
11  *
12  * FFmpeg is distributed in the hope that it will be useful,
13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * Lesser General Public License for more details.
16  *
17  * You should have received a copy of the GNU Lesser General Public
18  * License along with FFmpeg; if not, write to the Free Software
19  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
20  */
21 
22 #include "libavutil/crc.h"
23 #include "libavutil/intreadwrite.h"
24 
25 #include "apetag.h"
26 #include "avformat.h"
27 #include "avio_internal.h"
28 #include "internal.h"
29 
30 typedef struct TTAMuxContext {
33  uint32_t nb_samples;
37 
39 {
40  TTAMuxContext *tta = s->priv_data;
41  AVCodecParameters *par;
42 
43  if (s->nb_streams != 1) {
44  av_log(s, AV_LOG_ERROR, "Only one stream is supported\n");
45  return AVERROR(EINVAL);
46  }
47  par = s->streams[0]->codecpar;
48 
49  if (par->codec_id != AV_CODEC_ID_TTA) {
50  av_log(s, AV_LOG_ERROR, "Unsupported codec\n");
51  return AVERROR(EINVAL);
52  }
53  if (par->extradata && par->extradata_size < 22) {
54  av_log(s, AV_LOG_ERROR, "Invalid TTA extradata\n");
55  return AVERROR_INVALIDDATA;
56  }
57 
58  /* Prevent overflow */
59  if (par->sample_rate > 0x7FFFFFu) {
60  av_log(s, AV_LOG_ERROR, "Sample rate too large\n");
61  return AVERROR(EINVAL);
62  }
63  tta->frame_size = par->sample_rate * 256 / 245;
64  avpriv_set_pts_info(s->streams[0], 64, 1, par->sample_rate);
65 
66  return 0;
67 }
68 
70 {
71  TTAMuxContext *tta = s->priv_data;
72  AVCodecParameters *par = s->streams[0]->codecpar;
73  int ret;
74 
75  if ((ret = avio_open_dyn_buf(&tta->seek_table)) < 0)
76  return ret;
77 
78  /* Ignore most extradata information if present. It can be innacurate
79  if for example remuxing from Matroska */
82  avio_write(s->pb, "TTA1", 4);
83  avio_wl16(s->pb, par->extradata ? AV_RL16(par->extradata + 4) : 1);
84  avio_wl16(s->pb, par->channels);
86  avio_wl32(s->pb, par->sample_rate);
87 
88  return 0;
89 }
90 
92 {
93  TTAMuxContext *tta = s->priv_data;
94  int ret;
95 
96  ret = ff_packet_list_put(&tta->queue, &tta->queue_end, pkt,
98  if (ret < 0) {
99  return ret;
100  }
101 
102  avio_wl32(tta->seek_table, pkt->size);
103  tta->nb_samples += pkt->duration;
104 
105  if (tta->frame_size != pkt->duration) {
106  if (tta->last_frame) {
107  /* Two frames with a different duration than the default frame
108  size means the TTA stream comes from a faulty container, and
109  there's no way the last frame duration will be correct. */
110  av_log(s, AV_LOG_ERROR, "Invalid frame durations\n");
111 
112  return AVERROR_INVALIDDATA;
113  }
114  /* First frame with a different duration than the default frame size.
115  Assume it's the last frame in the stream and continue. */
116  tta->last_frame++;
117  }
118 
119  return 0;
120 }
121 
123 {
124  TTAMuxContext *tta = s->priv_data;
125  AVPacket pkt;
126 
127  while (tta->queue) {
128  ff_packet_list_get(&tta->queue, &tta->queue_end, &pkt);
129  avio_write(s->pb, pkt.data, pkt.size);
130  av_packet_unref(&pkt);
131  }
132 }
133 
135 {
136  TTAMuxContext *tta = s->priv_data;
137  uint8_t *ptr;
138  unsigned int crc;
139  int size;
140 
141  avio_wl32(s->pb, tta->nb_samples);
142  crc = ffio_get_checksum(s->pb) ^ UINT32_MAX;
143  avio_wl32(s->pb, crc);
144 
145  /* Write Seek table */
146  crc = ffio_get_checksum(tta->seek_table) ^ UINT32_MAX;
147  avio_wl32(tta->seek_table, crc);
148  size = avio_close_dyn_buf(tta->seek_table, &ptr);
149  avio_write(s->pb, ptr, size);
150  av_free(ptr);
151 
152  /* Write audio data */
153  tta_queue_flush(s);
154 
155  ff_ape_write_tag(s);
156  avio_flush(s->pb);
157 
158  return 0;
159 }
160 
162  .name = "tta",
163  .long_name = NULL_IF_CONFIG_SMALL("TTA (True Audio)"),
164  .mime_type = "audio/x-tta",
165  .extensions = "tta",
166  .priv_data_size = sizeof(TTAMuxContext),
167  .audio_codec = AV_CODEC_ID_TTA,
168  .video_codec = AV_CODEC_ID_NONE,
169  .init = tta_init,
173 };
static void write_packet(OutputFile *of, AVPacket *pkt, OutputStream *ost, int unqueue)
Definition: ffmpeg.c:689
void avio_wl16(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:472
Bytestream IO Context.
Definition: avio.h:161
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
int avio_close_dyn_buf(AVIOContext *s, uint8_t **pbuffer)
Return the written size and a pointer to the buffer.
Definition: aviobuf.c:1459
int ff_packet_list_put(AVPacketList **head, AVPacketList **tail, AVPacket *pkt, int flags)
Append an AVPacket to the list.
Definition: utils.c:447
void avpriv_set_pts_info(AVStream *s, int pts_wrap_bits, unsigned int pts_num, unsigned int pts_den)
Set the time base and wrapping info for a given stream.
Definition: utils.c:4892
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
static int tta_write_packet(AVFormatContext *s, AVPacket *pkt)
Definition: ttaenc.c:91
AVPacketList * queue
Definition: ttaenc.c:32
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: avcodec.h:3957
int size
Definition: avcodec.h:1478
static AVPacket pkt
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_WL32 unsigned int_TMPL AV_WL24 unsigned int_TMPL AV_RL16
Definition: bytestream.h:87
int avio_open_dyn_buf(AVIOContext **s)
Open a write only memory stream.
Definition: aviobuf.c:1430
This struct describes the properties of an encoded stream.
Definition: avcodec.h:3949
AVOutputFormat ff_tta_muxer
Definition: ttaenc.c:161
Format I/O context.
Definition: avformat.h:1358
static int tta_write_trailer(AVFormatContext *s)
Definition: ttaenc.c:134
void avio_wl32(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:372
uint8_t
int64_t duration
Duration of this packet in AVStream->time_base units, 0 if unknown.
Definition: avcodec.h:1495
unsigned long ff_crcEDB88320_update(unsigned long checksum, const uint8_t *buf, unsigned int len)
Definition: aviobuf.c:609
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1426
#define FF_PACKETLIST_FLAG_REF_PACKET
Create a new reference for the packet instead of transferring the ownership of the existing one to th...
Definition: internal.h:757
Public header for CRC hash function implementation.
uint8_t * data
Definition: avcodec.h:1477
ptrdiff_t size
Definition: opengl_enc.c:100
void avio_write(AVIOContext *s, const unsigned char *buf, int size)
Definition: aviobuf.c:218
#define av_log(a,...)
static int tta_init(AVFormatContext *s)
Definition: ttaenc.c:38
AVPacketList * queue_end
Definition: ttaenc.c:32
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
static void tta_queue_flush(AVFormatContext *s)
Definition: ttaenc.c:122
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:186
int extradata_size
Size of the extradata content in bytes.
Definition: avcodec.h:3975
unsigned int nb_streams
Number of elements in AVFormatContext.streams.
Definition: avformat.h:1414
void ffio_init_checksum(AVIOContext *s, unsigned long(*update_checksum)(unsigned long c, const uint8_t *p, unsigned int len), unsigned long checksum)
Definition: aviobuf.c:629
static int tta_write_header(AVFormatContext *s)
Definition: ttaenc.c:69
AVIOContext * seek_table
Definition: ttaenc.c:31
static int write_trailer(AVFormatContext *s1)
Definition: v4l2enc.c:94
const char * name
Definition: avformat.h:505
#define s(width, name)
Definition: cbs_vp9.c:257
AVIOContext * pb
I/O context.
Definition: avformat.h:1400
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:599
uint32_t nb_samples
Definition: ttaenc.c:33
int ff_ape_write_tag(AVFormatContext *s)
Write an APE tag into a file.
Definition: apetag.c:185
unsigned long ffio_get_checksum(AVIOContext *s)
Definition: aviobuf.c:621
void avio_flush(AVIOContext *s)
Force flushing of buffered data.
Definition: aviobuf.c:238
int bits_per_raw_sample
This is the number of valid bits in each output sample.
Definition: avcodec.h:4012
int sample_rate
Audio only.
Definition: avcodec.h:4067
int ff_packet_list_get(AVPacketList **head, AVPacketList **tail, AVPacket *pkt)
Remove the oldest AVPacket in the list and return it.
Definition: utils.c:1550
Main libavformat public API header.
#define av_free(p)
void * priv_data
Format private data.
Definition: avformat.h:1386
static void write_header(FFV1Context *f)
Definition: ffv1enc.c:337
int last_frame
Definition: ttaenc.c:35
uint8_t * extradata
Extra binary data needed for initializing the decoder, codec-dependent.
Definition: avcodec.h:3971
int channels
Audio only.
Definition: avcodec.h:4063
int frame_size
Definition: ttaenc.c:34
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1028
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
This structure stores compressed data.
Definition: avcodec.h:1454