FFmpeg
westwood_audenc.c
Go to the documentation of this file.
1 /*
2  * Copyright (c) 2021 Aidan Richmond
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 /**
22  * @file
23  * Westwood Studios AUD file muxer
24  * by Aidan Richmond (aidan.is@hotmail.co.uk)
25  *
26  * This muxer supports IMA ADPCM packed in westwoods format.
27  *
28  * @see http://xhp.xwis.net/documents/aud3.txt
29  */
30 
31 #include "avformat.h"
32 #include "avio_internal.h"
33 #include "internal.h"
34 #include "mux.h"
35 #include <stdint.h>
36 
37 #define AUD_CHUNK_SIGNATURE 0x0000DEAF
38 
39 typedef struct AUDMuxContext {
41  int size;
43 
45 {
46  AVIOContext *pb = ctx->pb;
47 
48  /* Stream must be seekable to correctly write the file. */
49  if (!(pb->seekable & AVIO_SEEKABLE_NORMAL)) {
50  av_log(ctx, AV_LOG_ERROR, "Cannot write Westwood AUD to"
51  " non-seekable stream.\n");
52  return AVERROR(EINVAL);
53  }
54 
55  return 0;
56 }
57 
59 {
60  AVStream *st = ctx->streams[0];
61  AVIOContext *pb = ctx->pb;
63  unsigned char flags = 0;
64 
65  a->uncomp_size = 0;
66  a->size = 0;
67 
68  /* Flag if we have stereo data. */
69  if (st->codecpar->ch_layout.nb_channels == 2)
70  flags |= 1;
71 
72  /* This flags that the file contains 16 bit samples rather than 8 bit
73  since the encoder only encodes 16 bit samples this should be set. */
75  flags |= 2;
76 
77  avio_wl16(pb, st->codecpar->sample_rate);
78  /* We don't know the file size yet, so just zero 8 bytes */
79  ffio_fill(pb, 0, 8);
80  avio_w8(pb, flags);
81  /* 99 indicates the ADPCM format. Other formats not supported. */
82  avio_w8(pb, 99);
83 
84  return 0;
85 }
86 
88 {
89  AVIOContext *pb = ctx->pb;
91 
92  if (pkt->size > UINT16_MAX / 4)
93  return AVERROR_INVALIDDATA;
94  /* Assumes ADPCM since this muxer doesn't support SND1 or PCM format. */
95  avio_wl16(pb, pkt->size);
96  avio_wl16(pb, pkt->size * 4);
98  avio_write(pb, pkt->data, pkt->size);
99  a->size += pkt->size + 8;
100  a->uncomp_size += pkt->size * 4;
101 
102  return 0;
103 }
104 
106 {
107  AVIOContext *pb = ctx->pb;
109 
110  avio_seek(pb, 2, SEEK_SET);
111  avio_wl32(pb, a->size);
112  avio_wl32(pb, a->uncomp_size);
113 
114  return 0;
115 }
116 
118  .p.name = "wsaud",
119  .p.long_name = NULL_IF_CONFIG_SMALL("Westwood Studios audio"),
120  .p.extensions = "aud",
121  .priv_data_size = sizeof(AUDMuxContext),
122  .p.audio_codec = AV_CODEC_ID_ADPCM_IMA_WS,
123  .p.video_codec = AV_CODEC_ID_NONE,
124  .p.subtitle_codec = AV_CODEC_ID_NONE,
125  .init = wsaud_write_init,
126  .write_header = wsaud_write_header,
127  .write_packet = wsaud_write_packet,
128  .write_trailer = wsaud_write_trailer,
129  .flags_internal = FF_OFMT_FLAG_MAX_ONE_OF_EACH |
131 };
AVOutputFormat::name
const char * name
Definition: avformat.h:510
AUDMuxContext
Definition: westwood_audenc.c:39
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
AVFormatContext::streams
AVStream ** streams
A list of all streams in the file.
Definition: avformat.h:1368
AVPacket::data
uint8_t * data
Definition: packet.h:539
FF_OFMT_FLAG_ONLY_DEFAULT_CODECS
#define FF_OFMT_FLAG_ONLY_DEFAULT_CODECS
If this flag is set, then the only permitted audio/video/subtitle codec ids are AVOutputFormat....
Definition: mux.h:59
AVChannelLayout::nb_channels
int nb_channels
Number of channels in this layout.
Definition: channel_layout.h:321
FFOutputFormat::p
AVOutputFormat p
The public AVOutputFormat.
Definition: mux.h:65
wsaud_write_header
static int wsaud_write_header(AVFormatContext *ctx)
Definition: westwood_audenc.c:58
ff_wsaud_muxer
const FFOutputFormat ff_wsaud_muxer
Definition: westwood_audenc.c:117
avio_wl16
void avio_wl16(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:437
av_get_bits_per_sample
int av_get_bits_per_sample(enum AVCodecID codec_id)
Return codec bits per sample.
Definition: utils.c:550
pkt
AVPacket * pkt
Definition: movenc.c:60
AV_LOG_ERROR
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:209
wsaud_write_init
static int wsaud_write_init(AVFormatContext *ctx)
Definition: westwood_audenc.c:44
ctx
AVFormatContext * ctx
Definition: movenc.c:49
AUDMuxContext::uncomp_size
int uncomp_size
Definition: westwood_audenc.c:40
AUDMuxContext::size
int size
Definition: westwood_audenc.c:41
AVFormatContext
Format I/O context.
Definition: avformat.h:1300
internal.h
AVStream::codecpar
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:771
wsaud_write_packet
static int wsaud_write_packet(AVFormatContext *ctx, AVPacket *pkt)
Definition: westwood_audenc.c:87
AV_CODEC_ID_ADPCM_IMA_WS
@ AV_CODEC_ID_ADPCM_IMA_WS
Definition: codec_id.h:371
AVFormatContext::pb
AVIOContext * pb
I/O context.
Definition: avformat.h:1342
FFOutputFormat
Definition: mux.h:61
avio_w8
void avio_w8(AVIOContext *s, int b)
Definition: aviobuf.c:179
ffio_fill
void ffio_fill(AVIOContext *s, int b, int64_t count)
Definition: aviobuf.c:187
AVCodecParameters::ch_layout
AVChannelLayout ch_layout
Audio only.
Definition: codec_par.h:180
AVCodecParameters::sample_rate
int sample_rate
Audio only.
Definition: codec_par.h:184
AVIOContext
Bytestream IO Context.
Definition: avio.h:160
AVPacket::size
int size
Definition: packet.h:540
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
AVIOContext::seekable
int seekable
A combination of AVIO_SEEKABLE_ flags or 0 when the stream is not seekable.
Definition: avio.h:261
avio_write
void avio_write(AVIOContext *s, const unsigned char *buf, int size)
Definition: aviobuf.c:201
a
The reader does not expect b to be semantically here and if the code is changed by maybe adding a a division or other the signedness will almost certainly be mistaken To avoid this confusion a new type was SUINT is the C unsigned type but it holds a signed int to use the same example SUINT a
Definition: undefined.txt:41
avio_wl32
void avio_wl32(AVIOContext *s, unsigned int val)
Definition: aviobuf.c:357
AV_CODEC_ID_NONE
@ AV_CODEC_ID_NONE
Definition: codec_id.h:50
avio_internal.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
wsaud_write_trailer
static int wsaud_write_trailer(AVFormatContext *ctx)
Definition: westwood_audenc.c:105
AVStream
Stream structure.
Definition: avformat.h:748
avio_seek
int64_t avio_seek(AVIOContext *s, int64_t offset, int whence)
fseek() equivalent for AVIOContext.
Definition: aviobuf.c:231
avformat.h
AVIO_SEEKABLE_NORMAL
#define AVIO_SEEKABLE_NORMAL
Seeking works like for a local file.
Definition: avio.h:41
AVCodecParameters::codec_id
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:55
AVPacket
This structure stores compressed data.
Definition: packet.h:516
AUD_CHUNK_SIGNATURE
#define AUD_CHUNK_SIGNATURE
Definition: westwood_audenc.c:37
flags
#define flags(name, subs,...)
Definition: cbs_av1.c:482
av_log
#define av_log(a,...)
Definition: tableprint_vlc.h:27
AVERROR_INVALIDDATA
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:61
AVFormatContext::priv_data
void * priv_data
Format private data.
Definition: avformat.h:1328
mux.h