annotate snow.c @ 2193:ea7715935e55 libavcodec

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