path: root/libavcodec/r210enc.c (plain)
blob: a55e5434f317f1a09857d380745b663594d9f56c

1	/*
2	* R210 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 "internal.h"
25	#include "bytestream.h"
26
27	static av_cold int encode_init(AVCodecContext *avctx)
28	{
29	int aligned_width = FFALIGN(avctx->width,
30	avctx->codec_id == AV_CODEC_ID_R10K ? 1 : 64);
31
32	avctx->bits_per_coded_sample = 32;
33	if (avctx->width > 0)
34	avctx->bit_rate = ff_guess_coded_bitrate(avctx) * aligned_width / avctx->width;
35
36	return 0;
37	}
38
39	static int encode_frame(AVCodecContext *avctx, AVPacket *pkt,
40	const AVFrame *pic, int *got_packet)
41	{
42	int i, j, ret;
43	int aligned_width = FFALIGN(avctx->width,
44	avctx->codec_id == AV_CODEC_ID_R10K ? 1 : 64);
45	int pad = (aligned_width - avctx->width) * 4;
46	uint8_t *src_line;
47	uint8_t *dst;
48
49	if ((ret = ff_alloc_packet2(avctx, pkt, 4 * aligned_width * avctx->height, 0)) < 0)
50	return ret;
51
52	src_line = pic->data[0];
53	dst = pkt->data;
54
55	for (i = 0; i < avctx->height; i++) {
56	uint16_t *src = (uint16_t *)src_line;
57	for (j = 0; j < avctx->width; j++) {
58	uint32_t pixel;
59	uint16_t r = *src++ >> 6;
60	uint16_t g = *src++ >> 6;
61	uint16_t b = *src++ >> 6;
62	if (avctx->codec_id == AV_CODEC_ID_R210)
63	pixel = (r << 20) | (g << 10) | b;
64	else
65	pixel = (r << 22) | (g << 12) | (b << 2);
66	if (avctx->codec_id == AV_CODEC_ID_AVRP)
67	bytestream_put_le32(&dst, pixel);
68	else
69	bytestream_put_be32(&dst, pixel);
70	}
71	memset(dst, 0, pad);
72	dst += pad;
73	src_line += pic->linesize[0];
74	}
75
76	pkt->flags |= AV_PKT_FLAG_KEY;
77	*got_packet = 1;
78	return 0;
79	}
80
81
82	#if CONFIG_R210_ENCODER
83	AVCodec ff_r210_encoder = {
84	.name = "r210",
85	.long_name = NULL_IF_CONFIG_SMALL("Uncompressed RGB 10-bit"),
86	.type = AVMEDIA_TYPE_VIDEO,
87	.id = AV_CODEC_ID_R210,
88	.init = encode_init,
89	.encode2 = encode_frame,
90	.pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_RGB48, AV_PIX_FMT_NONE },
91	.capabilities = AV_CODEC_CAP_INTRA_ONLY,
92	};
93	#endif
94	#if CONFIG_R10K_ENCODER
95	AVCodec ff_r10k_encoder = {
96	.name = "r10k",
97	.long_name = NULL_IF_CONFIG_SMALL("AJA Kona 10-bit RGB Codec"),
98	.type = AVMEDIA_TYPE_VIDEO,
99	.id = AV_CODEC_ID_R10K,
100	.init = encode_init,
101	.encode2 = encode_frame,
102	.pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_RGB48, AV_PIX_FMT_NONE },
103	.capabilities = AV_CODEC_CAP_INTRA_ONLY,
104	};
105	#endif
106	#if CONFIG_AVRP_ENCODER
107	AVCodec ff_avrp_encoder = {
108	.name = "avrp",
109	.long_name = NULL_IF_CONFIG_SMALL("Avid 1:1 10-bit RGB Packer"),
110	.type = AVMEDIA_TYPE_VIDEO,
111	.id = AV_CODEC_ID_AVRP,
112	.init = encode_init,
113	.encode2 = encode_frame,
114	.pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_RGB48, AV_PIX_FMT_NONE },
115	.capabilities = AV_CODEC_CAP_INTRA_ONLY,
116	};
117	#endif
118