FFmpeg
loco.c
Go to the documentation of this file.
1 /*
2  * LOCO codec
3  * Copyright (c) 2005 Konstantin Shishkov
4  *
5  * This file is part of FFmpeg.
6  *
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.
11  *
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.
16  *
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
20  */
21 
22 /**
23  * @file
24  * LOCO codec.
25  */
26 
27 #include "avcodec.h"
28 #include "get_bits.h"
29 #include "golomb.h"
30 #include "internal.h"
31 #include "mathops.h"
32 
33 enum LOCO_MODE {
34  LOCO_UNKN = 0,
35  LOCO_CYUY2 = -1,
36  LOCO_CRGB = -2,
37  LOCO_CRGBA = -3,
38  LOCO_CYV12 = -4,
39  LOCO_YUY2 = 1,
40  LOCO_UYVY = 2,
41  LOCO_RGB = 3,
42  LOCO_RGBA = 4,
43  LOCO_YV12 = 5,
44 };
45 
46 typedef struct LOCOContext {
48  int lossy;
50 } LOCOContext;
51 
52 typedef struct RICEContext {
54  int save, run, run2; /* internal rice decoder state */
55  int sum, count; /* sum and count for getting rice parameter */
56  int lossy;
57 } RICEContext;
58 
60 {
61  int cnt = 0;
62  int val = r->count;
63 
64  while (r->sum > val && cnt < 9) {
65  val <<= 1;
66  cnt++;
67  }
68 
69  return cnt;
70 }
71 
72 static inline void loco_update_rice_param(RICEContext *r, int val)
73 {
74  r->sum += val;
75  r->count++;
76 
77  if (r->count == 16) {
78  r->sum >>= 1;
79  r->count >>= 1;
80  }
81 }
82 
83 static inline int loco_get_rice(RICEContext *r)
84 {
85  int v;
86  if (r->run > 0) { /* we have zero run */
87  r->run--;
89  return 0;
90  }
91  if (get_bits_left(&r->gb) < 1)
92  return INT_MIN;
93  v = get_ur_golomb_jpegls(&r->gb, loco_get_rice_param(r), INT_MAX, 0);
94  loco_update_rice_param(r, (v + 1) >> 1);
95  if (!v) {
96  if (r->save >= 0) {
97  r->run = get_ur_golomb_jpegls(&r->gb, 2, INT_MAX, 0);
98  if (r->run > 1)
99  r->save += r->run + 1;
100  else
101  r->save -= 3;
102  } else
103  r->run2++;
104  } else {
105  v = ((v >> 1) + r->lossy) ^ -(v & 1);
106  if (r->run2 > 0) {
107  if (r->run2 > 2)
108  r->save += r->run2;
109  else
110  r->save -= 3;
111  r->run2 = 0;
112  }
113  }
114 
115  return v;
116 }
117 
118 /* LOCO main predictor - LOCO-I/JPEG-LS predictor */
119 static inline int loco_predict(uint8_t* data, int stride)
120 {
121  int a, b, c;
122 
123  a = data[-stride];
124  b = data[-1];
125  c = data[-stride - 1];
126 
127  return mid_pred(a, a + b - c, b);
128 }
129 
131  int stride, const uint8_t *buf, int buf_size)
132 {
133  RICEContext rc;
134  int val;
135  int ret;
136  int i, j;
137 
138  if(buf_size<=0)
139  return -1;
140 
141  if ((ret = init_get_bits8(&rc.gb, buf, buf_size)) < 0)
142  return ret;
143 
144  rc.save = 0;
145  rc.run = 0;
146  rc.run2 = 0;
147  rc.lossy = l->lossy;
148 
149  rc.sum = 8;
150  rc.count = 1;
151 
152  /* restore top left pixel */
153  val = loco_get_rice(&rc);
154  data[0] = 128 + val;
155  /* restore top line */
156  for (i = 1; i < width; i++) {
157  val = loco_get_rice(&rc);
158  if (val == INT_MIN)
159  return AVERROR_INVALIDDATA;
160  data[i] = data[i - 1] + val;
161  }
162  data += stride;
163  for (j = 1; j < height; j++) {
164  /* restore left column */
165  val = loco_get_rice(&rc);
166  if (val == INT_MIN)
167  return AVERROR_INVALIDDATA;
168  data[0] = data[-stride] + val;
169  /* restore all other pixels */
170  for (i = 1; i < width; i++) {
171  val = loco_get_rice(&rc);
172  if (val == INT_MIN)
173  return -1;
174  data[i] = loco_predict(&data[i], stride) + val;
175  }
176  data += stride;
177  }
178 
179  return (get_bits_count(&rc.gb) + 7) >> 3;
180 }
181 
182 static void rotate_faulty_loco(uint8_t *data, int width, int height, int stride)
183 {
184  int y;
185 
186  for (y=1; y<height; y++) {
187  if (width>=y) {
188  memmove(data + y*stride,
189  data + y*(stride + 1),
190  (width-y));
191  if (y+1 < height)
192  memmove(data + y*stride + (width-y),
193  data + (y+1)*stride, y);
194  }
195  }
196 }
197 
199  void *data, int *got_frame,
200  AVPacket *avpkt)
201 {
202  LOCOContext * const l = avctx->priv_data;
203  const uint8_t *buf = avpkt->data;
204  int buf_size = avpkt->size;
205  AVFrame * const p = data;
206  int decoded, ret;
207 
208  if ((ret = ff_get_buffer(avctx, p, 0)) < 0)
209  return ret;
210  p->key_frame = 1;
211 
212 #define ADVANCE_BY_DECODED do { \
213  if (decoded < 0 || decoded >= buf_size) goto buf_too_small; \
214  buf += decoded; buf_size -= decoded; \
215 } while(0)
216  switch(l->mode) {
217  case LOCO_CYUY2: case LOCO_YUY2: case LOCO_UYVY:
218  decoded = loco_decode_plane(l, p->data[0], avctx->width, avctx->height,
219  p->linesize[0], buf, buf_size);
221  decoded = loco_decode_plane(l, p->data[1], avctx->width / 2, avctx->height,
222  p->linesize[1], buf, buf_size);
224  decoded = loco_decode_plane(l, p->data[2], avctx->width / 2, avctx->height,
225  p->linesize[2], buf, buf_size);
226  break;
227  case LOCO_CYV12: case LOCO_YV12:
228  decoded = loco_decode_plane(l, p->data[0], avctx->width, avctx->height,
229  p->linesize[0], buf, buf_size);
231  decoded = loco_decode_plane(l, p->data[2], avctx->width / 2, avctx->height / 2,
232  p->linesize[2], buf, buf_size);
234  decoded = loco_decode_plane(l, p->data[1], avctx->width / 2, avctx->height / 2,
235  p->linesize[1], buf, buf_size);
236  break;
237  case LOCO_CRGB: case LOCO_RGB:
238  decoded = loco_decode_plane(l, p->data[1] + p->linesize[1]*(avctx->height-1), avctx->width, avctx->height,
239  -p->linesize[1], buf, buf_size);
241  decoded = loco_decode_plane(l, p->data[0] + p->linesize[0]*(avctx->height-1), avctx->width, avctx->height,
242  -p->linesize[0], buf, buf_size);
244  decoded = loco_decode_plane(l, p->data[2] + p->linesize[2]*(avctx->height-1), avctx->width, avctx->height,
245  -p->linesize[2], buf, buf_size);
246  if (avctx->width & 1) {
247  rotate_faulty_loco(p->data[0] + p->linesize[0]*(avctx->height-1), avctx->width, avctx->height, -p->linesize[0]);
248  rotate_faulty_loco(p->data[1] + p->linesize[1]*(avctx->height-1), avctx->width, avctx->height, -p->linesize[1]);
249  rotate_faulty_loco(p->data[2] + p->linesize[2]*(avctx->height-1), avctx->width, avctx->height, -p->linesize[2]);
250  }
251  break;
252  case LOCO_CRGBA:
253  case LOCO_RGBA:
254  decoded = loco_decode_plane(l, p->data[1] + p->linesize[1]*(avctx->height-1), avctx->width, avctx->height,
255  -p->linesize[1], buf, buf_size);
257  decoded = loco_decode_plane(l, p->data[0] + p->linesize[0]*(avctx->height-1), avctx->width, avctx->height,
258  -p->linesize[0], buf, buf_size);
260  decoded = loco_decode_plane(l, p->data[2] + p->linesize[2]*(avctx->height-1), avctx->width, avctx->height,
261  -p->linesize[2], buf, buf_size);
263  decoded = loco_decode_plane(l, p->data[3] + p->linesize[3]*(avctx->height-1), avctx->width, avctx->height,
264  -p->linesize[3], buf, buf_size);
265  break;
266  default:
267  av_assert0(0);
268  }
269 
270  if (decoded < 0 || decoded > buf_size)
271  goto buf_too_small;
272  buf_size -= decoded;
273 
274  *got_frame = 1;
275 
276  return avpkt->size - buf_size;
277 buf_too_small:
278  av_log(avctx, AV_LOG_ERROR, "Input data too small.\n");
279  return AVERROR(EINVAL);
280 }
281 
283 {
284  LOCOContext * const l = avctx->priv_data;
285  int version;
286 
287  l->avctx = avctx;
288  if (avctx->extradata_size < 12) {
289  av_log(avctx, AV_LOG_ERROR, "Extradata size must be >= 12 instead of %i\n",
290  avctx->extradata_size);
291  return AVERROR_INVALIDDATA;
292  }
293  version = AV_RL32(avctx->extradata);
294  switch (version) {
295  case 1:
296  l->lossy = 0;
297  break;
298  case 2:
299  l->lossy = AV_RL32(avctx->extradata + 8);
300  break;
301  default:
302  l->lossy = AV_RL32(avctx->extradata + 8);
303  avpriv_request_sample(avctx, "LOCO codec version %i", version);
304  }
305 
306  if (l->lossy > 65536U) {
307  av_log(avctx, AV_LOG_ERROR, "lossy %i is too large\n", l->lossy);
308  return AVERROR_INVALIDDATA;
309  }
310 
311  l->mode = AV_RL32(avctx->extradata + 4);
312  switch (l->mode) {
313  case LOCO_CYUY2:
314  case LOCO_YUY2:
315  case LOCO_UYVY:
316  avctx->pix_fmt = AV_PIX_FMT_YUV422P;
317  break;
318  case LOCO_CRGB:
319  case LOCO_RGB:
320  avctx->pix_fmt = AV_PIX_FMT_GBRP;
321  break;
322  case LOCO_CYV12:
323  case LOCO_YV12:
324  avctx->pix_fmt = AV_PIX_FMT_YUV420P;
325  break;
326  case LOCO_CRGBA:
327  case LOCO_RGBA:
328  avctx->pix_fmt = AV_PIX_FMT_GBRAP;
329  break;
330  default:
331  av_log(avctx, AV_LOG_INFO, "Unknown colorspace, index = %i\n", l->mode);
332  return AVERROR_INVALIDDATA;
333  }
334  if (avctx->debug & FF_DEBUG_PICT_INFO)
335  av_log(avctx, AV_LOG_INFO, "lossy:%i, version:%i, mode: %i\n", l->lossy, version, l->mode);
336 
337  return 0;
338 }
339 
341  .name = "loco",
342  .long_name = NULL_IF_CONFIG_SMALL("LOCO"),
343  .type = AVMEDIA_TYPE_VIDEO,
344  .id = AV_CODEC_ID_LOCO,
345  .priv_data_size = sizeof(LOCOContext),
346  .init = decode_init,
347  .decode = decode_frame,
348  .capabilities = AV_CODEC_CAP_DR1,
349 };
const char const char void * val
Definition: avisynth_c.h:863
#define AVERROR_INVALIDDATA
Invalid data found when processing input.
Definition: error.h:59
This structure describes decoded (raw) audio or video data.
Definition: frame.h:295
ptrdiff_t const GLvoid * data
Definition: opengl_enc.c:100
Definition: loco.c:41
static av_cold int init(AVCodecContext *avctx)
Definition: avrndec.c:35
#define avpriv_request_sample(...)
planar GBR 4:4:4 24bpp
Definition: pixfmt.h:168
int size
Definition: avcodec.h:1481
The reader does not expect b to be semantically here and if the code is changed by maybe adding a a division or other the signedness will almost certainly be mistaken To avoid this confusion a new type was SUINT is the C unsigned type but it holds a signed int to use the same example SUINT a
Definition: undefined.txt:36
enum AVPixelFormat pix_fmt
Pixel format, see AV_PIX_FMT_xxx.
Definition: avcodec.h:1778
int version
Definition: avisynth_c.h:858
static int get_ur_golomb_jpegls(GetBitContext *gb, int k, int limit, int esc_len)
read unsigned golomb rice code (jpegls).
Definition: golomb.h:428
uint8_t run
Definition: svq3.c:206
AVCodec.
Definition: avcodec.h:3492
Definition: loco.c:43
int run2
Definition: loco.c:54
static void decode(AVCodecContext *dec_ctx, AVPacket *pkt, AVFrame *frame, FILE *outfile)
Definition: decode_audio.c:71
#define av_assert0(cond)
assert() equivalent, that is always enabled.
Definition: avassert.h:37
static void rotate_faulty_loco(uint8_t *data, int width, int height, int stride)
Definition: loco.c:182
static int loco_get_rice_param(RICEContext *r)
Definition: loco.c:59
uint8_t
#define av_cold
Definition: attributes.h:82
static int loco_decode_plane(LOCOContext *l, uint8_t *data, int width, int height, int stride, const uint8_t *buf, int buf_size)
Definition: loco.c:130
#define FF_DEBUG_PICT_INFO
Definition: avcodec.h:2654
Undefined Behavior In the C some operations are like signed integer dereferencing freed accessing outside allocated Undefined Behavior must not occur in a C it is not safe even if the output of undefined operations is unused The unsafety may seem nit picking but Optimizing compilers have in fact optimized code on the assumption that no undefined Behavior occurs Optimizing code based on wrong assumptions can and has in some cases lead to effects beyond the output of computations The signed integer overflow problem in speed critical code Code which is highly optimized and works with signed integers sometimes has the problem that often the output of the computation does not c
Definition: undefined.txt:32
uint8_t * extradata
some codecs need / can use extradata like Huffman tables.
Definition: avcodec.h:1669
#define height
uint8_t * data
Definition: avcodec.h:1480
static int get_bits_count(const GetBitContext *s)
Definition: get_bits.h:219
bitstream reader API header.
int run
Definition: loco.c:54
#define av_log(a,...)
#define U(x)
Definition: vp56_arith.h:37
static int get_bits_left(GetBitContext *gb)
Definition: get_bits.h:849
#define i(width, name, range_min, range_max)
Definition: cbs_h2645.c:259
#define AV_LOG_ERROR
Something went wrong and cannot losslessly be recovered.
Definition: log.h:176
#define NULL_IF_CONFIG_SMALL(x)
Return NULL if CONFIG_SMALL is true, otherwise the argument without modification. ...
Definition: internal.h:186
const char * r
Definition: vf_curves.c:114
int count
Definition: loco.c:55
const char * name
Name of the codec implementation.
Definition: avcodec.h:3499
static int loco_get_rice(RICEContext *r)
Definition: loco.c:83
GLsizei count
Definition: opengl_enc.c:108
planar YUV 4:2:2, 16bpp, (1 Cr & Cb sample per 2x1 Y samples)
Definition: pixfmt.h:70
int save
Definition: loco.c:54
int lossy
Definition: loco.c:48
static void loco_update_rice_param(RICEContext *r, int val)
Definition: loco.c:72
#define b
Definition: input.c:41
static av_cold int decode_init(AVCodecContext *avctx)
Definition: loco.c:282
#define width
int width
picture width / height.
Definition: avcodec.h:1741
LOCO_MODE
Definition: loco.c:33
GetBitContext gb
Definition: loco.c:53
AVCodecContext * avctx
Definition: loco.c:47
int lossy
Definition: loco.c:56
Definition: loco.c:36
int sum
Definition: loco.c:55
#define AV_LOG_INFO
Standard information.
Definition: log.h:187
Libavcodec external API header.
Definition: loco.c:40
int linesize[AV_NUM_DATA_POINTERS]
For video, size in bytes of each picture line.
Definition: frame.h:326
static int init_get_bits8(GetBitContext *s, const uint8_t *buffer, int byte_size)
Initialize GetBitContext.
Definition: get_bits.h:677
int debug
debug
Definition: avcodec.h:2653
main external API structure.
Definition: avcodec.h:1568
int ff_get_buffer(AVCodecContext *avctx, AVFrame *frame, int flags)
Get a buffer for a frame.
Definition: decode.c:1968
void * buf
Definition: avisynth_c.h:766
int extradata_size
Definition: avcodec.h:1670
Definition: loco.c:42
#define mid_pred
Definition: mathops.h:97
#define ADVANCE_BY_DECODED
enum LOCO_MODE mode
Definition: loco.c:49
AVCodec ff_loco_decoder
Definition: loco.c:340
uint8_t * data[AV_NUM_DATA_POINTERS]
pointer to the picture/channel planes.
Definition: frame.h:309
GLint GLenum GLboolean GLsizei stride
Definition: opengl_enc.c:104
planar YUV 4:2:0, 12bpp, (1 Cr & Cb sample per 2x2 Y samples)
Definition: pixfmt.h:66
common internal api header.
planar GBRA 4:4:4:4 32bpp
Definition: pixfmt.h:215
static int loco_predict(uint8_t *data, int stride)
Definition: loco.c:119
void * priv_data
Definition: avcodec.h:1595
Definition: loco.c:39
int key_frame
1 -> keyframe, 0-> not
Definition: frame.h:373
Definition: loco.c:34
#define stride
Filter the word “frame” indicates either a video frame or a group of audio as stored in an AVFrame structure Format for each input and each output the list of supported formats For video that means pixel format For audio that means channel sample they are references to shared objects When the negotiation mechanism computes the intersection of the formats supported at each end of a all references to both lists are replaced with a reference to the intersection And when a single format is eventually chosen for a link amongst the remaining all references to the list are updated That means that if a filter requires that its input and output have the same format amongst a supported all it has to do is use a reference to the same list of formats query_formats can leave some formats unset and return AVERROR(EAGAIN) to cause the negotiation mechanism toagain later.That can be used by filters with complex requirements to use the format negotiated on one link to set the formats supported on another.Frame references ownership and permissions
static int decode_frame(AVCodecContext *avctx, void *data, int *got_frame, AVPacket *avpkt)
Definition: loco.c:198
exp golomb vlc stuff
uint64_t_TMPL AV_WL64 unsigned int_TMPL AV_RL32
Definition: bytestream.h:87
This structure stores compressed data.
Definition: avcodec.h:1457
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() for allocating buffers and supports custom allocators.
Definition: avcodec.h:984