2 * Gremlin Digital Video demuxer
3 * Copyright (c) 2017 Paul B Mahol
5 * This file is part of FFmpeg.
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.
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.
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
22 #include "libavutil/intreadwrite.h"
28 typedef struct GDVContext {
32 int audio_stream_index;
33 int video_stream_index;
37 static int gdv_read_probe(const AVProbeData *p)
39 if (AV_RL32(p->buf) == 0x29111994)
40 return AVPROBE_SCORE_MAX;
70 static int gdv_read_header(AVFormatContext *ctx)
72 GDVContext *gdv = ctx->priv_data;
73 AVIOContext *pb = ctx->pb;
75 unsigned fps, snd_flags, vid_depth, size_id;
78 size_id = avio_rl16(pb);
80 vst = avformat_new_stream(ctx, 0);
82 return AVERROR(ENOMEM);
86 vst->nb_frames = avio_rl16(pb);
90 return AVERROR_INVALIDDATA;
92 snd_flags = avio_rl16(pb);
94 ast = avformat_new_stream(ctx, 0);
96 return AVERROR(ENOMEM);
99 ast->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
100 ast->codecpar->codec_tag = 0;
101 ast->codecpar->sample_rate = avio_rl16(pb);
102 ast->codecpar->channels = 1 + !!(snd_flags & 2);
104 ast->codecpar->codec_id = AV_CODEC_ID_GREMLIN_DPCM;
106 ast->codecpar->codec_id = (snd_flags & 4) ? AV_CODEC_ID_PCM_S16LE : AV_CODEC_ID_PCM_U8;
109 avpriv_set_pts_info(ast, 64, 1, ast->codecpar->sample_rate);
110 gdv->audio_size = (ast->codecpar->sample_rate / fps) *
111 ast->codecpar->channels * (1 + !!(snd_flags & 4)) / (1 + !!(snd_flags & 8));
116 vid_depth = avio_rl16(pb);
119 vst->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
120 vst->codecpar->codec_id = AV_CODEC_ID_GDV;
121 vst->codecpar->codec_tag = 0;
122 vst->codecpar->width = avio_rl16(pb);
123 vst->codecpar->height = avio_rl16(pb);
125 if (vst->codecpar->width == 0 || vst->codecpar->height == 0) {
128 for (i = 0; i < FF_ARRAY_ELEMS(FixedSize) - 1; i++) {
129 if (FixedSize[i].id == size_id)
133 vst->codecpar->width = FixedSize[i].width;
134 vst->codecpar->height = FixedSize[i].height;
137 avpriv_set_pts_info(vst, 64, 1, fps);
142 for (i = 0; i < 256; i++) {
143 unsigned r = avio_r8(pb);
144 unsigned g = avio_r8(pb);
145 unsigned b = avio_r8(pb);
146 gdv->pal[i] = 0xFFU << 24 | r << 18 | g << 10 | b << 2;
150 gdv->is_first_video = 1;
155 static int gdv_read_packet(AVFormatContext *ctx, AVPacket *pkt)
157 GDVContext *gdv = ctx->priv_data;
158 AVIOContext *pb = ctx->pb;
162 return pb->error ? pb->error : AVERROR_EOF;
164 if (gdv->audio_size && gdv->is_audio) {
165 ret = av_get_packet(pb, pkt, gdv->audio_size);
168 pkt->stream_index = 1;
173 if (avio_rl16(pb) != 0x1305)
174 return AVERROR_INVALIDDATA;
175 ret = av_get_packet(pb, pkt, 4 + avio_rl16(pb));
178 pkt->stream_index = 0;
181 if (gdv->is_first_video) {
182 pal = av_packet_new_side_data(pkt, AV_PKT_DATA_PALETTE,
185 av_packet_unref(pkt);
186 return AVERROR(ENOMEM);
188 memcpy(pal, gdv->pal, AVPALETTE_SIZE);
189 pkt->flags |= AV_PKT_FLAG_KEY;
190 gdv->is_first_video = 0;
197 AVInputFormat ff_gdv_demuxer = {
199 .long_name = NULL_IF_CONFIG_SMALL("Gremlin Digital Video"),
200 .priv_data_size = sizeof(GDVContext),
201 .read_probe = gdv_read_probe,
202 .read_header = gdv_read_header,
203 .read_packet = gdv_read_packet,