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.

807 lines
25KB

  1. /*
  2. * software YUV to RGB converter
  3. *
  4. * Copyright (C) 2009 Konstantin Shishkov
  5. *
  6. * 1,4,8bpp support and context / deglobalize stuff
  7. * by Michael Niedermayer (michaelni@gmx.at)
  8. *
  9. * This file is part of Libav.
  10. *
  11. * Libav is free software; you can redistribute it and/or
  12. * modify it under the terms of the GNU Lesser General Public
  13. * License as published by the Free Software Foundation; either
  14. * version 2.1 of the License, or (at your option) any later version.
  15. *
  16. * Libav is distributed in the hope that it will be useful,
  17. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  18. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  19. * Lesser General Public License for more details.
  20. *
  21. * You should have received a copy of the GNU Lesser General Public
  22. * License along with Libav; if not, write to the Free Software
  23. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  24. */
  25. #include <stdio.h>
  26. #include <stdlib.h>
  27. #include <inttypes.h>
  28. #include <assert.h>
  29. #include "config.h"
  30. #include "rgb2rgb.h"
  31. #include "swscale.h"
  32. #include "swscale_internal.h"
  33. #include "libavutil/cpu.h"
  34. #include "libavutil/bswap.h"
  35. extern const uint8_t dither_4x4_16[4][8];
  36. extern const uint8_t dither_8x8_32[8][8];
  37. extern const uint8_t dither_8x8_73[8][8];
  38. extern const uint8_t dither_8x8_220[8][8];
  39. const int32_t ff_yuv2rgb_coeffs[8][4] = {
  40. {117504, 138453, 13954, 34903}, /* no sequence_display_extension */
  41. {117504, 138453, 13954, 34903}, /* ITU-R Rec. 709 (1990) */
  42. {104597, 132201, 25675, 53279}, /* unspecified */
  43. {104597, 132201, 25675, 53279}, /* reserved */
  44. {104448, 132798, 24759, 53109}, /* FCC */
  45. {104597, 132201, 25675, 53279}, /* ITU-R Rec. 624-4 System B, G */
  46. {104597, 132201, 25675, 53279}, /* SMPTE 170M */
  47. {117579, 136230, 16907, 35559} /* SMPTE 240M (1987) */
  48. };
  49. const int *sws_getCoefficients(int colorspace)
  50. {
  51. if (colorspace > 7 || colorspace < 0)
  52. colorspace = SWS_CS_DEFAULT;
  53. return ff_yuv2rgb_coeffs[colorspace];
  54. }
  55. #define LOADCHROMA(i) \
  56. U = pu[i]; \
  57. V = pv[i]; \
  58. r = (void *)c->table_rV[V]; \
  59. g = (void *)(c->table_gU[U] + c->table_gV[V]); \
  60. b = (void *)c->table_bU[U];
  61. #define PUTRGB(dst,src,i) \
  62. Y = src[2*i]; \
  63. dst[2*i ] = r[Y] + g[Y] + b[Y]; \
  64. Y = src[2*i+1]; \
  65. dst[2*i+1] = r[Y] + g[Y] + b[Y];
  66. #define PUTRGB24(dst,src,i) \
  67. Y = src[2*i]; \
  68. dst[6*i+0] = r[Y]; dst[6*i+1] = g[Y]; dst[6*i+2] = b[Y]; \
  69. Y = src[2*i+1]; \
  70. dst[6*i+3] = r[Y]; dst[6*i+4] = g[Y]; dst[6*i+5] = b[Y];
  71. #define PUTBGR24(dst,src,i) \
  72. Y = src[2*i]; \
  73. dst[6*i+0] = b[Y]; dst[6*i+1] = g[Y]; dst[6*i+2] = r[Y]; \
  74. Y = src[2*i+1]; \
  75. dst[6*i+3] = b[Y]; dst[6*i+4] = g[Y]; dst[6*i+5] = r[Y];
  76. #define PUTRGBA(dst,ysrc,asrc,i,s) \
  77. Y = ysrc[2*i]; \
  78. dst[2*i ] = r[Y] + g[Y] + b[Y] + (asrc[2*i ]<<s); \
  79. Y = ysrc[2*i+1]; \
  80. dst[2*i+1] = r[Y] + g[Y] + b[Y] + (asrc[2*i+1]<<s);
  81. #define PUTRGB48(dst,src,i) \
  82. Y = src[2*i]; \
  83. dst[12*i+ 0] = dst[12*i+ 1] = r[Y]; \
  84. dst[12*i+ 2] = dst[12*i+ 3] = g[Y]; \
  85. dst[12*i+ 4] = dst[12*i+ 5] = b[Y]; \
  86. Y = src[2*i+1]; \
  87. dst[12*i+ 6] = dst[12*i+ 7] = r[Y]; \
  88. dst[12*i+ 8] = dst[12*i+ 9] = g[Y]; \
  89. dst[12*i+10] = dst[12*i+11] = b[Y];
  90. #define PUTBGR48(dst,src,i) \
  91. Y = src[2*i]; \
  92. dst[12*i+ 0] = dst[12*i+ 1] = b[Y]; \
  93. dst[12*i+ 2] = dst[12*i+ 3] = g[Y]; \
  94. dst[12*i+ 4] = dst[12*i+ 5] = r[Y]; \
  95. Y = src[2*i+1]; \
  96. dst[12*i+ 6] = dst[12*i+ 7] = b[Y]; \
  97. dst[12*i+ 8] = dst[12*i+ 9] = g[Y]; \
  98. dst[12*i+10] = dst[12*i+11] = r[Y];
  99. #define YUV2RGBFUNC(func_name, dst_type, alpha) \
  100. static int func_name(SwsContext *c, const uint8_t* src[], int srcStride[], int srcSliceY, \
  101. int srcSliceH, uint8_t* dst[], int dstStride[]) \
  102. {\
  103. int y;\
  104. \
  105. if (!alpha && c->srcFormat == PIX_FMT_YUV422P) {\
  106. srcStride[1] *= 2;\
  107. srcStride[2] *= 2;\
  108. }\
  109. for (y=0; y<srcSliceH; y+=2) {\
  110. dst_type *dst_1 = (dst_type*)(dst[0] + (y+srcSliceY )*dstStride[0]);\
  111. dst_type *dst_2 = (dst_type*)(dst[0] + (y+srcSliceY+1)*dstStride[0]);\
  112. dst_type av_unused *r, *b;\
  113. dst_type *g;\
  114. const uint8_t *py_1 = src[0] + y*srcStride[0];\
  115. const uint8_t *py_2 = py_1 + srcStride[0];\
  116. const uint8_t *pu = src[1] + (y>>1)*srcStride[1];\
  117. const uint8_t *pv = src[2] + (y>>1)*srcStride[2];\
  118. const uint8_t av_unused *pa_1, *pa_2;\
  119. unsigned int h_size = c->dstW>>3;\
  120. if (alpha) {\
  121. pa_1 = src[3] + y*srcStride[3];\
  122. pa_2 = pa_1 + srcStride[3];\
  123. }\
  124. while (h_size--) {\
  125. int av_unused U, V;\
  126. int Y;\
  127. #define ENDYUV2RGBLINE(dst_delta)\
  128. pu += 4;\
  129. pv += 4;\
  130. py_1 += 8;\
  131. py_2 += 8;\
  132. dst_1 += dst_delta;\
  133. dst_2 += dst_delta;\
  134. }\
  135. if (c->dstW & 4) {\
  136. int av_unused Y, U, V;\
  137. #define ENDYUV2RGBFUNC()\
  138. }\
  139. }\
  140. return srcSliceH;\
  141. }
  142. #define CLOSEYUV2RGBFUNC(dst_delta)\
  143. ENDYUV2RGBLINE(dst_delta)\
  144. ENDYUV2RGBFUNC()
  145. YUV2RGBFUNC(yuv2rgb_c_48, uint8_t, 0)
  146. LOADCHROMA(0);
  147. PUTRGB48(dst_1,py_1,0);
  148. PUTRGB48(dst_2,py_2,0);
  149. LOADCHROMA(1);
  150. PUTRGB48(dst_2,py_2,1);
  151. PUTRGB48(dst_1,py_1,1);
  152. LOADCHROMA(2);
  153. PUTRGB48(dst_1,py_1,2);
  154. PUTRGB48(dst_2,py_2,2);
  155. LOADCHROMA(3);
  156. PUTRGB48(dst_2,py_2,3);
  157. PUTRGB48(dst_1,py_1,3);
  158. ENDYUV2RGBLINE(48)
  159. LOADCHROMA(0);
  160. PUTRGB48(dst_1,py_1,0);
  161. PUTRGB48(dst_2,py_2,0);
  162. LOADCHROMA(1);
  163. PUTRGB48(dst_2,py_2,1);
  164. PUTRGB48(dst_1,py_1,1);
  165. ENDYUV2RGBFUNC()
  166. YUV2RGBFUNC(yuv2rgb_c_bgr48, uint8_t, 0)
  167. LOADCHROMA(0);
  168. PUTBGR48(dst_1,py_1,0);
  169. PUTBGR48(dst_2,py_2,0);
  170. LOADCHROMA(1);
  171. PUTBGR48(dst_2,py_2,1);
  172. PUTBGR48(dst_1,py_1,1);
  173. LOADCHROMA(2);
  174. PUTBGR48(dst_1,py_1,2);
  175. PUTBGR48(dst_2,py_2,2);
  176. LOADCHROMA(3);
  177. PUTBGR48(dst_2,py_2,3);
  178. PUTBGR48(dst_1,py_1,3);
  179. ENDYUV2RGBLINE(48)
  180. LOADCHROMA(0);
  181. PUTBGR48(dst_1,py_1,0);
  182. PUTBGR48(dst_2,py_2,0);
  183. LOADCHROMA(1);
  184. PUTBGR48(dst_2,py_2,1);
  185. PUTBGR48(dst_1,py_1,1);
  186. ENDYUV2RGBFUNC()
  187. YUV2RGBFUNC(yuv2rgb_c_32, uint32_t, 0)
  188. LOADCHROMA(0);
  189. PUTRGB(dst_1,py_1,0);
  190. PUTRGB(dst_2,py_2,0);
  191. LOADCHROMA(1);
  192. PUTRGB(dst_2,py_2,1);
  193. PUTRGB(dst_1,py_1,1);
  194. LOADCHROMA(2);
  195. PUTRGB(dst_1,py_1,2);
  196. PUTRGB(dst_2,py_2,2);
  197. LOADCHROMA(3);
  198. PUTRGB(dst_2,py_2,3);
  199. PUTRGB(dst_1,py_1,3);
  200. ENDYUV2RGBLINE(8)
  201. LOADCHROMA(0);
  202. PUTRGB(dst_1,py_1,0);
  203. PUTRGB(dst_2,py_2,0);
  204. LOADCHROMA(1);
  205. PUTRGB(dst_2,py_2,1);
  206. PUTRGB(dst_1,py_1,1);
  207. ENDYUV2RGBFUNC()
  208. YUV2RGBFUNC(yuva2rgba_c, uint32_t, 1)
  209. LOADCHROMA(0);
  210. PUTRGBA(dst_1,py_1,pa_1,0,24);
  211. PUTRGBA(dst_2,py_2,pa_2,0,24);
  212. LOADCHROMA(1);
  213. PUTRGBA(dst_2,py_2,pa_1,1,24);
  214. PUTRGBA(dst_1,py_1,pa_2,1,24);
  215. LOADCHROMA(2);
  216. PUTRGBA(dst_1,py_1,pa_1,2,24);
  217. PUTRGBA(dst_2,py_2,pa_2,2,24);
  218. LOADCHROMA(3);
  219. PUTRGBA(dst_2,py_2,pa_1,3,24);
  220. PUTRGBA(dst_1,py_1,pa_2,3,24);
  221. pa_1 += 8;\
  222. pa_2 += 8;\
  223. ENDYUV2RGBLINE(8)
  224. LOADCHROMA(0);
  225. PUTRGBA(dst_1,py_1,pa_1,0,24);
  226. PUTRGBA(dst_2,py_2,pa_2,0,24);
  227. LOADCHROMA(1);
  228. PUTRGBA(dst_2,py_2,pa_1,1,24);
  229. PUTRGBA(dst_1,py_1,pa_2,1,24);
  230. ENDYUV2RGBFUNC()
  231. YUV2RGBFUNC(yuva2argb_c, uint32_t, 1)
  232. LOADCHROMA(0);
  233. PUTRGBA(dst_1,py_1,pa_1,0,0);
  234. PUTRGBA(dst_2,py_2,pa_2,0,0);
  235. LOADCHROMA(1);
  236. PUTRGBA(dst_2,py_2,pa_2,1,0);
  237. PUTRGBA(dst_1,py_1,pa_1,1,0);
  238. LOADCHROMA(2);
  239. PUTRGBA(dst_1,py_1,pa_1,2,0);
  240. PUTRGBA(dst_2,py_2,pa_2,2,0);
  241. LOADCHROMA(3);
  242. PUTRGBA(dst_2,py_2,pa_2,3,0);
  243. PUTRGBA(dst_1,py_1,pa_1,3,0);
  244. pa_1 += 8;\
  245. pa_2 += 8;\
  246. ENDYUV2RGBLINE(8)
  247. LOADCHROMA(0);
  248. PUTRGBA(dst_1,py_1,pa_1,0,0);
  249. PUTRGBA(dst_2,py_2,pa_2,0,0);
  250. LOADCHROMA(1);
  251. PUTRGBA(dst_2,py_2,pa_2,1,0);
  252. PUTRGBA(dst_1,py_1,pa_1,1,0);
  253. ENDYUV2RGBFUNC()
  254. YUV2RGBFUNC(yuv2rgb_c_24_rgb, uint8_t, 0)
  255. LOADCHROMA(0);
  256. PUTRGB24(dst_1,py_1,0);
  257. PUTRGB24(dst_2,py_2,0);
  258. LOADCHROMA(1);
  259. PUTRGB24(dst_2,py_2,1);
  260. PUTRGB24(dst_1,py_1,1);
  261. LOADCHROMA(2);
  262. PUTRGB24(dst_1,py_1,2);
  263. PUTRGB24(dst_2,py_2,2);
  264. LOADCHROMA(3);
  265. PUTRGB24(dst_2,py_2,3);
  266. PUTRGB24(dst_1,py_1,3);
  267. ENDYUV2RGBLINE(24)
  268. LOADCHROMA(0);
  269. PUTRGB24(dst_1,py_1,0);
  270. PUTRGB24(dst_2,py_2,0);
  271. LOADCHROMA(1);
  272. PUTRGB24(dst_2,py_2,1);
  273. PUTRGB24(dst_1,py_1,1);
  274. ENDYUV2RGBFUNC()
  275. // only trivial mods from yuv2rgb_c_24_rgb
  276. YUV2RGBFUNC(yuv2rgb_c_24_bgr, uint8_t, 0)
  277. LOADCHROMA(0);
  278. PUTBGR24(dst_1,py_1,0);
  279. PUTBGR24(dst_2,py_2,0);
  280. LOADCHROMA(1);
  281. PUTBGR24(dst_2,py_2,1);
  282. PUTBGR24(dst_1,py_1,1);
  283. LOADCHROMA(2);
  284. PUTBGR24(dst_1,py_1,2);
  285. PUTBGR24(dst_2,py_2,2);
  286. LOADCHROMA(3);
  287. PUTBGR24(dst_2,py_2,3);
  288. PUTBGR24(dst_1,py_1,3);
  289. ENDYUV2RGBLINE(24)
  290. LOADCHROMA(0);
  291. PUTBGR24(dst_1,py_1,0);
  292. PUTBGR24(dst_2,py_2,0);
  293. LOADCHROMA(1);
  294. PUTBGR24(dst_2,py_2,1);
  295. PUTBGR24(dst_1,py_1,1);
  296. ENDYUV2RGBFUNC()
  297. // This is exactly the same code as yuv2rgb_c_32 except for the types of
  298. // r, g, b, dst_1, dst_2
  299. YUV2RGBFUNC(yuv2rgb_c_16, uint16_t, 0)
  300. LOADCHROMA(0);
  301. PUTRGB(dst_1,py_1,0);
  302. PUTRGB(dst_2,py_2,0);
  303. LOADCHROMA(1);
  304. PUTRGB(dst_2,py_2,1);
  305. PUTRGB(dst_1,py_1,1);
  306. LOADCHROMA(2);
  307. PUTRGB(dst_1,py_1,2);
  308. PUTRGB(dst_2,py_2,2);
  309. LOADCHROMA(3);
  310. PUTRGB(dst_2,py_2,3);
  311. PUTRGB(dst_1,py_1,3);
  312. CLOSEYUV2RGBFUNC(8)
  313. // r, g, b, dst_1, dst_2
  314. YUV2RGBFUNC(yuv2rgb_c_12_ordered_dither, uint16_t, 0)
  315. const uint8_t *d16 = dither_4x4_16[y&3];
  316. #define PUTRGB12(dst,src,i,o) \
  317. Y = src[2*i]; \
  318. dst[2*i] = r[Y+d16[0+o]] + g[Y+d16[0+o]] + b[Y+d16[0+o]]; \
  319. Y = src[2*i+1]; \
  320. dst[2*i+1] = r[Y+d16[1+o]] + g[Y+d16[1+o]] + b[Y+d16[1+o]];
  321. LOADCHROMA(0);
  322. PUTRGB12(dst_1,py_1,0,0);
  323. PUTRGB12(dst_2,py_2,0,0+8);
  324. LOADCHROMA(1);
  325. PUTRGB12(dst_2,py_2,1,2+8);
  326. PUTRGB12(dst_1,py_1,1,2);
  327. LOADCHROMA(2);
  328. PUTRGB12(dst_1,py_1,2,4);
  329. PUTRGB12(dst_2,py_2,2,4+8);
  330. LOADCHROMA(3);
  331. PUTRGB12(dst_2,py_2,3,6+8);
  332. PUTRGB12(dst_1,py_1,3,6);
  333. CLOSEYUV2RGBFUNC(8)
  334. // r, g, b, dst_1, dst_2
  335. YUV2RGBFUNC(yuv2rgb_c_8_ordered_dither, uint8_t, 0)
  336. const uint8_t *d32 = dither_8x8_32[y&7];
  337. const uint8_t *d64 = dither_8x8_73[y&7];
  338. #define PUTRGB8(dst,src,i,o) \
  339. Y = src[2*i]; \
  340. dst[2*i] = r[Y+d32[0+o]] + g[Y+d32[0+o]] + b[Y+d64[0+o]]; \
  341. Y = src[2*i+1]; \
  342. dst[2*i+1] = r[Y+d32[1+o]] + g[Y+d32[1+o]] + b[Y+d64[1+o]];
  343. LOADCHROMA(0);
  344. PUTRGB8(dst_1,py_1,0,0);
  345. PUTRGB8(dst_2,py_2,0,0+8);
  346. LOADCHROMA(1);
  347. PUTRGB8(dst_2,py_2,1,2+8);
  348. PUTRGB8(dst_1,py_1,1,2);
  349. LOADCHROMA(2);
  350. PUTRGB8(dst_1,py_1,2,4);
  351. PUTRGB8(dst_2,py_2,2,4+8);
  352. LOADCHROMA(3);
  353. PUTRGB8(dst_2,py_2,3,6+8);
  354. PUTRGB8(dst_1,py_1,3,6);
  355. CLOSEYUV2RGBFUNC(8)
  356. YUV2RGBFUNC(yuv2rgb_c_4_ordered_dither, uint8_t, 0)
  357. const uint8_t *d64 = dither_8x8_73[y&7];
  358. const uint8_t *d128 = dither_8x8_220[y&7];
  359. int acc;
  360. #define PUTRGB4D(dst,src,i,o) \
  361. Y = src[2*i]; \
  362. acc = r[Y+d128[0+o]] + g[Y+d64[0+o]] + b[Y+d128[0+o]]; \
  363. Y = src[2*i+1]; \
  364. acc |= (r[Y+d128[1+o]] + g[Y+d64[1+o]] + b[Y+d128[1+o]])<<4; \
  365. dst[i]= acc;
  366. LOADCHROMA(0);
  367. PUTRGB4D(dst_1,py_1,0,0);
  368. PUTRGB4D(dst_2,py_2,0,0+8);
  369. LOADCHROMA(1);
  370. PUTRGB4D(dst_2,py_2,1,2+8);
  371. PUTRGB4D(dst_1,py_1,1,2);
  372. LOADCHROMA(2);
  373. PUTRGB4D(dst_1,py_1,2,4);
  374. PUTRGB4D(dst_2,py_2,2,4+8);
  375. LOADCHROMA(3);
  376. PUTRGB4D(dst_2,py_2,3,6+8);
  377. PUTRGB4D(dst_1,py_1,3,6);
  378. CLOSEYUV2RGBFUNC(4)
  379. YUV2RGBFUNC(yuv2rgb_c_4b_ordered_dither, uint8_t, 0)
  380. const uint8_t *d64 = dither_8x8_73[y&7];
  381. const uint8_t *d128 = dither_8x8_220[y&7];
  382. #define PUTRGB4DB(dst,src,i,o) \
  383. Y = src[2*i]; \
  384. dst[2*i] = r[Y+d128[0+o]] + g[Y+d64[0+o]] + b[Y+d128[0+o]]; \
  385. Y = src[2*i+1]; \
  386. dst[2*i+1] = r[Y+d128[1+o]] + g[Y+d64[1+o]] + b[Y+d128[1+o]];
  387. LOADCHROMA(0);
  388. PUTRGB4DB(dst_1,py_1,0,0);
  389. PUTRGB4DB(dst_2,py_2,0,0+8);
  390. LOADCHROMA(1);
  391. PUTRGB4DB(dst_2,py_2,1,2+8);
  392. PUTRGB4DB(dst_1,py_1,1,2);
  393. LOADCHROMA(2);
  394. PUTRGB4DB(dst_1,py_1,2,4);
  395. PUTRGB4DB(dst_2,py_2,2,4+8);
  396. LOADCHROMA(3);
  397. PUTRGB4DB(dst_2,py_2,3,6+8);
  398. PUTRGB4DB(dst_1,py_1,3,6);
  399. CLOSEYUV2RGBFUNC(8)
  400. YUV2RGBFUNC(yuv2rgb_c_1_ordered_dither, uint8_t, 0)
  401. const uint8_t *d128 = dither_8x8_220[y&7];
  402. char out_1 = 0, out_2 = 0;
  403. g= c->table_gU[128] + c->table_gV[128];
  404. #define PUTRGB1(out,src,i,o) \
  405. Y = src[2*i]; \
  406. out+= out + g[Y+d128[0+o]]; \
  407. Y = src[2*i+1]; \
  408. out+= out + g[Y+d128[1+o]];
  409. PUTRGB1(out_1,py_1,0,0);
  410. PUTRGB1(out_2,py_2,0,0+8);
  411. PUTRGB1(out_2,py_2,1,2+8);
  412. PUTRGB1(out_1,py_1,1,2);
  413. PUTRGB1(out_1,py_1,2,4);
  414. PUTRGB1(out_2,py_2,2,4+8);
  415. PUTRGB1(out_2,py_2,3,6+8);
  416. PUTRGB1(out_1,py_1,3,6);
  417. dst_1[0]= out_1;
  418. dst_2[0]= out_2;
  419. CLOSEYUV2RGBFUNC(1)
  420. SwsFunc ff_yuv2rgb_get_func_ptr(SwsContext *c)
  421. {
  422. SwsFunc t = NULL;
  423. if (HAVE_MMX) {
  424. t = ff_yuv2rgb_init_mmx(c);
  425. } else if (HAVE_VIS) {
  426. t = ff_yuv2rgb_init_vis(c);
  427. } else if (HAVE_ALTIVEC) {
  428. t = ff_yuv2rgb_init_altivec(c);
  429. } else if (ARCH_BFIN) {
  430. t = ff_yuv2rgb_get_func_ptr_bfin(c);
  431. }
  432. if (t)
  433. return t;
  434. av_log(c, AV_LOG_WARNING, "No accelerated colorspace conversion found from %s to %s.\n", sws_format_name(c->srcFormat), sws_format_name(c->dstFormat));
  435. switch (c->dstFormat) {
  436. case PIX_FMT_BGR48BE:
  437. case PIX_FMT_BGR48LE: return yuv2rgb_c_bgr48;
  438. case PIX_FMT_RGB48BE:
  439. case PIX_FMT_RGB48LE: return yuv2rgb_c_48;
  440. case PIX_FMT_ARGB:
  441. case PIX_FMT_ABGR: if (CONFIG_SWSCALE_ALPHA && c->srcFormat == PIX_FMT_YUVA420P) return yuva2argb_c;
  442. case PIX_FMT_RGBA:
  443. case PIX_FMT_BGRA: return (CONFIG_SWSCALE_ALPHA && c->srcFormat == PIX_FMT_YUVA420P) ? yuva2rgba_c : yuv2rgb_c_32;
  444. case PIX_FMT_RGB24: return yuv2rgb_c_24_rgb;
  445. case PIX_FMT_BGR24: return yuv2rgb_c_24_bgr;
  446. case PIX_FMT_RGB565:
  447. case PIX_FMT_BGR565:
  448. case PIX_FMT_RGB555:
  449. case PIX_FMT_BGR555: return yuv2rgb_c_16;
  450. case PIX_FMT_RGB444:
  451. case PIX_FMT_BGR444: return yuv2rgb_c_12_ordered_dither;
  452. case PIX_FMT_RGB8:
  453. case PIX_FMT_BGR8: return yuv2rgb_c_8_ordered_dither;
  454. case PIX_FMT_RGB4:
  455. case PIX_FMT_BGR4: return yuv2rgb_c_4_ordered_dither;
  456. case PIX_FMT_RGB4_BYTE:
  457. case PIX_FMT_BGR4_BYTE: return yuv2rgb_c_4b_ordered_dither;
  458. case PIX_FMT_MONOBLACK: return yuv2rgb_c_1_ordered_dither;
  459. default:
  460. assert(0);
  461. }
  462. return NULL;
  463. }
  464. static void fill_table(uint8_t* table[256], const int elemsize, const int inc, void *y_tab)
  465. {
  466. int i;
  467. int64_t cb = 0;
  468. uint8_t *y_table = y_tab;
  469. y_table -= elemsize * (inc >> 9);
  470. for (i = 0; i < 256; i++) {
  471. table[i] = y_table + elemsize * (cb >> 16);
  472. cb += inc;
  473. }
  474. }
  475. static void fill_gv_table(int table[256], const int elemsize, const int inc)
  476. {
  477. int i;
  478. int64_t cb = 0;
  479. int off = -(inc >> 9);
  480. for (i = 0; i < 256; i++) {
  481. table[i] = elemsize * (off + (cb >> 16));
  482. cb += inc;
  483. }
  484. }
  485. static uint16_t roundToInt16(int64_t f)
  486. {
  487. int r= (f + (1<<15))>>16;
  488. if (r<-0x7FFF) return 0x8000;
  489. else if (r> 0x7FFF) return 0x7FFF;
  490. else return r;
  491. }
  492. av_cold int ff_yuv2rgb_c_init_tables(SwsContext *c, const int inv_table[4], int fullRange,
  493. int brightness, int contrast, int saturation)
  494. {
  495. const int isRgb = c->dstFormat==PIX_FMT_RGB32
  496. || c->dstFormat==PIX_FMT_RGB32_1
  497. || c->dstFormat==PIX_FMT_BGR24
  498. || c->dstFormat==PIX_FMT_RGB565BE
  499. || c->dstFormat==PIX_FMT_RGB565LE
  500. || c->dstFormat==PIX_FMT_RGB555BE
  501. || c->dstFormat==PIX_FMT_RGB555LE
  502. || c->dstFormat==PIX_FMT_RGB444BE
  503. || c->dstFormat==PIX_FMT_RGB444LE
  504. || c->dstFormat==PIX_FMT_RGB8
  505. || c->dstFormat==PIX_FMT_RGB4
  506. || c->dstFormat==PIX_FMT_RGB4_BYTE
  507. || c->dstFormat==PIX_FMT_MONOBLACK;
  508. const int isNotNe = c->dstFormat==PIX_FMT_NE(RGB565LE,RGB565BE)
  509. || c->dstFormat==PIX_FMT_NE(RGB555LE,RGB555BE)
  510. || c->dstFormat==PIX_FMT_NE(RGB444LE,RGB444BE)
  511. || c->dstFormat==PIX_FMT_NE(BGR565LE,BGR565BE)
  512. || c->dstFormat==PIX_FMT_NE(BGR555LE,BGR555BE)
  513. || c->dstFormat==PIX_FMT_NE(BGR444LE,BGR444BE);
  514. const int bpp = c->dstFormatBpp;
  515. uint8_t *y_table;
  516. uint16_t *y_table16;
  517. uint32_t *y_table32;
  518. int i, base, rbase, gbase, bbase, abase, needAlpha;
  519. const int yoffs = fullRange ? 384 : 326;
  520. int64_t crv = inv_table[0];
  521. int64_t cbu = inv_table[1];
  522. int64_t cgu = -inv_table[2];
  523. int64_t cgv = -inv_table[3];
  524. int64_t cy = 1<<16;
  525. int64_t oy = 0;
  526. int64_t yb = 0;
  527. if (!fullRange) {
  528. cy = (cy*255) / 219;
  529. oy = 16<<16;
  530. } else {
  531. crv = (crv*224) / 255;
  532. cbu = (cbu*224) / 255;
  533. cgu = (cgu*224) / 255;
  534. cgv = (cgv*224) / 255;
  535. }
  536. cy = (cy *contrast ) >> 16;
  537. crv = (crv*contrast * saturation) >> 32;
  538. cbu = (cbu*contrast * saturation) >> 32;
  539. cgu = (cgu*contrast * saturation) >> 32;
  540. cgv = (cgv*contrast * saturation) >> 32;
  541. oy -= 256*brightness;
  542. c->uOffset= 0x0400040004000400LL;
  543. c->vOffset= 0x0400040004000400LL;
  544. c->yCoeff= roundToInt16(cy *8192) * 0x0001000100010001ULL;
  545. c->vrCoeff= roundToInt16(crv*8192) * 0x0001000100010001ULL;
  546. c->ubCoeff= roundToInt16(cbu*8192) * 0x0001000100010001ULL;
  547. c->vgCoeff= roundToInt16(cgv*8192) * 0x0001000100010001ULL;
  548. c->ugCoeff= roundToInt16(cgu*8192) * 0x0001000100010001ULL;
  549. c->yOffset= roundToInt16(oy * 8) * 0x0001000100010001ULL;
  550. c->yuv2rgb_y_coeff = (int16_t)roundToInt16(cy <<13);
  551. c->yuv2rgb_y_offset = (int16_t)roundToInt16(oy << 9);
  552. c->yuv2rgb_v2r_coeff= (int16_t)roundToInt16(crv<<13);
  553. c->yuv2rgb_v2g_coeff= (int16_t)roundToInt16(cgv<<13);
  554. c->yuv2rgb_u2g_coeff= (int16_t)roundToInt16(cgu<<13);
  555. c->yuv2rgb_u2b_coeff= (int16_t)roundToInt16(cbu<<13);
  556. //scale coefficients by cy
  557. crv = ((crv << 16) + 0x8000) / cy;
  558. cbu = ((cbu << 16) + 0x8000) / cy;
  559. cgu = ((cgu << 16) + 0x8000) / cy;
  560. cgv = ((cgv << 16) + 0x8000) / cy;
  561. av_free(c->yuvTable);
  562. switch (bpp) {
  563. case 1:
  564. c->yuvTable = av_malloc(1024);
  565. y_table = c->yuvTable;
  566. yb = -(384<<16) - oy;
  567. for (i = 0; i < 1024-110; i++) {
  568. y_table[i+110] = av_clip_uint8((yb + 0x8000) >> 16) >> 7;
  569. yb += cy;
  570. }
  571. fill_table(c->table_gU, 1, cgu, y_table + yoffs);
  572. fill_gv_table(c->table_gV, 1, cgv);
  573. break;
  574. case 4:
  575. case 4|128:
  576. rbase = isRgb ? 3 : 0;
  577. gbase = 1;
  578. bbase = isRgb ? 0 : 3;
  579. c->yuvTable = av_malloc(1024*3);
  580. y_table = c->yuvTable;
  581. yb = -(384<<16) - oy;
  582. for (i = 0; i < 1024-110; i++) {
  583. int yval = av_clip_uint8((yb + 0x8000) >> 16);
  584. y_table[i+110 ] = (yval >> 7) << rbase;
  585. y_table[i+ 37+1024] = ((yval + 43) / 85) << gbase;
  586. y_table[i+110+2048] = (yval >> 7) << bbase;
  587. yb += cy;
  588. }
  589. fill_table(c->table_rV, 1, crv, y_table + yoffs);
  590. fill_table(c->table_gU, 1, cgu, y_table + yoffs + 1024);
  591. fill_table(c->table_bU, 1, cbu, y_table + yoffs + 2048);
  592. fill_gv_table(c->table_gV, 1, cgv);
  593. break;
  594. case 8:
  595. rbase = isRgb ? 5 : 0;
  596. gbase = isRgb ? 2 : 3;
  597. bbase = isRgb ? 0 : 6;
  598. c->yuvTable = av_malloc(1024*3);
  599. y_table = c->yuvTable;
  600. yb = -(384<<16) - oy;
  601. for (i = 0; i < 1024-38; i++) {
  602. int yval = av_clip_uint8((yb + 0x8000) >> 16);
  603. y_table[i+16 ] = ((yval + 18) / 36) << rbase;
  604. y_table[i+16+1024] = ((yval + 18) / 36) << gbase;
  605. y_table[i+37+2048] = ((yval + 43) / 85) << bbase;
  606. yb += cy;
  607. }
  608. fill_table(c->table_rV, 1, crv, y_table + yoffs);
  609. fill_table(c->table_gU, 1, cgu, y_table + yoffs + 1024);
  610. fill_table(c->table_bU, 1, cbu, y_table + yoffs + 2048);
  611. fill_gv_table(c->table_gV, 1, cgv);
  612. break;
  613. case 12:
  614. rbase = isRgb ? 8 : 0;
  615. gbase = 4;
  616. bbase = isRgb ? 0 : 8;
  617. c->yuvTable = av_malloc(1024*3*2);
  618. y_table16 = c->yuvTable;
  619. yb = -(384<<16) - oy;
  620. for (i = 0; i < 1024; i++) {
  621. uint8_t yval = av_clip_uint8((yb + 0x8000) >> 16);
  622. y_table16[i ] = (yval >> 4) << rbase;
  623. y_table16[i+1024] = (yval >> 4) << gbase;
  624. y_table16[i+2048] = (yval >> 4) << bbase;
  625. yb += cy;
  626. }
  627. if (isNotNe)
  628. for (i = 0; i < 1024*3; i++)
  629. y_table16[i] = av_bswap16(y_table16[i]);
  630. fill_table(c->table_rV, 2, crv, y_table16 + yoffs);
  631. fill_table(c->table_gU, 2, cgu, y_table16 + yoffs + 1024);
  632. fill_table(c->table_bU, 2, cbu, y_table16 + yoffs + 2048);
  633. fill_gv_table(c->table_gV, 2, cgv);
  634. break;
  635. case 15:
  636. case 16:
  637. rbase = isRgb ? bpp - 5 : 0;
  638. gbase = 5;
  639. bbase = isRgb ? 0 : (bpp - 5);
  640. c->yuvTable = av_malloc(1024*3*2);
  641. y_table16 = c->yuvTable;
  642. yb = -(384<<16) - oy;
  643. for (i = 0; i < 1024; i++) {
  644. uint8_t yval = av_clip_uint8((yb + 0x8000) >> 16);
  645. y_table16[i ] = (yval >> 3) << rbase;
  646. y_table16[i+1024] = (yval >> (18 - bpp)) << gbase;
  647. y_table16[i+2048] = (yval >> 3) << bbase;
  648. yb += cy;
  649. }
  650. if(isNotNe)
  651. for (i = 0; i < 1024*3; i++)
  652. y_table16[i] = av_bswap16(y_table16[i]);
  653. fill_table(c->table_rV, 2, crv, y_table16 + yoffs);
  654. fill_table(c->table_gU, 2, cgu, y_table16 + yoffs + 1024);
  655. fill_table(c->table_bU, 2, cbu, y_table16 + yoffs + 2048);
  656. fill_gv_table(c->table_gV, 2, cgv);
  657. break;
  658. case 24:
  659. case 48:
  660. c->yuvTable = av_malloc(1024);
  661. y_table = c->yuvTable;
  662. yb = -(384<<16) - oy;
  663. for (i = 0; i < 1024; i++) {
  664. y_table[i] = av_clip_uint8((yb + 0x8000) >> 16);
  665. yb += cy;
  666. }
  667. fill_table(c->table_rV, 1, crv, y_table + yoffs);
  668. fill_table(c->table_gU, 1, cgu, y_table + yoffs);
  669. fill_table(c->table_bU, 1, cbu, y_table + yoffs);
  670. fill_gv_table(c->table_gV, 1, cgv);
  671. break;
  672. case 32:
  673. base = (c->dstFormat == PIX_FMT_RGB32_1 || c->dstFormat == PIX_FMT_BGR32_1) ? 8 : 0;
  674. rbase = base + (isRgb ? 16 : 0);
  675. gbase = base + 8;
  676. bbase = base + (isRgb ? 0 : 16);
  677. needAlpha = CONFIG_SWSCALE_ALPHA && isALPHA(c->srcFormat);
  678. if (!needAlpha)
  679. abase = (base + 24) & 31;
  680. c->yuvTable = av_malloc(1024*3*4);
  681. y_table32 = c->yuvTable;
  682. yb = -(384<<16) - oy;
  683. for (i = 0; i < 1024; i++) {
  684. unsigned yval = av_clip_uint8((yb + 0x8000) >> 16);
  685. y_table32[i ] = (yval << rbase) + (needAlpha ? 0 : (255u << abase));
  686. y_table32[i+1024] = yval << gbase;
  687. y_table32[i+2048] = yval << bbase;
  688. yb += cy;
  689. }
  690. fill_table(c->table_rV, 4, crv, y_table32 + yoffs);
  691. fill_table(c->table_gU, 4, cgu, y_table32 + yoffs + 1024);
  692. fill_table(c->table_bU, 4, cbu, y_table32 + yoffs + 2048);
  693. fill_gv_table(c->table_gV, 4, cgv);
  694. break;
  695. default:
  696. c->yuvTable = NULL;
  697. av_log(c, AV_LOG_ERROR, "%ibpp not supported by yuv2rgb\n", bpp);
  698. return -1;
  699. }
  700. return 0;
  701. }