annotate snow.c @ 2156:f06fb6fc0ff9 libavcodec

cleanup
author michael
date Mon, 02 Aug 2004 01:06:55 +0000
parents 274a01d80f4a
children 7f42295c1517
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1 /*
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2 * Copyright (C) 2004 Michael Niedermayer <michaelni@gmx.at>
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3 *
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
4 * This library is free software; you can redistribute it and/or
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
5 * modify it under the terms of the GNU Lesser General Public
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
6 * License as published by the Free Software Foundation; either
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
7 * version 2 of the License, or (at your option) any later version.
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
8 *
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
9 * This library is distributed in the hope that it will be useful,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
10 * but WITHOUT ANY WARRANTY; without even the implied warranty of
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
12 * Lesser General Public License for more details.
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
13 *
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
14 * You should have received a copy of the GNU Lesser General Public
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
15 * License along with this library; if not, write to the Free Software
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
16 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
17 */
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
18
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
19 #include "avcodec.h"
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
20 #include "common.h"
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
21 #include "dsputil.h"
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
22 #include "cabac.h"
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
23
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
24 #include "mpegvideo.h"
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
25
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
26 #undef NDEBUG
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
27 #include <assert.h>
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
28
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
29 #define MAX_DECOMPOSITIONS 8
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
30 #define MAX_PLANES 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
31 #define DWTELEM int
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
32 #define QROOT 8
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
33
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
34 static const int8_t quant3[256]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
35 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
36 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
37 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
38 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
39 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
40 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
41 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
42 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
43 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
44 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
45 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
46 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
47 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
48 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
49 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
50 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
51 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
52 static const int8_t quant3b[256]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
53 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
54 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
55 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
56 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
57 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
58 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
59 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
60 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
61 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
62 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
63 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
64 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
65 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
66 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
67 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
68 -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
69 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
70 static const int8_t quant5[256]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
71 0, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
72 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
73 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
74 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
75 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
76 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
77 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
78 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
79 -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
80 -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
81 -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
82 -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
83 -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
84 -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
85 -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
86 -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-1,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
87 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
88 static const int8_t quant7[256]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
89 0, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
90 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
91 2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
92 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
93 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
94 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
95 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
96 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
97 -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
98 -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
99 -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
100 -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
101 -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
102 -3,-3,-3,-3,-3,-3,-3,-3,-3,-2,-2,-2,-2,-2,-2,-2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
103 -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
104 -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
105 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
106 static const int8_t quant9[256]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
107 0, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
108 3, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
109 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
110 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
111 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
112 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
113 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
114 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
115 -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
116 -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
117 -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
118 -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
119 -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
120 -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
121 -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-3,-3,-3,-3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
122 -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-2,-2,-2,-2,-1,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
123 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
124 static const int8_t quant11[256]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
125 0, 1, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
126 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
127 4, 4, 4, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
128 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
129 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
130 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
131 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
132 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
133 -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
134 -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
135 -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
136 -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
137 -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
138 -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-4,-4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
139 -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
140 -4,-4,-4,-4,-4,-3,-3,-3,-3,-3,-3,-3,-2,-2,-2,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
141 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
142 static const int8_t quant13[256]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
143 0, 1, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
144 4, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
145 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
146 5, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
147 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
148 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
149 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
150 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
151 -6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
152 -6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
153 -6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
154 -6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
155 -6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
156 -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
157 -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
158 -4,-4,-4,-4,-4,-4,-4,-4,-4,-3,-3,-3,-3,-2,-2,-1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
159 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
160
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
161 #define OBMC_MAX 64
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
162 #if 0 //64*cubic
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
163 static const uint8_t obmc32[1024]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
164 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
165 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
166 0, 0, 0, 1, 1, 1, 1, 2, 2, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4, 3, 3, 3, 2, 2, 1, 1, 1, 1, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
167 0, 0, 1, 1, 2, 2, 3, 4, 4, 5, 6, 6, 7, 7, 8, 8, 8, 8, 7, 7, 6, 6, 5, 4, 4, 3, 2, 2, 1, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
168 0, 0, 1, 2, 2, 3, 4, 6, 7, 8, 9,10,11,12,12,12,12,12,12,11,10, 9, 8, 7, 6, 4, 3, 2, 2, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
169 0, 1, 1, 2, 3, 5, 6, 8,10,11,13,14,15,16,17,18,18,17,16,15,14,13,11,10, 8, 6, 5, 3, 2, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
170 0, 1, 1, 3, 4, 6, 8,10,13,15,17,19,20,22,22,23,23,22,22,20,19,17,15,13,10, 8, 6, 4, 3, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
171 0, 1, 2, 4, 6, 8,10,13,16,19,21,23,25,27,28,29,29,28,27,25,23,21,19,16,13,10, 8, 6, 4, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
172 0, 1, 2, 4, 7,10,13,16,19,22,25,28,31,33,34,35,35,34,33,31,28,25,22,19,16,13,10, 7, 4, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
173 0, 1, 3, 5, 8,11,15,19,22,26,30,33,36,38,40,41,41,40,38,36,33,30,26,22,19,15,11, 8, 5, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
174 0, 1, 3, 6, 9,12,17,21,25,30,34,38,41,44,45,46,46,45,44,41,38,34,30,25,21,17,12, 9, 6, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
175 0, 1, 3, 6,10,14,19,23,28,33,38,42,45,48,51,52,52,51,48,45,42,38,33,28,23,19,14,10, 6, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
176 0, 1, 4, 7,11,15,20,25,31,36,41,45,49,52,55,56,56,55,52,49,45,41,36,31,25,20,15,11, 7, 4, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
177 0, 2, 4, 7,12,16,22,27,33,38,44,48,52,56,58,60,60,58,56,52,48,44,38,33,27,22,16,12, 7, 4, 2, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
178 0, 1, 4, 8,12,17,22,28,34,40,45,51,55,58,61,62,62,61,58,55,51,45,40,34,28,22,17,12, 8, 4, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
179 0, 2, 4, 8,12,18,23,29,35,41,46,52,56,60,62,64,64,62,60,56,52,46,41,35,29,23,18,12, 8, 4, 2, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
180 0, 2, 4, 8,12,18,23,29,35,41,46,52,56,60,62,64,64,62,60,56,52,46,41,35,29,23,18,12, 8, 4, 2, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
181 0, 1, 4, 8,12,17,22,28,34,40,45,51,55,58,61,62,62,61,58,55,51,45,40,34,28,22,17,12, 8, 4, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
182 0, 2, 4, 7,12,16,22,27,33,38,44,48,52,56,58,60,60,58,56,52,48,44,38,33,27,22,16,12, 7, 4, 2, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
183 0, 1, 4, 7,11,15,20,25,31,36,41,45,49,52,55,56,56,55,52,49,45,41,36,31,25,20,15,11, 7, 4, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
184 0, 1, 3, 6,10,14,19,23,28,33,38,42,45,48,51,52,52,51,48,45,42,38,33,28,23,19,14,10, 6, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
185 0, 1, 3, 6, 9,12,17,21,25,30,34,38,41,44,45,46,46,45,44,41,38,34,30,25,21,17,12, 9, 6, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
186 0, 1, 3, 5, 8,11,15,19,22,26,30,33,36,38,40,41,41,40,38,36,33,30,26,22,19,15,11, 8, 5, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
187 0, 1, 2, 4, 7,10,13,16,19,22,25,28,31,33,34,35,35,34,33,31,28,25,22,19,16,13,10, 7, 4, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
188 0, 1, 2, 4, 6, 8,10,13,16,19,21,23,25,27,28,29,29,28,27,25,23,21,19,16,13,10, 8, 6, 4, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
189 0, 1, 1, 3, 4, 6, 8,10,13,15,17,19,20,22,22,23,23,22,22,20,19,17,15,13,10, 8, 6, 4, 3, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
190 0, 1, 1, 2, 3, 5, 6, 8,10,11,13,14,15,16,17,18,18,17,16,15,14,13,11,10, 8, 6, 5, 3, 2, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
191 0, 0, 1, 2, 2, 3, 4, 6, 7, 8, 9,10,11,12,12,12,12,12,12,11,10, 9, 8, 7, 6, 4, 3, 2, 2, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
192 0, 0, 1, 1, 2, 2, 3, 4, 4, 5, 6, 6, 7, 7, 8, 8, 8, 8, 7, 7, 6, 6, 5, 4, 4, 3, 2, 2, 1, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
193 0, 0, 0, 1, 1, 1, 1, 2, 2, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4, 3, 3, 3, 2, 2, 1, 1, 1, 1, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
194 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
195 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
196 //error:0.000022
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
197 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
198 static const uint8_t obmc16[256]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
199 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
200 0, 1, 1, 2, 4, 5, 5, 6, 6, 5, 5, 4, 2, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
201 0, 1, 4, 6, 9,11,13,15,15,13,11, 9, 6, 4, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
202 0, 2, 6,11,15,20,24,26,26,24,20,15,11, 6, 2, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
203 0, 4, 9,15,23,29,34,38,38,34,29,23,15, 9, 4, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
204 0, 5,11,20,29,38,45,49,49,45,38,29,20,11, 5, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
205 1, 5,13,24,34,45,53,57,57,53,45,34,24,13, 5, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
206 1, 6,15,26,38,49,57,62,62,57,49,38,26,15, 6, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
207 1, 6,15,26,38,49,57,62,62,57,49,38,26,15, 6, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
208 1, 5,13,24,34,45,53,57,57,53,45,34,24,13, 5, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
209 0, 5,11,20,29,38,45,49,49,45,38,29,20,11, 5, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
210 0, 4, 9,15,23,29,34,38,38,34,29,23,15, 9, 4, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
211 0, 2, 6,11,15,20,24,26,26,24,20,15,11, 6, 2, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
212 0, 1, 4, 6, 9,11,13,15,15,13,11, 9, 6, 4, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
213 0, 1, 1, 2, 4, 5, 5, 6, 6, 5, 5, 4, 2, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
214 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
215 //error:0.000033
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
216 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
217 #elif 1 // 64*linear
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
218 static const uint8_t obmc32[1024]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
219 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
220 0, 1, 1, 1, 2, 2, 2, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 5, 5, 5, 4, 4, 4, 3, 3, 2, 2, 2, 1, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
221 0, 1, 2, 2, 3, 3, 4, 5, 5, 6, 7, 7, 8, 8, 9,10,10, 9, 8, 8, 7, 7, 6, 5, 5, 4, 3, 3, 2, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
222 0, 1, 2, 3, 4, 5, 6, 7, 7, 8, 9,10,11,12,13,14,14,13,12,11,10, 9, 8, 7, 7, 6, 5, 4, 3, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
223 1, 2, 3, 4, 5, 6, 7, 8,10,11,12,13,14,15,16,17,17,16,15,14,13,12,11,10, 8, 7, 6, 5, 4, 3, 2, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
224 1, 2, 3, 5, 6, 8, 9,10,12,13,14,16,17,19,20,21,21,20,19,17,16,14,13,12,10, 9, 8, 6, 5, 3, 2, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
225 1, 2, 4, 6, 7, 9,11,12,14,15,17,19,20,22,24,25,25,24,22,20,19,17,15,14,12,11, 9, 7, 6, 4, 2, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
226 1, 3, 5, 7, 8,10,12,14,16,18,20,22,23,25,27,29,29,27,25,23,22,20,18,16,14,12,10, 8, 7, 5, 3, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
227 1, 3, 5, 7,10,12,14,16,18,20,22,24,27,29,31,33,33,31,29,27,24,22,20,18,16,14,12,10, 7, 5, 3, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
228 1, 4, 6, 8,11,13,15,18,20,23,25,27,30,32,34,37,37,34,32,30,27,25,23,20,18,15,13,11, 8, 6, 4, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
229 1, 4, 7, 9,12,14,17,20,22,25,28,30,33,35,38,41,41,38,35,33,30,28,25,22,20,17,14,12, 9, 7, 4, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
230 1, 4, 7,10,13,16,19,22,24,27,30,33,36,39,42,45,45,42,39,36,33,30,27,24,22,19,16,13,10, 7, 4, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
231 2, 5, 8,11,14,17,20,23,27,30,33,36,39,42,45,48,48,45,42,39,36,33,30,27,23,20,17,14,11, 8, 5, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
232 2, 5, 8,12,15,19,22,25,29,32,35,39,42,46,49,52,52,49,46,42,39,35,32,29,25,22,19,15,12, 8, 5, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
233 2, 5, 9,13,16,20,24,27,31,34,38,42,45,49,53,56,56,53,49,45,42,38,34,31,27,24,20,16,13, 9, 5, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
234 2, 6,10,14,17,21,25,29,33,37,41,45,48,52,56,60,60,56,52,48,45,41,37,33,29,25,21,17,14,10, 6, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
235 2, 6,10,14,17,21,25,29,33,37,41,45,48,52,56,60,60,56,52,48,45,41,37,33,29,25,21,17,14,10, 6, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
236 2, 5, 9,13,16,20,24,27,31,34,38,42,45,49,53,56,56,53,49,45,42,38,34,31,27,24,20,16,13, 9, 5, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
237 2, 5, 8,12,15,19,22,25,29,32,35,39,42,46,49,52,52,49,46,42,39,35,32,29,25,22,19,15,12, 8, 5, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
238 2, 5, 8,11,14,17,20,23,27,30,33,36,39,42,45,48,48,45,42,39,36,33,30,27,23,20,17,14,11, 8, 5, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
239 1, 4, 7,10,13,16,19,22,24,27,30,33,36,39,42,45,45,42,39,36,33,30,27,24,22,19,16,13,10, 7, 4, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
240 1, 4, 7, 9,12,14,17,20,22,25,28,30,33,35,38,41,41,38,35,33,30,28,25,22,20,17,14,12, 9, 7, 4, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
241 1, 4, 6, 8,11,13,15,18,20,23,25,27,30,32,34,37,37,34,32,30,27,25,23,20,18,15,13,11, 8, 6, 4, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
242 1, 3, 5, 7,10,12,14,16,18,20,22,24,27,29,31,33,33,31,29,27,24,22,20,18,16,14,12,10, 7, 5, 3, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
243 1, 3, 5, 7, 8,10,12,14,16,18,20,22,23,25,27,29,29,27,25,23,22,20,18,16,14,12,10, 8, 7, 5, 3, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
244 1, 2, 4, 6, 7, 9,11,12,14,15,17,19,20,22,24,25,25,24,22,20,19,17,15,14,12,11, 9, 7, 6, 4, 2, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
245 1, 2, 3, 5, 6, 8, 9,10,12,13,14,16,17,19,20,21,21,20,19,17,16,14,13,12,10, 9, 8, 6, 5, 3, 2, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
246 1, 2, 3, 4, 5, 6, 7, 8,10,11,12,13,14,15,16,17,17,16,15,14,13,12,11,10, 8, 7, 6, 5, 4, 3, 2, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
247 0, 1, 2, 3, 4, 5, 6, 7, 7, 8, 9,10,11,12,13,14,14,13,12,11,10, 9, 8, 7, 7, 6, 5, 4, 3, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
248 0, 1, 2, 2, 3, 3, 4, 5, 5, 6, 7, 7, 8, 8, 9,10,10, 9, 8, 8, 7, 7, 6, 5, 5, 4, 3, 3, 2, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
249 0, 1, 1, 1, 2, 2, 2, 3, 3, 4, 4, 4, 5, 5, 5, 6, 6, 5, 5, 5, 4, 4, 4, 3, 3, 2, 2, 2, 1, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
250 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
251 //error:0.000020
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
252 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
253 static const uint8_t obmc16[256]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
254 0, 1, 1, 2, 2, 3, 3, 4, 4, 3, 3, 2, 2, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
255 1, 2, 4, 5, 7, 8,10,11,11,10, 8, 7, 5, 4, 2, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
256 1, 4, 6, 9,11,14,16,19,19,16,14,11, 9, 6, 4, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
257 2, 5, 9,12,16,19,23,26,26,23,19,16,12, 9, 5, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
258 2, 7,11,16,20,25,29,34,34,29,25,20,16,11, 7, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
259 3, 8,14,19,25,30,36,41,41,36,30,25,19,14, 8, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
260 3,10,16,23,29,36,42,49,49,42,36,29,23,16,10, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
261 4,11,19,26,34,41,49,56,56,49,41,34,26,19,11, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
262 4,11,19,26,34,41,49,56,56,49,41,34,26,19,11, 4,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
263 3,10,16,23,29,36,42,49,49,42,36,29,23,16,10, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
264 3, 8,14,19,25,30,36,41,41,36,30,25,19,14, 8, 3,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
265 2, 7,11,16,20,25,29,34,34,29,25,20,16,11, 7, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
266 2, 5, 9,12,16,19,23,26,26,23,19,16,12, 9, 5, 2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
267 1, 4, 6, 9,11,14,16,19,19,16,14,11, 9, 6, 4, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
268 1, 2, 4, 5, 7, 8,10,11,11,10, 8, 7, 5, 4, 2, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
269 0, 1, 1, 2, 2, 3, 3, 4, 4, 3, 3, 2, 2, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
270 //error:0.000015
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
271 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
272 #else //64*cos
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
273 static const uint8_t obmc32[1024]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
274 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
275 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 2, 1, 1, 2, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
276 0, 0, 0, 1, 1, 1, 1, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 3, 3, 3, 3, 2, 2, 1, 1, 1, 1, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
277 0, 0, 1, 1, 1, 2, 2, 3, 4, 5, 5, 6, 7, 7, 7, 7, 7, 7, 7, 7, 6, 5, 5, 4, 3, 2, 2, 1, 1, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
278 0, 0, 1, 1, 2, 3, 4, 5, 6, 7, 9,10,11,11,12,12,12,12,11,11,10, 9, 7, 6, 5, 4, 3, 2, 1, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
279 0, 0, 1, 2, 3, 5, 6, 8, 9,11,12,14,15,16,17,17,17,17,16,15,14,12,11, 9, 8, 6, 5, 3, 2, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
280 0, 1, 1, 2, 4, 6, 8,10,12,15,17,19,20,21,22,23,23,22,21,20,19,17,15,12,10, 8, 6, 4, 2, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
281 0, 1, 2, 3, 5, 8,10,13,16,19,21,24,26,27,28,29,29,28,27,26,24,21,19,16,13,10, 8, 5, 3, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
282 0, 1, 2, 4, 6, 9,12,16,19,23,26,29,31,33,34,35,35,34,33,31,29,26,23,19,16,12, 9, 6, 4, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
283 0, 1, 3, 5, 7,11,15,19,23,26,30,34,37,39,40,41,41,40,39,37,34,30,26,23,19,15,11, 7, 5, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
284 0, 1, 3, 5, 9,12,17,21,26,30,35,38,42,44,46,47,47,46,44,42,38,35,30,26,21,17,12, 9, 5, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
285 0, 1, 3, 6, 9,14,19,24,29,34,38,43,46,49,51,52,52,51,49,46,43,38,34,29,24,19,14, 9, 6, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
286 0, 1, 3, 6,11,15,20,26,31,37,42,46,50,53,56,57,57,56,53,50,46,42,37,31,26,20,15,11, 6, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
287 0, 1, 3, 7,11,16,21,27,33,39,44,49,53,57,59,60,60,59,57,53,49,44,39,33,27,21,16,11, 7, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
288 0, 1, 4, 7,12,17,22,28,34,40,46,51,56,59,61,63,63,61,59,56,51,46,40,34,28,22,17,12, 7, 4, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
289 0, 1, 4, 7,12,17,23,29,35,41,47,52,57,60,63,64,64,63,60,57,52,47,41,35,29,23,17,12, 7, 4, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
290 0, 1, 4, 7,12,17,23,29,35,41,47,52,57,60,63,64,64,63,60,57,52,47,41,35,29,23,17,12, 7, 4, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
291 0, 1, 4, 7,12,17,22,28,34,40,46,51,56,59,61,63,63,61,59,56,51,46,40,34,28,22,17,12, 7, 4, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
292 0, 1, 3, 7,11,16,21,27,33,39,44,49,53,57,59,60,60,59,57,53,49,44,39,33,27,21,16,11, 7, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
293 0, 1, 3, 6,11,15,20,26,31,37,42,46,50,53,56,57,57,56,53,50,46,42,37,31,26,20,15,11, 6, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
294 0, 1, 3, 6, 9,14,19,24,29,34,38,43,46,49,51,52,52,51,49,46,43,38,34,29,24,19,14, 9, 6, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
295 0, 1, 3, 5, 9,12,17,21,26,30,35,38,42,44,46,47,47,46,44,42,38,35,30,26,21,17,12, 9, 5, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
296 0, 1, 3, 5, 7,11,15,19,23,26,30,34,37,39,40,41,41,40,39,37,34,30,26,23,19,15,11, 7, 5, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
297 0, 1, 2, 4, 6, 9,12,16,19,23,26,29,31,33,34,35,35,34,33,31,29,26,23,19,16,12, 9, 6, 4, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
298 0, 1, 2, 3, 5, 8,10,13,16,19,21,24,26,27,28,29,29,28,27,26,24,21,19,16,13,10, 8, 5, 3, 2, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
299 0, 1, 1, 2, 4, 6, 8,10,12,15,17,19,20,21,22,23,23,22,21,20,19,17,15,12,10, 8, 6, 4, 2, 1, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
300 0, 0, 1, 2, 3, 5, 6, 8, 9,11,12,14,15,16,17,17,17,17,16,15,14,12,11, 9, 8, 6, 5, 3, 2, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
301 0, 0, 1, 1, 2, 3, 4, 5, 6, 7, 9,10,11,11,12,12,12,12,11,11,10, 9, 7, 6, 5, 4, 3, 2, 1, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
302 0, 0, 1, 1, 1, 2, 2, 3, 4, 5, 5, 6, 7, 7, 7, 7, 7, 7, 7, 7, 6, 5, 5, 4, 3, 2, 2, 1, 1, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
303 0, 0, 0, 1, 1, 1, 1, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 3, 3, 3, 3, 2, 2, 1, 1, 1, 1, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
304 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 2, 1, 1, 2, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
305 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
306 //error:0.000022
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
307 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
308 static const uint8_t obmc16[256]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
309 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
310 0, 0, 1, 2, 3, 4, 5, 5, 5, 5, 4, 3, 2, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
311 0, 1, 3, 6, 8,11,13,14,14,13,11, 8, 6, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
312 0, 2, 6,10,15,20,24,26,26,24,20,15,10, 6, 2, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
313 0, 3, 8,16,23,30,35,38,38,35,30,23,16, 8, 3, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
314 1, 4,11,20,30,39,46,49,49,46,39,30,20,11, 4, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
315 1, 5,13,24,35,46,54,58,58,54,46,35,24,13, 5, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
316 0, 5,14,26,38,49,58,63,63,58,49,38,26,14, 5, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
317 0, 5,14,26,38,49,58,63,63,58,49,38,26,14, 5, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
318 1, 5,13,24,35,46,54,58,58,54,46,35,24,13, 5, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
319 1, 4,11,20,30,39,46,49,49,46,39,30,20,11, 4, 1,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
320 0, 3, 8,16,23,30,35,38,38,35,30,23,16, 8, 3, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
321 0, 2, 6,10,15,20,24,26,26,24,20,15,10, 6, 2, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
322 0, 1, 3, 6, 8,11,13,14,14,13,11, 8, 6, 3, 1, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
323 0, 0, 1, 2, 3, 4, 5, 5, 5, 5, 4, 3, 2, 1, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
324 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
325 //error:0.000022
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
326 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
327 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
328
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
329 typedef struct SubBand{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
330 int level;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
331 int stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
332 int width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
333 int height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
334 int qlog; ///< log(qscale)/log[2^(1/6)]
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
335 DWTELEM *buf;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
336 struct SubBand *parent;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
337 uint8_t state[/*7*2*/ 7 + 512][32];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
338 }SubBand;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
339
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
340 typedef struct Plane{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
341 int width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
342 int height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
343 SubBand band[MAX_DECOMPOSITIONS][4];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
344 }Plane;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
345
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
346 typedef struct SnowContext{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
347 // MpegEncContext m; // needed for motion estimation, should not be used for anything else, the idea is to make the motion estimation eventually independant of MpegEncContext, so this will be removed then (FIXME/XXX)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
348
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
349 AVCodecContext *avctx;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
350 CABACContext c;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
351 DSPContext dsp;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
352 AVFrame input_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
353 AVFrame current_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
354 AVFrame last_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
355 AVFrame mconly_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
356 // uint8_t q_context[16];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
357 uint8_t header_state[32];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
358 int keyframe;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
359 int version;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
360 int spatial_decomposition_type;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
361 int temporal_decomposition_type;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
362 int spatial_decomposition_count;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
363 int temporal_decomposition_count;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
364 DWTELEM *spatial_dwt_buffer;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
365 DWTELEM *pred_buffer;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
366 int colorspace_type;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
367 int chroma_h_shift;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
368 int chroma_v_shift;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
369 int spatial_scalability;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
370 int qlog;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
371 int mv_scale;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
372 int qbias;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
373 #define QBIAS_SHIFT 3
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
374 int b_width; //FIXME remove?
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
375 int b_height; //FIXME remove?
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
376 Plane plane[MAX_PLANES];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
377 SubBand mb_band;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
378 SubBand mv_band[2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
379
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
380 uint16_t *mb_type;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
381 uint8_t *mb_mean;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
382 uint32_t *dummy;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
383 int16_t (*motion_val8)[2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
384 int16_t (*motion_val16)[2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
385 MpegEncContext m; // needed for motion estimation, should not be used for anything else, the idea is to make the motion estimation eventually independant of MpegEncContext, so this will be removed then (FIXME/XXX)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
386 }SnowContext;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
387
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
388 #define QEXPSHIFT 7 //FIXME try to change this to 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
389 static const uint8_t qexp[8]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
390 128, 140, 152, 166, 181, 197, 215, 235
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
391 // 64, 70, 76, 83, 91, 99, 108, 117
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
392 // 32, 35, 38, 41, 45, 49, 54, 59
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
393 // 16, 17, 19, 21, 23, 25, 27, 29
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
394 // 8, 9, 10, 10, 11, 12, 13, 15
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
395 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
396
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
397 static inline int mirror(int v, int m){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
398 if (v<0) return -v;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
399 else if(v>m) return 2*m-v;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
400 else return v;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
401 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
402
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
403 static inline void put_symbol(CABACContext *c, uint8_t *state, int v, int is_signed){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
404 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
405
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
406 if(v){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
407 const int a= ABS(v);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
408 const int e= av_log2(a);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
409 #if 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
410 const int el= FFMIN(e, 10);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
411 put_cabac(c, state+0, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
412
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
413 for(i=0; i<el; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
414 put_cabac(c, state+1+i, 1); //1..10
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
415 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
416 for(; i<e; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
417 put_cabac(c, state+1+9, 1); //1..10
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
418 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
419 put_cabac(c, state+1+FFMIN(i,9), 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
420
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
421 for(i=e-1; i>=el; i--){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
422 put_cabac(c, state+22+9, (a>>i)&1); //22..31
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
423 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
424 for(; i>=0; i--){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
425 put_cabac(c, state+22+i, (a>>i)&1); //22..31
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
426 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
427
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
428 if(is_signed)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
429 put_cabac(c, state+11 + el, v < 0); //11..21
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
430 #else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
431
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
432 put_cabac(c, state+0, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
433 if(e<=9){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
434 for(i=0; i<e; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
435 put_cabac(c, state+1+i, 1); //1..10
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
436 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
437 put_cabac(c, state+1+i, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
438
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
439 for(i=e-1; i>=0; i--){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
440 put_cabac(c, state+22+i, (a>>i)&1); //22..31
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
441 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
442
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
443 if(is_signed)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
444 put_cabac(c, state+11 + e, v < 0); //11..21
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
445 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
446 for(i=0; i<e; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
447 put_cabac(c, state+1+FFMIN(i,9), 1); //1..10
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
448 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
449 put_cabac(c, state+1+FFMIN(i,9), 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
450
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
451 for(i=e-1; i>=0; i--){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
452 put_cabac(c, state+22+FFMIN(i,9), (a>>i)&1); //22..31
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
453 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
454
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
455 if(is_signed)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
456 put_cabac(c, state+11 + FFMIN(e,10), v < 0); //11..21
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
457 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
458 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
459 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
460 put_cabac(c, state+0, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
461 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
462 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
463
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
464 static inline int get_symbol(CABACContext *c, uint8_t *state, int is_signed){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
465 if(get_cabac(c, state+0))
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
466 return 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
467 else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
468 int i, e, a, el;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
469 //FIXME try to merge loops with FFMIN() maybe they are equally fast and they are surly cuter
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
470 for(e=0; e<10; e++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
471 if(get_cabac(c, state + 1 + e)==0) // 1..10
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
472 break;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
473 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
474 el= e;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
475
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
476 if(e==10){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
477 while(get_cabac(c, state + 1 + 9)) //10
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
478 e++;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
479 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
480 a= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
481 for(i=e-1; i>=el; i--){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
482 a += a + get_cabac(c, state+22+9); //31
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
483 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
484 for(; i>=0; i--){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
485 a += a + get_cabac(c, state+22+i); //22..31
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
486 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
487
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
488 if(is_signed && get_cabac(c, state+11 + el)) //11..21
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
489 return -a;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
490 else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
491 return a;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
492 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
493 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
494
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
495 static inline void put_symbol2(CABACContext *c, uint8_t *state, int v, int log2){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
496 int i;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
497 int e= av_log2(v<<1);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
498
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
499 assert(v>=0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
500 if(v==0) assert(e==0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
501
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
502 while(e > log2){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
503 put_cabac(c, state+log2, 1);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
504 v -= 1<<log2;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
505 assert(v>=0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
506 e= av_log2(v<<1);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
507 log2++;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
508 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
509 put_cabac(c, state+log2, 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
510
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
511 for(i=log2-1; i>=0; i--){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
512 put_cabac(c, state+31-i, (v>>i)&1);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
513 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
514 assert(!((v>>i)&1));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
515 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
516
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
517 static inline int get_symbol2(CABACContext *c, uint8_t *state, int log2){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
518 int i;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
519 int v=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
520
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
521 while(get_cabac(c, state+log2)){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
522 v+= 1<<log2;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
523 log2++;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
524 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
525
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
526 for(i=log2-1; i>=0; i--){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
527 v+= get_cabac(c, state+31-i)<<i;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
528 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
529
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
530 return v;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
531 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
532
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
533 static always_inline void lift(DWTELEM *dst, DWTELEM *src, DWTELEM *ref, int dst_step, int src_step, int ref_step, int width, int mul, int add, int shift, int highpass, int inverse){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
534 const int mirror_left= !highpass;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
535 const int mirror_right= (width&1) ^ highpass;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
536 const int w= (width>>1) - 1 + (highpass & width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
537 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
538
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
539 #define LIFT(src, ref, inv) ((src) + ((inv) ? - (ref) : + (ref)))
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
540 if(mirror_left){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
541 dst[0] = LIFT(src[0], ((mul*2*ref[0]+add)>>shift), inverse);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
542 dst += dst_step;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
543 src += src_step;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
544 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
545
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
546 for(i=0; i<w; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
547 dst[i*dst_step] = LIFT(src[i*src_step], ((mul*(ref[i*ref_step] + ref[(i+1)*ref_step])+add)>>shift), inverse);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
548 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
549
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
550 if(mirror_right){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
551 dst[w*dst_step] = LIFT(src[w*src_step], ((mul*2*ref[w*ref_step]+add)>>shift), inverse);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
552 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
553 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
554
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
555 static always_inline void lift5(DWTELEM *dst, DWTELEM *src, DWTELEM *ref, int dst_step, int src_step, int ref_step, int width, int mul, int add, int shift, int highpass, int inverse){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
556 const int mirror_left= !highpass;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
557 const int mirror_right= (width&1) ^ highpass;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
558 const int w= (width>>1) - 1 + (highpass & width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
559 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
560
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
561 if(mirror_left){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
562 int r= 3*2*ref[0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
563 r += r>>4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
564 r += r>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
565 dst[0] = LIFT(src[0], ((r+add)>>shift), inverse);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
566 dst += dst_step;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
567 src += src_step;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
568 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
569
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
570 for(i=0; i<w; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
571 int r= 3*(ref[i*ref_step] + ref[(i+1)*ref_step]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
572 r += r>>4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
573 r += r>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
574 dst[i*dst_step] = LIFT(src[i*src_step], ((r+add)>>shift), inverse);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
575 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
576
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
577 if(mirror_right){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
578 int r= 3*2*ref[w*ref_step];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
579 r += r>>4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
580 r += r>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
581 dst[w*dst_step] = LIFT(src[w*src_step], ((r+add)>>shift), inverse);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
582 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
583 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
584
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
585
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
586 static void inplace_lift(int *dst, int width, int *coeffs, int n, int shift, int start, int inverse){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
587 int x, i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
588
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
589 for(x=start; x<width; x+=2){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
590 int64_t sum=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
591
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
592 for(i=0; i<n; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
593 int x2= x + 2*i - n + 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
594 if (x2< 0) x2= -x2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
595 else if(x2>=width) x2= 2*width-x2-2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
596 sum += coeffs[i]*(int64_t)dst[x2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
597 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
598 if(inverse) dst[x] -= (sum + (1<<shift)/2)>>shift;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
599 else dst[x] += (sum + (1<<shift)/2)>>shift;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
600 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
601 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
602
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
603 static void inplace_liftV(int *dst, int width, int height, int stride, int *coeffs, int n, int shift, int start, int inverse){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
604 int x, y, i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
605 for(y=start; y<height; y+=2){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
606 for(x=0; x<width; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
607 int64_t sum=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
608
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
609 for(i=0; i<n; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
610 int y2= y + 2*i - n + 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
611 if (y2< 0) y2= -y2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
612 else if(y2>=height) y2= 2*height-y2-2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
613 sum += coeffs[i]*(int64_t)dst[x + y2*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
614 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
615 if(inverse) dst[x + y*stride] -= (sum + (1<<shift)/2)>>shift;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
616 else dst[x + y*stride] += (sum + (1<<shift)/2)>>shift;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
617 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
618 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
619 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
620
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
621 #define SCALEX 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
622 #define LX0 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
623 #define LX1 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
624
2139
michael
parents: 2138
diff changeset
625 #if 0 // more accurate 9/7
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
626 #define N1 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
627 #define SHIFT1 14
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
628 #define COEFFS1 (int[]){-25987,-25987}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
629 #define N2 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
630 #define SHIFT2 19
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
631 #define COEFFS2 (int[]){-27777,-27777}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
632 #define N3 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
633 #define SHIFT3 15
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
634 #define COEFFS3 (int[]){28931,28931}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
635 #define N4 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
636 #define SHIFT4 15
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
637 #define COEFFS4 (int[]){14533,14533}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
638 #elif 1 // 13/7 CRF
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
639 #define N1 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
640 #define SHIFT1 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
641 #define COEFFS1 (int[]){1,-9,-9,1}
2139
michael
parents: 2138
diff changeset
642 #define N2 4
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
643 #define SHIFT2 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
644 #define COEFFS2 (int[]){-1,5,5,-1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
645 #define N3 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
646 #define SHIFT3 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
647 #define COEFFS3 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
648 #define N4 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
649 #define SHIFT4 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
650 #define COEFFS4 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
651 #elif 1 // 3/5
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
652 #define LX0 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
653 #define LX1 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
654 #define SCALEX 0.5
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
655 #define N1 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
656 #define SHIFT1 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
657 #define COEFFS1 (int[]){1,1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
658 #define N2 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
659 #define SHIFT2 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
660 #define COEFFS2 (int[]){-1,-1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
661 #define N3 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
662 #define SHIFT3 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
663 #define COEFFS3 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
664 #define N4 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
665 #define SHIFT4 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
666 #define COEFFS4 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
667 #elif 1 // 11/5
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
668 #define N1 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
669 #define SHIFT1 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
670 #define COEFFS1 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
671 #define N2 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
672 #define SHIFT2 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
673 #define COEFFS2 (int[]){-1,-1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
674 #define N3 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
675 #define SHIFT3 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
676 #define COEFFS3 (int[]){-1,-1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
677 #define N4 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
678 #define SHIFT4 7
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
679 #define COEFFS4 (int[]){-5,29,29,-5}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
680 #define SCALEX 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
681 #elif 1 // 9/7 CDF
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
682 #define N1 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
683 #define SHIFT1 7
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
684 #define COEFFS1 (int[]){-203,-203}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
685 #define N2 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
686 #define SHIFT2 12
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
687 #define COEFFS2 (int[]){-217,-217}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
688 #define N3 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
689 #define SHIFT3 7
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
690 #define COEFFS3 (int[]){113,113}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
691 #define N4 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
692 #define SHIFT4 9
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
693 #define COEFFS4 (int[]){227,227}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
694 #define SCALEX 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
695 #elif 1 // 7/5 CDF
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
696 #define N1 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
697 #define SHIFT1 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
698 #define COEFFS1 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
699 #define N2 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
700 #define SHIFT2 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
701 #define COEFFS2 (int[]){-1,-1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
702 #define N3 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
703 #define SHIFT3 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
704 #define COEFFS3 (int[]){-1,-1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
705 #define N4 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
706 #define SHIFT4 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
707 #define COEFFS4 (int[]){3,3}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
708 #elif 1 // 9/7 MN
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
709 #define N1 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
710 #define SHIFT1 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
711 #define COEFFS1 (int[]){1,-9,-9,1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
712 #define N2 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
713 #define SHIFT2 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
714 #define COEFFS2 (int[]){1,1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
715 #define N3 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
716 #define SHIFT3 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
717 #define COEFFS3 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
718 #define N4 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
719 #define SHIFT4 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
720 #define COEFFS4 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
721 #else // 13/7 CRF
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
722 #define N1 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
723 #define SHIFT1 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
724 #define COEFFS1 (int[]){1,-9,-9,1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
725 #define N2 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
726 #define SHIFT2 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
727 #define COEFFS2 (int[]){-1,5,5,-1}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
728 #define N3 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
729 #define SHIFT3 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
730 #define COEFFS3 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
731 #define N4 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
732 #define SHIFT4 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
733 #define COEFFS4 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
734 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
735 static void horizontal_decomposeX(int *b, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
736 int temp[width];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
737 const int width2= width>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
738 const int w2= (width+1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
739 int A1,A2,A3,A4, x;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
740
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
741 inplace_lift(b, width, COEFFS1, N1, SHIFT1, LX1, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
742 inplace_lift(b, width, COEFFS2, N2, SHIFT2, LX0, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
743 inplace_lift(b, width, COEFFS3, N3, SHIFT3, LX1, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
744 inplace_lift(b, width, COEFFS4, N4, SHIFT4, LX0, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
745
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
746 for(x=0; x<width2; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
747 temp[x ]= b[2*x ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
748 temp[x+w2]= b[2*x + 1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
749 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
750 if(width&1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
751 temp[x ]= b[2*x ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
752 memcpy(b, temp, width*sizeof(int));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
753 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
754
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
755 static void horizontal_composeX(int *b, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
756 int temp[width];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
757 const int width2= width>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
758 int A1,A2,A3,A4, x;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
759 const int w2= (width+1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
760
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
761 memcpy(temp, b, width*sizeof(int));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
762 for(x=0; x<width2; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
763 b[2*x ]= temp[x ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
764 b[2*x + 1]= temp[x+w2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
765 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
766 if(width&1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
767 b[2*x ]= temp[x ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
768
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
769 inplace_lift(b, width, COEFFS4, N4, SHIFT4, LX0, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
770 inplace_lift(b, width, COEFFS3, N3, SHIFT3, LX1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
771 inplace_lift(b, width, COEFFS2, N2, SHIFT2, LX0, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
772 inplace_lift(b, width, COEFFS1, N1, SHIFT1, LX1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
773 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
774
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
775 static void spatial_decomposeX(int *buffer, int width, int height, int stride){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
776 int x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
777
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
778 for(y=0; y<height; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
779 for(x=0; x<width; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
780 buffer[y*stride + x] *= SCALEX;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
781 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
782 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
783
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
784 for(y=0; y<height; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
785 horizontal_decomposeX(buffer + y*stride, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
786 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
787
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
788 inplace_liftV(buffer, width, height, stride, COEFFS1, N1, SHIFT1, LX1, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
789 inplace_liftV(buffer, width, height, stride, COEFFS2, N2, SHIFT2, LX0, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
790 inplace_liftV(buffer, width, height, stride, COEFFS3, N3, SHIFT3, LX1, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
791 inplace_liftV(buffer, width, height, stride, COEFFS4, N4, SHIFT4, LX0, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
792 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
793
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
794 static void spatial_composeX(int *buffer, int width, int height, int stride){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
795 int x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
796
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
797 inplace_liftV(buffer, width, height, stride, COEFFS4, N4, SHIFT4, LX0, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
798 inplace_liftV(buffer, width, height, stride, COEFFS3, N3, SHIFT3, LX1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
799 inplace_liftV(buffer, width, height, stride, COEFFS2, N2, SHIFT2, LX0, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
800 inplace_liftV(buffer, width, height, stride, COEFFS1, N1, SHIFT1, LX1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
801
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
802 for(y=0; y<height; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
803 horizontal_composeX(buffer + y*stride, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
804 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
805
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
806 for(y=0; y<height; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
807 for(x=0; x<width; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
808 buffer[y*stride + x] /= SCALEX;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
809 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
810 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
811 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
812
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
813 static void horizontal_decompose53i(int *b, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
814 int temp[width];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
815 const int width2= width>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
816 int A1,A2,A3,A4, x;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
817 const int w2= (width+1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
818
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
819 for(x=0; x<width2; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
820 temp[x ]= b[2*x ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
821 temp[x+w2]= b[2*x + 1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
822 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
823 if(width&1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
824 temp[x ]= b[2*x ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
825 #if 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
826 A2= temp[1 ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
827 A4= temp[0 ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
828 A1= temp[0+width2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
829 A1 -= (A2 + A4)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
830 A4 += (A1 + 1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
831 b[0+width2] = A1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
832 b[0 ] = A4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
833 for(x=1; x+1<width2; x+=2){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
834 A3= temp[x+width2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
835 A4= temp[x+1 ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
836 A3 -= (A2 + A4)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
837 A2 += (A1 + A3 + 2)>>2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
838 b[x+width2] = A3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
839 b[x ] = A2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
840
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
841 A1= temp[x+1+width2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
842 A2= temp[x+2 ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
843 A1 -= (A2 + A4)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
844 A4 += (A1 + A3 + 2)>>2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
845 b[x+1+width2] = A1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
846 b[x+1 ] = A4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
847 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
848 A3= temp[width-1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
849 A3 -= A2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
850 A2 += (A1 + A3 + 2)>>2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
851 b[width -1] = A3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
852 b[width2-1] = A2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
853 #else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
854 lift(b+w2, temp+w2, temp, 1, 1, 1, width, -1, 0, 1, 1, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
855 lift(b , temp , b+w2, 1, 1, 1, width, 1, 2, 2, 0, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
856 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
857 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
858
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
859 static void vertical_decompose53iH0(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
860 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
861
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
862 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
863 b1[i] -= (b0[i] + b2[i])>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
864 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
865 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
866
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
867 static void vertical_decompose53iL0(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
868 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
869
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
870 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
871 b1[i] += (b0[i] + b2[i] + 2)>>2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
872 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
873 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
874
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
875 static void spatial_decompose53i(int *buffer, int width, int height, int stride){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
876 int x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
877 DWTELEM *b0= buffer + mirror(-2-1, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
878 DWTELEM *b1= buffer + mirror(-2 , height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
879
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
880 for(y=-2; y<height; y+=2){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
881 DWTELEM *b2= buffer + mirror(y+1, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
882 DWTELEM *b3= buffer + mirror(y+2, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
883
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
884 {START_TIMER
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
885 if(b1 <= b3) horizontal_decompose53i(b2, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
886 if(y+2 < height) horizontal_decompose53i(b3, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
887 STOP_TIMER("horizontal_decompose53i")}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
888
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
889 {START_TIMER
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
890 if(b1 <= b3) vertical_decompose53iH0(b1, b2, b3, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
891 if(b0 <= b2) vertical_decompose53iL0(b0, b1, b2, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
892 STOP_TIMER("vertical_decompose53i*")}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
893
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
894 b0=b2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
895 b1=b3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
896 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
897 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
898
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
899 #define lift5 lift
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
900 #if 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
901 #define W_AM 3
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
902 #define W_AO 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
903 #define W_AS 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
904
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
905 #define W_BM 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
906 #define W_BO 8
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
907 #define W_BS 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
908
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
909 #undef lift5
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
910 #define W_CM 9999
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
911 #define W_CO 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
912 #define W_CS 2
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
913
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
914 #define W_DM 15
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
915 #define W_DO 16
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
916 #define W_DS 5
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
917 #elif 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
918 #define W_AM 55
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
919 #define W_AO 16
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
920 #define W_AS 5
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
921
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
922 #define W_BM 3
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
923 #define W_BO 32
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
924 #define W_BS 6
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
925
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
926 #define W_CM 127
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
927 #define W_CO 64
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
928 #define W_CS 7
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
929
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
930 #define W_DM 7
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
931 #define W_DO 8
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
932 #define W_DS 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
933 #elif 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
934 #define W_AM 97
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
935 #define W_AO 32
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
936 #define W_AS 6
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
937
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
938 #define W_BM 63
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
939 #define W_BO 512
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
940 #define W_BS 10
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
941
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
942 #define W_CM 13
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
943 #define W_CO 8
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
944 #define W_CS 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
945
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
946 #define W_DM 15
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
947 #define W_DO 16
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
948 #define W_DS 5
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
949
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
950 #else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
951
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
952 #define W_AM 203
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
953 #define W_AO 64
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
954 #define W_AS 7
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
955
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
956 #define W_BM 217
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
957 #define W_BO 2048
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
958 #define W_BS 12
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
959
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
960 #define W_CM 113
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
961 #define W_CO 64
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
962 #define W_CS 7
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
963
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
964 #define W_DM 227
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
965 #define W_DO 128
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
966 #define W_DS 9
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
967 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
968 static void horizontal_decompose97i(int *b, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
969 int temp[width];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
970 const int w2= (width+1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
971
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
972 lift (temp+w2, b +1, b , 1, 2, 2, width, -W_AM, W_AO, W_AS, 1, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
973 lift (temp , b , temp+w2, 1, 2, 1, width, -W_BM, W_BO, W_BS, 0, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
974 lift5(b +w2, temp+w2, temp , 1, 1, 1, width, W_CM, W_CO, W_CS, 1, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
975 lift (b , temp , b +w2, 1, 1, 1, width, W_DM, W_DO, W_DS, 0, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
976 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
977
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
978
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
979 static void vertical_decompose97iH0(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
980 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
981
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
982 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
983 b1[i] -= (W_AM*(b0[i] + b2[i])+W_AO)>>W_AS;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
984 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
985 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
986
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
987 static void vertical_decompose97iH1(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
988 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
989
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
990 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
991 #ifdef lift5
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
992 b1[i] += (W_CM*(b0[i] + b2[i])+W_CO)>>W_CS;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
993 #else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
994 int r= 3*(b0[i] + b2[i]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
995 r+= r>>4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
996 r+= r>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
997 b1[i] += (r+W_CO)>>W_CS;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
998 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
999 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1000 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1001
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1002 static void vertical_decompose97iL0(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1003 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1004
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1005 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1006 b1[i] -= (W_BM*(b0[i] + b2[i])+W_BO)>>W_BS;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1007 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1008 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1009
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1010 static void vertical_decompose97iL1(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1011 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1012
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1013 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1014 b1[i] += (W_DM*(b0[i] + b2[i])+W_DO)>>W_DS;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1015 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1016 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1017
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1018 static void spatial_decompose97i(int *buffer, int width, int height, int stride){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1019 int x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1020 DWTELEM *b0= buffer + mirror(-4-1, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1021 DWTELEM *b1= buffer + mirror(-4 , height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1022 DWTELEM *b2= buffer + mirror(-4+1, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1023 DWTELEM *b3= buffer + mirror(-4+2, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1024
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1025 for(y=-4; y<height; y+=2){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1026 DWTELEM *b4= buffer + mirror(y+3, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1027 DWTELEM *b5= buffer + mirror(y+4, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1028
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1029 {START_TIMER
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1030 if(b3 <= b5) horizontal_decompose97i(b4, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1031 if(y+4 < height) horizontal_decompose97i(b5, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1032 if(width>400){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1033 STOP_TIMER("horizontal_decompose97i")
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1034 }}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1035
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1036 {START_TIMER
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1037 if(b3 <= b5) vertical_decompose97iH0(b3, b4, b5, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1038 if(b2 <= b4) vertical_decompose97iL0(b2, b3, b4, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1039 if(b1 <= b3) vertical_decompose97iH1(b1, b2, b3, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1040 if(b0 <= b2) vertical_decompose97iL1(b0, b1, b2, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1041
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1042 if(width>400){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1043 STOP_TIMER("vertical_decompose97i")
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1044 }}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1045
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1046 b0=b2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1047 b1=b3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1048 b2=b4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1049 b3=b5;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1050 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1051 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1052
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1053 static void spatial_dwt(SnowContext *s, int *buffer, int width, int height, int stride){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1054 int level;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1055
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1056 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1057 switch(s->spatial_decomposition_type){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1058 case 0: spatial_decompose97i(buffer, width>>level, height>>level, stride<<level); break;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1059 case 1: spatial_decompose53i(buffer, width>>level, height>>level, stride<<level); break;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1060 case 2: spatial_decomposeX (buffer, width>>level, height>>level, stride<<level); break;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1061 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1062 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1063 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1064
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1065 static void horizontal_compose53i(int *b, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1066 int temp[width];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1067 const int width2= width>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1068 const int w2= (width+1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1069 int A1,A2,A3,A4, x;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1070
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1071 #if 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1072 A2= temp[1 ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1073 A4= temp[0 ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1074 A1= temp[0+width2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1075 A1 -= (A2 + A4)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1076 A4 += (A1 + 1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1077 b[0+width2] = A1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1078 b[0 ] = A4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1079 for(x=1; x+1<width2; x+=2){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1080 A3= temp[x+width2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1081 A4= temp[x+1 ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1082 A3 -= (A2 + A4)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1083 A2 += (A1 + A3 + 2)>>2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1084 b[x+width2] = A3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1085 b[x ] = A2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1086
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1087 A1= temp[x+1+width2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1088 A2= temp[x+2 ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1089 A1 -= (A2 + A4)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1090 A4 += (A1 + A3 + 2)>>2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1091 b[x+1+width2] = A1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1092 b[x+1 ] = A4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1093 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1094 A3= temp[width-1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1095 A3 -= A2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1096 A2 += (A1 + A3 + 2)>>2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1097 b[width -1] = A3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1098 b[width2-1] = A2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1099 #else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1100 lift(temp , b , b+w2, 1, 1, 1, width, 1, 2, 2, 0, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1101 lift(temp+w2, b+w2, temp, 1, 1, 1, width, -1, 0, 1, 1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1102 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1103 for(x=0; x<width2; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1104 b[2*x ]= temp[x ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1105 b[2*x + 1]= temp[x+w2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1106 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1107 if(width&1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1108 b[2*x ]= temp[x ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1109 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1110
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1111 static void vertical_compose53iH0(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1112 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1113
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1114 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1115 b1[i] += (b0[i] + b2[i])>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1116 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1117 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1118
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1119 static void vertical_compose53iL0(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1120 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1121
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1122 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1123 b1[i] -= (b0[i] + b2[i] + 2)>>2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1124 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1125 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1126
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1127 static void spatial_compose53i(int *buffer, int width, int height, int stride){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1128 int x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1129 DWTELEM *b0= buffer + mirror(-1-1, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1130 DWTELEM *b1= buffer + mirror(-1 , height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1131
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1132 for(y=-1; y<=height; y+=2){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1133 DWTELEM *b2= buffer + mirror(y+1, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1134 DWTELEM *b3= buffer + mirror(y+2, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1135
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1136 {START_TIMER
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1137 if(b1 <= b3) vertical_compose53iL0(b1, b2, b3, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1138 if(b0 <= b2) vertical_compose53iH0(b0, b1, b2, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1139 STOP_TIMER("vertical_compose53i*")}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1140
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1141 {START_TIMER
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1142 if(y-1 >= 0) horizontal_compose53i(b0, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1143 if(b0 <= b2) horizontal_compose53i(b1, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1144 STOP_TIMER("horizontal_compose53i")}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1145
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1146 b0=b2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1147 b1=b3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1148 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1149 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1150
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1151
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1152 static void horizontal_compose97i(int *b, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1153 int temp[width];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1154 const int w2= (width+1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1155
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1156 lift (temp , b , b +w2, 1, 1, 1, width, W_DM, W_DO, W_DS, 0, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1157 lift5(temp+w2, b +w2, temp , 1, 1, 1, width, W_CM, W_CO, W_CS, 1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1158 lift (b , temp , temp+w2, 2, 1, 1, width, -W_BM, W_BO, W_BS, 0, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1159 lift (b+1 , temp+w2, b , 2, 1, 2, width, -W_AM, W_AO, W_AS, 1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1160 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1161
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1162 static void vertical_compose97iH0(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1163 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1164
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1165 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1166 b1[i] += (W_AM*(b0[i] + b2[i])+W_AO)>>W_AS;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1167 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1168 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1169
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1170 static void vertical_compose97iH1(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1171 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1172
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1173 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1174 #ifdef lift5
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1175 b1[i] -= (W_CM*(b0[i] + b2[i])+W_CO)>>W_CS;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1176 #else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1177 int r= 3*(b0[i] + b2[i]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1178 r+= r>>4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1179 r+= r>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1180 b1[i] -= (r+W_CO)>>W_CS;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1181 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1182 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1183 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1184
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1185 static void vertical_compose97iL0(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1186 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1187
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1188 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1189 b1[i] += (W_BM*(b0[i] + b2[i])+W_BO)>>W_BS;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1190 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1191 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1192
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1193 static void vertical_compose97iL1(int *b0, int *b1, int *b2, int width){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1194 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1195
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1196 for(i=0; i<width; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1197 b1[i] -= (W_DM*(b0[i] + b2[i])+W_DO)>>W_DS;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1198 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1199 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1200
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1201 static void spatial_compose97i(int *buffer, int width, int height, int stride){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1202 int x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1203 DWTELEM *b0= buffer + mirror(-3-1, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1204 DWTELEM *b1= buffer + mirror(-3 , height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1205 DWTELEM *b2= buffer + mirror(-3+1, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1206 DWTELEM *b3= buffer + mirror(-3+2, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1207
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1208 for(y=-3; y<=height; y+=2){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1209 DWTELEM *b4= buffer + mirror(y+3, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1210 DWTELEM *b5= buffer + mirror(y+4, height-1)*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1211
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1212 if(stride == width && y+4 < height && 0){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1213 int x;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1214 for(x=0; x<width/2; x++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1215 b5[x] += 64*2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1216 for(; x<width; x++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1217 b5[x] += 169*2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1218 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1219
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1220 {START_TIMER
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1221 if(b3 <= b5) vertical_compose97iL1(b3, b4, b5, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1222 if(b2 <= b4) vertical_compose97iH1(b2, b3, b4, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1223 if(b1 <= b3) vertical_compose97iL0(b1, b2, b3, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1224 if(b0 <= b2) vertical_compose97iH0(b0, b1, b2, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1225 if(width>400){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1226 STOP_TIMER("vertical_compose97i")}}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1227
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1228 {START_TIMER
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1229 if(y-1>= 0) horizontal_compose97i(b0, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1230 if(b0 <= b2) horizontal_compose97i(b1, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1231 if(width>400 && b0 <= b2){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1232 STOP_TIMER("horizontal_compose97i")}}
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1233
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1234 b0=b2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1235 b1=b3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1236 b2=b4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1237 b3=b5;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1238 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1239 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1240
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1241 static void spatial_idwt(SnowContext *s, int *buffer, int width, int height, int stride){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1242 int level;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1243
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1244 for(level=s->spatial_decomposition_count-1; level>=0; level--){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1245 switch(s->spatial_decomposition_type){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1246 case 0: spatial_compose97i(buffer, width>>level, height>>level, stride<<level); break;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1247 case 1: spatial_compose53i(buffer, width>>level, height>>level, stride<<level); break;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1248 case 2: spatial_composeX (buffer, width>>level, height>>level, stride<<level); break;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1249 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1250 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1251 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1252
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1253 static const int hilbert[16][2]={
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1254 {0,0}, {1,0}, {1,1}, {0,1},
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1255 {0,2}, {0,3}, {1,3}, {1,2},
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1256 {2,2}, {2,3}, {3,3}, {3,2},
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1257 {3,1}, {2,1}, {2,0}, {3,0},
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1258 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1259 #if 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1260 -o o-
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1261 | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1262 o-o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1263
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1264 -o-o o-o-
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1265 | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1266 o-o o-o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1267 | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1268 o o-o o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1269 | | | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1270 o-o o-o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1271
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1272 0112122312232334122323342334
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1273 0123456789ABCDEF0123456789AB
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1274 RLLRMRRLLRRMRLLMLRRLMLLRRLLM
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1275
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1276 4 B F 14 1B
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1277 4 11 15 20 27
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1278
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1279 -o o-o-o o-o-o o-
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1280 | | | | | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1281 o-o o-o o-o o-o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1282 | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1283 o-o o-o o-o o-o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1284 | | | | | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1285 o o-o-o o-o-o o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1286 | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1287 o-o o-o-o-o o-o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1288 | | | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1289 o-o o-o o-o o-o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1290 | | | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1291 o o-o o o o-o o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1292 | | | | | | | |
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1293 o-o o-o o-o o-o
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1294
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1295 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1296
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1297 #define SVI(a, i, x, y) \
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1298 {\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1299 a[i][0]= x;\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1300 a[i][1]= y;\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1301 i++;\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1302 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1303
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1304 static int sig_cmp(const void *a, const void *b){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1305 const int16_t* da = (const int16_t *) a;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1306 const int16_t* db = (const int16_t *) b;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1307
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1308 if(da[1] != db[1]) return da[1] - db[1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1309 else return da[0] - db[0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1310 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1311
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1312 static int deint(unsigned int a){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1313 a &= 0x55555555; //0 1 2 3 4 5 6 7 8 9 A B C D E F
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1314 a += a & 0x11111111; // 01 23 45 67 89 AB CD EF
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1315 a += 3*(a & 0x0F0F0F0F);// 0123 4567 89AB CDEF
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1316 a += 15*(a & 0x00FF00FF);// 01234567 89ABCDEF
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1317 a +=255*(a & 0x0000FFFF);// 0123456789ABCDEF
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1318 return a>>15;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1319 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1320
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1321 static void encode_subband_z0run(SnowContext *s, SubBand *b, DWTELEM *src, DWTELEM *parent, int stride, int orientation){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1322 const int level= b->level;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1323 const int w= b->width;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1324 const int h= b->height;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1325 int x, y, pos;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1326
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1327 if(1){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1328 int run=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1329 int runs[w*h];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1330 int run_index=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1331 int count=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1332
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1333 for(pos=0; ; pos++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1334 int x= deint(pos );
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1335 int y= deint(pos>>1);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1336 int v, p=0, pr=0, pd=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1337 int /*ll=0, */l=0, lt=0, t=0/*, rt=0*/;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1338
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1339 if(x>=w || y>=h){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1340 if(x>=w && y>=h)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1341 break;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1342 continue;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1343 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1344 count++;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1345
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1346 v= src[x + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1347
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1348 if(y){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1349 t= src[x + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1350 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1351 lt= src[x - 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1352 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1353 if(x + 1 < w){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1354 /*rt= src[x + 1 + (y-1)*stride]*/;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1355 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1356 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1357 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1358 l= src[x - 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1359 /*if(x > 1){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1360 if(orientation==1) ll= src[y + (x-2)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1361 else ll= src[x - 2 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1362 }*/
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1363 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1364 if(parent){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1365 int px= x>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1366 int py= y>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1367 if(px<b->parent->width && py<b->parent->height){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1368 p= parent[px + py*2*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1369 /*if(px+1<b->parent->width)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1370 pr= parent[px + 1 + py*2*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1371 if(py+1<b->parent->height)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1372 pd= parent[px + (py+1)*2*stride];*/
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1373 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1374 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1375 if(!(/*ll|*/l|lt|t|/*rt|*/p)){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1376 if(v){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1377 runs[run_index++]= run;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1378 run=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1379 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1380 run++;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1381 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1382 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1383 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1384 assert(count==w*h);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1385 runs[run_index++]= run;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1386 run_index=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1387 run= runs[run_index++];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1388
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1389 put_symbol(&s->c, b->state[1], run, 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1390
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1391 for(pos=0; ; pos++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1392 int x= deint(pos );
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1393 int y= deint(pos>>1);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1394 int v, p=0, pr=0, pd=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1395 int /*ll=0, */l=0, lt=0, t=0/*, rt=0*/;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1396
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1397 if(x>=w || y>=h){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1398 if(x>=w && y>=h)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1399 break;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1400 continue;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1401 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1402 v= src[x + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1403
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1404 if(y){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1405 t= src[x + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1406 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1407 lt= src[x - 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1408 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1409 if(x + 1 < w){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1410 // rt= src[x + 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1411 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1412 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1413 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1414 l= src[x - 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1415 /*if(x > 1){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1416 if(orientation==1) ll= src[y + (x-2)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1417 else ll= src[x - 2 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1418 }*/
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1419 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1420
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1421 if(parent){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1422 int px= x>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1423 int py= y>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1424 if(px<b->parent->width && py<b->parent->height){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1425 p= parent[px + py*2*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1426 /* if(px+1<b->parent->width)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1427 pr= parent[px + 1 + py*2*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1428 if(py+1<b->parent->height)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1429 pd= parent[px + (py+1)*2*stride];*/
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1430 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1431 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1432 if(/*ll|*/l|lt|t|/*rt|*/p){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1433 int context= av_log2(/*ABS(ll) + */2*(3*ABS(l) + ABS(lt) + 2*ABS(t) + /*ABS(rt) +*/ ABS(p)));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1434
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1435 put_cabac(&s->c, &b->state[0][context], !!v);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1436 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1437 if(!run){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1438 run= runs[run_index++];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1439 put_symbol(&s->c, b->state[1], run, 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1440 assert(v);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1441 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1442 run--;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1443 assert(!v);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1444 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1445 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1446 if(v){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1447 int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + /*ABS(rt) +*/ ABS(p));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1448
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1449 put_symbol(&s->c, b->state[context + 2], ABS(v)-1, 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1450 put_cabac(&s->c, &b->state[0][16 + 1 + 3 + quant3b[l&0xFF] + 3*quant3b[t&0xFF]], v<0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1451 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1452 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1453 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1454 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1455
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1456 static void encode_subband_bp(SnowContext *s, SubBand *b, DWTELEM *src, DWTELEM *parent, int stride, int orientation){
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1457 const int level= b->level;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1458 const int w= b->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1459 const int h= b->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1460 int x, y;
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1461
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1462 #if 0
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1463 int plane;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1464 for(plane=24; plane>=0; plane--){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1465 int run=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1466 int runs[w*h];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1467 int run_index=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1468
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1469 for(y=0; y<h; y++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1470 for(x=0; x<w; x++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1471 int v, lv, p=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1472 int d=0, r=0, rd=0, ld=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1473 int /*ll=0, */l=0, lt=0, t=0, rt=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1474 v= src[x + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1475
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1476 if(y){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1477 t= src[x + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1478 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1479 lt= src[x - 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1480 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1481 if(x + 1 < w){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1482 rt= src[x + 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1483 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1484 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1485 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1486 l= src[x - 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1487 /*if(x > 1){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1488 if(orientation==1) ll= src[y + (x-2)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1489 else ll= src[x - 2 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1490 }*/
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1491 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1492 if(y+1<h){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1493 d= src[x + (y+1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1494 if(x) ld= src[x - 1 + (y+1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1495 if(x + 1 < w) rd= src[x + 1 + (y+1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1496 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1497 if(x + 1 < w)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1498 r= src[x + 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1499 if(parent){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1500 int px= x>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1501 int py= y>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1502 if(px<b->parent->width && py<b->parent->height)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1503 p= parent[px + py*2*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1504 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1505 #define HIDE(c, plane) c= c>=0 ? c&((-1)<<(plane)) : -((-c)&((-1)<<(plane)));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1506 lv=v;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1507 HIDE( v, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1508 HIDE(lv, plane+1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1509 HIDE( p, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1510 HIDE( l, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1511 HIDE(lt, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1512 HIDE( t, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1513 HIDE(rt, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1514 HIDE( r, plane+1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1515 HIDE(rd, plane+1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1516 HIDE( d, plane+1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1517 HIDE(ld, plane+1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1518 if(!(/*ll|*/l|lt|t|rt|r|rd|ld|d|p|lv)){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1519 if(v){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1520 runs[run_index++]= run;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1521 run=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1522 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1523 run++;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1524 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1525 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1526 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1527 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1528 runs[run_index++]= run;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1529 run_index=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1530 run= runs[run_index++];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1531
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1532 put_symbol(&s->c, b->state[1], run, 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1533
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1534 for(y=0; y<h; y++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1535 for(x=0; x<w; x++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1536 int v, p=0, lv;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1537 int /*ll=0, */l=0, lt=0, t=0, rt=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1538 int d=0, r=0, rd=0, ld=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1539 v= src[x + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1540
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1541 if(y){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1542 t= src[x + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1543 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1544 lt= src[x - 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1545 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1546 if(x + 1 < w){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1547 rt= src[x + 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1548 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1549 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1550 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1551 l= src[x - 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1552 /*if(x > 1){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1553 if(orientation==1) ll= src[y + (x-2)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1554 else ll= src[x - 2 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1555 }*/
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1556 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1557 if(y+1<h){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1558 d= src[x + (y+1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1559 if(x) ld= src[x - 1 + (y+1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1560 if(x + 1 < w) rd= src[x + 1 + (y+1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1561 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1562 if(x + 1 < w)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1563 r= src[x + 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1564
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1565 if(parent){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1566 int px= x>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1567 int py= y>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1568 if(px<b->parent->width && py<b->parent->height)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1569 p= parent[px + py*2*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1570 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1571 lv=v;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1572 HIDE( v, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1573 HIDE(lv, plane+1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1574 HIDE( p, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1575 HIDE( l, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1576 HIDE(lt, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1577 HIDE( t, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1578 HIDE(rt, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1579 HIDE( r, plane+1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1580 HIDE(rd, plane+1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1581 HIDE( d, plane+1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1582 HIDE(ld, plane+1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1583 if(/*ll|*/l|lt|t|rt|r|rd|ld|d|p|lv){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1584 int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1585 +3*ABS(r) + ABS(rd) + 2*ABS(d) + ABS(ld));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1586
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1587 if(lv) put_cabac(&s->c, &b->state[99][context + 8*(av_log2(ABS(lv))-plane)], !!(v-lv));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1588 else put_cabac(&s->c, &b->state[ 0][context], !!v);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1589 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1590 assert(!lv);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1591 if(!run){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1592 run= runs[run_index++];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1593 put_symbol(&s->c, b->state[1], run, 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1594 assert(v);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1595 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1596 run--;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1597 assert(!v);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1598 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1599 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1600 if(v && !lv){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1601 int context= clip(quant3b[l&0xFF] + quant3b[r&0xFF], -1,1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1602 + 3*clip(quant3b[t&0xFF] + quant3b[d&0xFF], -1,1);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1603 put_cabac(&s->c, &b->state[0][16 + 1 + 3 + context], v<0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1604 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1605 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1606 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1607 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1608 return;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1609 #endif
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1610 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1611
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1612 static void encode_subband_X(SnowContext *s, SubBand *b, DWTELEM *src, DWTELEM *parent, int stride, int orientation){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1613 const int level= b->level;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1614 const int w= b->width;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1615 const int h= b->height;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1616 int x, y;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1617
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1618 #if 0
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1619 if(orientation==3 && parent && 0){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1620 int16_t candidate[w*h][2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1621 uint8_t state[w*h];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1622 int16_t boarder[3][w*h*4][2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1623 int16_t significant[w*h][2];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1624 int candidate_count=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1625 int boarder_count[3]={0,0,0};
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1626 int significant_count=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1627 int rle_pos=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1628 int v, last_v;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1629 int primary= orientation==1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1630
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1631 memset(candidate, 0, sizeof(candidate));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1632 memset(state, 0, sizeof(state));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1633 memset(boarder, 0, sizeof(boarder));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1634
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1635 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1636 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1637 if(parent[(x>>1) + (y>>1)*2*stride])
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1638 SVI(candidate, candidate_count, x, y)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1639 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1640 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1641
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1642 for(;;){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1643 while(candidate_count && !boarder_count[0] && !boarder_count[1] && !boarder_count[2]){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1644 candidate_count--;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1645 x= candidate[ candidate_count][0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1646 y= candidate[ candidate_count][1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1647 if(state[x + y*w])
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1648 continue;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1649 state[x + y*w]= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1650 v= !!src[x + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1651 put_cabac(&s->c, &b->state[0][0], v);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1652 if(v){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1653 SVI(significant, significant_count, x,y)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1654 if(x && !state[x - 1 + y *w]) SVI(boarder[0],boarder_count[0],x-1,y )
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1655 if(y && !state[x + (y-1)*w]) SVI(boarder[1],boarder_count[1],x ,y-1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1656 if(x+1<w && !state[x + 1 + y *w]) SVI(boarder[0],boarder_count[0],x+1,y )
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1657 if(y+1<h && !state[x + (y+1)*w]) SVI(boarder[1],boarder_count[1],x ,y+1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1658 if(x && y && !state[x - 1 + (y-1)*w]) SVI(boarder[2],boarder_count[2],x-1,y-1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1659 if(x && y+1<h && !state[x - 1 + (y+1)*w]) SVI(boarder[2],boarder_count[2],x-1,y+1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1660 if(x+1<w && y+1<h && !state[x + 1 + (y+1)*w]) SVI(boarder[2],boarder_count[2],x+1,y+1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1661 if(x+1<w && y && !state[x + 1 + (y-1)*w]) SVI(boarder[2],boarder_count[2],x+1,y-1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1662 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1663 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1664 while(!boarder_count[0] && !boarder_count[1] && !boarder_count[2] && rle_pos < w*h){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1665 int run=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1666 for(; rle_pos < w*h;){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1667 x= rle_pos % w; //FIXME speed
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1668 y= rle_pos / w;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1669 rle_pos++;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1670 if(state[x + y*w])
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1671 continue;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1672 state[x + y*w]= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1673 v= !!src[x + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1674 if(v){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1675 put_symbol(&s->c, b->state[1], run, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1676 SVI(significant, significant_count, x,y)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1677 if(x && !state[x - 1 + y *w]) SVI(boarder[0],boarder_count[0],x-1,y )
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1678 if(y && !state[x + (y-1)*w]) SVI(boarder[1],boarder_count[1],x ,y-1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1679 if(x+1<w && !state[x + 1 + y *w]) SVI(boarder[0],boarder_count[0],x+1,y )
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1680 if(y+1<h && !state[x + (y+1)*w]) SVI(boarder[1],boarder_count[1],x ,y+1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1681 if(x && y && !state[x - 1 + (y-1)*w]) SVI(boarder[2],boarder_count[2],x-1,y-1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1682 if(x && y+1<h && !state[x - 1 + (y+1)*w]) SVI(boarder[2],boarder_count[2],x-1,y+1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1683 if(x+1<w && y+1<h && !state[x + 1 + (y+1)*w]) SVI(boarder[2],boarder_count[2],x+1,y+1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1684 if(x+1<w && y && !state[x + 1 + (y-1)*w]) SVI(boarder[2],boarder_count[2],x+1,y-1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1685 break;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1686 //FIXME note only right & down can be boarders
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1687 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1688 run++;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1689 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1690 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1691 if(!boarder_count[0] && !boarder_count[1] && !boarder_count[2])
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1692 break;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1693
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1694 while(boarder_count[0] || boarder_count[1] || boarder_count[2]){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1695 int index;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1696
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1697 if (boarder_count[ primary]) index= primary;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1698 else if(boarder_count[1-primary]) index=1-primary;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1699 else index=2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1700
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1701 boarder_count[index]--;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1702 x= boarder[index][ boarder_count[index] ][0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1703 y= boarder[index][ boarder_count[index] ][1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1704 if(state[x + y*w]) //FIXME maybe check earlier
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1705 continue;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1706 state[x + y*w]= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1707 v= !!src[x + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1708 put_cabac(&s->c, &b->state[0][index+1], v);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1709 if(v){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1710 SVI(significant, significant_count, x,y)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1711 if(x && !state[x - 1 + y *w]) SVI(boarder[0],boarder_count[0],x-1,y )
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1712 if(y && !state[x + (y-1)*w]) SVI(boarder[1],boarder_count[1],x ,y-1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1713 if(x+1<w && !state[x + 1 + y *w]) SVI(boarder[0],boarder_count[0],x+1,y )
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1714 if(y+1<h && !state[x + (y+1)*w]) SVI(boarder[1],boarder_count[1],x ,y+1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1715 if(x && y && !state[x - 1 + (y-1)*w]) SVI(boarder[2],boarder_count[2],x-1,y-1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1716 if(x && y+1<h && !state[x - 1 + (y+1)*w]) SVI(boarder[2],boarder_count[2],x-1,y+1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1717 if(x+1<w && y+1<h && !state[x + 1 + (y+1)*w]) SVI(boarder[2],boarder_count[2],x+1,y+1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1718 if(x+1<w && y && !state[x + 1 + (y-1)*w]) SVI(boarder[2],boarder_count[2],x+1,y-1)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1719 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1720 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1721 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1722 //FIXME sort significant coeffs maybe
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1723 if(1){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1724 qsort(significant, significant_count, sizeof(int16_t[2]), sig_cmp);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1725 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1726
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1727 last_v=1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1728 while(significant_count){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1729 int context= 3 + quant7[last_v&0xFF]; //use significance of suroundings
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1730 significant_count--;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1731 x= significant[significant_count][0];//FIXME try opposit direction
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1732 y= significant[significant_count][1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1733 v= src[x + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1734 put_symbol(&s->c, b->state[context + 2], v, 1); //FIXME try to avoid first bit, try this with the old code too!!
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1735 last_v= v;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1736 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1737 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1738 #endif
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1739 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1740
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1741 static void encode_subband_c0run(SnowContext *s, SubBand *b, DWTELEM *src, DWTELEM *parent, int stride, int orientation){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1742 const int level= b->level;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1743 const int w= b->width;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1744 const int h= b->height;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1745 int x, y;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1746
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1747 if(1){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1748 int run=0;
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1749 int runs[w*h];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1750 int run_index=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1751
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1752 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1753 for(x=0; x<w; x++){
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1754 int v, p=0;
2144
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1755 int /*ll=0, */l=0, lt=0, t=0, rt=0;
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1756 v= src[x + y*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1757
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1758 if(y){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1759 t= src[x + (y-1)*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1760 if(x){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1761 lt= src[x - 1 + (y-1)*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1762 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1763 if(x + 1 < w){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1764 rt= src[x + 1 + (y-1)*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1765 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1766 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1767 if(x){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1768 l= src[x - 1 + y*stride];
2144
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1769 /*if(x > 1){
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1770 if(orientation==1) ll= src[y + (x-2)*stride];
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1771 else ll= src[x - 2 + y*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1772 }*/
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1773 }
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1774 if(parent){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1775 int px= x>>1;
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1776 int py= y>>1;
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1777 if(px<b->parent->width && py<b->parent->height)
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1778 p= parent[px + py*2*stride];
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1779 }
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1780 if(!(/*ll|*/l|lt|t|rt|p)){
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1781 if(v){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1782 runs[run_index++]= run;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1783 run=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1784 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1785 run++;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1786 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1787 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1788 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1789 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1790 runs[run_index++]= run;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1791 run_index=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1792 run= runs[run_index++];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1793
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1794 put_symbol2(&s->c, b->state[1], run, 3);
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1795
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1796 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1797 for(x=0; x<w; x++){
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1798 int v, p=0;
2144
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1799 int /*ll=0, */l=0, lt=0, t=0, rt=0;
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1800 v= src[x + y*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1801
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1802 if(y){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1803 t= src[x + (y-1)*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1804 if(x){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1805 lt= src[x - 1 + (y-1)*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1806 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1807 if(x + 1 < w){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1808 rt= src[x + 1 + (y-1)*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1809 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1810 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1811 if(x){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1812 l= src[x - 1 + y*stride];
2144
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1813 /*if(x > 1){
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1814 if(orientation==1) ll= src[y + (x-2)*stride];
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1815 else ll= src[x - 2 + y*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1816 }*/
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1817 }
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1818 if(parent){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1819 int px= x>>1;
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
1820 int py= y>>1;
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1821 if(px<b->parent->width && py<b->parent->height)
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1822 p= parent[px + py*2*stride];
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1823 }
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1824 if(/*ll|*/l|lt|t|rt|p){
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1825 int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p));
2144
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1826
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1827 put_cabac(&s->c, &b->state[0][context], !!v);
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1828 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1829 if(!run){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1830 run= runs[run_index++];
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1831
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1832 put_symbol2(&s->c, b->state[1], run, 3);
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1833 assert(v);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1834 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1835 run--;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1836 assert(!v);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1837 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1838 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1839 if(v){
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
1840 int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p));
2144
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1841
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
1842 put_symbol(&s->c, b->state[context + 2], ABS(v)-1, 0);
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1843 put_cabac(&s->c, &b->state[0][16 + 1 + 3 + quant3b[l&0xFF] + 3*quant3b[t&0xFF]], v<0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1844 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1845 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1846 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1847 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1848 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
1849
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1850 static void encode_subband_dzr(SnowContext *s, SubBand *b, DWTELEM *src, DWTELEM *parent, int stride, int orientation){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1851 const int level= b->level;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1852 const int w= b->width;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1853 const int h= b->height;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1854 int x, y;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1855
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1856 if(1){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1857 int run[16]={0};
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1858 int runs[16][w*h]; //FIXME do something about the size
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1859 int run_index[16]={0};
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1860 int positions[2][w];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1861 int distances[2][w];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1862 int dist_count=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1863 int i;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1864
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1865 for(y=0; y<h; y++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1866 int * pos = positions[ y&1];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1867 int *last_pos = positions[(y&1)^1];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1868 int * dist= distances[ y&1];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1869 int *last_dist= distances[(y&1)^1];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1870 int dist_index=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1871 int last_dist_index=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1872
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1873 for(x=0; x<w; x++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1874 int p=0, l=0, lt=0, t=0, rt=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1875 int v= src[x + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1876
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1877 if(y){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1878 t= src[x + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1879 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1880 lt= src[x - 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1881 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1882 if(x + 1 < w){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1883 rt= src[x + 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1884 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1885 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1886 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1887 l= src[x - 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1888 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1889 if(parent){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1890 int px= x>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1891 int py= y>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1892 if(px<b->parent->width && py<b->parent->height)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1893 p= parent[px + py*2*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1894 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1895 if(last_dist_index < dist_count && last_pos[last_dist_index] == x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1896 if(dist_index==0 || x - pos[dist_index-1] > dist[dist_index-1] - last_dist[last_dist_index]){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1897 pos[dist_index]= x;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1898 dist[dist_index++]= last_dist[last_dist_index];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1899 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1900 last_dist_index++;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1901 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1902
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1903 if(!(l|lt|t|rt|p)){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1904 int cur_dist=w>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1905 int run_class;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1906
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1907 if(last_dist_index < dist_count)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1908 cur_dist= last_pos[last_dist_index] - x + y - last_dist[last_dist_index];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1909 if(dist_index)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1910 cur_dist= FFMIN(cur_dist, x - pos[dist_index-1] + y - dist[dist_index-1]);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1911 assert(cur_dist>=2);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1912 run_class= av_log2(cur_dist+62);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1913
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1914 if(v){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1915 runs[run_class][run_index[run_class]++]= run[run_class];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1916 run[run_class]=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1917 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1918 run[run_class]++;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1919 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1920 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1921 if(v){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1922 while(dist_index && x - pos[dist_index-1] <= y - dist[dist_index-1])
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1923 dist_index--;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1924 pos[dist_index]= x;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1925 dist[dist_index++]= y;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1926 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1927 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1928 dist_count= dist_index;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1929 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1930 for(i=0; i<12; i++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1931 runs[i][run_index[i]++]= run[i];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1932 run_index[i]=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1933 run[i]=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1934 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1935
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1936 dist_count=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1937
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1938 for(y=0; y<h; y++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1939 int * pos = positions[ y&1];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1940 int *last_pos = positions[(y&1)^1];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1941 int * dist= distances[ y&1];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1942 int *last_dist= distances[(y&1)^1];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1943 int dist_index=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1944 int last_dist_index=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1945
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1946 for(x=0; x<w; x++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1947 int p=0, l=0, lt=0, t=0, rt=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1948 int v= src[x + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1949
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1950 if(y){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1951 t= src[x + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1952 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1953 lt= src[x - 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1954 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1955 if(x + 1 < w){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1956 rt= src[x + 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1957 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1958 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1959 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1960 l= src[x - 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1961 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1962 if(parent){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1963 int px= x>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1964 int py= y>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1965 if(px<b->parent->width && py<b->parent->height)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1966 p= parent[px + py*2*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1967 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1968 if(last_dist_index < dist_count && last_pos[last_dist_index] == x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1969 if(dist_index==0 || x - pos[dist_index-1] > dist[dist_index-1] - last_dist[last_dist_index]){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1970 pos[dist_index]= x;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1971 dist[dist_index++]= last_dist[last_dist_index];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1972 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1973 last_dist_index++;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1974 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1975 if(l|lt|t|rt|p){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1976 int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1977
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1978 put_cabac(&s->c, &b->state[0][context], !!v);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1979 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1980 int cur_dist=w>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1981 int run_class;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1982
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1983 if(last_dist_index < dist_count)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1984 cur_dist= last_pos[last_dist_index] - x + y - last_dist[last_dist_index];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1985 if(dist_index)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1986 cur_dist= FFMIN(cur_dist, x - pos[dist_index-1] + y - dist[dist_index-1]);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1987 assert(cur_dist>=2);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1988 assert(!dist_index || (pos[dist_index-1] >= 0 && pos[dist_index-1] <w));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1989 assert(last_dist_index >= dist_count || (last_pos[last_dist_index] >= 0 && last_pos[last_dist_index] <w));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1990 assert(!dist_index || dist[dist_index-1] <= y);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1991 assert(last_dist_index >= dist_count || last_dist[last_dist_index] < y);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1992 assert(cur_dist <= y + FFMAX(x, w-x-1));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1993 run_class= av_log2(cur_dist+62);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1994
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1995 if(!run_index[run_class]){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1996 run[run_class]= runs[run_class][run_index[run_class]++];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1997 put_symbol(&s->c, b->state[run_class+1], run[run_class], 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1998 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
1999 if(!run[run_class]){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2000 run[run_class]= runs[run_class][run_index[run_class]++];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2001 put_symbol(&s->c, b->state[run_class+1], run[run_class], 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2002 assert(v);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2003 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2004 run[run_class]--;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2005 assert(!v);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2006 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2007 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2008 if(v){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2009 int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2010
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2011 put_symbol(&s->c, b->state[context + 16], ABS(v)-1, 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2012 put_cabac(&s->c, &b->state[0][16 + 1 + 3 + quant3b[l&0xFF] + 3*quant3b[t&0xFF]], v<0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2013
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2014 while(dist_index && x - pos[dist_index-1] <= y - dist[dist_index-1])
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2015 dist_index--;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2016 pos[dist_index]= x;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2017 dist[dist_index++]= y;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2018 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2019 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2020 dist_count= dist_index;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2021 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2022 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2023 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2024
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2025 static void encode_subband(SnowContext *s, SubBand *b, DWTELEM *src, DWTELEM *parent, int stride, int orientation){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2026 // encode_subband_qtree(s, b, src, parent, stride, orientation);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2027 // encode_subband_z0run(s, b, src, parent, stride, orientation);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2028 encode_subband_c0run(s, b, src, parent, stride, orientation);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2029 // encode_subband_dzr(s, b, src, parent, stride, orientation);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2030 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2031
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
2032 static inline void decode_subband(SnowContext *s, SubBand *b, DWTELEM *src, DWTELEM *parent, int stride, int orientation){
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2033 const int level= b->level;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2034 const int w= b->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2035 const int h= b->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2036 int x,y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2037
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2038 START_TIMER
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2039 #if 0
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2040 for(y=0; y<b->height; y++)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2041 memset(&src[y*stride], 0, b->width*sizeof(DWTELEM));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2042
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2043 int plane;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2044 for(plane=24; plane>=0; plane--){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2045 int run;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2046
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2047 run= get_symbol(&s->c, b->state[1], 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2048
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2049 #define HIDE(c, plane) c= c>=0 ? c&((-1)<<(plane)) : -((-c)&((-1)<<(plane)));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2050
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2051 for(y=0; y<h; y++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2052 for(x=0; x<w; x++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2053 int v, p=0, lv;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2054 int /*ll=0, */l=0, lt=0, t=0, rt=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2055 int d=0, r=0, rd=0, ld=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2056 lv= src[x + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2057
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2058 if(y){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2059 t= src[x + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2060 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2061 lt= src[x - 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2062 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2063 if(x + 1 < w){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2064 rt= src[x + 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2065 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2066 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2067 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2068 l= src[x - 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2069 /*if(x > 1){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2070 if(orientation==1) ll= src[y + (x-2)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2071 else ll= src[x - 2 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2072 }*/
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2073 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2074 if(y+1<h){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2075 d= src[x + (y+1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2076 if(x) ld= src[x - 1 + (y+1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2077 if(x + 1 < w) rd= src[x + 1 + (y+1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2078 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2079 if(x + 1 < w)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2080 r= src[x + 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2081
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2082 if(parent){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2083 int px= x>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2084 int py= y>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2085 if(px<b->parent->width && py<b->parent->height)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2086 p= parent[px + py*2*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2087 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2088 HIDE( p, plane)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2089 if(/*ll|*/l|lt|t|rt|r|rd|ld|d|p|lv){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2090 int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2091 +3*ABS(r) + ABS(rd) + 2*ABS(d) + ABS(ld));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2092
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2093 if(lv){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2094 assert(context + 8*av_log2(ABS(lv)) < 512 - 100);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2095 if(get_cabac(&s->c, &b->state[99][context + 8*(av_log2(ABS(lv))-plane)])){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2096 if(lv<0) v= lv - (1<<plane);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2097 else v= lv + (1<<plane);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2098 }else
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2099 v=lv;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2100 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2101 v= get_cabac(&s->c, &b->state[ 0][context]) << plane;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2102 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2103 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2104 assert(!lv);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2105 if(!run){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2106 run= get_symbol(&s->c, b->state[1], 0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2107 v= 1<<plane;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2108 }else{
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2109 run--;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2110 v=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2111 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2112 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2113 if(v && !lv){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2114 int context= clip(quant3b[l&0xFF] + quant3b[r&0xFF], -1,1)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2115 + 3*clip(quant3b[t&0xFF] + quant3b[d&0xFF], -1,1);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2116 if(get_cabac(&s->c, &b->state[0][16 + 1 + 3 + context]))
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2117 v= -v;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2118 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2119 src[x + y*stride]= v;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2120 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2121 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2122 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2123 return;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2124 #endif
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2125 #if 0
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2126 int tree[10][w*h]; //FIXME space waste ...
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2127 int treedim[10][2];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2128 int lev;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2129 const int max_level= av_log2(2*FFMAX(w,h)-1);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2130 int w2=w, h2=h;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2131 memset(tree, 0, sizeof(tree));
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2132
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2133 // assert(w%2==0 && h%2==0);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2134
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2135 for(lev=max_level; lev>=0; lev--){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2136 treedim[lev][0]= w2;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2137 treedim[lev][1]= h2;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2138 w2= (w2+1)>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2139 h2= (h2+1)>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2140 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2141
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2142 for(lev=0; lev<=max_level; lev++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2143 w2= treedim[lev][0];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2144 h2= treedim[lev][1];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2145 for(y=0; y<h2; y++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2146 for(x=0; x<w2; x++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2147 int l= 0, t=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2148 int context;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2149 if(lev && !tree[lev-1][x/2 + y/2*w])
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2150 continue;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2151
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2152 if(x) l= tree[lev][x - 1 + y*w];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2153 if(y) t= tree[lev][x + (y-1)*w];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2154
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2155 context= lev + 8*(!!l) + 16*(!!t);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2156 tree[lev][x + y*w]= get_cabac(&s->c, &b->state[98][context]);
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2157 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2158 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2159 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2160 if(1){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2161 for(y=0; y<b->height; y++)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2162 memset(&src[y*stride], 0, b->width*sizeof(DWTELEM));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2163
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2164 for(y=0; y<h; y++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2165 for(x=0; x<w; x++){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2166 int v, p=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2167 int /*ll=0, */l=0, lt=0, t=0, rt=0;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2168
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2169 if(y){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2170 t= src[x + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2171 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2172 lt= src[x - 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2173 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2174 if(x + 1 < w){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2175 rt= src[x + 1 + (y-1)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2176 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2177 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2178 if(x){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2179 l= src[x - 1 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2180 /*if(x > 1){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2181 if(orientation==1) ll= src[y + (x-2)*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2182 else ll= src[x - 2 + y*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2183 }*/
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2184 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2185 if(parent){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2186 int px= x>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2187 int py= y>>1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2188 if(px<b->parent->width && py<b->parent->height)
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2189 p= parent[px + py*2*stride];
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2190 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2191 if(tree[max_level][x + y*w]){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2192 int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p));
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2193 v= get_symbol(&s->c, b->state[context + 2], 0) + 1;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2194 if(get_cabac(&s->c, &b->state[0][16 + 1 + 3 + quant3b[l&0xFF] + 3*quant3b[t&0xFF]]))
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2195 v= -v;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2196 src[x + y*stride]= v;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2197 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2198 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2199 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2200 if(level+1 == s->spatial_decomposition_count){
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2201 STOP_TIMER("decode_subband")
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2202 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2203
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2204 return;
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2205 }
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2206 #endif
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2207 if(1){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2208 int run;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2209
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2210 for(y=0; y<b->height; y++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2211 memset(&src[y*stride], 0, b->width*sizeof(DWTELEM));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2212
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2213 run= get_symbol2(&s->c, b->state[1], 3);
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2214 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2215 for(x=0; x<w; x++){
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
2216 int v, p=0;
2144
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
2217 int /*ll=0, */l=0, lt=0, t=0, rt=0;
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2218
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2219 if(y){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
2220 t= src[x + (y-1)*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2221 if(x){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
2222 lt= src[x - 1 + (y-1)*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2223 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2224 if(x + 1 < w){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
2225 rt= src[x + 1 + (y-1)*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2226 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2227 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2228 if(x){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
2229 l= src[x - 1 + y*stride];
2144
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
2230 /*if(x > 1){
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
2231 if(orientation==1) ll= src[y + (x-2)*stride];
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
2232 else ll= src[x - 2 + y*stride];
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2233 }*/
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2234 }
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
2235 if(parent){
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
2236 int px= x>>1;
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
2237 int py= y>>1;
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
2238 if(px<b->parent->width && py<b->parent->height)
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
2239 p= parent[px + py*2*stride];
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
2240 }
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
2241 if(/*ll|*/l|lt|t|rt|p){
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
2242 int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p));
2144
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
2243
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
2244 v=get_cabac(&s->c, &b->state[0][context]);
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2245 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2246 if(!run){
2155
274a01d80f4a various subband encoders (all either worse or complicated so they are commented out)
michael
parents: 2152
diff changeset
2247 run= get_symbol2(&s->c, b->state[1], 3);
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2248 //FIXME optimize this here
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2249 //FIXME try to store a more naive run
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2250 v=1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2251 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2252 run--;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2253 v=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2254 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2255 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2256 if(v){
2148
678be5a8f282 use parent sample to predict significance & magnitude
michael
parents: 2146
diff changeset
2257 int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p));
2144
e33371c1e2b4 better context model (0.2-1% lower bitrate)
michael
parents: 2139
diff changeset
2258 v= get_symbol(&s->c, b->state[context + 2], 0) + 1;
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2259 if(get_cabac(&s->c, &b->state[0][16 + 1 + 3 + quant3b[l&0xFF] + 3*quant3b[t&0xFF]]))
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2260 v= -v;
2149
48dc4ec06e88 cleanup
michael
parents: 2148
diff changeset
2261 src[x + y*stride]= v;
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2262 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2263 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2264 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2265 if(level+1 == s->spatial_decomposition_count){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2266 STOP_TIMER("decode_subband")
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2267 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2268
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2269 return;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2270 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2271 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2272
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2273 static void reset_contexts(SnowContext *s){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2274 int plane_index, level, orientation;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2275
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2276 for(plane_index=0; plane_index<2; plane_index++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2277 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2278 for(orientation=level ? 1:0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2279 memset(s->plane[plane_index].band[level][orientation].state, 0, sizeof(s->plane[plane_index].band[level][orientation].state));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2280 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2281 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2282 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2283 memset(s->mb_band.state, 0, sizeof(s->mb_band.state));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2284 memset(s->mv_band[0].state, 0, sizeof(s->mv_band[0].state));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2285 memset(s->mv_band[1].state, 0, sizeof(s->mv_band[1].state));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2286 memset(s->header_state, 0, sizeof(s->header_state));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2287 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2288
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2289 static void mc_block(uint8_t *dst, uint8_t *src, uint8_t *tmp, int stride, int b_w, int b_h, int dx, int dy){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2290 int x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2291
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2292 for(y=0; y < b_h+5; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2293 for(x=0; x < b_w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2294 int a0= src[x + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2295 int a1= src[x + 1 + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2296 int a2= src[x + 2 + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2297 int a3= src[x + 3 + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2298 int a4= src[x + 4 + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2299 int a5= src[x + 5 + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2300 // int am= 9*(a1+a2) - (a0+a3);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2301 int am= 20*(a2+a3) - 5*(a1+a4) + (a0+a5);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2302 // int am= 18*(a2+a3) - 2*(a1+a4);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2303 // int aL= (-7*a0 + 105*a1 + 35*a2 - 5*a3)>>3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2304 // int aR= (-7*a3 + 105*a2 + 35*a1 - 5*a0)>>3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2305
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2306 // if(b_w==16) am= 8*(a1+a2);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2307
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2308 if(dx<8) tmp[x + y*stride]= (32*a2*( 8-dx) + am* dx + 128)>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2309 else tmp[x + y*stride]= ( am*(16-dx) + 32*a3*(dx-8) + 128)>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2310
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2311 /* if (dx< 4) tmp[x + y*stride]= (16*a1*( 4-dx) + aL* dx + 32)>>6;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2312 else if(dx< 8) tmp[x + y*stride]= ( aL*( 8-dx) + am*(dx- 4) + 32)>>6;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2313 else if(dx<12) tmp[x + y*stride]= ( am*(12-dx) + aR*(dx- 8) + 32)>>6;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2314 else tmp[x + y*stride]= ( aR*(16-dx) + 16*a2*(dx-12) + 32)>>6;*/
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2315 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2316 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2317 for(y=0; y < b_h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2318 for(x=0; x < b_w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2319 int a0= tmp[x + y *stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2320 int a1= tmp[x + (y + 1)*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2321 int a2= tmp[x + (y + 2)*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2322 int a3= tmp[x + (y + 3)*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2323 int a4= tmp[x + (y + 4)*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2324 int a5= tmp[x + (y + 5)*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2325 int am= 20*(a2+a3) - 5*(a1+a4) + (a0+a5);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2326 // int am= 18*(a2+a3) - 2*(a1+a4);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2327 /* int aL= (-7*a0 + 105*a1 + 35*a2 - 5*a3)>>3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2328 int aR= (-7*a3 + 105*a2 + 35*a1 - 5*a0)>>3;*/
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2329
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2330 // if(b_w==16) am= 8*(a1+a2);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2331
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2332 if(dy<8) dst[x + y*stride]= (32*a2*( 8-dy) + am* dy + 128)>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2333 else dst[x + y*stride]= ( am*(16-dy) + 32*a3*(dy-8) + 128)>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2334
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2335 /* if (dy< 4) tmp[x + y*stride]= (16*a1*( 4-dy) + aL* dy + 32)>>6;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2336 else if(dy< 8) tmp[x + y*stride]= ( aL*( 8-dy) + am*(dy- 4) + 32)>>6;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2337 else if(dy<12) tmp[x + y*stride]= ( am*(12-dy) + aR*(dy- 8) + 32)>>6;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2338 else tmp[x + y*stride]= ( aR*(16-dy) + 16*a2*(dy-12) + 32)>>6;*/
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2339 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2340 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2341 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2342
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2343 #define mcb(dx,dy,b_w)\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2344 static void mc_block ## dx ## dy(uint8_t *dst, uint8_t *src, int stride){\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2345 uint8_t tmp[stride*(b_w+5)];\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2346 mc_block(dst, src-2-2*stride, tmp, stride, b_w, b_w, dx, dy);\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2347 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2348
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2349 mcb( 0, 0,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2350 mcb( 4, 0,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2351 mcb( 8, 0,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2352 mcb(12, 0,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2353 mcb( 0, 4,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2354 mcb( 4, 4,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2355 mcb( 8, 4,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2356 mcb(12, 4,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2357 mcb( 0, 8,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2358 mcb( 4, 8,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2359 mcb( 8, 8,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2360 mcb(12, 8,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2361 mcb( 0,12,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2362 mcb( 4,12,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2363 mcb( 8,12,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2364 mcb(12,12,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2365
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2366 #define mca(dx,dy,b_w)\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2367 static void mc_block_hpel ## dx ## dy(uint8_t *dst, uint8_t *src, int stride, int h){\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2368 uint8_t tmp[stride*(b_w+5)];\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2369 assert(h==b_w);\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2370 mc_block(dst, src-2-2*stride, tmp, stride, b_w, b_w, dx, dy);\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2371 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2372
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2373 mca( 0, 0,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2374 mca( 8, 0,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2375 mca( 0, 8,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2376 mca( 8, 8,16)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2377
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2378 static void add_xblock(DWTELEM *dst, uint8_t *src, uint8_t *obmc, int s_x, int s_y, int b_w, int b_h, int mv_x, int mv_y, int w, int h, int dst_stride, int src_stride, int obmc_stride, int mb_type, int add){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2379 uint8_t tmp[src_stride*(b_h+5)]; //FIXME move to context to gurantee alignment
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2380 int x,y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2381
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2382 if(s_x<0){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2383 obmc -= s_x;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2384 b_w += s_x;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2385 s_x=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2386 }else if(s_x + b_w > w){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2387 b_w = w - s_x;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2388 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2389 if(s_y<0){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2390 obmc -= s_y*obmc_stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2391 b_h += s_y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2392 s_y=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2393 }else if(s_y + b_h> h){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2394 b_h = h - s_y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2395 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2396
2152
2f6518c0e6cc 1/0 fix
michael
parents: 2151
diff changeset
2397 if(b_w<=0 || b_h<=0) return;
2f6518c0e6cc 1/0 fix
michael
parents: 2151
diff changeset
2398
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2399 dst += s_x + s_y*dst_stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2400
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2401 if(mb_type==1){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2402 src += s_x + s_y*src_stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2403 for(y=0; y < b_h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2404 for(x=0; x < b_w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2405 if(add) dst[x + y*dst_stride] += obmc[x + y*obmc_stride] * 128 * (256/OBMC_MAX);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2406 else dst[x + y*dst_stride] -= obmc[x + y*obmc_stride] * 128 * (256/OBMC_MAX);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2407 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2408 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2409 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2410 int dx= mv_x&15;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2411 int dy= mv_y&15;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2412 // int dxy= (mv_x&1) + 2*(mv_y&1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2413
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2414 s_x += (mv_x>>4) - 2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2415 s_y += (mv_y>>4) - 2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2416 src += s_x + s_y*src_stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2417 //use dsputil
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2418
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2419 if( (unsigned)s_x >= w - b_w - 4
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2420 || (unsigned)s_y >= h - b_h - 4){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2421 ff_emulated_edge_mc(tmp + 32, src, src_stride, b_w+5, b_h+5, s_x, s_y, w, h);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2422 src= tmp + 32;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2423 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2424
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2425 if(mb_type==0){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2426 mc_block(tmp, src, tmp + 64+8, src_stride, b_w, b_h, dx, dy);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2427 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2428 int sum=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2429 for(y=0; y < b_h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2430 for(x=0; x < b_w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2431 sum += src[x+ y*src_stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2432 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2433 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2434 sum= (sum + b_h*b_w/2) / (b_h*b_w);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2435 for(y=0; y < b_h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2436 for(x=0; x < b_w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2437 tmp[x + y*src_stride]= sum;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2438 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2439 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2440 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2441
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2442 for(y=0; y < b_h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2443 for(x=0; x < b_w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2444 if(add) dst[x + y*dst_stride] += obmc[x + y*obmc_stride] * tmp[x + y*src_stride] * (256/OBMC_MAX);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2445 else dst[x + y*dst_stride] -= obmc[x + y*obmc_stride] * tmp[x + y*src_stride] * (256/OBMC_MAX);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2446 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2447 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2448 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2449 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2450
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2451 static void predict_plane(SnowContext *s, DWTELEM *buf, int plane_index, int add){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2452 Plane *p= &s->plane[plane_index];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2453 const int mb_w= s->mb_band.width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2454 const int mb_h= s->mb_band.height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2455 const int mb_stride= s->mb_band.stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2456 int x, y, mb_x, mb_y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2457 int scale = plane_index ? s->mv_scale : 2*s->mv_scale;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2458 int block_w = plane_index ? 8 : 16;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2459 uint8_t *obmc = plane_index ? obmc16 : obmc32;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2460 int obmc_stride= plane_index ? 16 : 32;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2461 int ref_stride= s->last_picture.linesize[plane_index];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2462 uint8_t *ref = s->last_picture.data[plane_index];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2463 int w= p->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2464 int h= p->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2465
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2466 if(s->avctx->debug&512){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2467 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2468 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2469 if(add) buf[x + y*w]+= 128*256;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2470 else buf[x + y*w]-= 128*256;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2471 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2472 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2473
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2474 return;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2475 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2476 for(mb_y=-1; mb_y<=mb_h; mb_y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2477 for(mb_x=-1; mb_x<=mb_w; mb_x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2478 int index= clip(mb_x, 0, mb_w-1) + clip(mb_y, 0, mb_h-1)*mb_stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2479
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2480 add_xblock(buf, ref, obmc,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2481 block_w*mb_x - block_w/2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2482 block_w*mb_y - block_w/2,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2483 2*block_w, 2*block_w,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2484 s->mv_band[0].buf[index]*scale, s->mv_band[1].buf[index]*scale,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2485 w, h,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2486 w, ref_stride, obmc_stride,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2487 s->mb_band.buf[index], add);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2488
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2489 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2490 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2491 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2492
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2493 static void quantize(SnowContext *s, SubBand *b, DWTELEM *src, int stride, int bias){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2494 const int level= b->level;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2495 const int w= b->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2496 const int h= b->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2497 const int qlog= clip(s->qlog + b->qlog, 0, 128);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2498 const int qmul= qexp[qlog&7]<<(qlog>>3);
2150
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2499 int x,y, thres1, thres2;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2500 START_TIMER
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2501
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2502 assert(QROOT==8);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2503
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2504 bias= bias ? 0 : (3*qmul)>>3;
2150
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2505 thres1= ((qmul - bias)>>QEXPSHIFT) - 1;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2506 thres2= 2*thres1;
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2507
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2508 if(!bias){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2509 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2510 for(x=0; x<w; x++){
2150
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2511 int i= src[x + y*stride];
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2512
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2513 if((unsigned)(i+thres1) > thres2){
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2514 if(i>=0){
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2515 i<<= QEXPSHIFT;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2516 i/= qmul; //FIXME optimize
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2517 src[x + y*stride]= i;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2518 }else{
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2519 i= -i;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2520 i<<= QEXPSHIFT;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2521 i/= qmul; //FIXME optimize
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2522 src[x + y*stride]= -i;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2523 }
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2524 }else
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2525 src[x + y*stride]= 0;
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2526 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2527 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2528 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2529 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2530 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2531 int i= src[x + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2532
2150
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2533 if((unsigned)(i+thres1) > thres2){
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2534 if(i>=0){
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2535 i<<= QEXPSHIFT;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2536 i= (i + bias) / qmul; //FIXME optimize
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2537 src[x + y*stride]= i;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2538 }else{
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2539 i= -i;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2540 i<<= QEXPSHIFT;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2541 i= (i + bias) / qmul; //FIXME optimize
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2542 src[x + y*stride]= -i;
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2543 }
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2544 }else
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2545 src[x + y*stride]= 0;
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2546 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2547 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2548 }
2150
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2549 if(level+1 == s->spatial_decomposition_count){
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2550 // STOP_TIMER("quantize")
8aff375a986b optimize quantizaton (about 3x faster)
michael
parents: 2149
diff changeset
2551 }
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2552 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2553
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2554 static void dequantize(SnowContext *s, SubBand *b, DWTELEM *src, int stride){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2555 const int level= b->level;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2556 const int w= b->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2557 const int h= b->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2558 const int qlog= clip(s->qlog + b->qlog, 0, 128);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2559 const int qmul= qexp[qlog&7]<<(qlog>>3);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2560 const int qadd= (s->qbias*qmul)>>QBIAS_SHIFT;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2561 int x,y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2562
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2563 assert(QROOT==8);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2564
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2565 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2566 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2567 int i= src[x + y*stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2568 if(i<0){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2569 src[x + y*stride]= -((-i*qmul + qadd)>>(QEXPSHIFT)); //FIXME try different bias
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2570 }else if(i>0){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2571 src[x + y*stride]= (( i*qmul + qadd)>>(QEXPSHIFT));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2572 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2573 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2574 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2575 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2576
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2577 static void decorrelate(SnowContext *s, SubBand *b, DWTELEM *src, int stride, int inverse, int use_median){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2578 const int w= b->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2579 const int h= b->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2580 int x,y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2581
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2582 for(y=h-1; y>=0; y--){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2583 for(x=w-1; x>=0; x--){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2584 int i= x + y*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2585
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2586 if(x){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2587 if(use_median){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2588 if(y && x+1<w) src[i] -= mid_pred(src[i - 1], src[i - stride], src[i - stride + 1]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2589 else src[i] -= src[i - 1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2590 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2591 if(y) src[i] -= mid_pred(src[i - 1], src[i - stride], src[i - 1] + src[i - stride] - src[i - 1 - stride]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2592 else src[i] -= src[i - 1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2593 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2594 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2595 if(y) src[i] -= src[i - stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2596 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2597 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2598 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2599 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2600
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2601 static void correlate(SnowContext *s, SubBand *b, DWTELEM *src, int stride, int inverse, int use_median){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2602 const int w= b->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2603 const int h= b->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2604 int x,y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2605
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2606 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2607 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2608 int i= x + y*stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2609
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2610 if(x){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2611 if(use_median){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2612 if(y && x+1<w) src[i] += mid_pred(src[i - 1], src[i - stride], src[i - stride + 1]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2613 else src[i] += src[i - 1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2614 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2615 if(y) src[i] += mid_pred(src[i - 1], src[i - stride], src[i - 1] + src[i - stride] - src[i - 1 - stride]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2616 else src[i] += src[i - 1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2617 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2618 }else{
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2619 if(y) src[i] += src[i - stride];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2620 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2621 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2622 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2623 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2624
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2625 static void encode_header(SnowContext *s){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2626 int plane_index, level, orientation;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2627
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2628 put_cabac(&s->c, s->header_state, s->keyframe); // state clearing stuff?
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2629 if(s->keyframe){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2630 put_symbol(&s->c, s->header_state, s->version, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2631 put_symbol(&s->c, s->header_state, s->temporal_decomposition_type, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2632 put_symbol(&s->c, s->header_state, s->temporal_decomposition_count, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2633 put_symbol(&s->c, s->header_state, s->spatial_decomposition_count, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2634 put_symbol(&s->c, s->header_state, s->colorspace_type, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2635 put_symbol(&s->c, s->header_state, s->b_width, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2636 put_symbol(&s->c, s->header_state, s->b_height, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2637 put_symbol(&s->c, s->header_state, s->chroma_h_shift, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2638 put_symbol(&s->c, s->header_state, s->chroma_v_shift, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2639 put_cabac(&s->c, s->header_state, s->spatial_scalability);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2640 // put_cabac(&s->c, s->header_state, s->rate_scalability);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2641
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2642 for(plane_index=0; plane_index<2; plane_index++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2643 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2644 for(orientation=level ? 1:0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2645 if(orientation==2) continue;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2646 put_symbol(&s->c, s->header_state, s->plane[plane_index].band[level][orientation].qlog, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2647 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2648 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2649 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2650 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2651 put_symbol(&s->c, s->header_state, s->spatial_decomposition_type, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2652 put_symbol(&s->c, s->header_state, s->qlog, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2653 put_symbol(&s->c, s->header_state, s->mv_scale, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2654 put_symbol(&s->c, s->header_state, s->qbias, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2655 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2656
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2657 static int decode_header(SnowContext *s){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2658 int plane_index, level, orientation;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2659
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2660 s->keyframe= get_cabac(&s->c, s->header_state);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2661 if(s->keyframe){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2662 s->version= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2663 if(s->version>0){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2664 av_log(s->avctx, AV_LOG_ERROR, "version %d not supported", s->version);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2665 return -1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2666 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2667 s->temporal_decomposition_type= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2668 s->temporal_decomposition_count= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2669 s->spatial_decomposition_count= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2670 s->colorspace_type= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2671 s->b_width= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2672 s->b_height= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2673 s->chroma_h_shift= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2674 s->chroma_v_shift= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2675 s->spatial_scalability= get_cabac(&s->c, s->header_state);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2676 // s->rate_scalability= get_cabac(&s->c, s->header_state);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2677
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2678 for(plane_index=0; plane_index<3; plane_index++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2679 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2680 for(orientation=level ? 1:0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2681 int q;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2682 if (plane_index==2) q= s->plane[1].band[level][orientation].qlog;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2683 else if(orientation==2) q= s->plane[plane_index].band[level][1].qlog;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2684 else q= get_symbol(&s->c, s->header_state, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2685 s->plane[plane_index].band[level][orientation].qlog= q;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2686 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2687 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2688 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2689 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2690
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2691 s->spatial_decomposition_type= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2692 if(s->spatial_decomposition_type > 2){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2693 av_log(s->avctx, AV_LOG_ERROR, "spatial_decomposition_type %d not supported", s->spatial_decomposition_type);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2694 return -1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2695 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2696
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2697 s->qlog= get_symbol(&s->c, s->header_state, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2698 s->mv_scale= get_symbol(&s->c, s->header_state, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2699 s->qbias= get_symbol(&s->c, s->header_state, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2700
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2701 return 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2702 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2703
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2704 static int common_init(AVCodecContext *avctx){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2705 SnowContext *s = avctx->priv_data;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2706 int width, height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2707 int level, orientation, plane_index, dec;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2708
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2709 s->avctx= avctx;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2710
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2711 dsputil_init(&s->dsp, avctx);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2712
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2713 #define mcf(dx,dy)\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2714 s->dsp.put_qpel_pixels_tab [0][dy+dx/4]=\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2715 s->dsp.put_no_rnd_qpel_pixels_tab[0][dy+dx/4]=\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2716 mc_block ## dx ## dy;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2717
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2718 mcf( 0, 0)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2719 mcf( 4, 0)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2720 mcf( 8, 0)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2721 mcf(12, 0)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2722 mcf( 0, 4)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2723 mcf( 4, 4)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2724 mcf( 8, 4)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2725 mcf(12, 4)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2726 mcf( 0, 8)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2727 mcf( 4, 8)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2728 mcf( 8, 8)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2729 mcf(12, 8)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2730 mcf( 0,12)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2731 mcf( 4,12)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2732 mcf( 8,12)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2733 mcf(12,12)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2734
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2735 #define mcfh(dx,dy)\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2736 s->dsp.put_pixels_tab [0][dy/4+dx/8]=\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2737 s->dsp.put_no_rnd_pixels_tab[0][dy/4+dx/8]=\
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2738 mc_block_hpel ## dx ## dy;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2739
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2740 mcfh(0, 0)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2741 mcfh(8, 0)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2742 mcfh(0, 8)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2743 mcfh(8, 8)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2744
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2745 dec= s->spatial_decomposition_count= 5;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2746 s->spatial_decomposition_type= avctx->prediction_method; //FIXME add decorrelator type r transform_type
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2747
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2748 s->chroma_h_shift= 1; //FIXME XXX
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2749 s->chroma_v_shift= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2750
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2751 // dec += FFMAX(s->chroma_h_shift, s->chroma_v_shift);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2752
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2753 s->b_width = (s->avctx->width +(1<<dec)-1)>>dec;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2754 s->b_height= (s->avctx->height+(1<<dec)-1)>>dec;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2755
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2756 s->spatial_dwt_buffer= av_mallocz(s->b_width*s->b_height*sizeof(DWTELEM)<<(2*dec));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2757 s->pred_buffer= av_mallocz(s->b_width*s->b_height*sizeof(DWTELEM)<<(2*dec));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2758
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2759 s->mv_scale= (s->avctx->flags & CODEC_FLAG_QPEL) ? 2 : 4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2760
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2761 for(plane_index=0; plane_index<3; plane_index++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2762 int w= s->avctx->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2763 int h= s->avctx->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2764
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2765 if(plane_index){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2766 w>>= s->chroma_h_shift;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2767 h>>= s->chroma_v_shift;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2768 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2769 s->plane[plane_index].width = w;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2770 s->plane[plane_index].height= h;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2771 av_log(NULL, AV_LOG_DEBUG, "%d %d\n", w, h);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2772 for(level=s->spatial_decomposition_count-1; level>=0; level--){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2773 for(orientation=level ? 1 : 0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2774 SubBand *b= &s->plane[plane_index].band[level][orientation];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2775
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2776 b->buf= s->spatial_dwt_buffer;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2777 b->level= level;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2778 b->stride= s->plane[plane_index].width << (s->spatial_decomposition_count - level);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2779 b->width = (w + !(orientation&1))>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2780 b->height= (h + !(orientation>1))>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2781
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2782 if(orientation&1) b->buf += (w+1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2783 if(orientation>1) b->buf += b->stride>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2784
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2785 if(level)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2786 b->parent= &s->plane[plane_index].band[level-1][orientation];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2787 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2788 w= (w+1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2789 h= (h+1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2790 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2791 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2792
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2793 //FIXME init_subband() ?
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2794 s->mb_band.stride= s->mv_band[0].stride= s->mv_band[1].stride=
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2795 s->mb_band.width = s->mv_band[0].width = s->mv_band[1].width = (s->avctx->width + 15)>>4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2796 s->mb_band.height= s->mv_band[0].height= s->mv_band[1].height= (s->avctx->height+ 15)>>4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2797 s->mb_band .buf= av_mallocz(s->mb_band .stride * s->mb_band .height*sizeof(DWTELEM));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2798 s->mv_band[0].buf= av_mallocz(s->mv_band[0].stride * s->mv_band[0].height*sizeof(DWTELEM));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2799 s->mv_band[1].buf= av_mallocz(s->mv_band[1].stride * s->mv_band[1].height*sizeof(DWTELEM));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2800
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2801 reset_contexts(s);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2802 /*
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2803 width= s->width= avctx->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2804 height= s->height= avctx->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2805
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2806 assert(width && height);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2807 */
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2808 s->avctx->get_buffer(s->avctx, &s->mconly_picture);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2809
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2810 return 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2811 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2812
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2813
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2814 static void calculate_vissual_weight(SnowContext *s, Plane *p){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2815 int width = p->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2816 int height= p->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2817 int i, level, orientation, x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2818
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2819 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2820 for(orientation=level ? 1 : 0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2821 SubBand *b= &p->band[level][orientation];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2822 DWTELEM *buf= b->buf;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2823 int64_t error=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2824
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2825 memset(s->spatial_dwt_buffer, 0, sizeof(int)*width*height);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2826 buf[b->width/2 + b->height/2*b->stride]= 256*256;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2827 spatial_idwt(s, s->spatial_dwt_buffer, width, height, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2828 for(y=0; y<height; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2829 for(x=0; x<width; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2830 int64_t d= s->spatial_dwt_buffer[x + y*width];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2831 error += d*d;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2832 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2833 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2834
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2835 b->qlog= (int)(log(352256.0/sqrt(error)) / log(pow(2.0, 1.0/QROOT))+0.5);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2836 av_log(NULL, AV_LOG_DEBUG, "%d %d %d\n", level, orientation, b->qlog/*, sqrt(error)*/);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2837 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2838 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2839 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2840
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2841 static int encode_init(AVCodecContext *avctx)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2842 {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2843 SnowContext *s = avctx->priv_data;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2844 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2845 int level, orientation, plane_index;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2846
2151
44c973bfec0f dont segfault without -strict -1
michael
parents: 2150
diff changeset
2847 if(avctx->strict_std_compliance >= 0){
44c973bfec0f dont segfault without -strict -1
michael
parents: 2150
diff changeset
2848 av_log(avctx, AV_LOG_ERROR, "this codec is under development, files encoded with it wont be decodeable with future versions!!!\n"
44c973bfec0f dont segfault without -strict -1
michael
parents: 2150
diff changeset
2849 "use vstrict=-1 to use it anyway\n");
44c973bfec0f dont segfault without -strict -1
michael
parents: 2150
diff changeset
2850 return -1;
44c973bfec0f dont segfault without -strict -1
michael
parents: 2150
diff changeset
2851 }
44c973bfec0f dont segfault without -strict -1
michael
parents: 2150
diff changeset
2852
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2853 common_init(avctx);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2854
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2855 s->version=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2856
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2857 s->m.me.scratchpad= av_mallocz((avctx->width+64)*2*16*2*sizeof(uint8_t));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2858 s->m.me.map = av_mallocz(ME_MAP_SIZE*sizeof(uint32_t));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2859 s->m.me.score_map = av_mallocz(ME_MAP_SIZE*sizeof(uint32_t));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2860 s->mb_type = av_mallocz((s->mb_band.width+1)*s->mb_band.height*sizeof(int16_t));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2861 s->mb_mean = av_mallocz((s->mb_band.width+1)*s->mb_band.height*sizeof(int8_t ));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2862 s->dummy = av_mallocz((s->mb_band.width+1)*s->mb_band.height*sizeof(int32_t));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2863 h263_encode_init(&s->m); //mv_penalty
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2864
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2865 for(plane_index=0; plane_index<3; plane_index++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2866 calculate_vissual_weight(s, &s->plane[plane_index]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2867 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2868
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2869
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2870 avctx->coded_frame= &s->current_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2871 switch(avctx->pix_fmt){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2872 // case PIX_FMT_YUV444P:
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2873 // case PIX_FMT_YUV422P:
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2874 case PIX_FMT_YUV420P:
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2875 case PIX_FMT_GRAY8:
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2876 // case PIX_FMT_YUV411P:
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2877 // case PIX_FMT_YUV410P:
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2878 s->colorspace_type= 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2879 break;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2880 /* case PIX_FMT_RGBA32:
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2881 s->colorspace= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2882 break;*/
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2883 default:
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2884 av_log(avctx, AV_LOG_ERROR, "format not supported\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2885 return -1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2886 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2887 // avcodec_get_chroma_sub_sample(avctx->pix_fmt, &s->chroma_h_shift, &s->chroma_v_shift);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2888 s->chroma_h_shift= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2889 s->chroma_v_shift= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2890 return 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2891 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2892
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2893 static int frame_start(SnowContext *s){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2894 AVFrame tmp;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2895
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2896 if(s->keyframe)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2897 reset_contexts(s);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2898
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2899 tmp= s->last_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2900 s->last_picture= s->current_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2901 s->current_picture= tmp;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2902
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2903 s->current_picture.reference= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2904 if(s->avctx->get_buffer(s->avctx, &s->current_picture) < 0){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2905 av_log(s->avctx, AV_LOG_ERROR, "get_buffer() failed\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2906 return -1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2907 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2908
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2909 return 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2910 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2911
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2912 static int encode_frame(AVCodecContext *avctx, unsigned char *buf, int buf_size, void *data){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2913 SnowContext *s = avctx->priv_data;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2914 CABACContext * const c= &s->c;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2915 AVFrame *pict = data;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2916 const int width= s->avctx->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2917 const int height= s->avctx->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2918 int used_count= 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2919 int log2_threshold, level, orientation, plane_index, i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2920
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2921 ff_init_cabac_encoder(c, buf, buf_size);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2922 ff_init_cabac_states(c, ff_h264_lps_range, ff_h264_mps_state, ff_h264_lps_state, 64);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2923
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2924 s->input_picture = *pict;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2925
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2926 memset(s->header_state, 0, sizeof(s->header_state));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2927
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2928 s->keyframe=avctx->gop_size==0 || avctx->frame_number % avctx->gop_size == 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2929 pict->pict_type= s->keyframe ? FF_I_TYPE : FF_P_TYPE;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2930
2146
b288cdc2ad68 log2() & bsd
michael
parents: 2144
diff changeset
2931 s->qlog= rint(QROOT*log(pict->quality / (float)FF_QP2LAMBDA)/log(2));
2138
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2932 //<64 >60
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2933 s->qlog += 61;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2934
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2935 for(i=0; i<s->mb_band.stride * s->mb_band.height; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2936 s->mb_band.buf[i]= s->keyframe;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2937 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2938
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2939 frame_start(s);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2940
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2941 if(pict->pict_type == P_TYPE){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2942 int block_width = (width +15)>>4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2943 int block_height= (height+15)>>4;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2944 int stride= s->current_picture.linesize[0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2945 uint8_t *src_plane= s->input_picture.data[0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2946 int src_stride= s->input_picture.linesize[0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2947 int x,y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2948
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2949 assert(s->current_picture.data[0]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2950 assert(s->last_picture.data[0]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2951
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2952 s->m.avctx= s->avctx;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2953 s->m.current_picture.data[0]= s->current_picture.data[0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2954 s->m. last_picture.data[0]= s-> last_picture.data[0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2955 s->m. new_picture.data[0]= s-> input_picture.data[0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2956 s->m.current_picture_ptr= &s->m.current_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2957 s->m. last_picture_ptr= &s->m. last_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2958 s->m.linesize=
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2959 s->m. last_picture.linesize[0]=
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2960 s->m. new_picture.linesize[0]=
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2961 s->m.current_picture.linesize[0]= stride;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2962 s->m.width = width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2963 s->m.height= height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2964 s->m.mb_width = block_width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2965 s->m.mb_height= block_height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2966 s->m.mb_stride= s->m.mb_width+1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2967 s->m.b8_stride= 2*s->m.mb_width+1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2968 s->m.f_code=1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2969 s->m.pict_type= pict->pict_type;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2970 s->m.me_method= s->avctx->me_method;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2971 s->m.me.scene_change_score=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2972 s->m.flags= s->avctx->flags;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2973 s->m.quarter_sample= (s->avctx->flags & CODEC_FLAG_QPEL)!=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2974 s->m.out_format= FMT_H263;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2975 s->m.unrestricted_mv= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2976
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2977 s->m.lambda= pict->quality * 3/2; //FIXME bug somewhere else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2978 s->m.qscale= (s->m.lambda*139 + FF_LAMBDA_SCALE*64) >> (FF_LAMBDA_SHIFT + 7);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2979 s->m.lambda2= (s->m.lambda*s->m.lambda + FF_LAMBDA_SCALE/2) >> FF_LAMBDA_SHIFT;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2980
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2981 if(!s->motion_val8){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2982 s->motion_val8 = av_mallocz(s->m.b8_stride*block_height*2*2*sizeof(int16_t));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2983 s->motion_val16= av_mallocz(s->m.mb_stride*block_height*2*sizeof(int16_t));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2984 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2985
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2986 s->m.mb_type= s->mb_type;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2987
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2988 //dummies, to avoid segfaults
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2989 s->m.current_picture.mb_mean = s->mb_mean;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2990 s->m.current_picture.mb_var = (int16_t*)s->dummy;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2991 s->m.current_picture.mc_mb_var= (int16_t*)s->dummy;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2992 s->m.current_picture.mb_type = s->dummy;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2993
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2994 s->m.current_picture.motion_val[0]= s->motion_val8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2995 s->m.p_mv_table= s->motion_val16;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2996 s->m.dsp= s->dsp; //move
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2997 ff_init_me(&s->m);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2998
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
2999
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3000 s->m.me.pre_pass=1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3001 s->m.me.dia_size= s->avctx->pre_dia_size;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3002 s->m.first_slice_line=1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3003 for(y= block_height-1; y >= 0; y--) {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3004 uint8_t src[stride*16];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3005
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3006 s->m.new_picture.data[0]= src - y*16*stride; //ugly
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3007 s->m.mb_y= y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3008 for(i=0; i<16 && i + 16*y<height; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3009 memcpy(&src[i*stride], &src_plane[(i+16*y)*src_stride], width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3010 for(x=width; x<16*block_width; x++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3011 src[i*stride+x]= src[i*stride+x-1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3012 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3013 for(; i<16 && i + 16*y<16*block_height; i++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3014 memcpy(&src[i*stride], &src[(i-1)*stride], 16*block_width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3015
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3016 for(x=block_width-1; x >=0 ;x--) {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3017 s->m.mb_x= x;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3018 ff_init_block_index(&s->m);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3019 ff_update_block_index(&s->m);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3020 ff_pre_estimate_p_frame_motion(&s->m, x, y);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3021 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3022 s->m.first_slice_line=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3023 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3024 s->m.me.pre_pass=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3025
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3026
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3027 s->m.me.dia_size= s->avctx->dia_size;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3028 s->m.first_slice_line=1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3029 for (y = 0; y < block_height; y++) {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3030 uint8_t src[stride*16];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3031
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3032 s->m.new_picture.data[0]= src - y*16*stride; //ugly
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3033 s->m.mb_y= y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3034
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3035 assert(width <= stride);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3036 assert(width <= 16*block_width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3037
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3038 for(i=0; i<16 && i + 16*y<height; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3039 memcpy(&src[i*stride], &src_plane[(i+16*y)*src_stride], width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3040 for(x=width; x<16*block_width; x++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3041 src[i*stride+x]= src[i*stride+x-1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3042 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3043 for(; i<16 && i + 16*y<16*block_height; i++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3044 memcpy(&src[i*stride], &src[(i-1)*stride], 16*block_width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3045
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3046 for (x = 0; x < block_width; x++) {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3047 int mb_xy= x + y*(s->mb_band.stride);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3048 s->m.mb_x= x;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3049 ff_init_block_index(&s->m);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3050 ff_update_block_index(&s->m);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3051
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3052 ff_estimate_p_frame_motion(&s->m, x, y);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3053
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3054 s->mb_band .buf[mb_xy]= (s->m.mb_type[x + y*s->m.mb_stride]&CANDIDATE_MB_TYPE_INTER)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3055 ? 0 : 2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3056 s->mv_band[0].buf[mb_xy]= s->motion_val16[x + y*s->m.mb_stride][0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3057 s->mv_band[1].buf[mb_xy]= s->motion_val16[x + y*s->m.mb_stride][1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3058
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3059 if(s->mb_band .buf[x + y*(s->mb_band.stride)]==2 && 0){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3060 int dc0=128, dc1=128, dc, dc2, dir;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3061 int offset= (s->avctx->flags & CODEC_FLAG_QPEL) ? 64 : 32;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3062
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3063 dc =s->mb_mean[x + y *s->m.mb_stride ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3064 if(x) dc0=s->mb_mean[x + y *s->m.mb_stride - 1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3065 if(y) dc1=s->mb_mean[x + (y-1)*s->m.mb_stride ];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3066 dc2= (dc0+dc1)>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3067 #if 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3068 if (ABS(dc0 - dc) < ABS(dc1 - dc) && ABS(dc0 - dc) < ABS(dc2 - dc))
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3069 dir= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3070 else if(ABS(dc0 - dc) >=ABS(dc1 - dc) && ABS(dc1 - dc) < ABS(dc2 - dc))
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3071 dir=-1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3072 else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3073 dir=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3074 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3075 if(ABS(dc0 - dc) < ABS(dc1 - dc) && x){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3076 s->mv_band[0].buf[mb_xy]= s->mv_band[0].buf[x + y*(s->mb_band.stride)-1] - offset;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3077 s->mv_band[1].buf[mb_xy]= s->mv_band[1].buf[x + y*(s->mb_band.stride)-1];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3078 s->mb_mean[x + y *s->m.mb_stride ]= dc0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3079 }else if(y){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3080 s->mv_band[0].buf[mb_xy]= s->mv_band[0].buf[x + (y-1)*(s->mb_band.stride)];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3081 s->mv_band[1].buf[mb_xy]= s->mv_band[1].buf[x + (y-1)*(s->mb_band.stride)] - offset;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3082 s->mb_mean[x + y *s->m.mb_stride ]= dc1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3083 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3084 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3085 // s->mb_band .buf[x + y*(s->mb_band.stride)]=1; //FIXME intra only test
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3086 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3087 s->m.first_slice_line=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3088 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3089 assert(s->m.pict_type == P_TYPE);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3090 if(s->m.me.scene_change_score > s->avctx->scenechange_threshold){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3091 s->m.pict_type=
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3092 pict->pict_type =I_TYPE;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3093 for(i=0; i<s->mb_band.stride * s->mb_band.height; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3094 s->mb_band.buf[i]= 1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3095 s->mv_band[0].buf[i]=
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3096 s->mv_band[1].buf[i]= 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3097 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3098 //printf("Scene change detected, encoding as I Frame %d %d\n", s->current_picture.mb_var_sum, s->current_picture.mc_mb_var_sum);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3099 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3100 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3101
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3102 s->m.first_slice_line=1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3103
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3104 s->qbias= pict->pict_type == P_TYPE ? 2 : 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3105
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3106 encode_header(s);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3107
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3108 decorrelate(s, &s->mb_band , s->mb_band .buf, s->mb_band .stride, 0, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3109 decorrelate(s, &s->mv_band[0], s->mv_band[0].buf, s->mv_band[0].stride, 0, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3110 decorrelate(s, &s->mv_band[1], s->mv_band[1].buf, s->mv_band[1].stride, 0 ,1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3111 encode_subband(s, &s->mb_band , s->mb_band .buf, NULL, s->mb_band .stride, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3112 encode_subband(s, &s->mv_band[0], s->mv_band[0].buf, NULL, s->mv_band[0].stride, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3113 encode_subband(s, &s->mv_band[1], s->mv_band[1].buf, NULL, s->mv_band[1].stride, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3114
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3115 //FIXME avoid this
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3116 correlate(s, &s->mb_band , s->mb_band .buf, s->mb_band .stride, 1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3117 correlate(s, &s->mv_band[0], s->mv_band[0].buf, s->mv_band[0].stride, 1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3118 correlate(s, &s->mv_band[1], s->mv_band[1].buf, s->mv_band[1].stride, 1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3119
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3120 for(plane_index=0; plane_index<3; plane_index++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3121 Plane *p= &s->plane[plane_index];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3122 int w= p->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3123 int h= p->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3124 int x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3125 int bits= put_bits_count(&s->c.pb);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3126
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3127 //FIXME optimize
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3128 #if QPRED
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3129 memset(s->pred_buffer, 0, sizeof(DWTELEM)*w*h);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3130 predict_plane(s, s->pred_buffer, plane_index, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3131 spatial_dwt(s, s->pred_buffer, w, h, w);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3132 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3133 for(orientation=level ? 1 : 0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3134 SubBand *b= &p->band[level][orientation];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3135 int delta= ((int)s->pred_buffer - (int)s->spatial_dwt_buffer)/sizeof(DWTELEM);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3136
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3137 quantize (s, b, b->buf + delta, b->stride, s->qbias);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3138 dequantize(s, b, b->buf + delta, b->stride);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3139 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3140 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3141 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3142 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3143 s->spatial_dwt_buffer[y*w + x]= pict->data[plane_index][y*pict->linesize[plane_index] + x]<<8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3144 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3145 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3146 spatial_dwt(s, s->spatial_dwt_buffer, w, h, w);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3147 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3148 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3149 s->spatial_dwt_buffer[y*w + x]-= s->pred_buffer[y*w + x];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3150 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3151 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3152 #else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3153 if(pict->data[plane_index]) //FIXME gray hack
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3154 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3155 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3156 s->spatial_dwt_buffer[y*w + x]= pict->data[plane_index][y*pict->linesize[plane_index] + x]<<8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3157 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3158 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3159 predict_plane(s, s->spatial_dwt_buffer, plane_index, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3160 spatial_dwt(s, s->spatial_dwt_buffer, w, h, w);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3161 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3162
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3163 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3164 for(orientation=level ? 1 : 0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3165 SubBand *b= &p->band[level][orientation];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3166
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3167 quantize(s, b, b->buf, b->stride, s->qbias);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3168 if(orientation==0)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3169 decorrelate(s, b, b->buf, b->stride, pict->pict_type == P_TYPE, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3170 encode_subband(s, b, b->buf, b->parent ? b->parent->buf : NULL, b->stride, orientation);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3171 assert(b->parent==NULL || b->parent->stride == b->stride*2);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3172 if(orientation==0)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3173 correlate(s, b, b->buf, b->stride, 1, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3174 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3175 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3176 // av_log(NULL, AV_LOG_DEBUG, "plane:%d bits:%d\n", plane_index, put_bits_count(&s->c.pb) - bits);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3177
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3178 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3179 for(orientation=level ? 1 : 0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3180 SubBand *b= &p->band[level][orientation];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3181
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3182 dequantize(s, b, b->buf, b->stride);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3183 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3184 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3185
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3186 #if QPRED
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3187 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3188 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3189 s->spatial_dwt_buffer[y*w + x]+= s->pred_buffer[y*w + x];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3190 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3191 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3192 spatial_idwt(s, s->spatial_dwt_buffer, w, h, w);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3193 #else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3194 spatial_idwt(s, s->spatial_dwt_buffer, w, h, w);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3195 predict_plane(s, s->spatial_dwt_buffer, plane_index, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3196 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3197 //FIXME optimize
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3198 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3199 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3200 int v= (s->spatial_dwt_buffer[y*w + x]+128)>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3201 if(v&(~255)) v= ~(v>>31);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3202 s->current_picture.data[plane_index][y*s->current_picture.linesize[plane_index] + x]= v;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3203 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3204 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3205 if(s->avctx->flags&CODEC_FLAG_PSNR){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3206 int64_t error= 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3207
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3208 if(pict->data[plane_index]) //FIXME gray hack
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3209 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3210 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3211 int d= s->spatial_dwt_buffer[y*w + x] - pict->data[plane_index][y*pict->linesize[plane_index] + x]*256;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3212 error += d*d;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3213 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3214 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3215 error= (error + 128*256)>>16;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3216 s->avctx->error[plane_index] += error;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3217 s->avctx->error[3] += error;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3218 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3219 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3220
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3221 if(s->last_picture.data[0])
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3222 avctx->release_buffer(avctx, &s->last_picture);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3223
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3224 emms_c();
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3225
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3226 return put_cabac_terminate(c, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3227 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3228
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3229 static void common_end(SnowContext *s){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3230 av_freep(&s->spatial_dwt_buffer);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3231 av_freep(&s->mb_band.buf);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3232 av_freep(&s->mv_band[0].buf);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3233 av_freep(&s->mv_band[1].buf);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3234
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3235 av_freep(&s->m.me.scratchpad);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3236 av_freep(&s->m.me.map);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3237 av_freep(&s->m.me.score_map);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3238 av_freep(&s->mb_type);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3239 av_freep(&s->mb_mean);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3240 av_freep(&s->dummy);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3241 av_freep(&s->motion_val8);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3242 av_freep(&s->motion_val16);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3243 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3244
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3245 static int encode_end(AVCodecContext *avctx)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3246 {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3247 SnowContext *s = avctx->priv_data;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3248
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3249 common_end(s);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3250
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3251 return 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3252 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3253
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3254 static int decode_init(AVCodecContext *avctx)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3255 {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3256 // SnowContext *s = avctx->priv_data;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3257
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3258 common_init(avctx);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3259
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3260 return 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3261 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3262
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3263 static int decode_frame(AVCodecContext *avctx, void *data, int *data_size, uint8_t *buf, int buf_size){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3264 SnowContext *s = avctx->priv_data;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3265 CABACContext * const c= &s->c;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3266 const int width= s->avctx->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3267 const int height= s->avctx->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3268 int bytes_read;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3269 AVFrame *picture = data;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3270 int log2_threshold, level, orientation, plane_index;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3271
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3272
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3273 /* no supplementary picture */
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3274 if (buf_size == 0)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3275 return 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3276
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3277 ff_init_cabac_decoder(c, buf, buf_size);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3278 ff_init_cabac_states(c, ff_h264_lps_range, ff_h264_mps_state, ff_h264_lps_state, 64);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3279
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3280 memset(s->header_state, 0, sizeof(s->header_state));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3281
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3282 s->current_picture.pict_type= FF_I_TYPE; //FIXME I vs. P
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3283 decode_header(s);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3284
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3285 frame_start(s);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3286 //keyframe flag dupliaction mess FIXME
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3287 if(avctx->debug&FF_DEBUG_PICT_INFO)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3288 av_log(avctx, AV_LOG_ERROR, "keyframe:%d qlog:%d\n", s->keyframe, s->qlog);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3289
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3290 decode_subband(s, &s->mb_band , s->mb_band .buf, NULL, s->mb_band .stride, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3291 decode_subband(s, &s->mv_band[0], s->mv_band[0].buf, NULL, s->mv_band[0].stride, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3292 decode_subband(s, &s->mv_band[1], s->mv_band[1].buf, NULL, s->mv_band[1].stride, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3293 correlate(s, &s->mb_band , s->mb_band .buf, s->mb_band .stride, 1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3294 correlate(s, &s->mv_band[0], s->mv_band[0].buf, s->mv_band[0].stride, 1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3295 correlate(s, &s->mv_band[1], s->mv_band[1].buf, s->mv_band[1].stride, 1, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3296
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3297 for(plane_index=0; plane_index<3; plane_index++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3298 Plane *p= &s->plane[plane_index];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3299 int w= p->width;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3300 int h= p->height;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3301 int x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3302
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3303 if(s->avctx->debug&2048){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3304 memset(s->spatial_dwt_buffer, 0, sizeof(DWTELEM)*w*h);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3305 predict_plane(s, s->spatial_dwt_buffer, plane_index, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3306
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3307 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3308 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3309 int v= (s->spatial_dwt_buffer[y*w + x]+128)>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3310 if(v&(~255)) v= ~(v>>31);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3311 s->mconly_picture.data[plane_index][y*s->mconly_picture.linesize[plane_index] + x]= v;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3312 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3313 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3314 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3315 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3316 for(orientation=level ? 1 : 0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3317 SubBand *b= &p->band[level][orientation];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3318
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3319 decode_subband(s, b, b->buf, b->parent ? b->parent->buf : NULL, b->stride, orientation);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3320 if(orientation==0)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3321 correlate(s, b, b->buf, b->stride, 1, 0);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3322 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3323 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3324 if(!(s->avctx->debug&1024))
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3325 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3326 for(orientation=level ? 1 : 0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3327 SubBand *b= &p->band[level][orientation];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3328
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3329 dequantize(s, b, b->buf, b->stride);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3330 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3331 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3332
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3333 #if QPRED
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3334 memset(s->pred_buffer, 0, sizeof(DWTELEM)*w*h);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3335 predict_plane(s, s->pred_buffer, plane_index, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3336 spatial_dwt(s, s->pred_buffer, w, h, w);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3337 for(level=0; level<s->spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3338 for(orientation=level ? 1 : 0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3339 SubBand *b= &p->band[level][orientation];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3340 int delta= ((int)s->pred_buffer - (int)s->spatial_dwt_buffer)/sizeof(DWTELEM);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3341
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3342 quantize (s, b, b->buf + delta, b->stride, s->qbias);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3343 dequantize(s, b, b->buf + delta, b->stride);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3344 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3345 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3346 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3347 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3348 s->spatial_dwt_buffer[y*w + x]+= s->pred_buffer[y*w + x];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3349 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3350 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3351 spatial_idwt(s, s->spatial_dwt_buffer, w, h, w);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3352 #else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3353 spatial_idwt(s, s->spatial_dwt_buffer, w, h, w);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3354 predict_plane(s, s->spatial_dwt_buffer, plane_index, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3355 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3356
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3357 //FIXME optimize
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3358 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3359 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3360 int v= (s->spatial_dwt_buffer[y*w + x]+128)>>8;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3361 if(v&(~255)) v= ~(v>>31);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3362 s->current_picture.data[plane_index][y*s->current_picture.linesize[plane_index] + x]= v;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3363 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3364 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3365 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3366
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3367 emms_c();
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3368
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3369 if(s->last_picture.data[0])
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3370 avctx->release_buffer(avctx, &s->last_picture);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3371
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3372 if(!(s->avctx->debug&2048))
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3373 *picture= s->current_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3374 else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3375 *picture= s->mconly_picture;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3376
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3377 *data_size = sizeof(AVFrame);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3378
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3379 bytes_read= get_cabac_terminate(c);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3380 if(bytes_read ==0) av_log(s->avctx, AV_LOG_ERROR, "error at end of frame\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3381
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3382 return bytes_read;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3383 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3384
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3385 static int decode_end(AVCodecContext *avctx)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3386 {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3387 SnowContext *s = avctx->priv_data;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3388
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3389 common_end(s);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3390
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3391 return 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3392 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3393
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3394 AVCodec snow_decoder = {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3395 "snow",
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3396 CODEC_TYPE_VIDEO,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3397 CODEC_ID_SNOW,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3398 sizeof(SnowContext),
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3399 decode_init,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3400 NULL,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3401 decode_end,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3402 decode_frame,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3403 0 /*CODEC_CAP_DR1*/ /*| CODEC_CAP_DRAW_HORIZ_BAND*/,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3404 NULL
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3405 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3406
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3407 AVCodec snow_encoder = {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3408 "snow",
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3409 CODEC_TYPE_VIDEO,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3410 CODEC_ID_SNOW,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3411 sizeof(SnowContext),
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3412 encode_init,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3413 encode_frame,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3414 encode_end,
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3415 };
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3416
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3417
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3418 #if 0
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3419 #undef malloc
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3420 #undef free
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3421 #undef printf
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3422
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3423 int main(){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3424 int width=256;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3425 int height=256;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3426 int buffer[2][width*height];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3427 SnowContext s;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3428 int i;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3429 s.spatial_decomposition_count=6;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3430 s.spatial_decomposition_type=1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3431
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3432 printf("testing 5/3 DWT\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3433 for(i=0; i<width*height; i++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3434 buffer[0][i]= buffer[1][i]= random()%54321 - 12345;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3435
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3436 spatial_dwt(&s, buffer[0], width, height, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3437 spatial_idwt(&s, buffer[0], width, height, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3438
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3439 for(i=0; i<width*height; i++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3440 if(buffer[0][i]!= buffer[1][i]) printf("fsck: %d %d %d\n",i, buffer[0][i], buffer[1][i]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3441
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3442 printf("testing 9/7 DWT\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3443 s.spatial_decomposition_type=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3444 for(i=0; i<width*height; i++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3445 buffer[0][i]= buffer[1][i]= random()%54321 - 12345;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3446
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3447 spatial_dwt(&s, buffer[0], width, height, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3448 spatial_idwt(&s, buffer[0], width, height, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3449
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3450 for(i=0; i<width*height; i++)
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3451 if(buffer[0][i]!= buffer[1][i]) printf("fsck: %d %d %d\n",i, buffer[0][i], buffer[1][i]);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3452
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3453 printf("testing AC coder\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3454 memset(s.header_state, 0, sizeof(s.header_state));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3455 ff_init_cabac_encoder(&s.c, buffer[0], 256*256);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3456 ff_init_cabac_states(&s.c, ff_h264_lps_range, ff_h264_mps_state, ff_h264_lps_state, 64);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3457
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3458 for(i=-256; i<256; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3459 START_TIMER
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3460 put_symbol(&s.c, s.header_state, i*i*i/3*ABS(i), 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3461 STOP_TIMER("put_symbol")
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3462 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3463 put_cabac_terminate(&s.c, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3464
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3465 memset(s.header_state, 0, sizeof(s.header_state));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3466 ff_init_cabac_decoder(&s.c, buffer[0], 256*256);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3467 ff_init_cabac_states(&s.c, ff_h264_lps_range, ff_h264_mps_state, ff_h264_lps_state, 64);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3468
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3469 for(i=-256; i<256; i++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3470 int j;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3471 START_TIMER
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3472 j= get_symbol(&s.c, s.header_state, 1);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3473 STOP_TIMER("get_symbol")
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3474 if(j!=i*i*i/3*ABS(i)) printf("fsck: %d != %d\n", i, j);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3475 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3476 {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3477 int level, orientation, x, y;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3478 int64_t errors[8][4];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3479 int64_t g=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3480
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3481 memset(errors, 0, sizeof(errors));
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3482 s.spatial_decomposition_count=3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3483 s.spatial_decomposition_type=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3484 for(level=0; level<s.spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3485 for(orientation=level ? 1 : 0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3486 int w= width >> (s.spatial_decomposition_count-level);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3487 int h= height >> (s.spatial_decomposition_count-level);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3488 int stride= width << (s.spatial_decomposition_count-level);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3489 DWTELEM *buf= buffer[0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3490 int64_t error=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3491
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3492 if(orientation&1) buf+=w;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3493 if(orientation>1) buf+=stride>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3494
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3495 memset(buffer[0], 0, sizeof(int)*width*height);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3496 buf[w/2 + h/2*stride]= 256*256;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3497 spatial_idwt(&s, buffer[0], width, height, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3498 for(y=0; y<height; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3499 for(x=0; x<width; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3500 int64_t d= buffer[0][x + y*width];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3501 error += d*d;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3502 if(ABS(width/2-x)<9 && ABS(height/2-y)<9 && level==2) printf("%8lld ", d);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3503 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3504 if(ABS(height/2-y)<9 && level==2) printf("\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3505 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3506 error= (int)(sqrt(error)+0.5);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3507 errors[level][orientation]= error;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3508 if(g) g=ff_gcd(g, error);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3509 else g= error;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3510 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3511 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3512 printf("static int const visual_weight[][4]={\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3513 for(level=0; level<s.spatial_decomposition_count; level++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3514 printf(" {");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3515 for(orientation=0; orientation<4; orientation++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3516 printf("%8lld,", errors[level][orientation]/g);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3517 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3518 printf("},\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3519 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3520 printf("};\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3521 {
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3522 int level=2;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3523 int orientation=3;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3524 int w= width >> (s.spatial_decomposition_count-level);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3525 int h= height >> (s.spatial_decomposition_count-level);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3526 int stride= width << (s.spatial_decomposition_count-level);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3527 DWTELEM *buf= buffer[0];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3528 int64_t error=0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3529
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3530 buf+=w;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3531 buf+=stride>>1;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3532
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3533 memset(buffer[0], 0, sizeof(int)*width*height);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3534 #if 1
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3535 for(y=0; y<height; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3536 for(x=0; x<width; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3537 int tab[4]={0,2,3,1};
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3538 buffer[0][x+width*y]= 256*256*tab[(x&1) + 2*(y&1)];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3539 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3540 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3541 spatial_dwt(&s, buffer[0], width, height, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3542 #else
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3543 for(y=0; y<h; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3544 for(x=0; x<w; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3545 buf[x + y*stride ]=169;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3546 buf[x + y*stride-w]=64;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3547 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3548 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3549 spatial_idwt(&s, buffer[0], width, height, width);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3550 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3551 for(y=0; y<height; y++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3552 for(x=0; x<width; x++){
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3553 int64_t d= buffer[0][x + y*width];
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3554 error += d*d;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3555 if(ABS(width/2-x)<9 && ABS(height/2-y)<9) printf("%8lld ", d);
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3556 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3557 if(ABS(height/2-y)<9) printf("\n");
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3558 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3559 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3560
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3561 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3562 return 0;
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3563 }
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3564 #endif
1ac13c350369 my experimental wavelet codec
michael
parents:
diff changeset
3565