Main Page
Related Pages
Modules
Data Structures
Files
Examples
File List
Globals
libavcodec
v410enc.c
Go to the documentation of this file.
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
=
AV_PICTURE_TYPE_I
;
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
};