mirror of
https://github.com/FFmpeg/FFmpeg.git
synced 2024-12-23 12:43:46 +02:00
avcodec/snow: Move decoder/encoder-only inline funcs to snow(dec|enc).c
Reviewed-by: Michael Niedermayer <michael@niedermayer.cc> Signed-off-by: Andreas Rheinhardt <andreas.rheinhardt@outlook.com>
This commit is contained in:
Andreas Rheinhardt
parent
10dad7ccc2
commit
abaf02a7e7
@ -485,227 +485,8 @@ static inline void set_blocks(SnowContext *s, int level, int x, int y, int l, in
|
||||
}
|
||||
}
|
||||
|
||||
static inline void init_ref(MotionEstContext *c, const uint8_t *const src[3], uint8_t *const ref[3], uint8_t *const ref2[3], int x, int y, int ref_index){
|
||||
SnowContext *s = c->avctx->priv_data;
|
||||
const int offset[3]= {
|
||||
y*c-> stride + x,
|
||||
((y*c->uvstride + x)>>s->chroma_h_shift),
|
||||
((y*c->uvstride + x)>>s->chroma_h_shift),
|
||||
};
|
||||
int i;
|
||||
for(i=0; i<3; i++){
|
||||
c->src[0][i]= src [i];
|
||||
c->ref[0][i]= ref [i] + offset[i];
|
||||
}
|
||||
av_assert2(!ref_index);
|
||||
}
|
||||
|
||||
|
||||
/* bitstream functions */
|
||||
|
||||
extern const int8_t ff_quant3bA[256];
|
||||
|
||||
#define QEXPSHIFT (7-FRAC_BITS+8) //FIXME try to change this to 0
|
||||
|
||||
static inline void put_symbol(RangeCoder *c, uint8_t *state, int v, int is_signed){
|
||||
int i;
|
||||
|
||||
if(v){
|
||||
const int a= FFABS(v);
|
||||
const int e= av_log2(a);
|
||||
const int el= FFMIN(e, 10);
|
||||
put_rac(c, state+0, 0);
|
||||
|
||||
for(i=0; i<el; i++){
|
||||
put_rac(c, state+1+i, 1); //1..10
|
||||
}
|
||||
for(; i<e; i++){
|
||||
put_rac(c, state+1+9, 1); //1..10
|
||||
}
|
||||
put_rac(c, state+1+FFMIN(i,9), 0);
|
||||
|
||||
for(i=e-1; i>=el; i--){
|
||||
put_rac(c, state+22+9, (a>>i)&1); //22..31
|
||||
}
|
||||
for(; i>=0; i--){
|
||||
put_rac(c, state+22+i, (a>>i)&1); //22..31
|
||||
}
|
||||
|
||||
if(is_signed)
|
||||
put_rac(c, state+11 + el, v < 0); //11..21
|
||||
}else{
|
||||
put_rac(c, state+0, 1);
|
||||
}
|
||||
}
|
||||
|
||||
static inline int get_symbol(RangeCoder *c, uint8_t *state, int is_signed){
|
||||
if(get_rac(c, state+0))
|
||||
return 0;
|
||||
else{
|
||||
int i, e;
|
||||
unsigned a;
|
||||
e= 0;
|
||||
while(get_rac(c, state+1 + FFMIN(e,9))){ //1..10
|
||||
e++;
|
||||
if (e > 31)
|
||||
return AVERROR_INVALIDDATA;
|
||||
}
|
||||
|
||||
a= 1;
|
||||
for(i=e-1; i>=0; i--){
|
||||
a += a + get_rac(c, state+22 + FFMIN(i,9)); //22..31
|
||||
}
|
||||
|
||||
e= -(is_signed && get_rac(c, state+11 + FFMIN(e,10))); //11..21
|
||||
return (a^e)-e;
|
||||
}
|
||||
}
|
||||
|
||||
static inline void put_symbol2(RangeCoder *c, uint8_t *state, int v, int log2){
|
||||
int i;
|
||||
int r= log2>=0 ? 1<<log2 : 1;
|
||||
|
||||
av_assert2(v>=0);
|
||||
av_assert2(log2>=-4);
|
||||
|
||||
while(v >= r){
|
||||
put_rac(c, state+4+log2, 1);
|
||||
v -= r;
|
||||
log2++;
|
||||
if(log2>0) r+=r;
|
||||
}
|
||||
put_rac(c, state+4+log2, 0);
|
||||
|
||||
for(i=log2-1; i>=0; i--){
|
||||
put_rac(c, state+31-i, (v>>i)&1);
|
||||
}
|
||||
}
|
||||
|
||||
static inline int get_symbol2(RangeCoder *c, uint8_t *state, int log2){
|
||||
int i;
|
||||
int r= log2>=0 ? 1<<log2 : 1;
|
||||
int v=0;
|
||||
|
||||
av_assert2(log2>=-4);
|
||||
|
||||
while(log2<28 && get_rac(c, state+4+log2)){
|
||||
v+= r;
|
||||
log2++;
|
||||
if(log2>0) r+=r;
|
||||
}
|
||||
|
||||
for(i=log2-1; i>=0; i--){
|
||||
v+= get_rac(c, state+31-i)<<i;
|
||||
}
|
||||
|
||||
return v;
|
||||
}
|
||||
|
||||
static inline void unpack_coeffs(SnowContext *s, SubBand *b, SubBand * parent, int orientation){
|
||||
const int w= b->width;
|
||||
const int h= b->height;
|
||||
int x,y;
|
||||
|
||||
int run, runs;
|
||||
x_and_coeff *xc= b->x_coeff;
|
||||
x_and_coeff *prev_xc= NULL;
|
||||
x_and_coeff *prev2_xc= xc;
|
||||
x_and_coeff *parent_xc= parent ? parent->x_coeff : NULL;
|
||||
x_and_coeff *prev_parent_xc= parent_xc;
|
||||
|
||||
runs= get_symbol2(&s->c, b->state[30], 0);
|
||||
if(runs-- > 0) run= get_symbol2(&s->c, b->state[1], 3);
|
||||
else run= INT_MAX;
|
||||
|
||||
for(y=0; y<h; y++){
|
||||
int v=0;
|
||||
int lt=0, t=0, rt=0;
|
||||
|
||||
if(y && prev_xc->x == 0){
|
||||
rt= prev_xc->coeff;
|
||||
}
|
||||
for(x=0; x<w; x++){
|
||||
int p=0;
|
||||
const int l= v;
|
||||
|
||||
lt= t; t= rt;
|
||||
|
||||
if(y){
|
||||
if(prev_xc->x <= x)
|
||||
prev_xc++;
|
||||
if(prev_xc->x == x + 1)
|
||||
rt= prev_xc->coeff;
|
||||
else
|
||||
rt=0;
|
||||
}
|
||||
if(parent_xc){
|
||||
if(x>>1 > parent_xc->x){
|
||||
parent_xc++;
|
||||
}
|
||||
if(x>>1 == parent_xc->x){
|
||||
p= parent_xc->coeff;
|
||||
}
|
||||
}
|
||||
if(/*ll|*/l|lt|t|rt|p){
|
||||
int context= av_log2(/*FFABS(ll) + */3*(l>>1) + (lt>>1) + (t&~1) + (rt>>1) + (p>>1));
|
||||
|
||||
v=get_rac(&s->c, &b->state[0][context]);
|
||||
if(v){
|
||||
v= 2*(get_symbol2(&s->c, b->state[context + 2], context-4) + 1);
|
||||
v+=get_rac(&s->c, &b->state[0][16 + 1 + 3 + ff_quant3bA[l&0xFF] + 3*ff_quant3bA[t&0xFF]]);
|
||||
if ((uint16_t)v != v) {
|
||||
av_log(s->avctx, AV_LOG_ERROR, "Coefficient damaged\n");
|
||||
v = 1;
|
||||
}
|
||||
xc->x=x;
|
||||
(xc++)->coeff= v;
|
||||
}
|
||||
}else{
|
||||
if(!run){
|
||||
if(runs-- > 0) run= get_symbol2(&s->c, b->state[1], 3);
|
||||
else run= INT_MAX;
|
||||
v= 2*(get_symbol2(&s->c, b->state[0 + 2], 0-4) + 1);
|
||||
v+=get_rac(&s->c, &b->state[0][16 + 1 + 3]);
|
||||
if ((uint16_t)v != v) {
|
||||
av_log(s->avctx, AV_LOG_ERROR, "Coefficient damaged\n");
|
||||
v = 1;
|
||||
}
|
||||
|
||||
xc->x=x;
|
||||
(xc++)->coeff= v;
|
||||
}else{
|
||||
int max_run;
|
||||
run--;
|
||||
v=0;
|
||||
av_assert2(run >= 0);
|
||||
if(y) max_run= FFMIN(run, prev_xc->x - x - 2);
|
||||
else max_run= FFMIN(run, w-x-1);
|
||||
if(parent_xc)
|
||||
max_run= FFMIN(max_run, 2*parent_xc->x - x - 1);
|
||||
av_assert2(max_run >= 0 && max_run <= run);
|
||||
|
||||
x+= max_run;
|
||||
run-= max_run;
|
||||
}
|
||||
}
|
||||
}
|
||||
(xc++)->x= w+1; //end marker
|
||||
prev_xc= prev2_xc;
|
||||
prev2_xc= xc;
|
||||
|
||||
if(parent_xc){
|
||||
if(y&1){
|
||||
while(parent_xc->x != parent->width+1)
|
||||
parent_xc++;
|
||||
parent_xc++;
|
||||
prev_parent_xc= parent_xc;
|
||||
}else{
|
||||
parent_xc= prev_parent_xc;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
(xc++)->x= w+1; //end marker
|
||||
}
|
||||
|
||||
#endif /* AVCODEC_SNOW_H */
|
||||
|
||||
@ -31,6 +31,153 @@
|
||||
#include "rangecoder.h"
|
||||
#include "mathops.h"
|
||||
|
||||
static inline int get_symbol(RangeCoder *c, uint8_t *state, int is_signed)
|
||||
{
|
||||
if (get_rac(c, state + 0))
|
||||
return 0;
|
||||
else {
|
||||
int e;
|
||||
unsigned a;
|
||||
e = 0;
|
||||
while (get_rac(c, state + 1 + FFMIN(e, 9))) { //1..10
|
||||
e++;
|
||||
if (e > 31)
|
||||
return AVERROR_INVALIDDATA;
|
||||
}
|
||||
|
||||
a = 1;
|
||||
for (int i = e - 1; i >= 0; i--)
|
||||
a += a + get_rac(c, state + 22 + FFMIN(i, 9)); //22..31
|
||||
|
||||
e = -(is_signed && get_rac(c, state + 11 + FFMIN(e, 10))); //11..21
|
||||
return (a ^ e) - e;
|
||||
}
|
||||
}
|
||||
|
||||
static inline int get_symbol2(RangeCoder *c, uint8_t *state, int log2)
|
||||
{
|
||||
int r = log2 >= 0 ? 1 << log2 : 1;
|
||||
int v = 0;
|
||||
|
||||
av_assert2(log2 >= -4);
|
||||
|
||||
while (log2 < 28 && get_rac(c, state + 4 + log2)) {
|
||||
v += r;
|
||||
log2++;
|
||||
if (log2 > 0) r += r;
|
||||
}
|
||||
|
||||
for (int i = log2 - 1; i >= 0; i--)
|
||||
v += get_rac(c, state + 31 - i) << i;
|
||||
|
||||
return v;
|
||||
}
|
||||
|
||||
static void unpack_coeffs(SnowContext *s, SubBand *b, SubBand * parent, int orientation)
|
||||
{
|
||||
const int w = b->width;
|
||||
const int h = b->height;
|
||||
|
||||
int run, runs;
|
||||
x_and_coeff *xc = b->x_coeff;
|
||||
x_and_coeff *prev_xc = NULL;
|
||||
x_and_coeff *prev2_xc = xc;
|
||||
x_and_coeff *parent_xc = parent ? parent->x_coeff : NULL;
|
||||
x_and_coeff *prev_parent_xc = parent_xc;
|
||||
|
||||
runs = get_symbol2(&s->c, b->state[30], 0);
|
||||
if (runs-- > 0) run = get_symbol2(&s->c, b->state[1], 3);
|
||||
else run = INT_MAX;
|
||||
|
||||
for (int y = 0; y < h; y++) {
|
||||
int v = 0;
|
||||
int lt = 0, t = 0, rt = 0;
|
||||
|
||||
if (y && prev_xc->x == 0)
|
||||
rt = prev_xc->coeff;
|
||||
|
||||
for (int x = 0; x < w; x++) {
|
||||
int p = 0;
|
||||
const int l = v;
|
||||
|
||||
lt= t; t= rt;
|
||||
|
||||
if (y) {
|
||||
if (prev_xc->x <= x)
|
||||
prev_xc++;
|
||||
if (prev_xc->x == x + 1)
|
||||
rt = prev_xc->coeff;
|
||||
else
|
||||
rt = 0;
|
||||
}
|
||||
if (parent_xc) {
|
||||
if (x>>1 > parent_xc->x)
|
||||
parent_xc++;
|
||||
if (x>>1 == parent_xc->x)
|
||||
p = parent_xc->coeff;
|
||||
}
|
||||
if (/*ll|*/l|lt|t|rt|p) {
|
||||
int context = av_log2(/*FFABS(ll) + */3*(l>>1) + (lt>>1) + (t&~1) + (rt>>1) + (p>>1));
|
||||
|
||||
v = get_rac(&s->c, &b->state[0][context]);
|
||||
if (v) {
|
||||
v = 2*(get_symbol2(&s->c, b->state[context + 2], context-4) + 1);
|
||||
v += get_rac(&s->c, &b->state[0][16 + 1 + 3 + ff_quant3bA[l&0xFF] + 3 * ff_quant3bA[t&0xFF]]);
|
||||
if ((uint16_t)v != v) {
|
||||
av_log(s->avctx, AV_LOG_ERROR, "Coefficient damaged\n");
|
||||
v = 1;
|
||||
}
|
||||
xc->x = x;
|
||||
(xc++)->coeff = v;
|
||||
}
|
||||
} else {
|
||||
if (!run) {
|
||||
if (runs-- > 0) run = get_symbol2(&s->c, b->state[1], 3);
|
||||
else run = INT_MAX;
|
||||
v = 2 * (get_symbol2(&s->c, b->state[0 + 2], 0-4) + 1);
|
||||
v += get_rac(&s->c, &b->state[0][16 + 1 + 3]);
|
||||
if ((uint16_t)v != v) {
|
||||
av_log(s->avctx, AV_LOG_ERROR, "Coefficient damaged\n");
|
||||
v = 1;
|
||||
}
|
||||
|
||||
xc->x = x;
|
||||
(xc++)->coeff = v;
|
||||
} else {
|
||||
int max_run;
|
||||
run--;
|
||||
v = 0;
|
||||
av_assert2(run >= 0);
|
||||
if (y) max_run = FFMIN(run, prev_xc->x - x - 2);
|
||||
else max_run = FFMIN(run, w-x-1);
|
||||
if (parent_xc)
|
||||
max_run = FFMIN(max_run, 2*parent_xc->x - x - 1);
|
||||
av_assert2(max_run >= 0 && max_run <= run);
|
||||
|
||||
x += max_run;
|
||||
run -= max_run;
|
||||
}
|
||||
}
|
||||
}
|
||||
(xc++)->x = w+1; //end marker
|
||||
prev_xc = prev2_xc;
|
||||
prev2_xc = xc;
|
||||
|
||||
if (parent_xc) {
|
||||
if (y & 1) {
|
||||
while (parent_xc->x != parent->width+1)
|
||||
parent_xc++;
|
||||
parent_xc++;
|
||||
prev_parent_xc= parent_xc;
|
||||
} else {
|
||||
parent_xc= prev_parent_xc;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
(xc++)->x = w + 1; //end marker
|
||||
}
|
||||
|
||||
static av_always_inline void predict_slice_buffered(SnowContext *s, slice_buffer * sb, IDWTELEM * old_buffer, int plane_index, int add, int mb_y){
|
||||
Plane *p= &s->plane[plane_index];
|
||||
const int mb_w= s->b_width << s->block_max_depth;
|
||||
|
||||
@ -39,6 +39,70 @@
|
||||
#include "mpegvideo.h"
|
||||
#include "h263enc.h"
|
||||
|
||||
static void init_ref(MotionEstContext *c, const uint8_t *const src[3],
|
||||
uint8_t *const ref[3], uint8_t *const ref2[3],
|
||||
int x, int y, int ref_index)
|
||||
{
|
||||
SnowContext *s = c->avctx->priv_data;
|
||||
const int offset[3] = {
|
||||
y*c-> stride + x,
|
||||
((y*c->uvstride + x) >> s->chroma_h_shift),
|
||||
((y*c->uvstride + x) >> s->chroma_h_shift),
|
||||
};
|
||||
for (int i = 0; i < 3; i++) {
|
||||
c->src[0][i] = src [i];
|
||||
c->ref[0][i] = ref [i] + offset[i];
|
||||
}
|
||||
av_assert2(!ref_index);
|
||||
}
|
||||
|
||||
static inline void put_symbol(RangeCoder *c, uint8_t *state, int v, int is_signed)
|
||||
{
|
||||
if (v) {
|
||||
const int a = FFABS(v);
|
||||
const int e = av_log2(a);
|
||||
const int el = FFMIN(e, 10);
|
||||
int i;
|
||||
|
||||
put_rac(c, state + 0, 0);
|
||||
|
||||
for (i = 0; i < el; i++)
|
||||
put_rac(c, state + 1 + i, 1); //1..10
|
||||
for(; i < e; i++)
|
||||
put_rac(c, state + 1 + 9, 1); //1..10
|
||||
put_rac(c, state + 1 + FFMIN(i, 9), 0);
|
||||
|
||||
for (i = e - 1; i >= el; i--)
|
||||
put_rac(c, state + 22 + 9, (a >> i) & 1); //22..31
|
||||
for(; i >= 0; i--)
|
||||
put_rac(c, state + 22 + i, (a >> i) & 1); //22..31
|
||||
|
||||
if (is_signed)
|
||||
put_rac(c, state + 11 + el, v < 0); //11..21
|
||||
} else {
|
||||
put_rac(c, state + 0, 1);
|
||||
}
|
||||
}
|
||||
|
||||
static inline void put_symbol2(RangeCoder *c, uint8_t *state, int v, int log2)
|
||||
{
|
||||
int r = log2 >= 0 ? 1<<log2 : 1;
|
||||
|
||||
av_assert2(v >= 0);
|
||||
av_assert2(log2 >= -4);
|
||||
|
||||
while (v >= r) {
|
||||
put_rac(c, state + 4 + log2, 1);
|
||||
v -= r;
|
||||
log2++;
|
||||
if (log2 > 0) r += r;
|
||||
}
|
||||
put_rac(c, state + 4 + log2, 0);
|
||||
|
||||
for (int i = log2 - 1; i >= 0; i--)
|
||||
put_rac(c, state + 31 - i, (v >> i) & 1);
|
||||
}
|
||||
|
||||
static int get_encode_buffer(SnowContext *s, AVFrame *frame)
|
||||
{
|
||||
int ret;
|
||||
|
||||
Loading…
Reference in New Issue
Block a user