Commit d85b37a9 authored by Alexandra Hájková's avatar Alexandra Hájková Committed by Diego Biurrun

loco: Convert to the new bitstream reader

parent 0f94de8a
...@@ -25,8 +25,8 @@ ...@@ -25,8 +25,8 @@
*/ */
#include "avcodec.h" #include "avcodec.h"
#include "get_bits.h" #include "bitstream.h"
#include "golomb_legacy.h" #include "golomb.h"
#include "internal.h" #include "internal.h"
#include "mathops.h" #include "mathops.h"
...@@ -50,7 +50,7 @@ typedef struct LOCOContext { ...@@ -50,7 +50,7 @@ typedef struct LOCOContext {
} LOCOContext; } LOCOContext;
typedef struct RICEContext { typedef struct RICEContext {
GetBitContext gb; BitstreamContext bc;
int save, run, run2; /* internal rice decoder state */ int save, run, run2; /* internal rice decoder state */
int sum, count; /* sum and count for getting rice parameter */ int sum, count; /* sum and count for getting rice parameter */
int lossy; int lossy;
...@@ -88,11 +88,11 @@ static inline int loco_get_rice(RICEContext *r) ...@@ -88,11 +88,11 @@ static inline int loco_get_rice(RICEContext *r)
loco_update_rice_param(r, 0); loco_update_rice_param(r, 0);
return 0; return 0;
} }
v = get_ur_golomb_jpegls(&r->gb, loco_get_rice_param(r), INT_MAX, 0); v = get_ur_golomb_jpegls(&r->bc, loco_get_rice_param(r), INT_MAX, 0);
loco_update_rice_param(r, (v + 1) >> 1); loco_update_rice_param(r, (v + 1) >> 1);
if (!v) { if (!v) {
if (r->save >= 0) { if (r->save >= 0) {
r->run = get_ur_golomb_jpegls(&r->gb, 2, INT_MAX, 0); r->run = get_ur_golomb_jpegls(&r->bc, 2, INT_MAX, 0);
if (r->run > 1) if (r->run > 1)
r->save += r->run + 1; r->save += r->run + 1;
else else
...@@ -132,7 +132,7 @@ static int loco_decode_plane(LOCOContext *l, uint8_t *data, int width, int heigh ...@@ -132,7 +132,7 @@ static int loco_decode_plane(LOCOContext *l, uint8_t *data, int width, int heigh
int val; int val;
int i, j; int i, j;
init_get_bits(&rc.gb, buf, buf_size*8); bitstream_init8(&rc.bc, buf, buf_size);
rc.save = 0; rc.save = 0;
rc.run = 0; rc.run = 0;
rc.run2 = 0; rc.run2 = 0;
...@@ -162,7 +162,7 @@ static int loco_decode_plane(LOCOContext *l, uint8_t *data, int width, int heigh ...@@ -162,7 +162,7 @@ static int loco_decode_plane(LOCOContext *l, uint8_t *data, int width, int heigh
data += stride; data += stride;
} }
return (get_bits_count(&rc.gb) + 7) >> 3; return (bitstream_tell(&rc.bc) + 7) >> 3;
} }
static int decode_frame(AVCodecContext *avctx, static int decode_frame(AVCodecContext *avctx,
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment