3 * Copyright (c) 2001 Fabrice Bellard
5 * This file is part of FFmpeg.
7 * FFmpeg is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * FFmpeg is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
28 #include "imgconvert.h"
30 #include "libavutil/intreadwrite.h"
31 #include "libavcore/imgutils.h"
32 #include "libavcore/internal.h"
34 typedef struct RawVideoContext
{
35 uint32_t palette
[AVPALETTE_COUNT
];
36 unsigned char * buffer
; /* block of memory for holding one frame */
37 int length
; /* number of bytes in buffer */
39 AVFrame pic
; ///< AVCodecContext.coded_frame
42 static const PixelFormatTag pix_fmt_bps_avi
[] = {
45 { PIX_FMT_RGB444
, 12 },
46 { PIX_FMT_RGB555
, 15 },
47 { PIX_FMT_RGB555
, 16 },
48 { PIX_FMT_BGR24
, 24 },
49 { PIX_FMT_RGB32
, 32 },
53 static const PixelFormatTag pix_fmt_bps_mov
[] = {
54 { PIX_FMT_MONOWHITE
, 1 },
58 // FIXME swscale does not support 16 bit in .mov, sample 16bit.mov
59 // http://developer.apple.com/documentation/QuickTime/QTFF/QTFFChap3/qtff3.html
60 { PIX_FMT_RGB555BE
, 16 },
61 { PIX_FMT_RGB24
, 24 },
66 static enum PixelFormat
find_pix_fmt(const PixelFormatTag
*tags
, unsigned int fourcc
)
68 while (tags
->pix_fmt
>= 0) {
69 if (tags
->fourcc
== fourcc
)
73 return PIX_FMT_YUV420P
;
76 static av_cold
int raw_init_decoder(AVCodecContext
*avctx
)
78 RawVideoContext
*context
= avctx
->priv_data
;
80 if (avctx
->codec_tag
== MKTAG('r','a','w',' '))
81 avctx
->pix_fmt
= find_pix_fmt(pix_fmt_bps_mov
, avctx
->bits_per_coded_sample
);
82 else if (avctx
->codec_tag
)
83 avctx
->pix_fmt
= find_pix_fmt(ff_raw_pix_fmt_tags
, avctx
->codec_tag
);
84 else if (avctx
->pix_fmt
== PIX_FMT_NONE
&& avctx
->bits_per_coded_sample
)
85 avctx
->pix_fmt
= find_pix_fmt(pix_fmt_bps_avi
, avctx
->bits_per_coded_sample
);
87 ff_set_systematic_pal2(context
->palette
, avctx
->pix_fmt
);
88 context
->length
= avpicture_get_size(avctx
->pix_fmt
, avctx
->width
, avctx
->height
);
89 if((avctx
->bits_per_coded_sample
== 4 || avctx
->bits_per_coded_sample
== 2) &&
90 avctx
->pix_fmt
==PIX_FMT_PAL8
&&
91 (!avctx
->codec_tag
|| avctx
->codec_tag
== MKTAG('r','a','w',' '))){
92 context
->buffer
= av_malloc(context
->length
);
96 context
->pic
.pict_type
= FF_I_TYPE
;
97 context
->pic
.key_frame
= 1;
99 avctx
->coded_frame
= &context
->pic
;
101 if((avctx
->extradata_size
>= 9 && !memcmp(avctx
->extradata
+ avctx
->extradata_size
- 9, "BottomUp", 9)) ||
102 avctx
->codec_tag
== MKTAG( 3 , 0 , 0 , 0 ))
108 static void flip(AVCodecContext
*avctx
, AVPicture
* picture
){
109 picture
->data
[0] += picture
->linesize
[0] * (avctx
->height
-1);
110 picture
->linesize
[0] *= -1;
113 static int raw_decode(AVCodecContext
*avctx
,
114 void *data
, int *data_size
,
117 const uint8_t *buf
= avpkt
->data
;
118 int buf_size
= avpkt
->size
;
119 RawVideoContext
*context
= avctx
->priv_data
;
121 AVFrame
* frame
= (AVFrame
*) data
;
122 AVPicture
* picture
= (AVPicture
*) data
;
124 frame
->interlaced_frame
= avctx
->coded_frame
->interlaced_frame
;
125 frame
->top_field_first
= avctx
->coded_frame
->top_field_first
;
126 frame
->reordered_opaque
= avctx
->reordered_opaque
;
128 //2bpp and 4bpp raw in avi and mov (yes this is ugly ...)
129 if (context
->buffer
) {
131 uint8_t *dst
= context
->buffer
;
132 buf_size
= context
->length
- 256*4;
133 if (avctx
->bits_per_coded_sample
== 4){
134 for(i
=0; 2*i
+1 < buf_size
; i
++){
135 dst
[2*i
+0]= buf
[i
]>>4;
136 dst
[2*i
+1]= buf
[i
]&15;
139 for(i
=0; 4*i
+3 < buf_size
; i
++){
140 dst
[4*i
+0]= buf
[i
]>>6;
141 dst
[4*i
+1]= buf
[i
]>>4&3;
142 dst
[4*i
+2]= buf
[i
]>>2&3;
143 dst
[4*i
+3]= buf
[i
] &3;
148 if(avctx
->codec_tag
== MKTAG('A', 'V', '1', 'x') ||
149 avctx
->codec_tag
== MKTAG('A', 'V', 'u', 'p'))
150 buf
+= buf_size
- context
->length
;
152 if(buf_size
< context
->length
- (avctx
->pix_fmt
==PIX_FMT_PAL8 ?
256*4 : 0))
155 avpicture_fill(picture
, buf
, avctx
->pix_fmt
, avctx
->width
, avctx
->height
);
156 if((avctx
->pix_fmt
==PIX_FMT_PAL8
&& buf_size
< context
->length
) ||
157 (avctx
->pix_fmt
!=PIX_FMT_PAL8
&&
158 (av_pix_fmt_descriptors
[avctx
->pix_fmt
].flags
& PIX_FMT_PAL
))){
159 frame
->data
[1]= context
->palette
;
161 if (avctx
->palctrl
&& avctx
->palctrl
->palette_changed
) {
162 memcpy(frame
->data
[1], avctx
->palctrl
->palette
, AVPALETTE_SIZE
);
163 avctx
->palctrl
->palette_changed
= 0;
165 if(avctx
->pix_fmt
==PIX_FMT_BGR24
&& ((frame
->linesize
[0]+3)&~3)*avctx
->height
<= buf_size
)
166 frame
->linesize
[0] = (frame
->linesize
[0]+3)&~3;
169 flip(avctx
, picture
);
171 if ( avctx
->codec_tag
== MKTAG('Y', 'V', '1', '2')
172 || avctx
->codec_tag
== MKTAG('Y', 'V', 'U', '9'))
173 FFSWAP(uint8_t *, picture
->data
[1], picture
->data
[2]);
175 if(avctx
->codec_tag
== AV_RL32("yuv2") &&
176 avctx
->pix_fmt
== PIX_FMT_YUYV422
) {
178 uint8_t *line
= picture
->data
[0];
179 for(y
= 0; y
< avctx
->height
; y
++) {
180 for(x
= 0; x
< avctx
->width
; x
++)
181 line
[2*x
+ 1] ^= 0x80;
182 line
+= picture
->linesize
[0];
186 *data_size
= sizeof(AVPicture
);
190 static av_cold
int raw_close_decoder(AVCodecContext
*avctx
)
192 RawVideoContext
*context
= avctx
->priv_data
;
194 av_freep(&context
->buffer
);
198 AVCodec ff_rawvideo_decoder
= {
202 sizeof(RawVideoContext
),
207 .long_name
= NULL_IF_CONFIG_SMALL("raw video"),