FFmpeg  4.4
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(const 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 
70  s->ctx_flags |= AVFMTCTX_NOHEADER;
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  return AVERROR(EIO);
118  }
119 
120  pkt->size = ret + palette_size;
121  pkt->stream_index = avs->st_video->index;
122  if (sub_type == 0)
124 
125  return 0;
126 }
127 
129 {
130  AvsFormat *avs = s->priv_data;
131  int ret;
132  int64_t size;
133 
134  size = avio_tell(s->pb);
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;
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) {
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) {
210  if (!avs->st_audio)
211  return AVERROR(ENOMEM);
213  }
214  avs->remaining_audio_size = size - 4;
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  .name = "avs",
229  .long_name = NULL_IF_CONFIG_SMALL("Argonaut Games Creature Shock"),
230  .priv_data_size = sizeof(AvsFormat),
234 };
uint8_t
Main libavformat public API header.
#define AVFMTCTX_NOHEADER
signal that no header is present (streams are added dynamically)
Definition: avformat.h:1177
#define AVPROBE_SCORE_EXTENSION
score for file extension
Definition: avformat.h:451
static av_always_inline int64_t avio_tell(AVIOContext *s)
ftell() equivalent for AVIOContext.
Definition: avio.h:557
unsigned int avio_rl16(AVIOContext *s)
Definition: aviobuf.c:734
int64_t avio_skip(AVIOContext *s, int64_t offset)
Skip given number of bytes forward.
Definition: aviobuf.c:337
int avio_read(AVIOContext *s, unsigned char *buf, int size)
Read size bytes from AVIOContext into buf.
Definition: aviobuf.c:633
unsigned int avio_rl32(AVIOContext *s)
Definition: aviobuf.c:750
int avio_r8(AVIOContext *s)
Definition: aviobuf.c:624
static int read_packet(void *opaque, uint8_t *buf, int buf_size)
Definition: avio_reading.c:42
#define s(width, name)
Definition: cbs_vp9.c:257
#define NULL
Definition: coverity.c:32
static int read_header(FFV1Context *f)
Definition: ffv1dec.c:527
@ AV_CODEC_ID_AVS
Definition: codec_id.h:131
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: packet.h:410
int av_new_packet(AVPacket *pkt, int size)
Allocate the payload of a packet and initialize its fields with default values.
Definition: avpacket.c:99
AVStream * avformat_new_stream(AVFormatContext *s, const AVCodec *c)
Add a new stream to a media file.
Definition: utils.c:4505
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
#define AVERROR(e)
Definition: error.h:43
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:194
@ AVMEDIA_TYPE_AUDIO
Definition: avutil.h:202
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
cl_device_type type
AvsBlockType
Definition: avs.c:30
@ AVS_PALETTE
Definition: avs.c:33
@ AVS_AUDIO
Definition: avs.c:32
@ AVS_GAME_DATA
Definition: avs.c:34
@ AVS_VIDEO
Definition: avs.c:31
@ AVS_NONE
Definition: avs.c:46
static int avs_read_audio_packet(AVFormatContext *s, AVPacket *pkt)
Definition: avs.c:128
AVInputFormat ff_avs_demuxer
Definition: avs.c:227
static int avs_read_header(AVFormatContext *s)
Definition: avs.c:66
static int avs_probe(const AVProbeData *p)
Definition: avs.c:53
static int avs_read_packet(AVFormatContext *s, AVPacket *pkt)
Definition: avs.c:150
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(const AVProbeData *pd)
Definition: jvdec.c:55
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification.
Definition: internal.h:117
int bits_per_coded_sample
The number of bits per sample in the codedwords.
Definition: codec_par.h:102
int width
Video only.
Definition: codec_par.h:126
enum AVMediaType codec_type
General type of the encoded data.
Definition: codec_par.h:56
enum AVCodecID codec_id
Specific type of the encoded data (the codec used).
Definition: codec_par.h:60
Format I/O context.
Definition: avformat.h:1232
const char * name
A comma separated list of short names for the format.
Definition: avformat.h:645
This structure stores compressed data.
Definition: packet.h:346
int stream_index
Definition: packet.h:371
int flags
A combination of AV_PKT_FLAG values.
Definition: packet.h:375
int size
Definition: packet.h:370
uint8_t * data
Definition: packet.h:369
This structure contains the data a format has to probe a file.
Definition: avformat.h:441
unsigned char * buf
Buffer must have AVPROBE_PADDING_SIZE of extra allocated bytes filled with zero.
Definition: avformat.h:443
Rational number (pair of numerator and denominator).
Definition: rational.h:58
Stream structure.
Definition: avformat.h:873
AVCodecParameters * codecpar
Codec parameters associated with this stream.
Definition: avformat.h:1038
int64_t nb_frames
number of frames in this stream if known or 0
Definition: avformat.h:924
int index
stream index in AVFormatContext
Definition: avformat.h:874
AVRational avg_frame_rate
Average framerate.
Definition: avformat.h:946
AVRational r_frame_rate
Real base framerate of the stream.
Definition: avformat.h:1015
Definition: avs.c:32
int remaining_frame_size
Definition: avs.c:41
AVStream * st_audio
Definition: avs.c:35
int fps
Definition: avs.c:39
int height
Definition: avs.c:37
int bits_per_sample
Definition: avs.c:38
int nb_frames
Definition: avs.c:40
AVStream * st_video
Definition: avs.c:34
VocDecContext voc
Definition: avs.c:33
int remaining_audio_size
Definition: avs.c:42
int width
Definition: avs.c:36
#define av_log(a,...)
AVPacket * pkt
Definition: movenc.c:59
int size
int ff_voc_get_packet(AVFormatContext *s, AVPacket *pkt, AVStream *st, int max_size)
Definition: voc_packet.c:27