mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2025-08-06 15:49:50 +08:00
split rawvideo encoder and decoder in their own files
Originally committed as revision 9560 to svn://svn.ffmpeg.org/ffmpeg/trunk
This commit is contained in:
166
libavcodec/raw.c
166
libavcodec/raw.c
@ -25,19 +25,9 @@
|
||||
*/
|
||||
|
||||
#include "avcodec.h"
|
||||
#include "raw.h"
|
||||
|
||||
typedef struct RawVideoContext {
|
||||
unsigned char * buffer; /* block of memory for holding one frame */
|
||||
int length; /* number of bytes in buffer */
|
||||
AVFrame pic; ///< AVCodecContext.coded_frame
|
||||
} RawVideoContext;
|
||||
|
||||
typedef struct PixelFormatTag {
|
||||
int pix_fmt;
|
||||
unsigned int fourcc;
|
||||
} PixelFormatTag;
|
||||
|
||||
static const PixelFormatTag pixelFormatTags[] = {
|
||||
const PixelFormatTag ff_raw_pixelFormatTags[] = {
|
||||
{ PIX_FMT_YUV420P, MKTAG('I', '4', '2', '0') }, /* Planar formats */
|
||||
{ PIX_FMT_YUV420P, MKTAG('I', 'Y', 'U', 'V') },
|
||||
{ PIX_FMT_YUV420P, MKTAG('Y', 'V', '1', '2') },
|
||||
@ -64,38 +54,9 @@ static const PixelFormatTag pixelFormatTags[] = {
|
||||
{ -1, 0 },
|
||||
};
|
||||
|
||||
static const PixelFormatTag pixelFormatBpsAVI[] = {
|
||||
{ PIX_FMT_PAL8, 8 },
|
||||
{ PIX_FMT_RGB555, 15 },
|
||||
{ PIX_FMT_RGB555, 16 },
|
||||
{ PIX_FMT_BGR24, 24 },
|
||||
{ PIX_FMT_RGB32, 32 },
|
||||
{ -1, 0 },
|
||||
};
|
||||
|
||||
static const PixelFormatTag pixelFormatBpsMOV[] = {
|
||||
/* FIXME fix swscaler to support those */
|
||||
/* http://developer.apple.com/documentation/QuickTime/QTFF/QTFFChap3/chapter_4_section_2.html */
|
||||
{ PIX_FMT_PAL8, 8 },
|
||||
{ PIX_FMT_BGR555, 16 },
|
||||
{ PIX_FMT_RGB24, 24 },
|
||||
{ PIX_FMT_BGR32_1, 32 },
|
||||
{ -1, 0 },
|
||||
};
|
||||
|
||||
static int findPixelFormat(const PixelFormatTag *tags, unsigned int fourcc)
|
||||
{
|
||||
while (tags->pix_fmt >= 0) {
|
||||
if (tags->fourcc == fourcc)
|
||||
return tags->pix_fmt;
|
||||
tags++;
|
||||
}
|
||||
return PIX_FMT_YUV420P;
|
||||
}
|
||||
|
||||
unsigned int avcodec_pix_fmt_to_codec_tag(enum PixelFormat fmt)
|
||||
{
|
||||
const PixelFormatTag * tags = pixelFormatTags;
|
||||
const PixelFormatTag * tags = ff_raw_pixelFormatTags;
|
||||
while (tags->pix_fmt >= 0) {
|
||||
if (tags->pix_fmt == fmt)
|
||||
return tags->fourcc;
|
||||
@ -103,124 +64,3 @@ unsigned int avcodec_pix_fmt_to_codec_tag(enum PixelFormat fmt)
|
||||
}
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* RAW Decoder Implementation */
|
||||
|
||||
static int raw_init_decoder(AVCodecContext *avctx)
|
||||
{
|
||||
RawVideoContext *context = avctx->priv_data;
|
||||
|
||||
if (avctx->codec_tag == MKTAG('r','a','w',' '))
|
||||
avctx->pix_fmt = findPixelFormat(pixelFormatBpsMOV, avctx->bits_per_sample);
|
||||
else if (avctx->codec_tag)
|
||||
avctx->pix_fmt = findPixelFormat(pixelFormatTags, avctx->codec_tag);
|
||||
else if (avctx->bits_per_sample)
|
||||
avctx->pix_fmt = findPixelFormat(pixelFormatBpsAVI, avctx->bits_per_sample);
|
||||
|
||||
context->length = avpicture_get_size(avctx->pix_fmt, avctx->width, avctx->height);
|
||||
context->buffer = av_malloc(context->length);
|
||||
context->pic.pict_type = FF_I_TYPE;
|
||||
context->pic.key_frame = 1;
|
||||
|
||||
avctx->coded_frame= &context->pic;
|
||||
|
||||
if (!context->buffer)
|
||||
return -1;
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
static void flip(AVCodecContext *avctx, AVPicture * picture){
|
||||
if(!avctx->codec_tag && avctx->bits_per_sample && picture->linesize[2]==0){
|
||||
picture->data[0] += picture->linesize[0] * (avctx->height-1);
|
||||
picture->linesize[0] *= -1;
|
||||
}
|
||||
}
|
||||
|
||||
static int raw_decode(AVCodecContext *avctx,
|
||||
void *data, int *data_size,
|
||||
uint8_t *buf, int buf_size)
|
||||
{
|
||||
RawVideoContext *context = avctx->priv_data;
|
||||
|
||||
AVFrame * frame = (AVFrame *) data;
|
||||
AVPicture * picture = (AVPicture *) data;
|
||||
|
||||
frame->interlaced_frame = avctx->coded_frame->interlaced_frame;
|
||||
frame->top_field_first = avctx->coded_frame->top_field_first;
|
||||
|
||||
if(buf_size < context->length - (avctx->pix_fmt==PIX_FMT_PAL8 ? 256*4 : 0))
|
||||
return -1;
|
||||
|
||||
avpicture_fill(picture, buf, avctx->pix_fmt, avctx->width, avctx->height);
|
||||
if(avctx->pix_fmt==PIX_FMT_PAL8 && buf_size < context->length){
|
||||
frame->data[1]= context->buffer;
|
||||
}
|
||||
if (avctx->palctrl && avctx->palctrl->palette_changed) {
|
||||
memcpy(frame->data[1], avctx->palctrl->palette, AVPALETTE_SIZE);
|
||||
avctx->palctrl->palette_changed = 0;
|
||||
}
|
||||
|
||||
flip(avctx, picture);
|
||||
|
||||
if (avctx->codec_tag == MKTAG('Y', 'V', '1', '2'))
|
||||
{
|
||||
// swap fields
|
||||
unsigned char *tmp = picture->data[1];
|
||||
picture->data[1] = picture->data[2];
|
||||
picture->data[2] = tmp;
|
||||
}
|
||||
|
||||
*data_size = sizeof(AVPicture);
|
||||
return buf_size;
|
||||
}
|
||||
|
||||
static int raw_close_decoder(AVCodecContext *avctx)
|
||||
{
|
||||
RawVideoContext *context = avctx->priv_data;
|
||||
|
||||
av_freep(&context->buffer);
|
||||
return 0;
|
||||
}
|
||||
|
||||
/* RAW Encoder Implementation */
|
||||
#ifdef CONFIG_RAWVIDEO_ENCODER
|
||||
static int raw_init_encoder(AVCodecContext *avctx)
|
||||
{
|
||||
avctx->coded_frame = (AVFrame *)avctx->priv_data;
|
||||
avctx->coded_frame->pict_type = FF_I_TYPE;
|
||||
avctx->coded_frame->key_frame = 1;
|
||||
if(!avctx->codec_tag)
|
||||
avctx->codec_tag = avcodec_pix_fmt_to_codec_tag(avctx->pix_fmt);
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int raw_encode(AVCodecContext *avctx,
|
||||
unsigned char *frame, int buf_size, void *data)
|
||||
{
|
||||
return avpicture_layout((AVPicture *)data, avctx->pix_fmt, avctx->width,
|
||||
avctx->height, frame, buf_size);
|
||||
}
|
||||
|
||||
AVCodec rawvideo_encoder = {
|
||||
"rawvideo",
|
||||
CODEC_TYPE_VIDEO,
|
||||
CODEC_ID_RAWVIDEO,
|
||||
sizeof(AVFrame),
|
||||
raw_init_encoder,
|
||||
raw_encode,
|
||||
};
|
||||
#endif // CONFIG_RAWVIDEO_ENCODER
|
||||
|
||||
#ifdef CONFIG_RAWVIDEO_DECODER
|
||||
AVCodec rawvideo_decoder = {
|
||||
"rawvideo",
|
||||
CODEC_TYPE_VIDEO,
|
||||
CODEC_ID_RAWVIDEO,
|
||||
sizeof(RawVideoContext),
|
||||
raw_init_decoder,
|
||||
NULL,
|
||||
raw_close_decoder,
|
||||
raw_decode,
|
||||
};
|
||||
#endif // CONFIG_RAWVIDEO_DECODER
|
||||
|
Reference in New Issue
Block a user