00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00022
00031 #include "libavutil/intreadwrite.h"
00032 #include "libavcodec/iff.h"
00033 #include "avformat.h"
00034
00035 #define ID_8SVX MKTAG('8','S','V','X')
00036 #define ID_VHDR MKTAG('V','H','D','R')
00037 #define ID_ATAK MKTAG('A','T','A','K')
00038 #define ID_RLSE MKTAG('R','L','S','E')
00039 #define ID_CHAN MKTAG('C','H','A','N')
00040 #define ID_PBM MKTAG('P','B','M',' ')
00041 #define ID_ILBM MKTAG('I','L','B','M')
00042 #define ID_BMHD MKTAG('B','M','H','D')
00043 #define ID_CMAP MKTAG('C','M','A','P')
00044
00045 #define ID_FORM MKTAG('F','O','R','M')
00046 #define ID_ANNO MKTAG('A','N','N','O')
00047 #define ID_AUTH MKTAG('A','U','T','H')
00048 #define ID_CHRS MKTAG('C','H','R','S')
00049 #define ID_COPYRIGHT MKTAG('(','c',')',' ')
00050 #define ID_CSET MKTAG('C','S','E','T')
00051 #define ID_FVER MKTAG('F','V','E','R')
00052 #define ID_NAME MKTAG('N','A','M','E')
00053 #define ID_TEXT MKTAG('T','E','X','T')
00054 #define ID_BODY MKTAG('B','O','D','Y')
00055 #define ID_ANNO MKTAG('A','N','N','O')
00056
00057 #define LEFT 2
00058 #define RIGHT 4
00059 #define STEREO 6
00060
00061 #define PACKET_SIZE 1024
00062
00063 typedef enum {COMP_NONE, COMP_FIB, COMP_EXP} svx8_compression_type;
00064 typedef enum {BITMAP_RAW, BITMAP_BYTERUN1} bitmap_compression_type;
00065
00066 typedef struct {
00067 uint32_t body_size;
00068 uint32_t sent_bytes;
00069 uint32_t audio_frame_count;
00070 } IffDemuxContext;
00071
00072
00073 static void interleave_stereo(const uint8_t *src, uint8_t *dest, int size)
00074 {
00075 uint8_t *end = dest + size;
00076 size = size>>1;
00077
00078 while(dest < end) {
00079 *dest++ = *src;
00080 *dest++ = *(src+size);
00081 src++;
00082 }
00083 }
00084
00085 static int iff_probe(AVProbeData *p)
00086 {
00087 const uint8_t *d = p->buf;
00088
00089 if ( AV_RL32(d) == ID_FORM &&
00090 (AV_RL32(d+8) == ID_8SVX || AV_RL32(d+8) == ID_PBM || AV_RL32(d+8) == ID_ILBM) )
00091 return AVPROBE_SCORE_MAX;
00092 return 0;
00093 }
00094
00095 static int iff_read_header(AVFormatContext *s,
00096 AVFormatParameters *ap)
00097 {
00098 IffDemuxContext *iff = s->priv_data;
00099 ByteIOContext *pb = s->pb;
00100 AVStream *st;
00101 uint32_t chunk_id, data_size;
00102 int padding, done = 0;
00103 int compression = -1;
00104 char *buf;
00105
00106 st = av_new_stream(s, 0);
00107 if (!st)
00108 return AVERROR(ENOMEM);
00109
00110 st->codec->channels = 1;
00111 url_fskip(pb, 8);
00112
00113 st->codec->codec_tag = get_le32(pb);
00114
00115 while(!done && !url_feof(pb)) {
00116 chunk_id = get_le32(pb);
00117 data_size = get_be32(pb);
00118 padding = data_size & 1;
00119
00120 switch(chunk_id) {
00121 case ID_VHDR:
00122 st->codec->codec_type = AVMEDIA_TYPE_AUDIO;
00123 url_fskip(pb, 12);
00124 st->codec->sample_rate = get_be16(pb);
00125 url_fskip(pb, 1);
00126 compression = get_byte(pb);
00127 url_fskip(pb, 4);
00128 break;
00129
00130 case ID_BODY:
00131 iff->body_size = data_size;
00132 done = 1;
00133 break;
00134
00135 case ID_CHAN:
00136 st->codec->channels = (get_be32(pb) < 6) ? 1 : 2;
00137 break;
00138
00139 case ID_CMAP:
00140 st->codec->extradata_size = data_size;
00141 st->codec->extradata = av_malloc(data_size);
00142 if (!st->codec->extradata)
00143 return AVERROR(ENOMEM);
00144 if (get_buffer(pb, st->codec->extradata, data_size) < 0)
00145 return AVERROR(EIO);
00146 break;
00147
00148 case ID_BMHD:
00149 st->codec->codec_type = AVMEDIA_TYPE_VIDEO;
00150 st->codec->width = get_be16(pb);
00151 st->codec->height = get_be16(pb);
00152 url_fskip(pb, 4);
00153 st->codec->bits_per_coded_sample = get_byte(pb);
00154 url_fskip(pb, 1);
00155 compression = get_byte(pb);
00156 url_fskip(pb, 3);
00157 st->sample_aspect_ratio.num = get_byte(pb);
00158 st->sample_aspect_ratio.den = get_byte(pb);
00159 url_fskip(pb, 4);
00160 break;
00161
00162 case ID_ANNO:
00163 buf = av_malloc(data_size + 1);
00164 if (!buf)
00165 break;
00166 get_buffer(pb, buf, data_size);
00167 buf[data_size] = 0;
00168 av_metadata_set2(&s->metadata, "comment", buf, AV_METADATA_DONT_STRDUP_VAL);
00169 break;
00170
00171 default:
00172 url_fseek(pb, data_size + padding, SEEK_CUR);
00173 break;
00174 }
00175 }
00176
00177 switch(st->codec->codec_type) {
00178 case AVMEDIA_TYPE_AUDIO:
00179 av_set_pts_info(st, 32, 1, st->codec->sample_rate);
00180
00181 switch(compression) {
00182 case COMP_NONE:
00183 st->codec->codec_id = CODEC_ID_PCM_S8;
00184 break;
00185 case COMP_FIB:
00186 st->codec->codec_id = CODEC_ID_8SVX_FIB;
00187 break;
00188 case COMP_EXP:
00189 st->codec->codec_id = CODEC_ID_8SVX_EXP;
00190 break;
00191 default:
00192 av_log(s, AV_LOG_ERROR, "iff: unknown compression method\n");
00193 return -1;
00194 }
00195
00196 st->codec->bits_per_coded_sample = 8;
00197 st->codec->bit_rate = st->codec->channels * st->codec->sample_rate * st->codec->bits_per_coded_sample;
00198 st->codec->block_align = st->codec->channels * st->codec->bits_per_coded_sample;
00199 break;
00200
00201 case AVMEDIA_TYPE_VIDEO:
00202 switch (compression) {
00203 case BITMAP_RAW:
00204 if (st->codec->codec_tag == ID_ILBM) {
00205 st->codec->codec_id = CODEC_ID_IFF_ILBM;
00206 } else {
00207 st->codec->codec_id = CODEC_ID_RAWVIDEO;
00208 st->codec->pix_fmt = PIX_FMT_PAL8;
00209 st->codec->codec_tag = 0;
00210 }
00211 break;
00212 case BITMAP_BYTERUN1:
00213 st->codec->codec_id = CODEC_ID_IFF_BYTERUN1;
00214 break;
00215 default:
00216 av_log(s, AV_LOG_ERROR, "unknown compression method\n");
00217 return AVERROR_INVALIDDATA;
00218 }
00219 break;
00220 default:
00221 return -1;
00222 }
00223
00224 return 0;
00225 }
00226
00227 static int iff_read_packet(AVFormatContext *s,
00228 AVPacket *pkt)
00229 {
00230 IffDemuxContext *iff = s->priv_data;
00231 ByteIOContext *pb = s->pb;
00232 AVStream *st = s->streams[0];
00233 int ret;
00234
00235 if(iff->sent_bytes >= iff->body_size)
00236 return AVERROR(EIO);
00237
00238 if(s->streams[0]->codec->channels == 2) {
00239 uint8_t sample_buffer[PACKET_SIZE];
00240
00241 ret = get_buffer(pb, sample_buffer, PACKET_SIZE);
00242 if(av_new_packet(pkt, PACKET_SIZE) < 0) {
00243 av_log(s, AV_LOG_ERROR, "iff: cannot allocate packet \n");
00244 return AVERROR(ENOMEM);
00245 }
00246 interleave_stereo(sample_buffer, pkt->data, PACKET_SIZE);
00247 } else if (s->streams[0]->codec->codec_id == CODEC_ID_RAWVIDEO) {
00248 if(av_new_packet(pkt, iff->body_size + AVPALETTE_SIZE) < 0) {
00249 return AVERROR(ENOMEM);
00250 }
00251
00252 ret = ff_cmap_read_palette(st->codec, (uint32_t*)(pkt->data + iff->body_size));
00253 if (ret < 0)
00254 return ret;
00255 av_freep(&st->codec->extradata);
00256 st->codec->extradata_size = 0;
00257
00258 ret = get_buffer(pb, pkt->data, iff->body_size);
00259 } else if (s->streams[0]->codec->codec_type == AVMEDIA_TYPE_VIDEO) {
00260 ret = av_get_packet(pb, pkt, iff->body_size);
00261 } else {
00262 ret = av_get_packet(pb, pkt, PACKET_SIZE);
00263 }
00264
00265 if(iff->sent_bytes == 0)
00266 pkt->flags |= AV_PKT_FLAG_KEY;
00267
00268 if(s->streams[0]->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
00269 iff->sent_bytes += PACKET_SIZE;
00270 } else {
00271 iff->sent_bytes = iff->body_size;
00272 }
00273 pkt->stream_index = 0;
00274 if(s->streams[0]->codec->codec_type == AVMEDIA_TYPE_AUDIO) {
00275 pkt->pts = iff->audio_frame_count;
00276 iff->audio_frame_count += ret / s->streams[0]->codec->channels;
00277 }
00278 return ret;
00279 }
00280
00281 AVInputFormat iff_demuxer = {
00282 "IFF",
00283 NULL_IF_CONFIG_SMALL("IFF format"),
00284 sizeof(IffDemuxContext),
00285 iff_probe,
00286 iff_read_header,
00287 iff_read_packet,
00288 };