3 * Copyright (c) 2001 Fabrice Bellard
5 * This file is part of Libav.
7 * Libav 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 * Libav 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 Libav; 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 "libavutil/imgutils.h"
33 typedef struct RawVideoContext
{
34 uint32_t palette
[AVPALETTE_COUNT
];
35 unsigned char * buffer
; /* block of memory for holding one frame */
36 int length
; /* number of bytes in buffer */
38 AVFrame pic
; ///< AVCodecContext.coded_frame
41 static const PixelFormatTag pix_fmt_bps_avi
[] = {
44 { PIX_FMT_RGB444
, 12 },
45 { PIX_FMT_RGB555
, 15 },
46 { PIX_FMT_RGB555
, 16 },
47 { PIX_FMT_BGR24
, 24 },
48 { PIX_FMT_RGB32
, 32 },
52 static const PixelFormatTag pix_fmt_bps_mov
[] = {
53 { PIX_FMT_MONOWHITE
, 1 },
57 // FIXME swscale does not support 16 bit in .mov, sample 16bit.mov
58 // http://developer.apple.com/documentation/QuickTime/QTFF/QTFFChap3/qtff3.html
59 { PIX_FMT_RGB555BE
, 16 },
60 { PIX_FMT_RGB24
, 24 },
62 { PIX_FMT_MONOWHITE
,33 },
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
== MKTAG('W','R','A','W'))
83 avctx
->pix_fmt
= find_pix_fmt(pix_fmt_bps_avi
, avctx
->bits_per_coded_sample
);
84 else if (avctx
->codec_tag
)
85 avctx
->pix_fmt
= find_pix_fmt(ff_raw_pix_fmt_tags
, avctx
->codec_tag
);
86 else if (avctx
->pix_fmt
== PIX_FMT_NONE
&& avctx
->bits_per_coded_sample
)
87 avctx
->pix_fmt
= find_pix_fmt(pix_fmt_bps_avi
, avctx
->bits_per_coded_sample
);
89 ff_set_systematic_pal2(context
->palette
, avctx
->pix_fmt
);
90 context
->length
= avpicture_get_size(avctx
->pix_fmt
, avctx
->width
, avctx
->height
);
91 if((avctx
->bits_per_coded_sample
== 4 || avctx
->bits_per_coded_sample
== 2) &&
92 avctx
->pix_fmt
==PIX_FMT_PAL8
&&
93 (!avctx
->codec_tag
|| avctx
->codec_tag
== MKTAG('r','a','w',' '))){
94 context
->buffer
= av_malloc(context
->length
);
98 context
->pic
.pict_type
= AV_PICTURE_TYPE_I
;
99 context
->pic
.key_frame
= 1;
101 avctx
->coded_frame
= &context
->pic
;
103 if((avctx
->extradata_size
>= 9 && !memcmp(avctx
->extradata
+ avctx
->extradata_size
- 9, "BottomUp", 9)) ||
104 avctx
->codec_tag
== MKTAG(3, 0, 0, 0) || avctx
->codec_tag
== MKTAG('W','R','A','W'))
110 static void flip(AVCodecContext
*avctx
, AVPicture
* picture
){
111 picture
->data
[0] += picture
->linesize
[0] * (avctx
->height
-1);
112 picture
->linesize
[0] *= -1;
115 static int raw_decode(AVCodecContext
*avctx
,
116 void *data
, int *data_size
,
119 const uint8_t *buf
= avpkt
->data
;
120 int buf_size
= avpkt
->size
;
121 RawVideoContext
*context
= avctx
->priv_data
;
124 AVFrame
*frame
= data
;
125 AVPicture
*picture
= data
;
127 frame
->pict_type
= avctx
->coded_frame
->pict_type
;
128 frame
->interlaced_frame
= avctx
->coded_frame
->interlaced_frame
;
129 frame
->top_field_first
= avctx
->coded_frame
->top_field_first
;
130 frame
->reordered_opaque
= avctx
->reordered_opaque
;
131 frame
->pkt_pts
= avctx
->pkt
->pts
;
133 if(buf_size
< context
->length
- (avctx
->pix_fmt
==PIX_FMT_PAL8 ?
256*4 : 0))
136 //2bpp and 4bpp raw in avi and mov (yes this is ugly ...)
137 if (context
->buffer
) {
139 uint8_t *dst
= context
->buffer
;
140 buf_size
= context
->length
- 256*4;
141 if (avctx
->bits_per_coded_sample
== 4){
142 for(i
=0; 2*i
+1 < buf_size
; i
++){
143 dst
[2*i
+0]= buf
[i
]>>4;
144 dst
[2*i
+1]= buf
[i
]&15;
147 for(i
=0; 4*i
+3 < buf_size
; i
++){
148 dst
[4*i
+0]= buf
[i
]>>6;
149 dst
[4*i
+1]= buf
[i
]>>4&3;
150 dst
[4*i
+2]= buf
[i
]>>2&3;
151 dst
[4*i
+3]= buf
[i
] &3;
156 if(avctx
->codec_tag
== MKTAG('A', 'V', '1', 'x') ||
157 avctx
->codec_tag
== MKTAG('A', 'V', 'u', 'p'))
158 buf
+= buf_size
- context
->length
;
160 if ((res
= avpicture_fill(picture
, buf
, avctx
->pix_fmt
,
161 avctx
->width
, avctx
->height
)) < 0)
163 if((avctx
->pix_fmt
==PIX_FMT_PAL8
&& buf_size
< context
->length
) ||
164 (av_pix_fmt_descriptors
[avctx
->pix_fmt
].flags
& PIX_FMT_PSEUDOPAL
)) {
165 frame
->data
[1]= context
->palette
;
167 if (avctx
->pix_fmt
== PIX_FMT_PAL8
) {
168 const uint8_t *pal
= av_packet_get_side_data(avpkt
, AV_PKT_DATA_PALETTE
, NULL
);
171 memcpy(frame
->data
[1], pal
, AVPALETTE_SIZE
);
172 frame
->palette_has_changed
= 1;
175 if(avctx
->pix_fmt
==PIX_FMT_BGR24
&& ((frame
->linesize
[0]+3)&~3)*avctx
->height
<= buf_size
)
176 frame
->linesize
[0] = (frame
->linesize
[0]+3)&~3;
179 flip(avctx
, picture
);
181 if ( avctx
->codec_tag
== MKTAG('Y', 'V', '1', '2')
182 || avctx
->codec_tag
== MKTAG('Y', 'V', '1', '6')
183 || avctx
->codec_tag
== MKTAG('Y', 'V', '2', '4')
184 || avctx
->codec_tag
== MKTAG('Y', 'V', 'U', '9'))
185 FFSWAP(uint8_t *, picture
->data
[1], picture
->data
[2]);
187 if(avctx
->codec_tag
== AV_RL32("yuv2") &&
188 avctx
->pix_fmt
== PIX_FMT_YUYV422
) {
190 uint8_t *line
= picture
->data
[0];
191 for(y
= 0; y
< avctx
->height
; y
++) {
192 for(x
= 0; x
< avctx
->width
; x
++)
193 line
[2*x
+ 1] ^= 0x80;
194 line
+= picture
->linesize
[0];
198 *data_size
= sizeof(AVPicture
);
202 static av_cold
int raw_close_decoder(AVCodecContext
*avctx
)
204 RawVideoContext
*context
= avctx
->priv_data
;
206 av_freep(&context
->buffer
);
210 AVCodec ff_rawvideo_decoder
= {
212 .type
= AVMEDIA_TYPE_VIDEO
,
213 .id
= CODEC_ID_RAWVIDEO
,
214 .priv_data_size
= sizeof(RawVideoContext
),
215 .init
= raw_init_decoder
,
216 .close
= raw_close_decoder
,
217 .decode
= raw_decode
,
218 .long_name
= NULL_IF_CONFIG_SMALL("raw video"),