Fix raw rgb/bgr vertical flip in avi based on info from http://www.fourcc.org/fccbihgt.php.

partially fixes issue862.

Originally committed as revision 17475 to svn://svn.ffmpeg.org/ffmpeg/trunk
This commit is contained in:
Michael Niedermayer 2009-02-21 15:32:56 +00:00
parent 431ac29046
commit 31f2616db8
3 changed files with 17 additions and 4 deletions

View File

@ -30,6 +30,7 @@
typedef struct RawVideoContext { typedef struct RawVideoContext {
unsigned char * buffer; /* block of memory for holding one frame */ unsigned char * buffer; /* block of memory for holding one frame */
int length; /* number of bytes in buffer */ int length; /* number of bytes in buffer */
int flip;
AVFrame pic; ///< AVCodecContext.coded_frame AVFrame pic; ///< AVCodecContext.coded_frame
} RawVideoContext; } RawVideoContext;
@ -85,15 +86,16 @@ static av_cold int raw_init_decoder(AVCodecContext *avctx)
if (!context->buffer) if (!context->buffer)
return -1; return -1;
if(avctx->extradata_size >= 9 && !memcmp(avctx->extradata + avctx->extradata_size - 9, "BottomUp", 9))
context->flip=1;
return 0; return 0;
} }
static void flip(AVCodecContext *avctx, AVPicture * picture){ static void flip(AVCodecContext *avctx, AVPicture * picture){
if(!avctx->codec_tag && avctx->bits_per_coded_sample && picture->linesize[2]==0){
picture->data[0] += picture->linesize[0] * (avctx->height-1); picture->data[0] += picture->linesize[0] * (avctx->height-1);
picture->linesize[0] *= -1; picture->linesize[0] *= -1;
} }
}
static int raw_decode(AVCodecContext *avctx, static int raw_decode(AVCodecContext *avctx,
void *data, int *data_size, void *data, int *data_size,
@ -131,6 +133,7 @@ static int raw_decode(AVCodecContext *avctx,
avctx->palctrl->palette_changed = 0; avctx->palctrl->palette_changed = 0;
} }
if(context->flip)
flip(avctx, picture); flip(avctx, picture);
if (avctx->codec_tag == MKTAG('Y', 'V', '1', '2')) if (avctx->codec_tag == MKTAG('Y', 'V', '1', '2'))

View File

@ -451,7 +451,7 @@ static int avi_read_header(AVFormatContext *s, AVFormatParameters *ap)
} }
get_le32(pb); /* size */ get_le32(pb); /* size */
st->codec->width = get_le32(pb); st->codec->width = get_le32(pb);
st->codec->height = get_le32(pb); st->codec->height = (int32_t)get_le32(pb);
get_le16(pb); /* panes */ get_le16(pb); /* panes */
st->codec->bits_per_coded_sample= get_le16(pb); /* depth */ st->codec->bits_per_coded_sample= get_le16(pb); /* depth */
tag1 = get_le32(pb); tag1 = get_le32(pb);
@ -499,6 +499,15 @@ static int avi_read_header(AVFormatContext *s, AVFormatParameters *ap)
st->codec->codec_tag = tag1; st->codec->codec_tag = tag1;
st->codec->codec_id = codec_get_id(codec_bmp_tags, tag1); st->codec->codec_id = codec_get_id(codec_bmp_tags, tag1);
st->need_parsing = AVSTREAM_PARSE_HEADERS; // This is needed to get the pict type which is necessary for generating correct pts. st->need_parsing = AVSTREAM_PARSE_HEADERS; // This is needed to get the pict type which is necessary for generating correct pts.
if(st->codec->codec_tag==0 && st->codec->height > 0 && st->codec->extradata_size < 1U<<30){
st->codec->extradata_size+= 9;
st->codec->extradata= av_realloc(st->codec->extradata, st->codec->extradata_size + FF_INPUT_BUFFER_PADDING_SIZE);
if(st->codec->extradata)
memcpy(st->codec->extradata + st->codec->extradata_size - 9, "BottomUp", 9);
}
st->codec->height= FFABS(st->codec->height);
// url_fskip(pb, size - 5 * 4); // url_fskip(pb, size - 5 * 4);
break; break;
case CODEC_TYPE_AUDIO: case CODEC_TYPE_AUDIO:

View File

@ -368,7 +368,8 @@ void put_bmp_header(ByteIOContext *pb, AVCodecContext *enc, const AVCodecTag *ta
{ {
put_le32(pb, 40 + enc->extradata_size); /* size */ put_le32(pb, 40 + enc->extradata_size); /* size */
put_le32(pb, enc->width); put_le32(pb, enc->width);
put_le32(pb, enc->height); //We always store RGB TopDown
put_le32(pb, enc->codec_tag ? enc->height : -enc->height);
put_le16(pb, 1); /* planes */ put_le16(pb, 1); /* planes */
put_le16(pb, enc->bits_per_coded_sample ? enc->bits_per_coded_sample : 24); /* depth */ put_le16(pb, enc->bits_per_coded_sample ? enc->bits_per_coded_sample : 24); /* depth */