27
27
#include "avcodec.h"
29
typedef struct RawVideoContext {
30
unsigned char * buffer; /* block of memory for holding one frame */
31
int length; /* number of bytes in buffer */
32
AVFrame pic; ///< AVCodecContext.coded_frame
35
typedef struct PixelFormatTag {
40
static const PixelFormatTag pixelFormatTags[] = {
30
const PixelFormatTag ff_raw_pixelFormatTags[] = {
41
31
{ PIX_FMT_YUV420P, MKTAG('I', '4', '2', '0') }, /* Planar formats */
42
32
{ PIX_FMT_YUV420P, MKTAG('I', 'Y', 'U', 'V') },
43
33
{ PIX_FMT_YUV420P, MKTAG('Y', 'V', '1', '2') },
51
41
{ PIX_FMT_YUYV422, MKTAG('Y', 'U', 'Y', '2') }, /* Packed formats */
52
42
{ PIX_FMT_YUYV422, MKTAG('Y', '4', '2', '2') },
53
43
{ PIX_FMT_UYVY422, MKTAG('U', 'Y', 'V', 'Y') },
44
{ PIX_FMT_UYVY422, MKTAG('H', 'D', 'Y', 'C') },
54
45
{ PIX_FMT_GRAY8, MKTAG('G', 'R', 'E', 'Y') },
55
46
{ PIX_FMT_RGB555, MKTAG('R', 'G', 'B', 15) },
56
47
{ PIX_FMT_BGR555, MKTAG('B', 'G', 'R', 15) },
61
52
{ PIX_FMT_UYVY422, MKTAG('2', 'v', 'u', 'y') },
66
static const PixelFormatTag pixelFormatBpsAVI[] = {
68
{ PIX_FMT_RGB555, 15 },
69
{ PIX_FMT_RGB555, 16 },
70
{ PIX_FMT_BGR24, 24 },
71
{ PIX_FMT_RGB32, 32 },
75
static const PixelFormatTag pixelFormatBpsMOV[] = {
76
/* FIXME fix swscaler to support those */
77
/* http://developer.apple.com/documentation/QuickTime/QTFF/QTFFChap3/chapter_4_section_2.html */
79
{ PIX_FMT_BGR555, 16 },
80
{ PIX_FMT_RGB24, 24 },
81
{ PIX_FMT_BGR32_1, 32 },
85
static int findPixelFormat(const PixelFormatTag *tags, unsigned int fourcc)
87
while (tags->pix_fmt >= 0) {
88
if (tags->fourcc == fourcc)
92
return PIX_FMT_YUV420P;
53
{ PIX_FMT_UYVY422, MKTAG('A', 'V', 'U', 'I') }, /* FIXME merge both fields */
95
58
unsigned int avcodec_pix_fmt_to_codec_tag(enum PixelFormat fmt)
97
const PixelFormatTag * tags = pixelFormatTags;
60
const PixelFormatTag * tags = ff_raw_pixelFormatTags;
98
61
while (tags->pix_fmt >= 0) {
99
62
if (tags->pix_fmt == fmt)
100
63
return tags->fourcc;
106
/* RAW Decoder Implementation */
108
static int raw_init_decoder(AVCodecContext *avctx)
110
RawVideoContext *context = avctx->priv_data;
112
if (avctx->codec_tag == MKTAG('r','a','w',' '))
113
avctx->pix_fmt = findPixelFormat(pixelFormatBpsMOV, avctx->bits_per_sample);
114
else if (avctx->codec_tag)
115
avctx->pix_fmt = findPixelFormat(pixelFormatTags, avctx->codec_tag);
116
else if (avctx->bits_per_sample)
117
avctx->pix_fmt = findPixelFormat(pixelFormatBpsAVI, avctx->bits_per_sample);
119
context->length = avpicture_get_size(avctx->pix_fmt, avctx->width, avctx->height);
120
context->buffer = av_malloc(context->length);
121
context->pic.pict_type = FF_I_TYPE;
122
context->pic.key_frame = 1;
124
avctx->coded_frame= &context->pic;
126
if (!context->buffer)
132
static void flip(AVCodecContext *avctx, AVPicture * picture){
133
if(!avctx->codec_tag && avctx->bits_per_sample && picture->linesize[2]==0){
134
picture->data[0] += picture->linesize[0] * (avctx->height-1);
135
picture->linesize[0] *= -1;
139
static int raw_decode(AVCodecContext *avctx,
140
void *data, int *data_size,
141
uint8_t *buf, int buf_size)
143
RawVideoContext *context = avctx->priv_data;
145
AVFrame * frame = (AVFrame *) data;
146
AVPicture * picture = (AVPicture *) data;
148
frame->interlaced_frame = avctx->coded_frame->interlaced_frame;
149
frame->top_field_first = avctx->coded_frame->top_field_first;
151
if(buf_size < context->length - (avctx->pix_fmt==PIX_FMT_PAL8 ? 256*4 : 0))
154
avpicture_fill(picture, buf, avctx->pix_fmt, avctx->width, avctx->height);
155
if(avctx->pix_fmt==PIX_FMT_PAL8 && buf_size < context->length){
156
frame->data[1]= context->buffer;
158
if (avctx->palctrl && avctx->palctrl->palette_changed) {
159
memcpy(frame->data[1], avctx->palctrl->palette, AVPALETTE_SIZE);
160
avctx->palctrl->palette_changed = 0;
163
flip(avctx, picture);
165
if (avctx->codec_tag == MKTAG('Y', 'V', '1', '2'))
168
unsigned char *tmp = picture->data[1];
169
picture->data[1] = picture->data[2];
170
picture->data[2] = tmp;
173
*data_size = sizeof(AVPicture);
177
static int raw_close_decoder(AVCodecContext *avctx)
179
RawVideoContext *context = avctx->priv_data;
181
av_freep(&context->buffer);
185
/* RAW Encoder Implementation */
186
#ifdef CONFIG_RAWVIDEO_ENCODER
187
static int raw_init_encoder(AVCodecContext *avctx)
189
avctx->coded_frame = (AVFrame *)avctx->priv_data;
190
avctx->coded_frame->pict_type = FF_I_TYPE;
191
avctx->coded_frame->key_frame = 1;
192
if(!avctx->codec_tag)
193
avctx->codec_tag = avcodec_pix_fmt_to_codec_tag(avctx->pix_fmt);
197
static int raw_encode(AVCodecContext *avctx,
198
unsigned char *frame, int buf_size, void *data)
200
return avpicture_layout((AVPicture *)data, avctx->pix_fmt, avctx->width,
201
avctx->height, frame, buf_size);
204
AVCodec rawvideo_encoder = {
212
#endif // CONFIG_RAWVIDEO_ENCODER
214
AVCodec rawvideo_decoder = {
218
sizeof(RawVideoContext),