00001
00002
00003
00004
00005
00006
00007
00008
00009
00010
00011
00012
00013
00014
00015
00016
00017
00018
00019
00020
00021
00071 #include "avformat.h"
00072
00073 #define HUFFMAN_TABLE_SIZE (64 * 1024)
00074 #define FRAME_PTS_INC (90000 / 14)
00075
00076 typedef struct IdcinDemuxContext {
00077 int video_stream_index;
00078 int audio_stream_index;
00079 int audio_chunk_size1;
00080 int audio_chunk_size2;
00081
00082
00083 int current_audio_chunk;
00084 int next_chunk_is_video;
00085 int audio_present;
00086
00087 int64_t pts;
00088
00089 AVPaletteControl palctrl;
00090 } IdcinDemuxContext;
00091
00092 static int idcin_probe(AVProbeData *p)
00093 {
00094 unsigned int number;
00095
00096
00097
00098
00099
00100
00101
00102
00103
00104
00105
00106
00107
00108 number = AV_RL32(&p->buf[0]);
00109 if ((number == 0) || (number > 1024))
00110 return 0;
00111
00112
00113 number = AV_RL32(&p->buf[4]);
00114 if ((number == 0) || (number > 1024))
00115 return 0;
00116
00117
00118 number = AV_RL32(&p->buf[8]);
00119 if ((number != 0) && ((number < 8000) | (number > 48000)))
00120 return 0;
00121
00122
00123 number = AV_RL32(&p->buf[12]);
00124 if (number > 2)
00125 return 0;
00126
00127
00128 number = AV_RL32(&p->buf[16]);
00129 if (number > 2)
00130 return 0;
00131
00132
00133 return AVPROBE_SCORE_MAX / 2;
00134 }
00135
00136 static int idcin_read_header(AVFormatContext *s,
00137 AVFormatParameters *ap)
00138 {
00139 ByteIOContext *pb = s->pb;
00140 IdcinDemuxContext *idcin = s->priv_data;
00141 AVStream *st;
00142 unsigned int width, height;
00143 unsigned int sample_rate, bytes_per_sample, channels;
00144
00145
00146 width = get_le32(pb);
00147 height = get_le32(pb);
00148 sample_rate = get_le32(pb);
00149 bytes_per_sample = get_le32(pb);
00150 channels = get_le32(pb);
00151
00152 st = av_new_stream(s, 0);
00153 if (!st)
00154 return AVERROR(ENOMEM);
00155 av_set_pts_info(st, 33, 1, 90000);
00156 idcin->video_stream_index = st->index;
00157 st->codec->codec_type = CODEC_TYPE_VIDEO;
00158 st->codec->codec_id = CODEC_ID_IDCIN;
00159 st->codec->codec_tag = 0;
00160 st->codec->width = width;
00161 st->codec->height = height;
00162
00163
00164 st->codec->extradata_size = HUFFMAN_TABLE_SIZE;
00165 st->codec->extradata = av_malloc(HUFFMAN_TABLE_SIZE);
00166 if (get_buffer(pb, st->codec->extradata, HUFFMAN_TABLE_SIZE) !=
00167 HUFFMAN_TABLE_SIZE)
00168 return AVERROR(EIO);
00169
00170 st->codec->palctrl = &idcin->palctrl;
00171
00172
00173 if (sample_rate) {
00174 idcin->audio_present = 1;
00175 st = av_new_stream(s, 0);
00176 if (!st)
00177 return AVERROR(ENOMEM);
00178 av_set_pts_info(st, 33, 1, 90000);
00179 idcin->audio_stream_index = st->index;
00180 st->codec->codec_type = CODEC_TYPE_AUDIO;
00181 st->codec->codec_tag = 1;
00182 st->codec->channels = channels;
00183 st->codec->sample_rate = sample_rate;
00184 st->codec->bits_per_sample = bytes_per_sample * 8;
00185 st->codec->bit_rate = sample_rate * bytes_per_sample * 8 * channels;
00186 st->codec->block_align = bytes_per_sample * channels;
00187 if (bytes_per_sample == 1)
00188 st->codec->codec_id = CODEC_ID_PCM_U8;
00189 else
00190 st->codec->codec_id = CODEC_ID_PCM_S16LE;
00191
00192 if (sample_rate % 14 != 0) {
00193 idcin->audio_chunk_size1 = (sample_rate / 14) *
00194 bytes_per_sample * channels;
00195 idcin->audio_chunk_size2 = (sample_rate / 14 + 1) *
00196 bytes_per_sample * channels;
00197 } else {
00198 idcin->audio_chunk_size1 = idcin->audio_chunk_size2 =
00199 (sample_rate / 14) * bytes_per_sample * channels;
00200 }
00201 idcin->current_audio_chunk = 0;
00202 } else
00203 idcin->audio_present = 1;
00204
00205 idcin->next_chunk_is_video = 1;
00206 idcin->pts = 0;
00207
00208 return 0;
00209 }
00210
00211 static int idcin_read_packet(AVFormatContext *s,
00212 AVPacket *pkt)
00213 {
00214 int ret;
00215 unsigned int command;
00216 unsigned int chunk_size;
00217 IdcinDemuxContext *idcin = s->priv_data;
00218 ByteIOContext *pb = s->pb;
00219 int i;
00220 int palette_scale;
00221 unsigned char r, g, b;
00222 unsigned char palette_buffer[768];
00223
00224 if (url_feof(s->pb))
00225 return AVERROR(EIO);
00226
00227 if (idcin->next_chunk_is_video) {
00228 command = get_le32(pb);
00229 if (command == 2) {
00230 return AVERROR(EIO);
00231 } else if (command == 1) {
00232
00233 idcin->palctrl.palette_changed = 1;
00234 if (get_buffer(pb, palette_buffer, 768) != 768)
00235 return AVERROR(EIO);
00236
00237 palette_scale = 2;
00238 for (i = 0; i < 768; i++)
00239 if (palette_buffer[i] > 63) {
00240 palette_scale = 0;
00241 break;
00242 }
00243
00244 for (i = 0; i < 256; i++) {
00245 r = palette_buffer[i * 3 ] << palette_scale;
00246 g = palette_buffer[i * 3 + 1] << palette_scale;
00247 b = palette_buffer[i * 3 + 2] << palette_scale;
00248 idcin->palctrl.palette[i] = (r << 16) | (g << 8) | (b);
00249 }
00250 }
00251
00252 chunk_size = get_le32(pb);
00253
00254 url_fseek(pb, 4, SEEK_CUR);
00255 chunk_size -= 4;
00256 ret= av_get_packet(pb, pkt, chunk_size);
00257 if (ret != chunk_size)
00258 return AVERROR(EIO);
00259 pkt->stream_index = idcin->video_stream_index;
00260 pkt->pts = idcin->pts;
00261 } else {
00262
00263 if (idcin->current_audio_chunk)
00264 chunk_size = idcin->audio_chunk_size2;
00265 else
00266 chunk_size = idcin->audio_chunk_size1;
00267 ret= av_get_packet(pb, pkt, chunk_size);
00268 if (ret != chunk_size)
00269 return AVERROR(EIO);
00270 pkt->stream_index = idcin->audio_stream_index;
00271 pkt->pts = idcin->pts;
00272
00273 idcin->current_audio_chunk ^= 1;
00274 idcin->pts += FRAME_PTS_INC;
00275 }
00276
00277 if (idcin->audio_present)
00278 idcin->next_chunk_is_video ^= 1;
00279
00280 return ret;
00281 }
00282
00283 static int idcin_read_close(AVFormatContext *s)
00284 {
00285
00286 return 0;
00287 }
00288
00289 AVInputFormat idcin_demuxer = {
00290 "idcin",
00291 "Id CIN format",
00292 sizeof(IdcinDemuxContext),
00293 idcin_probe,
00294 idcin_read_header,
00295 idcin_read_packet,
00296 idcin_read_close,
00297 };