snow.h 2.39 KB
Newer Older
1 2 3 4
/*
 * Copyright (C) 2004 Michael Niedermayer <michaelni@gmx.at>
 * Copyright (C) 2006 Robert Edele <yartrebo@earthlink.net>
 *
5 6 7
 * This file is part of FFmpeg.
 *
 * FFmpeg is free software; you can redistribute it and/or
8 9
 * modify it under the terms of the GNU Lesser General Public
 * 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 14 15 16 17
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * 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 20 21
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
 */

22 23
#ifndef AVCODEC_SNOW_H
#define AVCODEC_SNOW_H
24 25

#include "dsputil.h"
26
#include "dwt.h"
27 28 29 30 31 32 33

#define MID_STATE 128

#define MAX_PLANES 4
#define QSHIFT 5
#define QROOT (1<<QSHIFT)
#define LOSSLESS_QLOG -128
34
#define FRAC_BITS 4
35
#define MAX_REF_FRAMES 8
36

Robert Edele's avatar
Robert Edele committed
37
#define LOG2_OBMC_MAX 8
38 39
#define OBMC_MAX (1<<(LOG2_OBMC_MAX))

Luca Barbato's avatar
Luca Barbato committed
40 41
/* C bits used by mmx/sse2/altivec */

42
static av_always_inline void snow_interleave_line_header(int * i, int width, IDWTELEM * low, IDWTELEM * high){
Luca Barbato's avatar
Luca Barbato committed
43 44 45 46 47 48 49 50
    (*i) = (width) - 2;

    if (width & 1){
        low[(*i)+1] = low[((*i)+1)>>1];
        (*i)--;
    }
}

51
static av_always_inline void snow_interleave_line_footer(int * i, IDWTELEM * low, IDWTELEM * high){
Luca Barbato's avatar
Luca Barbato committed
52 53 54 55 56 57
    for (; (*i)>=0; (*i)-=2){
        low[(*i)+1] = high[(*i)>>1];
        low[*i] = low[(*i)>>1];
    }
}

58
static av_always_inline void snow_horizontal_compose_lift_lead_out(int i, IDWTELEM * dst, IDWTELEM * src, IDWTELEM * ref, int width, int w, int lift_high, int mul, int add, int shift){
Luca Barbato's avatar
Luca Barbato committed
59 60 61 62 63 64 65 66 67
    for(; i<w; i++){
        dst[i] = src[i] - ((mul * (ref[i] + ref[i + 1]) + add) >> shift);
    }

    if((width^lift_high)&1){
        dst[w] = src[w] - ((mul * 2 * ref[w] + add) >> shift);
    }
}

68
static av_always_inline void snow_horizontal_compose_liftS_lead_out(int i, IDWTELEM * dst, IDWTELEM * src, IDWTELEM * ref, int width, int w){
Luca Barbato's avatar
Luca Barbato committed
69
        for(; i<w; i++){
70
            dst[i] = src[i] + ((ref[i] + ref[(i+1)]+W_BO + 4 * src[i]) >> W_BS);
Luca Barbato's avatar
Luca Barbato committed
71 72 73
        }

        if(width&1){
74
            dst[w] = src[w] + ((2 * ref[w] + W_BO + 4 * src[w]) >> W_BS);
Luca Barbato's avatar
Luca Barbato committed
75 76 77
        }
}

78
#endif /* AVCODEC_SNOW_H */