h264pred.c 39 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175
  1. /*
  2. * H.26L/H.264/AVC/JVT/14496-10/... encoder/decoder
  3. * Copyright (c) 2003 Michael Niedermayer <michaelni@gmx.at>
  4. *
  5. * This file is part of FFmpeg.
  6. *
  7. * FFmpeg is free software; you can redistribute it and/or
  8. * modify it under the terms of the GNU Lesser General Public
  9. * License as published by the Free Software Foundation; either
  10. * version 2.1 of the License, or (at your option) any later version.
  11. *
  12. * FFmpeg is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  15. * Lesser General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU Lesser General Public
  18. * License along with FFmpeg; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  20. */
  21. /**
  22. * @file libavcodec/h264pred.c
  23. * H.264 / AVC / MPEG4 part10 prediction functions.
  24. * @author Michael Niedermayer <michaelni@gmx.at>
  25. */
  26. #include "avcodec.h"
  27. #include "mpegvideo.h"
  28. #include "h264pred.h"
  29. static void pred4x4_vertical_c(uint8_t *src, uint8_t *topright, int stride){
  30. const uint32_t a= ((uint32_t*)(src-stride))[0];
  31. ((uint32_t*)(src+0*stride))[0]= a;
  32. ((uint32_t*)(src+1*stride))[0]= a;
  33. ((uint32_t*)(src+2*stride))[0]= a;
  34. ((uint32_t*)(src+3*stride))[0]= a;
  35. }
  36. static void pred4x4_horizontal_c(uint8_t *src, uint8_t *topright, int stride){
  37. ((uint32_t*)(src+0*stride))[0]= src[-1+0*stride]*0x01010101;
  38. ((uint32_t*)(src+1*stride))[0]= src[-1+1*stride]*0x01010101;
  39. ((uint32_t*)(src+2*stride))[0]= src[-1+2*stride]*0x01010101;
  40. ((uint32_t*)(src+3*stride))[0]= src[-1+3*stride]*0x01010101;
  41. }
  42. static void pred4x4_dc_c(uint8_t *src, uint8_t *topright, int stride){
  43. const int dc= ( src[-stride] + src[1-stride] + src[2-stride] + src[3-stride]
  44. + src[-1+0*stride] + src[-1+1*stride] + src[-1+2*stride] + src[-1+3*stride] + 4) >>3;
  45. ((uint32_t*)(src+0*stride))[0]=
  46. ((uint32_t*)(src+1*stride))[0]=
  47. ((uint32_t*)(src+2*stride))[0]=
  48. ((uint32_t*)(src+3*stride))[0]= dc* 0x01010101;
  49. }
  50. static void pred4x4_left_dc_c(uint8_t *src, uint8_t *topright, int stride){
  51. const int dc= ( src[-1+0*stride] + src[-1+1*stride] + src[-1+2*stride] + src[-1+3*stride] + 2) >>2;
  52. ((uint32_t*)(src+0*stride))[0]=
  53. ((uint32_t*)(src+1*stride))[0]=
  54. ((uint32_t*)(src+2*stride))[0]=
  55. ((uint32_t*)(src+3*stride))[0]= dc* 0x01010101;
  56. }
  57. static void pred4x4_top_dc_c(uint8_t *src, uint8_t *topright, int stride){
  58. const int dc= ( src[-stride] + src[1-stride] + src[2-stride] + src[3-stride] + 2) >>2;
  59. ((uint32_t*)(src+0*stride))[0]=
  60. ((uint32_t*)(src+1*stride))[0]=
  61. ((uint32_t*)(src+2*stride))[0]=
  62. ((uint32_t*)(src+3*stride))[0]= dc* 0x01010101;
  63. }
  64. static void pred4x4_128_dc_c(uint8_t *src, uint8_t *topright, int stride){
  65. ((uint32_t*)(src+0*stride))[0]=
  66. ((uint32_t*)(src+1*stride))[0]=
  67. ((uint32_t*)(src+2*stride))[0]=
  68. ((uint32_t*)(src+3*stride))[0]= 128U*0x01010101U;
  69. }
  70. #define LOAD_TOP_RIGHT_EDGE\
  71. const int av_unused t4= topright[0];\
  72. const int av_unused t5= topright[1];\
  73. const int av_unused t6= topright[2];\
  74. const int av_unused t7= topright[3];\
  75. #define LOAD_DOWN_LEFT_EDGE\
  76. const int av_unused l4= src[-1+4*stride];\
  77. const int av_unused l5= src[-1+5*stride];\
  78. const int av_unused l6= src[-1+6*stride];\
  79. const int av_unused l7= src[-1+7*stride];\
  80. #define LOAD_LEFT_EDGE\
  81. const int av_unused l0= src[-1+0*stride];\
  82. const int av_unused l1= src[-1+1*stride];\
  83. const int av_unused l2= src[-1+2*stride];\
  84. const int av_unused l3= src[-1+3*stride];\
  85. #define LOAD_TOP_EDGE\
  86. const int av_unused t0= src[ 0-1*stride];\
  87. const int av_unused t1= src[ 1-1*stride];\
  88. const int av_unused t2= src[ 2-1*stride];\
  89. const int av_unused t3= src[ 3-1*stride];\
  90. static void pred4x4_down_right_c(uint8_t *src, uint8_t *topright, int stride){
  91. const int lt= src[-1-1*stride];
  92. LOAD_TOP_EDGE
  93. LOAD_LEFT_EDGE
  94. src[0+3*stride]=(l3 + 2*l2 + l1 + 2)>>2;
  95. src[0+2*stride]=
  96. src[1+3*stride]=(l2 + 2*l1 + l0 + 2)>>2;
  97. src[0+1*stride]=
  98. src[1+2*stride]=
  99. src[2+3*stride]=(l1 + 2*l0 + lt + 2)>>2;
  100. src[0+0*stride]=
  101. src[1+1*stride]=
  102. src[2+2*stride]=
  103. src[3+3*stride]=(l0 + 2*lt + t0 + 2)>>2;
  104. src[1+0*stride]=
  105. src[2+1*stride]=
  106. src[3+2*stride]=(lt + 2*t0 + t1 + 2)>>2;
  107. src[2+0*stride]=
  108. src[3+1*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  109. src[3+0*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  110. }
  111. static void pred4x4_down_left_c(uint8_t *src, uint8_t *topright, int stride){
  112. LOAD_TOP_EDGE
  113. LOAD_TOP_RIGHT_EDGE
  114. // LOAD_LEFT_EDGE
  115. src[0+0*stride]=(t0 + t2 + 2*t1 + 2)>>2;
  116. src[1+0*stride]=
  117. src[0+1*stride]=(t1 + t3 + 2*t2 + 2)>>2;
  118. src[2+0*stride]=
  119. src[1+1*stride]=
  120. src[0+2*stride]=(t2 + t4 + 2*t3 + 2)>>2;
  121. src[3+0*stride]=
  122. src[2+1*stride]=
  123. src[1+2*stride]=
  124. src[0+3*stride]=(t3 + t5 + 2*t4 + 2)>>2;
  125. src[3+1*stride]=
  126. src[2+2*stride]=
  127. src[1+3*stride]=(t4 + t6 + 2*t5 + 2)>>2;
  128. src[3+2*stride]=
  129. src[2+3*stride]=(t5 + t7 + 2*t6 + 2)>>2;
  130. src[3+3*stride]=(t6 + 3*t7 + 2)>>2;
  131. }
  132. static void pred4x4_down_left_svq3_c(uint8_t *src, uint8_t *topright, int stride){
  133. LOAD_TOP_EDGE
  134. LOAD_LEFT_EDGE
  135. const av_unused int unu0= t0;
  136. const av_unused int unu1= l0;
  137. src[0+0*stride]=(l1 + t1)>>1;
  138. src[1+0*stride]=
  139. src[0+1*stride]=(l2 + t2)>>1;
  140. src[2+0*stride]=
  141. src[1+1*stride]=
  142. src[0+2*stride]=
  143. src[3+0*stride]=
  144. src[2+1*stride]=
  145. src[1+2*stride]=
  146. src[0+3*stride]=
  147. src[3+1*stride]=
  148. src[2+2*stride]=
  149. src[1+3*stride]=
  150. src[3+2*stride]=
  151. src[2+3*stride]=
  152. src[3+3*stride]=(l3 + t3)>>1;
  153. }
  154. static void pred4x4_down_left_rv40_c(uint8_t *src, uint8_t *topright, int stride){
  155. LOAD_TOP_EDGE
  156. LOAD_TOP_RIGHT_EDGE
  157. LOAD_LEFT_EDGE
  158. LOAD_DOWN_LEFT_EDGE
  159. src[0+0*stride]=(t0 + t2 + 2*t1 + 2 + l0 + l2 + 2*l1 + 2)>>3;
  160. src[1+0*stride]=
  161. src[0+1*stride]=(t1 + t3 + 2*t2 + 2 + l1 + l3 + 2*l2 + 2)>>3;
  162. src[2+0*stride]=
  163. src[1+1*stride]=
  164. src[0+2*stride]=(t2 + t4 + 2*t3 + 2 + l2 + l4 + 2*l3 + 2)>>3;
  165. src[3+0*stride]=
  166. src[2+1*stride]=
  167. src[1+2*stride]=
  168. src[0+3*stride]=(t3 + t5 + 2*t4 + 2 + l3 + l5 + 2*l4 + 2)>>3;
  169. src[3+1*stride]=
  170. src[2+2*stride]=
  171. src[1+3*stride]=(t4 + t6 + 2*t5 + 2 + l4 + l6 + 2*l5 + 2)>>3;
  172. src[3+2*stride]=
  173. src[2+3*stride]=(t5 + t7 + 2*t6 + 2 + l5 + l7 + 2*l6 + 2)>>3;
  174. src[3+3*stride]=(t6 + t7 + 1 + l6 + l7 + 1)>>2;
  175. }
  176. static void pred4x4_down_left_rv40_nodown_c(uint8_t *src, uint8_t *topright, int stride){
  177. LOAD_TOP_EDGE
  178. LOAD_TOP_RIGHT_EDGE
  179. LOAD_LEFT_EDGE
  180. src[0+0*stride]=(t0 + t2 + 2*t1 + 2 + l0 + l2 + 2*l1 + 2)>>3;
  181. src[1+0*stride]=
  182. src[0+1*stride]=(t1 + t3 + 2*t2 + 2 + l1 + l3 + 2*l2 + 2)>>3;
  183. src[2+0*stride]=
  184. src[1+1*stride]=
  185. src[0+2*stride]=(t2 + t4 + 2*t3 + 2 + l2 + 3*l3 + 2)>>3;
  186. src[3+0*stride]=
  187. src[2+1*stride]=
  188. src[1+2*stride]=
  189. src[0+3*stride]=(t3 + t5 + 2*t4 + 2 + l3*4 + 2)>>3;
  190. src[3+1*stride]=
  191. src[2+2*stride]=
  192. src[1+3*stride]=(t4 + t6 + 2*t5 + 2 + l3*4 + 2)>>3;
  193. src[3+2*stride]=
  194. src[2+3*stride]=(t5 + t7 + 2*t6 + 2 + l3*4 + 2)>>3;
  195. src[3+3*stride]=(t6 + t7 + 1 + 2*l3 + 1)>>2;
  196. }
  197. static void pred4x4_vertical_right_c(uint8_t *src, uint8_t *topright, int stride){
  198. const int lt= src[-1-1*stride];
  199. LOAD_TOP_EDGE
  200. LOAD_LEFT_EDGE
  201. src[0+0*stride]=
  202. src[1+2*stride]=(lt + t0 + 1)>>1;
  203. src[1+0*stride]=
  204. src[2+2*stride]=(t0 + t1 + 1)>>1;
  205. src[2+0*stride]=
  206. src[3+2*stride]=(t1 + t2 + 1)>>1;
  207. src[3+0*stride]=(t2 + t3 + 1)>>1;
  208. src[0+1*stride]=
  209. src[1+3*stride]=(l0 + 2*lt + t0 + 2)>>2;
  210. src[1+1*stride]=
  211. src[2+3*stride]=(lt + 2*t0 + t1 + 2)>>2;
  212. src[2+1*stride]=
  213. src[3+3*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  214. src[3+1*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  215. src[0+2*stride]=(lt + 2*l0 + l1 + 2)>>2;
  216. src[0+3*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  217. }
  218. static void pred4x4_vertical_left_c(uint8_t *src, uint8_t *topright, int stride){
  219. LOAD_TOP_EDGE
  220. LOAD_TOP_RIGHT_EDGE
  221. src[0+0*stride]=(t0 + t1 + 1)>>1;
  222. src[1+0*stride]=
  223. src[0+2*stride]=(t1 + t2 + 1)>>1;
  224. src[2+0*stride]=
  225. src[1+2*stride]=(t2 + t3 + 1)>>1;
  226. src[3+0*stride]=
  227. src[2+2*stride]=(t3 + t4+ 1)>>1;
  228. src[3+2*stride]=(t4 + t5+ 1)>>1;
  229. src[0+1*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  230. src[1+1*stride]=
  231. src[0+3*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  232. src[2+1*stride]=
  233. src[1+3*stride]=(t2 + 2*t3 + t4 + 2)>>2;
  234. src[3+1*stride]=
  235. src[2+3*stride]=(t3 + 2*t4 + t5 + 2)>>2;
  236. src[3+3*stride]=(t4 + 2*t5 + t6 + 2)>>2;
  237. }
  238. static void pred4x4_vertical_left_rv40(uint8_t *src, uint8_t *topright, int stride,
  239. const int l0, const int l1, const int l2, const int l3, const int l4){
  240. LOAD_TOP_EDGE
  241. LOAD_TOP_RIGHT_EDGE
  242. src[0+0*stride]=(2*t0 + 2*t1 + l1 + 2*l2 + l3 + 4)>>3;
  243. src[1+0*stride]=
  244. src[0+2*stride]=(t1 + t2 + 1)>>1;
  245. src[2+0*stride]=
  246. src[1+2*stride]=(t2 + t3 + 1)>>1;
  247. src[3+0*stride]=
  248. src[2+2*stride]=(t3 + t4+ 1)>>1;
  249. src[3+2*stride]=(t4 + t5+ 1)>>1;
  250. src[0+1*stride]=(t0 + 2*t1 + t2 + l2 + 2*l3 + l4 + 4)>>3;
  251. src[1+1*stride]=
  252. src[0+3*stride]=(t1 + 2*t2 + t3 + 2)>>2;
  253. src[2+1*stride]=
  254. src[1+3*stride]=(t2 + 2*t3 + t4 + 2)>>2;
  255. src[3+1*stride]=
  256. src[2+3*stride]=(t3 + 2*t4 + t5 + 2)>>2;
  257. src[3+3*stride]=(t4 + 2*t5 + t6 + 2)>>2;
  258. }
  259. static void pred4x4_vertical_left_rv40_c(uint8_t *src, uint8_t *topright, int stride){
  260. LOAD_LEFT_EDGE
  261. LOAD_DOWN_LEFT_EDGE
  262. pred4x4_vertical_left_rv40(src, topright, stride, l0, l1, l2, l3, l4);
  263. }
  264. static void pred4x4_vertical_left_rv40_nodown_c(uint8_t *src, uint8_t *topright, int stride){
  265. LOAD_LEFT_EDGE
  266. pred4x4_vertical_left_rv40(src, topright, stride, l0, l1, l2, l3, l3);
  267. }
  268. static void pred4x4_horizontal_up_c(uint8_t *src, uint8_t *topright, int stride){
  269. LOAD_LEFT_EDGE
  270. src[0+0*stride]=(l0 + l1 + 1)>>1;
  271. src[1+0*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  272. src[2+0*stride]=
  273. src[0+1*stride]=(l1 + l2 + 1)>>1;
  274. src[3+0*stride]=
  275. src[1+1*stride]=(l1 + 2*l2 + l3 + 2)>>2;
  276. src[2+1*stride]=
  277. src[0+2*stride]=(l2 + l3 + 1)>>1;
  278. src[3+1*stride]=
  279. src[1+2*stride]=(l2 + 2*l3 + l3 + 2)>>2;
  280. src[3+2*stride]=
  281. src[1+3*stride]=
  282. src[0+3*stride]=
  283. src[2+2*stride]=
  284. src[2+3*stride]=
  285. src[3+3*stride]=l3;
  286. }
  287. static void pred4x4_horizontal_up_rv40_c(uint8_t *src, uint8_t *topright, int stride){
  288. LOAD_LEFT_EDGE
  289. LOAD_DOWN_LEFT_EDGE
  290. LOAD_TOP_EDGE
  291. LOAD_TOP_RIGHT_EDGE
  292. src[0+0*stride]=(t1 + 2*t2 + t3 + 2*l0 + 2*l1 + 4)>>3;
  293. src[1+0*stride]=(t2 + 2*t3 + t4 + l0 + 2*l1 + l2 + 4)>>3;
  294. src[2+0*stride]=
  295. src[0+1*stride]=(t3 + 2*t4 + t5 + 2*l1 + 2*l2 + 4)>>3;
  296. src[3+0*stride]=
  297. src[1+1*stride]=(t4 + 2*t5 + t6 + l1 + 2*l2 + l3 + 4)>>3;
  298. src[2+1*stride]=
  299. src[0+2*stride]=(t5 + 2*t6 + t7 + 2*l2 + 2*l3 + 4)>>3;
  300. src[3+1*stride]=
  301. src[1+2*stride]=(t6 + 3*t7 + l2 + 3*l3 + 4)>>3;
  302. src[3+2*stride]=
  303. src[1+3*stride]=(l3 + 2*l4 + l5 + 2)>>2;
  304. src[0+3*stride]=
  305. src[2+2*stride]=(t6 + t7 + l3 + l4 + 2)>>2;
  306. src[2+3*stride]=(l4 + l5 + 1)>>1;
  307. src[3+3*stride]=(l4 + 2*l5 + l6 + 2)>>2;
  308. }
  309. static void pred4x4_horizontal_up_rv40_nodown_c(uint8_t *src, uint8_t *topright, int stride){
  310. LOAD_LEFT_EDGE
  311. LOAD_TOP_EDGE
  312. LOAD_TOP_RIGHT_EDGE
  313. src[0+0*stride]=(t1 + 2*t2 + t3 + 2*l0 + 2*l1 + 4)>>3;
  314. src[1+0*stride]=(t2 + 2*t3 + t4 + l0 + 2*l1 + l2 + 4)>>3;
  315. src[2+0*stride]=
  316. src[0+1*stride]=(t3 + 2*t4 + t5 + 2*l1 + 2*l2 + 4)>>3;
  317. src[3+0*stride]=
  318. src[1+1*stride]=(t4 + 2*t5 + t6 + l1 + 2*l2 + l3 + 4)>>3;
  319. src[2+1*stride]=
  320. src[0+2*stride]=(t5 + 2*t6 + t7 + 2*l2 + 2*l3 + 4)>>3;
  321. src[3+1*stride]=
  322. src[1+2*stride]=(t6 + 3*t7 + l2 + 3*l3 + 4)>>3;
  323. src[3+2*stride]=
  324. src[1+3*stride]=l3;
  325. src[0+3*stride]=
  326. src[2+2*stride]=(t6 + t7 + 2*l3 + 2)>>2;
  327. src[2+3*stride]=
  328. src[3+3*stride]=l3;
  329. }
  330. static void pred4x4_horizontal_down_c(uint8_t *src, uint8_t *topright, int stride){
  331. const int lt= src[-1-1*stride];
  332. LOAD_TOP_EDGE
  333. LOAD_LEFT_EDGE
  334. src[0+0*stride]=
  335. src[2+1*stride]=(lt + l0 + 1)>>1;
  336. src[1+0*stride]=
  337. src[3+1*stride]=(l0 + 2*lt + t0 + 2)>>2;
  338. src[2+0*stride]=(lt + 2*t0 + t1 + 2)>>2;
  339. src[3+0*stride]=(t0 + 2*t1 + t2 + 2)>>2;
  340. src[0+1*stride]=
  341. src[2+2*stride]=(l0 + l1 + 1)>>1;
  342. src[1+1*stride]=
  343. src[3+2*stride]=(lt + 2*l0 + l1 + 2)>>2;
  344. src[0+2*stride]=
  345. src[2+3*stride]=(l1 + l2+ 1)>>1;
  346. src[1+2*stride]=
  347. src[3+3*stride]=(l0 + 2*l1 + l2 + 2)>>2;
  348. src[0+3*stride]=(l2 + l3 + 1)>>1;
  349. src[1+3*stride]=(l1 + 2*l2 + l3 + 2)>>2;
  350. }
  351. static void pred16x16_vertical_c(uint8_t *src, int stride){
  352. int i;
  353. const uint32_t a= ((uint32_t*)(src-stride))[0];
  354. const uint32_t b= ((uint32_t*)(src-stride))[1];
  355. const uint32_t c= ((uint32_t*)(src-stride))[2];
  356. const uint32_t d= ((uint32_t*)(src-stride))[3];
  357. for(i=0; i<16; i++){
  358. ((uint32_t*)(src+i*stride))[0]= a;
  359. ((uint32_t*)(src+i*stride))[1]= b;
  360. ((uint32_t*)(src+i*stride))[2]= c;
  361. ((uint32_t*)(src+i*stride))[3]= d;
  362. }
  363. }
  364. static void pred16x16_horizontal_c(uint8_t *src, int stride){
  365. int i;
  366. for(i=0; i<16; i++){
  367. ((uint32_t*)(src+i*stride))[0]=
  368. ((uint32_t*)(src+i*stride))[1]=
  369. ((uint32_t*)(src+i*stride))[2]=
  370. ((uint32_t*)(src+i*stride))[3]= src[-1+i*stride]*0x01010101;
  371. }
  372. }
  373. static void pred16x16_dc_c(uint8_t *src, int stride){
  374. int i, dc=0;
  375. for(i=0;i<16; i++){
  376. dc+= src[-1+i*stride];
  377. }
  378. for(i=0;i<16; i++){
  379. dc+= src[i-stride];
  380. }
  381. dc= 0x01010101*((dc + 16)>>5);
  382. for(i=0; i<16; i++){
  383. ((uint32_t*)(src+i*stride))[0]=
  384. ((uint32_t*)(src+i*stride))[1]=
  385. ((uint32_t*)(src+i*stride))[2]=
  386. ((uint32_t*)(src+i*stride))[3]= dc;
  387. }
  388. }
  389. static void pred16x16_left_dc_c(uint8_t *src, int stride){
  390. int i, dc=0;
  391. for(i=0;i<16; i++){
  392. dc+= src[-1+i*stride];
  393. }
  394. dc= 0x01010101*((dc + 8)>>4);
  395. for(i=0; i<16; i++){
  396. ((uint32_t*)(src+i*stride))[0]=
  397. ((uint32_t*)(src+i*stride))[1]=
  398. ((uint32_t*)(src+i*stride))[2]=
  399. ((uint32_t*)(src+i*stride))[3]= dc;
  400. }
  401. }
  402. static void pred16x16_top_dc_c(uint8_t *src, int stride){
  403. int i, dc=0;
  404. for(i=0;i<16; i++){
  405. dc+= src[i-stride];
  406. }
  407. dc= 0x01010101*((dc + 8)>>4);
  408. for(i=0; i<16; i++){
  409. ((uint32_t*)(src+i*stride))[0]=
  410. ((uint32_t*)(src+i*stride))[1]=
  411. ((uint32_t*)(src+i*stride))[2]=
  412. ((uint32_t*)(src+i*stride))[3]= dc;
  413. }
  414. }
  415. static void pred16x16_128_dc_c(uint8_t *src, int stride){
  416. int i;
  417. for(i=0; i<16; i++){
  418. ((uint32_t*)(src+i*stride))[0]=
  419. ((uint32_t*)(src+i*stride))[1]=
  420. ((uint32_t*)(src+i*stride))[2]=
  421. ((uint32_t*)(src+i*stride))[3]= 0x01010101U*128U;
  422. }
  423. }
  424. static inline void pred16x16_plane_compat_c(uint8_t *src, int stride, const int svq3, const int rv40){
  425. int i, j, k;
  426. int a;
  427. uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  428. const uint8_t * const src0 = src+7-stride;
  429. const uint8_t *src1 = src+8*stride-1;
  430. const uint8_t *src2 = src1-2*stride; // == src+6*stride-1;
  431. int H = src0[1] - src0[-1];
  432. int V = src1[0] - src2[ 0];
  433. for(k=2; k<=8; ++k) {
  434. src1 += stride; src2 -= stride;
  435. H += k*(src0[k] - src0[-k]);
  436. V += k*(src1[0] - src2[ 0]);
  437. }
  438. if(svq3){
  439. H = ( 5*(H/4) ) / 16;
  440. V = ( 5*(V/4) ) / 16;
  441. /* required for 100% accuracy */
  442. i = H; H = V; V = i;
  443. }else if(rv40){
  444. H = ( H + (H>>2) ) >> 4;
  445. V = ( V + (V>>2) ) >> 4;
  446. }else{
  447. H = ( 5*H+32 ) >> 6;
  448. V = ( 5*V+32 ) >> 6;
  449. }
  450. a = 16*(src1[0] + src2[16] + 1) - 7*(V+H);
  451. for(j=16; j>0; --j) {
  452. int b = a;
  453. a += V;
  454. for(i=-16; i<0; i+=4) {
  455. src[16+i] = cm[ (b ) >> 5 ];
  456. src[17+i] = cm[ (b+ H) >> 5 ];
  457. src[18+i] = cm[ (b+2*H) >> 5 ];
  458. src[19+i] = cm[ (b+3*H) >> 5 ];
  459. b += 4*H;
  460. }
  461. src += stride;
  462. }
  463. }
  464. static void pred16x16_plane_c(uint8_t *src, int stride){
  465. pred16x16_plane_compat_c(src, stride, 0, 0);
  466. }
  467. static void pred16x16_plane_svq3_c(uint8_t *src, int stride){
  468. pred16x16_plane_compat_c(src, stride, 1, 0);
  469. }
  470. static void pred16x16_plane_rv40_c(uint8_t *src, int stride){
  471. pred16x16_plane_compat_c(src, stride, 0, 1);
  472. }
  473. static void pred8x8_vertical_c(uint8_t *src, int stride){
  474. int i;
  475. const uint32_t a= ((uint32_t*)(src-stride))[0];
  476. const uint32_t b= ((uint32_t*)(src-stride))[1];
  477. for(i=0; i<8; i++){
  478. ((uint32_t*)(src+i*stride))[0]= a;
  479. ((uint32_t*)(src+i*stride))[1]= b;
  480. }
  481. }
  482. static void pred8x8_horizontal_c(uint8_t *src, int stride){
  483. int i;
  484. for(i=0; i<8; i++){
  485. ((uint32_t*)(src+i*stride))[0]=
  486. ((uint32_t*)(src+i*stride))[1]= src[-1+i*stride]*0x01010101;
  487. }
  488. }
  489. static void pred8x8_128_dc_c(uint8_t *src, int stride){
  490. int i;
  491. for(i=0; i<8; i++){
  492. ((uint32_t*)(src+i*stride))[0]=
  493. ((uint32_t*)(src+i*stride))[1]= 0x01010101U*128U;
  494. }
  495. }
  496. static void pred8x8_left_dc_c(uint8_t *src, int stride){
  497. int i;
  498. int dc0, dc2;
  499. dc0=dc2=0;
  500. for(i=0;i<4; i++){
  501. dc0+= src[-1+i*stride];
  502. dc2+= src[-1+(i+4)*stride];
  503. }
  504. dc0= 0x01010101*((dc0 + 2)>>2);
  505. dc2= 0x01010101*((dc2 + 2)>>2);
  506. for(i=0; i<4; i++){
  507. ((uint32_t*)(src+i*stride))[0]=
  508. ((uint32_t*)(src+i*stride))[1]= dc0;
  509. }
  510. for(i=4; i<8; i++){
  511. ((uint32_t*)(src+i*stride))[0]=
  512. ((uint32_t*)(src+i*stride))[1]= dc2;
  513. }
  514. }
  515. static void pred8x8_left_dc_rv40_c(uint8_t *src, int stride){
  516. int i;
  517. int dc0;
  518. dc0=0;
  519. for(i=0;i<8; i++)
  520. dc0+= src[-1+i*stride];
  521. dc0= 0x01010101*((dc0 + 4)>>3);
  522. for(i=0; i<8; i++){
  523. ((uint32_t*)(src+i*stride))[0]=
  524. ((uint32_t*)(src+i*stride))[1]= dc0;
  525. }
  526. }
  527. static void pred8x8_top_dc_c(uint8_t *src, int stride){
  528. int i;
  529. int dc0, dc1;
  530. dc0=dc1=0;
  531. for(i=0;i<4; i++){
  532. dc0+= src[i-stride];
  533. dc1+= src[4+i-stride];
  534. }
  535. dc0= 0x01010101*((dc0 + 2)>>2);
  536. dc1= 0x01010101*((dc1 + 2)>>2);
  537. for(i=0; i<4; i++){
  538. ((uint32_t*)(src+i*stride))[0]= dc0;
  539. ((uint32_t*)(src+i*stride))[1]= dc1;
  540. }
  541. for(i=4; i<8; i++){
  542. ((uint32_t*)(src+i*stride))[0]= dc0;
  543. ((uint32_t*)(src+i*stride))[1]= dc1;
  544. }
  545. }
  546. static void pred8x8_top_dc_rv40_c(uint8_t *src, int stride){
  547. int i;
  548. int dc0;
  549. dc0=0;
  550. for(i=0;i<8; i++)
  551. dc0+= src[i-stride];
  552. dc0= 0x01010101*((dc0 + 4)>>3);
  553. for(i=0; i<8; i++){
  554. ((uint32_t*)(src+i*stride))[0]=
  555. ((uint32_t*)(src+i*stride))[1]= dc0;
  556. }
  557. }
  558. static void pred8x8_dc_c(uint8_t *src, int stride){
  559. int i;
  560. int dc0, dc1, dc2, dc3;
  561. dc0=dc1=dc2=0;
  562. for(i=0;i<4; i++){
  563. dc0+= src[-1+i*stride] + src[i-stride];
  564. dc1+= src[4+i-stride];
  565. dc2+= src[-1+(i+4)*stride];
  566. }
  567. dc3= 0x01010101*((dc1 + dc2 + 4)>>3);
  568. dc0= 0x01010101*((dc0 + 4)>>3);
  569. dc1= 0x01010101*((dc1 + 2)>>2);
  570. dc2= 0x01010101*((dc2 + 2)>>2);
  571. for(i=0; i<4; i++){
  572. ((uint32_t*)(src+i*stride))[0]= dc0;
  573. ((uint32_t*)(src+i*stride))[1]= dc1;
  574. }
  575. for(i=4; i<8; i++){
  576. ((uint32_t*)(src+i*stride))[0]= dc2;
  577. ((uint32_t*)(src+i*stride))[1]= dc3;
  578. }
  579. }
  580. //the following 4 function should not be optimized!
  581. static void pred8x8_mad_cow_dc_l0t(uint8_t *src, int stride){
  582. pred8x8_top_dc_c(src, stride);
  583. pred4x4_dc_c(src, NULL, stride);
  584. }
  585. static void pred8x8_mad_cow_dc_0lt(uint8_t *src, int stride){
  586. pred8x8_dc_c(src, stride);
  587. pred4x4_top_dc_c(src, NULL, stride);
  588. }
  589. static void pred8x8_mad_cow_dc_l00(uint8_t *src, int stride){
  590. pred8x8_left_dc_c(src, stride);
  591. pred4x4_128_dc_c(src + 4*stride , NULL, stride);
  592. pred4x4_128_dc_c(src + 4*stride + 4, NULL, stride);
  593. }
  594. static void pred8x8_mad_cow_dc_0l0(uint8_t *src, int stride){
  595. pred8x8_left_dc_c(src, stride);
  596. pred4x4_128_dc_c(src , NULL, stride);
  597. pred4x4_128_dc_c(src + 4, NULL, stride);
  598. }
  599. static void pred8x8_dc_rv40_c(uint8_t *src, int stride){
  600. int i;
  601. int dc0=0;
  602. for(i=0;i<4; i++){
  603. dc0+= src[-1+i*stride] + src[i-stride];
  604. dc0+= src[4+i-stride];
  605. dc0+= src[-1+(i+4)*stride];
  606. }
  607. dc0= 0x01010101*((dc0 + 8)>>4);
  608. for(i=0; i<4; i++){
  609. ((uint32_t*)(src+i*stride))[0]= dc0;
  610. ((uint32_t*)(src+i*stride))[1]= dc0;
  611. }
  612. for(i=4; i<8; i++){
  613. ((uint32_t*)(src+i*stride))[0]= dc0;
  614. ((uint32_t*)(src+i*stride))[1]= dc0;
  615. }
  616. }
  617. static void pred8x8_plane_c(uint8_t *src, int stride){
  618. int j, k;
  619. int a;
  620. uint8_t *cm = ff_cropTbl + MAX_NEG_CROP;
  621. const uint8_t * const src0 = src+3-stride;
  622. const uint8_t *src1 = src+4*stride-1;
  623. const uint8_t *src2 = src1-2*stride; // == src+2*stride-1;
  624. int H = src0[1] - src0[-1];
  625. int V = src1[0] - src2[ 0];
  626. for(k=2; k<=4; ++k) {
  627. src1 += stride; src2 -= stride;
  628. H += k*(src0[k] - src0[-k]);
  629. V += k*(src1[0] - src2[ 0]);
  630. }
  631. H = ( 17*H+16 ) >> 5;
  632. V = ( 17*V+16 ) >> 5;
  633. a = 16*(src1[0] + src2[8]+1) - 3*(V+H);
  634. for(j=8; j>0; --j) {
  635. int b = a;
  636. a += V;
  637. src[0] = cm[ (b ) >> 5 ];
  638. src[1] = cm[ (b+ H) >> 5 ];
  639. src[2] = cm[ (b+2*H) >> 5 ];
  640. src[3] = cm[ (b+3*H) >> 5 ];
  641. src[4] = cm[ (b+4*H) >> 5 ];
  642. src[5] = cm[ (b+5*H) >> 5 ];
  643. src[6] = cm[ (b+6*H) >> 5 ];
  644. src[7] = cm[ (b+7*H) >> 5 ];
  645. src += stride;
  646. }
  647. }
  648. #define SRC(x,y) src[(x)+(y)*stride]
  649. #define PL(y) \
  650. const int l##y = (SRC(-1,y-1) + 2*SRC(-1,y) + SRC(-1,y+1) + 2) >> 2;
  651. #define PREDICT_8x8_LOAD_LEFT \
  652. const int l0 = ((has_topleft ? SRC(-1,-1) : SRC(-1,0)) \
  653. + 2*SRC(-1,0) + SRC(-1,1) + 2) >> 2; \
  654. PL(1) PL(2) PL(3) PL(4) PL(5) PL(6) \
  655. const int l7 av_unused = (SRC(-1,6) + 3*SRC(-1,7) + 2) >> 2
  656. #define PT(x) \
  657. const int t##x = (SRC(x-1,-1) + 2*SRC(x,-1) + SRC(x+1,-1) + 2) >> 2;
  658. #define PREDICT_8x8_LOAD_TOP \
  659. const int t0 = ((has_topleft ? SRC(-1,-1) : SRC(0,-1)) \
  660. + 2*SRC(0,-1) + SRC(1,-1) + 2) >> 2; \
  661. PT(1) PT(2) PT(3) PT(4) PT(5) PT(6) \
  662. const int t7 av_unused = ((has_topright ? SRC(8,-1) : SRC(7,-1)) \
  663. + 2*SRC(7,-1) + SRC(6,-1) + 2) >> 2
  664. #define PTR(x) \
  665. t##x = (SRC(x-1,-1) + 2*SRC(x,-1) + SRC(x+1,-1) + 2) >> 2;
  666. #define PREDICT_8x8_LOAD_TOPRIGHT \
  667. int t8, t9, t10, t11, t12, t13, t14, t15; \
  668. if(has_topright) { \
  669. PTR(8) PTR(9) PTR(10) PTR(11) PTR(12) PTR(13) PTR(14) \
  670. t15 = (SRC(14,-1) + 3*SRC(15,-1) + 2) >> 2; \
  671. } else t8=t9=t10=t11=t12=t13=t14=t15= SRC(7,-1);
  672. #define PREDICT_8x8_LOAD_TOPLEFT \
  673. const int lt = (SRC(-1,0) + 2*SRC(-1,-1) + SRC(0,-1) + 2) >> 2
  674. #define PREDICT_8x8_DC(v) \
  675. int y; \
  676. for( y = 0; y < 8; y++ ) { \
  677. ((uint32_t*)src)[0] = \
  678. ((uint32_t*)src)[1] = v; \
  679. src += stride; \
  680. }
  681. static void pred8x8l_128_dc_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  682. {
  683. PREDICT_8x8_DC(0x80808080);
  684. }
  685. static void pred8x8l_left_dc_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  686. {
  687. PREDICT_8x8_LOAD_LEFT;
  688. const uint32_t dc = ((l0+l1+l2+l3+l4+l5+l6+l7+4) >> 3) * 0x01010101;
  689. PREDICT_8x8_DC(dc);
  690. }
  691. static void pred8x8l_top_dc_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  692. {
  693. PREDICT_8x8_LOAD_TOP;
  694. const uint32_t dc = ((t0+t1+t2+t3+t4+t5+t6+t7+4) >> 3) * 0x01010101;
  695. PREDICT_8x8_DC(dc);
  696. }
  697. static void pred8x8l_dc_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  698. {
  699. PREDICT_8x8_LOAD_LEFT;
  700. PREDICT_8x8_LOAD_TOP;
  701. const uint32_t dc = ((l0+l1+l2+l3+l4+l5+l6+l7
  702. +t0+t1+t2+t3+t4+t5+t6+t7+8) >> 4) * 0x01010101;
  703. PREDICT_8x8_DC(dc);
  704. }
  705. static void pred8x8l_horizontal_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  706. {
  707. PREDICT_8x8_LOAD_LEFT;
  708. #define ROW(y) ((uint32_t*)(src+y*stride))[0] =\
  709. ((uint32_t*)(src+y*stride))[1] = 0x01010101 * l##y
  710. ROW(0); ROW(1); ROW(2); ROW(3); ROW(4); ROW(5); ROW(6); ROW(7);
  711. #undef ROW
  712. }
  713. static void pred8x8l_vertical_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  714. {
  715. int y;
  716. PREDICT_8x8_LOAD_TOP;
  717. src[0] = t0;
  718. src[1] = t1;
  719. src[2] = t2;
  720. src[3] = t3;
  721. src[4] = t4;
  722. src[5] = t5;
  723. src[6] = t6;
  724. src[7] = t7;
  725. for( y = 1; y < 8; y++ )
  726. *(uint64_t*)(src+y*stride) = *(uint64_t*)src;
  727. }
  728. static void pred8x8l_down_left_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  729. {
  730. PREDICT_8x8_LOAD_TOP;
  731. PREDICT_8x8_LOAD_TOPRIGHT;
  732. SRC(0,0)= (t0 + 2*t1 + t2 + 2) >> 2;
  733. SRC(0,1)=SRC(1,0)= (t1 + 2*t2 + t3 + 2) >> 2;
  734. SRC(0,2)=SRC(1,1)=SRC(2,0)= (t2 + 2*t3 + t4 + 2) >> 2;
  735. SRC(0,3)=SRC(1,2)=SRC(2,1)=SRC(3,0)= (t3 + 2*t4 + t5 + 2) >> 2;
  736. SRC(0,4)=SRC(1,3)=SRC(2,2)=SRC(3,1)=SRC(4,0)= (t4 + 2*t5 + t6 + 2) >> 2;
  737. SRC(0,5)=SRC(1,4)=SRC(2,3)=SRC(3,2)=SRC(4,1)=SRC(5,0)= (t5 + 2*t6 + t7 + 2) >> 2;
  738. SRC(0,6)=SRC(1,5)=SRC(2,4)=SRC(3,3)=SRC(4,2)=SRC(5,1)=SRC(6,0)= (t6 + 2*t7 + t8 + 2) >> 2;
  739. SRC(0,7)=SRC(1,6)=SRC(2,5)=SRC(3,4)=SRC(4,3)=SRC(5,2)=SRC(6,1)=SRC(7,0)= (t7 + 2*t8 + t9 + 2) >> 2;
  740. SRC(1,7)=SRC(2,6)=SRC(3,5)=SRC(4,4)=SRC(5,3)=SRC(6,2)=SRC(7,1)= (t8 + 2*t9 + t10 + 2) >> 2;
  741. SRC(2,7)=SRC(3,6)=SRC(4,5)=SRC(5,4)=SRC(6,3)=SRC(7,2)= (t9 + 2*t10 + t11 + 2) >> 2;
  742. SRC(3,7)=SRC(4,6)=SRC(5,5)=SRC(6,4)=SRC(7,3)= (t10 + 2*t11 + t12 + 2) >> 2;
  743. SRC(4,7)=SRC(5,6)=SRC(6,5)=SRC(7,4)= (t11 + 2*t12 + t13 + 2) >> 2;
  744. SRC(5,7)=SRC(6,6)=SRC(7,5)= (t12 + 2*t13 + t14 + 2) >> 2;
  745. SRC(6,7)=SRC(7,6)= (t13 + 2*t14 + t15 + 2) >> 2;
  746. SRC(7,7)= (t14 + 3*t15 + 2) >> 2;
  747. }
  748. static void pred8x8l_down_right_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  749. {
  750. PREDICT_8x8_LOAD_TOP;
  751. PREDICT_8x8_LOAD_LEFT;
  752. PREDICT_8x8_LOAD_TOPLEFT;
  753. SRC(0,7)= (l7 + 2*l6 + l5 + 2) >> 2;
  754. SRC(0,6)=SRC(1,7)= (l6 + 2*l5 + l4 + 2) >> 2;
  755. SRC(0,5)=SRC(1,6)=SRC(2,7)= (l5 + 2*l4 + l3 + 2) >> 2;
  756. SRC(0,4)=SRC(1,5)=SRC(2,6)=SRC(3,7)= (l4 + 2*l3 + l2 + 2) >> 2;
  757. SRC(0,3)=SRC(1,4)=SRC(2,5)=SRC(3,6)=SRC(4,7)= (l3 + 2*l2 + l1 + 2) >> 2;
  758. SRC(0,2)=SRC(1,3)=SRC(2,4)=SRC(3,5)=SRC(4,6)=SRC(5,7)= (l2 + 2*l1 + l0 + 2) >> 2;
  759. SRC(0,1)=SRC(1,2)=SRC(2,3)=SRC(3,4)=SRC(4,5)=SRC(5,6)=SRC(6,7)= (l1 + 2*l0 + lt + 2) >> 2;
  760. SRC(0,0)=SRC(1,1)=SRC(2,2)=SRC(3,3)=SRC(4,4)=SRC(5,5)=SRC(6,6)=SRC(7,7)= (l0 + 2*lt + t0 + 2) >> 2;
  761. SRC(1,0)=SRC(2,1)=SRC(3,2)=SRC(4,3)=SRC(5,4)=SRC(6,5)=SRC(7,6)= (lt + 2*t0 + t1 + 2) >> 2;
  762. SRC(2,0)=SRC(3,1)=SRC(4,2)=SRC(5,3)=SRC(6,4)=SRC(7,5)= (t0 + 2*t1 + t2 + 2) >> 2;
  763. SRC(3,0)=SRC(4,1)=SRC(5,2)=SRC(6,3)=SRC(7,4)= (t1 + 2*t2 + t3 + 2) >> 2;
  764. SRC(4,0)=SRC(5,1)=SRC(6,2)=SRC(7,3)= (t2 + 2*t3 + t4 + 2) >> 2;
  765. SRC(5,0)=SRC(6,1)=SRC(7,2)= (t3 + 2*t4 + t5 + 2) >> 2;
  766. SRC(6,0)=SRC(7,1)= (t4 + 2*t5 + t6 + 2) >> 2;
  767. SRC(7,0)= (t5 + 2*t6 + t7 + 2) >> 2;
  768. }
  769. static void pred8x8l_vertical_right_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  770. {
  771. PREDICT_8x8_LOAD_TOP;
  772. PREDICT_8x8_LOAD_LEFT;
  773. PREDICT_8x8_LOAD_TOPLEFT;
  774. SRC(0,6)= (l5 + 2*l4 + l3 + 2) >> 2;
  775. SRC(0,7)= (l6 + 2*l5 + l4 + 2) >> 2;
  776. SRC(0,4)=SRC(1,6)= (l3 + 2*l2 + l1 + 2) >> 2;
  777. SRC(0,5)=SRC(1,7)= (l4 + 2*l3 + l2 + 2) >> 2;
  778. SRC(0,2)=SRC(1,4)=SRC(2,6)= (l1 + 2*l0 + lt + 2) >> 2;
  779. SRC(0,3)=SRC(1,5)=SRC(2,7)= (l2 + 2*l1 + l0 + 2) >> 2;
  780. SRC(0,1)=SRC(1,3)=SRC(2,5)=SRC(3,7)= (l0 + 2*lt + t0 + 2) >> 2;
  781. SRC(0,0)=SRC(1,2)=SRC(2,4)=SRC(3,6)= (lt + t0 + 1) >> 1;
  782. SRC(1,1)=SRC(2,3)=SRC(3,5)=SRC(4,7)= (lt + 2*t0 + t1 + 2) >> 2;
  783. SRC(1,0)=SRC(2,2)=SRC(3,4)=SRC(4,6)= (t0 + t1 + 1) >> 1;
  784. SRC(2,1)=SRC(3,3)=SRC(4,5)=SRC(5,7)= (t0 + 2*t1 + t2 + 2) >> 2;
  785. SRC(2,0)=SRC(3,2)=SRC(4,4)=SRC(5,6)= (t1 + t2 + 1) >> 1;
  786. SRC(3,1)=SRC(4,3)=SRC(5,5)=SRC(6,7)= (t1 + 2*t2 + t3 + 2) >> 2;
  787. SRC(3,0)=SRC(4,2)=SRC(5,4)=SRC(6,6)= (t2 + t3 + 1) >> 1;
  788. SRC(4,1)=SRC(5,3)=SRC(6,5)=SRC(7,7)= (t2 + 2*t3 + t4 + 2) >> 2;
  789. SRC(4,0)=SRC(5,2)=SRC(6,4)=SRC(7,6)= (t3 + t4 + 1) >> 1;
  790. SRC(5,1)=SRC(6,3)=SRC(7,5)= (t3 + 2*t4 + t5 + 2) >> 2;
  791. SRC(5,0)=SRC(6,2)=SRC(7,4)= (t4 + t5 + 1) >> 1;
  792. SRC(6,1)=SRC(7,3)= (t4 + 2*t5 + t6 + 2) >> 2;
  793. SRC(6,0)=SRC(7,2)= (t5 + t6 + 1) >> 1;
  794. SRC(7,1)= (t5 + 2*t6 + t7 + 2) >> 2;
  795. SRC(7,0)= (t6 + t7 + 1) >> 1;
  796. }
  797. static void pred8x8l_horizontal_down_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  798. {
  799. PREDICT_8x8_LOAD_TOP;
  800. PREDICT_8x8_LOAD_LEFT;
  801. PREDICT_8x8_LOAD_TOPLEFT;
  802. SRC(0,7)= (l6 + l7 + 1) >> 1;
  803. SRC(1,7)= (l5 + 2*l6 + l7 + 2) >> 2;
  804. SRC(0,6)=SRC(2,7)= (l5 + l6 + 1) >> 1;
  805. SRC(1,6)=SRC(3,7)= (l4 + 2*l5 + l6 + 2) >> 2;
  806. SRC(0,5)=SRC(2,6)=SRC(4,7)= (l4 + l5 + 1) >> 1;
  807. SRC(1,5)=SRC(3,6)=SRC(5,7)= (l3 + 2*l4 + l5 + 2) >> 2;
  808. SRC(0,4)=SRC(2,5)=SRC(4,6)=SRC(6,7)= (l3 + l4 + 1) >> 1;
  809. SRC(1,4)=SRC(3,5)=SRC(5,6)=SRC(7,7)= (l2 + 2*l3 + l4 + 2) >> 2;
  810. SRC(0,3)=SRC(2,4)=SRC(4,5)=SRC(6,6)= (l2 + l3 + 1) >> 1;
  811. SRC(1,3)=SRC(3,4)=SRC(5,5)=SRC(7,6)= (l1 + 2*l2 + l3 + 2) >> 2;
  812. SRC(0,2)=SRC(2,3)=SRC(4,4)=SRC(6,5)= (l1 + l2 + 1) >> 1;
  813. SRC(1,2)=SRC(3,3)=SRC(5,4)=SRC(7,5)= (l0 + 2*l1 + l2 + 2) >> 2;
  814. SRC(0,1)=SRC(2,2)=SRC(4,3)=SRC(6,4)= (l0 + l1 + 1) >> 1;
  815. SRC(1,1)=SRC(3,2)=SRC(5,3)=SRC(7,4)= (lt + 2*l0 + l1 + 2) >> 2;
  816. SRC(0,0)=SRC(2,1)=SRC(4,2)=SRC(6,3)= (lt + l0 + 1) >> 1;
  817. SRC(1,0)=SRC(3,1)=SRC(5,2)=SRC(7,3)= (l0 + 2*lt + t0 + 2) >> 2;
  818. SRC(2,0)=SRC(4,1)=SRC(6,2)= (t1 + 2*t0 + lt + 2) >> 2;
  819. SRC(3,0)=SRC(5,1)=SRC(7,2)= (t2 + 2*t1 + t0 + 2) >> 2;
  820. SRC(4,0)=SRC(6,1)= (t3 + 2*t2 + t1 + 2) >> 2;
  821. SRC(5,0)=SRC(7,1)= (t4 + 2*t3 + t2 + 2) >> 2;
  822. SRC(6,0)= (t5 + 2*t4 + t3 + 2) >> 2;
  823. SRC(7,0)= (t6 + 2*t5 + t4 + 2) >> 2;
  824. }
  825. static void pred8x8l_vertical_left_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  826. {
  827. PREDICT_8x8_LOAD_TOP;
  828. PREDICT_8x8_LOAD_TOPRIGHT;
  829. SRC(0,0)= (t0 + t1 + 1) >> 1;
  830. SRC(0,1)= (t0 + 2*t1 + t2 + 2) >> 2;
  831. SRC(0,2)=SRC(1,0)= (t1 + t2 + 1) >> 1;
  832. SRC(0,3)=SRC(1,1)= (t1 + 2*t2 + t3 + 2) >> 2;
  833. SRC(0,4)=SRC(1,2)=SRC(2,0)= (t2 + t3 + 1) >> 1;
  834. SRC(0,5)=SRC(1,3)=SRC(2,1)= (t2 + 2*t3 + t4 + 2) >> 2;
  835. SRC(0,6)=SRC(1,4)=SRC(2,2)=SRC(3,0)= (t3 + t4 + 1) >> 1;
  836. SRC(0,7)=SRC(1,5)=SRC(2,3)=SRC(3,1)= (t3 + 2*t4 + t5 + 2) >> 2;
  837. SRC(1,6)=SRC(2,4)=SRC(3,2)=SRC(4,0)= (t4 + t5 + 1) >> 1;
  838. SRC(1,7)=SRC(2,5)=SRC(3,3)=SRC(4,1)= (t4 + 2*t5 + t6 + 2) >> 2;
  839. SRC(2,6)=SRC(3,4)=SRC(4,2)=SRC(5,0)= (t5 + t6 + 1) >> 1;
  840. SRC(2,7)=SRC(3,5)=SRC(4,3)=SRC(5,1)= (t5 + 2*t6 + t7 + 2) >> 2;
  841. SRC(3,6)=SRC(4,4)=SRC(5,2)=SRC(6,0)= (t6 + t7 + 1) >> 1;
  842. SRC(3,7)=SRC(4,5)=SRC(5,3)=SRC(6,1)= (t6 + 2*t7 + t8 + 2) >> 2;
  843. SRC(4,6)=SRC(5,4)=SRC(6,2)=SRC(7,0)= (t7 + t8 + 1) >> 1;
  844. SRC(4,7)=SRC(5,5)=SRC(6,3)=SRC(7,1)= (t7 + 2*t8 + t9 + 2) >> 2;
  845. SRC(5,6)=SRC(6,4)=SRC(7,2)= (t8 + t9 + 1) >> 1;
  846. SRC(5,7)=SRC(6,5)=SRC(7,3)= (t8 + 2*t9 + t10 + 2) >> 2;
  847. SRC(6,6)=SRC(7,4)= (t9 + t10 + 1) >> 1;
  848. SRC(6,7)=SRC(7,5)= (t9 + 2*t10 + t11 + 2) >> 2;
  849. SRC(7,6)= (t10 + t11 + 1) >> 1;
  850. SRC(7,7)= (t10 + 2*t11 + t12 + 2) >> 2;
  851. }
  852. static void pred8x8l_horizontal_up_c(uint8_t *src, int has_topleft, int has_topright, int stride)
  853. {
  854. PREDICT_8x8_LOAD_LEFT;
  855. SRC(0,0)= (l0 + l1 + 1) >> 1;
  856. SRC(1,0)= (l0 + 2*l1 + l2 + 2) >> 2;
  857. SRC(0,1)=SRC(2,0)= (l1 + l2 + 1) >> 1;
  858. SRC(1,1)=SRC(3,0)= (l1 + 2*l2 + l3 + 2) >> 2;
  859. SRC(0,2)=SRC(2,1)=SRC(4,0)= (l2 + l3 + 1) >> 1;
  860. SRC(1,2)=SRC(3,1)=SRC(5,0)= (l2 + 2*l3 + l4 + 2) >> 2;
  861. SRC(0,3)=SRC(2,2)=SRC(4,1)=SRC(6,0)= (l3 + l4 + 1) >> 1;
  862. SRC(1,3)=SRC(3,2)=SRC(5,1)=SRC(7,0)= (l3 + 2*l4 + l5 + 2) >> 2;
  863. SRC(0,4)=SRC(2,3)=SRC(4,2)=SRC(6,1)= (l4 + l5 + 1) >> 1;
  864. SRC(1,4)=SRC(3,3)=SRC(5,2)=SRC(7,1)= (l4 + 2*l5 + l6 + 2) >> 2;
  865. SRC(0,5)=SRC(2,4)=SRC(4,3)=SRC(6,2)= (l5 + l6 + 1) >> 1;
  866. SRC(1,5)=SRC(3,4)=SRC(5,3)=SRC(7,2)= (l5 + 2*l6 + l7 + 2) >> 2;
  867. SRC(0,6)=SRC(2,5)=SRC(4,4)=SRC(6,3)= (l6 + l7 + 1) >> 1;
  868. SRC(1,6)=SRC(3,5)=SRC(5,4)=SRC(7,3)= (l6 + 3*l7 + 2) >> 2;
  869. SRC(0,7)=SRC(1,7)=SRC(2,6)=SRC(2,7)=SRC(3,6)=
  870. SRC(3,7)=SRC(4,5)=SRC(4,6)=SRC(4,7)=SRC(5,5)=
  871. SRC(5,6)=SRC(5,7)=SRC(6,4)=SRC(6,5)=SRC(6,6)=
  872. SRC(6,7)=SRC(7,4)=SRC(7,5)=SRC(7,6)=SRC(7,7)= l7;
  873. }
  874. #undef PREDICT_8x8_LOAD_LEFT
  875. #undef PREDICT_8x8_LOAD_TOP
  876. #undef PREDICT_8x8_LOAD_TOPLEFT
  877. #undef PREDICT_8x8_LOAD_TOPRIGHT
  878. #undef PREDICT_8x8_DC
  879. #undef PTR
  880. #undef PT
  881. #undef PL
  882. #undef SRC
  883. static void pred4x4_vertical_add_c(uint8_t *pix, const DCTELEM *block, int stride){
  884. int i;
  885. pix -= stride;
  886. for(i=0; i<4; i++){
  887. uint8_t v = pix[0];
  888. pix[1*stride]= v += block[0];
  889. pix[2*stride]= v += block[4];
  890. pix[3*stride]= v += block[8];
  891. pix[4*stride]= v += block[12];
  892. pix++;
  893. block++;
  894. }
  895. }
  896. static void pred4x4_horizontal_add_c(uint8_t *pix, const DCTELEM *block, int stride){
  897. int i;
  898. for(i=0; i<4; i++){
  899. uint8_t v = pix[-1];
  900. pix[0]= v += block[0];
  901. pix[1]= v += block[1];
  902. pix[2]= v += block[2];
  903. pix[3]= v += block[3];
  904. pix+= stride;
  905. block+= 4;
  906. }
  907. }
  908. static void pred8x8l_vertical_add_c(uint8_t *pix, const DCTELEM *block, int stride){
  909. int i;
  910. pix -= stride;
  911. for(i=0; i<8; i++){
  912. uint8_t v = pix[0];
  913. pix[1*stride]= v += block[0];
  914. pix[2*stride]= v += block[8];
  915. pix[3*stride]= v += block[16];
  916. pix[4*stride]= v += block[24];
  917. pix[5*stride]= v += block[32];
  918. pix[6*stride]= v += block[40];
  919. pix[7*stride]= v += block[48];
  920. pix[8*stride]= v += block[56];
  921. pix++;
  922. block++;
  923. }
  924. }
  925. static void pred8x8l_horizontal_add_c(uint8_t *pix, const DCTELEM *block, int stride){
  926. int i;
  927. for(i=0; i<8; i++){
  928. uint8_t v = pix[-1];
  929. pix[0]= v += block[0];
  930. pix[1]= v += block[1];
  931. pix[2]= v += block[2];
  932. pix[3]= v += block[3];
  933. pix[4]= v += block[4];
  934. pix[5]= v += block[5];
  935. pix[6]= v += block[6];
  936. pix[7]= v += block[7];
  937. pix+= stride;
  938. block+= 8;
  939. }
  940. }
  941. static void pred16x16_vertical_add_c(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  942. int i;
  943. for(i=0; i<16; i++)
  944. pred4x4_vertical_add_c(pix + block_offset[i], block + i*16, stride);
  945. }
  946. static void pred16x16_horizontal_add_c(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  947. int i;
  948. for(i=0; i<16; i++)
  949. pred4x4_horizontal_add_c(pix + block_offset[i], block + i*16, stride);
  950. }
  951. static void pred8x8_vertical_add_c(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  952. int i;
  953. for(i=0; i<4; i++)
  954. pred4x4_vertical_add_c(pix + block_offset[i], block + i*16, stride);
  955. }
  956. static void pred8x8_horizontal_add_c(uint8_t *pix, const int *block_offset, const DCTELEM *block, int stride){
  957. int i;
  958. for(i=0; i<4; i++)
  959. pred4x4_horizontal_add_c(pix + block_offset[i], block + i*16, stride);
  960. }
  961. /**
  962. * Sets the intra prediction function pointers.
  963. */
  964. void ff_h264_pred_init(H264PredContext *h, int codec_id){
  965. // MpegEncContext * const s = &h->s;
  966. if(codec_id != CODEC_ID_RV40){
  967. h->pred4x4[VERT_PRED ]= pred4x4_vertical_c;
  968. h->pred4x4[HOR_PRED ]= pred4x4_horizontal_c;
  969. h->pred4x4[DC_PRED ]= pred4x4_dc_c;
  970. if(codec_id == CODEC_ID_SVQ3)
  971. h->pred4x4[DIAG_DOWN_LEFT_PRED ]= pred4x4_down_left_svq3_c;
  972. else
  973. h->pred4x4[DIAG_DOWN_LEFT_PRED ]= pred4x4_down_left_c;
  974. h->pred4x4[DIAG_DOWN_RIGHT_PRED]= pred4x4_down_right_c;
  975. h->pred4x4[VERT_RIGHT_PRED ]= pred4x4_vertical_right_c;
  976. h->pred4x4[HOR_DOWN_PRED ]= pred4x4_horizontal_down_c;
  977. h->pred4x4[VERT_LEFT_PRED ]= pred4x4_vertical_left_c;
  978. h->pred4x4[HOR_UP_PRED ]= pred4x4_horizontal_up_c;
  979. h->pred4x4[LEFT_DC_PRED ]= pred4x4_left_dc_c;
  980. h->pred4x4[TOP_DC_PRED ]= pred4x4_top_dc_c;
  981. h->pred4x4[DC_128_PRED ]= pred4x4_128_dc_c;
  982. }else{
  983. h->pred4x4[VERT_PRED ]= pred4x4_vertical_c;
  984. h->pred4x4[HOR_PRED ]= pred4x4_horizontal_c;
  985. h->pred4x4[DC_PRED ]= pred4x4_dc_c;
  986. h->pred4x4[DIAG_DOWN_LEFT_PRED ]= pred4x4_down_left_rv40_c;
  987. h->pred4x4[DIAG_DOWN_RIGHT_PRED]= pred4x4_down_right_c;
  988. h->pred4x4[VERT_RIGHT_PRED ]= pred4x4_vertical_right_c;
  989. h->pred4x4[HOR_DOWN_PRED ]= pred4x4_horizontal_down_c;
  990. h->pred4x4[VERT_LEFT_PRED ]= pred4x4_vertical_left_rv40_c;
  991. h->pred4x4[HOR_UP_PRED ]= pred4x4_horizontal_up_rv40_c;
  992. h->pred4x4[LEFT_DC_PRED ]= pred4x4_left_dc_c;
  993. h->pred4x4[TOP_DC_PRED ]= pred4x4_top_dc_c;
  994. h->pred4x4[DC_128_PRED ]= pred4x4_128_dc_c;
  995. h->pred4x4[DIAG_DOWN_LEFT_PRED_RV40_NODOWN]= pred4x4_down_left_rv40_nodown_c;
  996. h->pred4x4[HOR_UP_PRED_RV40_NODOWN]= pred4x4_horizontal_up_rv40_nodown_c;
  997. h->pred4x4[VERT_LEFT_PRED_RV40_NODOWN]= pred4x4_vertical_left_rv40_nodown_c;
  998. }
  999. h->pred8x8l[VERT_PRED ]= pred8x8l_vertical_c;
  1000. h->pred8x8l[HOR_PRED ]= pred8x8l_horizontal_c;
  1001. h->pred8x8l[DC_PRED ]= pred8x8l_dc_c;
  1002. h->pred8x8l[DIAG_DOWN_LEFT_PRED ]= pred8x8l_down_left_c;
  1003. h->pred8x8l[DIAG_DOWN_RIGHT_PRED]= pred8x8l_down_right_c;
  1004. h->pred8x8l[VERT_RIGHT_PRED ]= pred8x8l_vertical_right_c;
  1005. h->pred8x8l[HOR_DOWN_PRED ]= pred8x8l_horizontal_down_c;
  1006. h->pred8x8l[VERT_LEFT_PRED ]= pred8x8l_vertical_left_c;
  1007. h->pred8x8l[HOR_UP_PRED ]= pred8x8l_horizontal_up_c;
  1008. h->pred8x8l[LEFT_DC_PRED ]= pred8x8l_left_dc_c;
  1009. h->pred8x8l[TOP_DC_PRED ]= pred8x8l_top_dc_c;
  1010. h->pred8x8l[DC_128_PRED ]= pred8x8l_128_dc_c;
  1011. h->pred8x8[VERT_PRED8x8 ]= pred8x8_vertical_c;
  1012. h->pred8x8[HOR_PRED8x8 ]= pred8x8_horizontal_c;
  1013. h->pred8x8[PLANE_PRED8x8 ]= pred8x8_plane_c;
  1014. if(codec_id != CODEC_ID_RV40){
  1015. h->pred8x8[DC_PRED8x8 ]= pred8x8_dc_c;
  1016. h->pred8x8[LEFT_DC_PRED8x8]= pred8x8_left_dc_c;
  1017. h->pred8x8[TOP_DC_PRED8x8 ]= pred8x8_top_dc_c;
  1018. h->pred8x8[ALZHEIMER_DC_L0T_PRED8x8 ]= pred8x8_mad_cow_dc_l0t;
  1019. h->pred8x8[ALZHEIMER_DC_0LT_PRED8x8 ]= pred8x8_mad_cow_dc_0lt;
  1020. h->pred8x8[ALZHEIMER_DC_L00_PRED8x8 ]= pred8x8_mad_cow_dc_l00;
  1021. h->pred8x8[ALZHEIMER_DC_0L0_PRED8x8 ]= pred8x8_mad_cow_dc_0l0;
  1022. }else{
  1023. h->pred8x8[DC_PRED8x8 ]= pred8x8_dc_rv40_c;
  1024. h->pred8x8[LEFT_DC_PRED8x8]= pred8x8_left_dc_rv40_c;
  1025. h->pred8x8[TOP_DC_PRED8x8 ]= pred8x8_top_dc_rv40_c;
  1026. }
  1027. h->pred8x8[DC_128_PRED8x8 ]= pred8x8_128_dc_c;
  1028. h->pred16x16[DC_PRED8x8 ]= pred16x16_dc_c;
  1029. h->pred16x16[VERT_PRED8x8 ]= pred16x16_vertical_c;
  1030. h->pred16x16[HOR_PRED8x8 ]= pred16x16_horizontal_c;
  1031. h->pred16x16[PLANE_PRED8x8 ]= pred16x16_plane_c;
  1032. switch(codec_id){
  1033. case CODEC_ID_SVQ3:
  1034. h->pred16x16[PLANE_PRED8x8 ]= pred16x16_plane_svq3_c;
  1035. break;
  1036. case CODEC_ID_RV40:
  1037. h->pred16x16[PLANE_PRED8x8 ]= pred16x16_plane_rv40_c;
  1038. break;
  1039. default:
  1040. h->pred16x16[PLANE_PRED8x8 ]= pred16x16_plane_c;
  1041. }
  1042. h->pred16x16[LEFT_DC_PRED8x8]= pred16x16_left_dc_c;
  1043. h->pred16x16[TOP_DC_PRED8x8 ]= pred16x16_top_dc_c;
  1044. h->pred16x16[DC_128_PRED8x8 ]= pred16x16_128_dc_c;
  1045. //special lossless h/v prediction for h264
  1046. h->pred4x4_add [VERT_PRED ]= pred4x4_vertical_add_c;
  1047. h->pred4x4_add [ HOR_PRED ]= pred4x4_horizontal_add_c;
  1048. h->pred8x8l_add [VERT_PRED ]= pred8x8l_vertical_add_c;
  1049. h->pred8x8l_add [ HOR_PRED ]= pred8x8l_horizontal_add_c;
  1050. h->pred8x8_add [VERT_PRED8x8]= pred8x8_vertical_add_c;
  1051. h->pred8x8_add [ HOR_PRED8x8]= pred8x8_horizontal_add_c;
  1052. h->pred16x16_add[VERT_PRED8x8]= pred16x16_vertical_add_c;
  1053. h->pred16x16_add[ HOR_PRED8x8]= pred16x16_horizontal_add_c;
  1054. }