2 * Digital Pictures SGA game demuxer
4 * Copyright (C) 2021 Paul B Mahol
6 * This file is part of FFmpeg.
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 #include "libavutil/intreadwrite.h"
24 #include "libavutil/avassert.h"
25 #include "libavutil/internal.h"
28 #include "avio_internal.h"
30 #define SEGA_CD_PCM_NUM 12500000
31 #define SEGA_CD_PCM_DEN 786432
33 typedef struct SGADemuxContext {
34 int video_stream_index;
35 int audio_stream_index;
37 uint8_t sector[65536 * 2];
49 static int sga_probe(const AVProbeData *p)
51 const uint8_t *src = p->buf;
52 int score = 0, sectors = 1;
56 if (p->buf_size < 2048)
59 for (int i = 0; i + 2 < p->buf_size; i += 2048) {
60 int header = AV_RB16(src + i);
62 if ((header > 0x07FE && header < 0x8100) ||
63 (header > 0x8200 && header < 0xA100) ||
64 (header > 0xA200 && header < 0xC100)) {
70 for (int i = 0; i + 4 < p->buf_size;) {
71 int header = AV_RB16(src + i);
72 int left = AV_RB16(src + i + 2);
73 int offset, type, size;
77 if (sectors && header && last_left == 0) {
81 last_left = left = header;
83 } else if (sectors && header) {
86 if (header != 0x7FE && left < 7)
91 i += sectors ? 2048 : left + 4;
96 if (sectors && (i > 0 && left < 0x7fe) &&
97 (i + left + 14 < p->buf_size)) {
98 offset = i + left + 2;
99 } else if (sectors && i > 0) {
101 last_left -= FFMIN(last_left, 2046);
108 header = AV_RB16(src + offset);
109 size = AV_RB16(src + offset + 2) + 4;
111 while ((header & 0xFF00) == 0) {
113 if (offset + 4 >= p->buf_size)
115 header = AV_RB16(src + offset);
116 size = AV_RB16(src + offset + 2) + 4;
119 if (offset + 12 >= p->buf_size)
121 if ((header & 0xFF) > 1)
133 new_rate = AV_RB16(src + offset + 8);
135 sample_rate = new_rate;
136 if (sample_rate == 0 || new_rate != sample_rate)
138 if (src[offset + 10] != 1)
142 } else if (type == 0xC1 ||
150 int nb_pals = src[offset + 9];
151 int tiles_w = src[offset + 10];
152 int tiles_h = src[offset + 11];
156 if (nb_pals == 0 || nb_pals > 4)
158 if (tiles_w == 0 || tiles_w > 80)
160 if (tiles_h == 0 || tiles_h > 60)
164 } else if (header == 0x7FE) {
170 i += sectors ? 2048 : size + 4;
171 last_left -= FFMIN(last_left, 2046);
177 return av_clip(score, 0, AVPROBE_SCORE_MAX);
180 static int sga_read_header(AVFormatContext *s)
182 SGADemuxContext *sga = s->priv_data;
183 AVIOContext *pb = s->pb;
185 sga->sector_headers = 1;
186 sga->first_audio_size = 0;
187 sga->video_stream_index = -1;
188 sga->audio_stream_index = -1;
192 s->ctx_flags |= AVFMTCTX_NOHEADER;
194 if (pb->seekable & AVIO_SEEKABLE_NORMAL) {
195 while (!avio_feof(pb)) {
196 int header = avio_rb16(pb);
197 int type = header >> 8;
201 if (!sga->first_audio_size &&
206 sga->first_audio_size = avio_rb16(pb);
208 clock = avio_rb16(pb);
209 sga->sample_rate = av_rescale(clock,
214 if ((header > 0x07FE && header < 0x8100) ||
215 (header > 0x8200 && header < 0xA100) ||
216 (header > 0xA200 && header < 0xC100)) {
217 sga->sector_headers = 0;
224 avio_seek(pb, 0, SEEK_SET);
230 static void print_stats(AVFormatContext *s, const char *where)
232 SGADemuxContext *sga = s->priv_data;
234 av_log(s, AV_LOG_DEBUG, "START %s\n", where);
235 av_log(s, AV_LOG_DEBUG, "pos: %"PRIX64"\n", avio_tell(s->pb));
236 av_log(s, AV_LOG_DEBUG, "idx: %X\n", sga->idx);
237 av_log(s, AV_LOG_DEBUG, "packet_type: %X\n", sga->packet_type);
238 av_log(s, AV_LOG_DEBUG, "payload_size: %X\n", sga->payload_size);
239 av_log(s, AV_LOG_DEBUG, "SECTOR: %016"PRIX64"\n", AV_RB64(sga->sector));
240 av_log(s, AV_LOG_DEBUG, "stream: %X\n", sga->sector[1]);
241 av_log(s, AV_LOG_DEBUG, "END %s\n", where);
244 static void update_type_size(AVFormatContext *s)
246 SGADemuxContext *sga = s->priv_data;
249 sga->packet_type = sga->sector[0];
250 sga->payload_size = AV_RB16(sga->sector + 2);
252 sga->packet_type = 0;
253 sga->payload_size = 0;
257 static int sga_video_packet(AVFormatContext *s, AVPacket *pkt)
259 SGADemuxContext *sga = s->priv_data;
262 if (sga->payload_size <= 8)
263 return AVERROR_INVALIDDATA;
265 if (sga->video_stream_index == -1) {
266 AVRational frame_rate;
268 AVStream *st = avformat_new_stream(s, NULL);
270 return AVERROR(ENOMEM);
273 st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
274 st->codecpar->codec_tag = 0;
275 st->codecpar->codec_id = AV_CODEC_ID_SGA_VIDEO;
276 sga->video_stream_index = st->index;
278 if (sga->first_audio_size > 0 && sga->sample_rate > 0) {
279 frame_rate.num = sga->sample_rate;
280 frame_rate.den = sga->first_audio_size;
285 avpriv_set_pts_info(st, 64, frame_rate.den, frame_rate.num);
288 ret = av_new_packet(pkt, sga->payload_size + 4);
290 return AVERROR(ENOMEM);
291 memcpy(pkt->data, sga->sector, sga->payload_size + 4);
292 av_assert0(sga->idx >= sga->payload_size + 4);
293 memmove(sga->sector, sga->sector + sga->payload_size + 4, sga->idx - sga->payload_size - 4);
295 pkt->stream_index = sga->video_stream_index;
297 pkt->pos = sga->pkt_pos;
298 pkt->flags |= sga->flags;
299 sga->idx -= sga->payload_size + 4;
303 av_log(s, AV_LOG_DEBUG, "VIDEO PACKET: %d:%016"PRIX64" i:%X\n", pkt->size, AV_RB64(sga->sector), sga->idx);
308 static int sga_audio_packet(AVFormatContext *s, AVPacket *pkt)
310 SGADemuxContext *sga = s->priv_data;
313 if (sga->payload_size <= 8)
314 return AVERROR_INVALIDDATA;
316 if (sga->audio_stream_index == -1) {
317 AVStream *st = avformat_new_stream(s, NULL);
319 return AVERROR(ENOMEM);
322 st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
323 st->codecpar->codec_tag = 0;
324 st->codecpar->codec_id = AV_CODEC_ID_PCM_SGA;
325 st->codecpar->channels = 1;
326 st->codecpar->channel_layout= AV_CH_LAYOUT_MONO;
327 st->codecpar->sample_rate = av_rescale(AV_RB16(sga->sector + 8),
330 sga->audio_stream_index = st->index;
332 avpriv_set_pts_info(st, 64, 1, st->codecpar->sample_rate);
335 ret = av_new_packet(pkt, sga->payload_size - 8);
337 return AVERROR(ENOMEM);
338 memcpy(pkt->data, sga->sector + 12, sga->payload_size - 8);
339 av_assert0(sga->idx >= sga->payload_size + 4);
340 memmove(sga->sector, sga->sector + sga->payload_size + 4, sga->idx - sga->payload_size - 4);
342 pkt->stream_index = sga->audio_stream_index;
343 pkt->duration = pkt->size;
344 pkt->pos = sga->pkt_pos;
345 pkt->flags |= sga->flags;
346 sga->idx -= sga->payload_size + 4;
350 av_log(s, AV_LOG_DEBUG, "AUDIO PACKET: %d:%016"PRIX64" i:%X\n", pkt->size, AV_RB64(sga->sector), sga->idx);
355 static int sga_packet(AVFormatContext *s, AVPacket *pkt)
357 SGADemuxContext *sga = s->priv_data;
360 if (sga->packet_type == 0xCD ||
361 sga->packet_type == 0xCB ||
362 sga->packet_type == 0xC9 ||
363 sga->packet_type == 0xC8 ||
364 sga->packet_type == 0xC7 ||
365 sga->packet_type == 0xC6 ||
366 sga->packet_type == 0xC1 ||
367 sga->packet_type == 0xE7) {
368 ret = sga_video_packet(s, pkt);
369 } else if (sga->packet_type == 0xA1 ||
370 sga->packet_type == 0xA2 ||
371 sga->packet_type == 0xA3 ||
372 sga->packet_type == 0xAA) {
373 ret = sga_audio_packet(s, pkt);
378 return AVERROR_INVALIDDATA;
379 memmove(sga->sector, sga->sector + 1, sga->idx - 1);
381 return AVERROR(EAGAIN);
387 static int try_packet(AVFormatContext *s, AVPacket *pkt)
389 SGADemuxContext *sga = s->priv_data;
390 int ret = AVERROR(EAGAIN);
393 if (sga->idx >= sga->payload_size + 4) {
394 print_stats(s, "before sga_packet");
395 ret = sga_packet(s, pkt);
396 print_stats(s, "after sga_packet");
397 if (ret != AVERROR(EAGAIN))
401 return sga->idx < sga->payload_size + 4 ? AVERROR(EAGAIN) : ret;
404 static int sga_read_packet(AVFormatContext *s, AVPacket *pkt)
406 SGADemuxContext *sga = s->priv_data;
407 AVIOContext *pb = s->pb;
410 sga->pkt_pos = avio_tell(pb);
415 print_stats(s, "start");
417 (!sga->payload_size || sga->idx < sga->payload_size + 4))
420 if (sga->idx < sga->payload_size + 4) {
421 ret = ffio_ensure_seekback(pb, 2);
425 print_stats(s, "before read header");
426 header = avio_rb16(pb);
430 } else if (!avio_feof(pb) &&
432 !sga->sector_headers)) {
433 avio_seek(pb, -2, SEEK_CUR);
434 sga->flags = AV_PKT_FLAG_KEY;
440 av_assert0(sga->idx + sga->left < sizeof(sga->sector));
441 ret = avio_read(pb, sga->sector + sga->idx, sga->left);
444 else if (ret != AVERROR_EOF && ret)
446 print_stats(s, "after read header");
451 ret = try_packet(s, pkt);
452 if (ret == AVERROR(EAGAIN))
458 static int sga_seek(AVFormatContext *s, int stream_index,
459 int64_t timestamp, int flags)
461 SGADemuxContext *sga = s->priv_data;
463 sga->packet_type = sga->payload_size = sga->idx = 0;
464 memset(sga->sector, 0, sizeof(sga->sector));
469 AVInputFormat ff_sga_demuxer = {
471 .long_name = NULL_IF_CONFIG_SMALL("Digital Pictures SGA"),
472 .priv_data_size = sizeof(SGADemuxContext),
473 .read_probe = sga_probe,
474 .read_header = sga_read_header,
475 .read_packet = sga_read_packet,
476 .read_seek = sga_seek,
478 .flags = AVFMT_GENERIC_INDEX,