3 * Copyright (c) 2005 Konstantin Shishkov
5 * This file is part of Libav.
7 * Libav is free software; you can redistribute it and/or
8 * modify it under the terms of the GNU Lesser General Public
9 * License as published by the Free Software Foundation; either
10 * version 2.1 of the License, or (at your option) any later version.
12 * Libav is distributed in the hope that it will be useful,
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15 * Lesser General Public License for more details.
17 * You should have received a copy of the GNU Lesser General Public
18 * License along with Libav; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
46 typedef struct LOCOContext
{
47 AVCodecContext
*avctx
;
53 typedef struct RICEContext
{
55 int save
, run
, run2
; /* internal rice decoder state */
56 int sum
, count
; /* sum and count for getting rice parameter */
60 static int loco_get_rice_param(RICEContext
*r
)
65 while (r
->sum
> val
&& cnt
< 9) {
73 static inline void loco_update_rice_param(RICEContext
*r
, int val
)
84 static inline int loco_get_rice(RICEContext
*r
)
87 if (r
->run
> 0) { /* we have zero run */
89 loco_update_rice_param(r
, 0);
92 v
= get_ur_golomb_jpegls(&r
->gb
, loco_get_rice_param(r
), INT_MAX
, 0);
93 loco_update_rice_param(r
, (v
+ 1) >> 1);
96 r
->run
= get_ur_golomb_jpegls(&r
->gb
, 2, INT_MAX
, 0);
98 r
->save
+= r
->run
+ 1;
104 v
= ((v
>> 1) + r
->lossy
) ^ -(v
& 1);
117 /* LOCO main predictor - LOCO-I/JPEG-LS predictor */
118 static inline int loco_predict(uint8_t* data
, int stride
, int step
)
124 c
= data
[-stride
- step
];
126 return mid_pred(a
, a
+ b
- c
, b
);
129 static int loco_decode_plane(LOCOContext
*l
, uint8_t *data
, int width
, int height
,
130 int stride
, const uint8_t *buf
, int buf_size
, int step
)
136 init_get_bits(&rc
.gb
, buf
, buf_size
*8);
145 /* restore top left pixel */
146 val
= loco_get_rice(&rc
);
148 /* restore top line */
149 for (i
= 1; i
< width
; i
++) {
150 val
= loco_get_rice(&rc
);
151 data
[i
* step
] = data
[i
* step
- step
] + val
;
154 for (j
= 1; j
< height
; j
++) {
155 /* restore left column */
156 val
= loco_get_rice(&rc
);
157 data
[0] = data
[-stride
] + val
;
158 /* restore all other pixels */
159 for (i
= 1; i
< width
; i
++) {
160 val
= loco_get_rice(&rc
);
161 data
[i
* step
] = loco_predict(&data
[i
* step
], stride
, step
) + val
;
166 return (get_bits_count(&rc
.gb
) + 7) >> 3;
169 static int decode_frame(AVCodecContext
*avctx
,
170 void *data
, int *got_frame
,
173 LOCOContext
* const l
= avctx
->priv_data
;
174 const uint8_t *buf
= avpkt
->data
;
175 int buf_size
= avpkt
->size
;
176 AVFrame
* const p
= &l
->pic
;
180 avctx
->release_buffer(avctx
, p
);
183 if ((ret
= ff_get_buffer(avctx
, p
)) < 0) {
184 av_log(avctx
, AV_LOG_ERROR
, "get_buffer() failed\n");
190 case LOCO_CYUY2
: case LOCO_YUY2
: case LOCO_UYVY
:
191 decoded
= loco_decode_plane(l
, p
->data
[0], avctx
->width
, avctx
->height
,
192 p
->linesize
[0], buf
, buf_size
, 1);
193 buf
+= decoded
; buf_size
-= decoded
;
194 decoded
= loco_decode_plane(l
, p
->data
[1], avctx
->width
/ 2, avctx
->height
,
195 p
->linesize
[1], buf
, buf_size
, 1);
196 buf
+= decoded
; buf_size
-= decoded
;
197 decoded
= loco_decode_plane(l
, p
->data
[2], avctx
->width
/ 2, avctx
->height
,
198 p
->linesize
[2], buf
, buf_size
, 1);
200 case LOCO_CYV12
: case LOCO_YV12
:
201 decoded
= loco_decode_plane(l
, p
->data
[0], avctx
->width
, avctx
->height
,
202 p
->linesize
[0], buf
, buf_size
, 1);
203 buf
+= decoded
; buf_size
-= decoded
;
204 decoded
= loco_decode_plane(l
, p
->data
[2], avctx
->width
/ 2, avctx
->height
/ 2,
205 p
->linesize
[2], buf
, buf_size
, 1);
206 buf
+= decoded
; buf_size
-= decoded
;
207 decoded
= loco_decode_plane(l
, p
->data
[1], avctx
->width
/ 2, avctx
->height
/ 2,
208 p
->linesize
[1], buf
, buf_size
, 1);
210 case LOCO_CRGB
: case LOCO_RGB
:
211 decoded
= loco_decode_plane(l
, p
->data
[0] + p
->linesize
[0]*(avctx
->height
-1), avctx
->width
, avctx
->height
,
212 -p
->linesize
[0], buf
, buf_size
, 3);
213 buf
+= decoded
; buf_size
-= decoded
;
214 decoded
= loco_decode_plane(l
, p
->data
[0] + p
->linesize
[0]*(avctx
->height
-1) + 1, avctx
->width
, avctx
->height
,
215 -p
->linesize
[0], buf
, buf_size
, 3);
216 buf
+= decoded
; buf_size
-= decoded
;
217 decoded
= loco_decode_plane(l
, p
->data
[0] + p
->linesize
[0]*(avctx
->height
-1) + 2, avctx
->width
, avctx
->height
,
218 -p
->linesize
[0], buf
, buf_size
, 3);
221 decoded
= loco_decode_plane(l
, p
->data
[0], avctx
->width
, avctx
->height
,
222 p
->linesize
[0], buf
, buf_size
, 4);
223 buf
+= decoded
; buf_size
-= decoded
;
224 decoded
= loco_decode_plane(l
, p
->data
[0] + 1, avctx
->width
, avctx
->height
,
225 p
->linesize
[0], buf
, buf_size
, 4);
226 buf
+= decoded
; buf_size
-= decoded
;
227 decoded
= loco_decode_plane(l
, p
->data
[0] + 2, avctx
->width
, avctx
->height
,
228 p
->linesize
[0], buf
, buf_size
, 4);
229 buf
+= decoded
; buf_size
-= decoded
;
230 decoded
= loco_decode_plane(l
, p
->data
[0] + 3, avctx
->width
, avctx
->height
,
231 p
->linesize
[0], buf
, buf_size
, 4);
236 *(AVFrame
*)data
= l
->pic
;
241 static av_cold
int decode_init(AVCodecContext
*avctx
)
243 LOCOContext
* const l
= avctx
->priv_data
;
247 if (avctx
->extradata_size
< 12) {
248 av_log(avctx
, AV_LOG_ERROR
, "Extradata size must be >= 12 instead of %i\n",
249 avctx
->extradata_size
);
250 return AVERROR_INVALIDDATA
;
252 version
= AV_RL32(avctx
->extradata
);
258 l
->lossy
= AV_RL32(avctx
->extradata
+ 8);
261 l
->lossy
= AV_RL32(avctx
->extradata
+ 8);
262 av_log_ask_for_sample(avctx
, "This is LOCO codec version %i.\n", version
);
265 l
->mode
= AV_RL32(avctx
->extradata
+ 4);
270 avctx
->pix_fmt
= AV_PIX_FMT_YUV422P
;
274 avctx
->pix_fmt
= AV_PIX_FMT_BGR24
;
278 avctx
->pix_fmt
= AV_PIX_FMT_YUV420P
;
282 avctx
->pix_fmt
= AV_PIX_FMT_RGB32
;
285 av_log(avctx
, AV_LOG_INFO
, "Unknown colorspace, index = %i\n", l
->mode
);
286 return AVERROR_INVALIDDATA
;
288 if (avctx
->debug
& FF_DEBUG_PICT_INFO
)
289 av_log(avctx
, AV_LOG_INFO
, "lossy:%i, version:%i, mode: %i\n", l
->lossy
, version
, l
->mode
);
294 static av_cold
int decode_end(AVCodecContext
*avctx
)
296 LOCOContext
* const l
= avctx
->priv_data
;
297 AVFrame
*pic
= &l
->pic
;
300 avctx
->release_buffer(avctx
, pic
);
305 AVCodec ff_loco_decoder
= {
307 .type
= AVMEDIA_TYPE_VIDEO
,
308 .id
= AV_CODEC_ID_LOCO
,
309 .priv_data_size
= sizeof(LOCOContext
),
312 .decode
= decode_frame
,
313 .capabilities
= CODEC_CAP_DR1
,
314 .long_name
= NULL_IF_CONFIG_SMALL("LOCO"),