simple_idct.c 5.86 KB
Newer Older
1
/*
2 3 4 5
 * Simple IDCT
 *
 * Copyright (c) 2001 Michael Niedermayer <michaelni@gmx.at>
 *
6
 * This file is part of Libav.
7
 *
8
 * Libav is free software; you can redistribute it and/or
9 10
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
11
 * version 2.1 of the License, or (at your option) any later version.
12
 *
13
 * Libav is distributed in the hope that it will be useful,
14 15 16 17 18
 * 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
19
 * License along with Libav; if not, write to the Free Software
20
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
21
 */
22

Michael Niedermayer's avatar
Michael Niedermayer committed
23
/**
24
 * @file
Michael Niedermayer's avatar
Michael Niedermayer committed
25 26
 * simpleidct in C.
 */
27

28
#include "libavutil/intreadwrite.h"
29
#include "avcodec.h"
30
#include "dsputil.h"
31
#include "mathops.h"
32 33
#include "simple_idct.h"

34 35 36
#define BIT_DEPTH 8
#include "simple_idct_template.c"
#undef BIT_DEPTH
37

38 39 40
#define BIT_DEPTH 10
#include "simple_idct_template.c"
#undef BIT_DEPTH
Michael Niedermayer's avatar
Michael Niedermayer committed
41

42 43 44 45
/* 2x4x8 idct */

#define CN_SHIFT 12
#define C_FIX(x) ((int)((x) * (1 << CN_SHIFT) + 0.5))
46 47
#define C1 C_FIX(0.6532814824)
#define C2 C_FIX(0.2705980501)
48

49 50 51
/* row idct is multiple by 16 * sqrt(2.0), col idct4 is normalized,
   and the butterfly must be multiplied by 0.5 * sqrt(2.0) */
#define C_SHIFT (4+1+12)
52

53
static inline void idct4col_put(uint8_t *dest, int line_size, const DCTELEM *col)
54 55
{
    int c0, c1, c2, c3, a0, a1, a2, a3;
56
    const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
57 58 59 60 61

    a0 = col[8*0];
    a1 = col[8*2];
    a2 = col[8*4];
    a3 = col[8*6];
62 63
    c0 = ((a0 + a2) << (CN_SHIFT - 1)) + (1 << (C_SHIFT - 1));
    c2 = ((a0 - a2) << (CN_SHIFT - 1)) + (1 << (C_SHIFT - 1));
64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
    c1 = a1 * C1 + a3 * C2;
    c3 = a1 * C2 - a3 * C1;
    dest[0] = cm[(c0 + c1) >> C_SHIFT];
    dest += line_size;
    dest[0] = cm[(c2 + c3) >> C_SHIFT];
    dest += line_size;
    dest[0] = cm[(c2 - c3) >> C_SHIFT];
    dest += line_size;
    dest[0] = cm[(c0 - c1) >> C_SHIFT];
}

#define BF(k) \
{\
    int a0, a1;\
    a0 = ptr[k];\
    a1 = ptr[8 + k];\
    ptr[k] = a0 + a1;\
    ptr[8 + k] = a0 - a1;\
}

/* only used by DV codec. The input must be interlaced. 128 is added
   to the pixels before clamping to avoid systematic error
   (1024*sqrt(2)) offset would be needed otherwise. */
/* XXX: I think a 1.0/sqrt(2) normalization should be needed to
   compensate the extra butterfly stage - I don't have the full DV
   specification */
90
void ff_simple_idct248_put(uint8_t *dest, int line_size, DCTELEM *block)
91 92
{
    int i;
93
    DCTELEM *ptr;
94

95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110
    /* butterfly */
    ptr = block;
    for(i=0;i<4;i++) {
        BF(0);
        BF(1);
        BF(2);
        BF(3);
        BF(4);
        BF(5);
        BF(6);
        BF(7);
        ptr += 2 * 8;
    }

    /* IDCT8 on each line */
    for(i=0; i<8; i++) {
111
        idctRowCondDC_8(block + i*8, 0);
112 113 114 115
    }

    /* IDCT4 and store */
    for(i=0;i<8;i++) {
116 117
        idct4col_put(dest + i, 2 * line_size, block + i);
        idct4col_put(dest + line_size + i, 2 * line_size, block + 8 + i);
118 119
    }
}
Michael Niedermayer's avatar
Michael Niedermayer committed
120 121 122 123 124 125 126 127 128 129 130 131 132

/* 8x4 & 4x8 WMV2 IDCT */
#undef CN_SHIFT
#undef C_SHIFT
#undef C_FIX
#undef C1
#undef C2
#define CN_SHIFT 12
#define C_FIX(x) ((int)((x) * 1.414213562 * (1 << CN_SHIFT) + 0.5))
#define C1 C_FIX(0.6532814824)
#define C2 C_FIX(0.2705980501)
#define C3 C_FIX(0.5)
#define C_SHIFT (4+1+12)
133
static inline void idct4col_add(uint8_t *dest, int line_size, const DCTELEM *col)
Michael Niedermayer's avatar
Michael Niedermayer committed
134 135
{
    int c0, c1, c2, c3, a0, a1, a2, a3;
136
    const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
Michael Niedermayer's avatar
Michael Niedermayer committed
137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160

    a0 = col[8*0];
    a1 = col[8*1];
    a2 = col[8*2];
    a3 = col[8*3];
    c0 = (a0 + a2)*C3 + (1 << (C_SHIFT - 1));
    c2 = (a0 - a2)*C3 + (1 << (C_SHIFT - 1));
    c1 = a1 * C1 + a3 * C2;
    c3 = a1 * C2 - a3 * C1;
    dest[0] = cm[dest[0] + ((c0 + c1) >> C_SHIFT)];
    dest += line_size;
    dest[0] = cm[dest[0] + ((c2 + c3) >> C_SHIFT)];
    dest += line_size;
    dest[0] = cm[dest[0] + ((c2 - c3) >> C_SHIFT)];
    dest += line_size;
    dest[0] = cm[dest[0] + ((c0 - c1) >> C_SHIFT)];
}

#define RN_SHIFT 15
#define R_FIX(x) ((int)((x) * 1.414213562 * (1 << RN_SHIFT) + 0.5))
#define R1 R_FIX(0.6532814824)
#define R2 R_FIX(0.2705980501)
#define R3 R_FIX(0.5)
#define R_SHIFT 11
161
static inline void idct4row(DCTELEM *row)
Michael Niedermayer's avatar
Michael Niedermayer committed
162 163
{
    int c0, c1, c2, c3, a0, a1, a2, a3;
164
    //const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
Michael Niedermayer's avatar
Michael Niedermayer committed
165 166 167 168 169 170 171 172 173 174 175 176 177 178 179

    a0 = row[0];
    a1 = row[1];
    a2 = row[2];
    a3 = row[3];
    c0 = (a0 + a2)*R3 + (1 << (R_SHIFT - 1));
    c2 = (a0 - a2)*R3 + (1 << (R_SHIFT - 1));
    c1 = a1 * R1 + a3 * R2;
    c3 = a1 * R2 - a3 * R1;
    row[0]= (c0 + c1) >> R_SHIFT;
    row[1]= (c2 + c3) >> R_SHIFT;
    row[2]= (c2 - c3) >> R_SHIFT;
    row[3]= (c0 - c1) >> R_SHIFT;
}

180
void ff_simple_idct84_add(uint8_t *dest, int line_size, DCTELEM *block)
Michael Niedermayer's avatar
Michael Niedermayer committed
181 182 183 184 185
{
    int i;

    /* IDCT8 on each line */
    for(i=0; i<4; i++) {
186
        idctRowCondDC_8(block + i*8, 0);
Michael Niedermayer's avatar
Michael Niedermayer committed
187 188 189 190 191 192 193 194
    }

    /* IDCT4 and store */
    for(i=0;i<8;i++) {
        idct4col_add(dest + i, line_size, block + i);
    }
}

195
void ff_simple_idct48_add(uint8_t *dest, int line_size, DCTELEM *block)
Michael Niedermayer's avatar
Michael Niedermayer committed
196 197 198 199 200 201 202 203 204 205
{
    int i;

    /* IDCT4 on each line */
    for(i=0; i<8; i++) {
        idct4row(block + i*8);
    }

    /* IDCT8 and store */
    for(i=0; i<4; i++){
206
        idctSparseColAdd_8(dest + i, line_size, block + i);
Michael Niedermayer's avatar
Michael Niedermayer committed
207 208 209
    }
}

Kostya Shishkov's avatar
Kostya Shishkov committed
210 211 212 213 214 215 216 217 218 219 220 221 222 223
void ff_simple_idct44_add(uint8_t *dest, int line_size, DCTELEM *block)
{
    int i;

    /* IDCT4 on each line */
    for(i=0; i<4; i++) {
        idct4row(block + i*8);
    }

    /* IDCT4 and store */
    for(i=0; i<4; i++){
        idct4col_add(dest + i, line_size, block + i);
    }
}
224 225 226 227 228 229 230 231 232

void ff_prores_idct(DCTELEM *block, const int16_t *qmat)
{
    int i;

    for (i = 0; i < 64; i++)
        block[i] *= qmat[i];

    for (i = 0; i < 8; i++)
233
        idctRowCondDC_10(block + i*8, 2);
234 235 236 237

    for (i = 0; i < 8; i++)
        idctSparseCol_10(block + i);
}