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
30 #include "libavutil/pixdesc.h"
31 #include "libavutil/intreadwrite.h"
32 #include "libavutil/internal.h"
34 static av_cold
int raw_encode_init(AVCodecContext
*avctx
)
36 const AVPixFmtDescriptor
*desc
= av_pix_fmt_desc_get(avctx
->pix_fmt
);
38 avctx
->coded_frame
= av_frame_alloc();
39 avctx
->coded_frame
->pict_type
= AV_PICTURE_TYPE_I
;
40 avctx
->bits_per_coded_sample
= av_get_bits_per_pixel(desc
);
42 avctx
->codec_tag
= avcodec_pix_fmt_to_codec_tag(avctx
->pix_fmt
);
46 static int raw_encode(AVCodecContext
*avctx
, AVPacket
*pkt
,
47 const AVFrame
*frame
, int *got_packet
)
49 int ret
= avpicture_get_size(avctx
->pix_fmt
, avctx
->width
, avctx
->height
);
54 if ((ret
= ff_alloc_packet2(avctx
, pkt
, ret
)) < 0)
56 if ((ret
= avpicture_layout((const AVPicture
*)frame
, avctx
->pix_fmt
, avctx
->width
,
57 avctx
->height
, pkt
->data
, pkt
->size
)) < 0)
60 if(avctx
->codec_tag
== AV_RL32("yuv2") && ret
> 0 &&
61 avctx
->pix_fmt
== AV_PIX_FMT_YUYV422
) {
63 for(x
= 1; x
< avctx
->height
*avctx
->width
*2; x
+= 2)
66 pkt
->flags
|= AV_PKT_FLAG_KEY
;
71 static av_cold
int raw_close(AVCodecContext
*avctx
)
73 av_frame_free(&avctx
->coded_frame
);
77 AVCodec ff_rawvideo_encoder
= {
79 .long_name
= NULL_IF_CONFIG_SMALL("raw video"),
80 .type
= AVMEDIA_TYPE_VIDEO
,
81 .id
= AV_CODEC_ID_RAWVIDEO
,
82 .init
= raw_encode_init
,
83 .encode2
= raw_encode
,