3 * Copyright (c) 2005 Konstantin Shishkov
5 * This file is part of FFmpeg.
7 * FFmpeg 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 * FFmpeg 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 FFmpeg; if not, write to the Free Software
19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
23 * @file libavcodec/loco.c
28 #include "bitstream.h"
32 enum LOCO_MODE
{LOCO_UNKN
=0, LOCO_CYUY2
=-1, LOCO_CRGB
=-2, LOCO_CRGBA
=-3, LOCO_CYV12
=-4,
33 LOCO_YUY2
=1, LOCO_UYVY
=2, LOCO_RGB
=3, LOCO_RGBA
=4, LOCO_YV12
=5};
35 typedef struct LOCOContext
{
36 AVCodecContext
*avctx
;
42 typedef struct RICEContext
{
44 int save
, run
, run2
; /* internal rice decoder state */
45 int sum
, count
; /* sum and count for getting rice parameter */
49 static int loco_get_rice_param(RICEContext
*r
)
54 while(r
->sum
> val
&& cnt
< 9) {
62 static inline void loco_update_rice_param(RICEContext
*r
, int val
)
73 static inline int loco_get_rice(RICEContext
*r
)
76 if (r
->run
> 0) { /* we have zero run */
78 loco_update_rice_param(r
, 0);
81 v
= get_ur_golomb_jpegls(&r
->gb
, loco_get_rice_param(r
), INT_MAX
, 0);
82 loco_update_rice_param(r
, (v
+1)>>1);
85 r
->run
= get_ur_golomb_jpegls(&r
->gb
, 2, INT_MAX
, 0);
87 r
->save
+= r
->run
+ 1;
94 v
= ((v
>>1) + r
->lossy
) ^ -(v
&1);
107 /* LOCO main predictor - LOCO-I/JPEG-LS predictor */
108 static inline int loco_predict(uint8_t* data
, int stride
, int step
)
114 c
= data
[-stride
- step
];
116 return mid_pred(a
, a
+ b
- c
, b
);
119 static int loco_decode_plane(LOCOContext
*l
, uint8_t *data
, int width
, int height
,
120 int stride
, const uint8_t *buf
, int buf_size
, int step
)
126 init_get_bits(&rc
.gb
, buf
, buf_size
*8);
135 /* restore top left pixel */
136 val
= loco_get_rice(&rc
);
138 /* restore top line */
139 for (i
= 1; i
< width
; i
++) {
140 val
= loco_get_rice(&rc
);
141 data
[i
* step
] = data
[i
* step
- step
] + val
;
144 for (j
= 1; j
< height
; j
++) {
145 /* restore left column */
146 val
= loco_get_rice(&rc
);
147 data
[0] = data
[-stride
] + val
;
148 /* restore all other pixels */
149 for (i
= 1; i
< width
; i
++) {
150 val
= loco_get_rice(&rc
);
151 data
[i
* step
] = loco_predict(&data
[i
* step
], stride
, step
) + val
;
156 return (get_bits_count(&rc
.gb
) + 7) >> 3;
159 static int decode_frame(AVCodecContext
*avctx
,
160 void *data
, int *data_size
,
161 const uint8_t *buf
, int buf_size
)
163 LOCOContext
* const l
= avctx
->priv_data
;
164 AVFrame
* const p
= (AVFrame
*)&l
->pic
;
168 avctx
->release_buffer(avctx
, p
);
171 if(avctx
->get_buffer(avctx
, p
) < 0){
172 av_log(avctx
, AV_LOG_ERROR
, "get_buffer() failed\n");
178 case LOCO_CYUY2
: case LOCO_YUY2
: case LOCO_UYVY
:
179 decoded
= loco_decode_plane(l
, p
->data
[0], avctx
->width
, avctx
->height
,
180 p
->linesize
[0], buf
, buf_size
, 1);
181 buf
+= decoded
; buf_size
-= decoded
;
182 decoded
= loco_decode_plane(l
, p
->data
[1], avctx
->width
/ 2, avctx
->height
,
183 p
->linesize
[1], buf
, buf_size
, 1);
184 buf
+= decoded
; buf_size
-= decoded
;
185 decoded
= loco_decode_plane(l
, p
->data
[2], avctx
->width
/ 2, avctx
->height
,
186 p
->linesize
[2], buf
, buf_size
, 1);
188 case LOCO_CYV12
: case LOCO_YV12
:
189 decoded
= loco_decode_plane(l
, p
->data
[0], avctx
->width
, avctx
->height
,
190 p
->linesize
[0], buf
, buf_size
, 1);
191 buf
+= decoded
; buf_size
-= decoded
;
192 decoded
= loco_decode_plane(l
, p
->data
[2], avctx
->width
/ 2, avctx
->height
/ 2,
193 p
->linesize
[2], buf
, buf_size
, 1);
194 buf
+= decoded
; buf_size
-= decoded
;
195 decoded
= loco_decode_plane(l
, p
->data
[1], avctx
->width
/ 2, avctx
->height
/ 2,
196 p
->linesize
[1], buf
, buf_size
, 1);
198 case LOCO_CRGB
: case LOCO_RGB
:
199 decoded
= loco_decode_plane(l
, p
->data
[0] + p
->linesize
[0]*(avctx
->height
-1), avctx
->width
, avctx
->height
,
200 -p
->linesize
[0], buf
, buf_size
, 3);
201 buf
+= decoded
; buf_size
-= decoded
;
202 decoded
= loco_decode_plane(l
, p
->data
[0] + p
->linesize
[0]*(avctx
->height
-1) + 1, avctx
->width
, avctx
->height
,
203 -p
->linesize
[0], buf
, buf_size
, 3);
204 buf
+= decoded
; buf_size
-= decoded
;
205 decoded
= loco_decode_plane(l
, p
->data
[0] + p
->linesize
[0]*(avctx
->height
-1) + 2, avctx
->width
, avctx
->height
,
206 -p
->linesize
[0], buf
, buf_size
, 3);
209 decoded
= loco_decode_plane(l
, p
->data
[0], avctx
->width
, avctx
->height
,
210 p
->linesize
[0], buf
, buf_size
, 4);
211 buf
+= decoded
; buf_size
-= decoded
;
212 decoded
= loco_decode_plane(l
, p
->data
[0] + 1, avctx
->width
, avctx
->height
,
213 p
->linesize
[0], buf
, buf_size
, 4);
214 buf
+= decoded
; buf_size
-= decoded
;
215 decoded
= loco_decode_plane(l
, p
->data
[0] + 2, avctx
->width
, avctx
->height
,
216 p
->linesize
[0], buf
, buf_size
, 4);
217 buf
+= decoded
; buf_size
-= decoded
;
218 decoded
= loco_decode_plane(l
, p
->data
[0] + 3, avctx
->width
, avctx
->height
,
219 p
->linesize
[0], buf
, buf_size
, 4);
223 *data_size
= sizeof(AVFrame
);
224 *(AVFrame
*)data
= l
->pic
;
229 static av_cold
int decode_init(AVCodecContext
*avctx
){
230 LOCOContext
* const l
= avctx
->priv_data
;
234 if (avctx
->extradata_size
< 12) {
235 av_log(avctx
, AV_LOG_ERROR
, "Extradata size must be >= 12 instead of %i\n",
236 avctx
->extradata_size
);
239 version
= AV_RL32(avctx
->extradata
);
245 l
->lossy
= AV_RL32(avctx
->extradata
+ 8);
248 l
->lossy
= AV_RL32(avctx
->extradata
+ 8);
249 av_log(avctx
, AV_LOG_INFO
, "This is LOCO codec version %i, please upload file for study\n", version
);
252 l
->mode
= AV_RL32(avctx
->extradata
+ 4);
254 case LOCO_CYUY2
: case LOCO_YUY2
: case LOCO_UYVY
:
255 avctx
->pix_fmt
= PIX_FMT_YUV422P
;
257 case LOCO_CRGB
: case LOCO_RGB
:
258 avctx
->pix_fmt
= PIX_FMT_BGR24
;
260 case LOCO_CYV12
: case LOCO_YV12
:
261 avctx
->pix_fmt
= PIX_FMT_YUV420P
;
263 case LOCO_CRGBA
: case LOCO_RGBA
:
264 avctx
->pix_fmt
= PIX_FMT_RGB32
;
267 av_log(avctx
, AV_LOG_INFO
, "Unknown colorspace, index = %i\n", l
->mode
);
270 if(avctx
->debug
& FF_DEBUG_PICT_INFO
)
271 av_log(avctx
, AV_LOG_INFO
, "lossy:%i, version:%i, mode: %i\n", l
->lossy
, version
, l
->mode
);
276 AVCodec loco_decoder
= {
286 .long_name
= NULL_IF_CONFIG_SMALL("LOCO"),