/src/ffmpeg/libavcodec/y41penc.c
Line | Count | Source |
1 | | /* |
2 | | * y41p encoder |
3 | | * |
4 | | * Copyright (c) 2012 Paul B Mahol |
5 | | * |
6 | | * This file is part of FFmpeg. |
7 | | * |
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. |
12 | | * |
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. |
17 | | * |
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 |
21 | | */ |
22 | | |
23 | | #include "avcodec.h" |
24 | | #include "codec_internal.h" |
25 | | #include "encode.h" |
26 | | #include "internal.h" |
27 | | |
28 | | static av_cold int y41p_encode_init(AVCodecContext *avctx) |
29 | 194 | { |
30 | 194 | if (avctx->width & 7) { |
31 | 10 | av_log(avctx, AV_LOG_ERROR, "y41p requires width to be divisible by 8.\n"); |
32 | 10 | return AVERROR_INVALIDDATA; |
33 | 10 | } |
34 | | |
35 | 184 | avctx->bits_per_coded_sample = 12; |
36 | 184 | avctx->bit_rate = ff_guess_coded_bitrate(avctx); |
37 | | |
38 | 184 | return 0; |
39 | 194 | } |
40 | | |
41 | | static int y41p_encode_frame(AVCodecContext *avctx, AVPacket *pkt, |
42 | | const AVFrame *pic, int *got_packet) |
43 | 5.80k | { |
44 | 5.80k | uint8_t *dst; |
45 | 5.80k | const uint8_t *y, *u, *v; |
46 | 5.80k | int i, j, ret; |
47 | | |
48 | 5.80k | ret = ff_get_encode_buffer(avctx, pkt, avctx->width * avctx->height * 1.5, 0); |
49 | 5.80k | if (ret < 0) |
50 | 0 | return ret; |
51 | | |
52 | 5.80k | dst = pkt->data; |
53 | | |
54 | 724k | for (i = avctx->height - 1; i >= 0; i--) { |
55 | 718k | y = &pic->data[0][i * pic->linesize[0]]; |
56 | 718k | u = &pic->data[1][i * pic->linesize[1]]; |
57 | 718k | v = &pic->data[2][i * pic->linesize[2]]; |
58 | 2.05M | for (j = 0; j < avctx->width; j += 8) { |
59 | 1.33M | *(dst++) = *(u++); |
60 | 1.33M | *(dst++) = *(y++); |
61 | 1.33M | *(dst++) = *(v++); |
62 | 1.33M | *(dst++) = *(y++); |
63 | | |
64 | 1.33M | *(dst++) = *(u++); |
65 | 1.33M | *(dst++) = *(y++); |
66 | 1.33M | *(dst++) = *(v++); |
67 | 1.33M | *(dst++) = *(y++); |
68 | | |
69 | 1.33M | *(dst++) = *(y++); |
70 | 1.33M | *(dst++) = *(y++); |
71 | 1.33M | *(dst++) = *(y++); |
72 | 1.33M | *(dst++) = *(y++); |
73 | 1.33M | } |
74 | 718k | } |
75 | | |
76 | 5.80k | *got_packet = 1; |
77 | 5.80k | return 0; |
78 | 5.80k | } |
79 | | |
80 | | const FFCodec ff_y41p_encoder = { |
81 | | .p.name = "y41p", |
82 | | CODEC_LONG_NAME("Uncompressed YUV 4:1:1 12-bit"), |
83 | | .p.type = AVMEDIA_TYPE_VIDEO, |
84 | | .p.id = AV_CODEC_ID_Y41P, |
85 | | .p.capabilities = AV_CODEC_CAP_DR1 | AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE, |
86 | | .init = y41p_encode_init, |
87 | | FF_CODEC_ENCODE_CB(y41p_encode_frame), |
88 | | CODEC_PIXFMTS(AV_PIX_FMT_YUV411P), |
89 | | }; |