00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00030 #include "avformat.h"
00031
00032 #define FILM_TAG MKBETAG('F', 'I', 'L', 'M')
00033 #define FDSC_TAG MKBETAG('F', 'D', 'S', 'C')
00034 #define STAB_TAG MKBETAG('S', 'T', 'A', 'B')
00035 #define CVID_TAG MKBETAG('c', 'v', 'i', 'd')
00036
00037 typedef struct {
00038 int stream;
00039 offset_t sample_offset;
00040 unsigned int sample_size;
00041 int64_t pts;
00042 int keyframe;
00043 } film_sample_t;
00044
00045 typedef struct FilmDemuxContext {
00046 int video_stream_index;
00047 int audio_stream_index;
00048
00049 unsigned int audio_type;
00050 unsigned int audio_samplerate;
00051 unsigned int audio_bits;
00052 unsigned int audio_channels;
00053
00054 unsigned int video_type;
00055 unsigned int sample_count;
00056 film_sample_t *sample_table;
00057 unsigned int current_sample;
00058
00059 unsigned int base_clock;
00060 unsigned int version;
00061
00062
00063 unsigned char *stereo_buffer;
00064 int stereo_buffer_size;
00065 } FilmDemuxContext;
00066
00067 static int film_probe(AVProbeData *p)
00068 {
00069 if (AV_RB32(&p->buf[0]) != FILM_TAG)
00070 return 0;
00071
00072 return AVPROBE_SCORE_MAX;
00073 }
00074
00075 static int film_read_header(AVFormatContext *s,
00076 AVFormatParameters *ap)
00077 {
00078 FilmDemuxContext *film = s->priv_data;
00079 ByteIOContext *pb = s->pb;
00080 AVStream *st;
00081 unsigned char scratch[256];
00082 int i;
00083 unsigned int data_offset;
00084 unsigned int audio_frame_counter;
00085
00086 film->sample_table = NULL;
00087 film->stereo_buffer = NULL;
00088 film->stereo_buffer_size = 0;
00089
00090
00091 if (get_buffer(pb, scratch, 16) != 16)
00092 return AVERROR(EIO);
00093 data_offset = AV_RB32(&scratch[4]);
00094 film->version = AV_RB32(&scratch[8]);
00095
00096
00097 if (film->version == 0) {
00098
00099 if (get_buffer(pb, scratch, 20) != 20)
00100 return AVERROR(EIO);
00101
00102 film->audio_type = CODEC_ID_PCM_S8;
00103 film->audio_samplerate = 22050;
00104 film->audio_channels = 1;
00105 film->audio_bits = 8;
00106 } else {
00107
00108 if (get_buffer(pb, scratch, 32) != 32)
00109 return AVERROR(EIO);
00110 film->audio_samplerate = AV_RB16(&scratch[24]);;
00111 film->audio_channels = scratch[21];
00112 film->audio_bits = scratch[22];
00113 if (film->audio_bits == 8)
00114 film->audio_type = CODEC_ID_PCM_S8;
00115 else if (film->audio_bits == 16)
00116 film->audio_type = CODEC_ID_PCM_S16BE;
00117 else
00118 film->audio_type = 0;
00119 }
00120
00121 if (AV_RB32(&scratch[0]) != FDSC_TAG)
00122 return AVERROR_INVALIDDATA;
00123
00124 if (AV_RB32(&scratch[8]) == CVID_TAG) {
00125 film->video_type = CODEC_ID_CINEPAK;
00126 } else
00127 film->video_type = 0;
00128
00129
00130 if (film->video_type) {
00131 st = av_new_stream(s, 0);
00132 if (!st)
00133 return AVERROR(ENOMEM);
00134 film->video_stream_index = st->index;
00135 st->codec->codec_type = CODEC_TYPE_VIDEO;
00136 st->codec->codec_id = film->video_type;
00137 st->codec->codec_tag = 0;
00138 st->codec->width = AV_RB32(&scratch[16]);
00139 st->codec->height = AV_RB32(&scratch[12]);
00140 }
00141
00142 if (film->audio_type) {
00143 st = av_new_stream(s, 0);
00144 if (!st)
00145 return AVERROR(ENOMEM);
00146 film->audio_stream_index = st->index;
00147 st->codec->codec_type = CODEC_TYPE_AUDIO;
00148 st->codec->codec_id = film->audio_type;
00149 st->codec->codec_tag = 1;
00150 st->codec->channels = film->audio_channels;
00151 st->codec->bits_per_sample = film->audio_bits;
00152 st->codec->sample_rate = film->audio_samplerate;
00153 st->codec->bit_rate = st->codec->channels * st->codec->sample_rate *
00154 st->codec->bits_per_sample;
00155 st->codec->block_align = st->codec->channels *
00156 st->codec->bits_per_sample / 8;
00157 }
00158
00159
00160 if (get_buffer(pb, scratch, 16) != 16)
00161 return AVERROR(EIO);
00162 if (AV_RB32(&scratch[0]) != STAB_TAG)
00163 return AVERROR_INVALIDDATA;
00164 film->base_clock = AV_RB32(&scratch[8]);
00165 film->sample_count = AV_RB32(&scratch[12]);
00166 if(film->sample_count >= UINT_MAX / sizeof(film_sample_t))
00167 return -1;
00168 film->sample_table = av_malloc(film->sample_count * sizeof(film_sample_t));
00169
00170 for(i=0; i<s->nb_streams; i++)
00171 av_set_pts_info(s->streams[i], 33, 1, film->base_clock);
00172
00173 audio_frame_counter = 0;
00174 for (i = 0; i < film->sample_count; i++) {
00175
00176 if (get_buffer(pb, scratch, 16) != 16) {
00177 av_free(film->sample_table);
00178 return AVERROR(EIO);
00179 }
00180 film->sample_table[i].sample_offset =
00181 data_offset + AV_RB32(&scratch[0]);
00182 film->sample_table[i].sample_size = AV_RB32(&scratch[4]);
00183 if (AV_RB32(&scratch[8]) == 0xFFFFFFFF) {
00184 film->sample_table[i].stream = film->audio_stream_index;
00185 film->sample_table[i].pts = audio_frame_counter;
00186 film->sample_table[i].pts *= film->base_clock;
00187 film->sample_table[i].pts /= film->audio_samplerate;
00188
00189 audio_frame_counter += (film->sample_table[i].sample_size /
00190 (film->audio_channels * film->audio_bits / 8));
00191 } else {
00192 film->sample_table[i].stream = film->video_stream_index;
00193 film->sample_table[i].pts = AV_RB32(&scratch[8]) & 0x7FFFFFFF;
00194 film->sample_table[i].keyframe = (scratch[8] & 0x80) ? 0 : 1;
00195 }
00196 }
00197
00198 film->current_sample = 0;
00199
00200 return 0;
00201 }
00202
00203 static int film_read_packet(AVFormatContext *s,
00204 AVPacket *pkt)
00205 {
00206 FilmDemuxContext *film = s->priv_data;
00207 ByteIOContext *pb = s->pb;
00208 film_sample_t *sample;
00209 int ret = 0;
00210 int i;
00211 int left, right;
00212
00213 if (film->current_sample >= film->sample_count)
00214 return AVERROR(EIO);
00215
00216 sample = &film->sample_table[film->current_sample];
00217
00218
00219 url_fseek(pb, sample->sample_offset, SEEK_SET);
00220
00221
00222 if ((sample->stream == film->video_stream_index) &&
00223 (film->video_type == CODEC_ID_CINEPAK)) {
00224 pkt->pos= url_ftell(pb);
00225 if (av_new_packet(pkt, sample->sample_size))
00226 return AVERROR(ENOMEM);
00227 get_buffer(pb, pkt->data, sample->sample_size);
00228 } else if ((sample->stream == film->audio_stream_index) &&
00229 (film->audio_channels == 2)) {
00230
00231
00232 if (av_new_packet(pkt, sample->sample_size))
00233 return AVERROR(ENOMEM);
00234
00235
00236 if (sample->sample_size > film->stereo_buffer_size) {
00237 av_free(film->stereo_buffer);
00238 film->stereo_buffer_size = sample->sample_size;
00239 film->stereo_buffer = av_malloc(film->stereo_buffer_size);
00240 }
00241
00242 pkt->pos= url_ftell(pb);
00243 ret = get_buffer(pb, film->stereo_buffer, sample->sample_size);
00244 if (ret != sample->sample_size)
00245 ret = AVERROR(EIO);
00246
00247 left = 0;
00248 right = sample->sample_size / 2;
00249 for (i = 0; i < sample->sample_size; ) {
00250 if (film->audio_bits == 8) {
00251 pkt->data[i++] = film->stereo_buffer[left++];
00252 pkt->data[i++] = film->stereo_buffer[right++];
00253 } else {
00254 pkt->data[i++] = film->stereo_buffer[left++];
00255 pkt->data[i++] = film->stereo_buffer[left++];
00256 pkt->data[i++] = film->stereo_buffer[right++];
00257 pkt->data[i++] = film->stereo_buffer[right++];
00258 }
00259 }
00260 } else {
00261 ret= av_get_packet(pb, pkt, sample->sample_size);
00262 if (ret != sample->sample_size)
00263 ret = AVERROR(EIO);
00264 }
00265
00266 pkt->stream_index = sample->stream;
00267 pkt->pts = sample->pts;
00268
00269 film->current_sample++;
00270
00271 return ret;
00272 }
00273
00274 static int film_read_close(AVFormatContext *s)
00275 {
00276 FilmDemuxContext *film = s->priv_data;
00277
00278 av_free(film->sample_table);
00279 av_free(film->stereo_buffer);
00280
00281 return 0;
00282 }
00283
00284 AVInputFormat segafilm_demuxer = {
00285 "film_cpk",
00286 "Sega FILM/CPK format",
00287 sizeof(FilmDemuxContext),
00288 film_probe,
00289 film_read_header,
00290 film_read_packet,
00291 film_read_close,
00292 };