* some minor modification by Philip Gladston
* grab containes new code Originally committed as revision 388 to svn://svn.ffmpeg.org/ffmpeg/trunk
This commit is contained in:
28
libav/asf.c
28
libav/asf.c
@@ -29,6 +29,7 @@ typedef struct {
|
||||
/* use for reading */
|
||||
AVPacket pkt;
|
||||
int frag_offset;
|
||||
INT64 duration;
|
||||
} ASFStream;
|
||||
|
||||
typedef struct {
|
||||
@@ -120,6 +121,21 @@ static const GUID my_guid = {
|
||||
0, 0, 0, { 0, 0, 0, 0, 0, 0, 0, 0 },
|
||||
};
|
||||
|
||||
CodecTag codec_asf_bmp_tags[] = {
|
||||
{ CODEC_ID_H263, MKTAG('U', '2', '6', '3') },
|
||||
{ CODEC_ID_H263P, MKTAG('U', '2', '6', '3') },
|
||||
{ CODEC_ID_H263I, MKTAG('I', '2', '6', '3') }, /* intel h263 */
|
||||
{ CODEC_ID_MJPEG, MKTAG('M', 'J', 'P', 'G') },
|
||||
{ CODEC_ID_MPEG4, MKTAG('D', 'I', 'V', 'X') },
|
||||
{ CODEC_ID_MPEG4, MKTAG('d', 'i', 'v', 'x') },
|
||||
{ CODEC_ID_MPEG4, MKTAG(0x04, 0, 0, 0) }, /* some broken avi use this */
|
||||
{ CODEC_ID_MSMPEG4, MKTAG('M', 'P', '4', '3') }, /* default signature when using MSMPEG4 */
|
||||
{ CODEC_ID_MSMPEG4, MKTAG('D', 'I', 'V', '3') },
|
||||
{ 0, 0 },
|
||||
};
|
||||
|
||||
|
||||
|
||||
static void put_guid(ByteIOContext *s, const GUID *g)
|
||||
{
|
||||
int i;
|
||||
@@ -237,7 +253,7 @@ static int asf_write_header1(AVFormatContext *s, INT64 file_size, INT64 data_chu
|
||||
put_le64(pb, asf->duration); /* duration (in 100ns units) */
|
||||
put_le32(pb, 0); /* start time stamp */
|
||||
put_le32(pb, 0); /* ??? */
|
||||
put_le32(pb, 0); /* ??? */
|
||||
put_le32(pb, url_is_streamed(&s->pb) ? 1 : 0); /* ??? */
|
||||
put_le32(pb, asf->packet_size); /* packet size */
|
||||
put_le32(pb, asf->packet_size); /* packet size */
|
||||
put_le32(pb, 80 * asf->packet_size); /* frame_size ??? */
|
||||
@@ -310,7 +326,7 @@ static int asf_write_header1(AVFormatContext *s, INT64 file_size, INT64 data_chu
|
||||
put_le16(pb, 40); /* size */
|
||||
|
||||
/* BITMAPINFOHEADER header */
|
||||
put_bmp_header(pb, enc);
|
||||
put_bmp_header(pb, enc, codec_asf_bmp_tags);
|
||||
}
|
||||
end_header(pb, hpos);
|
||||
}
|
||||
@@ -332,7 +348,7 @@ static int asf_write_header1(AVFormatContext *s, INT64 file_size, INT64 data_chu
|
||||
put_le16(pb, codec_get_tag(codec_wav_tags, enc->codec_id));
|
||||
} else {
|
||||
put_le16(pb, 4);
|
||||
put_le32(pb, codec_get_tag(codec_bmp_tags, enc->codec_id));
|
||||
put_le32(pb, codec_get_tag(codec_asf_bmp_tags, enc->codec_id));
|
||||
}
|
||||
}
|
||||
end_header(pb, hpos);
|
||||
@@ -735,7 +751,7 @@ static int asf_read_header(AVFormatContext *s, AVFormatParameters *ap)
|
||||
get_le16(pb); /* depth */
|
||||
tag1 = get_le32(pb);
|
||||
st->codec.codec_tag = tag1;
|
||||
st->codec.codec_id = codec_get_id(codec_bmp_tags, tag1);
|
||||
st->codec.codec_id = codec_get_id(codec_asf_bmp_tags, tag1);
|
||||
url_fskip(pb, size - 5 * 4);
|
||||
}
|
||||
pos2 = url_ftell(pb);
|
||||
@@ -944,7 +960,11 @@ AVFormat asf_format = {
|
||||
"asf format",
|
||||
"application/octet-stream",
|
||||
"asf,wmv",
|
||||
#ifdef CONFIG_MP3LAME
|
||||
CODEC_ID_MP3LAME,
|
||||
#else
|
||||
CODEC_ID_MP2,
|
||||
#endif
|
||||
CODEC_ID_MSMPEG4,
|
||||
asf_write_header,
|
||||
asf_write_packet,
|
||||
|
@@ -248,6 +248,11 @@ static int audio_read_packet(AVFormatContext *s1, AVPacket *pkt)
|
||||
ret = read(s->fd, pkt->data, pkt->size);
|
||||
if (ret > 0)
|
||||
break;
|
||||
if (ret == -1 && (errno == EAGAIN || errno == EINTR)) {
|
||||
av_free_packet(pkt);
|
||||
pkt->size = 0;
|
||||
return 0;
|
||||
}
|
||||
if (!(ret == 0 || (ret == -1 && (errno == EAGAIN || errno == EINTR)))) {
|
||||
av_free_packet(pkt);
|
||||
return -EIO;
|
||||
|
@@ -12,6 +12,7 @@ typedef struct AVPacket {
|
||||
int stream_index;
|
||||
int flags;
|
||||
#define PKT_FLAG_KEY 0x0001
|
||||
#define PKT_FLAG_DROPPED_FRAME 0x0002
|
||||
} AVPacket;
|
||||
|
||||
int av_new_packet(AVPacket *pkt, int size);
|
||||
@@ -29,7 +30,7 @@ typedef struct AVFormatParameters {
|
||||
int channels;
|
||||
int width;
|
||||
int height;
|
||||
int pix_fmt;
|
||||
enum PixelFormat pix_fmt;
|
||||
} AVFormatParameters;
|
||||
|
||||
typedef struct AVFormat {
|
||||
|
@@ -9,15 +9,15 @@
|
||||
offset_t start_tag(ByteIOContext *pb, char *tag);
|
||||
void end_tag(ByteIOContext *pb, offset_t start);
|
||||
|
||||
void put_bmp_header(ByteIOContext *pb, AVCodecContext *enc);
|
||||
int put_wav_header(ByteIOContext *pb, AVCodecContext *enc);
|
||||
int wav_codec_get_id(unsigned int tag, int bps);
|
||||
|
||||
typedef struct CodecTag {
|
||||
int id;
|
||||
unsigned int tag;
|
||||
} CodecTag;
|
||||
|
||||
void put_bmp_header(ByteIOContext *pb, AVCodecContext *enc, CodecTag *tags);
|
||||
int put_wav_header(ByteIOContext *pb, AVCodecContext *enc);
|
||||
int wav_codec_get_id(unsigned int tag, int bps);
|
||||
|
||||
extern CodecTag codec_bmp_tags[];
|
||||
extern CodecTag codec_wav_tags[];
|
||||
|
||||
|
@@ -19,7 +19,7 @@ typedef struct URLFormat {
|
||||
int channels;
|
||||
int height;
|
||||
int width;
|
||||
int pix_fmt;
|
||||
enum PixelFormat pix_fmt;
|
||||
} URLFormat;
|
||||
|
||||
typedef struct URLContext URLContext;
|
||||
|
@@ -101,7 +101,7 @@ offset_t url_fseek(ByteIOContext *s, offset_t offset, int whence)
|
||||
|
||||
if (s->write_flag) {
|
||||
if (whence == SEEK_CUR) {
|
||||
offset1 = s->pos + s->buf_ptr - s->buffer;
|
||||
offset1 = s->pos + (s->buf_ptr - s->buffer);
|
||||
if (offset == 0)
|
||||
return offset1;
|
||||
offset += offset1;
|
||||
|
10
libav/ffm.c
10
libav/ffm.c
@@ -159,6 +159,11 @@ static int ffm_write_header(AVFormatContext *s)
|
||||
put_be32(pb, (codec->frame_rate * 1000) / FRAME_RATE_BASE);
|
||||
put_be16(pb, codec->width);
|
||||
put_be16(pb, codec->height);
|
||||
put_byte(pb, codec->qmin);
|
||||
put_byte(pb, codec->qmax);
|
||||
put_byte(pb, codec->max_qdiff);
|
||||
put_be16(pb, (int) (codec->qcompress * 10000.0));
|
||||
put_be16(pb, (int) (codec->qblur * 10000.0));
|
||||
break;
|
||||
case CODEC_TYPE_AUDIO:
|
||||
put_be32(pb, codec->sample_rate);
|
||||
@@ -367,6 +372,11 @@ static int ffm_read_header(AVFormatContext *s, AVFormatParameters *ap)
|
||||
codec->frame_rate = ((INT64)get_be32(pb) * FRAME_RATE_BASE) / 1000;
|
||||
codec->width = get_be16(pb);
|
||||
codec->height = get_be16(pb);
|
||||
codec->qmin = get_byte(pb);
|
||||
codec->qmax = get_byte(pb);
|
||||
codec->max_qdiff = get_byte(pb);
|
||||
codec->qcompress = get_be16(pb) / 10000.0;
|
||||
codec->qblur = get_be16(pb) / 10000.0;
|
||||
break;
|
||||
case CODEC_TYPE_AUDIO:
|
||||
codec->sample_rate = get_be32(pb);
|
||||
|
93
libav/grab.c
93
libav/grab.c
@@ -53,6 +53,7 @@ static int grab_read_header(AVFormatContext *s1, AVFormatParameters *ap)
|
||||
int width, height;
|
||||
int video_fd, frame_size;
|
||||
int ret, frame_rate;
|
||||
int desired_palette;
|
||||
|
||||
if (!ap || ap->width <= 0 || ap->height <= 0 || ap->frame_rate <= 0)
|
||||
return -1;
|
||||
@@ -93,6 +94,15 @@ static int grab_read_header(AVFormatContext *s1, AVFormatParameters *ap)
|
||||
goto fail;
|
||||
}
|
||||
|
||||
desired_palette = -1;
|
||||
if (st->codec.pix_fmt == PIX_FMT_YUV420P) {
|
||||
desired_palette = VIDEO_PALETTE_YUV420P;
|
||||
} else if (st->codec.pix_fmt == PIX_FMT_YUV422) {
|
||||
desired_palette = VIDEO_PALETTE_YUV422;
|
||||
} else if (st->codec.pix_fmt == PIX_FMT_BGR24) {
|
||||
desired_palette = VIDEO_PALETTE_RGB24;
|
||||
}
|
||||
|
||||
/* unmute audio */
|
||||
ioctl(video_fd, VIDIOCGAUDIO, &audio);
|
||||
memcpy(&audio_saved, &audio, sizeof(audio));
|
||||
@@ -125,16 +135,19 @@ static int grab_read_header(AVFormatContext *s1, AVFormatParameters *ap)
|
||||
pict.whiteness);
|
||||
#endif
|
||||
/* try to choose a suitable video format */
|
||||
pict.palette=VIDEO_PALETTE_YUV420P;
|
||||
ret = ioctl(video_fd, VIDIOCSPICT, &pict);
|
||||
if (ret < 0) {
|
||||
pict.palette=VIDEO_PALETTE_YUV422;
|
||||
pict.palette = desired_palette;
|
||||
if (desired_palette == -1 || (ret = ioctl(video_fd, VIDIOCSPICT, &pict)) < 0) {
|
||||
pict.palette=VIDEO_PALETTE_YUV420P;
|
||||
ret = ioctl(video_fd, VIDIOCSPICT, &pict);
|
||||
if (ret < 0) {
|
||||
pict.palette=VIDEO_PALETTE_RGB24;
|
||||
pict.palette=VIDEO_PALETTE_YUV422;
|
||||
ret = ioctl(video_fd, VIDIOCSPICT, &pict);
|
||||
if (ret < 0)
|
||||
goto fail1;
|
||||
if (ret < 0) {
|
||||
pict.palette=VIDEO_PALETTE_RGB24;
|
||||
ret = ioctl(video_fd, VIDIOCSPICT, &pict);
|
||||
if (ret < 0)
|
||||
goto fail1;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -155,22 +168,26 @@ static int grab_read_header(AVFormatContext *s1, AVFormatParameters *ap)
|
||||
s->time_frame = gettime();
|
||||
|
||||
/* start to grab the first frame */
|
||||
gb_buf.frame = (gb_frame + 1) % gb_buffers.frames;
|
||||
gb_buf.frame = gb_frame % gb_buffers.frames;
|
||||
gb_buf.height = height;
|
||||
gb_buf.width = width;
|
||||
gb_buf.format = VIDEO_PALETTE_YUV420P;
|
||||
gb_buf.format = desired_palette;
|
||||
|
||||
ret = ioctl(video_fd, VIDIOCMCAPTURE, &gb_buf);
|
||||
if (ret < 0 && errno != EAGAIN) {
|
||||
/* try YUV422 */
|
||||
gb_buf.format = VIDEO_PALETTE_YUV422;
|
||||
if (desired_palette == -1 || (ret = ioctl(video_fd, VIDIOCMCAPTURE, &gb_buf)) < 0) {
|
||||
gb_buf.format = VIDEO_PALETTE_YUV420P;
|
||||
|
||||
ret = ioctl(video_fd, VIDIOCMCAPTURE, &gb_buf);
|
||||
if (ret < 0 && errno != EAGAIN) {
|
||||
/* try RGB24 */
|
||||
gb_buf.format = VIDEO_PALETTE_RGB24;
|
||||
ret = ioctl(video_fd, VIDIOCMCAPTURE, &gb_buf);
|
||||
}
|
||||
if (ret < 0 && errno != EAGAIN) {
|
||||
/* try YUV422 */
|
||||
gb_buf.format = VIDEO_PALETTE_YUV422;
|
||||
|
||||
ret = ioctl(video_fd, VIDIOCMCAPTURE, &gb_buf);
|
||||
if (ret < 0 && errno != EAGAIN) {
|
||||
/* try RGB24 */
|
||||
gb_buf.format = VIDEO_PALETTE_RGB24;
|
||||
ret = ioctl(video_fd, VIDIOCMCAPTURE, &gb_buf);
|
||||
}
|
||||
}
|
||||
}
|
||||
if (ret < 0) {
|
||||
if (errno != EAGAIN) {
|
||||
@@ -221,8 +238,11 @@ static int grab_read_header(AVFormatContext *s1, AVFormatParameters *ap)
|
||||
static int v4l_mm_read_picture(VideoData *s, UINT8 *buf)
|
||||
{
|
||||
UINT8 *ptr;
|
||||
struct timeval tv_s, tv_e;
|
||||
int delay;
|
||||
|
||||
gb_buf.frame = gb_frame;
|
||||
/* Setup to capture the next frame */
|
||||
gb_buf.frame = (gb_frame + 1) % gb_buffers.frames;
|
||||
if (ioctl(s->fd, VIDIOCMCAPTURE, &gb_buf) < 0) {
|
||||
if (errno == EAGAIN)
|
||||
fprintf(stderr,"Cannot Sync\n");
|
||||
@@ -230,13 +250,26 @@ static int v4l_mm_read_picture(VideoData *s, UINT8 *buf)
|
||||
perror("VIDIOCMCAPTURE");
|
||||
return -EIO;
|
||||
}
|
||||
gb_frame = (gb_frame + 1) % gb_buffers.frames;
|
||||
|
||||
gettimeofday(&tv_s, 0);
|
||||
|
||||
while (ioctl(s->fd, VIDIOCSYNC, &gb_frame) < 0 &&
|
||||
(errno == EAGAIN || errno == EINTR));
|
||||
|
||||
/*
|
||||
gettimeofday(&tv_e, 0);
|
||||
|
||||
delay = (tv_e.tv_sec - tv_s.tv_sec) * 1000000 + tv_e.tv_usec - tv_s.tv_usec;
|
||||
if (delay > 10000)
|
||||
printf("VIDIOCSYNC took %d us\n", delay);
|
||||
*/
|
||||
|
||||
ptr = video_buf + gb_buffers.offsets[gb_frame];
|
||||
memcpy(buf, ptr, s->frame_size);
|
||||
|
||||
/* This is now the grabbing frame */
|
||||
gb_frame = gb_buf.frame;
|
||||
|
||||
return s->frame_size;
|
||||
}
|
||||
|
||||
@@ -245,14 +278,25 @@ static int grab_read_packet(AVFormatContext *s1, AVPacket *pkt)
|
||||
VideoData *s = s1->priv_data;
|
||||
INT64 curtime, delay;
|
||||
struct timespec ts;
|
||||
int first;
|
||||
INT64 per_frame = (INT64_C(1000000) * FRAME_RATE_BASE) / s->frame_rate;
|
||||
int dropped = 0;
|
||||
|
||||
/* Calculate the time of the next frame */
|
||||
s->time_frame += per_frame;
|
||||
|
||||
/* wait based on the frame rate */
|
||||
s->time_frame += (INT64_C(1000000) * FRAME_RATE_BASE) / s->frame_rate;
|
||||
for(;;) {
|
||||
for(first = 1;; first = 0) {
|
||||
curtime = gettime();
|
||||
delay = s->time_frame - curtime;
|
||||
if (delay <= 0)
|
||||
if (delay <= 0) {
|
||||
if (delay < -per_frame) {
|
||||
/* printf("grabbing is %d frames late (dropping)\n", (int) -(delay / 16666)); */
|
||||
dropped = 1;
|
||||
s->time_frame += per_frame;
|
||||
}
|
||||
break;
|
||||
}
|
||||
ts.tv_sec = delay / 1000000;
|
||||
ts.tv_nsec = (delay % 1000000) * 1000;
|
||||
nanosleep(&ts, NULL);
|
||||
@@ -261,6 +305,9 @@ static int grab_read_packet(AVFormatContext *s1, AVPacket *pkt)
|
||||
if (av_new_packet(pkt, s->frame_size) < 0)
|
||||
return -EIO;
|
||||
|
||||
if (dropped)
|
||||
pkt->flags |= PKT_FLAG_DROPPED_FRAME;
|
||||
|
||||
/* read one frame */
|
||||
if (s->use_mmap) {
|
||||
return v4l_mm_read_picture(s, pkt->data);
|
||||
|
Reference in New Issue
Block a user