[deb_ffmpeg.git] / ffmpeg / libavformat / segafilm.c

/*
 * Sega FILM Format (CPK) Demuxer
 * Copyright (c) 2003 The FFmpeg Project
 *
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * FFmpeg is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with FFmpeg; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

/**
 * @file
 * Sega FILM (.cpk) file demuxer
 * by Mike Melanson (melanson@pcisys.net)
 * For more information regarding the Sega FILM file format, visit:
 *   http://www.pcisys.net/~melanson/codecs/
 */

#include "libavutil/intreadwrite.h"
#include "avformat.h"
#include "internal.h"
#include "avio_internal.h"

#define FILM_TAG MKBETAG('F', 'I', 'L', 'M')
#define FDSC_TAG MKBETAG('F', 'D', 'S', 'C')
#define STAB_TAG MKBETAG('S', 'T', 'A', 'B')
#define CVID_TAG MKBETAG('c', 'v', 'i', 'd')
#define RAW_TAG  MKBETAG('r', 'a', 'w', ' ')

typedef struct {
  int stream;
  int64_t sample_offset;
  unsigned int sample_size;
  int64_t pts;
  int keyframe;
} film_sample;

typedef struct FilmDemuxContext {
    int video_stream_index;
    int audio_stream_index;

    enum AVCodecID audio_type;
    unsigned int audio_samplerate;
    unsigned int audio_bits;
    unsigned int audio_channels;

    enum AVCodecID video_type;
    unsigned int sample_count;
    film_sample *sample_table;
    unsigned int current_sample;

    unsigned int base_clock;
    unsigned int version;
} FilmDemuxContext;

static int film_probe(AVProbeData *p)
{
    if (AV_RB32(&p->buf[0]) != FILM_TAG)
        return 0;

    if (AV_RB32(&p->buf[16]) != FDSC_TAG)
        return 0;

    return AVPROBE_SCORE_MAX;
}

static int film_read_close(AVFormatContext *s)
{
    FilmDemuxContext *film = s->priv_data;

    av_freep(&film->sample_table);

    return 0;
}

static int film_read_header(AVFormatContext *s)
{
    FilmDemuxContext *film = s->priv_data;
    AVIOContext *pb = s->pb;
    AVStream *st;
    unsigned char scratch[256];
    int i, ret;
    unsigned int data_offset;
    unsigned int audio_frame_counter;

    film->sample_table = NULL;

    /* load the main FILM header */
    if (avio_read(pb, scratch, 16) != 16)
        return AVERROR(EIO);
    data_offset = AV_RB32(&scratch[4]);
    film->version = AV_RB32(&scratch[8]);

    /* load the FDSC chunk */
    if (film->version == 0) {
        /* special case for Lemmings .film files; 20-byte header */
        if (avio_read(pb, scratch, 20) != 20)
            return AVERROR(EIO);
        /* make some assumptions about the audio parameters */
        film->audio_type = AV_CODEC_ID_PCM_S8;
        film->audio_samplerate = 22050;
        film->audio_channels = 1;
        film->audio_bits = 8;
    } else {
        /* normal Saturn .cpk files; 32-byte header */
        if (avio_read(pb, scratch, 32) != 32)
            return AVERROR(EIO);
        film->audio_samplerate = AV_RB16(&scratch[24]);
        film->audio_channels = scratch[21];
        film->audio_bits = scratch[22];
        if (scratch[23] == 2 && film->audio_channels > 0)
            film->audio_type = AV_CODEC_ID_ADPCM_ADX;
        else if (film->audio_channels > 0) {
            if (film->audio_bits == 8)
                film->audio_type = AV_CODEC_ID_PCM_S8_PLANAR;
            else if (film->audio_bits == 16)
                film->audio_type = AV_CODEC_ID_PCM_S16BE_PLANAR;
            else
                film->audio_type = AV_CODEC_ID_NONE;
        } else
            film->audio_type = AV_CODEC_ID_NONE;
    }

    if (AV_RB32(&scratch[0]) != FDSC_TAG)
        return AVERROR_INVALIDDATA;

    if (AV_RB32(&scratch[8]) == CVID_TAG) {
        film->video_type = AV_CODEC_ID_CINEPAK;
    } else if (AV_RB32(&scratch[8]) == RAW_TAG) {
        film->video_type = AV_CODEC_ID_RAWVIDEO;
    } else {
        film->video_type = AV_CODEC_ID_NONE;
    }

    /* initialize the decoder streams */
    if (film->video_type) {
        st = avformat_new_stream(s, NULL);
        if (!st)
            return AVERROR(ENOMEM);
        film->video_stream_index = st->index;
        st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
        st->codec->codec_id = film->video_type;
        st->codec->codec_tag = 0;  /* no fourcc */
        st->codec->width = AV_RB32(&scratch[16]);
        st->codec->height = AV_RB32(&scratch[12]);

        if (film->video_type == AV_CODEC_ID_RAWVIDEO) {
            if (scratch[20] == 24) {
                st->codec->pix_fmt = AV_PIX_FMT_RGB24;
            } else {
                av_log(s, AV_LOG_ERROR, "raw video is using unhandled %dbpp\n", scratch[20]);
                return -1;
            }
        }
    }

    if (film->audio_type) {
        st = avformat_new_stream(s, NULL);
        if (!st)
            return AVERROR(ENOMEM);
        film->audio_stream_index = st->index;
        st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
        st->codec->codec_id = film->audio_type;
        st->codec->codec_tag = 1;
        st->codec->channels = film->audio_channels;
        st->codec->sample_rate = film->audio_samplerate;

        if (film->audio_type == AV_CODEC_ID_ADPCM_ADX) {
            st->codec->bits_per_coded_sample = 18 * 8 / 32;
            st->codec->block_align = st->codec->channels * 18;
            st->need_parsing = AVSTREAM_PARSE_FULL;
        } else {
            st->codec->bits_per_coded_sample = film->audio_bits;
            st->codec->block_align = st->codec->channels *
                st->codec->bits_per_coded_sample / 8;
        }

        st->codec->bit_rate = st->codec->channels * st->codec->sample_rate *
            st->codec->bits_per_coded_sample;
    }

    /* load the sample table */
    if (avio_read(pb, scratch, 16) != 16)
        return AVERROR(EIO);
    if (AV_RB32(&scratch[0]) != STAB_TAG)
        return AVERROR_INVALIDDATA;
    film->base_clock = AV_RB32(&scratch[8]);
    film->sample_count = AV_RB32(&scratch[12]);
    if(film->sample_count >= UINT_MAX / sizeof(film_sample))
        return -1;
    film->sample_table = av_malloc(film->sample_count * sizeof(film_sample));
    if (!film->sample_table)
        return AVERROR(ENOMEM);

    for (i = 0; i < s->nb_streams; i++) {
        st = s->streams[i];
        if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
            avpriv_set_pts_info(st, 33, 1, film->base_clock);
        else
            avpriv_set_pts_info(st, 64, 1, film->audio_samplerate);
    }

    audio_frame_counter = 0;
    for (i = 0; i < film->sample_count; i++) {
        /* load the next sample record and transfer it to an internal struct */
        if (avio_read(pb, scratch, 16) != 16) {
            ret = AVERROR(EIO);
            goto fail;
        }
        film->sample_table[i].sample_offset =
            data_offset + AV_RB32(&scratch[0]);
        film->sample_table[i].sample_size = AV_RB32(&scratch[4]);
        if (film->sample_table[i].sample_size > INT_MAX / 4) {
            ret = AVERROR_INVALIDDATA;
            goto fail;
        }
        if (AV_RB32(&scratch[8]) == 0xFFFFFFFF) {
            film->sample_table[i].stream = film->audio_stream_index;
            film->sample_table[i].pts = audio_frame_counter;

            if (film->audio_type == AV_CODEC_ID_ADPCM_ADX)
                audio_frame_counter += (film->sample_table[i].sample_size * 32 /
                    (18 * film->audio_channels));
            else if (film->audio_type != AV_CODEC_ID_NONE)
                audio_frame_counter += (film->sample_table[i].sample_size /
                    (film->audio_channels * film->audio_bits / 8));
        } else {
            film->sample_table[i].stream = film->video_stream_index;
            film->sample_table[i].pts = AV_RB32(&scratch[8]) & 0x7FFFFFFF;
            film->sample_table[i].keyframe = (scratch[8] & 0x80) ? 0 : 1;
        }
    }

    film->current_sample = 0;

    return 0;
fail:
    film_read_close(s);
    return ret;
}

static int film_read_packet(AVFormatContext *s,
                            AVPacket *pkt)
{
    FilmDemuxContext *film = s->priv_data;
    AVIOContext *pb = s->pb;
    film_sample *sample;
    int ret = 0;

    if (film->current_sample >= film->sample_count)
        return AVERROR_EOF;

    sample = &film->sample_table[film->current_sample];

    /* position the stream (will probably be there anyway) */
    avio_seek(pb, sample->sample_offset, SEEK_SET);


    ret= av_get_packet(pb, pkt, sample->sample_size);
    if (ret != sample->sample_size)
        ret = AVERROR(EIO);

    pkt->stream_index = sample->stream;
    pkt->pts = sample->pts;

    film->current_sample++;

    return ret;
}

AVInputFormat ff_segafilm_demuxer = {
    .name           = "film_cpk",
    .long_name      = NULL_IF_CONFIG_SMALL("Sega FILM / CPK"),
    .priv_data_size = sizeof(FilmDemuxContext),
    .read_probe     = film_probe,
    .read_header    = film_read_header,
    .read_packet    = film_read_packet,
    .read_close     = film_read_close,
};
Commit	Line	Data
2ba45a60 DM	1	/*
	2	* Sega FILM Format (CPK) Demuxer
	3	* Copyright (c) 2003 The FFmpeg Project
	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	* Sega FILM (.cpk) file demuxer
	25	* by Mike Melanson (melanson@pcisys.net)
	26	* For more information regarding the Sega FILM file format, visit:
	27	* http://www.pcisys.net/~melanson/codecs/
	28	*/
	29
	30	#include "libavutil/intreadwrite.h"
	31	#include "avformat.h"
	32	#include "internal.h"
	33	#include "avio_internal.h"
	34
	35	#define FILM_TAG MKBETAG('F', 'I', 'L', 'M')
	36	#define FDSC_TAG MKBETAG('F', 'D', 'S', 'C')
	37	#define STAB_TAG MKBETAG('S', 'T', 'A', 'B')
	38	#define CVID_TAG MKBETAG('c', 'v', 'i', 'd')
	39	#define RAW_TAG MKBETAG('r', 'a', 'w', ' ')
	40
	41	typedef struct {
	42	int stream;
	43	int64_t sample_offset;
	44	unsigned int sample_size;
	45	int64_t pts;
	46	int keyframe;
	47	} film_sample;
	48
	49	typedef struct FilmDemuxContext {
	50	int video_stream_index;
	51	int audio_stream_index;
	52
	53	enum AVCodecID audio_type;
	54	unsigned int audio_samplerate;
	55	unsigned int audio_bits;
	56	unsigned int audio_channels;
	57
	58	enum AVCodecID video_type;
	59	unsigned int sample_count;
	60	film_sample *sample_table;
	61	unsigned int current_sample;
	62
	63	unsigned int base_clock;
	64	unsigned int version;
65	} FilmDemuxContext;
66
67	static int film_probe(AVProbeData *p)
68	{
69	if (AV_RB32(&p->buf[0]) != FILM_TAG)
70	return 0;
71
72	if (AV_RB32(&p->buf[16]) != FDSC_TAG)
73	return 0;
74
75	return AVPROBE_SCORE_MAX;
76	}
77
78	static int film_read_close(AVFormatContext *s)
79	{
80	FilmDemuxContext *film = s->priv_data;
81
82	av_freep(&film->sample_table);
83
84	return 0;
85	}
86
87	static int film_read_header(AVFormatContext *s)
88	{
89	FilmDemuxContext *film = s->priv_data;
90	AVIOContext *pb = s->pb;
91	AVStream *st;
92	unsigned char scratch[256];
93	int i, ret;
94	unsigned int data_offset;
95	unsigned int audio_frame_counter;
96
97	film->sample_table = NULL;
98
99	/* load the main FILM header */
100	if (avio_read(pb, scratch, 16) != 16)
101	return AVERROR(EIO);
102	data_offset = AV_RB32(&scratch[4]);
103	film->version = AV_RB32(&scratch[8]);
104
105	/* load the FDSC chunk */
106	if (film->version == 0) {
107	/* special case for Lemmings .film files; 20-byte header */
108	if (avio_read(pb, scratch, 20) != 20)
109	return AVERROR(EIO);
110	/* make some assumptions about the audio parameters */
111	film->audio_type = AV_CODEC_ID_PCM_S8;
112	film->audio_samplerate = 22050;
113	film->audio_channels = 1;
114	film->audio_bits = 8;
115	} else {
116	/* normal Saturn .cpk files; 32-byte header */
117	if (avio_read(pb, scratch, 32) != 32)
118	return AVERROR(EIO);
119	film->audio_samplerate = AV_RB16(&scratch[24]);
120	film->audio_channels = scratch[21];
121	film->audio_bits = scratch[22];
122	if (scratch[23] == 2 && film->audio_channels > 0)
123	film->audio_type = AV_CODEC_ID_ADPCM_ADX;
124	else if (film->audio_channels > 0) {
125	if (film->audio_bits == 8)
126	film->audio_type = AV_CODEC_ID_PCM_S8_PLANAR;
127	else if (film->audio_bits == 16)
128	film->audio_type = AV_CODEC_ID_PCM_S16BE_PLANAR;
129	else
130	film->audio_type = AV_CODEC_ID_NONE;
131	} else
132	film->audio_type = AV_CODEC_ID_NONE;
133	}
134
135	if (AV_RB32(&scratch[0]) != FDSC_TAG)
136	return AVERROR_INVALIDDATA;
137
138	if (AV_RB32(&scratch[8]) == CVID_TAG) {
139	film->video_type = AV_CODEC_ID_CINEPAK;
140	} else if (AV_RB32(&scratch[8]) == RAW_TAG) {
141	film->video_type = AV_CODEC_ID_RAWVIDEO;
142	} else {
143	film->video_type = AV_CODEC_ID_NONE;
144	}
145
146	/* initialize the decoder streams */
147	if (film->video_type) {
148	st = avformat_new_stream(s, NULL);
149	if (!st)
150	return AVERROR(ENOMEM);
151	film->video_stream_index = st->index;
152	st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
153	st->codec->codec_id = film->video_type;
154	st->codec->codec_tag = 0; /* no fourcc */
155	st->codec->width = AV_RB32(&scratch[16]);
156	st->codec->height = AV_RB32(&scratch[12]);
157
158	if (film->video_type == AV_CODEC_ID_RAWVIDEO) {
159	if (scratch[20] == 24) {
160	st->codec->pix_fmt = AV_PIX_FMT_RGB24;
161	} else {
162	av_log(s, AV_LOG_ERROR, "raw video is using unhandled %dbpp\n", scratch[20]);
163	return -1;
164	}
165	}
166	}
167
168	if (film->audio_type) {
169	st = avformat_new_stream(s, NULL);
170	if (!st)
171	return AVERROR(ENOMEM);
172	film->audio_stream_index = st->index;
173	st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
174	st->codec->codec_id = film->audio_type;
175	st->codec->codec_tag = 1;
176	st->codec->channels = film->audio_channels;
177	st->codec->sample_rate = film->audio_samplerate;
178
179	if (film->audio_type == AV_CODEC_ID_ADPCM_ADX) {
180	st->codec->bits_per_coded_sample = 18 * 8 / 32;
181	st->codec->block_align = st->codec->channels * 18;
182	st->need_parsing = AVSTREAM_PARSE_FULL;
183	} else {
184	st->codec->bits_per_coded_sample = film->audio_bits;
185	st->codec->block_align = st->codec->channels *
186	st->codec->bits_per_coded_sample / 8;
187	}
188
189	st->codec->bit_rate = st->codec->channels * st->codec->sample_rate *
190	st->codec->bits_per_coded_sample;
191	}
192
193	/* load the sample table */
194	if (avio_read(pb, scratch, 16) != 16)
195	return AVERROR(EIO);
196	if (AV_RB32(&scratch[0]) != STAB_TAG)
197	return AVERROR_INVALIDDATA;
198	film->base_clock = AV_RB32(&scratch[8]);
199	film->sample_count = AV_RB32(&scratch[12]);
200	if(film->sample_count >= UINT_MAX / sizeof(film_sample))
201	return -1;
202	film->sample_table = av_malloc(film->sample_count * sizeof(film_sample));
203	if (!film->sample_table)
204	return AVERROR(ENOMEM);
205
206	for (i = 0; i < s->nb_streams; i++) {
207	st = s->streams[i];
208	if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO)
209	avpriv_set_pts_info(st, 33, 1, film->base_clock);
210	else
211	avpriv_set_pts_info(st, 64, 1, film->audio_samplerate);
212	}
213
214	audio_frame_counter = 0;
215	for (i = 0; i < film->sample_count; i++) {
216	/* load the next sample record and transfer it to an internal struct */
217	if (avio_read(pb, scratch, 16) != 16) {
218	ret = AVERROR(EIO);
219	goto fail;
220	}
221	film->sample_table[i].sample_offset =
222	data_offset + AV_RB32(&scratch[0]);
223	film->sample_table[i].sample_size = AV_RB32(&scratch[4]);
224	if (film->sample_table[i].sample_size > INT_MAX / 4) {
225	ret = AVERROR_INVALIDDATA;
226	goto fail;
227	}
228	if (AV_RB32(&scratch[8]) == 0xFFFFFFFF) {
229	film->sample_table[i].stream = film->audio_stream_index;
230	film->sample_table[i].pts = audio_frame_counter;
231
232	if (film->audio_type == AV_CODEC_ID_ADPCM_ADX)
233	audio_frame_counter += (film->sample_table[i].sample_size * 32 /
234	(18 * film->audio_channels));
235	else if (film->audio_type != AV_CODEC_ID_NONE)
236	audio_frame_counter += (film->sample_table[i].sample_size /
237	(film->audio_channels * film->audio_bits / 8));
238	} else {
239	film->sample_table[i].stream = film->video_stream_index;
240	film->sample_table[i].pts = AV_RB32(&scratch[8]) & 0x7FFFFFFF;
241	film->sample_table[i].keyframe = (scratch[8] & 0x80) ? 0 : 1;
242	}
243	}
244
245	film->current_sample = 0;
246
247	return 0;
248	fail:
249	film_read_close(s);
250	return ret;
251	}
252
253	static int film_read_packet(AVFormatContext *s,
254	AVPacket *pkt)
255	{
256	FilmDemuxContext *film = s->priv_data;
257	AVIOContext *pb = s->pb;
258	film_sample *sample;
259	int ret = 0;
260
261	if (film->current_sample >= film->sample_count)
262	return AVERROR_EOF;
263
264	sample = &film->sample_table[film->current_sample];
265
266	/* position the stream (will probably be there anyway) */
267	avio_seek(pb, sample->sample_offset, SEEK_SET);
268
269
270	ret= av_get_packet(pb, pkt, sample->sample_size);
271	if (ret != sample->sample_size)
272	ret = AVERROR(EIO);
273
274	pkt->stream_index = sample->stream;
275	pkt->pts = sample->pts;
276
277	film->current_sample++;
278
279	return ret;
280	}
281
282	AVInputFormat ff_segafilm_demuxer = {
283	.name = "film_cpk",
284	.long_name = NULL_IF_CONFIG_SMALL("Sega FILM / CPK"),
285	.priv_data_size = sizeof(FilmDemuxContext),
286	.read_probe = film_probe,
287	.read_header = film_read_header,
288	.read_packet = film_read_packet,
289	.read_close = film_read_close,
290	};