Mercurial > libavcodec.hg
annotate dct.c @ 11517:e3b680f6c106 libavcodec
Cosmetics: move ff_dct_init() to the bottom of the file
author | vitor |
---|---|
date | Sun, 21 Mar 2010 11:28:38 +0000 |
parents | 4b3da727d832 |
children | c4d18d452f82 |
rev | line source |
---|---|
10944 | 1 /* |
2 * (I)DCT Transforms | |
3 * Copyright (c) 2009 Peter Ross <pross@xvid.org> | |
4 * Copyright (c) 2010 Alex Converse <alex.converse@gmail.com> | |
5 * Copyright (c) 2010 Vitor Sessak | |
6 * | |
7 * This file is part of FFmpeg. | |
8 * | |
9 * FFmpeg is free software; you can redistribute it and/or | |
10 * modify it under the terms of the GNU Lesser General Public | |
11 * License as published by the Free Software Foundation; either | |
12 * version 2.1 of the License, or (at your option) any later version. | |
13 * | |
14 * FFmpeg is distributed in the hope that it will be useful, | |
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU | |
17 * Lesser General Public License for more details. | |
18 * | |
19 * You should have received a copy of the GNU Lesser General Public | |
20 * License along with FFmpeg; if not, write to the Free Software | |
21 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA | |
22 */ | |
23 | |
24 /** | |
25 * @file libavcodec/dct.c | |
26 * (Inverse) Discrete Cosine Transforms. These are also known as the | |
27 * type II and type III DCTs respectively. | |
28 */ | |
29 | |
30 #include <math.h> | |
11370 | 31 #include "libavutil/mathematics.h" |
32 #include "fft.h" | |
10944 | 33 |
34 /* sin((M_PI * x / (2*n)) */ | |
35 #define SIN(s,n,x) (s->costab[(n) - (x)]) | |
36 | |
37 /* cos((M_PI * x / (2*n)) */ | |
38 #define COS(s,n,x) (s->costab[x]) | |
39 | |
40 static void ff_dct_calc_c(DCTContext *ctx, FFTSample *data) | |
41 { | |
42 int n = 1 << ctx->nbits; | |
43 int i; | |
44 | |
45 if (ctx->inverse) { | |
46 float next = data[n - 1]; | |
47 float inv_n = 1.0f / n; | |
48 | |
49 for (i = n - 2; i >= 2; i -= 2) { | |
50 float val1 = data[i ]; | |
51 float val2 = data[i - 1] - data[i + 1]; | |
52 float c = COS(ctx, n, i); | |
53 float s = SIN(ctx, n, i); | |
54 | |
55 data[i ] = c * val1 + s * val2; | |
56 data[i + 1] = s * val1 - c * val2; | |
57 } | |
58 | |
59 data[1] = 2 * next; | |
60 | |
61 ff_rdft_calc(&ctx->rdft, data); | |
62 | |
63 for (i = 0; i < n / 2; i++) { | |
64 float tmp1 = data[i ] * inv_n; | |
65 float tmp2 = data[n - i - 1] * inv_n; | |
66 float csc = ctx->csc2[i] * (tmp1 - tmp2); | |
67 | |
68 tmp1 += tmp2; | |
69 data[i ] = tmp1 + csc; | |
70 data[n - i - 1] = tmp1 - csc; | |
71 } | |
72 } else { | |
73 float next; | |
74 for (i=0; i < n/2; i++) { | |
75 float tmp1 = data[i ]; | |
76 float tmp2 = data[n - i - 1]; | |
77 float s = SIN(ctx, n, 2*i + 1); | |
78 | |
79 s *= tmp1 - tmp2; | |
80 tmp1 = (tmp1 + tmp2) * 0.5f; | |
81 | |
82 data[i ] = tmp1 + s; | |
83 data[n-i-1] = tmp1 - s; | |
84 } | |
85 | |
86 ff_rdft_calc(&ctx->rdft, data); | |
87 | |
88 next = data[1] * 0.5; | |
89 data[1] *= -1; | |
90 | |
91 for (i = n - 2; i >= 0; i -= 2) { | |
92 float inr = data[i ]; | |
93 float ini = data[i + 1]; | |
94 float c = COS(ctx, n, i); | |
95 float s = SIN(ctx, n, i); | |
96 | |
97 data[i ] = c * inr + s * ini; | |
98 | |
99 data[i+1] = next; | |
100 | |
101 next += s * inr - c * ini; | |
102 } | |
103 } | |
104 } | |
105 | |
106 void ff_dct_calc(DCTContext *s, FFTSample *data) | |
107 { | |
108 ff_dct_calc_c(s, data); | |
109 } | |
110 | |
11517
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
111 av_cold int ff_dct_init(DCTContext *s, int nbits, int inverse) |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
112 { |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
113 int n = 1 << nbits; |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
114 int i; |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
115 |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
116 s->nbits = nbits; |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
117 s->inverse = inverse; |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
118 |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
119 ff_init_ff_cos_tabs(nbits+2); |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
120 |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
121 s->costab = ff_cos_tabs[nbits+2]; |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
122 |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
123 s->csc2 = av_malloc(n/2 * sizeof(FFTSample)); |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
124 |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
125 if (ff_rdft_init(&s->rdft, nbits, inverse) < 0) { |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
126 av_free(s->csc2); |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
127 return -1; |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
128 } |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
129 |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
130 for (i = 0; i < n/2; i++) |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
131 s->csc2[i] = 0.5 / sin((M_PI / (2*n) * (2*i + 1))); |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
132 |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
133 return 0; |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
134 } |
e3b680f6c106
Cosmetics: move ff_dct_init() to the bottom of the file
vitor
parents:
11370
diff
changeset
|
135 |
10944 | 136 av_cold void ff_dct_end(DCTContext *s) |
137 { | |
138 ff_rdft_end(&s->rdft); | |
139 av_free(s->csc2); | |
140 } |