2
0

convert_from.cc 47 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505
  1. /*
  2. * Copyright 2012 The LibYuv Project Authors. All rights reserved.
  3. *
  4. * Use of this source code is governed by a BSD-style license
  5. * that can be found in the LICENSE file in the root of the source
  6. * tree. An additional intellectual property rights grant can be found
  7. * in the file PATENTS. All contributing project authors may
  8. * be found in the AUTHORS file in the root of the source tree.
  9. */
  10. #include "libyuv/convert_from.h"
  11. #include "libyuv/basic_types.h"
  12. #include "libyuv/convert.h" // For I420Copy
  13. #include "libyuv/cpu_id.h"
  14. #include "libyuv/planar_functions.h"
  15. #include "libyuv/rotate.h"
  16. #include "libyuv/row.h"
  17. #include "libyuv/scale.h" // For ScalePlane()
  18. #include "libyuv/video_common.h"
  19. #ifdef __cplusplus
  20. namespace libyuv {
  21. extern "C" {
  22. #endif
  23. #define SUBSAMPLE(v, a, s) (v < 0) ? (-((-v + a) >> s)) : ((v + a) >> s)
  24. static __inline int Abs(int v) {
  25. return v >= 0 ? v : -v;
  26. }
  27. // I420 To any I4xx YUV format with mirroring.
  28. static int I420ToI4xx(const uint8_t* src_y,
  29. int src_stride_y,
  30. const uint8_t* src_u,
  31. int src_stride_u,
  32. const uint8_t* src_v,
  33. int src_stride_v,
  34. uint8_t* dst_y,
  35. int dst_stride_y,
  36. uint8_t* dst_u,
  37. int dst_stride_u,
  38. uint8_t* dst_v,
  39. int dst_stride_v,
  40. int src_y_width,
  41. int src_y_height,
  42. int dst_uv_width,
  43. int dst_uv_height) {
  44. const int dst_y_width = Abs(src_y_width);
  45. const int dst_y_height = Abs(src_y_height);
  46. const int src_uv_width = SUBSAMPLE(src_y_width, 1, 1);
  47. const int src_uv_height = SUBSAMPLE(src_y_height, 1, 1);
  48. if (src_y_width == 0 || src_y_height == 0 || dst_uv_width <= 0 ||
  49. dst_uv_height <= 0) {
  50. return -1;
  51. }
  52. if (dst_y) {
  53. ScalePlane(src_y, src_stride_y, src_y_width, src_y_height, dst_y,
  54. dst_stride_y, dst_y_width, dst_y_height, kFilterBilinear);
  55. }
  56. ScalePlane(src_u, src_stride_u, src_uv_width, src_uv_height, dst_u,
  57. dst_stride_u, dst_uv_width, dst_uv_height, kFilterBilinear);
  58. ScalePlane(src_v, src_stride_v, src_uv_width, src_uv_height, dst_v,
  59. dst_stride_v, dst_uv_width, dst_uv_height, kFilterBilinear);
  60. return 0;
  61. }
  62. // Convert 8 bit YUV to 10 bit.
  63. LIBYUV_API
  64. int I420ToI010(const uint8_t* src_y,
  65. int src_stride_y,
  66. const uint8_t* src_u,
  67. int src_stride_u,
  68. const uint8_t* src_v,
  69. int src_stride_v,
  70. uint16_t* dst_y,
  71. int dst_stride_y,
  72. uint16_t* dst_u,
  73. int dst_stride_u,
  74. uint16_t* dst_v,
  75. int dst_stride_v,
  76. int width,
  77. int height) {
  78. int halfwidth = (width + 1) >> 1;
  79. int halfheight = (height + 1) >> 1;
  80. if (!src_u || !src_v || !dst_u || !dst_v || width <= 0 || height == 0) {
  81. return -1;
  82. }
  83. // Negative height means invert the image.
  84. if (height < 0) {
  85. height = -height;
  86. halfheight = (height + 1) >> 1;
  87. src_y = src_y + (height - 1) * src_stride_y;
  88. src_u = src_u + (halfheight - 1) * src_stride_u;
  89. src_v = src_v + (halfheight - 1) * src_stride_v;
  90. src_stride_y = -src_stride_y;
  91. src_stride_u = -src_stride_u;
  92. src_stride_v = -src_stride_v;
  93. }
  94. // Convert Y plane.
  95. Convert8To16Plane(src_y, src_stride_y, dst_y, dst_stride_y, 1024, width,
  96. height);
  97. // Convert UV planes.
  98. Convert8To16Plane(src_u, src_stride_u, dst_u, dst_stride_u, 1024, halfwidth,
  99. halfheight);
  100. Convert8To16Plane(src_v, src_stride_v, dst_v, dst_stride_v, 1024, halfwidth,
  101. halfheight);
  102. return 0;
  103. }
  104. // 420 chroma is 1/2 width, 1/2 height
  105. // 422 chroma is 1/2 width, 1x height
  106. LIBYUV_API
  107. int I420ToI422(const uint8_t* src_y,
  108. int src_stride_y,
  109. const uint8_t* src_u,
  110. int src_stride_u,
  111. const uint8_t* src_v,
  112. int src_stride_v,
  113. uint8_t* dst_y,
  114. int dst_stride_y,
  115. uint8_t* dst_u,
  116. int dst_stride_u,
  117. uint8_t* dst_v,
  118. int dst_stride_v,
  119. int width,
  120. int height) {
  121. const int dst_uv_width = (Abs(width) + 1) >> 1;
  122. const int dst_uv_height = Abs(height);
  123. return I420ToI4xx(src_y, src_stride_y, src_u, src_stride_u, src_v,
  124. src_stride_v, dst_y, dst_stride_y, dst_u, dst_stride_u,
  125. dst_v, dst_stride_v, width, height, dst_uv_width,
  126. dst_uv_height);
  127. }
  128. // 420 chroma is 1/2 width, 1/2 height
  129. // 444 chroma is 1x width, 1x height
  130. LIBYUV_API
  131. int I420ToI444(const uint8_t* src_y,
  132. int src_stride_y,
  133. const uint8_t* src_u,
  134. int src_stride_u,
  135. const uint8_t* src_v,
  136. int src_stride_v,
  137. uint8_t* dst_y,
  138. int dst_stride_y,
  139. uint8_t* dst_u,
  140. int dst_stride_u,
  141. uint8_t* dst_v,
  142. int dst_stride_v,
  143. int width,
  144. int height) {
  145. const int dst_uv_width = Abs(width);
  146. const int dst_uv_height = Abs(height);
  147. return I420ToI4xx(src_y, src_stride_y, src_u, src_stride_u, src_v,
  148. src_stride_v, dst_y, dst_stride_y, dst_u, dst_stride_u,
  149. dst_v, dst_stride_v, width, height, dst_uv_width,
  150. dst_uv_height);
  151. }
  152. // Copy to I400. Source can be I420,422,444,400,NV12,NV21
  153. LIBYUV_API
  154. int I400Copy(const uint8_t* src_y,
  155. int src_stride_y,
  156. uint8_t* dst_y,
  157. int dst_stride_y,
  158. int width,
  159. int height) {
  160. if (!src_y || !dst_y || width <= 0 || height == 0) {
  161. return -1;
  162. }
  163. // Negative height means invert the image.
  164. if (height < 0) {
  165. height = -height;
  166. src_y = src_y + (height - 1) * src_stride_y;
  167. src_stride_y = -src_stride_y;
  168. }
  169. CopyPlane(src_y, src_stride_y, dst_y, dst_stride_y, width, height);
  170. return 0;
  171. }
  172. LIBYUV_API
  173. int I422ToYUY2(const uint8_t* src_y,
  174. int src_stride_y,
  175. const uint8_t* src_u,
  176. int src_stride_u,
  177. const uint8_t* src_v,
  178. int src_stride_v,
  179. uint8_t* dst_yuy2,
  180. int dst_stride_yuy2,
  181. int width,
  182. int height) {
  183. int y;
  184. void (*I422ToYUY2Row)(const uint8_t* src_y, const uint8_t* src_u,
  185. const uint8_t* src_v, uint8_t* dst_yuy2, int width) =
  186. I422ToYUY2Row_C;
  187. if (!src_y || !src_u || !src_v || !dst_yuy2 || width <= 0 || height == 0) {
  188. return -1;
  189. }
  190. // Negative height means invert the image.
  191. if (height < 0) {
  192. height = -height;
  193. dst_yuy2 = dst_yuy2 + (height - 1) * dst_stride_yuy2;
  194. dst_stride_yuy2 = -dst_stride_yuy2;
  195. }
  196. // Coalesce rows.
  197. if (src_stride_y == width && src_stride_u * 2 == width &&
  198. src_stride_v * 2 == width && dst_stride_yuy2 == width * 2) {
  199. width *= height;
  200. height = 1;
  201. src_stride_y = src_stride_u = src_stride_v = dst_stride_yuy2 = 0;
  202. }
  203. #if defined(HAS_I422TOYUY2ROW_SSE2)
  204. if (TestCpuFlag(kCpuHasSSE2)) {
  205. I422ToYUY2Row = I422ToYUY2Row_Any_SSE2;
  206. if (IS_ALIGNED(width, 16)) {
  207. I422ToYUY2Row = I422ToYUY2Row_SSE2;
  208. }
  209. }
  210. #endif
  211. #if defined(HAS_I422TOYUY2ROW_AVX2)
  212. if (TestCpuFlag(kCpuHasAVX2)) {
  213. I422ToYUY2Row = I422ToYUY2Row_Any_AVX2;
  214. if (IS_ALIGNED(width, 32)) {
  215. I422ToYUY2Row = I422ToYUY2Row_AVX2;
  216. }
  217. }
  218. #endif
  219. #if defined(HAS_I422TOYUY2ROW_NEON)
  220. if (TestCpuFlag(kCpuHasNEON)) {
  221. I422ToYUY2Row = I422ToYUY2Row_Any_NEON;
  222. if (IS_ALIGNED(width, 16)) {
  223. I422ToYUY2Row = I422ToYUY2Row_NEON;
  224. }
  225. }
  226. #endif
  227. for (y = 0; y < height; ++y) {
  228. I422ToYUY2Row(src_y, src_u, src_v, dst_yuy2, width);
  229. src_y += src_stride_y;
  230. src_u += src_stride_u;
  231. src_v += src_stride_v;
  232. dst_yuy2 += dst_stride_yuy2;
  233. }
  234. return 0;
  235. }
  236. LIBYUV_API
  237. int I420ToYUY2(const uint8_t* src_y,
  238. int src_stride_y,
  239. const uint8_t* src_u,
  240. int src_stride_u,
  241. const uint8_t* src_v,
  242. int src_stride_v,
  243. uint8_t* dst_yuy2,
  244. int dst_stride_yuy2,
  245. int width,
  246. int height) {
  247. int y;
  248. void (*I422ToYUY2Row)(const uint8_t* src_y, const uint8_t* src_u,
  249. const uint8_t* src_v, uint8_t* dst_yuy2, int width) =
  250. I422ToYUY2Row_C;
  251. if (!src_y || !src_u || !src_v || !dst_yuy2 || width <= 0 || height == 0) {
  252. return -1;
  253. }
  254. // Negative height means invert the image.
  255. if (height < 0) {
  256. height = -height;
  257. dst_yuy2 = dst_yuy2 + (height - 1) * dst_stride_yuy2;
  258. dst_stride_yuy2 = -dst_stride_yuy2;
  259. }
  260. #if defined(HAS_I422TOYUY2ROW_SSE2)
  261. if (TestCpuFlag(kCpuHasSSE2)) {
  262. I422ToYUY2Row = I422ToYUY2Row_Any_SSE2;
  263. if (IS_ALIGNED(width, 16)) {
  264. I422ToYUY2Row = I422ToYUY2Row_SSE2;
  265. }
  266. }
  267. #endif
  268. #if defined(HAS_I422TOYUY2ROW_AVX2)
  269. if (TestCpuFlag(kCpuHasAVX2)) {
  270. I422ToYUY2Row = I422ToYUY2Row_Any_AVX2;
  271. if (IS_ALIGNED(width, 32)) {
  272. I422ToYUY2Row = I422ToYUY2Row_AVX2;
  273. }
  274. }
  275. #endif
  276. #if defined(HAS_I422TOYUY2ROW_NEON)
  277. if (TestCpuFlag(kCpuHasNEON)) {
  278. I422ToYUY2Row = I422ToYUY2Row_Any_NEON;
  279. if (IS_ALIGNED(width, 16)) {
  280. I422ToYUY2Row = I422ToYUY2Row_NEON;
  281. }
  282. }
  283. #endif
  284. #if defined(HAS_I422TOYUY2ROW_MSA)
  285. if (TestCpuFlag(kCpuHasMSA)) {
  286. I422ToYUY2Row = I422ToYUY2Row_Any_MSA;
  287. if (IS_ALIGNED(width, 32)) {
  288. I422ToYUY2Row = I422ToYUY2Row_MSA;
  289. }
  290. }
  291. #endif
  292. #if defined(HAS_I422TOYUY2ROW_MMI)
  293. if (TestCpuFlag(kCpuHasMMI)) {
  294. I422ToYUY2Row = I422ToYUY2Row_Any_MMI;
  295. if (IS_ALIGNED(width, 8)) {
  296. I422ToYUY2Row = I422ToYUY2Row_MMI;
  297. }
  298. }
  299. #endif
  300. for (y = 0; y < height - 1; y += 2) {
  301. I422ToYUY2Row(src_y, src_u, src_v, dst_yuy2, width);
  302. I422ToYUY2Row(src_y + src_stride_y, src_u, src_v,
  303. dst_yuy2 + dst_stride_yuy2, width);
  304. src_y += src_stride_y * 2;
  305. src_u += src_stride_u;
  306. src_v += src_stride_v;
  307. dst_yuy2 += dst_stride_yuy2 * 2;
  308. }
  309. if (height & 1) {
  310. I422ToYUY2Row(src_y, src_u, src_v, dst_yuy2, width);
  311. }
  312. return 0;
  313. }
  314. LIBYUV_API
  315. int I422ToUYVY(const uint8_t* src_y,
  316. int src_stride_y,
  317. const uint8_t* src_u,
  318. int src_stride_u,
  319. const uint8_t* src_v,
  320. int src_stride_v,
  321. uint8_t* dst_uyvy,
  322. int dst_stride_uyvy,
  323. int width,
  324. int height) {
  325. int y;
  326. void (*I422ToUYVYRow)(const uint8_t* src_y, const uint8_t* src_u,
  327. const uint8_t* src_v, uint8_t* dst_uyvy, int width) =
  328. I422ToUYVYRow_C;
  329. if (!src_y || !src_u || !src_v || !dst_uyvy || width <= 0 || height == 0) {
  330. return -1;
  331. }
  332. // Negative height means invert the image.
  333. if (height < 0) {
  334. height = -height;
  335. dst_uyvy = dst_uyvy + (height - 1) * dst_stride_uyvy;
  336. dst_stride_uyvy = -dst_stride_uyvy;
  337. }
  338. // Coalesce rows.
  339. if (src_stride_y == width && src_stride_u * 2 == width &&
  340. src_stride_v * 2 == width && dst_stride_uyvy == width * 2) {
  341. width *= height;
  342. height = 1;
  343. src_stride_y = src_stride_u = src_stride_v = dst_stride_uyvy = 0;
  344. }
  345. #if defined(HAS_I422TOUYVYROW_SSE2)
  346. if (TestCpuFlag(kCpuHasSSE2)) {
  347. I422ToUYVYRow = I422ToUYVYRow_Any_SSE2;
  348. if (IS_ALIGNED(width, 16)) {
  349. I422ToUYVYRow = I422ToUYVYRow_SSE2;
  350. }
  351. }
  352. #endif
  353. #if defined(HAS_I422TOUYVYROW_AVX2)
  354. if (TestCpuFlag(kCpuHasAVX2)) {
  355. I422ToUYVYRow = I422ToUYVYRow_Any_AVX2;
  356. if (IS_ALIGNED(width, 32)) {
  357. I422ToUYVYRow = I422ToUYVYRow_AVX2;
  358. }
  359. }
  360. #endif
  361. #if defined(HAS_I422TOUYVYROW_NEON)
  362. if (TestCpuFlag(kCpuHasNEON)) {
  363. I422ToUYVYRow = I422ToUYVYRow_Any_NEON;
  364. if (IS_ALIGNED(width, 16)) {
  365. I422ToUYVYRow = I422ToUYVYRow_NEON;
  366. }
  367. }
  368. #endif
  369. #if defined(HAS_I422TOUYVYROW_MSA)
  370. if (TestCpuFlag(kCpuHasMSA)) {
  371. I422ToUYVYRow = I422ToUYVYRow_Any_MSA;
  372. if (IS_ALIGNED(width, 32)) {
  373. I422ToUYVYRow = I422ToUYVYRow_MSA;
  374. }
  375. }
  376. #endif
  377. #if defined(HAS_I422TOUYVYROW_MMI)
  378. if (TestCpuFlag(kCpuHasMMI)) {
  379. I422ToUYVYRow = I422ToUYVYRow_Any_MMI;
  380. if (IS_ALIGNED(width, 8)) {
  381. I422ToUYVYRow = I422ToUYVYRow_MMI;
  382. }
  383. }
  384. #endif
  385. for (y = 0; y < height; ++y) {
  386. I422ToUYVYRow(src_y, src_u, src_v, dst_uyvy, width);
  387. src_y += src_stride_y;
  388. src_u += src_stride_u;
  389. src_v += src_stride_v;
  390. dst_uyvy += dst_stride_uyvy;
  391. }
  392. return 0;
  393. }
  394. LIBYUV_API
  395. int I420ToUYVY(const uint8_t* src_y,
  396. int src_stride_y,
  397. const uint8_t* src_u,
  398. int src_stride_u,
  399. const uint8_t* src_v,
  400. int src_stride_v,
  401. uint8_t* dst_uyvy,
  402. int dst_stride_uyvy,
  403. int width,
  404. int height) {
  405. int y;
  406. void (*I422ToUYVYRow)(const uint8_t* src_y, const uint8_t* src_u,
  407. const uint8_t* src_v, uint8_t* dst_uyvy, int width) =
  408. I422ToUYVYRow_C;
  409. if (!src_y || !src_u || !src_v || !dst_uyvy || width <= 0 || height == 0) {
  410. return -1;
  411. }
  412. // Negative height means invert the image.
  413. if (height < 0) {
  414. height = -height;
  415. dst_uyvy = dst_uyvy + (height - 1) * dst_stride_uyvy;
  416. dst_stride_uyvy = -dst_stride_uyvy;
  417. }
  418. #if defined(HAS_I422TOUYVYROW_SSE2)
  419. if (TestCpuFlag(kCpuHasSSE2)) {
  420. I422ToUYVYRow = I422ToUYVYRow_Any_SSE2;
  421. if (IS_ALIGNED(width, 16)) {
  422. I422ToUYVYRow = I422ToUYVYRow_SSE2;
  423. }
  424. }
  425. #endif
  426. #if defined(HAS_I422TOUYVYROW_AVX2)
  427. if (TestCpuFlag(kCpuHasAVX2)) {
  428. I422ToUYVYRow = I422ToUYVYRow_Any_AVX2;
  429. if (IS_ALIGNED(width, 32)) {
  430. I422ToUYVYRow = I422ToUYVYRow_AVX2;
  431. }
  432. }
  433. #endif
  434. #if defined(HAS_I422TOUYVYROW_NEON)
  435. if (TestCpuFlag(kCpuHasNEON)) {
  436. I422ToUYVYRow = I422ToUYVYRow_Any_NEON;
  437. if (IS_ALIGNED(width, 16)) {
  438. I422ToUYVYRow = I422ToUYVYRow_NEON;
  439. }
  440. }
  441. #endif
  442. #if defined(HAS_I422TOUYVYROW_MSA)
  443. if (TestCpuFlag(kCpuHasMSA)) {
  444. I422ToUYVYRow = I422ToUYVYRow_Any_MSA;
  445. if (IS_ALIGNED(width, 32)) {
  446. I422ToUYVYRow = I422ToUYVYRow_MSA;
  447. }
  448. }
  449. #endif
  450. #if defined(HAS_I422TOUYVYROW_MMI)
  451. if (TestCpuFlag(kCpuHasMMI)) {
  452. I422ToUYVYRow = I422ToUYVYRow_Any_MMI;
  453. if (IS_ALIGNED(width, 8)) {
  454. I422ToUYVYRow = I422ToUYVYRow_MMI;
  455. }
  456. }
  457. #endif
  458. for (y = 0; y < height - 1; y += 2) {
  459. I422ToUYVYRow(src_y, src_u, src_v, dst_uyvy, width);
  460. I422ToUYVYRow(src_y + src_stride_y, src_u, src_v,
  461. dst_uyvy + dst_stride_uyvy, width);
  462. src_y += src_stride_y * 2;
  463. src_u += src_stride_u;
  464. src_v += src_stride_v;
  465. dst_uyvy += dst_stride_uyvy * 2;
  466. }
  467. if (height & 1) {
  468. I422ToUYVYRow(src_y, src_u, src_v, dst_uyvy, width);
  469. }
  470. return 0;
  471. }
  472. // TODO(fbarchard): test negative height for invert.
  473. LIBYUV_API
  474. int I420ToNV12(const uint8_t* src_y,
  475. int src_stride_y,
  476. const uint8_t* src_u,
  477. int src_stride_u,
  478. const uint8_t* src_v,
  479. int src_stride_v,
  480. uint8_t* dst_y,
  481. int dst_stride_y,
  482. uint8_t* dst_uv,
  483. int dst_stride_uv,
  484. int width,
  485. int height) {
  486. if (!src_y || !src_u || !src_v || !dst_y || !dst_uv || width <= 0 ||
  487. height == 0) {
  488. return -1;
  489. }
  490. int halfwidth = (width + 1) / 2;
  491. int halfheight = height > 0 ? (height + 1) / 2 : (height - 1) / 2;
  492. if (dst_y) {
  493. CopyPlane(src_y, src_stride_y, dst_y, dst_stride_y, width, height);
  494. }
  495. MergeUVPlane(src_u, src_stride_u, src_v, src_stride_v, dst_uv, dst_stride_uv,
  496. halfwidth, halfheight);
  497. return 0;
  498. }
  499. LIBYUV_API
  500. int I420ToNV21(const uint8_t* src_y,
  501. int src_stride_y,
  502. const uint8_t* src_u,
  503. int src_stride_u,
  504. const uint8_t* src_v,
  505. int src_stride_v,
  506. uint8_t* dst_y,
  507. int dst_stride_y,
  508. uint8_t* dst_vu,
  509. int dst_stride_vu,
  510. int width,
  511. int height) {
  512. return I420ToNV12(src_y, src_stride_y, src_v, src_stride_v, src_u,
  513. src_stride_u, dst_y, dst_stride_y, dst_vu, dst_stride_vu,
  514. width, height);
  515. }
  516. // Convert I422 to RGBA with matrix
  517. static int I420ToRGBAMatrix(const uint8_t* src_y,
  518. int src_stride_y,
  519. const uint8_t* src_u,
  520. int src_stride_u,
  521. const uint8_t* src_v,
  522. int src_stride_v,
  523. uint8_t* dst_rgba,
  524. int dst_stride_rgba,
  525. const struct YuvConstants* yuvconstants,
  526. int width,
  527. int height) {
  528. int y;
  529. void (*I422ToRGBARow)(const uint8_t* y_buf, const uint8_t* u_buf,
  530. const uint8_t* v_buf, uint8_t* rgb_buf,
  531. const struct YuvConstants* yuvconstants, int width) =
  532. I422ToRGBARow_C;
  533. if (!src_y || !src_u || !src_v || !dst_rgba || width <= 0 || height == 0) {
  534. return -1;
  535. }
  536. // Negative height means invert the image.
  537. if (height < 0) {
  538. height = -height;
  539. dst_rgba = dst_rgba + (height - 1) * dst_stride_rgba;
  540. dst_stride_rgba = -dst_stride_rgba;
  541. }
  542. #if defined(HAS_I422TORGBAROW_SSSE3)
  543. if (TestCpuFlag(kCpuHasSSSE3)) {
  544. I422ToRGBARow = I422ToRGBARow_Any_SSSE3;
  545. if (IS_ALIGNED(width, 8)) {
  546. I422ToRGBARow = I422ToRGBARow_SSSE3;
  547. }
  548. }
  549. #endif
  550. #if defined(HAS_I422TORGBAROW_AVX2)
  551. if (TestCpuFlag(kCpuHasAVX2)) {
  552. I422ToRGBARow = I422ToRGBARow_Any_AVX2;
  553. if (IS_ALIGNED(width, 16)) {
  554. I422ToRGBARow = I422ToRGBARow_AVX2;
  555. }
  556. }
  557. #endif
  558. #if defined(HAS_I422TORGBAROW_NEON)
  559. if (TestCpuFlag(kCpuHasNEON)) {
  560. I422ToRGBARow = I422ToRGBARow_Any_NEON;
  561. if (IS_ALIGNED(width, 8)) {
  562. I422ToRGBARow = I422ToRGBARow_NEON;
  563. }
  564. }
  565. #endif
  566. #if defined(HAS_I422TORGBAROW_MSA)
  567. if (TestCpuFlag(kCpuHasMSA)) {
  568. I422ToRGBARow = I422ToRGBARow_Any_MSA;
  569. if (IS_ALIGNED(width, 8)) {
  570. I422ToRGBARow = I422ToRGBARow_MSA;
  571. }
  572. }
  573. #endif
  574. for (y = 0; y < height; ++y) {
  575. I422ToRGBARow(src_y, src_u, src_v, dst_rgba, yuvconstants, width);
  576. dst_rgba += dst_stride_rgba;
  577. src_y += src_stride_y;
  578. if (y & 1) {
  579. src_u += src_stride_u;
  580. src_v += src_stride_v;
  581. }
  582. }
  583. return 0;
  584. }
  585. // Convert I420 to RGBA.
  586. LIBYUV_API
  587. int I420ToRGBA(const uint8_t* src_y,
  588. int src_stride_y,
  589. const uint8_t* src_u,
  590. int src_stride_u,
  591. const uint8_t* src_v,
  592. int src_stride_v,
  593. uint8_t* dst_rgba,
  594. int dst_stride_rgba,
  595. int width,
  596. int height) {
  597. return I420ToRGBAMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  598. src_stride_v, dst_rgba, dst_stride_rgba,
  599. &kYuvI601Constants, width, height);
  600. }
  601. // Convert I420 to BGRA.
  602. LIBYUV_API
  603. int I420ToBGRA(const uint8_t* src_y,
  604. int src_stride_y,
  605. const uint8_t* src_u,
  606. int src_stride_u,
  607. const uint8_t* src_v,
  608. int src_stride_v,
  609. uint8_t* dst_bgra,
  610. int dst_stride_bgra,
  611. int width,
  612. int height) {
  613. return I420ToRGBAMatrix(src_y, src_stride_y, src_v,
  614. src_stride_v, // Swap U and V
  615. src_u, src_stride_u, dst_bgra, dst_stride_bgra,
  616. &kYvuI601Constants, // Use Yvu matrix
  617. width, height);
  618. }
  619. // Convert I420 to RGB24 with matrix
  620. static int I420ToRGB24Matrix(const uint8_t* src_y,
  621. int src_stride_y,
  622. const uint8_t* src_u,
  623. int src_stride_u,
  624. const uint8_t* src_v,
  625. int src_stride_v,
  626. uint8_t* dst_rgb24,
  627. int dst_stride_rgb24,
  628. const struct YuvConstants* yuvconstants,
  629. int width,
  630. int height) {
  631. int y;
  632. void (*I422ToRGB24Row)(const uint8_t* y_buf, const uint8_t* u_buf,
  633. const uint8_t* v_buf, uint8_t* rgb_buf,
  634. const struct YuvConstants* yuvconstants, int width) =
  635. I422ToRGB24Row_C;
  636. if (!src_y || !src_u || !src_v || !dst_rgb24 || width <= 0 || height == 0) {
  637. return -1;
  638. }
  639. // Negative height means invert the image.
  640. if (height < 0) {
  641. height = -height;
  642. dst_rgb24 = dst_rgb24 + (height - 1) * dst_stride_rgb24;
  643. dst_stride_rgb24 = -dst_stride_rgb24;
  644. }
  645. #if defined(HAS_I422TORGB24ROW_SSSE3)
  646. if (TestCpuFlag(kCpuHasSSSE3)) {
  647. I422ToRGB24Row = I422ToRGB24Row_Any_SSSE3;
  648. if (IS_ALIGNED(width, 16)) {
  649. I422ToRGB24Row = I422ToRGB24Row_SSSE3;
  650. }
  651. }
  652. #endif
  653. #if defined(HAS_I422TORGB24ROW_AVX2)
  654. if (TestCpuFlag(kCpuHasAVX2)) {
  655. I422ToRGB24Row = I422ToRGB24Row_Any_AVX2;
  656. if (IS_ALIGNED(width, 32)) {
  657. I422ToRGB24Row = I422ToRGB24Row_AVX2;
  658. }
  659. }
  660. #endif
  661. #if defined(HAS_I422TORGB24ROW_NEON)
  662. if (TestCpuFlag(kCpuHasNEON)) {
  663. I422ToRGB24Row = I422ToRGB24Row_Any_NEON;
  664. if (IS_ALIGNED(width, 8)) {
  665. I422ToRGB24Row = I422ToRGB24Row_NEON;
  666. }
  667. }
  668. #endif
  669. #if defined(HAS_I422TORGB24ROW_MSA)
  670. if (TestCpuFlag(kCpuHasMSA)) {
  671. I422ToRGB24Row = I422ToRGB24Row_Any_MSA;
  672. if (IS_ALIGNED(width, 16)) {
  673. I422ToRGB24Row = I422ToRGB24Row_MSA;
  674. }
  675. }
  676. #endif
  677. for (y = 0; y < height; ++y) {
  678. I422ToRGB24Row(src_y, src_u, src_v, dst_rgb24, yuvconstants, width);
  679. dst_rgb24 += dst_stride_rgb24;
  680. src_y += src_stride_y;
  681. if (y & 1) {
  682. src_u += src_stride_u;
  683. src_v += src_stride_v;
  684. }
  685. }
  686. return 0;
  687. }
  688. // Convert I420 to RGB24.
  689. LIBYUV_API
  690. int I420ToRGB24(const uint8_t* src_y,
  691. int src_stride_y,
  692. const uint8_t* src_u,
  693. int src_stride_u,
  694. const uint8_t* src_v,
  695. int src_stride_v,
  696. uint8_t* dst_rgb24,
  697. int dst_stride_rgb24,
  698. int width,
  699. int height) {
  700. return I420ToRGB24Matrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  701. src_stride_v, dst_rgb24, dst_stride_rgb24,
  702. &kYuvI601Constants, width, height);
  703. }
  704. // Convert I420 to RAW.
  705. LIBYUV_API
  706. int I420ToRAW(const uint8_t* src_y,
  707. int src_stride_y,
  708. const uint8_t* src_u,
  709. int src_stride_u,
  710. const uint8_t* src_v,
  711. int src_stride_v,
  712. uint8_t* dst_raw,
  713. int dst_stride_raw,
  714. int width,
  715. int height) {
  716. return I420ToRGB24Matrix(src_y, src_stride_y, src_v,
  717. src_stride_v, // Swap U and V
  718. src_u, src_stride_u, dst_raw, dst_stride_raw,
  719. &kYvuI601Constants, // Use Yvu matrix
  720. width, height);
  721. }
  722. // Convert H420 to RGB24.
  723. LIBYUV_API
  724. int H420ToRGB24(const uint8_t* src_y,
  725. int src_stride_y,
  726. const uint8_t* src_u,
  727. int src_stride_u,
  728. const uint8_t* src_v,
  729. int src_stride_v,
  730. uint8_t* dst_rgb24,
  731. int dst_stride_rgb24,
  732. int width,
  733. int height) {
  734. return I420ToRGB24Matrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  735. src_stride_v, dst_rgb24, dst_stride_rgb24,
  736. &kYuvH709Constants, width, height);
  737. }
  738. // Convert H420 to RAW.
  739. LIBYUV_API
  740. int H420ToRAW(const uint8_t* src_y,
  741. int src_stride_y,
  742. const uint8_t* src_u,
  743. int src_stride_u,
  744. const uint8_t* src_v,
  745. int src_stride_v,
  746. uint8_t* dst_raw,
  747. int dst_stride_raw,
  748. int width,
  749. int height) {
  750. return I420ToRGB24Matrix(src_y, src_stride_y, src_v,
  751. src_stride_v, // Swap U and V
  752. src_u, src_stride_u, dst_raw, dst_stride_raw,
  753. &kYvuH709Constants, // Use Yvu matrix
  754. width, height);
  755. }
  756. // Convert I420 to ARGB1555.
  757. LIBYUV_API
  758. int I420ToARGB1555(const uint8_t* src_y,
  759. int src_stride_y,
  760. const uint8_t* src_u,
  761. int src_stride_u,
  762. const uint8_t* src_v,
  763. int src_stride_v,
  764. uint8_t* dst_argb1555,
  765. int dst_stride_argb1555,
  766. int width,
  767. int height) {
  768. int y;
  769. void (*I422ToARGB1555Row)(const uint8_t* y_buf, const uint8_t* u_buf,
  770. const uint8_t* v_buf, uint8_t* rgb_buf,
  771. const struct YuvConstants* yuvconstants,
  772. int width) = I422ToARGB1555Row_C;
  773. if (!src_y || !src_u || !src_v || !dst_argb1555 || width <= 0 ||
  774. height == 0) {
  775. return -1;
  776. }
  777. // Negative height means invert the image.
  778. if (height < 0) {
  779. height = -height;
  780. dst_argb1555 = dst_argb1555 + (height - 1) * dst_stride_argb1555;
  781. dst_stride_argb1555 = -dst_stride_argb1555;
  782. }
  783. #if defined(HAS_I422TOARGB1555ROW_SSSE3)
  784. if (TestCpuFlag(kCpuHasSSSE3)) {
  785. I422ToARGB1555Row = I422ToARGB1555Row_Any_SSSE3;
  786. if (IS_ALIGNED(width, 8)) {
  787. I422ToARGB1555Row = I422ToARGB1555Row_SSSE3;
  788. }
  789. }
  790. #endif
  791. #if defined(HAS_I422TOARGB1555ROW_AVX2)
  792. if (TestCpuFlag(kCpuHasAVX2)) {
  793. I422ToARGB1555Row = I422ToARGB1555Row_Any_AVX2;
  794. if (IS_ALIGNED(width, 16)) {
  795. I422ToARGB1555Row = I422ToARGB1555Row_AVX2;
  796. }
  797. }
  798. #endif
  799. #if defined(HAS_I422TOARGB1555ROW_NEON)
  800. if (TestCpuFlag(kCpuHasNEON)) {
  801. I422ToARGB1555Row = I422ToARGB1555Row_Any_NEON;
  802. if (IS_ALIGNED(width, 8)) {
  803. I422ToARGB1555Row = I422ToARGB1555Row_NEON;
  804. }
  805. }
  806. #endif
  807. #if defined(HAS_I422TOARGB1555ROW_MSA)
  808. if (TestCpuFlag(kCpuHasMSA)) {
  809. I422ToARGB1555Row = I422ToARGB1555Row_Any_MSA;
  810. if (IS_ALIGNED(width, 8)) {
  811. I422ToARGB1555Row = I422ToARGB1555Row_MSA;
  812. }
  813. }
  814. #endif
  815. for (y = 0; y < height; ++y) {
  816. I422ToARGB1555Row(src_y, src_u, src_v, dst_argb1555, &kYuvI601Constants,
  817. width);
  818. dst_argb1555 += dst_stride_argb1555;
  819. src_y += src_stride_y;
  820. if (y & 1) {
  821. src_u += src_stride_u;
  822. src_v += src_stride_v;
  823. }
  824. }
  825. return 0;
  826. }
  827. // Convert I420 to ARGB4444.
  828. LIBYUV_API
  829. int I420ToARGB4444(const uint8_t* src_y,
  830. int src_stride_y,
  831. const uint8_t* src_u,
  832. int src_stride_u,
  833. const uint8_t* src_v,
  834. int src_stride_v,
  835. uint8_t* dst_argb4444,
  836. int dst_stride_argb4444,
  837. int width,
  838. int height) {
  839. int y;
  840. void (*I422ToARGB4444Row)(const uint8_t* y_buf, const uint8_t* u_buf,
  841. const uint8_t* v_buf, uint8_t* rgb_buf,
  842. const struct YuvConstants* yuvconstants,
  843. int width) = I422ToARGB4444Row_C;
  844. if (!src_y || !src_u || !src_v || !dst_argb4444 || width <= 0 ||
  845. height == 0) {
  846. return -1;
  847. }
  848. // Negative height means invert the image.
  849. if (height < 0) {
  850. height = -height;
  851. dst_argb4444 = dst_argb4444 + (height - 1) * dst_stride_argb4444;
  852. dst_stride_argb4444 = -dst_stride_argb4444;
  853. }
  854. #if defined(HAS_I422TOARGB4444ROW_SSSE3)
  855. if (TestCpuFlag(kCpuHasSSSE3)) {
  856. I422ToARGB4444Row = I422ToARGB4444Row_Any_SSSE3;
  857. if (IS_ALIGNED(width, 8)) {
  858. I422ToARGB4444Row = I422ToARGB4444Row_SSSE3;
  859. }
  860. }
  861. #endif
  862. #if defined(HAS_I422TOARGB4444ROW_AVX2)
  863. if (TestCpuFlag(kCpuHasAVX2)) {
  864. I422ToARGB4444Row = I422ToARGB4444Row_Any_AVX2;
  865. if (IS_ALIGNED(width, 16)) {
  866. I422ToARGB4444Row = I422ToARGB4444Row_AVX2;
  867. }
  868. }
  869. #endif
  870. #if defined(HAS_I422TOARGB4444ROW_NEON)
  871. if (TestCpuFlag(kCpuHasNEON)) {
  872. I422ToARGB4444Row = I422ToARGB4444Row_Any_NEON;
  873. if (IS_ALIGNED(width, 8)) {
  874. I422ToARGB4444Row = I422ToARGB4444Row_NEON;
  875. }
  876. }
  877. #endif
  878. #if defined(HAS_I422TOARGB4444ROW_MSA)
  879. if (TestCpuFlag(kCpuHasMSA)) {
  880. I422ToARGB4444Row = I422ToARGB4444Row_Any_MSA;
  881. if (IS_ALIGNED(width, 8)) {
  882. I422ToARGB4444Row = I422ToARGB4444Row_MSA;
  883. }
  884. }
  885. #endif
  886. for (y = 0; y < height; ++y) {
  887. I422ToARGB4444Row(src_y, src_u, src_v, dst_argb4444, &kYuvI601Constants,
  888. width);
  889. dst_argb4444 += dst_stride_argb4444;
  890. src_y += src_stride_y;
  891. if (y & 1) {
  892. src_u += src_stride_u;
  893. src_v += src_stride_v;
  894. }
  895. }
  896. return 0;
  897. }
  898. // Convert I420 to RGB565 with specified color matrix.
  899. LIBYUV_API
  900. int I420ToRGB565Matrix(const uint8_t* src_y,
  901. int src_stride_y,
  902. const uint8_t* src_u,
  903. int src_stride_u,
  904. const uint8_t* src_v,
  905. int src_stride_v,
  906. uint8_t* dst_rgb565,
  907. int dst_stride_rgb565,
  908. const struct YuvConstants* yuvconstants,
  909. int width,
  910. int height) {
  911. int y;
  912. void (*I422ToRGB565Row)(const uint8_t* y_buf, const uint8_t* u_buf,
  913. const uint8_t* v_buf, uint8_t* rgb_buf,
  914. const struct YuvConstants* yuvconstants, int width) =
  915. I422ToRGB565Row_C;
  916. if (!src_y || !src_u || !src_v || !dst_rgb565 || width <= 0 || height == 0) {
  917. return -1;
  918. }
  919. // Negative height means invert the image.
  920. if (height < 0) {
  921. height = -height;
  922. dst_rgb565 = dst_rgb565 + (height - 1) * dst_stride_rgb565;
  923. dst_stride_rgb565 = -dst_stride_rgb565;
  924. }
  925. #if defined(HAS_I422TORGB565ROW_SSSE3)
  926. if (TestCpuFlag(kCpuHasSSSE3)) {
  927. I422ToRGB565Row = I422ToRGB565Row_Any_SSSE3;
  928. if (IS_ALIGNED(width, 8)) {
  929. I422ToRGB565Row = I422ToRGB565Row_SSSE3;
  930. }
  931. }
  932. #endif
  933. #if defined(HAS_I422TORGB565ROW_AVX2)
  934. if (TestCpuFlag(kCpuHasAVX2)) {
  935. I422ToRGB565Row = I422ToRGB565Row_Any_AVX2;
  936. if (IS_ALIGNED(width, 16)) {
  937. I422ToRGB565Row = I422ToRGB565Row_AVX2;
  938. }
  939. }
  940. #endif
  941. #if defined(HAS_I422TORGB565ROW_NEON)
  942. if (TestCpuFlag(kCpuHasNEON)) {
  943. I422ToRGB565Row = I422ToRGB565Row_Any_NEON;
  944. if (IS_ALIGNED(width, 8)) {
  945. I422ToRGB565Row = I422ToRGB565Row_NEON;
  946. }
  947. }
  948. #endif
  949. #if defined(HAS_I422TORGB565ROW_MSA)
  950. if (TestCpuFlag(kCpuHasMSA)) {
  951. I422ToRGB565Row = I422ToRGB565Row_Any_MSA;
  952. if (IS_ALIGNED(width, 8)) {
  953. I422ToRGB565Row = I422ToRGB565Row_MSA;
  954. }
  955. }
  956. #endif
  957. for (y = 0; y < height; ++y) {
  958. I422ToRGB565Row(src_y, src_u, src_v, dst_rgb565, yuvconstants, width);
  959. dst_rgb565 += dst_stride_rgb565;
  960. src_y += src_stride_y;
  961. if (y & 1) {
  962. src_u += src_stride_u;
  963. src_v += src_stride_v;
  964. }
  965. }
  966. return 0;
  967. }
  968. // Convert I420 to RGB565.
  969. LIBYUV_API
  970. int I420ToRGB565(const uint8_t* src_y,
  971. int src_stride_y,
  972. const uint8_t* src_u,
  973. int src_stride_u,
  974. const uint8_t* src_v,
  975. int src_stride_v,
  976. uint8_t* dst_rgb565,
  977. int dst_stride_rgb565,
  978. int width,
  979. int height) {
  980. return I420ToRGB565Matrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  981. src_stride_v, dst_rgb565, dst_stride_rgb565,
  982. &kYuvI601Constants, width, height);
  983. }
  984. // Convert J420 to RGB565.
  985. LIBYUV_API
  986. int J420ToRGB565(const uint8_t* src_y,
  987. int src_stride_y,
  988. const uint8_t* src_u,
  989. int src_stride_u,
  990. const uint8_t* src_v,
  991. int src_stride_v,
  992. uint8_t* dst_rgb565,
  993. int dst_stride_rgb565,
  994. int width,
  995. int height) {
  996. return I420ToRGB565Matrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  997. src_stride_v, dst_rgb565, dst_stride_rgb565,
  998. &kYuvJPEGConstants, width, height);
  999. }
  1000. // Convert H420 to RGB565.
  1001. LIBYUV_API
  1002. int H420ToRGB565(const uint8_t* src_y,
  1003. int src_stride_y,
  1004. const uint8_t* src_u,
  1005. int src_stride_u,
  1006. const uint8_t* src_v,
  1007. int src_stride_v,
  1008. uint8_t* dst_rgb565,
  1009. int dst_stride_rgb565,
  1010. int width,
  1011. int height) {
  1012. return I420ToRGB565Matrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  1013. src_stride_v, dst_rgb565, dst_stride_rgb565,
  1014. &kYuvH709Constants, width, height);
  1015. }
  1016. // Convert I422 to RGB565.
  1017. LIBYUV_API
  1018. int I422ToRGB565(const uint8_t* src_y,
  1019. int src_stride_y,
  1020. const uint8_t* src_u,
  1021. int src_stride_u,
  1022. const uint8_t* src_v,
  1023. int src_stride_v,
  1024. uint8_t* dst_rgb565,
  1025. int dst_stride_rgb565,
  1026. int width,
  1027. int height) {
  1028. int y;
  1029. void (*I422ToRGB565Row)(const uint8_t* y_buf, const uint8_t* u_buf,
  1030. const uint8_t* v_buf, uint8_t* rgb_buf,
  1031. const struct YuvConstants* yuvconstants, int width) =
  1032. I422ToRGB565Row_C;
  1033. if (!src_y || !src_u || !src_v || !dst_rgb565 || width <= 0 || height == 0) {
  1034. return -1;
  1035. }
  1036. // Negative height means invert the image.
  1037. if (height < 0) {
  1038. height = -height;
  1039. dst_rgb565 = dst_rgb565 + (height - 1) * dst_stride_rgb565;
  1040. dst_stride_rgb565 = -dst_stride_rgb565;
  1041. }
  1042. #if defined(HAS_I422TORGB565ROW_SSSE3)
  1043. if (TestCpuFlag(kCpuHasSSSE3)) {
  1044. I422ToRGB565Row = I422ToRGB565Row_Any_SSSE3;
  1045. if (IS_ALIGNED(width, 8)) {
  1046. I422ToRGB565Row = I422ToRGB565Row_SSSE3;
  1047. }
  1048. }
  1049. #endif
  1050. #if defined(HAS_I422TORGB565ROW_AVX2)
  1051. if (TestCpuFlag(kCpuHasAVX2)) {
  1052. I422ToRGB565Row = I422ToRGB565Row_Any_AVX2;
  1053. if (IS_ALIGNED(width, 16)) {
  1054. I422ToRGB565Row = I422ToRGB565Row_AVX2;
  1055. }
  1056. }
  1057. #endif
  1058. #if defined(HAS_I422TORGB565ROW_NEON)
  1059. if (TestCpuFlag(kCpuHasNEON)) {
  1060. I422ToRGB565Row = I422ToRGB565Row_Any_NEON;
  1061. if (IS_ALIGNED(width, 8)) {
  1062. I422ToRGB565Row = I422ToRGB565Row_NEON;
  1063. }
  1064. }
  1065. #endif
  1066. #if defined(HAS_I422TORGB565ROW_MSA)
  1067. if (TestCpuFlag(kCpuHasMSA)) {
  1068. I422ToRGB565Row = I422ToRGB565Row_Any_MSA;
  1069. if (IS_ALIGNED(width, 8)) {
  1070. I422ToRGB565Row = I422ToRGB565Row_MSA;
  1071. }
  1072. }
  1073. #endif
  1074. for (y = 0; y < height; ++y) {
  1075. I422ToRGB565Row(src_y, src_u, src_v, dst_rgb565, &kYuvI601Constants, width);
  1076. dst_rgb565 += dst_stride_rgb565;
  1077. src_y += src_stride_y;
  1078. src_u += src_stride_u;
  1079. src_v += src_stride_v;
  1080. }
  1081. return 0;
  1082. }
  1083. // Ordered 8x8 dither for 888 to 565. Values from 0 to 7.
  1084. static const uint8_t kDither565_4x4[16] = {
  1085. 0, 4, 1, 5, 6, 2, 7, 3, 1, 5, 0, 4, 7, 3, 6, 2,
  1086. };
  1087. // Convert I420 to RGB565 with dithering.
  1088. LIBYUV_API
  1089. int I420ToRGB565Dither(const uint8_t* src_y,
  1090. int src_stride_y,
  1091. const uint8_t* src_u,
  1092. int src_stride_u,
  1093. const uint8_t* src_v,
  1094. int src_stride_v,
  1095. uint8_t* dst_rgb565,
  1096. int dst_stride_rgb565,
  1097. const uint8_t* dither4x4,
  1098. int width,
  1099. int height) {
  1100. int y;
  1101. void (*I422ToARGBRow)(const uint8_t* y_buf, const uint8_t* u_buf,
  1102. const uint8_t* v_buf, uint8_t* rgb_buf,
  1103. const struct YuvConstants* yuvconstants, int width) =
  1104. I422ToARGBRow_C;
  1105. void (*ARGBToRGB565DitherRow)(const uint8_t* src_argb, uint8_t* dst_rgb,
  1106. const uint32_t dither4, int width) =
  1107. ARGBToRGB565DitherRow_C;
  1108. if (!src_y || !src_u || !src_v || !dst_rgb565 || width <= 0 || height == 0) {
  1109. return -1;
  1110. }
  1111. // Negative height means invert the image.
  1112. if (height < 0) {
  1113. height = -height;
  1114. dst_rgb565 = dst_rgb565 + (height - 1) * dst_stride_rgb565;
  1115. dst_stride_rgb565 = -dst_stride_rgb565;
  1116. }
  1117. if (!dither4x4) {
  1118. dither4x4 = kDither565_4x4;
  1119. }
  1120. #if defined(HAS_I422TOARGBROW_SSSE3)
  1121. if (TestCpuFlag(kCpuHasSSSE3)) {
  1122. I422ToARGBRow = I422ToARGBRow_Any_SSSE3;
  1123. if (IS_ALIGNED(width, 8)) {
  1124. I422ToARGBRow = I422ToARGBRow_SSSE3;
  1125. }
  1126. }
  1127. #endif
  1128. #if defined(HAS_I422TOARGBROW_AVX2)
  1129. if (TestCpuFlag(kCpuHasAVX2)) {
  1130. I422ToARGBRow = I422ToARGBRow_Any_AVX2;
  1131. if (IS_ALIGNED(width, 16)) {
  1132. I422ToARGBRow = I422ToARGBRow_AVX2;
  1133. }
  1134. }
  1135. #endif
  1136. #if defined(HAS_I422TOARGBROW_NEON)
  1137. if (TestCpuFlag(kCpuHasNEON)) {
  1138. I422ToARGBRow = I422ToARGBRow_Any_NEON;
  1139. if (IS_ALIGNED(width, 8)) {
  1140. I422ToARGBRow = I422ToARGBRow_NEON;
  1141. }
  1142. }
  1143. #endif
  1144. #if defined(HAS_I422TOARGBROW_MSA)
  1145. if (TestCpuFlag(kCpuHasMSA)) {
  1146. I422ToARGBRow = I422ToARGBRow_Any_MSA;
  1147. if (IS_ALIGNED(width, 8)) {
  1148. I422ToARGBRow = I422ToARGBRow_MSA;
  1149. }
  1150. }
  1151. #endif
  1152. #if defined(HAS_ARGBTORGB565DITHERROW_SSE2)
  1153. if (TestCpuFlag(kCpuHasSSE2)) {
  1154. ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_Any_SSE2;
  1155. if (IS_ALIGNED(width, 4)) {
  1156. ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_SSE2;
  1157. }
  1158. }
  1159. #endif
  1160. #if defined(HAS_ARGBTORGB565DITHERROW_AVX2)
  1161. if (TestCpuFlag(kCpuHasAVX2)) {
  1162. ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_Any_AVX2;
  1163. if (IS_ALIGNED(width, 8)) {
  1164. ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_AVX2;
  1165. }
  1166. }
  1167. #endif
  1168. #if defined(HAS_ARGBTORGB565DITHERROW_NEON)
  1169. if (TestCpuFlag(kCpuHasNEON)) {
  1170. ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_Any_NEON;
  1171. if (IS_ALIGNED(width, 8)) {
  1172. ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_NEON;
  1173. }
  1174. }
  1175. #endif
  1176. #if defined(HAS_ARGBTORGB565DITHERROW_MSA)
  1177. if (TestCpuFlag(kCpuHasMSA)) {
  1178. ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_Any_MSA;
  1179. if (IS_ALIGNED(width, 8)) {
  1180. ARGBToRGB565DitherRow = ARGBToRGB565DitherRow_MSA;
  1181. }
  1182. }
  1183. #endif
  1184. {
  1185. // Allocate a row of argb.
  1186. align_buffer_64(row_argb, width * 4);
  1187. for (y = 0; y < height; ++y) {
  1188. I422ToARGBRow(src_y, src_u, src_v, row_argb, &kYuvI601Constants, width);
  1189. ARGBToRGB565DitherRow(row_argb, dst_rgb565,
  1190. *(const uint32_t*)(dither4x4 + ((y & 3) << 2)),
  1191. width);
  1192. dst_rgb565 += dst_stride_rgb565;
  1193. src_y += src_stride_y;
  1194. if (y & 1) {
  1195. src_u += src_stride_u;
  1196. src_v += src_stride_v;
  1197. }
  1198. }
  1199. free_aligned_buffer_64(row_argb);
  1200. }
  1201. return 0;
  1202. }
  1203. // Convert I420 to AR30 with matrix
  1204. static int I420ToAR30Matrix(const uint8_t* src_y,
  1205. int src_stride_y,
  1206. const uint8_t* src_u,
  1207. int src_stride_u,
  1208. const uint8_t* src_v,
  1209. int src_stride_v,
  1210. uint8_t* dst_ar30,
  1211. int dst_stride_ar30,
  1212. const struct YuvConstants* yuvconstants,
  1213. int width,
  1214. int height) {
  1215. int y;
  1216. void (*I422ToAR30Row)(const uint8_t* y_buf, const uint8_t* u_buf,
  1217. const uint8_t* v_buf, uint8_t* rgb_buf,
  1218. const struct YuvConstants* yuvconstants, int width) =
  1219. I422ToAR30Row_C;
  1220. if (!src_y || !src_u || !src_v || !dst_ar30 || width <= 0 || height == 0) {
  1221. return -1;
  1222. }
  1223. // Negative height means invert the image.
  1224. if (height < 0) {
  1225. height = -height;
  1226. dst_ar30 = dst_ar30 + (height - 1) * dst_stride_ar30;
  1227. dst_stride_ar30 = -dst_stride_ar30;
  1228. }
  1229. #if defined(HAS_I422TOAR30ROW_SSSE3)
  1230. if (TestCpuFlag(kCpuHasSSSE3)) {
  1231. I422ToAR30Row = I422ToAR30Row_Any_SSSE3;
  1232. if (IS_ALIGNED(width, 8)) {
  1233. I422ToAR30Row = I422ToAR30Row_SSSE3;
  1234. }
  1235. }
  1236. #endif
  1237. #if defined(HAS_I422TOAR30ROW_AVX2)
  1238. if (TestCpuFlag(kCpuHasAVX2)) {
  1239. I422ToAR30Row = I422ToAR30Row_Any_AVX2;
  1240. if (IS_ALIGNED(width, 16)) {
  1241. I422ToAR30Row = I422ToAR30Row_AVX2;
  1242. }
  1243. }
  1244. #endif
  1245. for (y = 0; y < height; ++y) {
  1246. I422ToAR30Row(src_y, src_u, src_v, dst_ar30, yuvconstants, width);
  1247. dst_ar30 += dst_stride_ar30;
  1248. src_y += src_stride_y;
  1249. if (y & 1) {
  1250. src_u += src_stride_u;
  1251. src_v += src_stride_v;
  1252. }
  1253. }
  1254. return 0;
  1255. }
  1256. // Convert I420 to AR30.
  1257. LIBYUV_API
  1258. int I420ToAR30(const uint8_t* src_y,
  1259. int src_stride_y,
  1260. const uint8_t* src_u,
  1261. int src_stride_u,
  1262. const uint8_t* src_v,
  1263. int src_stride_v,
  1264. uint8_t* dst_ar30,
  1265. int dst_stride_ar30,
  1266. int width,
  1267. int height) {
  1268. return I420ToAR30Matrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  1269. src_stride_v, dst_ar30, dst_stride_ar30,
  1270. &kYuvI601Constants, width, height);
  1271. }
  1272. // Convert H420 to AR30.
  1273. LIBYUV_API
  1274. int H420ToAR30(const uint8_t* src_y,
  1275. int src_stride_y,
  1276. const uint8_t* src_u,
  1277. int src_stride_u,
  1278. const uint8_t* src_v,
  1279. int src_stride_v,
  1280. uint8_t* dst_ar30,
  1281. int dst_stride_ar30,
  1282. int width,
  1283. int height) {
  1284. return I420ToAR30Matrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  1285. src_stride_v, dst_ar30, dst_stride_ar30,
  1286. &kYvuH709Constants, width, height);
  1287. }
  1288. // Convert I420 to specified format
  1289. LIBYUV_API
  1290. int ConvertFromI420(const uint8_t* y,
  1291. int y_stride,
  1292. const uint8_t* u,
  1293. int u_stride,
  1294. const uint8_t* v,
  1295. int v_stride,
  1296. uint8_t* dst_sample,
  1297. int dst_sample_stride,
  1298. int width,
  1299. int height,
  1300. uint32_t fourcc) {
  1301. uint32_t format = CanonicalFourCC(fourcc);
  1302. int r = 0;
  1303. if (!y || !u || !v || !dst_sample || width <= 0 || height == 0) {
  1304. return -1;
  1305. }
  1306. switch (format) {
  1307. // Single plane formats
  1308. case FOURCC_YUY2:
  1309. r = I420ToYUY2(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1310. dst_sample_stride ? dst_sample_stride : width * 2, width,
  1311. height);
  1312. break;
  1313. case FOURCC_UYVY:
  1314. r = I420ToUYVY(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1315. dst_sample_stride ? dst_sample_stride : width * 2, width,
  1316. height);
  1317. break;
  1318. case FOURCC_RGBP:
  1319. r = I420ToRGB565(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1320. dst_sample_stride ? dst_sample_stride : width * 2, width,
  1321. height);
  1322. break;
  1323. case FOURCC_RGBO:
  1324. r = I420ToARGB1555(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1325. dst_sample_stride ? dst_sample_stride : width * 2,
  1326. width, height);
  1327. break;
  1328. case FOURCC_R444:
  1329. r = I420ToARGB4444(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1330. dst_sample_stride ? dst_sample_stride : width * 2,
  1331. width, height);
  1332. break;
  1333. case FOURCC_24BG:
  1334. r = I420ToRGB24(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1335. dst_sample_stride ? dst_sample_stride : width * 3, width,
  1336. height);
  1337. break;
  1338. case FOURCC_RAW:
  1339. r = I420ToRAW(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1340. dst_sample_stride ? dst_sample_stride : width * 3, width,
  1341. height);
  1342. break;
  1343. case FOURCC_ARGB:
  1344. r = I420ToARGB(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1345. dst_sample_stride ? dst_sample_stride : width * 4, width,
  1346. height);
  1347. break;
  1348. case FOURCC_BGRA:
  1349. r = I420ToBGRA(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1350. dst_sample_stride ? dst_sample_stride : width * 4, width,
  1351. height);
  1352. break;
  1353. case FOURCC_ABGR:
  1354. r = I420ToABGR(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1355. dst_sample_stride ? dst_sample_stride : width * 4, width,
  1356. height);
  1357. break;
  1358. case FOURCC_RGBA:
  1359. r = I420ToRGBA(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1360. dst_sample_stride ? dst_sample_stride : width * 4, width,
  1361. height);
  1362. break;
  1363. case FOURCC_AR30:
  1364. r = I420ToAR30(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1365. dst_sample_stride ? dst_sample_stride : width * 4, width,
  1366. height);
  1367. break;
  1368. case FOURCC_I400:
  1369. r = I400Copy(y, y_stride, dst_sample,
  1370. dst_sample_stride ? dst_sample_stride : width, width,
  1371. height);
  1372. break;
  1373. case FOURCC_NV12: {
  1374. uint8_t* dst_uv = dst_sample + width * height;
  1375. r = I420ToNV12(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1376. dst_sample_stride ? dst_sample_stride : width, dst_uv,
  1377. dst_sample_stride ? dst_sample_stride : width, width,
  1378. height);
  1379. break;
  1380. }
  1381. case FOURCC_NV21: {
  1382. uint8_t* dst_vu = dst_sample + width * height;
  1383. r = I420ToNV21(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1384. dst_sample_stride ? dst_sample_stride : width, dst_vu,
  1385. dst_sample_stride ? dst_sample_stride : width, width,
  1386. height);
  1387. break;
  1388. }
  1389. // TODO(fbarchard): Add M420.
  1390. // Triplanar formats
  1391. case FOURCC_I420:
  1392. case FOURCC_YV12: {
  1393. dst_sample_stride = dst_sample_stride ? dst_sample_stride : width;
  1394. int halfstride = (dst_sample_stride + 1) / 2;
  1395. int halfheight = (height + 1) / 2;
  1396. uint8_t* dst_u;
  1397. uint8_t* dst_v;
  1398. if (format == FOURCC_YV12) {
  1399. dst_v = dst_sample + dst_sample_stride * height;
  1400. dst_u = dst_v + halfstride * halfheight;
  1401. } else {
  1402. dst_u = dst_sample + dst_sample_stride * height;
  1403. dst_v = dst_u + halfstride * halfheight;
  1404. }
  1405. r = I420Copy(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1406. dst_sample_stride, dst_u, halfstride, dst_v, halfstride,
  1407. width, height);
  1408. break;
  1409. }
  1410. case FOURCC_I422:
  1411. case FOURCC_YV16: {
  1412. dst_sample_stride = dst_sample_stride ? dst_sample_stride : width;
  1413. int halfstride = (dst_sample_stride + 1) / 2;
  1414. uint8_t* dst_u;
  1415. uint8_t* dst_v;
  1416. if (format == FOURCC_YV16) {
  1417. dst_v = dst_sample + dst_sample_stride * height;
  1418. dst_u = dst_v + halfstride * height;
  1419. } else {
  1420. dst_u = dst_sample + dst_sample_stride * height;
  1421. dst_v = dst_u + halfstride * height;
  1422. }
  1423. r = I420ToI422(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1424. dst_sample_stride, dst_u, halfstride, dst_v, halfstride,
  1425. width, height);
  1426. break;
  1427. }
  1428. case FOURCC_I444:
  1429. case FOURCC_YV24: {
  1430. dst_sample_stride = dst_sample_stride ? dst_sample_stride : width;
  1431. uint8_t* dst_u;
  1432. uint8_t* dst_v;
  1433. if (format == FOURCC_YV24) {
  1434. dst_v = dst_sample + dst_sample_stride * height;
  1435. dst_u = dst_v + dst_sample_stride * height;
  1436. } else {
  1437. dst_u = dst_sample + dst_sample_stride * height;
  1438. dst_v = dst_u + dst_sample_stride * height;
  1439. }
  1440. r = I420ToI444(y, y_stride, u, u_stride, v, v_stride, dst_sample,
  1441. dst_sample_stride, dst_u, dst_sample_stride, dst_v,
  1442. dst_sample_stride, width, height);
  1443. break;
  1444. }
  1445. // Formats not supported - MJPG, biplanar, some rgb formats.
  1446. default:
  1447. return -1; // unknown fourcc - return failure code.
  1448. }
  1449. return r;
  1450. }
  1451. #ifdef __cplusplus
  1452. } // extern "C"
  1453. } // namespace libyuv
  1454. #endif