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

   1 /*
   2  * Cryo Interactive Entertainment HNM4 demuxer
   3  *
   4  * Copyright (c) 2012 David Kment
   5  *
   6  * This file is part of Libav.
   7  *
   8  * Libav 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  * Libav 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 Libav; if not, write to the Free Software
  20  * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21  */
  22
  23 #include <inttypes.h>
  24
  25 #include "libavutil/intreadwrite.h"
  26 #include "avformat.h"
  27 #include "internal.h"
  28
  29 #define HNM4_TAG MKTAG('H', 'N', 'M', '4')
  30
  31 #define HNM4_SAMPLE_RATE 22050
  32 #define HNM4_FRAME_FPS 24
  33
  34 #define HNM4_CHUNK_ID_PL 19536
  35 #define HNM4_CHUNK_ID_IZ 23113
  36 #define HNM4_CHUNK_ID_IU 21833
  37 #define HNM4_CHUNK_ID_SD 17491
  38
  39 typedef struct Hnm4DemuxContext {
  40     uint8_t version;
  41     uint16_t width;
  42     uint16_t height;
  43     uint32_t filesize;
  44     uint32_t frames;
  45     uint32_t taboffset;
  46     uint16_t bits;
  47     uint16_t channels;
  48     uint32_t framesize;
  49     uint32_t currentframe;
  50     int64_t pts;
  51     uint32_t superchunk_remaining;
  52     AVPacket vpkt;
  53 } Hnm4DemuxContext;
  54
  55 static int hnm_probe(AVProbeData *p)
  56 {
  57     if (p->buf_size < 4)
  58         return 0;
  59
  60     // check for HNM4 header.
  61     // currently only HNM v4/v4A is supported
  62     if (AV_RL32(&p->buf[0]) == HNM4_TAG)
  63         return AVPROBE_SCORE_MAX;
  64
  65     return 0;
  66 }
  67
  68 static int hnm_read_header(AVFormatContext *s)
  69 {
  70     Hnm4DemuxContext *hnm = s->priv_data;
  71     AVIOContext *pb = s->pb;
  72     AVStream *vst;
  73
  74     /* default context members */
  75     hnm->pts = 0;
  76     av_init_packet(&hnm->vpkt);
  77     hnm->vpkt.data = NULL;
  78     hnm->vpkt.size = 0;
  79
  80     hnm->superchunk_remaining = 0;
  81
  82     avio_skip(pb, 8);
  83     hnm->width     = avio_rl16(pb);
  84     hnm->height    = avio_rl16(pb);
  85     hnm->filesize  = avio_rl32(pb);
  86     hnm->frames    = avio_rl32(pb);
  87     hnm->taboffset = avio_rl32(pb);
  88     hnm->bits      = avio_rl16(pb);
  89     hnm->channels  = avio_rl16(pb);
  90     hnm->framesize = avio_rl32(pb);
  91     avio_skip(pb, 32);
  92
  93     hnm->currentframe = 0;
  94
  95     if (hnm->width  < 320 || hnm->width  > 640 ||
  96         hnm->height < 150 || hnm->height > 480) {
  97         av_log(s, AV_LOG_ERROR,
  98                "invalid resolution: %ux%u\n", hnm->width, hnm->height);
  99         return AVERROR_INVALIDDATA;
 100     }
 101
 102     // TODO: find a better way to detect HNM4A
 103     if (hnm->width == 640)
 104         hnm->version = 0x4a;
 105     else
 106         hnm->version = 0x40;
 107
 108     if (!(vst = avformat_new_stream(s, NULL)))
 109         return AVERROR(ENOMEM);
 110
 111     vst->codec->codec_type = AVMEDIA_TYPE_VIDEO;
 112     vst->codec->codec_id   = AV_CODEC_ID_HNM4_VIDEO;
 113     vst->codec->codec_tag  = 0;
 114     vst->codec->width      = hnm->width;
 115     vst->codec->height     = hnm->height;
 116     vst->codec->extradata  = av_mallocz(1);
 117
 118     vst->codec->extradata_size = 1;
 119     memcpy(vst->codec->extradata, &hnm->version, 1);
 120
 121     vst->start_time = 0;
 122
 123     avpriv_set_pts_info(vst, 33, 1, HNM4_FRAME_FPS);
 124
 125     return 0;
 126 }
 127
 128 static int hnm_read_packet(AVFormatContext *s, AVPacket *pkt)
 129 {
 130     Hnm4DemuxContext *hnm = s->priv_data;
 131     AVIOContext *pb = s->pb;
 132     int ret = 0;
 133
 134     uint32_t superchunk_size, chunk_size;
 135     uint16_t chunk_id;
 136
 137     if (hnm->currentframe == hnm->frames || pb->eof_reached)
 138         return AVERROR_EOF;
 139
 140     if (hnm->superchunk_remaining == 0) {
 141         /* parse next superchunk */
 142         superchunk_size = avio_rl24(pb);
 143         avio_skip(pb, 1);
 144
 145         hnm->superchunk_remaining = superchunk_size - 4;
 146     }
 147
 148     chunk_size = avio_rl24(pb);
 149     avio_skip(pb, 1);
 150     chunk_id = avio_rl16(pb);
 151     avio_skip(pb, 2);
 152
 153     if (chunk_size > hnm->superchunk_remaining) {
 154         av_log(s, AV_LOG_ERROR,
 155                "invalid chunk size: %"PRIu32", offset: %"PRId64"\n",
 156                chunk_size, avio_tell(pb));
 157         avio_skip(pb, hnm->superchunk_remaining - 8);
 158         hnm->superchunk_remaining = 0;
 159     }
 160
 161     switch (chunk_id) {
 162     case HNM4_CHUNK_ID_PL:
 163     case HNM4_CHUNK_ID_IZ:
 164     case HNM4_CHUNK_ID_IU:
 165         avio_seek(pb, -8, SEEK_CUR);
 166         ret += av_get_packet(pb, pkt, chunk_size);
 167         hnm->superchunk_remaining -= chunk_size;
 168         if (chunk_id == HNM4_CHUNK_ID_IZ || chunk_id == HNM4_CHUNK_ID_IU)
 169             hnm->currentframe++;
 170         break;
 171
 172     case HNM4_CHUNK_ID_SD:
 173         avio_skip(pb, chunk_size - 8);
 174         hnm->superchunk_remaining -= chunk_size;
 175         break;
 176
 177     default:
 178         av_log(s, AV_LOG_WARNING, "unknown chunk found: %"PRIu16", offset: %"PRId64"\n",
 179                chunk_id, avio_tell(pb));
 180         avio_skip(pb, chunk_size - 8);
 181         hnm->superchunk_remaining -= chunk_size;
 182         break;
 183     }
 184
 185     return ret;
 186 }
 187
 188 static int hnm_read_close(AVFormatContext *s)
 189 {
 190     Hnm4DemuxContext *hnm = s->priv_data;
 191
 192     if (hnm->vpkt.size > 0)
 193         av_packet_unref(&hnm->vpkt);
 194
 195     return 0;
 196 }
 197
 198 AVInputFormat ff_hnm_demuxer = {
 199     .name           = "hnm",
 200     .long_name      = NULL_IF_CONFIG_SMALL("Cryo HNM v4"),
 201     .priv_data_size = sizeof(Hnm4DemuxContext),
 202     .read_probe     = hnm_probe,
 203     .read_header    = hnm_read_header,
 204     .read_packet    = hnm_read_packet,
 205     .read_close     = hnm_read_close,
 206     .flags          = AVFMT_NO_BYTE_SEEK | AVFMT_NOGENSEARCH | AVFMT_NOBINSEARCH
 207 };