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
  29 #define IMX_TAG MKTAG('I', 'M', 'A', 'X')
  30
  31 typedef struct SimbiosisIMXDemuxContext {
  32     uint8_t pal[AVPALETTE_SIZE];
  33     int pal_changed;
  34     int64_t first_video_packet_pos;
  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->duration =
  70     vst->nb_frames = avio_rl32(pb);
  71     rate = avio_rl16(pb);
  72     avio_skip(pb, 12);
  73
  74     avpriv_set_pts_info(vst, 64, 1, rate);
  75
  76     ast->codecpar->codec_type = AVMEDIA_TYPE_AUDIO;
  77     ast->codecpar->codec_tag  = 0;
  78     ast->codecpar->codec_id   = AV_CODEC_ID_PCM_U8;
  79     ast->codecpar->channels   = 1;
  80     ast->codecpar->channel_layout = AV_CH_LAYOUT_MONO;
  81     ast->codecpar->sample_rate = 22050;
  82     ast->start_time = 0;
  83
  84     avpriv_set_pts_info(ast, 64, 1, 22050);
  85
  86     return 0;
  87 }
  88
  89 static int simbiosis_imx_read_packet(AVFormatContext *s, AVPacket *pkt)
  90 {
  91     AVIOContext *pb = s->pb;
  92     SimbiosisIMXDemuxContext *imx = s->priv_data;
  93     uint32_t chunk_size, chunk_type;
  94     int64_t pos = avio_tell(pb);
  95     int ret, idx = -1;
  96
  97 retry:
  98     if (avio_feof(pb))
  99         return AVERROR_EOF;
 100
 101     chunk_size = avio_rl32(pb);
 102     chunk_type = avio_rl32(pb);
 103
 104     switch (chunk_type) {
 105     case 0xAAFF:
 106         return AVERROR_EOF;
 107     case 0xAA99:
 108         idx = 1;
 109         break;
 110     case 0xAA97:
 111         idx = 0;
 112         if (!imx->first_video_packet_pos)
 113             imx->first_video_packet_pos = pos;
 114         break;
 115     case 0xAA98:
 116         if (chunk_size > 256 * 3)
 117             return AVERROR_INVALIDDATA;
 118         for (int i = 0; i < chunk_size / 3; i++) {
 119             unsigned r = avio_r8(pb) << 18;
 120             unsigned g = avio_r8(pb) << 10;
 121             unsigned b = avio_r8(pb) <<  2;
 122
 123             AV_WL32(imx->pal + i * 4, (0xFFU << 24) | r | g | b);
 124         }
 125         imx->pal_changed = 1;
 126         idx = -1;
 127         break;
 128     default:
 129         return AVERROR_INVALIDDATA;
 130     }
 131
 132     if (idx == -1)
 133         goto retry;
 134
 135     ret = av_get_packet(pb, pkt, chunk_size);
 136     if (ret < 0)
 137         return ret;
 138
 139     if (imx->pal_changed && idx == 0) {
 140         uint8_t *pal = av_packet_new_side_data(pkt, AV_PKT_DATA_PALETTE,
 141                                                AVPALETTE_SIZE);
 142         if (!pal)
 143             return AVERROR(ENOMEM);
 144         memcpy(pal, imx->pal, AVPALETTE_SIZE);
 145         imx->pal_changed = 0;
 146         if (pos <= imx->first_video_packet_pos)
 147             pkt->flags |= AV_PKT_FLAG_KEY;
 148     } else if (idx == 1) {
 149         pkt->flags |= AV_PKT_FLAG_KEY;
 150     }
 151
 152     pkt->pos = pos;
 153     pkt->stream_index = idx;
 154     pkt->duration = idx ? chunk_size : 1;
 155
 156     return ret;
 157 }
 158
 159 AVInputFormat ff_simbiosis_imx_demuxer = {
 160     .name           = "simbiosis_imx",
 161     .long_name      = NULL_IF_CONFIG_SMALL("Simbiosis Interactive IMX"),
 162     .priv_data_size = sizeof(SimbiosisIMXDemuxContext),
 163     .read_probe     = simbiosis_imx_probe,
 164     .read_header    = simbiosis_imx_read_header,
 165     .read_packet    = simbiosis_imx_read_packet,
 166     .extensions     = "imx",
 167     .flags          = AVFMT_GENERIC_INDEX,
 168 };