git.sesse.net Git - ffmpeg/blob - libavformat/tiertexseq.c

   1 /*
   2  * Tiertex Limited SEQ File Demuxer
   3  * Copyright (c) 2006 Gregory Montoir (cyx@users.sourceforge.net)
   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  * Tiertex Limited SEQ file demuxer
  25  */
  26
  27 #include "libavutil/channel_layout.h"
  28 #include "avformat.h"
  29 #include "internal.h"
  30
  31 #define SEQ_FRAME_SIZE         6144
  32 #define SEQ_FRAME_W            256
  33 #define SEQ_FRAME_H            128
  34 #define SEQ_NUM_FRAME_BUFFERS  30
  35 #define SEQ_AUDIO_BUFFER_SIZE  882
  36 #define SEQ_SAMPLE_RATE        22050
  37 #define SEQ_FRAME_RATE         25
  38
  39
  40 typedef struct TiertexSeqFrameBuffer {
  41     int fill_size;
  42     int data_size;
  43     unsigned char *data;
  44 } TiertexSeqFrameBuffer;
  45
  46 typedef struct SeqDemuxContext {
  47     int audio_stream_index;
  48     int video_stream_index;
  49     int current_frame_pts;
  50     int current_frame_offs;
  51     TiertexSeqFrameBuffer frame_buffers[SEQ_NUM_FRAME_BUFFERS];
  52     int frame_buffers_count;
  53     unsigned int current_audio_data_size;
  54     unsigned int current_audio_data_offs;
  55     unsigned int current_pal_data_size;
  56     unsigned int current_pal_data_offs;
  57     unsigned int current_video_data_size;
  58     unsigned char *current_video_data_ptr;
  59     int audio_buffer_full;
  60 } SeqDemuxContext;
  61
  62
  63 static int seq_probe(const AVProbeData *p)
  64 {
  65     int i;
  66
  67     if (p->buf_size < 258)
  68         return 0;
  69
  70     /* there's no real header in a .seq file, the only thing they have in common */
  71     /* is the first 256 bytes of the file which are always filled with 0 */
  72     for (i = 0; i < 256; i++)
  73         if (p->buf[i])
  74             return 0;
  75
  76     if(p->buf[256]==0 && p->buf[257]==0)
  77         return 0;
  78
  79     /* only one fourth of the score since the previous check is too naive */
  80     return AVPROBE_SCORE_MAX / 4;
  81 }
  82
  83 static int seq_init_frame_buffers(SeqDemuxContext *seq, AVIOContext *pb)
  84 {
  85     int i, sz;
  86     TiertexSeqFrameBuffer *seq_buffer;
  87
  88     avio_seek(pb, 256, SEEK_SET);
  89
  90     for (i = 0; i < SEQ_NUM_FRAME_BUFFERS; i++) {
  91         sz = avio_rl16(pb);
  92         if (sz == 0)
  93             break;
  94         else {
  95             seq_buffer = &seq->frame_buffers[i];
  96             seq_buffer->fill_size = 0;
  97             seq_buffer->data_size = sz;
  98             seq_buffer->data = av_malloc(sz);
  99             if (!seq_buffer->data)
 100                 return AVERROR(ENOMEM);
 101         }
 102     }
 103     seq->frame_buffers_count = i;
 104     return 0;
 105 }
 106
 107 static int seq_fill_buffer(SeqDemuxContext *seq, AVIOContext *pb, int buffer_num, unsigned int data_offs, int data_size)
 108 {
 109     TiertexSeqFrameBuffer *seq_buffer;
 110
 111     if (buffer_num >= SEQ_NUM_FRAME_BUFFERS)
 112         return AVERROR_INVALIDDATA;
 113
 114     seq_buffer = &seq->frame_buffers[buffer_num];
 115     if (seq_buffer->fill_size + data_size > seq_buffer->data_size || data_size <= 0)
 116         return AVERROR_INVALIDDATA;
 117
 118     avio_seek(pb, seq->current_frame_offs + data_offs, SEEK_SET);
 119     if (avio_read(pb, seq_buffer->data + seq_buffer->fill_size, data_size) != data_size)
 120         return AVERROR(EIO);
 121
 122     seq_buffer->fill_size += data_size;
 123     return 0;
 124 }
 125
 126 static int seq_parse_frame_data(SeqDemuxContext *seq, AVIOContext *pb)
 127 {
 128     unsigned int offset_table[4], buffer_num[4];
 129     TiertexSeqFrameBuffer *seq_buffer;
 130     int i, e, err;
 131
 132     seq->current_frame_offs += SEQ_FRAME_SIZE;
 133     avio_seek(pb, seq->current_frame_offs, SEEK_SET);
 134
 135     /* sound data */
 136     seq->current_audio_data_offs = avio_rl16(pb);
 137     if (seq->current_audio_data_offs) {
 138         seq->current_audio_data_size = SEQ_AUDIO_BUFFER_SIZE * 2;
 139     } else {
 140         seq->current_audio_data_size = 0;
 141     }
 142
 143     /* palette data */
 144     seq->current_pal_data_offs = avio_rl16(pb);
 145     if (seq->current_pal_data_offs) {
 146         seq->current_pal_data_size = 768;
 147     } else {
 148         seq->current_pal_data_size = 0;
 149     }
 150
 151     /* video data */
 152     for (i = 0; i < 4; i++)
 153         buffer_num[i] = avio_r8(pb);
 154
 155     for (i = 0; i < 4; i++)
 156         offset_table[i] = avio_rl16(pb);
 157
 158     for (i = 0; i < 3; i++) {
 159         if (offset_table[i]) {
 160             for (e = i + 1; e < 3 && offset_table[e] == 0; e++);
 161             err = seq_fill_buffer(seq, pb, buffer_num[1 + i],
 162               offset_table[i],
 163               offset_table[e] - offset_table[i]);
 164             if (err)
 165                 return err;
 166         }
 167     }
 168
 169     if (buffer_num[0] != 255) {
 170         if (buffer_num[0] >= SEQ_NUM_FRAME_BUFFERS)
 171             return AVERROR_INVALIDDATA;
 172
 173         seq_buffer = &seq->frame_buffers[buffer_num[0]];
 174         seq->current_video_data_size = seq_buffer->fill_size;
 175         seq->current_video_data_ptr  = seq_buffer->data;
 176         seq_buffer->fill_size = 0;
 177     } else {
 178         seq->current_video_data_size = 0;
 179         seq->current_video_data_ptr  = 0;
 180     }
 181
 182     return 0;
 183 }
 184
 185 static int seq_read_close(AVFormatContext *s)
 186 {
 187     int i;
 188     SeqDemuxContext *seq = s->priv_data;
 189
 190     for (i = 0; i < SEQ_NUM_FRAME_BUFFERS; i++)
 191         av_freep(&seq->frame_buffers[i].data);
 192
 193     return 0;
 194 }
 195
 196 static int seq_read_header(AVFormatContext *s)
 197 {
 198     int i, rc;
 199     SeqDemuxContext *seq = s->priv_data;
 200     AVIOContext *pb = s->pb;
 201     AVStream *st;
 202
 203     /* init internal buffers */
 204     rc = seq_init_frame_buffers(seq, pb);
 205     if (rc)
 206         return rc;
 207
 208     seq->current_frame_offs = 0;
 209
 210     /* preload (no audio data, just buffer operations related data) */
 211     for (i = 1; i <= 100; i++) {
 212         rc = seq_parse_frame_data(seq, pb);
 213         if (rc)
 214             return rc;
 215     }
 216
 217     seq->current_frame_pts = 0;
 218
 219     seq->audio_buffer_full = 0;
 220
 221     /* initialize the video decoder stream */
 222     st = avformat_new_stream(s, NULL);
 223     if (!st)
 224         return AVERROR(ENOMEM);
 225
 226     avpriv_set_pts_info(st, 32, 1, SEQ_FRAME_RATE);
 227     seq->video_stream_index = st->index;
 228     st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
 229     st->codecpar->codec_id = AV_CODEC_ID_TIERTEXSEQVIDEO;
 230     st->codecpar->codec_tag = 0;  /* no fourcc */
 231     st->codecpar->width = SEQ_FRAME_W;
 232     st->codecpar->height = SEQ_FRAME_H;
 233
 234     /* initialize the audio decoder stream */
 235     st = avformat_new_stream(s, NULL);
 236     if (!st)
 237         return AVERROR(ENOMEM);
 238
 239     st->start_time = 0;
 240     avpriv_set_pts_info(st, 32, 1, SEQ_SAMPLE_RATE);
 241     seq->audio_stream_index = st->index;
 242     st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
 243     st->codecpar->codec_id = AV_CODEC_ID_PCM_S16BE;
 244     st->codecpar->codec_tag = 0;  /* no tag */
 245     st->codecpar->channels = 1;
 246     st->codecpar->channel_layout = AV_CH_LAYOUT_MONO;
 247     st->codecpar->sample_rate = SEQ_SAMPLE_RATE;
 248     st->codecpar->bits_per_coded_sample = 16;
 249     st->codecpar->bit_rate = st->codecpar->sample_rate * st->codecpar->bits_per_coded_sample * st->codecpar->channels;
 250     st->codecpar->block_align = st->codecpar->channels * st->codecpar->bits_per_coded_sample / 8;
 251
 252     return 0;
 253 }
 254
 255 static int seq_read_packet(AVFormatContext *s, AVPacket *pkt)
 256 {
 257     int rc;
 258     SeqDemuxContext *seq = s->priv_data;
 259     AVIOContext *pb = s->pb;
 260
 261     if (!seq->audio_buffer_full) {
 262         rc = seq_parse_frame_data(seq, pb);
 263         if (rc)
 264             return rc;
 265
 266         /* video packet */
 267         if (seq->current_pal_data_size + seq->current_video_data_size != 0) {
 268             if (av_new_packet(pkt, 1 + seq->current_pal_data_size + seq->current_video_data_size))
 269                 return AVERROR(ENOMEM);
 270
 271             pkt->data[0] = 0;
 272             if (seq->current_pal_data_size) {
 273                 pkt->data[0] |= 1;
 274                 avio_seek(pb, seq->current_frame_offs + seq->current_pal_data_offs, SEEK_SET);
 275                 if (avio_read(pb, &pkt->data[1], seq->current_pal_data_size) != seq->current_pal_data_size)
 276                     return AVERROR(EIO);
 277             }
 278             if (seq->current_video_data_size) {
 279                 pkt->data[0] |= 2;
 280                 memcpy(&pkt->data[1 + seq->current_pal_data_size],
 281                   seq->current_video_data_ptr,
 282                   seq->current_video_data_size);
 283             }
 284             pkt->stream_index = seq->video_stream_index;
 285             pkt->pts = seq->current_frame_pts;
 286
 287             /* sound buffer will be processed on next read_packet() call */
 288             seq->audio_buffer_full = 1;
 289             return 0;
 290        }
 291     }
 292
 293     /* audio packet */
 294     if (seq->current_audio_data_offs == 0) /* end of data reached */
 295         return AVERROR(EIO);
 296
 297     avio_seek(pb, seq->current_frame_offs + seq->current_audio_data_offs, SEEK_SET);
 298     rc = av_get_packet(pb, pkt, seq->current_audio_data_size);
 299     if (rc < 0)
 300         return rc;
 301
 302     pkt->stream_index = seq->audio_stream_index;
 303     seq->current_frame_pts++;
 304
 305     seq->audio_buffer_full = 0;
 306     return 0;
 307 }
 308
 309 AVInputFormat ff_tiertexseq_demuxer = {
 310     .name           = "tiertexseq",
 311     .long_name      = NULL_IF_CONFIG_SMALL("Tiertex Limited SEQ"),
 312     .priv_data_size = sizeof(SeqDemuxContext),
 313     .read_probe     = seq_probe,
 314     .read_header    = seq_read_header,
 315     .read_packet    = seq_read_packet,
 316     .read_close     = seq_read_close,
 317 };