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  tta->seek_table = NULL;
151  av_free(ptr);
152 
153  /* Write audio data */
154  tta_queue_flush(s);
155 
156  ff_ape_write_tag(s);
157  avio_flush(s->pb);
158 
159  return 0;
160 }
161 
163 {
164  TTAMuxContext *tta = s->priv_data;
165 
167  ff_packet_list_free(&tta->queue, &tta->queue_end);
168 }
169 
171  .name = "tta",
172  .long_name = NULL_IF_CONFIG_SMALL("TTA (True Audio)"),
173  .mime_type = "audio/x-tta",
174  .extensions = "tta",
175  .priv_data_size = sizeof(TTAMuxContext),
176  .audio_codec = AV_CODEC_ID_TTA,
177  .video_codec = AV_CODEC_ID_NONE,
178  .init = tta_init,
179  .deinit = tta_deinit,
183 };
static void write_packet(OutputFile *of, AVPacket *pkt, OutputStream *ost, int unqueue)
Definition: ffmpeg.c:690
#define NULL
Definition: coverity.c:32
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:4929
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:3968
int size
Definition: avcodec.h:1481
static AVPacket pkt
static void tta_deinit(AVFormatContext *s)
Definition: ttaenc.c:162
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:3960
AVOutputFormat ff_tta_muxer
Definition: ttaenc.c:170
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:1498
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:755
Public header for CRC hash function implementation.
uint8_t * data
Definition: avcodec.h:1480
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:3986
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
void ffio_free_dyn_buf(AVIOContext **s)
Free a dynamic buffer.
Definition: aviobuf.c:1489
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
void ff_packet_list_free(AVPacketList **head, AVPacketList **tail)
Wipe the list and unref all the packets in it.
Definition: utils.c:1427
int bits_per_raw_sample
This is the number of valid bits in each output sample.
Definition: avcodec.h:4023
int sample_rate
Audio only.
Definition: avcodec.h:4078
int ff_packet_list_get(AVPacketList **head, AVPacketList **tail, AVPacket *pkt)
Remove the oldest AVPacket in the list and return it.
Definition: utils.c:1557
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:349
int last_frame
Definition: ttaenc.c:35
uint8_t * extradata
Extra binary data needed for initializing the decoder, codec-dependent.
Definition: avcodec.h:3982
int channels
Audio only.
Definition: avcodec.h:4074
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:1457