FFmpeg
 All Data Structures Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Groups Pages
avs.c
Go to the documentation of this file.
1 /*
2  * AVS demuxer.
3  * Copyright (c) 2006 Aurelien Jacobs <aurel@gnuage.org>
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 /**
23  * @file
24  * Argonaut Games' Creature Shock demuxer
25  * @see http://wiki.multimedia.cx/index.php?title=AVS
26  */
27 
28 #include "avformat.h"
29 #include "voc.h"
30 
31 
32 typedef struct avs_format {
36  int width;
37  int height;
39  int fps;
40  int nb_frames;
43 } AvsFormat;
44 
45 typedef enum avs_block_type {
46  AVS_NONE = 0x00,
47  AVS_VIDEO = 0x01,
48  AVS_AUDIO = 0x02,
49  AVS_PALETTE = 0x03,
50  AVS_GAME_DATA = 0x04,
51 } AvsBlockType;
52 
53 static int avs_probe(AVProbeData * p)
54 {
55  const uint8_t *d;
56 
57  d = p->buf;
58  if (d[0] == 'w' && d[1] == 'W' && d[2] == 0x10 && d[3] == 0)
59  /* Ensure the buffer probe scores higher than the extension probe.
60  * This avoids problems with misdetection as AviSynth scripts. */
61  return AVPROBE_SCORE_EXTENSION + 5;
62 
63  return 0;
64 }
65 
67 {
68  AvsFormat *avs = s->priv_data;
69 
71 
72  avio_skip(s->pb, 4);
73  avs->width = avio_rl16(s->pb);
74  avs->height = avio_rl16(s->pb);
75  avs->bits_per_sample = avio_rl16(s->pb);
76  avs->fps = avio_rl16(s->pb);
77  avs->nb_frames = avio_rl32(s->pb);
78  avs->remaining_frame_size = 0;
79  avs->remaining_audio_size = 0;
80 
81  avs->st_video = avs->st_audio = NULL;
82 
83  if (avs->width != 318 || avs->height != 198)
84  av_log(s, AV_LOG_ERROR, "This avs pretend to be %dx%d "
85  "when the avs format is supposed to be 318x198 only.\n",
86  avs->width, avs->height);
87 
88  return 0;
89 }
90 
91 static int
93  AvsBlockType type, int sub_type, int size,
94  uint8_t * palette, int palette_size)
95 {
96  AvsFormat *avs = s->priv_data;
97  int ret;
98 
99  ret = av_new_packet(pkt, size + palette_size);
100  if (ret < 0)
101  return ret;
102 
103  if (palette_size) {
104  pkt->data[0] = 0x00;
105  pkt->data[1] = 0x03;
106  pkt->data[2] = palette_size & 0xFF;
107  pkt->data[3] = (palette_size >> 8) & 0xFF;
108  memcpy(pkt->data + 4, palette, palette_size - 4);
109  }
110 
111  pkt->data[palette_size + 0] = sub_type;
112  pkt->data[palette_size + 1] = type;
113  pkt->data[palette_size + 2] = size & 0xFF;
114  pkt->data[palette_size + 3] = (size >> 8) & 0xFF;
115  ret = avio_read(s->pb, pkt->data + palette_size + 4, size - 4) + 4;
116  if (ret < size) {
117  av_packet_unref(pkt);
118  return AVERROR(EIO);
119  }
120 
121  pkt->size = ret + palette_size;
122  pkt->stream_index = avs->st_video->index;
123  if (sub_type == 0)
124  pkt->flags |= AV_PKT_FLAG_KEY;
125 
126  return 0;
127 }
128 
130 {
131  AvsFormat *avs = s->priv_data;
132  int ret, size;
133 
134  size = avio_tell(s->pb);
135  ret = ff_voc_get_packet(s, pkt, avs->st_audio, avs->remaining_audio_size);
136  size = avio_tell(s->pb) - size;
137  avs->remaining_audio_size -= size;
138 
139  if (ret == AVERROR(EIO))
140  return 0; /* this indicate EOS */
141  if (ret < 0)
142  return ret;
143 
144  pkt->stream_index = avs->st_audio->index;
145  pkt->flags |= AV_PKT_FLAG_KEY;
146 
147  return size;
148 }
149 
151 {
152  AvsFormat *avs = s->priv_data;
153  int sub_type = 0, size = 0;
155  int palette_size = 0;
156  uint8_t palette[4 + 3 * 256];
157  int ret;
158 
159  if (avs->remaining_audio_size > 0)
160  if (avs_read_audio_packet(s, pkt) > 0)
161  return 0;
162 
163  while (1) {
164  if (avs->remaining_frame_size <= 0) {
165  if (!avio_rl16(s->pb)) /* found EOF */
166  return AVERROR(EIO);
167  avs->remaining_frame_size = avio_rl16(s->pb) - 4;
168  }
169 
170  while (avs->remaining_frame_size > 0) {
171  sub_type = avio_r8(s->pb);
172  type = avio_r8(s->pb);
173  size = avio_rl16(s->pb);
174  if (size < 4)
175  return AVERROR_INVALIDDATA;
176  avs->remaining_frame_size -= size;
177 
178  switch (type) {
179  case AVS_PALETTE:
180  if (size - 4 > sizeof(palette))
181  return AVERROR_INVALIDDATA;
182  ret = avio_read(s->pb, palette, size - 4);
183  if (ret < size - 4)
184  return AVERROR(EIO);
185  palette_size = size;
186  break;
187 
188  case AVS_VIDEO:
189  if (!avs->st_video) {
190  avs->st_video = avformat_new_stream(s, NULL);
191  if (!avs->st_video)
192  return AVERROR(ENOMEM);
195  avs->st_video->codecpar->width = avs->width;
196  avs->st_video->codecpar->height = avs->height;
198  avs->st_video->nb_frames = avs->nb_frames;
199 #if FF_API_R_FRAME_RATE
200  avs->st_video->r_frame_rate =
201 #endif
202  avs->st_video->avg_frame_rate = (AVRational){avs->fps, 1};
203  }
204  return avs_read_video_packet(s, pkt, type, sub_type, size,
205  palette, palette_size);
206 
207  case AVS_AUDIO:
208  if (!avs->st_audio) {
209  avs->st_audio = avformat_new_stream(s, NULL);
210  if (!avs->st_audio)
211  return AVERROR(ENOMEM);
213  }
214  avs->remaining_audio_size = size - 4;
215  size = avs_read_audio_packet(s, pkt);
216  if (size != 0)
217  return size;
218  break;
219 
220  default:
221  avio_skip(s->pb, size - 4);
222  }
223  }
224  }
225 }
226 
228 {
229  return 0;
230 }
231 
233  .name = "avs",
234  .long_name = NULL_IF_CONFIG_SMALL("Argonaut Games Creature Shock"),
235  .priv_data_size = sizeof(AvsFormat),
240 };
#define NULL
Definition: coverity.c:32
int width
Definition: avs.c:36
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
int ff_voc_get_packet(AVFormatContext *s, AVPacket *pkt, AVStream *st, int max_size)
Definition: voc_packet.c:27
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: avcodec.h:3900
int index
stream index in AVFormatContext
Definition: avformat.h:875
int size
Definition: avcodec.h:1446
int64_t avio_skip(AVIOContext *s, int64_t offset)
Skip given number of bytes forward.
Definition: aviobuf.c:331
static AVPacket pkt
int ctx_flags
Flags signalling stream properties.
Definition: avformat.h:1400
Format I/O context.
Definition: avformat.h:1351
AVStream * st_video
Definition: avs.c:34
uint8_t
int remaining_audio_size
Definition: avs.c:42
int width
Video only.
Definition: avcodec.h:3966
#define AVFMTCTX_NOHEADER
signal that no header is present (streams are added dynamically)
Definition: avformat.h:1295
static int avs_read_close(AVFormatContext *s)
Definition: avs.c:227
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
Definition: utils.c:4455
uint8_t * data
Definition: avcodec.h:1445
static av_cold int read_close(AVFormatContext *ctx)
Definition: libcdio.c:145
ptrdiff_t size
Definition: opengl_enc.c:101
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:557
#define av_log(a,...)
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:648
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: avcodec.h:1477
int av_new_packet(AVPacket *pkt, int size)
Allocate the payload of a packet and initialize its fields with default values.
Definition: avpacket.c:86
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
Definition: avs.c:32
unsigned int avio_rl32(AVIOContext *s)
Definition: aviobuf.c:770
#define AVERROR(e)
Definition: error.h:43
VocDecContext voc
Definition: avs.c:33
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:186
Definition: avs.c:31
AVInputFormat ff_avs_demuxer
Definition: avs.c:232
enum AVMediaType codec_type
General type of the encoded data.
Definition: avcodec.h:3896
int nb_frames
Definition: avs.c:40
int fps
Definition: avs.c:39
AVRational avg_frame_rate
Average framerate.
Definition: avformat.h:947
Definition: avs.c:46
int flags
A combination of AV_PKT_FLAG values.
Definition: avcodec.h:1451
int avio_r8(AVIOContext *s)
Definition: aviobuf.c:639
unsigned char * buf
Buffer must have AVPROBE_PADDING_SIZE of extra allocated bytes filled with zero.
Definition: avformat.h:450
int remaining_frame_size
Definition: avs.c:41
static int avs_read_video_packet(AVFormatContext *s, AVPacket *pkt, AvsBlockType type, int sub_type, int size, uint8_t *palette, int palette_size)
Definition: avs.c:92
static int read_probe(AVProbeData *pd)
Definition: jvdec.c:55
#define s(width, name)
Definition: cbs_vp9.c:257
static int avs_probe(AVProbeData *p)
Definition: avs.c:53
static int read_header(FFV1Context *f)
Definition: ffv1dec.c:530
Stream structure.
Definition: avformat.h:874
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
Definition: avio_reading.c:42
AVStream * st_audio
Definition: avs.c:35
AVIOContext * pb
I/O context.
Definition: avformat.h:1393
void av_packet_unref(AVPacket *pkt)
Wipe the packet.
Definition: avpacket.c:598
GLint GLenum type
Definition: opengl_enc.c:105
static int avs_read_audio_packet(AVFormatContext *s, AVPacket *pkt)
Definition: avs.c:129
Rational number (pair of numerator and denominator).
Definition: rational.h:58
#define AVPROBE_SCORE_EXTENSION
score for file extension
Definition: avformat.h:458
AvsBlockType
Definition: avs.c:30
This structure contains the data a format has to probe a file.
Definition: avformat.h:448
static int avs_read_header(AVFormatContext *s)
Definition: avs.c:66
unsigned int avio_rl16(AVIOContext *s)
Definition: aviobuf.c:754
Main libavformat public API header.
Definition: avs.c:32
int bits_per_sample
Definition: avs.c:38
int height
Definition: avs.c:37
int64_t nb_frames
number of frames in this stream if known or 0
Definition: avformat.h:925
void * priv_data
Format private data.
Definition: avformat.h:1379
int bits_per_coded_sample
The number of bits per sample in the codedwords.
Definition: avcodec.h:3942
static int avs_read_packet(AVFormatContext *s, AVPacket *pkt)
Definition: avs.c:150
const char * name
A comma separated list of short names for the format.
Definition: avformat.h:647
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1021
int stream_index
Definition: avcodec.h:1447
AVRational r_frame_rate
Real base framerate of the stream.
Definition: avformat.h:998
This structure stores compressed data.
Definition: avcodec.h:1422