FFmpeg
r210enc.c
Go to the documentation of this file.
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 
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 
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 *srcr_line, *srcg_line, *srcb_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  srcg_line = pic->data[0];
53  srcb_line = pic->data[1];
54  srcr_line = pic->data[2];
55  dst = pkt->data;
56 
57  for (i = 0; i < avctx->height; i++) {
58  uint16_t *srcr = (uint16_t *)srcr_line;
59  uint16_t *srcg = (uint16_t *)srcg_line;
60  uint16_t *srcb = (uint16_t *)srcb_line;
61  for (j = 0; j < avctx->width; j++) {
62  uint32_t pixel;
63  unsigned r = *srcr++;
64  unsigned g = *srcg++;
65  unsigned b = *srcb++;
66  if (avctx->codec_id == AV_CODEC_ID_R210)
67  pixel = (r << 20) | (g << 10) | b;
68  else
69  pixel = (r << 22) | (g << 12) | (b << 2);
70  if (avctx->codec_id == AV_CODEC_ID_AVRP)
71  bytestream_put_le32(&dst, pixel);
72  else
73  bytestream_put_be32(&dst, pixel);
74  }
75  memset(dst, 0, pad);
76  dst += pad;
77  srcr_line += pic->linesize[2];
78  srcg_line += pic->linesize[0];
79  srcb_line += pic->linesize[1];
80  }
81 
82  pkt->flags |= AV_PKT_FLAG_KEY;
83  *got_packet = 1;
84  return 0;
85 }
86 
87 
88 #if CONFIG_R210_ENCODER
90  .name = "r210",
91  .long_name = NULL_IF_CONFIG_SMALL("Uncompressed RGB 10-bit"),
92  .type = AVMEDIA_TYPE_VIDEO,
93  .id = AV_CODEC_ID_R210,
94  .init = encode_init,
95  .encode2 = encode_frame,
96  .pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_GBRP10, AV_PIX_FMT_NONE },
97 };
98 #endif
99 #if CONFIG_R10K_ENCODER
101  .name = "r10k",
102  .long_name = NULL_IF_CONFIG_SMALL("AJA Kona 10-bit RGB Codec"),
103  .type = AVMEDIA_TYPE_VIDEO,
104  .id = AV_CODEC_ID_R10K,
105  .init = encode_init,
106  .encode2 = encode_frame,
107  .pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_GBRP10, AV_PIX_FMT_NONE },
108 };
109 #endif
110 #if CONFIG_AVRP_ENCODER
112  .name = "avrp",
113  .long_name = NULL_IF_CONFIG_SMALL("Avid 1:1 10-bit RGB Packer"),
114  .type = AVMEDIA_TYPE_VIDEO,
115  .id = AV_CODEC_ID_AVRP,
116  .init = encode_init,
117  .encode2 = encode_frame,
118  .pix_fmts = (const enum AVPixelFormat[]) { AV_PIX_FMT_GBRP10, AV_PIX_FMT_NONE },
119 };
120 #endif
This structure describes decoded (raw) audio or video data.
Definition: frame.h:308
int64_t bit_rate
the average bitrate
Definition: avcodec.h:576
const char * g
Definition: vf_curves.c:115
#define AV_PIX_FMT_GBRP10
Definition: pixfmt.h:415
static av_cold int encode_init(AVCodecContext *avctx)
Definition: r210enc.c:27
static AVPacket pkt
AVCodec.
Definition: codec.h:190
int ff_alloc_packet2(AVCodecContext *avctx, AVPacket *avpkt, int64_t size, int64_t min_size)
Check AVPacket size and/or allocate data.
Definition: encode.c:33
uint8_t
#define av_cold
Definition: attributes.h:88
uint8_t * data
Definition: packet.h:355
int bits_per_coded_sample
bits per sample/pixel from the demuxer (needed for huffyuv).
Definition: avcodec.h:1750
#define FFALIGN(x, a)
Definition: macros.h:48
#define AV_PKT_FLAG_KEY
The packet contains a keyframe.
Definition: packet.h:388
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:153
const char * r
Definition: vf_curves.c:114
const char * name
Name of the codec implementation.
Definition: codec.h:197
int flags
A combination of AV_PKT_FLAG values.
Definition: packet.h:361
int64_t ff_guess_coded_bitrate(AVCodecContext *avctx)
Get an estimated video bitrate based on frame size, frame rate and coded bits per pixel...
Definition: utils.c:2310
#define b
Definition: input.c:41
int width
picture width / height.
Definition: avcodec.h:699
AVCodec ff_avrp_encoder
AVCodec ff_r10k_encoder
Libavcodec external API header.
enum AVCodecID codec_id
Definition: avcodec.h:536
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:339
main external API structure.
Definition: avcodec.h:526
AVCodec ff_r210_encoder
uint8_t pixel
Definition: tiny_ssim.c:42
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:322
common internal api header.
static int encode_frame(AVCodecContext *avctx, AVPacket *pkt, const AVFrame *pic, int *got_packet)
Definition: r210enc.c:39
AVPixelFormat
Pixel format.
Definition: pixfmt.h:64
This structure stores compressed data.
Definition: packet.h:332
int i
Definition: input.c:406