Commit 0ae2f6e9 authored by reimar's avatar reimar

Copy nuv header only when the video codec is nuv


git-svn-id: file:///var/local/repositories/ffmpeg/trunk@10439 9553f0bf-9b14-0410-a0b8-cfaf0461ba5b
parent 809619c0
...@@ -24,6 +24,7 @@ ...@@ -24,6 +24,7 @@
typedef struct { typedef struct {
int v_id; int v_id;
int a_id; int a_id;
int rtjpg_video;
} NUVContext; } NUVContext;
typedef enum { typedef enum {
...@@ -173,6 +174,8 @@ static int nuv_header(AVFormatContext *s, AVFormatParameters *ap) { ...@@ -173,6 +174,8 @@ static int nuv_header(AVFormatContext *s, AVFormatParameters *ap) {
ctx->a_id = -1; ctx->a_id = -1;
get_codec_data(pb, vst, ast, is_mythtv); get_codec_data(pb, vst, ast, is_mythtv);
ctx->rtjpg_video = vst->codec->codec_id == CODEC_ID_NUV;
ctx->rtjpg_video |= vst->codec->codec_tag == MKTAG('R', 'J', 'P', 'G');
return 0; return 0;
} }
...@@ -185,27 +188,32 @@ static int nuv_packet(AVFormatContext *s, AVPacket *pkt) { ...@@ -185,27 +188,32 @@ static int nuv_packet(AVFormatContext *s, AVPacket *pkt) {
frametype_t frametype; frametype_t frametype;
int ret, size; int ret, size;
while (!url_feof(pb)) { while (!url_feof(pb)) {
int copyhdrsize = ctx->rtjpg_video ? HDRSIZE : 0;
ret = get_buffer(pb, hdr, HDRSIZE); ret = get_buffer(pb, hdr, HDRSIZE);
if (ret <= 0) if (ret <= 0)
return ret ? ret : -1; return ret ? ret : -1;
frametype = hdr[0]; frametype = hdr[0];
size = PKTSIZE(AV_RL32(&hdr[8])); size = PKTSIZE(AV_RL32(&hdr[8]));
switch (frametype) { switch (frametype) {
case NUV_VIDEO:
case NUV_EXTRADATA: case NUV_EXTRADATA:
if (!ctx->rtjpg_video) {
url_fskip(pb, size);
break;
}
case NUV_VIDEO:
if (ctx->v_id < 0) { if (ctx->v_id < 0) {
av_log(s, AV_LOG_ERROR, "Video packet in file without video stream!\n"); av_log(s, AV_LOG_ERROR, "Video packet in file without video stream!\n");
url_fskip(pb, size); url_fskip(pb, size);
break; break;
} }
ret = av_new_packet(pkt, HDRSIZE + size); ret = av_new_packet(pkt, copyhdrsize + size);
if (ret < 0) if (ret < 0)
return ret; return ret;
pkt->pos = url_ftell(pb); pkt->pos = url_ftell(pb) - copyhdrsize;
pkt->pts = AV_RL32(&hdr[4]); pkt->pts = AV_RL32(&hdr[4]);
pkt->stream_index = ctx->v_id; pkt->stream_index = ctx->v_id;
memcpy(pkt->data, hdr, HDRSIZE); memcpy(pkt->data, hdr, copyhdrsize);
ret = get_buffer(pb, pkt->data + HDRSIZE, size); ret = get_buffer(pb, pkt->data + copyhdrsize, size);
return ret; return ret;
case NUV_AUDIO: case NUV_AUDIO:
if (ctx->a_id < 0) { if (ctx->a_id < 0) {
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment