2 * libquicktime yuv4 encoder
4 * Copyright (c) 2011 Carl Eugen Hoyos
6 * This file is part of FFmpeg.
8 * FFmpeg is free software; you can redistribute it and/or
9 * modify it under the terms of the GNU Lesser General Public
10 * License as published by the Free Software Foundation; either
11 * version 2.1 of the License, or (at your option) any later version.
13 * FFmpeg is distributed in the hope that it will be useful,
14 * but WITHOUT ANY WARRANTY; without even the implied warranty of
15 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
16 * Lesser General Public License for more details.
18 * You should have received a copy of the GNU Lesser General Public
19 * License along with FFmpeg; if not, write to the Free Software
20 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
26 static av_cold
int yuv4_encode_init(AVCodecContext
*avctx
)
28 avctx
->coded_frame
= av_frame_alloc();
30 if (!avctx
->coded_frame
) {
31 av_log(avctx
, AV_LOG_ERROR
, "Could not allocate frame.\n");
32 return AVERROR(ENOMEM
);
38 static int yuv4_encode_frame(AVCodecContext
*avctx
, AVPacket
*pkt
,
39 const AVFrame
*pic
, int *got_packet
)
45 if ((ret
= ff_alloc_packet2(avctx
, pkt
, 6 * (avctx
->width
+ 1 >> 1) * (avctx
->height
+ 1 >> 1))) < 0)
49 avctx
->coded_frame
->key_frame
= 1;
50 avctx
->coded_frame
->pict_type
= AV_PICTURE_TYPE_I
;
56 for (i
= 0; i
< avctx
->height
+ 1 >> 1; i
++) {
57 for (j
= 0; j
< avctx
->width
+ 1 >> 1; j
++) {
61 *dst
++ = y
[ 2 * j
+ 1];
62 *dst
++ = y
[pic
->linesize
[0] + 2 * j
];
63 *dst
++ = y
[pic
->linesize
[0] + 2 * j
+ 1];
65 y
+= 2 * pic
->linesize
[0];
66 u
+= pic
->linesize
[1];
67 v
+= pic
->linesize
[2];
70 pkt
->flags
|= AV_PKT_FLAG_KEY
;
75 static av_cold
int yuv4_encode_close(AVCodecContext
*avctx
)
77 av_freep(&avctx
->coded_frame
);
82 AVCodec ff_yuv4_encoder
= {
84 .long_name
= NULL_IF_CONFIG_SMALL("Uncompressed packed 4:2:0"),
85 .type
= AVMEDIA_TYPE_VIDEO
,
86 .id
= AV_CODEC_ID_YUV4
,
87 .init
= yuv4_encode_init
,
88 .encode2
= yuv4_encode_frame
,
89 .close
= yuv4_encode_close
,
90 .pix_fmts
= (const enum AVPixelFormat
[]){ AV_PIX_FMT_YUV420P
, AV_PIX_FMT_NONE
},