You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

571 lines
15KB

  1. /*
  2. * Simple IDCT
  3. *
  4. * Copyright (c) 2001 Michael Niedermayer <michaelni@gmx.at>
  5. *
  6. * This file is part of Libav.
  7. *
  8. * Libav is free software; you can redistribute it and/or
  9. * modify it under the terms of the GNU Lesser General Public
  10. * License as published by the Free Software Foundation; either
  11. * version 2.1 of the License, or (at your option) any later version.
  12. *
  13. * Libav is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  16. * Lesser General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU Lesser General Public
  19. * License along with Libav; if not, write to the Free Software
  20. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  21. */
  22. /**
  23. * @file
  24. * simpleidct in C.
  25. */
  26. /*
  27. based upon some outcommented c code from mpeg2dec (idct_mmx.c
  28. written by Aaron Holtzman <aholtzma@ess.engr.uvic.ca>)
  29. */
  30. #include "avcodec.h"
  31. #include "dsputil.h"
  32. #include "mathops.h"
  33. #include "simple_idct.h"
  34. #define W1 22725 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
  35. #define W2 21407 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
  36. #define W3 19266 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
  37. #define W4 16383 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
  38. #define W5 12873 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
  39. #define W6 8867 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
  40. #define W7 4520 //cos(i*M_PI/16)*sqrt(2)*(1<<14) + 0.5
  41. #define ROW_SHIFT 11
  42. #define COL_SHIFT 20 // 6
  43. static inline void idctRowCondDC (DCTELEM * row)
  44. {
  45. int a0, a1, a2, a3, b0, b1, b2, b3;
  46. #if HAVE_FAST_64BIT
  47. uint64_t temp;
  48. #else
  49. uint32_t temp;
  50. #endif
  51. #if HAVE_FAST_64BIT
  52. #if HAVE_BIGENDIAN
  53. #define ROW0_MASK 0xffff000000000000LL
  54. #else
  55. #define ROW0_MASK 0xffffLL
  56. #endif
  57. if(sizeof(DCTELEM)==2){
  58. if ( ((((uint64_t *)row)[0] & ~ROW0_MASK) |
  59. ((uint64_t *)row)[1]) == 0) {
  60. temp = (row[0] << 3) & 0xffff;
  61. temp += temp << 16;
  62. temp += temp << 32;
  63. ((uint64_t *)row)[0] = temp;
  64. ((uint64_t *)row)[1] = temp;
  65. return;
  66. }
  67. }else{
  68. if (!(row[1]|row[2]|row[3]|row[4]|row[5]|row[6]|row[7])) {
  69. row[0]=row[1]=row[2]=row[3]=row[4]=row[5]=row[6]=row[7]= row[0] << 3;
  70. return;
  71. }
  72. }
  73. #else
  74. if(sizeof(DCTELEM)==2){
  75. if (!(((uint32_t*)row)[1] |
  76. ((uint32_t*)row)[2] |
  77. ((uint32_t*)row)[3] |
  78. row[1])) {
  79. temp = (row[0] << 3) & 0xffff;
  80. temp += temp << 16;
  81. ((uint32_t*)row)[0]=((uint32_t*)row)[1] =
  82. ((uint32_t*)row)[2]=((uint32_t*)row)[3] = temp;
  83. return;
  84. }
  85. }else{
  86. if (!(row[1]|row[2]|row[3]|row[4]|row[5]|row[6]|row[7])) {
  87. row[0]=row[1]=row[2]=row[3]=row[4]=row[5]=row[6]=row[7]= row[0] << 3;
  88. return;
  89. }
  90. }
  91. #endif
  92. a0 = (W4 * row[0]) + (1 << (ROW_SHIFT - 1));
  93. a1 = a0;
  94. a2 = a0;
  95. a3 = a0;
  96. /* no need to optimize : gcc does it */
  97. a0 += W2 * row[2];
  98. a1 += W6 * row[2];
  99. a2 -= W6 * row[2];
  100. a3 -= W2 * row[2];
  101. b0 = MUL16(W1, row[1]);
  102. MAC16(b0, W3, row[3]);
  103. b1 = MUL16(W3, row[1]);
  104. MAC16(b1, -W7, row[3]);
  105. b2 = MUL16(W5, row[1]);
  106. MAC16(b2, -W1, row[3]);
  107. b3 = MUL16(W7, row[1]);
  108. MAC16(b3, -W5, row[3]);
  109. #if HAVE_FAST_64BIT
  110. temp = ((uint64_t*)row)[1];
  111. #else
  112. temp = ((uint32_t*)row)[2] | ((uint32_t*)row)[3];
  113. #endif
  114. if (temp != 0) {
  115. a0 += W4*row[4] + W6*row[6];
  116. a1 += - W4*row[4] - W2*row[6];
  117. a2 += - W4*row[4] + W2*row[6];
  118. a3 += W4*row[4] - W6*row[6];
  119. MAC16(b0, W5, row[5]);
  120. MAC16(b0, W7, row[7]);
  121. MAC16(b1, -W1, row[5]);
  122. MAC16(b1, -W5, row[7]);
  123. MAC16(b2, W7, row[5]);
  124. MAC16(b2, W3, row[7]);
  125. MAC16(b3, W3, row[5]);
  126. MAC16(b3, -W1, row[7]);
  127. }
  128. row[0] = (a0 + b0) >> ROW_SHIFT;
  129. row[7] = (a0 - b0) >> ROW_SHIFT;
  130. row[1] = (a1 + b1) >> ROW_SHIFT;
  131. row[6] = (a1 - b1) >> ROW_SHIFT;
  132. row[2] = (a2 + b2) >> ROW_SHIFT;
  133. row[5] = (a2 - b2) >> ROW_SHIFT;
  134. row[3] = (a3 + b3) >> ROW_SHIFT;
  135. row[4] = (a3 - b3) >> ROW_SHIFT;
  136. }
  137. static inline void idctSparseColPut (uint8_t *dest, int line_size,
  138. DCTELEM * col)
  139. {
  140. int a0, a1, a2, a3, b0, b1, b2, b3;
  141. uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  142. /* XXX: I did that only to give same values as previous code */
  143. a0 = W4 * (col[8*0] + ((1<<(COL_SHIFT-1))/W4));
  144. a1 = a0;
  145. a2 = a0;
  146. a3 = a0;
  147. a0 += + W2*col[8*2];
  148. a1 += + W6*col[8*2];
  149. a2 += - W6*col[8*2];
  150. a3 += - W2*col[8*2];
  151. b0 = MUL16(W1, col[8*1]);
  152. b1 = MUL16(W3, col[8*1]);
  153. b2 = MUL16(W5, col[8*1]);
  154. b3 = MUL16(W7, col[8*1]);
  155. MAC16(b0, + W3, col[8*3]);
  156. MAC16(b1, - W7, col[8*3]);
  157. MAC16(b2, - W1, col[8*3]);
  158. MAC16(b3, - W5, col[8*3]);
  159. if(col[8*4]){
  160. a0 += + W4*col[8*4];
  161. a1 += - W4*col[8*4];
  162. a2 += - W4*col[8*4];
  163. a3 += + W4*col[8*4];
  164. }
  165. if (col[8*5]) {
  166. MAC16(b0, + W5, col[8*5]);
  167. MAC16(b1, - W1, col[8*5]);
  168. MAC16(b2, + W7, col[8*5]);
  169. MAC16(b3, + W3, col[8*5]);
  170. }
  171. if(col[8*6]){
  172. a0 += + W6*col[8*6];
  173. a1 += - W2*col[8*6];
  174. a2 += + W2*col[8*6];
  175. a3 += - W6*col[8*6];
  176. }
  177. if (col[8*7]) {
  178. MAC16(b0, + W7, col[8*7]);
  179. MAC16(b1, - W5, col[8*7]);
  180. MAC16(b2, + W3, col[8*7]);
  181. MAC16(b3, - W1, col[8*7]);
  182. }
  183. dest[0] = cm[(a0 + b0) >> COL_SHIFT];
  184. dest += line_size;
  185. dest[0] = cm[(a1 + b1) >> COL_SHIFT];
  186. dest += line_size;
  187. dest[0] = cm[(a2 + b2) >> COL_SHIFT];
  188. dest += line_size;
  189. dest[0] = cm[(a3 + b3) >> COL_SHIFT];
  190. dest += line_size;
  191. dest[0] = cm[(a3 - b3) >> COL_SHIFT];
  192. dest += line_size;
  193. dest[0] = cm[(a2 - b2) >> COL_SHIFT];
  194. dest += line_size;
  195. dest[0] = cm[(a1 - b1) >> COL_SHIFT];
  196. dest += line_size;
  197. dest[0] = cm[(a0 - b0) >> COL_SHIFT];
  198. }
  199. static inline void idctSparseColAdd (uint8_t *dest, int line_size,
  200. DCTELEM * col)
  201. {
  202. int a0, a1, a2, a3, b0, b1, b2, b3;
  203. uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  204. /* XXX: I did that only to give same values as previous code */
  205. a0 = W4 * (col[8*0] + ((1<<(COL_SHIFT-1))/W4));
  206. a1 = a0;
  207. a2 = a0;
  208. a3 = a0;
  209. a0 += + W2*col[8*2];
  210. a1 += + W6*col[8*2];
  211. a2 += - W6*col[8*2];
  212. a3 += - W2*col[8*2];
  213. b0 = MUL16(W1, col[8*1]);
  214. b1 = MUL16(W3, col[8*1]);
  215. b2 = MUL16(W5, col[8*1]);
  216. b3 = MUL16(W7, col[8*1]);
  217. MAC16(b0, + W3, col[8*3]);
  218. MAC16(b1, - W7, col[8*3]);
  219. MAC16(b2, - W1, col[8*3]);
  220. MAC16(b3, - W5, col[8*3]);
  221. if(col[8*4]){
  222. a0 += + W4*col[8*4];
  223. a1 += - W4*col[8*4];
  224. a2 += - W4*col[8*4];
  225. a3 += + W4*col[8*4];
  226. }
  227. if (col[8*5]) {
  228. MAC16(b0, + W5, col[8*5]);
  229. MAC16(b1, - W1, col[8*5]);
  230. MAC16(b2, + W7, col[8*5]);
  231. MAC16(b3, + W3, col[8*5]);
  232. }
  233. if(col[8*6]){
  234. a0 += + W6*col[8*6];
  235. a1 += - W2*col[8*6];
  236. a2 += + W2*col[8*6];
  237. a3 += - W6*col[8*6];
  238. }
  239. if (col[8*7]) {
  240. MAC16(b0, + W7, col[8*7]);
  241. MAC16(b1, - W5, col[8*7]);
  242. MAC16(b2, + W3, col[8*7]);
  243. MAC16(b3, - W1, col[8*7]);
  244. }
  245. dest[0] = cm[dest[0] + ((a0 + b0) >> COL_SHIFT)];
  246. dest += line_size;
  247. dest[0] = cm[dest[0] + ((a1 + b1) >> COL_SHIFT)];
  248. dest += line_size;
  249. dest[0] = cm[dest[0] + ((a2 + b2) >> COL_SHIFT)];
  250. dest += line_size;
  251. dest[0] = cm[dest[0] + ((a3 + b3) >> COL_SHIFT)];
  252. dest += line_size;
  253. dest[0] = cm[dest[0] + ((a3 - b3) >> COL_SHIFT)];
  254. dest += line_size;
  255. dest[0] = cm[dest[0] + ((a2 - b2) >> COL_SHIFT)];
  256. dest += line_size;
  257. dest[0] = cm[dest[0] + ((a1 - b1) >> COL_SHIFT)];
  258. dest += line_size;
  259. dest[0] = cm[dest[0] + ((a0 - b0) >> COL_SHIFT)];
  260. }
  261. static inline void idctSparseCol (DCTELEM * col)
  262. {
  263. int a0, a1, a2, a3, b0, b1, b2, b3;
  264. /* XXX: I did that only to give same values as previous code */
  265. a0 = W4 * (col[8*0] + ((1<<(COL_SHIFT-1))/W4));
  266. a1 = a0;
  267. a2 = a0;
  268. a3 = a0;
  269. a0 += + W2*col[8*2];
  270. a1 += + W6*col[8*2];
  271. a2 += - W6*col[8*2];
  272. a3 += - W2*col[8*2];
  273. b0 = MUL16(W1, col[8*1]);
  274. b1 = MUL16(W3, col[8*1]);
  275. b2 = MUL16(W5, col[8*1]);
  276. b3 = MUL16(W7, col[8*1]);
  277. MAC16(b0, + W3, col[8*3]);
  278. MAC16(b1, - W7, col[8*3]);
  279. MAC16(b2, - W1, col[8*3]);
  280. MAC16(b3, - W5, col[8*3]);
  281. if(col[8*4]){
  282. a0 += + W4*col[8*4];
  283. a1 += - W4*col[8*4];
  284. a2 += - W4*col[8*4];
  285. a3 += + W4*col[8*4];
  286. }
  287. if (col[8*5]) {
  288. MAC16(b0, + W5, col[8*5]);
  289. MAC16(b1, - W1, col[8*5]);
  290. MAC16(b2, + W7, col[8*5]);
  291. MAC16(b3, + W3, col[8*5]);
  292. }
  293. if(col[8*6]){
  294. a0 += + W6*col[8*6];
  295. a1 += - W2*col[8*6];
  296. a2 += + W2*col[8*6];
  297. a3 += - W6*col[8*6];
  298. }
  299. if (col[8*7]) {
  300. MAC16(b0, + W7, col[8*7]);
  301. MAC16(b1, - W5, col[8*7]);
  302. MAC16(b2, + W3, col[8*7]);
  303. MAC16(b3, - W1, col[8*7]);
  304. }
  305. col[0 ] = ((a0 + b0) >> COL_SHIFT);
  306. col[8 ] = ((a1 + b1) >> COL_SHIFT);
  307. col[16] = ((a2 + b2) >> COL_SHIFT);
  308. col[24] = ((a3 + b3) >> COL_SHIFT);
  309. col[32] = ((a3 - b3) >> COL_SHIFT);
  310. col[40] = ((a2 - b2) >> COL_SHIFT);
  311. col[48] = ((a1 - b1) >> COL_SHIFT);
  312. col[56] = ((a0 - b0) >> COL_SHIFT);
  313. }
  314. void ff_simple_idct_put(uint8_t *dest, int line_size, DCTELEM *block)
  315. {
  316. int i;
  317. for(i=0; i<8; i++)
  318. idctRowCondDC(block + i*8);
  319. for(i=0; i<8; i++)
  320. idctSparseColPut(dest + i, line_size, block + i);
  321. }
  322. void ff_simple_idct_add(uint8_t *dest, int line_size, DCTELEM *block)
  323. {
  324. int i;
  325. for(i=0; i<8; i++)
  326. idctRowCondDC(block + i*8);
  327. for(i=0; i<8; i++)
  328. idctSparseColAdd(dest + i, line_size, block + i);
  329. }
  330. void ff_simple_idct(DCTELEM *block)
  331. {
  332. int i;
  333. for(i=0; i<8; i++)
  334. idctRowCondDC(block + i*8);
  335. for(i=0; i<8; i++)
  336. idctSparseCol(block + i);
  337. }
  338. /* 2x4x8 idct */
  339. #define CN_SHIFT 12
  340. #define C_FIX(x) ((int)((x) * (1 << CN_SHIFT) + 0.5))
  341. #define C1 C_FIX(0.6532814824)
  342. #define C2 C_FIX(0.2705980501)
  343. /* row idct is multiple by 16 * sqrt(2.0), col idct4 is normalized,
  344. and the butterfly must be multiplied by 0.5 * sqrt(2.0) */
  345. #define C_SHIFT (4+1+12)
  346. static inline void idct4col_put(uint8_t *dest, int line_size, const DCTELEM *col)
  347. {
  348. int c0, c1, c2, c3, a0, a1, a2, a3;
  349. const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  350. a0 = col[8*0];
  351. a1 = col[8*2];
  352. a2 = col[8*4];
  353. a3 = col[8*6];
  354. c0 = ((a0 + a2) << (CN_SHIFT - 1)) + (1 << (C_SHIFT - 1));
  355. c2 = ((a0 - a2) << (CN_SHIFT - 1)) + (1 << (C_SHIFT - 1));
  356. c1 = a1 * C1 + a3 * C2;
  357. c3 = a1 * C2 - a3 * C1;
  358. dest[0] = cm[(c0 + c1) >> C_SHIFT];
  359. dest += line_size;
  360. dest[0] = cm[(c2 + c3) >> C_SHIFT];
  361. dest += line_size;
  362. dest[0] = cm[(c2 - c3) >> C_SHIFT];
  363. dest += line_size;
  364. dest[0] = cm[(c0 - c1) >> C_SHIFT];
  365. }
  366. #define BF(k) \
  367. {\
  368. int a0, a1;\
  369. a0 = ptr[k];\
  370. a1 = ptr[8 + k];\
  371. ptr[k] = a0 + a1;\
  372. ptr[8 + k] = a0 - a1;\
  373. }
  374. /* only used by DV codec. The input must be interlaced. 128 is added
  375. to the pixels before clamping to avoid systematic error
  376. (1024*sqrt(2)) offset would be needed otherwise. */
  377. /* XXX: I think a 1.0/sqrt(2) normalization should be needed to
  378. compensate the extra butterfly stage - I don't have the full DV
  379. specification */
  380. void ff_simple_idct248_put(uint8_t *dest, int line_size, DCTELEM *block)
  381. {
  382. int i;
  383. DCTELEM *ptr;
  384. /* butterfly */
  385. ptr = block;
  386. for(i=0;i<4;i++) {
  387. BF(0);
  388. BF(1);
  389. BF(2);
  390. BF(3);
  391. BF(4);
  392. BF(5);
  393. BF(6);
  394. BF(7);
  395. ptr += 2 * 8;
  396. }
  397. /* IDCT8 on each line */
  398. for(i=0; i<8; i++) {
  399. idctRowCondDC(block + i*8);
  400. }
  401. /* IDCT4 and store */
  402. for(i=0;i<8;i++) {
  403. idct4col_put(dest + i, 2 * line_size, block + i);
  404. idct4col_put(dest + line_size + i, 2 * line_size, block + 8 + i);
  405. }
  406. }
  407. /* 8x4 & 4x8 WMV2 IDCT */
  408. #undef CN_SHIFT
  409. #undef C_SHIFT
  410. #undef C_FIX
  411. #undef C1
  412. #undef C2
  413. #define CN_SHIFT 12
  414. #define C_FIX(x) ((int)((x) * 1.414213562 * (1 << CN_SHIFT) + 0.5))
  415. #define C1 C_FIX(0.6532814824)
  416. #define C2 C_FIX(0.2705980501)
  417. #define C3 C_FIX(0.5)
  418. #define C_SHIFT (4+1+12)
  419. static inline void idct4col_add(uint8_t *dest, int line_size, const DCTELEM *col)
  420. {
  421. int c0, c1, c2, c3, a0, a1, a2, a3;
  422. const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  423. a0 = col[8*0];
  424. a1 = col[8*1];
  425. a2 = col[8*2];
  426. a3 = col[8*3];
  427. c0 = (a0 + a2)*C3 + (1 << (C_SHIFT - 1));
  428. c2 = (a0 - a2)*C3 + (1 << (C_SHIFT - 1));
  429. c1 = a1 * C1 + a3 * C2;
  430. c3 = a1 * C2 - a3 * C1;
  431. dest[0] = cm[dest[0] + ((c0 + c1) >> C_SHIFT)];
  432. dest += line_size;
  433. dest[0] = cm[dest[0] + ((c2 + c3) >> C_SHIFT)];
  434. dest += line_size;
  435. dest[0] = cm[dest[0] + ((c2 - c3) >> C_SHIFT)];
  436. dest += line_size;
  437. dest[0] = cm[dest[0] + ((c0 - c1) >> C_SHIFT)];
  438. }
  439. #define RN_SHIFT 15
  440. #define R_FIX(x) ((int)((x) * 1.414213562 * (1 << RN_SHIFT) + 0.5))
  441. #define R1 R_FIX(0.6532814824)
  442. #define R2 R_FIX(0.2705980501)
  443. #define R3 R_FIX(0.5)
  444. #define R_SHIFT 11
  445. static inline void idct4row(DCTELEM *row)
  446. {
  447. int c0, c1, c2, c3, a0, a1, a2, a3;
  448. //const uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  449. a0 = row[0];
  450. a1 = row[1];
  451. a2 = row[2];
  452. a3 = row[3];
  453. c0 = (a0 + a2)*R3 + (1 << (R_SHIFT - 1));
  454. c2 = (a0 - a2)*R3 + (1 << (R_SHIFT - 1));
  455. c1 = a1 * R1 + a3 * R2;
  456. c3 = a1 * R2 - a3 * R1;
  457. row[0]= (c0 + c1) >> R_SHIFT;
  458. row[1]= (c2 + c3) >> R_SHIFT;
  459. row[2]= (c2 - c3) >> R_SHIFT;
  460. row[3]= (c0 - c1) >> R_SHIFT;
  461. }
  462. void ff_simple_idct84_add(uint8_t *dest, int line_size, DCTELEM *block)
  463. {
  464. int i;
  465. /* IDCT8 on each line */
  466. for(i=0; i<4; i++) {
  467. idctRowCondDC(block + i*8);
  468. }
  469. /* IDCT4 and store */
  470. for(i=0;i<8;i++) {
  471. idct4col_add(dest + i, line_size, block + i);
  472. }
  473. }
  474. void ff_simple_idct48_add(uint8_t *dest, int line_size, DCTELEM *block)
  475. {
  476. int i;
  477. /* IDCT4 on each line */
  478. for(i=0; i<8; i++) {
  479. idct4row(block + i*8);
  480. }
  481. /* IDCT8 and store */
  482. for(i=0; i<4; i++){
  483. idctSparseColAdd(dest + i, line_size, block + i);
  484. }
  485. }
  486. void ff_simple_idct44_add(uint8_t *dest, int line_size, DCTELEM *block)
  487. {
  488. int i;
  489. /* IDCT4 on each line */
  490. for(i=0; i<4; i++) {
  491. idct4row(block + i*8);
  492. }
  493. /* IDCT4 and store */
  494. for(i=0; i<4; i++){
  495. idct4col_add(dest + i, line_size, block + i);
  496. }
  497. }