开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 嵌入式/单片机编程 > Windows CE > 查看源码
snow.c
资源名称：tcpmp.rar　[点击查看]
上传用户：wstnjxml
上传日期：2014-04-03
资源大小：7248k
文件大小：135k
源码类别：

Windows CE

开发平台：

C/C++

snow.c：源码内容
  1. /*
  2.  * Copyright (C) 2004 Michael Niedermayer <michaelni@gmx.at>
  3.  *
  4.  * This library is free software; you can redistribute it and/or
  5.  * modify it under the terms of the GNU Lesser General Public
  6.  * License as published by the Free Software Foundation; either
  7.  * version 2 of the License, or (at your option) any later version.
  8.  *
  9.  * This library is distributed in the hope that it will be useful,
  10.  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12.  * Lesser General Public License for more details.
  13.  *
  14.  * You should have received a copy of the GNU Lesser General Public
  15.  * License along with this library; if not, write to the Free Software
  16.  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  17.  */
  19. #include "avcodec.h"
  20. #include "common.h"
  21. #include "dsputil.h"
  23. #include "rangecoder.h"
  24. #define MID_STATE 128
  26. #include "mpegvideo.h"
  28. #undef NDEBUG
  29. #include <assert.h>
  31. #define MAX_DECOMPOSITIONS 8
  32. #define MAX_PLANES 4
  33. #define DWTELEM int
  34. #define QSHIFT 5
  35. #define QROOT (1<<QSHIFT)
  36. #define LOSSLESS_QLOG -128
  37. #define FRAC_BITS 8
  39. static const int8_t quant3[256]={
  40.  0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  41.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  42.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  43.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  44.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  45.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  46.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  47.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  48. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  49. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  50. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  51. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  52. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  53. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  54. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  55. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1, 0,
  56. };
  57. static const int8_t quant3b[256]={
  58.  0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  59.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  60.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  61.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  62.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  63.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  64.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  65.  1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1,
  66. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  67. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  68. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  69. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  70. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  71. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  72. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  73. -1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,-1,
  74. };
  75. static const int8_t quant3bA[256]={
  76.  0, 0, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  77.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  78.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  79.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  80.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  81.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  82.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  83.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  84.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  85.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  86.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  87.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  88.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  89.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  90.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  91.  1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1, 1,-1,
  92. };
  93. static const int8_t quant5[256]={
  94.  0, 1, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  95.  2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  96.  2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  97.  2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  98.  2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  99.  2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  100.  2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  101.  2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  102. -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
  103. -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
  104. -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
  105. -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
  106. -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
  107. -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
  108. -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
  109. -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-1,-1,-1,
  110. };
  111. static const int8_t quant7[256]={
  112.  0, 1, 1, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  113.  2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2, 2,
  114.  2, 2, 2, 2, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3,
  115.  3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
  116.  3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
  117.  3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
  118.  3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
  119.  3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3, 3,
  120. -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,
  121. -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,
  122. -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,
  123. -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,
  124. -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-3,
  125. -3,-3,-3,-3,-3,-3,-3,-3,-3,-2,-2,-2,-2,-2,-2,-2,
  126. -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,
  127. -2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-2,-1,-1,
  128. };
  129. static const int8_t quant9[256]={
  130.  0, 1, 1, 2, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 3, 3,
  131.  3, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  132.  4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  133.  4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  134.  4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  135.  4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  136.  4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  137.  4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  138. -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
  139. -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
  140. -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
  141. -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
  142. -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
  143. -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
  144. -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-3,-3,-3,-3,
  145. -3,-3,-3,-3,-3,-3,-3,-3,-3,-3,-2,-2,-2,-2,-1,-1,
  146. };
  147. static const int8_t quant11[256]={
  148.  0, 1, 2, 2, 2, 3, 3, 3, 3, 3, 3, 3, 4, 4, 4, 4,
  149.  4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4, 4,
  150.  4, 4, 4, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
  151.  5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
  152.  5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
  153.  5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
  154.  5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
  155.  5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
  156. -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
  157. -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
  158. -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
  159. -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
  160. -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
  161. -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-4,-4,
  162. -4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,-4,
  163. -4,-4,-4,-4,-4,-3,-3,-3,-3,-3,-3,-3,-2,-2,-2,-1,
  164. };
  165. static const int8_t quant13[256]={
  166.  0, 1, 2, 2, 3, 3, 3, 3, 4, 4, 4, 4, 4, 4, 4, 4,
  167.  4, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
  168.  5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5, 5,
  169.  5, 5, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
  170.  6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
  171.  6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
  172.  6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
  173.  6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6, 6,
  174. -6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,
  175. -6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,
  176. -6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,
  177. -6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,
  178. -6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-6,-5,
  179. -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
  180. -5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,-5,
  181. -4,-4,-4,-4,-4,-4,-4,-4,-4,-3,-3,-3,-3,-2,-2,-1,
  182. };
  184. #define LOG2_OBMC_MAX 6
  185. #define OBMC_MAX (1<<(LOG2_OBMC_MAX))
  186. #if 0 //64*cubic
  187. static const uint8_t obmc32[1024]={
  188.  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,
  189.  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,
  190.  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,
  191.  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,
  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,
  193.  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,
  194.  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,
  195.  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,
  196.  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,
  197.  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,
  198.  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,
  199.  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,
  200.  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,
  201.  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,
  202.  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,
  203.  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,
  204.  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,
  205.  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,
  206.  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,
  207.  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,
  208.  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,
  209.  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,
  210.  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,
  211.  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,
  212.  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,
  213.  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,
  214.  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,
  215.  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,
  216.  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,
  217.  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,
  218.  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,
  219.  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,
  220. //error:0.000022
  221. };
  222. static const uint8_t obmc16[256]={
  223.  0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0,
  224.  0, 1, 1, 2, 4, 5, 5, 6, 6, 5, 5, 4, 2, 1, 1, 0,
  225.  0, 1, 4, 6, 9,11,13,15,15,13,11, 9, 6, 4, 1, 0,
  226.  0, 2, 6,11,15,20,24,26,26,24,20,15,11, 6, 2, 0,
  227.  0, 4, 9,15,23,29,34,38,38,34,29,23,15, 9, 4, 0,
  228.  0, 5,11,20,29,38,45,49,49,45,38,29,20,11, 5, 0,
  229.  1, 5,13,24,34,45,53,57,57,53,45,34,24,13, 5, 1,
  230.  1, 6,15,26,38,49,57,62,62,57,49,38,26,15, 6, 1,
  231.  1, 6,15,26,38,49,57,62,62,57,49,38,26,15, 6, 1,
  232.  1, 5,13,24,34,45,53,57,57,53,45,34,24,13, 5, 1,
  233.  0, 5,11,20,29,38,45,49,49,45,38,29,20,11, 5, 0,
  234.  0, 4, 9,15,23,29,34,38,38,34,29,23,15, 9, 4, 0,
  235.  0, 2, 6,11,15,20,24,26,26,24,20,15,11, 6, 2, 0,
  236.  0, 1, 4, 6, 9,11,13,15,15,13,11, 9, 6, 4, 1, 0,
  237.  0, 1, 1, 2, 4, 5, 5, 6, 6, 5, 5, 4, 2, 1, 1, 0,
  238.  0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0,
  239. //error:0.000033
  240. };
  241. #elif 1 // 64*linear
  242. static const uint8_t obmc32[1024]={
  243.  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,
  244.  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,
  245.  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,
  246.  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,
  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,
  248.  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,
  249.  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,
  250.  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,
  251.  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,
  252.  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,
  253.  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,
  254.  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,
  255.  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,
  256.  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,
  257.  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,
  258.  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,
  259.  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,
  260.  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,
  261.  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,
  262.  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,
  263.  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,
  264.  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,
  265.  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,
  266.  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,
  267.  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,
  268.  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,
  269.  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,
  270.  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,
  271.  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,
  272.  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,
  273.  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,
  274.  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,
  275.  //error:0.000020
  276. };
  277. static const uint8_t obmc16[256]={
  278.  0, 1, 1, 2, 2, 3, 3, 4, 4, 3, 3, 2, 2, 1, 1, 0,
  279.  1, 2, 4, 5, 7, 8,10,11,11,10, 8, 7, 5, 4, 2, 1,
  280.  1, 4, 6, 9,11,14,16,19,19,16,14,11, 9, 6, 4, 1,
  281.  2, 5, 9,12,16,19,23,26,26,23,19,16,12, 9, 5, 2,
  282.  2, 7,11,16,20,25,29,34,34,29,25,20,16,11, 7, 2,
  283.  3, 8,14,19,25,30,36,41,41,36,30,25,19,14, 8, 3,
  284.  3,10,16,23,29,36,42,49,49,42,36,29,23,16,10, 3,
  285.  4,11,19,26,34,41,49,56,56,49,41,34,26,19,11, 4,
  286.  4,11,19,26,34,41,49,56,56,49,41,34,26,19,11, 4,
  287.  3,10,16,23,29,36,42,49,49,42,36,29,23,16,10, 3,
  288.  3, 8,14,19,25,30,36,41,41,36,30,25,19,14, 8, 3,
  289.  2, 7,11,16,20,25,29,34,34,29,25,20,16,11, 7, 2,
  290.  2, 5, 9,12,16,19,23,26,26,23,19,16,12, 9, 5, 2,
  291.  1, 4, 6, 9,11,14,16,19,19,16,14,11, 9, 6, 4, 1,
  292.  1, 2, 4, 5, 7, 8,10,11,11,10, 8, 7, 5, 4, 2, 1,
  293.  0, 1, 1, 2, 2, 3, 3, 4, 4, 3, 3, 2, 2, 1, 1, 0,
  294. //error:0.000015
  295. };
  296. #else //64*cos
  297. static const uint8_t obmc32[1024]={
  298.  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,
  299.  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,
  300.  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,
  301.  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,
  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,
  303.  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,
  304.  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,
  305.  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,
  306.  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,
  307.  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,
  308.  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,
  309.  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,
  310.  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,
  311.  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,
  312.  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,
  313.  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,
  314.  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,
  315.  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,
  316.  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,
  317.  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,
  318.  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,
  319.  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,
  320.  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,
  321.  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,
  322.  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,
  323.  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,
  324.  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,
  325.  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,
  326.  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,
  327.  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,
  328.  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,
  329.  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,
  330. //error:0.000022
  331. };
  332. static const uint8_t obmc16[256]={
  333.  0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0,
  334.  0, 0, 1, 2, 3, 4, 5, 5, 5, 5, 4, 3, 2, 1, 0, 0,
  335.  0, 1, 3, 6, 8,11,13,14,14,13,11, 8, 6, 3, 1, 0,
  336.  0, 2, 6,10,15,20,24,26,26,24,20,15,10, 6, 2, 0,
  337.  0, 3, 8,16,23,30,35,38,38,35,30,23,16, 8, 3, 0,
  338.  1, 4,11,20,30,39,46,49,49,46,39,30,20,11, 4, 1,
  339.  1, 5,13,24,35,46,54,58,58,54,46,35,24,13, 5, 1,
  340.  0, 5,14,26,38,49,58,63,63,58,49,38,26,14, 5, 0,
  341.  0, 5,14,26,38,49,58,63,63,58,49,38,26,14, 5, 0,
  342.  1, 5,13,24,35,46,54,58,58,54,46,35,24,13, 5, 1,
  343.  1, 4,11,20,30,39,46,49,49,46,39,30,20,11, 4, 1,
  344.  0, 3, 8,16,23,30,35,38,38,35,30,23,16, 8, 3, 0,
  345.  0, 2, 6,10,15,20,24,26,26,24,20,15,10, 6, 2, 0,
  346.  0, 1, 3, 6, 8,11,13,14,14,13,11, 8, 6, 3, 1, 0,
  347.  0, 0, 1, 2, 3, 4, 5, 5, 5, 5, 4, 3, 2, 1, 0, 0,
  348.  0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0,
  349. //error:0.000022
  350. };
  351. #endif
  353. //linear *64
  354. static const uint8_t obmc8[64]={
  355.  1, 3, 5, 7, 7, 5, 3, 1,
  356.  3, 9,15,21,21,15, 9, 3,
  357.  5,15,25,35,35,25,15, 5,
  358.  7,21,35,49,49,35,21, 7,
  359.  7,21,35,49,49,35,21, 7,
  360.  5,15,25,35,35,25,15, 5,
  361.  3, 9,15,21,21,15, 9, 3,
  362.  1, 3, 5, 7, 7, 5, 3, 1,
  363. //error:0.000000
  364. };
  366. //linear *64
  367. static const uint8_t obmc4[16]={
  368.  4,12,12, 4,
  369. 12,36,36,12,
  370. 12,36,36,12,
  371.  4,12,12, 4,
  372. //error:0.000000
  373. };
  375. static const uint8_t *obmc_tab[4]={
  376.     obmc32, obmc16, obmc8, obmc4
  377. };
  379. typedef struct BlockNode{
  380.     int16_t mx;
  381.     int16_t my;
  382.     uint8_t color[3];
  383.     uint8_t type;
  384. //#define TYPE_SPLIT    1
  385. #define BLOCK_INTRA   1
  386. //#define TYPE_NOCOLOR  4
  387.     uint8_t level; //FIXME merge into type?
  388. }BlockNode;
  390. #define LOG2_MB_SIZE 4
  391. #define MB_SIZE (1<<LOG2_MB_SIZE)
  393. typedef struct x_and_coeff{
  394.     int16_t x;
  395.     uint16_t coeff;
  396. } x_and_coeff;
  398. typedef struct SubBand{
  399.     int level;
  400.     int stride;
  401.     int width;
  402.     int height;
  403.     int qlog;                                   ///< log(qscale)/log[2^(1/6)]
  404.     DWTELEM *buf;
  405.     int buf_x_offset;
  406.     int buf_y_offset;
  407.     int stride_line; ///< Stride measured in lines, not pixels.
  408.     x_and_coeff * x_coeff;
  409.     struct SubBand *parent;
  410.     uint8_t state[/*7*2*/ 7 + 512][32];
  411. }SubBand;
  413. typedef struct Plane{
  414.     int width;
  415.     int height;
  416.     SubBand band[MAX_DECOMPOSITIONS][4];
  417. }Plane;
  419. /** Used to minimize the amount of memory used in order to optimize cache performance. **/
  420. typedef struct {
  421.     DWTELEM * * line; ///< For use by idwt and predict_slices.
  422.     DWTELEM * * data_stack; ///< Used for internal purposes.
  423.     int data_stack_top;
  424.     int line_count;
  425.     int line_width;
  426.     int data_count;
  427.     DWTELEM * base_buffer; ///< Buffer that this structure is caching.
  428. } slice_buffer;
  430. typedef struct SnowContext{
  431. //    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)
  433.     AVCodecContext *avctx;
  434.     RangeCoder c;
  435.     DSPContext dsp;
  436.     AVFrame input_picture;
  437.     AVFrame current_picture;
  438.     AVFrame last_picture;
  439.     AVFrame mconly_picture;
  440. //     uint8_t q_context[16];
  441.     uint8_t header_state[32];
  442.     uint8_t block_state[128 + 32*128];
  443.     int keyframe;
  444.     int always_reset;
  445.     int version;
  446.     int spatial_decomposition_type;
  447.     int temporal_decomposition_type;
  448.     int spatial_decomposition_count;
  449.     int temporal_decomposition_count;
  450.     DWTELEM *spatial_dwt_buffer;
  451.     int colorspace_type;
  452.     int chroma_h_shift;
  453.     int chroma_v_shift;
  454.     int spatial_scalability;
  455.     int qlog;
  456.     int lambda;
  457.     int lambda2;
  458.     int mv_scale;
  459.     int qbias;
  460. #define QBIAS_SHIFT 3
  461.     int b_width;
  462.     int b_height;
  463.     int block_max_depth;
  464.     Plane plane[MAX_PLANES];
  465.     BlockNode *block;
  466.     slice_buffer sb;
  468.     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)
  469. }SnowContext;
  471. typedef struct {
  472.     DWTELEM *b0;
  473.     DWTELEM *b1;
  474.     DWTELEM *b2;
  475.     DWTELEM *b3;
  476.     int y;
  477. } dwt_compose_t;
  479. #define slice_buffer_get_line(slice_buf, line_num) ((slice_buf)->line[line_num] ? (slice_buf)->line[line_num] : slice_buffer_load_line((slice_buf), (line_num)))
  480. //#define slice_buffer_get_line(slice_buf, line_num) (slice_buffer_load_line((slice_buf), (line_num)))
  482. static void slice_buffer_init(slice_buffer * buf, int line_count, int max_allocated_lines, int line_width, DWTELEM * base_buffer)
  483. {
  484.     int i;
  485.   
  486.     buf->base_buffer = base_buffer;
  487.     buf->line_count = line_count;
  488.     buf->line_width = line_width;
  489.     buf->data_count = max_allocated_lines;
  490.     buf->line = (DWTELEM * *) av_mallocz (sizeof(DWTELEM *) * line_count);
  491.     buf->data_stack = (DWTELEM * *) av_malloc (sizeof(DWTELEM *) * max_allocated_lines);
  492.   
  493.     for (i = 0; i < max_allocated_lines; i++)
  494.     {
  495.       buf->data_stack[i] = (DWTELEM *) av_malloc (sizeof(DWTELEM) * line_width);
  496.     }
  497.     
  498.     buf->data_stack_top = max_allocated_lines - 1;
  499. }
  501. static DWTELEM * slice_buffer_load_line(slice_buffer * buf, int line)
  502. {
  503.     int i;
  504.     int offset;
  505.     DWTELEM * buffer;
  506.   
  507. //  av_log(NULL, AV_LOG_DEBUG, "Cache hit: %dn", line);  
  508.   
  509.     assert(buf->data_stack_top >= 0);
  510. //  assert(!buf->line[line]);
  511.     if (buf->line[line])
  512.         return buf->line[line];
  513.     
  514.     offset = buf->line_width * line;
  515.     buffer = buf->data_stack[buf->data_stack_top];
  516.     buf->data_stack_top--;
  517.     buf->line[line] = buffer;
  518.   
  519. //  av_log(NULL, AV_LOG_DEBUG, "slice_buffer_load_line: line: %d remaining: %dn", line, buf->data_stack_top + 1);
  520.   
  521.     return buffer;
  522. }
  524. static void slice_buffer_release(slice_buffer * buf, int line)
  525. {
  526.     int i;
  527.     int offset;
  528.     DWTELEM * buffer;
  530.     assert(line >= 0 && line < buf->line_count);
  531.     assert(buf->line[line]);
  533.     offset = buf->line_width * line;
  534.     buffer = buf->line[line];
  535.     buf->data_stack_top++;
  536.     buf->data_stack[buf->data_stack_top] = buffer;
  537.     buf->line[line] = NULL;
  538.   
  539. //  av_log(NULL, AV_LOG_DEBUG, "slice_buffer_release: line: %d remaining: %dn", line, buf->data_stack_top + 1);
  540. }
  542. static void slice_buffer_flush(slice_buffer * buf)
  543. {
  544.     int i;
  545.     for (i = 0; i < buf->line_count; i++)
  546.     {
  547.         if (buf->line[i])
  548.         {
  549. //      av_log(NULL, AV_LOG_DEBUG, "slice_buffer_flush: line: %d n", i);
  550.             slice_buffer_release(buf, i);
  551.         }
  552.     }
  553. }
  555. static void slice_buffer_destroy(slice_buffer * buf)
  556. {
  557.     int i;
  558.     slice_buffer_flush(buf);
  559.   
  560.     for (i = buf->data_count - 1; i >= 0; i--)
  561.     {
  562.         assert(buf->data_stack[i]);
  563.         av_free(buf->data_stack[i]);
  564.     }
  565.     assert(buf->data_stack);
  566.     av_free(buf->data_stack);
  567.     assert(buf->line);
  568.     av_free(buf->line);
  569. }
  571. #ifdef __sgi
  572. // Avoid a name clash on SGI IRIX
  573. #undef qexp
  574. #endif
  575. #define QEXPSHIFT (7-FRAC_BITS+8) //FIXME try to change this to 0
  576. static uint8_t qexp[QROOT];
  578. static inline int mirror(int v, int m){
  579.     if     (v<0) return -v;
  580.     else if(v>m) return 2*m-v;
  581.     else         return v;
  582. }
  584. static inline void put_symbol(RangeCoder *c, uint8_t *state, int v, int is_signed){
  585.     int i;
  587.     if(v){
  588.         const int a= ABS(v);
  589.         const int e= av_log2(a);
  590. #if 1
  591.         const int el= FFMIN(e, 10);   
  592.         put_rac(c, state+0, 0);
  594.         for(i=0; i<el; i++){
  595.             put_rac(c, state+1+i, 1);  //1..10
  596.         }
  597.         for(; i<e; i++){
  598.             put_rac(c, state+1+9, 1);  //1..10
  599.         }
  600.         put_rac(c, state+1+FFMIN(i,9), 0);
  602.         for(i=e-1; i>=el; i--){
  603.             put_rac(c, state+22+9, (a>>i)&1); //22..31
  604.         }
  605.         for(; i>=0; i--){
  606.             put_rac(c, state+22+i, (a>>i)&1); //22..31
  607.         }
  609.         if(is_signed)
  610.             put_rac(c, state+11 + el, v < 0); //11..21
  611. #else
  612.         
  613.         put_rac(c, state+0, 0);
  614.         if(e<=9){
  615.             for(i=0; i<e; i++){
  616.                 put_rac(c, state+1+i, 1);  //1..10
  617.             }
  618.             put_rac(c, state+1+i, 0);
  620.             for(i=e-1; i>=0; i--){
  621.                 put_rac(c, state+22+i, (a>>i)&1); //22..31
  622.             }
  624.             if(is_signed)
  625.                 put_rac(c, state+11 + e, v < 0); //11..21
  626.         }else{
  627.             for(i=0; i<e; i++){
  628.                 put_rac(c, state+1+FFMIN(i,9), 1);  //1..10
  629.             }
  630.             put_rac(c, state+1+FFMIN(i,9), 0);
  632.             for(i=e-1; i>=0; i--){
  633.                 put_rac(c, state+22+FFMIN(i,9), (a>>i)&1); //22..31
  634.             }
  636.             if(is_signed)
  637.                 put_rac(c, state+11 + FFMIN(e,10), v < 0); //11..21
  638.         }
  639. #endif
  640.     }else{
  641.         put_rac(c, state+0, 1);
  642.     }
  643. }
  645. static inline int get_symbol(RangeCoder *c, uint8_t *state, int is_signed){
  646.     if(get_rac(c, state+0))
  647.         return 0;
  648.     else{
  649.         int i, e, a;
  650.         e= 0;
  651.         while(get_rac(c, state+1 + FFMIN(e,9))){ //1..10
  652.             e++;
  653.         }
  655.         a= 1;
  656.         for(i=e-1; i>=0; i--){
  657.             a += a + get_rac(c, state+22 + FFMIN(i,9)); //22..31
  658.         }
  660.         if(is_signed && get_rac(c, state+11 + FFMIN(e,10))) //11..21
  661.             return -a;
  662.         else
  663.             return a;
  664.     }
  665. }
  667. static inline void put_symbol2(RangeCoder *c, uint8_t *state, int v, int log2){
  668.     int i;
  669.     int r= log2>=0 ? 1<<log2 : 1;
  671.     assert(v>=0);
  672.     assert(log2>=-4);
  674.     while(v >= r){
  675.         put_rac(c, state+4+log2, 1);
  676.         v -= r;
  677.         log2++;
  678.         if(log2>0) r+=r;
  679.     }
  680.     put_rac(c, state+4+log2, 0);
  681.     
  682.     for(i=log2-1; i>=0; i--){
  683.         put_rac(c, state+31-i, (v>>i)&1);
  684.     }
  685. }
  687. static inline int get_symbol2(RangeCoder *c, uint8_t *state, int log2){
  688.     int i;
  689.     int r= log2>=0 ? 1<<log2 : 1;
  690.     int v=0;
  692.     assert(log2>=-4);
  694.     while(get_rac(c, state+4+log2)){
  695.         v+= r;
  696.         log2++;
  697.         if(log2>0) r+=r;
  698.     }
  699.     
  700.     for(i=log2-1; i>=0; i--){
  701.         v+= get_rac(c, state+31-i)<<i;
  702.     }
  704.     return v;
  705. }
  707. 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){
  708.     const int mirror_left= !highpass;
  709.     const int mirror_right= (width&1) ^ highpass;
  710.     const int w= (width>>1) - 1 + (highpass & width);
  711.     int i;
  713. #define LIFT(src, ref, inv) ((src) + ((inv) ? - (ref) : + (ref)))
  714.     if(mirror_left){
  715.         dst[0] = LIFT(src[0], ((mul*2*ref[0]+add)>>shift), inverse);
  716.         dst += dst_step;
  717.         src += src_step;
  718.     }
  719.     
  720.     for(i=0; i<w; i++){
  721.         dst[i*dst_step] = LIFT(src[i*src_step], ((mul*(ref[i*ref_step] + ref[(i+1)*ref_step])+add)>>shift), inverse);
  722.     }
  723.     
  724.     if(mirror_right){
  725.         dst[w*dst_step] = LIFT(src[w*src_step], ((mul*2*ref[w*ref_step]+add)>>shift), inverse);
  726.     }
  727. }
  729. 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){
  730.     const int mirror_left= !highpass;
  731.     const int mirror_right= (width&1) ^ highpass;
  732.     const int w= (width>>1) - 1 + (highpass & width);
  733.     int i;
  735.     if(mirror_left){
  736.         int r= 3*2*ref[0];
  737.         r += r>>4;
  738.         r += r>>8;
  739.         dst[0] = LIFT(src[0], ((r+add)>>shift), inverse);
  740.         dst += dst_step;
  741.         src += src_step;
  742.     }
  743.     
  744.     for(i=0; i<w; i++){
  745.         int r= 3*(ref[i*ref_step] + ref[(i+1)*ref_step]);
  746.         r += r>>4;
  747.         r += r>>8;
  748.         dst[i*dst_step] = LIFT(src[i*src_step], ((r+add)>>shift), inverse);
  749.     }
  750.     
  751.     if(mirror_right){
  752.         int r= 3*2*ref[w*ref_step];
  753.         r += r>>4;
  754.         r += r>>8;
  755.         dst[w*dst_step] = LIFT(src[w*src_step], ((r+add)>>shift), inverse);
  756.     }
  757. }
  759. static always_inline void liftS(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){
  760.     const int mirror_left= !highpass;
  761.     const int mirror_right= (width&1) ^ highpass;
  762.     const int w= (width>>1) - 1 + (highpass & width);
  763.     int i;
  765.     assert(shift == 4);
  766. #define LIFTS(src, ref, inv) ((inv) ? (src) - (((ref) - 4*(src))>>shift): (16*4*(src) + 4*(ref) + 8 + (5<<27))/(5*16) - (1<<23))
  767.     if(mirror_left){
  768.         dst[0] = LIFTS(src[0], mul*2*ref[0]+add, inverse);
  769.         dst += dst_step;
  770.         src += src_step;
  771.     }
  772.     
  773.     for(i=0; i<w; i++){
  774.         dst[i*dst_step] = LIFTS(src[i*src_step], mul*(ref[i*ref_step] + ref[(i+1)*ref_step])+add, inverse);
  775.     }
  776.     
  777.     if(mirror_right){
  778.         dst[w*dst_step] = LIFTS(src[w*src_step], mul*2*ref[w*ref_step]+add, inverse);
  779.     }
  780. }
  783. static void inplace_lift(DWTELEM *dst, int width, int *coeffs, int n, int shift, int start, int inverse){
  784.     int x, i;
  785.     
  786.     for(x=start; x<width; x+=2){
  787.         int64_t sum=0;
  789.         for(i=0; i<n; i++){
  790.             int x2= x + 2*i - n + 1;
  791.             if     (x2<     0) x2= -x2;
  792.             else if(x2>=width) x2= 2*width-x2-2;
  793.             sum += coeffs[i]*(int64_t)dst[x2];
  794.         }
  795.         if(inverse) dst[x] -= (sum + (1<<shift)/2)>>shift;
  796.         else        dst[x] += (sum + (1<<shift)/2)>>shift;
  797.     }
  798. }
  800. static void inplace_liftV(DWTELEM *dst, int width, int height, int stride, int *coeffs, int n, int shift, int start, int inverse){
  801.     int x, y, i;
  802.     for(y=start; y<height; y+=2){
  803.         for(x=0; x<width; x++){
  804.             int64_t sum=0;
  805.     
  806.             for(i=0; i<n; i++){
  807.                 int y2= y + 2*i - n + 1;
  808.                 if     (y2<      0) y2= -y2;
  809.                 else if(y2>=height) y2= 2*height-y2-2;
  810.                 sum += coeffs[i]*(int64_t)dst[x + y2*stride];
  811.             }
  812.             if(inverse) dst[x + y*stride] -= (sum + (1<<shift)/2)>>shift;
  813.             else        dst[x + y*stride] += (sum + (1<<shift)/2)>>shift;
  814.         }
  815.     }
  816. }
  818. #define SCALEX 1
  819. #define LX0 0
  820. #define LX1 1
  822. #if 0 // more accurate 9/7
  823. #define N1 2
  824. #define SHIFT1 14
  825. #define COEFFS1 (int[]){-25987,-25987}
  826. #define N2 2
  827. #define SHIFT2 19
  828. #define COEFFS2 (int[]){-27777,-27777}
  829. #define N3 2
  830. #define SHIFT3 15
  831. #define COEFFS3 (int[]){28931,28931}
  832. #define N4 2
  833. #define SHIFT4 15
  834. #define COEFFS4 (int[]){14533,14533}
  835. #elif 1 // 13/7 CRF
  836. #define N1 4
  837. #define SHIFT1 4
  838. #define COEFFS1 (int[]){1,-9,-9,1}
  839. #define N2 4
  840. #define SHIFT2 4
  841. #define COEFFS2 (int[]){-1,5,5,-1}
  842. #define N3 0
  843. #define SHIFT3 1
  844. #define COEFFS3 NULL
  845. #define N4 0
  846. #define SHIFT4 1
  847. #define COEFFS4 NULL
  848. #elif 1 // 3/5
  849. #define LX0 1
  850. #define LX1 0
  851. #define SCALEX 0.5
  852. #define N1 2
  853. #define SHIFT1 1
  854. #define COEFFS1 (int[]){1,1}
  855. #define N2 2
  856. #define SHIFT2 2
  857. #define COEFFS2 (int[]){-1,-1}
  858. #define N3 0
  859. #define SHIFT3 0
  860. #define COEFFS3 NULL
  861. #define N4 0
  862. #define SHIFT4 0
  863. #define COEFFS4 NULL
  864. #elif 1 // 11/5 
  865. #define N1 0
  866. #define SHIFT1 1
  867. #define COEFFS1 NULL
  868. #define N2 2
  869. #define SHIFT2 2
  870. #define COEFFS2 (int[]){-1,-1}
  871. #define N3 2
  872. #define SHIFT3 0
  873. #define COEFFS3 (int[]){-1,-1}
  874. #define N4 4
  875. #define SHIFT4 7
  876. #define COEFFS4 (int[]){-5,29,29,-5}
  877. #define SCALEX 4
  878. #elif 1 // 9/7 CDF
  879. #define N1 2
  880. #define SHIFT1 7
  881. #define COEFFS1 (int[]){-203,-203}
  882. #define N2 2
  883. #define SHIFT2 12
  884. #define COEFFS2 (int[]){-217,-217}
  885. #define N3 2
  886. #define SHIFT3 7
  887. #define COEFFS3 (int[]){113,113}
  888. #define N4 2
  889. #define SHIFT4 9
  890. #define COEFFS4 (int[]){227,227}
  891. #define SCALEX 1
  892. #elif 1 // 7/5 CDF
  893. #define N1 0
  894. #define SHIFT1 1
  895. #define COEFFS1 NULL
  896. #define N2 2
  897. #define SHIFT2 2
  898. #define COEFFS2 (int[]){-1,-1}
  899. #define N3 2
  900. #define SHIFT3 0
  901. #define COEFFS3 (int[]){-1,-1}
  902. #define N4 2
  903. #define SHIFT4 4
  904. #define COEFFS4 (int[]){3,3}
  905. #elif 1 // 9/7 MN
  906. #define N1 4
  907. #define SHIFT1 4
  908. #define COEFFS1 (int[]){1,-9,-9,1}
  909. #define N2 2
  910. #define SHIFT2 2
  911. #define COEFFS2 (int[]){1,1}
  912. #define N3 0
  913. #define SHIFT3 1
  914. #define COEFFS3 NULL
  915. #define N4 0
  916. #define SHIFT4 1
  917. #define COEFFS4 NULL
  918. #else // 13/7 CRF
  919. #define N1 4
  920. #define SHIFT1 4
  921. #define COEFFS1 (int[]){1,-9,-9,1}
  922. #define N2 4
  923. #define SHIFT2 4
  924. #define COEFFS2 (int[]){-1,5,5,-1}
  925. #define N3 0
  926. #define SHIFT3 1
  927. #define COEFFS3 NULL
  928. #define N4 0
  929. #define SHIFT4 1
  930. #define COEFFS4 NULL
  931. #endif
  932. static void horizontal_decomposeX(DWTELEM *b, int width){
  933.     DWTELEM temp[width];
  934.     const int width2= width>>1;
  935.     const int w2= (width+1)>>1;
  936.     int A1,A2,A3,A4, x;
  938.     inplace_lift(b, width, COEFFS1, N1, SHIFT1, LX1, 0);
  939.     inplace_lift(b, width, COEFFS2, N2, SHIFT2, LX0, 0);
  940.     inplace_lift(b, width, COEFFS3, N3, SHIFT3, LX1, 0);
  941.     inplace_lift(b, width, COEFFS4, N4, SHIFT4, LX0, 0);
  942.     
  943.     for(x=0; x<width2; x++){
  944.         temp[x   ]= b[2*x    ];
  945.         temp[x+w2]= b[2*x + 1];
  946.     }
  947.     if(width&1)
  948.         temp[x   ]= b[2*x    ];
  949.     memcpy(b, temp, width*sizeof(int));
  950. }
  952. static void horizontal_composeX(DWTELEM *b, int width){
  953.     DWTELEM temp[width];
  954.     const int width2= width>>1;
  955.     int A1,A2,A3,A4, x;
  956.     const int w2= (width+1)>>1;
  958.     memcpy(temp, b, width*sizeof(int));
  959.     for(x=0; x<width2; x++){
  960.         b[2*x    ]= temp[x   ];
  961.         b[2*x + 1]= temp[x+w2];
  962.     }
  963.     if(width&1)
  964.         b[2*x    ]= temp[x   ];
  966.     inplace_lift(b, width, COEFFS4, N4, SHIFT4, LX0, 1);
  967.     inplace_lift(b, width, COEFFS3, N3, SHIFT3, LX1, 1);
  968.     inplace_lift(b, width, COEFFS2, N2, SHIFT2, LX0, 1);
  969.     inplace_lift(b, width, COEFFS1, N1, SHIFT1, LX1, 1);
  970. }
  972. static void spatial_decomposeX(DWTELEM *buffer, int width, int height, int stride){
  973.     int x, y;
  974.   
  975.     for(y=0; y<height; y++){
  976.         for(x=0; x<width; x++){
  977.             buffer[y*stride + x] *= SCALEX;
  978.         }
  979.     }
  981.     for(y=0; y<height; y++){
  982.         horizontal_decomposeX(buffer + y*stride, width);
  983.     }
  984.     
  985.     inplace_liftV(buffer, width, height, stride, COEFFS1, N1, SHIFT1, LX1, 0);
  986.     inplace_liftV(buffer, width, height, stride, COEFFS2, N2, SHIFT2, LX0, 0);
  987.     inplace_liftV(buffer, width, height, stride, COEFFS3, N3, SHIFT3, LX1, 0);
  988.     inplace_liftV(buffer, width, height, stride, COEFFS4, N4, SHIFT4, LX0, 0);    
  989. }
  991. static void spatial_composeX(DWTELEM *buffer, int width, int height, int stride){
  992.     int x, y;
  993.   
  994.     inplace_liftV(buffer, width, height, stride, COEFFS4, N4, SHIFT4, LX0, 1);
  995.     inplace_liftV(buffer, width, height, stride, COEFFS3, N3, SHIFT3, LX1, 1);
  996.     inplace_liftV(buffer, width, height, stride, COEFFS2, N2, SHIFT2, LX0, 1);
  997.     inplace_liftV(buffer, width, height, stride, COEFFS1, N1, SHIFT1, LX1, 1);
  999.     for(y=0; y<height; y++){
  1000.         horizontal_composeX(buffer + y*stride, width);
  1001.     }
  1003.     for(y=0; y<height; y++){
  1004.         for(x=0; x<width; x++){
  1005.             buffer[y*stride + x] /= SCALEX;
  1006.         }
  1007.     }
  1008. }
  1010. static void horizontal_decompose53i(DWTELEM *b, int width){
  1011.     DWTELEM temp[width];
  1012.     const int width2= width>>1;
  1013.     int A1,A2,A3,A4, x;
  1014.     const int w2= (width+1)>>1;
  1016.     for(x=0; x<width2; x++){
  1017.         temp[x   ]= b[2*x    ];
  1018.         temp[x+w2]= b[2*x + 1];
  1019.     }
  1020.     if(width&1)
  1021.         temp[x   ]= b[2*x    ];
  1022. #if 0
  1023.     A2= temp[1       ];
  1024.     A4= temp[0       ];
  1025.     A1= temp[0+width2];
  1026.     A1 -= (A2 + A4)>>1;
  1027.     A4 += (A1 + 1)>>1;
  1028.     b[0+width2] = A1;
  1029.     b[0       ] = A4;
  1030.     for(x=1; x+1<width2; x+=2){
  1031.         A3= temp[x+width2];
  1032.         A4= temp[x+1     ];
  1033.         A3 -= (A2 + A4)>>1;
  1034.         A2 += (A1 + A3 + 2)>>2;
  1035.         b[x+width2] = A3;
  1036.         b[x       ] = A2;
  1038.         A1= temp[x+1+width2];
  1039.         A2= temp[x+2       ];
  1040.         A1 -= (A2 + A4)>>1;
  1041.         A4 += (A1 + A3 + 2)>>2;
  1042.         b[x+1+width2] = A1;
  1043.         b[x+1       ] = A4;
  1044.     }
  1045.     A3= temp[width-1];
  1046.     A3 -= A2;
  1047.     A2 += (A1 + A3 + 2)>>2;
  1048.     b[width -1] = A3;
  1049.     b[width2-1] = A2;
  1050. #else        
  1051.     lift(b+w2, temp+w2, temp, 1, 1, 1, width, -1, 0, 1, 1, 0);
  1052.     lift(b   , temp   , b+w2, 1, 1, 1, width,  1, 2, 2, 0, 0);
  1053. #endif
  1054. }
  1056. static void vertical_decompose53iH0(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1057.     int i;
  1058.     
  1059.     for(i=0; i<width; i++){
  1060.         b1[i] -= (b0[i] + b2[i])>>1;
  1061.     }
  1062. }
  1064. static void vertical_decompose53iL0(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1065.     int i;
  1066.     
  1067.     for(i=0; i<width; i++){
  1068.         b1[i] += (b0[i] + b2[i] + 2)>>2;
  1069.     }
  1070. }
  1072. static void spatial_decompose53i(DWTELEM *buffer, int width, int height, int stride){
  1073.     int y;
  1074.     DWTELEM *b0= buffer + mirror(-2-1, height-1)*stride;
  1075.     DWTELEM *b1= buffer + mirror(-2  , height-1)*stride;
  1076.   
  1077.     for(y=-2; y<height; y+=2){
  1078.         DWTELEM *b2= buffer + mirror(y+1, height-1)*stride;
  1079.         DWTELEM *b3= buffer + mirror(y+2, height-1)*stride;
  1081. {START_TIMER
  1082.         if(b1 <= b3)     horizontal_decompose53i(b2, width);
  1083.         if(y+2 < height) horizontal_decompose53i(b3, width);
  1084. STOP_TIMER("horizontal_decompose53i")}
  1085.         
  1086. {START_TIMER
  1087.         if(b1 <= b3) vertical_decompose53iH0(b1, b2, b3, width);
  1088.         if(b0 <= b2) vertical_decompose53iL0(b0, b1, b2, width);
  1089. STOP_TIMER("vertical_decompose53i*")}
  1090.         
  1091.         b0=b2;
  1092.         b1=b3;
  1093.     }
  1094. }
  1096. #define liftS lift
  1097. #define lift5 lift
  1098. #if 1
  1099. #define W_AM 3
  1100. #define W_AO 0
  1101. #define W_AS 1
  1103. #undef liftS
  1104. #define W_BM 1
  1105. #define W_BO 8
  1106. #define W_BS 4
  1108. #define W_CM 1
  1109. #define W_CO 0
  1110. #define W_CS 0
  1112. #define W_DM 3
  1113. #define W_DO 4
  1114. #define W_DS 3
  1115. #elif 0
  1116. #define W_AM 55
  1117. #define W_AO 16
  1118. #define W_AS 5
  1120. #define W_BM 3
  1121. #define W_BO 32
  1122. #define W_BS 6
  1124. #define W_CM 127
  1125. #define W_CO 64
  1126. #define W_CS 7
  1128. #define W_DM 7
  1129. #define W_DO 8
  1130. #define W_DS 4
  1131. #elif 0
  1132. #define W_AM 97
  1133. #define W_AO 32
  1134. #define W_AS 6
  1136. #define W_BM 63
  1137. #define W_BO 512
  1138. #define W_BS 10
  1140. #define W_CM 13
  1141. #define W_CO 8
  1142. #define W_CS 4
  1144. #define W_DM 15
  1145. #define W_DO 16
  1146. #define W_DS 5
  1148. #else
  1150. #define W_AM 203
  1151. #define W_AO 64
  1152. #define W_AS 7
  1154. #define W_BM 217
  1155. #define W_BO 2048
  1156. #define W_BS 12
  1158. #define W_CM 113
  1159. #define W_CO 64
  1160. #define W_CS 7
  1162. #define W_DM 227
  1163. #define W_DO 128
  1164. #define W_DS 9
  1165. #endif
  1166. static void horizontal_decompose97i(DWTELEM *b, int width){
  1167.     DWTELEM temp[width];
  1168.     const int w2= (width+1)>>1;
  1170.     lift (temp+w2, b    +1, b      , 1, 2, 2, width, -W_AM, W_AO, W_AS, 1, 0);
  1171.     liftS(temp   , b      , temp+w2, 1, 2, 1, width, -W_BM, W_BO, W_BS, 0, 0);
  1172.     lift5(b   +w2, temp+w2, temp   , 1, 1, 1, width,  W_CM, W_CO, W_CS, 1, 0);
  1173.     lift (b      , temp   , b   +w2, 1, 1, 1, width,  W_DM, W_DO, W_DS, 0, 0);
  1174. }
  1177. static void vertical_decompose97iH0(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1178.     int i;
  1179.     
  1180.     for(i=0; i<width; i++){
  1181.         b1[i] -= (W_AM*(b0[i] + b2[i])+W_AO)>>W_AS;
  1182.     }
  1183. }
  1185. static void vertical_decompose97iH1(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1186.     int i;
  1187.     
  1188.     for(i=0; i<width; i++){
  1189. #ifdef lift5
  1190.         b1[i] += (W_CM*(b0[i] + b2[i])+W_CO)>>W_CS;
  1191. #else
  1192.         int r= 3*(b0[i] + b2[i]);
  1193.         r+= r>>4;
  1194.         r+= r>>8;
  1195.         b1[i] += (r+W_CO)>>W_CS;
  1196. #endif
  1197.     }
  1198. }
  1200. static void vertical_decompose97iL0(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1201.     int i;
  1202.     
  1203.     for(i=0; i<width; i++){
  1204. #ifdef liftS
  1205.         b1[i] -= (W_BM*(b0[i] + b2[i])+W_BO)>>W_BS;
  1206. #else
  1207.         b1[i] = (16*4*b1[i] - 4*(b0[i] + b2[i]) + 8*5 + (5<<27)) / (5*16) - (1<<23);
  1208. #endif
  1209.     }
  1210. }
  1212. static void vertical_decompose97iL1(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1213.     int i;
  1214.     
  1215.     for(i=0; i<width; i++){
  1216.         b1[i] += (W_DM*(b0[i] + b2[i])+W_DO)>>W_DS;
  1217.     }
  1218. }
  1220. static void spatial_decompose97i(DWTELEM *buffer, int width, int height, int stride){
  1221.     int y;
  1222.     DWTELEM *b0= buffer + mirror(-4-1, height-1)*stride;
  1223.     DWTELEM *b1= buffer + mirror(-4  , height-1)*stride;
  1224.     DWTELEM *b2= buffer + mirror(-4+1, height-1)*stride;
  1225.     DWTELEM *b3= buffer + mirror(-4+2, height-1)*stride;
  1226.   
  1227.     for(y=-4; y<height; y+=2){
  1228.         DWTELEM *b4= buffer + mirror(y+3, height-1)*stride;
  1229.         DWTELEM *b5= buffer + mirror(y+4, height-1)*stride;
  1231. {START_TIMER
  1232.         if(b3 <= b5)     horizontal_decompose97i(b4, width);
  1233.         if(y+4 < height) horizontal_decompose97i(b5, width);
  1234. if(width>400){
  1235. STOP_TIMER("horizontal_decompose97i")
  1236. }}
  1237.         
  1238. {START_TIMER
  1239.         if(b3 <= b5) vertical_decompose97iH0(b3, b4, b5, width);
  1240.         if(b2 <= b4) vertical_decompose97iL0(b2, b3, b4, width);
  1241.         if(b1 <= b3) vertical_decompose97iH1(b1, b2, b3, width);
  1242.         if(b0 <= b2) vertical_decompose97iL1(b0, b1, b2, width);
  1244. if(width>400){
  1245. STOP_TIMER("vertical_decompose97i")
  1246. }}
  1247.         
  1248.         b0=b2;
  1249.         b1=b3;
  1250.         b2=b4;
  1251.         b3=b5;
  1252.     }
  1253. }
  1255. void ff_spatial_dwt(DWTELEM *buffer, int width, int height, int stride, int type, int decomposition_count){
  1256.     int level;
  1257.     
  1258.     for(level=0; level<decomposition_count; level++){
  1259.         switch(type){
  1260.         case 0: spatial_decompose97i(buffer, width>>level, height>>level, stride<<level); break;
  1261.         case 1: spatial_decompose53i(buffer, width>>level, height>>level, stride<<level); break;
  1262.         case 2: spatial_decomposeX  (buffer, width>>level, height>>level, stride<<level); break;
  1263.         }
  1264.     }
  1265. }
  1267. static void horizontal_compose53i(DWTELEM *b, int width){
  1268.     DWTELEM temp[width];
  1269.     const int width2= width>>1;
  1270.     const int w2= (width+1)>>1;
  1271.     int A1,A2,A3,A4, x;
  1273. #if 0
  1274.     A2= temp[1       ];
  1275.     A4= temp[0       ];
  1276.     A1= temp[0+width2];
  1277.     A1 -= (A2 + A4)>>1;
  1278.     A4 += (A1 + 1)>>1;
  1279.     b[0+width2] = A1;
  1280.     b[0       ] = A4;
  1281.     for(x=1; x+1<width2; x+=2){
  1282.         A3= temp[x+width2];
  1283.         A4= temp[x+1     ];
  1284.         A3 -= (A2 + A4)>>1;
  1285.         A2 += (A1 + A3 + 2)>>2;
  1286.         b[x+width2] = A3;
  1287.         b[x       ] = A2;
  1289.         A1= temp[x+1+width2];
  1290.         A2= temp[x+2       ];
  1291.         A1 -= (A2 + A4)>>1;
  1292.         A4 += (A1 + A3 + 2)>>2;
  1293.         b[x+1+width2] = A1;
  1294.         b[x+1       ] = A4;
  1295.     }
  1296.     A3= temp[width-1];
  1297.     A3 -= A2;
  1298.     A2 += (A1 + A3 + 2)>>2;
  1299.     b[width -1] = A3;
  1300.     b[width2-1] = A2;
  1301. #else   
  1302.     lift(temp   , b   , b+w2, 1, 1, 1, width,  1, 2, 2, 0, 1);
  1303.     lift(temp+w2, b+w2, temp, 1, 1, 1, width, -1, 0, 1, 1, 1);
  1304. #endif
  1305.     for(x=0; x<width2; x++){
  1306.         b[2*x    ]= temp[x   ];
  1307.         b[2*x + 1]= temp[x+w2];
  1308.     }
  1309.     if(width&1)
  1310.         b[2*x    ]= temp[x   ];
  1311. }
  1313. static void vertical_compose53iH0(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1314.     int i;
  1315.     
  1316.     for(i=0; i<width; i++){
  1317.         b1[i] += (b0[i] + b2[i])>>1;
  1318.     }
  1319. }
  1321. static void vertical_compose53iL0(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1322.     int i;
  1323.     
  1324.     for(i=0; i<width; i++){
  1325.         b1[i] -= (b0[i] + b2[i] + 2)>>2;
  1326.     }
  1327. }
  1329. static void spatial_compose53i_buffered_init(dwt_compose_t *cs, slice_buffer * sb, int height, int stride_line){
  1330.     cs->b0 = slice_buffer_get_line(sb, mirror(-1-1, height-1) * stride_line);
  1331.     cs->b1 = slice_buffer_get_line(sb, mirror(-1  , height-1) * stride_line);
  1332.     cs->y = -1;
  1333. }
  1335. static void spatial_compose53i_init(dwt_compose_t *cs, DWTELEM *buffer, int height, int stride){
  1336.     cs->b0 = buffer + mirror(-1-1, height-1)*stride;
  1337.     cs->b1 = buffer + mirror(-1  , height-1)*stride;
  1338.     cs->y = -1;
  1339. }
  1341. static void spatial_compose53i_dy_buffered(dwt_compose_t *cs, slice_buffer * sb, int width, int height, int stride_line){
  1342.     int y= cs->y;
  1343.     int mirror0 = mirror(y-1, height-1);
  1344.     int mirror1 = mirror(y  , height-1);
  1345.     int mirror2 = mirror(y+1, height-1);
  1346.     int mirror3 = mirror(y+2, height-1);
  1347.     
  1348.     DWTELEM *b0= cs->b0;
  1349.     DWTELEM *b1= cs->b1;
  1350.     DWTELEM *b2= slice_buffer_get_line(sb, mirror2 * stride_line);
  1351.     DWTELEM *b3= slice_buffer_get_line(sb, mirror3 * stride_line);
  1353. {START_TIMER
  1354.         if(mirror1 <= mirror3) vertical_compose53iL0(b1, b2, b3, width);
  1355.         if(mirror0 <= mirror2) vertical_compose53iH0(b0, b1, b2, width);
  1356. STOP_TIMER("vertical_compose53i*")}
  1358. {START_TIMER
  1359.         if(y-1 >= 0) horizontal_compose53i(b0, width);
  1360.         if(mirror0 <= mirror2) horizontal_compose53i(b1, width);
  1361. STOP_TIMER("horizontal_compose53i")}
  1363.     cs->b0 = b2;
  1364.     cs->b1 = b3;
  1365.     cs->y += 2;
  1366. }
  1368. static void spatial_compose53i_dy(dwt_compose_t *cs, DWTELEM *buffer, int width, int height, int stride){
  1369.     int y= cs->y;
  1370.     DWTELEM *b0= cs->b0;
  1371.     DWTELEM *b1= cs->b1;
  1372.     DWTELEM *b2= buffer + mirror(y+1, height-1)*stride;
  1373.     DWTELEM *b3= buffer + mirror(y+2, height-1)*stride;
  1375. {START_TIMER
  1376.         if(b1 <= b3) vertical_compose53iL0(b1, b2, b3, width);
  1377.         if(b0 <= b2) vertical_compose53iH0(b0, b1, b2, width);
  1378. STOP_TIMER("vertical_compose53i*")}
  1380. {START_TIMER
  1381.         if(y-1 >= 0) horizontal_compose53i(b0, width);
  1382.         if(b0 <= b2) horizontal_compose53i(b1, width);
  1383. STOP_TIMER("horizontal_compose53i")}
  1385.     cs->b0 = b2;
  1386.     cs->b1 = b3;
  1387.     cs->y += 2;
  1388. }
  1390. static void spatial_compose53i(DWTELEM *buffer, int width, int height, int stride){
  1391.     dwt_compose_t cs;
  1392.     spatial_compose53i_init(&cs, buffer, height, stride);
  1393.     while(cs.y <= height)
  1394.         spatial_compose53i_dy(&cs, buffer, width, height, stride);
  1395. }   
  1397.  
  1398. static void horizontal_compose97i(DWTELEM *b, int width){
  1399.     DWTELEM temp[width];
  1400.     const int w2= (width+1)>>1;
  1402.     lift (temp   , b      , b   +w2, 1, 1, 1, width,  W_DM, W_DO, W_DS, 0, 1);
  1403.     lift5(temp+w2, b   +w2, temp   , 1, 1, 1, width,  W_CM, W_CO, W_CS, 1, 1);
  1404.     liftS(b      , temp   , temp+w2, 2, 1, 1, width, -W_BM, W_BO, W_BS, 0, 1);
  1405.     lift (b+1    , temp+w2, b      , 2, 1, 2, width, -W_AM, W_AO, W_AS, 1, 1);
  1406. }
  1408. static void vertical_compose97iH0(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1409.     int i;
  1410.     
  1411.     for(i=0; i<width; i++){
  1412.         b1[i] += (W_AM*(b0[i] + b2[i])+W_AO)>>W_AS;
  1413.     }
  1414. }
  1416. static void vertical_compose97iH1(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1417.     int i;
  1418.     
  1419.     for(i=0; i<width; i++){
  1420. #ifdef lift5
  1421.         b1[i] -= (W_CM*(b0[i] + b2[i])+W_CO)>>W_CS;
  1422. #else
  1423.         int r= 3*(b0[i] + b2[i]);
  1424.         r+= r>>4;
  1425.         r+= r>>8;
  1426.         b1[i] -= (r+W_CO)>>W_CS;
  1427. #endif
  1428.     }
  1429. }
  1431. static void vertical_compose97iL0(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1432.     int i;
  1433.     
  1434.     for(i=0; i<width; i++){
  1435. #ifdef liftS
  1436.         b1[i] += (W_BM*(b0[i] + b2[i])+W_BO)>>W_BS;
  1437. #else
  1438.         b1[i] += (W_BM*(b0[i] + b2[i])+4*b1[i]+W_BO)>>W_BS;
  1439. #endif
  1440.     }
  1441. }
  1443. static void vertical_compose97iL1(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, int width){
  1444.     int i;
  1445.     
  1446.     for(i=0; i<width; i++){
  1447.         b1[i] -= (W_DM*(b0[i] + b2[i])+W_DO)>>W_DS;
  1448.     }
  1449. }
  1451. static void vertical_compose97i(DWTELEM *b0, DWTELEM *b1, DWTELEM *b2, DWTELEM *b3, DWTELEM *b4, DWTELEM *b5, int width){
  1452.     int i;
  1453.     
  1454.     for(i=0; i<width; i++){
  1455.         int r;
  1456.         b4[i] -= (W_DM*(b3[i] + b5[i])+W_DO)>>W_DS;
  1457. #ifdef lift5
  1458.         b3[i] -= (W_CM*(b2[i] + b4[i])+W_CO)>>W_CS;
  1459. #else
  1460.         r= 3*(b2[i] + b4[i]);
  1461.         r+= r>>4;
  1462.         r+= r>>8;
  1463.         b3[i] -= (r+W_CO)>>W_CS;
  1464. #endif
  1465. #ifdef liftS
  1466.         b2[i] += (W_BM*(b1[i] + b3[i])+W_BO)>>W_BS;
  1467. #else
  1468.         b2[i] += (W_BM*(b1[i] + b3[i])+4*b2[i]+W_BO)>>W_BS;
  1469. #endif
  1470.         b1[i] += (W_AM*(b0[i] + b2[i])+W_AO)>>W_AS;
  1471.     }
  1472. }
  1474. static void spatial_compose97i_buffered_init(dwt_compose_t *cs, slice_buffer * sb, int height, int stride_line){
  1475.     cs->b0 = slice_buffer_get_line(sb, mirror(-3-1, height-1) * stride_line);
  1476.     cs->b1 = slice_buffer_get_line(sb, mirror(-3  , height-1) * stride_line);
  1477.     cs->b2 = slice_buffer_get_line(sb, mirror(-3+1, height-1) * stride_line);
  1478.     cs->b3 = slice_buffer_get_line(sb, mirror(-3+2, height-1) * stride_line);
  1479.     cs->y = -3;
  1480. }
  1482. static void spatial_compose97i_init(dwt_compose_t *cs, DWTELEM *buffer, int height, int stride){
  1483.     cs->b0 = buffer + mirror(-3-1, height-1)*stride;
  1484.     cs->b1 = buffer + mirror(-3  , height-1)*stride;
  1485.     cs->b2 = buffer + mirror(-3+1, height-1)*stride;
  1486.     cs->b3 = buffer + mirror(-3+2, height-1)*stride;
  1487.     cs->y = -3;
  1488. }
  1490. static void spatial_compose97i_dy_buffered(dwt_compose_t *cs, slice_buffer * sb, int width, int height, int stride_line){
  1491.     int y = cs->y;
  1492.     
  1493.     int mirror0 = mirror(y - 1, height - 1);
  1494.     int mirror1 = mirror(y + 0, height - 1);
  1495.     int mirror2 = mirror(y + 1, height - 1);
  1496.     int mirror3 = mirror(y + 2, height - 1);
  1497.     int mirror4 = mirror(y + 3, height - 1);
  1498.     int mirror5 = mirror(y + 4, height - 1);
  1499.     DWTELEM *b0= cs->b0;
  1500.     DWTELEM *b1= cs->b1;
  1501.     DWTELEM *b2= cs->b2;
  1502.     DWTELEM *b3= cs->b3;
  1503.     DWTELEM *b4= slice_buffer_get_line(sb, mirror4 * stride_line);
  1504.     DWTELEM *b5= slice_buffer_get_line(sb, mirror5 * stride_line);
  1505.         
  1506. {START_TIMER
  1507.     if(y>0 && y+4<height){
  1508.         vertical_compose97i(b0, b1, b2, b3, b4, b5, width);
  1509.     }else{
  1510.         if(mirror3 <= mirror5) vertical_compose97iL1(b3, b4, b5, width);
  1511.         if(mirror2 <= mirror4) vertical_compose97iH1(b2, b3, b4, width);
  1512.         if(mirror1 <= mirror3) vertical_compose97iL0(b1, b2, b3, width);
  1513.         if(mirror0 <= mirror2) vertical_compose97iH0(b0, b1, b2, width);
  1514.     }
  1515. if(width>400){
  1516. STOP_TIMER("vertical_compose97i")}}
  1518. {START_TIMER
  1519.         if(y-1>=  0) horizontal_compose97i(b0, width);
  1520.         if(mirror0 <= mirror2) horizontal_compose97i(b1, width);
  1521. if(width>400 && mirror0 <= mirror2){
  1522. STOP_TIMER("horizontal_compose97i")}}
  1524.     cs->b0=b2;
  1525.     cs->b1=b3;
  1526.     cs->b2=b4;
  1527.     cs->b3=b5;
  1528.     cs->y += 2;
  1529. }
  1531. static void spatial_compose97i_dy(dwt_compose_t *cs, DWTELEM *buffer, int width, int height, int stride){
  1532.     int y = cs->y;
  1533.     DWTELEM *b0= cs->b0;
  1534.     DWTELEM *b1= cs->b1;
  1535.     DWTELEM *b2= cs->b2;
  1536.     DWTELEM *b3= cs->b3;
  1537.     DWTELEM *b4= buffer + mirror(y+3, height-1)*stride;
  1538.     DWTELEM *b5= buffer + mirror(y+4, height-1)*stride;
  1540.         if(stride == width && y+4 < height && 0){ 
  1541.             int x;
  1542.             for(x=0; x<width/2; x++)
  1543.                 b5[x] += 64*2;
  1544.             for(; x<width; x++)
  1545.                 b5[x] += 169*2;
  1546.         }
  1547.         
  1548. {START_TIMER
  1549.         if(b3 <= b5) vertical_compose97iL1(b3, b4, b5, width);
  1550.         if(b2 <= b4) vertical_compose97iH1(b2, b3, b4, width);
  1551.         if(b1 <= b3) vertical_compose97iL0(b1, b2, b3, width);
  1552.         if(b0 <= b2) vertical_compose97iH0(b0, b1, b2, width);
  1553. if(width>400){
  1554. STOP_TIMER("vertical_compose97i")}}
  1556. {START_TIMER
  1557.         if(y-1>=  0) horizontal_compose97i(b0, width);
  1558.         if(b0 <= b2) horizontal_compose97i(b1, width);
  1559. if(width>400 && b0 <= b2){
  1560. STOP_TIMER("horizontal_compose97i")}}
  1562.     cs->b0=b2;
  1563.     cs->b1=b3;
  1564.     cs->b2=b4;
  1565.     cs->b3=b5;
  1566.     cs->y += 2;
  1567. }
  1569. static void spatial_compose97i(DWTELEM *buffer, int width, int height, int stride){
  1570.     dwt_compose_t cs;
  1571.     spatial_compose97i_init(&cs, buffer, height, stride);
  1572.     while(cs.y <= height)
  1573.         spatial_compose97i_dy(&cs, buffer, width, height, stride);
  1574. }
  1576. void ff_spatial_idwt_buffered_init(dwt_compose_t *cs, slice_buffer * sb, int width, int height, int stride_line, int type, int decomposition_count){
  1577.     int level;
  1578.     for(level=decomposition_count-1; level>=0; level--){
  1579.         switch(type){
  1580.         case 0: spatial_compose97i_buffered_init(cs+level, sb, height>>level, stride_line<<level); break;
  1581.         case 1: spatial_compose53i_buffered_init(cs+level, sb, height>>level, stride_line<<level); break;
  1582.         /* not slicified yet */
  1583.         case 2: /*spatial_composeX(buffer, width>>level, height>>level, stride<<level); break;*/
  1584.           av_log(NULL, AV_LOG_ERROR, "spatial_composeX neither buffered nor slicified yet.n"); break;
  1585.         }
  1586.     }
  1587. }
  1589. void ff_spatial_idwt_init(dwt_compose_t *cs, DWTELEM *buffer, int width, int height, int stride, int type, int decomposition_count){
  1590.     int level;
  1591.     for(level=decomposition_count-1; level>=0; level--){
  1592.         switch(type){
  1593.         case 0: spatial_compose97i_init(cs+level, buffer, height>>level, stride<<level); break;
  1594.         case 1: spatial_compose53i_init(cs+level, buffer, height>>level, stride<<level); break;
  1595.         /* not slicified yet */
  1596.         case 2: spatial_composeX(buffer, width>>level, height>>level, stride<<level); break;
  1597.         }
  1598.     }
  1599. }
  1601. void ff_spatial_idwt_slice(dwt_compose_t *cs, DWTELEM *buffer, int width, int height, int stride, int type, int decomposition_count, int y){
  1602.     const int support = type==1 ? 3 : 5;
  1603.     int level;
  1604.     if(type==2) return;
  1606.     for(level=decomposition_count-1; level>=0; level--){
  1607.         while(cs[level].y <= FFMIN((y>>level)+support, height>>level)){
  1608.             switch(type){
  1609.             case 0: spatial_compose97i_dy(cs+level, buffer, width>>level, height>>level, stride<<level);
  1610.                     break;
  1611.             case 1: spatial_compose53i_dy(cs+level, buffer, width>>level, height>>level, stride<<level);
  1612.                     break;
  1613.             case 2: break;
  1614.             }
  1615.         }
  1616.     }
  1617. }
  1619. void ff_spatial_idwt_buffered_slice(dwt_compose_t *cs, slice_buffer * slice_buf, int width, int height, int stride_line, int type, int decomposition_count, int y){
  1620.     const int support = type==1 ? 3 : 5;
  1621.     int level;
  1622.     if(type==2) return;
  1624.     for(level=decomposition_count-1; level>=0; level--){
  1625.         while(cs[level].y <= FFMIN((y>>level)+support, height>>level)){
  1626.             switch(type){
  1627.             case 0: spatial_compose97i_dy_buffered(cs+level, slice_buf, width>>level, height>>level, stride_line<<level);
  1628.                     break;
  1629.             case 1: spatial_compose53i_dy_buffered(cs+level, slice_buf, width>>level, height>>level, stride_line<<level);
  1630.                     break;
  1631.             case 2: break;
  1632.             }
  1633.         }
  1634.     }
  1635. }
  1637. void ff_spatial_idwt(DWTELEM *buffer, int width, int height, int stride, int type, int decomposition_count){
  1638.     if(type==2){
  1639.         int level;
  1640.         for(level=decomposition_count-1; level>=0; level--)
  1641.             spatial_composeX  (buffer, width>>level, height>>level, stride<<level);
  1642.     }else{
  1643.         dwt_compose_t cs[MAX_DECOMPOSITIONS];
  1644.         int y;
  1645.         ff_spatial_idwt_init(cs, buffer, width, height, stride, type, decomposition_count);
  1646.         for(y=0; y<height; y+=4)
  1647.             ff_spatial_idwt_slice(cs, buffer, width, height, stride, type, decomposition_count, y);
  1648.     }
  1649. }
  1651. static int encode_subband_c0run(SnowContext *s, SubBand *b, DWTELEM *src, DWTELEM *parent, int stride, int orientation){
  1652.     const int w= b->width;
  1653.     const int h= b->height;
  1654.     int x, y;
  1656.     if(1){
  1657.         int run=0;
  1658.         int runs[w*h];
  1659.         int run_index=0;
  1660.         int max_index;
  1661.                 
  1662.         for(y=0; y<h; y++){
  1663.             for(x=0; x<w; x++){
  1664.                 int v, p=0;
  1665.                 int /*ll=0, */l=0, lt=0, t=0, rt=0;
  1666.                 v= src[x + y*stride];
  1668.                 if(y){
  1669.                     t= src[x + (y-1)*stride];
  1670.                     if(x){
  1671.                         lt= src[x - 1 + (y-1)*stride];
  1672.                     }
  1673.                     if(x + 1 < w){
  1674.                         rt= src[x + 1 + (y-1)*stride];
  1675.                     }
  1676.                 }
  1677.                 if(x){
  1678.                     l= src[x - 1 + y*stride];
  1679.                     /*if(x > 1){
  1680.                         if(orientation==1) ll= src[y + (x-2)*stride];
  1681.                         else               ll= src[x - 2 + y*stride];
  1682.                     }*/
  1683.                 }
  1684.                 if(parent){
  1685.                     int px= x>>1;
  1686.                     int py= y>>1;
  1687.                     if(px<b->parent->width && py<b->parent->height) 
  1688.                         p= parent[px + py*2*stride];
  1689.                 }
  1690.                 if(!(/*ll|*/l|lt|t|rt|p)){
  1691.                     if(v){
  1692.                         runs[run_index++]= run;
  1693.                         run=0;
  1694.                     }else{
  1695.                         run++;
  1696.                     }
  1697.                 }
  1698.             }
  1699.         }
  1700.         max_index= run_index;
  1701.         runs[run_index++]= run;
  1702.         run_index=0;
  1703.         run= runs[run_index++];
  1705.         put_symbol2(&s->c, b->state[30], max_index, 0);
  1706.         if(run_index <= max_index)
  1707.             put_symbol2(&s->c, b->state[1], run, 3);
  1708.         
  1709.         for(y=0; y<h; y++){
  1710.             if(s->c.bytestream_end - s->c.bytestream < w*40){
  1711.                 av_log(s->avctx, AV_LOG_ERROR, "encoded frame too largen");
  1712.                 return -1;
  1713.             }
  1714.             for(x=0; x<w; x++){
  1715.                 int v, p=0;
  1716.                 int /*ll=0, */l=0, lt=0, t=0, rt=0;
  1717.                 v= src[x + y*stride];
  1719.                 if(y){
  1720.                     t= src[x + (y-1)*stride];
  1721.                     if(x){
  1722.                         lt= src[x - 1 + (y-1)*stride];
  1723.                     }
  1724.                     if(x + 1 < w){
  1725.                         rt= src[x + 1 + (y-1)*stride];
  1726.                     }
  1727.                 }
  1728.                 if(x){
  1729.                     l= src[x - 1 + y*stride];
  1730.                     /*if(x > 1){
  1731.                         if(orientation==1) ll= src[y + (x-2)*stride];
  1732.                         else               ll= src[x - 2 + y*stride];
  1733.                     }*/
  1734.                 }
  1735.                 if(parent){
  1736.                     int px= x>>1;
  1737.                     int py= y>>1;
  1738.                     if(px<b->parent->width && py<b->parent->height) 
  1739.                         p= parent[px + py*2*stride];
  1740.                 }
  1741.                 if(/*ll|*/l|lt|t|rt|p){
  1742.                     int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p));
  1744.                     put_rac(&s->c, &b->state[0][context], !!v);
  1745.                 }else{
  1746.                     if(!run){
  1747.                         run= runs[run_index++];
  1749.                         if(run_index <= max_index)
  1750.                             put_symbol2(&s->c, b->state[1], run, 3);
  1751.                         assert(v);
  1752.                     }else{
  1753.                         run--;
  1754.                         assert(!v);
  1755.                     }
  1756.                 }
  1757.                 if(v){
  1758.                     int context= av_log2(/*ABS(ll) + */3*ABS(l) + ABS(lt) + 2*ABS(t) + ABS(rt) + ABS(p));
  1759.                     int l2= 2*ABS(l) + (l<0);
  1760.                     int t2= 2*ABS(t) + (t<0);
  1762.                     put_symbol2(&s->c, b->state[context + 2], ABS(v)-1, context-4);
  1763.                     put_rac(&s->c, &b->state[0][16 + 1 + 3 + quant3bA[l2&0xFF] + 3*quant3bA[t2&0xFF]], v<0);
  1764.                 }
  1765.             }
  1766.         }
  1767.     }
  1768.     return 0;
  1769. }
  1771. static int encode_subband(SnowContext *s, SubBand *b, DWTELEM *src, DWTELEM *parent, int stride, int orientation){    
  1772. //    encode_subband_qtree(s, b, src, parent, stride, orientation);
  1773. //    encode_subband_z0run(s, b, src, parent, stride, orientation);
  1774.     return encode_subband_c0run(s, b, src, parent, stride, orientation);
  1775. //    encode_subband_dzr(s, b, src, parent, stride, orientation);
  1776. }
  1778. static inline void unpack_coeffs(SnowContext *s, SubBand *b, SubBand * parent, int orientation){
  1779.     const int w= b->width;
  1780.     const int h= b->height;
  1781.     int x,y;
  1782.     
  1783.     if(1){
  1784.         int run, runs;
  1785.         x_and_coeff *xc= b->x_coeff;
  1786.         x_and_coeff *prev_xc= NULL;
  1787.         x_and_coeff *prev2_xc= xc;
  1788.         x_and_coeff *parent_xc= parent ? parent->x_coeff : NULL;
  1789.         x_and_coeff *prev_parent_xc= parent_xc;
  1791.         runs= get_symbol2(&s->c, b->state[30], 0);
  1792.         if(runs-- > 0) run= get_symbol2(&s->c, b->state[1], 3);
  1793.         else           run= INT_MAX;
  1795.         for(y=0; y<h; y++){
  1796.             int v=0;
  1797.             int lt=0, t=0, rt=0;
  1799.             if(y && prev_xc->x == 0){
  1800.                 rt= prev_xc->coeff;
  1801.             }
  1802.             for(x=0; x<w; x++){
  1803.                 int p=0;
  1804.                 const int l= v;
  1805.                 
  1806.                 lt= t; t= rt;
  1808.                 if(y){
  1809.                     if(prev_xc->x <= x)
  1810.                         prev_xc++;
  1811.                     if(prev_xc->x == x + 1)
  1812.                         rt= prev_xc->coeff;
  1813.                     else
  1814.                         rt=0;
  1815.                 }
  1816.                 if(parent_xc){
  1817.                     if(x>>1 > parent_xc->x){
  1818.                         parent_xc++;
  1819.                     }
  1820.                     if(x>>1 == parent_xc->x){
  1821.                         p= parent_xc->coeff;
  1822.                     }
  1823.                 }
  1824.                 if(/*ll|*/l|lt|t|rt|p){
  1825.                     int context= av_log2(/*ABS(ll) + */3*(l>>1) + (lt>>1) + (t&~1) + (rt>>1) + (p>>1));
  1827.                     v=get_rac(&s->c, &b->state[0][context]);
  1828.                     if(v){
  1829.                         v= 2*(get_symbol2(&s->c, b->state[context + 2], context-4) + 1);
  1830.                         v+=get_rac(&s->c, &b->state[0][16 + 1 + 3 + quant3bA[l&0xFF] + 3*quant3bA[t&0xFF]]);
  1831.                         
  1832.                         xc->x=x;
  1833.                         (xc++)->coeff= v;
  1834.                     }
  1835.                 }else{
  1836.                     if(!run){
  1837.                         if(runs-- > 0) run= get_symbol2(&s->c, b->state[1], 3);
  1838.                         else           run= INT_MAX;
  1839.                         v= 2*(get_symbol2(&s->c, b->state[0 + 2], 0-4) + 1);
  1840.                         v+=get_rac(&s->c, &b->state[0][16 + 1 + 3]);
  1841.                         
  1842.                         xc->x=x;
  1843.                         (xc++)->coeff= v;
  1844.                     }else{
  1845.                         int max_run;
  1846.                         run--;
  1847.                         v=0;
  1849.                         if(y) max_run= FFMIN(run, prev_xc->x - x - 2);
  1850.                         else  max_run= FFMIN(run, w-x-1);
  1851.                         if(parent_xc)
  1852.                             max_run= FFMIN(max_run, 2*parent_xc->x - x - 1);
  1853.                         x+= max_run;
  1854.                         run-= max_run;
  1855.                     }
  1856.                 }
  1857.             }
  1858.             (xc++)->x= w+1; //end marker
  1859.             prev_xc= prev2_xc;
  1860.             prev2_xc= xc;
  1861.             
  1862.             if(parent_xc){
  1863.                 if(y&1){
  1864.                     while(parent_xc->x != parent->width+1)
  1865.                         parent_xc++;
  1866.                     parent_xc++;
  1867.                     prev_parent_xc= parent_xc;
  1868.                 }else{
  1869.                     parent_xc= prev_parent_xc;
  1870.                 }
  1871.             }
  1872.         }
  1874.         (xc++)->x= w+1; //end marker
  1875.     }
  1876. }
  1878. static inline void decode_subband_slice_buffered(SnowContext *s, SubBand *b, slice_buffer * sb, int start_y, int h, int save_state[1]){
  1879.     const int w= b->width;
  1880.     int x,y;
  1881.     const int qlog= clip(s->qlog + b->qlog, 0, QROOT*16);
  1882.     int qmul= qexp[qlog&(QROOT-1)]<<(qlog>>QSHIFT);
  1883.     int qadd= (s->qbias*qmul)>>QBIAS_SHIFT;
  1884.     int new_index = 0;
  1885.     
  1886.     START_TIMER
  1888.     if(b->buf == s->spatial_dwt_buffer || s->qlog == LOSSLESS_QLOG){
  1889.         qadd= 0;
  1890.         qmul= 1<<QEXPSHIFT;
  1891.     }
  1893.     /* If we are on the second or later slice, restore our index. */
  1894.     if (start_y != 0)
  1895.         new_index = save_state[0];
  1897.         
  1898.     for(y=start_y; y<h; y++){
  1899.         int x = 0;
  1900.         int v;
  1901.         DWTELEM * line = slice_buffer_get_line(sb, y * b->stride_line + b->buf_y_offset) + b->buf_x_offset;
  1902.         memset(line, 0, b->width*sizeof(DWTELEM));
  1903.         v = b->x_coeff[new_index].coeff;
  1904.         x = b->x_coeff[new_index++].x;
  1905.         while(x < w)
  1906.         {
  1907.             register int t= ( (v>>1)*qmul + qadd)>>QEXPSHIFT;
  1908.             register int u= -(v&1);
  1909.             line[x] = (t^u) - u;
  1911.             v = b->x_coeff[new_index].coeff;
  1912.             x = b->x_coeff[new_index++].x;
  1913.         }
  1914.     }
  1915.     if(w > 200 && start_y != 0/*level+1 == s->spatial_decomposition_count*/){
  1916.         STOP_TIMER("decode_subband")
  1917.     }
  1918.         
  1919.     /* Save our variables for the next slice. */
  1920.     save_state[0] = new_index;
  1921.         
  1922.     return;
  1923. }
  1925. static void reset_contexts(SnowContext *s){
  1926.     int plane_index, level, orientation;
  1928.     for(plane_index=0; plane_index<3; plane_index++){
  1929.         for(level=0; level<s->spatial_decomposition_count; level++){
  1930.             for(orientation=level ? 1:0; orientation<4; orientation++){
  1931.                 memset(s->plane[plane_index].band[level][orientation].state, MID_STATE, sizeof(s->plane[plane_index].band[level][orientation].state));
  1932.             }
  1933.         }
  1934.     }
  1935.     memset(s->header_state, MID_STATE, sizeof(s->header_state));
  1936.     memset(s->block_state, MID_STATE, sizeof(s->block_state));
  1937. }
  1939. static int alloc_blocks(SnowContext *s){
  1940.     int w= -((-s->avctx->width )>>LOG2_MB_SIZE);
  1941.     int h= -((-s->avctx->height)>>LOG2_MB_SIZE);
  1942.     
  1943.     s->b_width = w;
  1944.     s->b_height= h;
  1945.     
  1946.     s->block= av_mallocz(w * h * sizeof(BlockNode) << (s->block_max_depth*2));
  1947.     return 0;
  1948. }
  1950. static inline void copy_rac_state(RangeCoder *d, RangeCoder *s){
  1951.     uint8_t *bytestream= d->bytestream;
  1952.     uint8_t *bytestream_start= d->bytestream_start;
  1953.     *d= *s;
  1954.     d->bytestream= bytestream;
  1955.     d->bytestream_start= bytestream_start;
  1956. }
  1958. //near copy & paste from dsputil, FIXME
  1959. static int pix_sum(uint8_t * pix, int line_size, int w)
  1960. {
  1961.     int s, i, j;
  1963.     s = 0;
  1964.     for (i = 0; i < w; i++) {
  1965.         for (j = 0; j < w; j++) {
  1966.             s += pix[0];
  1967.             pix ++;
  1968.         }
  1969.         pix += line_size - w;
  1970.     }
  1971.     return s;
  1972. }
  1974. //near copy & paste from dsputil, FIXME
  1975. static int pix_norm1(uint8_t * pix, int line_size, int w)
  1976. {
  1977.     int s, i, j;
  1978.     uint32_t *sq = squareTbl + 256;
  1980.     s = 0;
  1981.     for (i = 0; i < w; i++) {
  1982.         for (j = 0; j < w; j ++) {
  1983.             s += sq[pix[0]];
  1984.             pix ++;
  1985.         }
  1986.         pix += line_size - w;
  1987.     }
  1988.     return s;
  1989. }
  1991. 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){
  1992.     const int w= s->b_width << s->block_max_depth;
  1993.     const int rem_depth= s->block_max_depth - level;
  1994.     const int index= (x + y*w) << rem_depth;
  1995.     const int block_w= 1<<rem_depth;
  1996.     BlockNode block;
  1997.     int i,j;
  1998.     
  1999.     block.color[0]= l;
  2000.     block.color[1]= cb;
  2001.     block.color[2]= cr;
  2002.     block.mx= mx;
  2003.     block.my= my;
  2004.     block.type= type;
  2005.     block.level= level;
  2007.     for(j=0; j<block_w; j++){
  2008.         for(i=0; i<block_w; i++){