mmxloop.h 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216
  1. #if !defined(_x86_mmxloop_H)
  2. # define _x86_mmxloop_H (1)
  3. # include <stddef.h>
  4. # include "x86int.h"
  5. #if defined(OC_X86_ASM)
  6. /*On entry, mm0={a0,...,a7}, mm1={b0,...,b7}, mm2={c0,...,c7}, mm3={d0,...d7}.
  7. On exit, mm1={b0+lflim(R_0,L),...,b7+lflim(R_7,L)} and
  8. mm2={c0-lflim(R_0,L),...,c7-lflim(R_7,L)}; mm0 and mm3 are clobbered.*/
  9. #define OC_LOOP_FILTER8_MMX \
  10. "#OC_LOOP_FILTER8_MMX\n\t" \
  11. /*mm7=0*/ \
  12. "pxor %%mm7,%%mm7\n\t" \
  13. /*mm6:mm0={a0,...,a7}*/ \
  14. "movq %%mm0,%%mm6\n\t" \
  15. "punpcklbw %%mm7,%%mm0\n\t" \
  16. "punpckhbw %%mm7,%%mm6\n\t" \
  17. /*mm3:mm5={d0,...,d7}*/ \
  18. "movq %%mm3,%%mm5\n\t" \
  19. "punpcklbw %%mm7,%%mm3\n\t" \
  20. "punpckhbw %%mm7,%%mm5\n\t" \
  21. /*mm6:mm0={a0-d0,...,a7-d7}*/ \
  22. "psubw %%mm3,%%mm0\n\t" \
  23. "psubw %%mm5,%%mm6\n\t" \
  24. /*mm3:mm1={b0,...,b7}*/ \
  25. "movq %%mm1,%%mm3\n\t" \
  26. "punpcklbw %%mm7,%%mm1\n\t" \
  27. "movq %%mm2,%%mm4\n\t" \
  28. "punpckhbw %%mm7,%%mm3\n\t" \
  29. /*mm5:mm4={c0,...,c7}*/ \
  30. "movq %%mm2,%%mm5\n\t" \
  31. "punpcklbw %%mm7,%%mm4\n\t" \
  32. "punpckhbw %%mm7,%%mm5\n\t" \
  33. /*mm7={3}x4 \
  34. mm5:mm4={c0-b0,...,c7-b7}*/ \
  35. "pcmpeqw %%mm7,%%mm7\n\t" \
  36. "psubw %%mm1,%%mm4\n\t" \
  37. "psrlw $14,%%mm7\n\t" \
  38. "psubw %%mm3,%%mm5\n\t" \
  39. /*Scale by 3.*/ \
  40. "pmullw %%mm7,%%mm4\n\t" \
  41. "pmullw %%mm7,%%mm5\n\t" \
  42. /*mm7={4}x4 \
  43. mm5:mm4=f={a0-d0+3*(c0-b0),...,a7-d7+3*(c7-b7)}*/ \
  44. "psrlw $1,%%mm7\n\t" \
  45. "paddw %%mm0,%%mm4\n\t" \
  46. "psllw $2,%%mm7\n\t" \
  47. "movq (%[ll]),%%mm0\n\t" \
  48. "paddw %%mm6,%%mm5\n\t" \
  49. /*R_i has the range [-127,128], so we compute -R_i instead. \
  50. mm4=-R_i=-(f+4>>3)=0xFF^(f-4>>3)*/ \
  51. "psubw %%mm7,%%mm4\n\t" \
  52. "psubw %%mm7,%%mm5\n\t" \
  53. "psraw $3,%%mm4\n\t" \
  54. "psraw $3,%%mm5\n\t" \
  55. "pcmpeqb %%mm7,%%mm7\n\t" \
  56. "packsswb %%mm5,%%mm4\n\t" \
  57. "pxor %%mm6,%%mm6\n\t" \
  58. "pxor %%mm7,%%mm4\n\t" \
  59. "packuswb %%mm3,%%mm1\n\t" \
  60. /*Now compute lflim of -mm4 cf. Section 7.10 of the sepc.*/ \
  61. /*There's no unsigned byte+signed byte with unsigned saturation op code, so \
  62. we have to split things by sign (the other option is to work in 16 bits, \
  63. but working in 8 bits gives much better parallelism). \
  64. We compute abs(R_i), but save a mask of which terms were negative in mm6. \
  65. Then we compute mm4=abs(lflim(R_i,L))=min(abs(R_i),max(2*L-abs(R_i),0)). \
  66. Finally, we split mm4 into positive and negative pieces using the mask in \
  67. mm6, and add and subtract them as appropriate.*/ \
  68. /*mm4=abs(-R_i)*/ \
  69. /*mm7=255-2*L*/ \
  70. "pcmpgtb %%mm4,%%mm6\n\t" \
  71. "psubb %%mm0,%%mm7\n\t" \
  72. "pxor %%mm6,%%mm4\n\t" \
  73. "psubb %%mm0,%%mm7\n\t" \
  74. "psubb %%mm6,%%mm4\n\t" \
  75. /*mm7=255-max(2*L-abs(R_i),0)*/ \
  76. "paddusb %%mm4,%%mm7\n\t" \
  77. /*mm4=min(abs(R_i),max(2*L-abs(R_i),0))*/ \
  78. "paddusb %%mm7,%%mm4\n\t" \
  79. "psubusb %%mm7,%%mm4\n\t" \
  80. /*Now split mm4 by the original sign of -R_i.*/ \
  81. "movq %%mm4,%%mm5\n\t" \
  82. "pand %%mm6,%%mm4\n\t" \
  83. "pandn %%mm5,%%mm6\n\t" \
  84. /*mm1={b0+lflim(R_0,L),...,b7+lflim(R_7,L)}*/ \
  85. /*mm2={c0-lflim(R_0,L),...,c7-lflim(R_7,L)}*/ \
  86. "paddusb %%mm4,%%mm1\n\t" \
  87. "psubusb %%mm4,%%mm2\n\t" \
  88. "psubusb %%mm6,%%mm1\n\t" \
  89. "paddusb %%mm6,%%mm2\n\t" \
  90. #define OC_LOOP_FILTER_V_MMX(_pix,_ystride,_ll) \
  91. do{ \
  92. ptrdiff_t ystride3__; \
  93. __asm__ __volatile__( \
  94. /*mm0={a0,...,a7}*/ \
  95. "movq (%[pix]),%%mm0\n\t" \
  96. /*ystride3=_ystride*3*/ \
  97. "lea (%[ystride],%[ystride],2),%[ystride3]\n\t" \
  98. /*mm3={d0,...,d7}*/ \
  99. "movq (%[pix],%[ystride3]),%%mm3\n\t" \
  100. /*mm1={b0,...,b7}*/ \
  101. "movq (%[pix],%[ystride]),%%mm1\n\t" \
  102. /*mm2={c0,...,c7}*/ \
  103. "movq (%[pix],%[ystride],2),%%mm2\n\t" \
  104. OC_LOOP_FILTER8_MMX \
  105. /*Write it back out.*/ \
  106. "movq %%mm1,(%[pix],%[ystride])\n\t" \
  107. "movq %%mm2,(%[pix],%[ystride],2)\n\t" \
  108. :[ystride3]"=&r"(ystride3__) \
  109. :[pix]"r"(_pix-_ystride*2),[ystride]"r"((ptrdiff_t)(_ystride)), \
  110. [ll]"r"(_ll) \
  111. :"memory" \
  112. ); \
  113. } \
  114. while(0)
  115. #define OC_LOOP_FILTER_H_MMX(_pix,_ystride,_ll) \
  116. do{ \
  117. unsigned char *pix__; \
  118. ptrdiff_t ystride3__; \
  119. ptrdiff_t d__; \
  120. pix__=(_pix)-2; \
  121. __asm__ __volatile__( \
  122. /*x x x x d0 c0 b0 a0*/ \
  123. "movd (%[pix]),%%mm0\n\t" \
  124. /*x x x x d1 c1 b1 a1*/ \
  125. "movd (%[pix],%[ystride]),%%mm1\n\t" \
  126. /*ystride3=_ystride*3*/ \
  127. "lea (%[ystride],%[ystride],2),%[ystride3]\n\t" \
  128. /*x x x x d2 c2 b2 a2*/ \
  129. "movd (%[pix],%[ystride],2),%%mm2\n\t" \
  130. /*x x x x d3 c3 b3 a3*/ \
  131. "lea (%[pix],%[ystride],4),%[d]\n\t" \
  132. "movd (%[pix],%[ystride3]),%%mm3\n\t" \
  133. /*x x x x d4 c4 b4 a4*/ \
  134. "movd (%[d]),%%mm4\n\t" \
  135. /*x x x x d5 c5 b5 a5*/ \
  136. "movd (%[d],%[ystride]),%%mm5\n\t" \
  137. /*x x x x d6 c6 b6 a6*/ \
  138. "movd (%[d],%[ystride],2),%%mm6\n\t" \
  139. /*x x x x d7 c7 b7 a7*/ \
  140. "movd (%[d],%[ystride3]),%%mm7\n\t" \
  141. /*mm0=d1 d0 c1 c0 b1 b0 a1 a0*/ \
  142. "punpcklbw %%mm1,%%mm0\n\t" \
  143. /*mm2=d3 d2 c3 c2 b3 b2 a3 a2*/ \
  144. "punpcklbw %%mm3,%%mm2\n\t" \
  145. /*mm3=d1 d0 c1 c0 b1 b0 a1 a0*/ \
  146. "movq %%mm0,%%mm3\n\t" \
  147. /*mm0=b3 b2 b1 b0 a3 a2 a1 a0*/ \
  148. "punpcklwd %%mm2,%%mm0\n\t" \
  149. /*mm3=d3 d2 d1 d0 c3 c2 c1 c0*/ \
  150. "punpckhwd %%mm2,%%mm3\n\t" \
  151. /*mm1=b3 b2 b1 b0 a3 a2 a1 a0*/ \
  152. "movq %%mm0,%%mm1\n\t" \
  153. /*mm4=d5 d4 c5 c4 b5 b4 a5 a4*/ \
  154. "punpcklbw %%mm5,%%mm4\n\t" \
  155. /*mm6=d7 d6 c7 c6 b7 b6 a7 a6*/ \
  156. "punpcklbw %%mm7,%%mm6\n\t" \
  157. /*mm5=d5 d4 c5 c4 b5 b4 a5 a4*/ \
  158. "movq %%mm4,%%mm5\n\t" \
  159. /*mm4=b7 b6 b5 b4 a7 a6 a5 a4*/ \
  160. "punpcklwd %%mm6,%%mm4\n\t" \
  161. /*mm5=d7 d6 d5 d4 c7 c6 c5 c4*/ \
  162. "punpckhwd %%mm6,%%mm5\n\t" \
  163. /*mm2=d3 d2 d1 d0 c3 c2 c1 c0*/ \
  164. "movq %%mm3,%%mm2\n\t" \
  165. /*mm0=a7 a6 a5 a4 a3 a2 a1 a0*/ \
  166. "punpckldq %%mm4,%%mm0\n\t" \
  167. /*mm1=b7 b6 b5 b4 b3 b2 b1 b0*/ \
  168. "punpckhdq %%mm4,%%mm1\n\t" \
  169. /*mm2=c7 c6 c5 c4 c3 c2 c1 c0*/ \
  170. "punpckldq %%mm5,%%mm2\n\t" \
  171. /*mm3=d7 d6 d5 d4 d3 d2 d1 d0*/ \
  172. "punpckhdq %%mm5,%%mm3\n\t" \
  173. OC_LOOP_FILTER8_MMX \
  174. /*mm2={b0+R_0'',...,b7+R_7''}*/ \
  175. "movq %%mm1,%%mm0\n\t" \
  176. /*mm1={b0+R_0'',c0-R_0'',...,b3+R_3'',c3-R_3''}*/ \
  177. "punpcklbw %%mm2,%%mm1\n\t" \
  178. /*mm2={b4+R_4'',c4-R_4'',...,b7+R_7'',c7-R_7''}*/ \
  179. "punpckhbw %%mm2,%%mm0\n\t" \
  180. /*[d]=c1 b1 c0 b0*/ \
  181. "movd %%mm1,%[d]\n\t" \
  182. "movw %w[d],1(%[pix])\n\t" \
  183. "psrlq $32,%%mm1\n\t" \
  184. "shr $16,%[d]\n\t" \
  185. "movw %w[d],1(%[pix],%[ystride])\n\t" \
  186. /*[d]=c3 b3 c2 b2*/ \
  187. "movd %%mm1,%[d]\n\t" \
  188. "movw %w[d],1(%[pix],%[ystride],2)\n\t" \
  189. "shr $16,%[d]\n\t" \
  190. "movw %w[d],1(%[pix],%[ystride3])\n\t" \
  191. "lea (%[pix],%[ystride],4),%[pix]\n\t" \
  192. /*[d]=c5 b5 c4 b4*/ \
  193. "movd %%mm0,%[d]\n\t" \
  194. "movw %w[d],1(%[pix])\n\t" \
  195. "psrlq $32,%%mm0\n\t" \
  196. "shr $16,%[d]\n\t" \
  197. "movw %w[d],1(%[pix],%[ystride])\n\t" \
  198. /*[d]=c7 b7 c6 b6*/ \
  199. "movd %%mm0,%[d]\n\t" \
  200. "movw %w[d],1(%[pix],%[ystride],2)\n\t" \
  201. "shr $16,%[d]\n\t" \
  202. "movw %w[d],1(%[pix],%[ystride3])\n\t" \
  203. :[pix]"+r"(pix__),[ystride3]"=&r"(ystride3__),[d]"=&r"(d__) \
  204. :[ystride]"r"((ptrdiff_t)(_ystride)),[ll]"r"(_ll) \
  205. :"memory" \
  206. ); \
  207. } \
  208. while(0)
  209. # endif
  210. #endif