4 * This file is part of Libav.
6 * Libav is free software; you can redistribute it and/or
7 * modify it under the terms of the GNU Lesser General Public
8 * License as published by the Free Software Foundation; either
9 * version 2.1 of the License, or (at your option) any later version.
11 * Libav is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14 * Lesser General Public License for more details.
16 * You should have received a copy of the GNU Lesser General Public
17 * License along with Libav; if not, write to the Free Software
18 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
28 #include "libavutil/internal.h"
30 typedef struct AuraDecodeContext
{
31 AVCodecContext
*avctx
;
35 static av_cold
int aura_decode_init(AVCodecContext
*avctx
)
37 AuraDecodeContext
*s
= avctx
->priv_data
;
40 /* width needs to be divisible by 4 for this codec to work */
41 if (avctx
->width
& 0x3)
42 return AVERROR(EINVAL
);
43 avctx
->pix_fmt
= AV_PIX_FMT_YUV422P
;
48 static int aura_decode_frame(AVCodecContext
*avctx
,
49 void *data
, int *got_frame
,
52 AuraDecodeContext
*s
= avctx
->priv_data
;
56 const uint8_t *buf
= pkt
->data
;
58 /* prediction error tables (make it clear that they are signed values) */
59 const int8_t *delta_table
= (const int8_t*)buf
+ 16;
61 if (pkt
->size
!= 48 + avctx
->height
* avctx
->width
) {
62 av_log(avctx
, AV_LOG_ERROR
, "got a buffer with %d bytes when %d were expected\n",
63 pkt
->size
, 48 + avctx
->height
* avctx
->width
);
64 return AVERROR_INVALIDDATA
;
67 /* pixel data starts 48 bytes in, after 3x16-byte tables */
71 avctx
->release_buffer(avctx
, &s
->frame
);
73 s
->frame
.buffer_hints
= FF_BUFFER_HINTS_VALID
;
74 s
->frame
.reference
= 0;
75 if ((ret
= ff_get_buffer(avctx
, &s
->frame
)) < 0) {
76 av_log(avctx
, AV_LOG_ERROR
, "get_buffer() failed\n");
84 /* iterate through each line in the height */
85 for (y
= 0; y
< avctx
->height
; y
++) {
86 /* reset predictors */
92 Y
[1] = Y
[0] + delta_table
[val
& 0xF];
95 /* iterate through the remaining pixel groups (4 pixels/group) */
96 for (x
= 1; x
< (avctx
->width
>> 1); x
++) {
98 U
[0] = U
[-1] + delta_table
[val
>> 4];
99 Y
[0] = Y
[-1] + delta_table
[val
& 0xF];
101 V
[0] = V
[-1] + delta_table
[val
>> 4];
102 Y
[1] = Y
[ 0] + delta_table
[val
& 0xF];
105 Y
+= s
->frame
.linesize
[0] - avctx
->width
;
106 U
+= s
->frame
.linesize
[1] - (avctx
->width
>> 1);
107 V
+= s
->frame
.linesize
[2] - (avctx
->width
>> 1);
111 *(AVFrame
*)data
= s
->frame
;
116 static av_cold
int aura_decode_end(AVCodecContext
*avctx
)
118 AuraDecodeContext
*s
= avctx
->priv_data
;
120 if (s
->frame
.data
[0])
121 avctx
->release_buffer(avctx
, &s
->frame
);
126 AVCodec ff_aura2_decoder
= {
128 .type
= AVMEDIA_TYPE_VIDEO
,
129 .id
= AV_CODEC_ID_AURA2
,
130 .priv_data_size
= sizeof(AuraDecodeContext
),
131 .init
= aura_decode_init
,
132 .close
= aura_decode_end
,
133 .decode
= aura_decode_frame
,
134 .capabilities
= CODEC_CAP_DR1
,
135 .long_name
= NULL_IF_CONFIG_SMALL("Auravision Aura 2"),