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

   1 /*
   2  * Simbiosis game demuxer
   3  *
   4  * Copyright (C) 2021 Paul B Mahol
   5  *
   6  * This file is part of FFmpeg.
   7  *
   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.
  12  *
  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.
  17  *
  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
  21  */
  22
  23 #include "avformat.h"
  24 #include "internal.h"
  25 #include "libavutil/intreadwrite.h"
  26 #include "libavutil/avassert.h"
  27 #include "libavutil/internal.h"
  28 #include "argo_asf.h"
  29
  30 #define IMX_TAG MKTAG('I', 'M', 'A', 'X')
  31
  32 typedef struct SimbiosisIMXDemuxContext {
  33     uint8_t pal[AVPALETTE_SIZE];
  34     int pal_changed;
  35 } SimbiosisIMXDemuxContext;
  36
  37 static int simbiosis_imx_probe(const AVProbeData *p)
  38 {
  39     if (AV_RL32(p->buf) != IMX_TAG)
  40         return 0;
  41     if (AV_RN32(p->buf+4) == 0)
  42         return 0;
  43     if (AV_RN16(p->buf+8) == 0)
  44         return 0;
  45     if (AV_RL16(p->buf+10) != 0x102)
  46         return 0;
  47
  48     return AVPROBE_SCORE_EXTENSION + 10;
  49 }
  50
  51 static int simbiosis_imx_read_header(AVFormatContext *s)
  52 {
  53     AVIOContext *pb = s->pb;
  54     AVStream *vst, *ast;
  55     int rate;
  56
  57     vst = avformat_new_stream(s, NULL);
  58     ast = avformat_new_stream(s, NULL);
  59     if (!vst || !ast)
  60         return AVERROR(ENOMEM);
  61
  62     avio_skip(pb, 4);
  63
  64     vst->codecpar->codec_type = AVMEDIA_TYPE_VIDEO;
  65     vst->codecpar->codec_tag  = 0;
  66     vst->codecpar->format     = AV_PIX_FMT_PAL8;
  67     vst->codecpar->codec_id   = AV_CODEC_ID_SIMBIOSIS_IMX;
  68     vst->start_time = 0;
  69     vst->nb_frames = avio_rl32(pb);
  70     rate = avio_rl16(pb);
  71     avio_skip(pb, 12);
  72
  73     avpriv_set_pts_info(vst, 64, 1, rate);
  74
  75     ast->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
  76     ast->codecpar->codec_tag  = 0;
  77     ast->codecpar->codec_id   = AV_CODEC_ID_PCM_U8;
  78     ast->codecpar->channels   = 1;
  79     ast->codecpar->channel_layout = AV_CH_LAYOUT_MONO;
  80     ast->codecpar->sample_rate = 22050;
  81     ast->start_time = 0;
  82
  83     avpriv_set_pts_info(ast, 64, 1, 22050);
  84
  85     return 0;
  86 }
  87
  88 static int simbiosis_imx_read_packet(AVFormatContext *s, AVPacket *pkt)
  89 {
  90     AVIOContext *pb = s->pb;
  91     SimbiosisIMXDemuxContext *imx = s->priv_data;
  92     uint32_t chunk_size, chunk_type;
  93     int64_t pos = avio_tell(pb);
  94     int ret, idx = -1;
  95
  96 retry:
  97     if (avio_feof(pb))
  98         return AVERROR_EOF;
  99
 100     chunk_size = avio_rl32(pb);
 101     chunk_type = avio_rl32(pb);
 102
 103     switch (chunk_type) {
 104     case 0xAAFF:
 105         return AVERROR_EOF;
 106     case 0xAA99:
 107         idx = 1;
 108         break;
 109     case 0xAA97:
 110         idx = 0;
 111         break;
 112     case 0xAA98:
 113         for (int i = 0; i < chunk_size / 3; i++) {
 114             unsigned r = avio_r8(pb) << 18;
 115             unsigned g = avio_r8(pb) << 10;
 116             unsigned b = avio_r8(pb) <<  2;
 117
 118             AV_WL32(imx->pal + i * 4, (0xFFU << 24) | r | g | b);
 119         }
 120         imx->pal_changed = 1;
 121         idx = -1;
 122         break;
 123     default:
 124         return AVERROR_INVALIDDATA;
 125     }
 126
 127     if (idx == -1)
 128         goto retry;
 129
 130     ret = av_get_packet(pb, pkt, chunk_size);
 131     if (ret < 0)
 132         return ret;
 133
 134     if (imx->pal_changed && idx == 0) {
 135         uint8_t *pal = av_packet_new_side_data(pkt, AV_PKT_DATA_PALETTE,
 136                                                AVPALETTE_SIZE);
 137         if (!pal)
 138             return AVERROR(ENOMEM);
 139         memcpy(pal, imx->pal, AVPALETTE_SIZE);
 140         imx->pal_changed = 0;
 141         pkt->flags |= AV_PKT_FLAG_KEY;
 142     }
 143
 144     pkt->pos = pos;
 145     pkt->stream_index = idx;
 146     pkt->duration = idx ? chunk_size : 1;
 147
 148     return ret;
 149 }
 150
 151 AVInputFormat ff_simbiosis_imx_demuxer = {
 152     .name           = "simbiosis_imx",
 153     .long_name      = NULL_IF_CONFIG_SMALL("Simbiosis Interactive IMX"),
 154     .priv_data_size = sizeof(SimbiosisIMXDemuxContext),
 155     .read_probe     = simbiosis_imx_probe,
 156     .read_header    = simbiosis_imx_read_header,
 157     .read_packet    = simbiosis_imx_read_packet,
 158     .extensions     = "imx",
 159     .flags          = AVFMT_GENERIC_INDEX,
 160 };