LCOV - code coverage report
Current view: top level - libavcodec - r210enc.c (source / functions) Hit Total Coverage
Test: coverage.info Lines: 28 31 90.3 %
Date: 2017-12-12 11:08:38 Functions: 2 2 100.0 %

          Line data    Source code
       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           4 : static av_cold int encode_init(AVCodecContext *avctx)
      28             : {
      29           4 :     int aligned_width = FFALIGN(avctx->width,
      30             :                                 avctx->codec_id == AV_CODEC_ID_R10K ? 1 : 64);
      31             : 
      32           4 :     avctx->bits_per_coded_sample = 32;
      33           4 :     if (avctx->width > 0)
      34           4 :         avctx->bit_rate = ff_guess_coded_bitrate(avctx) * aligned_width / avctx->width;
      35             : 
      36           4 :     return 0;
      37             : }
      38             : 
      39         200 : static int encode_frame(AVCodecContext *avctx, AVPacket *pkt,
      40             :                         const AVFrame *pic, int *got_packet)
      41             : {
      42             :     int i, j, ret;
      43         200 :     int aligned_width = FFALIGN(avctx->width,
      44             :                                 avctx->codec_id == AV_CODEC_ID_R10K ? 1 : 64);
      45         200 :     int pad = (aligned_width - avctx->width) * 4;
      46             :     uint8_t *src_line;
      47             :     uint8_t *dst;
      48             : 
      49         200 :     if ((ret = ff_alloc_packet2(avctx, pkt, 4 * aligned_width * avctx->height, 0)) < 0)
      50           0 :         return ret;
      51             : 
      52         200 :     src_line = pic->data[0];
      53         200 :     dst = pkt->data;
      54             : 
      55       45100 :     for (i = 0; i < avctx->height; i++) {
      56       44900 :         uint16_t *src = (uint16_t *)src_line;
      57    15309100 :         for (j = 0; j < avctx->width; j++) {
      58             :             uint32_t pixel;
      59    15264200 :             uint16_t r = *src++ >> 6;
      60    15264200 :             uint16_t g = *src++ >> 6;
      61    15264200 :             uint16_t b = *src++ >> 6;
      62    15264200 :             if (avctx->codec_id == AV_CODEC_ID_R210)
      63    15264200 :                 pixel = (r << 20) | (g << 10) | b;
      64             :             else
      65           0 :                 pixel = (r << 22) | (g << 12) | (b << 2);
      66    15264200 :             if (avctx->codec_id == AV_CODEC_ID_AVRP)
      67           0 :                 bytestream_put_le32(&dst, pixel);
      68             :             else
      69    15264200 :                 bytestream_put_be32(&dst, pixel);
      70             :         }
      71       44900 :         memset(dst, 0, pad);
      72       44900 :         dst += pad;
      73       44900 :         src_line += pic->linesize[0];
      74             :     }
      75             : 
      76         200 :     pkt->flags |= AV_PKT_FLAG_KEY;
      77         200 :     *got_packet = 1;
      78         200 :     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

Generated by: LCOV version 1.13