mirror of https://github.com/FFmpeg/FFmpeg.git
Originally committed as revision 8587 to svn://svn.ffmpeg.org/ffmpeg/trunkpull/126/head
Diego Biurrun
18 years ago
parent
b966340546
commit
efb0c399ca
9 changed files with 195 additions and 0 deletions
@ -0,0 +1,170 @@ |
||||
/*
|
||||
* THP Demuxer |
||||
* Copyright (c) 2007 Marco Gerards. |
||||
* |
||||
* This file is part of FFmpeg. |
||||
* |
||||
* FFmpeg is free software; you can redistribute it and/or |
||||
* modify it under the terms of the GNU Lesser General Public |
||||
* License as published by the Free Software Foundation; either |
||||
* version 2.1 of the License, or (at your option) any later version. |
||||
* |
||||
* FFmpeg is distributed in the hope that it will be useful, |
||||
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
||||
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
||||
* Lesser General Public License for more details. |
||||
* |
||||
* You should have received a copy of the GNU Lesser General Public |
||||
* License along with FFmpeg; if not, write to the Free Software |
||||
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
||||
*/ |
||||
|
||||
|
||||
#include "avformat.h" |
||||
#include "allformats.h" |
||||
|
||||
typedef struct ThpDemuxContext { |
||||
int version; |
||||
int first_frame; |
||||
int first_framesz; |
||||
int last_frame; |
||||
int compoff; |
||||
int framecnt; |
||||
AVRational fps; |
||||
int frame; |
||||
int next_frame; |
||||
int next_framesz; |
||||
int video_stream_index; |
||||
int compcount; |
||||
unsigned char components[16]; |
||||
AVStream* vst; |
||||
int has_audio; |
||||
} ThpDemuxContext; |
||||
|
||||
|
||||
static int thp_probe(AVProbeData *p) |
||||
{ |
||||
/* check file header */ |
||||
if (p->buf_size < 4) |
||||
return 0; |
||||
|
||||
if (AV_RL32(p->buf) == MKTAG('T', 'H', 'P', '\0')) |
||||
return AVPROBE_SCORE_MAX; |
||||
else |
||||
return 0; |
||||
} |
||||
|
||||
static int thp_read_header(AVFormatContext *s, |
||||
AVFormatParameters *ap) |
||||
{ |
||||
ThpDemuxContext *thp = s->priv_data; |
||||
AVStream *st; |
||||
ByteIOContext *pb = &s->pb; |
||||
int i; |
||||
|
||||
/* Read the file header. */ |
||||
|
||||
get_be32(pb); /* Skip Magic. */ |
||||
thp->version = get_be32(pb); |
||||
|
||||
get_be32(pb); /* Max buf size. */ |
||||
get_be32(pb); /* Max samples. */ |
||||
|
||||
thp->fps = av_d2q(av_int2flt(get_be32(pb)), INT_MAX); |
||||
thp->framecnt = get_be32(pb); |
||||
thp->first_framesz = get_be32(pb); |
||||
get_be32(pb); /* Data size. */ |
||||
|
||||
thp->compoff = get_be32(pb); |
||||
get_be32(pb); /* offsetDataOffset. */ |
||||
thp->first_frame = get_be32(pb); |
||||
thp->last_frame = get_be32(pb); |
||||
|
||||
thp->next_framesz = thp->first_framesz; |
||||
thp->next_frame = thp->first_frame; |
||||
|
||||
/* Read the component structure. */ |
||||
url_fseek (pb, thp->compoff, SEEK_SET); |
||||
thp->compcount = get_be32(pb); |
||||
|
||||
/* Read the list of component types. */ |
||||
get_buffer(pb, thp->components, 16); |
||||
|
||||
for (i = 0; i < thp->compcount; i++) { |
||||
if (thp->components[i] == 0) { |
||||
if (thp->vst != 0) |
||||
break; |
||||
|
||||
/* Video component. */ |
||||
st = av_new_stream(s, 0); |
||||
if (!st) |
||||
return AVERROR_NOMEM; |
||||
|
||||
/* The denominator and numerator are switched because 1/fps
|
||||
is required. */ |
||||
av_set_pts_info(st, 64, thp->fps.den, thp->fps.num); |
||||
st->codec->codec_type = CODEC_TYPE_VIDEO; |
||||
st->codec->codec_id = CODEC_ID_THP; |
||||
st->codec->codec_tag = 0; /* no fourcc */ |
||||
st->codec->width = get_be32(pb); |
||||
st->codec->height = get_be32(pb); |
||||
st->codec->sample_rate = av_q2d(thp->fps); |
||||
thp->vst = st; |
||||
thp->video_stream_index = st->index; |
||||
|
||||
if (thp->version == 0x11000) |
||||
get_be32(pb); /* Unknown. */ |
||||
} |
||||
else if (thp->components[i] == 1) { |
||||
/* XXX: Required for audio playback. */ |
||||
thp->has_audio = 1; |
||||
} |
||||
} |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
static int thp_read_packet(AVFormatContext *s, |
||||
AVPacket *pkt) |
||||
{ |
||||
ThpDemuxContext *thp = s->priv_data; |
||||
ByteIOContext *pb = &s->pb; |
||||
int size; |
||||
int ret; |
||||
|
||||
/* Terminate when last frame is reached. */ |
||||
if (thp->frame >= thp->framecnt) |
||||
return AVERROR_IO; |
||||
|
||||
url_fseek(pb, thp->next_frame, SEEK_SET); |
||||
|
||||
/* Locate the next frame and read out its size. */ |
||||
thp->next_frame += thp->next_framesz; |
||||
thp->next_framesz = get_be32(pb); |
||||
|
||||
get_be32(pb); /* Previous total size. */ |
||||
size = get_be32(pb); /* Total size of this frame. */ |
||||
|
||||
if (thp->has_audio) |
||||
get_be32(pb); /* Audio size. */ |
||||
|
||||
ret = av_get_packet(pb, pkt, size); |
||||
if (ret != size) { |
||||
av_free_packet(pkt); |
||||
return AVERROR_IO; |
||||
} |
||||
|
||||
pkt->stream_index = thp->video_stream_index; |
||||
thp->frame++; |
||||
|
||||
return 0; |
||||
} |
||||
|
||||
AVInputFormat thp_demuxer = { |
||||
"thp", |
||||
"THP", |
||||
sizeof(ThpDemuxContext), |
||||
thp_probe, |
||||
thp_read_header, |
||||
thp_read_packet |
||||
}; |
||||
Loading…
Reference in new issue