~medibuntu-maintainers/mplayer/medibuntu.precise

« back to all changes in this revision

Viewing changes to ffmpeg/libavcodec/v410enc.c

  • Committer: Package Import Robot
  • Author(s): Reinhard Tartler
  • Date: 2012-01-12 22:23:28 UTC
  • mfrom: (0.4.7 sid)
  • mto: This revision was merged to the branch mainline in revision 76.
  • Revision ID: package-import@ubuntu.com-20120112222328-8jqdyodym3p84ygu
Tags: 2:1.0~rc4.dfsg1+svn34540-1
* New upstream snapshot
* upload to unstable

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/*
 
2
 * v410 encoder
 
3
 *
 
4
 * Copyright (c) 2011 Derek Buitenhuis
 
5
 *
 
6
 * This file is part of Libav.
 
7
 *
 
8
 * Libav 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
 * Libav 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 Libav; if not, write to the Free Software
 
20
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 
21
 */
 
22
 
 
23
#include "libavutil/intreadwrite.h"
 
24
#include "avcodec.h"
 
25
 
 
26
static av_cold int v410_encode_init(AVCodecContext *avctx)
 
27
{
 
28
    if (avctx->width & 1) {
 
29
        av_log(avctx, AV_LOG_ERROR, "v410 requires even width.\n");
 
30
        return AVERROR_INVALIDDATA;
 
31
    }
 
32
 
 
33
    avctx->coded_frame = avcodec_alloc_frame();
 
34
 
 
35
    if (!avctx->coded_frame) {
 
36
        av_log(avctx, AV_LOG_ERROR, "Could not allocate frame.\n");
 
37
        return AVERROR(ENOMEM);
 
38
    }
 
39
 
 
40
    return 0;
 
41
}
 
42
 
 
43
static int v410_encode_frame(AVCodecContext *avctx, uint8_t *buf,
 
44
                             int buf_size, void *data)
 
45
{
 
46
    AVFrame *pic = data;
 
47
    uint8_t *dst = buf;
 
48
    uint16_t *y, *u, *v;
 
49
    uint32_t val;
 
50
    int i, j;
 
51
    int output_size = 0;
 
52
 
 
53
    if (buf_size < avctx->width * avctx->height * 4) {
 
54
        av_log(avctx, AV_LOG_ERROR, "Out buffer is too small.\n");
 
55
        return AVERROR(ENOMEM);
 
56
    }
 
57
 
 
58
    avctx->coded_frame->reference = 0;
 
59
    avctx->coded_frame->key_frame = 1;
 
60
    avctx->coded_frame->pict_type = FF_I_TYPE;
 
61
 
 
62
    y = (uint16_t *)pic->data[0];
 
63
    u = (uint16_t *)pic->data[1];
 
64
    v = (uint16_t *)pic->data[2];
 
65
 
 
66
    for (i = 0; i < avctx->height; i++) {
 
67
        for (j = 0; j < avctx->width; j++) {
 
68
            val  = u[j] << 2;
 
69
            val |= y[j] << 12;
 
70
            val |= (uint32_t) v[j] << 22;
 
71
            AV_WL32(dst, val);
 
72
            dst += 4;
 
73
            output_size += 4;
 
74
        }
 
75
        y += pic->linesize[0] >> 1;
 
76
        u += pic->linesize[1] >> 1;
 
77
        v += pic->linesize[2] >> 1;
 
78
    }
 
79
 
 
80
    return output_size;
 
81
}
 
82
 
 
83
static av_cold int v410_encode_close(AVCodecContext *avctx)
 
84
{
 
85
    av_freep(&avctx->coded_frame);
 
86
 
 
87
    return 0;
 
88
}
 
89
 
 
90
AVCodec ff_v410_encoder = {
 
91
    .name         = "v410",
 
92
    .type         = AVMEDIA_TYPE_VIDEO,
 
93
    .id           = CODEC_ID_V410,
 
94
    .init         = v410_encode_init,
 
95
    .encode       = v410_encode_frame,
 
96
    .close        = v410_encode_close,
 
97
    .pix_fmts     = (const enum PixelFormat[]){ PIX_FMT_YUV444P10, PIX_FMT_NONE },
 
98
    .long_name    = NULL_IF_CONFIG_SMALL("Uncompressed 4:4:4 10-bit"),
 
99
};