2
0

convert_argb.cc 72 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371
  1. /*
  2. * Copyright 2011 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_argb.h"
  11. #include "libyuv/cpu_id.h"
  12. #ifdef HAVE_JPEG
  13. #include "libyuv/mjpeg_decoder.h"
  14. #endif
  15. #include "libyuv/planar_functions.h" // For CopyPlane and ARGBShuffle.
  16. #include "libyuv/rotate_argb.h"
  17. #include "libyuv/row.h"
  18. #include "libyuv/video_common.h"
  19. #ifdef __cplusplus
  20. namespace libyuv {
  21. extern "C" {
  22. #endif
  23. // Copy ARGB with optional flipping
  24. LIBYUV_API
  25. int ARGBCopy(const uint8_t* src_argb,
  26. int src_stride_argb,
  27. uint8_t* dst_argb,
  28. int dst_stride_argb,
  29. int width,
  30. int height) {
  31. if (!src_argb || !dst_argb || width <= 0 || height == 0) {
  32. return -1;
  33. }
  34. // Negative height means invert the image.
  35. if (height < 0) {
  36. height = -height;
  37. src_argb = src_argb + (height - 1) * src_stride_argb;
  38. src_stride_argb = -src_stride_argb;
  39. }
  40. CopyPlane(src_argb, src_stride_argb, dst_argb, dst_stride_argb, width * 4,
  41. height);
  42. return 0;
  43. }
  44. // Convert I420 to ARGB with matrix
  45. static int I420ToARGBMatrix(const uint8_t* src_y,
  46. int src_stride_y,
  47. const uint8_t* src_u,
  48. int src_stride_u,
  49. const uint8_t* src_v,
  50. int src_stride_v,
  51. uint8_t* dst_argb,
  52. int dst_stride_argb,
  53. const struct YuvConstants* yuvconstants,
  54. int width,
  55. int height) {
  56. int y;
  57. void (*I422ToARGBRow)(const uint8_t* y_buf, const uint8_t* u_buf,
  58. const uint8_t* v_buf, uint8_t* rgb_buf,
  59. const struct YuvConstants* yuvconstants, int width) =
  60. I422ToARGBRow_C;
  61. if (!src_y || !src_u || !src_v || !dst_argb || width <= 0 || height == 0) {
  62. return -1;
  63. }
  64. // Negative height means invert the image.
  65. if (height < 0) {
  66. height = -height;
  67. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  68. dst_stride_argb = -dst_stride_argb;
  69. }
  70. #if defined(HAS_I422TOARGBROW_SSSE3)
  71. if (TestCpuFlag(kCpuHasSSSE3)) {
  72. I422ToARGBRow = I422ToARGBRow_Any_SSSE3;
  73. if (IS_ALIGNED(width, 8)) {
  74. I422ToARGBRow = I422ToARGBRow_SSSE3;
  75. }
  76. }
  77. #endif
  78. #if defined(HAS_I422TOARGBROW_AVX2)
  79. if (TestCpuFlag(kCpuHasAVX2)) {
  80. I422ToARGBRow = I422ToARGBRow_Any_AVX2;
  81. if (IS_ALIGNED(width, 16)) {
  82. I422ToARGBRow = I422ToARGBRow_AVX2;
  83. }
  84. }
  85. #endif
  86. #if defined(HAS_I422TOARGBROW_NEON)
  87. if (TestCpuFlag(kCpuHasNEON)) {
  88. I422ToARGBRow = I422ToARGBRow_Any_NEON;
  89. if (IS_ALIGNED(width, 8)) {
  90. I422ToARGBRow = I422ToARGBRow_NEON;
  91. }
  92. }
  93. #endif
  94. #if defined(HAS_I422TOARGBROW_MSA)
  95. if (TestCpuFlag(kCpuHasMSA)) {
  96. I422ToARGBRow = I422ToARGBRow_Any_MSA;
  97. if (IS_ALIGNED(width, 8)) {
  98. I422ToARGBRow = I422ToARGBRow_MSA;
  99. }
  100. }
  101. #endif
  102. for (y = 0; y < height; ++y) {
  103. I422ToARGBRow(src_y, src_u, src_v, dst_argb, yuvconstants, width);
  104. dst_argb += dst_stride_argb;
  105. src_y += src_stride_y;
  106. if (y & 1) {
  107. src_u += src_stride_u;
  108. src_v += src_stride_v;
  109. }
  110. }
  111. return 0;
  112. }
  113. // Convert I420 to ARGB.
  114. LIBYUV_API
  115. int I420ToARGB(const uint8_t* src_y,
  116. int src_stride_y,
  117. const uint8_t* src_u,
  118. int src_stride_u,
  119. const uint8_t* src_v,
  120. int src_stride_v,
  121. uint8_t* dst_argb,
  122. int dst_stride_argb,
  123. int width,
  124. int height) {
  125. return I420ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  126. src_stride_v, dst_argb, dst_stride_argb,
  127. &kYuvI601Constants, width, height);
  128. }
  129. // Convert I420 to ABGR.
  130. LIBYUV_API
  131. int I420ToABGR(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_abgr,
  138. int dst_stride_abgr,
  139. int width,
  140. int height) {
  141. return I420ToARGBMatrix(src_y, src_stride_y, src_v,
  142. src_stride_v, // Swap U and V
  143. src_u, src_stride_u, dst_abgr, dst_stride_abgr,
  144. &kYvuI601Constants, // Use Yvu matrix
  145. width, height);
  146. }
  147. // Convert J420 to ARGB.
  148. LIBYUV_API
  149. int J420ToARGB(const uint8_t* src_y,
  150. int src_stride_y,
  151. const uint8_t* src_u,
  152. int src_stride_u,
  153. const uint8_t* src_v,
  154. int src_stride_v,
  155. uint8_t* dst_argb,
  156. int dst_stride_argb,
  157. int width,
  158. int height) {
  159. return I420ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  160. src_stride_v, dst_argb, dst_stride_argb,
  161. &kYuvJPEGConstants, width, height);
  162. }
  163. // Convert J420 to ABGR.
  164. LIBYUV_API
  165. int J420ToABGR(const uint8_t* src_y,
  166. int src_stride_y,
  167. const uint8_t* src_u,
  168. int src_stride_u,
  169. const uint8_t* src_v,
  170. int src_stride_v,
  171. uint8_t* dst_abgr,
  172. int dst_stride_abgr,
  173. int width,
  174. int height) {
  175. return I420ToARGBMatrix(src_y, src_stride_y, src_v,
  176. src_stride_v, // Swap U and V
  177. src_u, src_stride_u, dst_abgr, dst_stride_abgr,
  178. &kYvuJPEGConstants, // Use Yvu matrix
  179. width, height);
  180. }
  181. // Convert H420 to ARGB.
  182. LIBYUV_API
  183. int H420ToARGB(const uint8_t* src_y,
  184. int src_stride_y,
  185. const uint8_t* src_u,
  186. int src_stride_u,
  187. const uint8_t* src_v,
  188. int src_stride_v,
  189. uint8_t* dst_argb,
  190. int dst_stride_argb,
  191. int width,
  192. int height) {
  193. return I420ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  194. src_stride_v, dst_argb, dst_stride_argb,
  195. &kYuvH709Constants, width, height);
  196. }
  197. // Convert H420 to ABGR.
  198. LIBYUV_API
  199. int H420ToABGR(const uint8_t* src_y,
  200. int src_stride_y,
  201. const uint8_t* src_u,
  202. int src_stride_u,
  203. const uint8_t* src_v,
  204. int src_stride_v,
  205. uint8_t* dst_abgr,
  206. int dst_stride_abgr,
  207. int width,
  208. int height) {
  209. return I420ToARGBMatrix(src_y, src_stride_y, src_v,
  210. src_stride_v, // Swap U and V
  211. src_u, src_stride_u, dst_abgr, dst_stride_abgr,
  212. &kYvuH709Constants, // Use Yvu matrix
  213. width, height);
  214. }
  215. // Convert I422 to ARGB with matrix
  216. static int I422ToARGBMatrix(const uint8_t* src_y,
  217. int src_stride_y,
  218. const uint8_t* src_u,
  219. int src_stride_u,
  220. const uint8_t* src_v,
  221. int src_stride_v,
  222. uint8_t* dst_argb,
  223. int dst_stride_argb,
  224. const struct YuvConstants* yuvconstants,
  225. int width,
  226. int height) {
  227. int y;
  228. void (*I422ToARGBRow)(const uint8_t* y_buf, const uint8_t* u_buf,
  229. const uint8_t* v_buf, uint8_t* rgb_buf,
  230. const struct YuvConstants* yuvconstants, int width) =
  231. I422ToARGBRow_C;
  232. if (!src_y || !src_u || !src_v || !dst_argb || width <= 0 || height == 0) {
  233. return -1;
  234. }
  235. // Negative height means invert the image.
  236. if (height < 0) {
  237. height = -height;
  238. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  239. dst_stride_argb = -dst_stride_argb;
  240. }
  241. // Coalesce rows.
  242. if (src_stride_y == width && src_stride_u * 2 == width &&
  243. src_stride_v * 2 == width && dst_stride_argb == width * 4) {
  244. width *= height;
  245. height = 1;
  246. src_stride_y = src_stride_u = src_stride_v = dst_stride_argb = 0;
  247. }
  248. #if defined(HAS_I422TOARGBROW_SSSE3)
  249. if (TestCpuFlag(kCpuHasSSSE3)) {
  250. I422ToARGBRow = I422ToARGBRow_Any_SSSE3;
  251. if (IS_ALIGNED(width, 8)) {
  252. I422ToARGBRow = I422ToARGBRow_SSSE3;
  253. }
  254. }
  255. #endif
  256. #if defined(HAS_I422TOARGBROW_AVX2)
  257. if (TestCpuFlag(kCpuHasAVX2)) {
  258. I422ToARGBRow = I422ToARGBRow_Any_AVX2;
  259. if (IS_ALIGNED(width, 16)) {
  260. I422ToARGBRow = I422ToARGBRow_AVX2;
  261. }
  262. }
  263. #endif
  264. #if defined(HAS_I422TOARGBROW_NEON)
  265. if (TestCpuFlag(kCpuHasNEON)) {
  266. I422ToARGBRow = I422ToARGBRow_Any_NEON;
  267. if (IS_ALIGNED(width, 8)) {
  268. I422ToARGBRow = I422ToARGBRow_NEON;
  269. }
  270. }
  271. #endif
  272. #if defined(HAS_I422TOARGBROW_MSA)
  273. if (TestCpuFlag(kCpuHasMSA)) {
  274. I422ToARGBRow = I422ToARGBRow_Any_MSA;
  275. if (IS_ALIGNED(width, 8)) {
  276. I422ToARGBRow = I422ToARGBRow_MSA;
  277. }
  278. }
  279. #endif
  280. for (y = 0; y < height; ++y) {
  281. I422ToARGBRow(src_y, src_u, src_v, dst_argb, yuvconstants, width);
  282. dst_argb += dst_stride_argb;
  283. src_y += src_stride_y;
  284. src_u += src_stride_u;
  285. src_v += src_stride_v;
  286. }
  287. return 0;
  288. }
  289. // Convert I422 to ARGB.
  290. LIBYUV_API
  291. int I422ToARGB(const uint8_t* src_y,
  292. int src_stride_y,
  293. const uint8_t* src_u,
  294. int src_stride_u,
  295. const uint8_t* src_v,
  296. int src_stride_v,
  297. uint8_t* dst_argb,
  298. int dst_stride_argb,
  299. int width,
  300. int height) {
  301. return I422ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  302. src_stride_v, dst_argb, dst_stride_argb,
  303. &kYuvI601Constants, width, height);
  304. }
  305. // Convert I422 to ABGR.
  306. LIBYUV_API
  307. int I422ToABGR(const uint8_t* src_y,
  308. int src_stride_y,
  309. const uint8_t* src_u,
  310. int src_stride_u,
  311. const uint8_t* src_v,
  312. int src_stride_v,
  313. uint8_t* dst_abgr,
  314. int dst_stride_abgr,
  315. int width,
  316. int height) {
  317. return I422ToARGBMatrix(src_y, src_stride_y, src_v,
  318. src_stride_v, // Swap U and V
  319. src_u, src_stride_u, dst_abgr, dst_stride_abgr,
  320. &kYvuI601Constants, // Use Yvu matrix
  321. width, height);
  322. }
  323. // Convert J422 to ARGB.
  324. LIBYUV_API
  325. int J422ToARGB(const uint8_t* src_y,
  326. int src_stride_y,
  327. const uint8_t* src_u,
  328. int src_stride_u,
  329. const uint8_t* src_v,
  330. int src_stride_v,
  331. uint8_t* dst_argb,
  332. int dst_stride_argb,
  333. int width,
  334. int height) {
  335. return I422ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  336. src_stride_v, dst_argb, dst_stride_argb,
  337. &kYuvJPEGConstants, width, height);
  338. }
  339. // Convert J422 to ABGR.
  340. LIBYUV_API
  341. int J422ToABGR(const uint8_t* src_y,
  342. int src_stride_y,
  343. const uint8_t* src_u,
  344. int src_stride_u,
  345. const uint8_t* src_v,
  346. int src_stride_v,
  347. uint8_t* dst_abgr,
  348. int dst_stride_abgr,
  349. int width,
  350. int height) {
  351. return I422ToARGBMatrix(src_y, src_stride_y, src_v,
  352. src_stride_v, // Swap U and V
  353. src_u, src_stride_u, dst_abgr, dst_stride_abgr,
  354. &kYvuJPEGConstants, // Use Yvu matrix
  355. width, height);
  356. }
  357. // Convert H422 to ARGB.
  358. LIBYUV_API
  359. int H422ToARGB(const uint8_t* src_y,
  360. int src_stride_y,
  361. const uint8_t* src_u,
  362. int src_stride_u,
  363. const uint8_t* src_v,
  364. int src_stride_v,
  365. uint8_t* dst_argb,
  366. int dst_stride_argb,
  367. int width,
  368. int height) {
  369. return I422ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  370. src_stride_v, dst_argb, dst_stride_argb,
  371. &kYuvH709Constants, width, height);
  372. }
  373. // Convert H422 to ABGR.
  374. LIBYUV_API
  375. int H422ToABGR(const uint8_t* src_y,
  376. int src_stride_y,
  377. const uint8_t* src_u,
  378. int src_stride_u,
  379. const uint8_t* src_v,
  380. int src_stride_v,
  381. uint8_t* dst_abgr,
  382. int dst_stride_abgr,
  383. int width,
  384. int height) {
  385. return I422ToARGBMatrix(src_y, src_stride_y, src_v,
  386. src_stride_v, // Swap U and V
  387. src_u, src_stride_u, dst_abgr, dst_stride_abgr,
  388. &kYvuH709Constants, // Use Yvu matrix
  389. width, height);
  390. }
  391. // Convert 10 bit YUV to ARGB with matrix
  392. // TODO(fbarchard): Consider passing scale multiplier to I210ToARGB to
  393. // multiply 10 bit yuv into high bits to allow any number of bits.
  394. static int I010ToAR30Matrix(const uint16_t* src_y,
  395. int src_stride_y,
  396. const uint16_t* src_u,
  397. int src_stride_u,
  398. const uint16_t* src_v,
  399. int src_stride_v,
  400. uint8_t* dst_ar30,
  401. int dst_stride_ar30,
  402. const struct YuvConstants* yuvconstants,
  403. int width,
  404. int height) {
  405. int y;
  406. void (*I210ToAR30Row)(const uint16_t* y_buf, const uint16_t* u_buf,
  407. const uint16_t* v_buf, uint8_t* rgb_buf,
  408. const struct YuvConstants* yuvconstants, int width) =
  409. I210ToAR30Row_C;
  410. if (!src_y || !src_u || !src_v || !dst_ar30 || width <= 0 || height == 0) {
  411. return -1;
  412. }
  413. // Negative height means invert the image.
  414. if (height < 0) {
  415. height = -height;
  416. dst_ar30 = dst_ar30 + (height - 1) * dst_stride_ar30;
  417. dst_stride_ar30 = -dst_stride_ar30;
  418. }
  419. #if defined(HAS_I210TOAR30ROW_SSSE3)
  420. if (TestCpuFlag(kCpuHasSSSE3)) {
  421. I210ToAR30Row = I210ToAR30Row_Any_SSSE3;
  422. if (IS_ALIGNED(width, 8)) {
  423. I210ToAR30Row = I210ToAR30Row_SSSE3;
  424. }
  425. }
  426. #endif
  427. #if defined(HAS_I210TOAR30ROW_AVX2)
  428. if (TestCpuFlag(kCpuHasAVX2)) {
  429. I210ToAR30Row = I210ToAR30Row_Any_AVX2;
  430. if (IS_ALIGNED(width, 16)) {
  431. I210ToAR30Row = I210ToAR30Row_AVX2;
  432. }
  433. }
  434. #endif
  435. for (y = 0; y < height; ++y) {
  436. I210ToAR30Row(src_y, src_u, src_v, dst_ar30, yuvconstants, width);
  437. dst_ar30 += dst_stride_ar30;
  438. src_y += src_stride_y;
  439. if (y & 1) {
  440. src_u += src_stride_u;
  441. src_v += src_stride_v;
  442. }
  443. }
  444. return 0;
  445. }
  446. // Convert I010 to AR30.
  447. LIBYUV_API
  448. int I010ToAR30(const uint16_t* src_y,
  449. int src_stride_y,
  450. const uint16_t* src_u,
  451. int src_stride_u,
  452. const uint16_t* src_v,
  453. int src_stride_v,
  454. uint8_t* dst_ar30,
  455. int dst_stride_ar30,
  456. int width,
  457. int height) {
  458. return I010ToAR30Matrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  459. src_stride_v, dst_ar30, dst_stride_ar30,
  460. &kYuvI601Constants, width, height);
  461. }
  462. // Convert H010 to AR30.
  463. LIBYUV_API
  464. int H010ToAR30(const uint16_t* src_y,
  465. int src_stride_y,
  466. const uint16_t* src_u,
  467. int src_stride_u,
  468. const uint16_t* src_v,
  469. int src_stride_v,
  470. uint8_t* dst_ar30,
  471. int dst_stride_ar30,
  472. int width,
  473. int height) {
  474. return I010ToAR30Matrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  475. src_stride_v, dst_ar30, dst_stride_ar30,
  476. &kYuvH709Constants, width, height);
  477. }
  478. // Convert I010 to AB30.
  479. LIBYUV_API
  480. int I010ToAB30(const uint16_t* src_y,
  481. int src_stride_y,
  482. const uint16_t* src_u,
  483. int src_stride_u,
  484. const uint16_t* src_v,
  485. int src_stride_v,
  486. uint8_t* dst_ab30,
  487. int dst_stride_ab30,
  488. int width,
  489. int height) {
  490. return I010ToAR30Matrix(src_y, src_stride_y, src_v, src_stride_v, src_u,
  491. src_stride_u, dst_ab30, dst_stride_ab30,
  492. &kYvuI601Constants, width, height);
  493. }
  494. // Convert H010 to AB30.
  495. LIBYUV_API
  496. int H010ToAB30(const uint16_t* src_y,
  497. int src_stride_y,
  498. const uint16_t* src_u,
  499. int src_stride_u,
  500. const uint16_t* src_v,
  501. int src_stride_v,
  502. uint8_t* dst_ab30,
  503. int dst_stride_ab30,
  504. int width,
  505. int height) {
  506. return I010ToAR30Matrix(src_y, src_stride_y, src_v, src_stride_v, src_u,
  507. src_stride_u, dst_ab30, dst_stride_ab30,
  508. &kYvuH709Constants, width, height);
  509. }
  510. // Convert 10 bit YUV to ARGB with matrix
  511. static int I010ToARGBMatrix(const uint16_t* src_y,
  512. int src_stride_y,
  513. const uint16_t* src_u,
  514. int src_stride_u,
  515. const uint16_t* src_v,
  516. int src_stride_v,
  517. uint8_t* dst_argb,
  518. int dst_stride_argb,
  519. const struct YuvConstants* yuvconstants,
  520. int width,
  521. int height) {
  522. int y;
  523. void (*I210ToARGBRow)(const uint16_t* y_buf, const uint16_t* u_buf,
  524. const uint16_t* v_buf, uint8_t* rgb_buf,
  525. const struct YuvConstants* yuvconstants, int width) =
  526. I210ToARGBRow_C;
  527. if (!src_y || !src_u || !src_v || !dst_argb || width <= 0 || height == 0) {
  528. return -1;
  529. }
  530. // Negative height means invert the image.
  531. if (height < 0) {
  532. height = -height;
  533. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  534. dst_stride_argb = -dst_stride_argb;
  535. }
  536. #if defined(HAS_I210TOARGBROW_SSSE3)
  537. if (TestCpuFlag(kCpuHasSSSE3)) {
  538. I210ToARGBRow = I210ToARGBRow_Any_SSSE3;
  539. if (IS_ALIGNED(width, 8)) {
  540. I210ToARGBRow = I210ToARGBRow_SSSE3;
  541. }
  542. }
  543. #endif
  544. #if defined(HAS_I210TOARGBROW_AVX2)
  545. if (TestCpuFlag(kCpuHasAVX2)) {
  546. I210ToARGBRow = I210ToARGBRow_Any_AVX2;
  547. if (IS_ALIGNED(width, 16)) {
  548. I210ToARGBRow = I210ToARGBRow_AVX2;
  549. }
  550. }
  551. #endif
  552. for (y = 0; y < height; ++y) {
  553. I210ToARGBRow(src_y, src_u, src_v, dst_argb, yuvconstants, width);
  554. dst_argb += dst_stride_argb;
  555. src_y += src_stride_y;
  556. if (y & 1) {
  557. src_u += src_stride_u;
  558. src_v += src_stride_v;
  559. }
  560. }
  561. return 0;
  562. }
  563. // Convert I010 to ARGB.
  564. LIBYUV_API
  565. int I010ToARGB(const uint16_t* src_y,
  566. int src_stride_y,
  567. const uint16_t* src_u,
  568. int src_stride_u,
  569. const uint16_t* src_v,
  570. int src_stride_v,
  571. uint8_t* dst_argb,
  572. int dst_stride_argb,
  573. int width,
  574. int height) {
  575. return I010ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  576. src_stride_v, dst_argb, dst_stride_argb,
  577. &kYuvI601Constants, width, height);
  578. }
  579. // Convert I010 to ABGR.
  580. LIBYUV_API
  581. int I010ToABGR(const uint16_t* src_y,
  582. int src_stride_y,
  583. const uint16_t* src_u,
  584. int src_stride_u,
  585. const uint16_t* src_v,
  586. int src_stride_v,
  587. uint8_t* dst_abgr,
  588. int dst_stride_abgr,
  589. int width,
  590. int height) {
  591. return I010ToARGBMatrix(src_y, src_stride_y, src_v,
  592. src_stride_v, // Swap U and V
  593. src_u, src_stride_u, dst_abgr, dst_stride_abgr,
  594. &kYvuI601Constants, // Use Yvu matrix
  595. width, height);
  596. }
  597. // Convert H010 to ARGB.
  598. LIBYUV_API
  599. int H010ToARGB(const uint16_t* src_y,
  600. int src_stride_y,
  601. const uint16_t* src_u,
  602. int src_stride_u,
  603. const uint16_t* src_v,
  604. int src_stride_v,
  605. uint8_t* dst_argb,
  606. int dst_stride_argb,
  607. int width,
  608. int height) {
  609. return I010ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  610. src_stride_v, dst_argb, dst_stride_argb,
  611. &kYuvH709Constants, width, height);
  612. }
  613. // Convert H010 to ABGR.
  614. LIBYUV_API
  615. int H010ToABGR(const uint16_t* src_y,
  616. int src_stride_y,
  617. const uint16_t* src_u,
  618. int src_stride_u,
  619. const uint16_t* src_v,
  620. int src_stride_v,
  621. uint8_t* dst_abgr,
  622. int dst_stride_abgr,
  623. int width,
  624. int height) {
  625. return I010ToARGBMatrix(src_y, src_stride_y, src_v,
  626. src_stride_v, // Swap U and V
  627. src_u, src_stride_u, dst_abgr, dst_stride_abgr,
  628. &kYvuH709Constants, // Use Yvu matrix
  629. width, height);
  630. }
  631. // Convert I444 to ARGB with matrix
  632. static int I444ToARGBMatrix(const uint8_t* src_y,
  633. int src_stride_y,
  634. const uint8_t* src_u,
  635. int src_stride_u,
  636. const uint8_t* src_v,
  637. int src_stride_v,
  638. uint8_t* dst_argb,
  639. int dst_stride_argb,
  640. const struct YuvConstants* yuvconstants,
  641. int width,
  642. int height) {
  643. int y;
  644. void (*I444ToARGBRow)(const uint8_t* y_buf, const uint8_t* u_buf,
  645. const uint8_t* v_buf, uint8_t* rgb_buf,
  646. const struct YuvConstants* yuvconstants, int width) =
  647. I444ToARGBRow_C;
  648. if (!src_y || !src_u || !src_v || !dst_argb || width <= 0 || height == 0) {
  649. return -1;
  650. }
  651. // Negative height means invert the image.
  652. if (height < 0) {
  653. height = -height;
  654. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  655. dst_stride_argb = -dst_stride_argb;
  656. }
  657. // Coalesce rows.
  658. if (src_stride_y == width && src_stride_u == width && src_stride_v == width &&
  659. dst_stride_argb == width * 4) {
  660. width *= height;
  661. height = 1;
  662. src_stride_y = src_stride_u = src_stride_v = dst_stride_argb = 0;
  663. }
  664. #if defined(HAS_I444TOARGBROW_SSSE3)
  665. if (TestCpuFlag(kCpuHasSSSE3)) {
  666. I444ToARGBRow = I444ToARGBRow_Any_SSSE3;
  667. if (IS_ALIGNED(width, 8)) {
  668. I444ToARGBRow = I444ToARGBRow_SSSE3;
  669. }
  670. }
  671. #endif
  672. #if defined(HAS_I444TOARGBROW_AVX2)
  673. if (TestCpuFlag(kCpuHasAVX2)) {
  674. I444ToARGBRow = I444ToARGBRow_Any_AVX2;
  675. if (IS_ALIGNED(width, 16)) {
  676. I444ToARGBRow = I444ToARGBRow_AVX2;
  677. }
  678. }
  679. #endif
  680. #if defined(HAS_I444TOARGBROW_NEON)
  681. if (TestCpuFlag(kCpuHasNEON)) {
  682. I444ToARGBRow = I444ToARGBRow_Any_NEON;
  683. if (IS_ALIGNED(width, 8)) {
  684. I444ToARGBRow = I444ToARGBRow_NEON;
  685. }
  686. }
  687. #endif
  688. #if defined(HAS_I444TOARGBROW_MSA)
  689. if (TestCpuFlag(kCpuHasMSA)) {
  690. I444ToARGBRow = I444ToARGBRow_Any_MSA;
  691. if (IS_ALIGNED(width, 8)) {
  692. I444ToARGBRow = I444ToARGBRow_MSA;
  693. }
  694. }
  695. #endif
  696. for (y = 0; y < height; ++y) {
  697. I444ToARGBRow(src_y, src_u, src_v, dst_argb, yuvconstants, width);
  698. dst_argb += dst_stride_argb;
  699. src_y += src_stride_y;
  700. src_u += src_stride_u;
  701. src_v += src_stride_v;
  702. }
  703. return 0;
  704. }
  705. // Convert I444 to ARGB.
  706. LIBYUV_API
  707. int I444ToARGB(const uint8_t* src_y,
  708. int src_stride_y,
  709. const uint8_t* src_u,
  710. int src_stride_u,
  711. const uint8_t* src_v,
  712. int src_stride_v,
  713. uint8_t* dst_argb,
  714. int dst_stride_argb,
  715. int width,
  716. int height) {
  717. return I444ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  718. src_stride_v, dst_argb, dst_stride_argb,
  719. &kYuvI601Constants, width, height);
  720. }
  721. // Convert I444 to ABGR.
  722. LIBYUV_API
  723. int I444ToABGR(const uint8_t* src_y,
  724. int src_stride_y,
  725. const uint8_t* src_u,
  726. int src_stride_u,
  727. const uint8_t* src_v,
  728. int src_stride_v,
  729. uint8_t* dst_abgr,
  730. int dst_stride_abgr,
  731. int width,
  732. int height) {
  733. return I444ToARGBMatrix(src_y, src_stride_y, src_v,
  734. src_stride_v, // Swap U and V
  735. src_u, src_stride_u, dst_abgr, dst_stride_abgr,
  736. &kYvuI601Constants, // Use Yvu matrix
  737. width, height);
  738. }
  739. // Convert J444 to ARGB.
  740. LIBYUV_API
  741. int J444ToARGB(const uint8_t* src_y,
  742. int src_stride_y,
  743. const uint8_t* src_u,
  744. int src_stride_u,
  745. const uint8_t* src_v,
  746. int src_stride_v,
  747. uint8_t* dst_argb,
  748. int dst_stride_argb,
  749. int width,
  750. int height) {
  751. return I444ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  752. src_stride_v, dst_argb, dst_stride_argb,
  753. &kYuvJPEGConstants, width, height);
  754. }
  755. // Convert I420 with Alpha to preattenuated ARGB.
  756. static int I420AlphaToARGBMatrix(const uint8_t* src_y,
  757. int src_stride_y,
  758. const uint8_t* src_u,
  759. int src_stride_u,
  760. const uint8_t* src_v,
  761. int src_stride_v,
  762. const uint8_t* src_a,
  763. int src_stride_a,
  764. uint8_t* dst_argb,
  765. int dst_stride_argb,
  766. const struct YuvConstants* yuvconstants,
  767. int width,
  768. int height,
  769. int attenuate) {
  770. int y;
  771. void (*I422AlphaToARGBRow)(const uint8_t* y_buf, const uint8_t* u_buf,
  772. const uint8_t* v_buf, const uint8_t* a_buf,
  773. uint8_t* dst_argb,
  774. const struct YuvConstants* yuvconstants,
  775. int width) = I422AlphaToARGBRow_C;
  776. void (*ARGBAttenuateRow)(const uint8_t* src_argb, uint8_t* dst_argb,
  777. int width) = ARGBAttenuateRow_C;
  778. if (!src_y || !src_u || !src_v || !dst_argb || width <= 0 || height == 0) {
  779. return -1;
  780. }
  781. // Negative height means invert the image.
  782. if (height < 0) {
  783. height = -height;
  784. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  785. dst_stride_argb = -dst_stride_argb;
  786. }
  787. #if defined(HAS_I422ALPHATOARGBROW_SSSE3)
  788. if (TestCpuFlag(kCpuHasSSSE3)) {
  789. I422AlphaToARGBRow = I422AlphaToARGBRow_Any_SSSE3;
  790. if (IS_ALIGNED(width, 8)) {
  791. I422AlphaToARGBRow = I422AlphaToARGBRow_SSSE3;
  792. }
  793. }
  794. #endif
  795. #if defined(HAS_I422ALPHATOARGBROW_AVX2)
  796. if (TestCpuFlag(kCpuHasAVX2)) {
  797. I422AlphaToARGBRow = I422AlphaToARGBRow_Any_AVX2;
  798. if (IS_ALIGNED(width, 16)) {
  799. I422AlphaToARGBRow = I422AlphaToARGBRow_AVX2;
  800. }
  801. }
  802. #endif
  803. #if defined(HAS_I422ALPHATOARGBROW_NEON)
  804. if (TestCpuFlag(kCpuHasNEON)) {
  805. I422AlphaToARGBRow = I422AlphaToARGBRow_Any_NEON;
  806. if (IS_ALIGNED(width, 8)) {
  807. I422AlphaToARGBRow = I422AlphaToARGBRow_NEON;
  808. }
  809. }
  810. #endif
  811. #if defined(HAS_I422ALPHATOARGBROW_MSA)
  812. if (TestCpuFlag(kCpuHasMSA)) {
  813. I422AlphaToARGBRow = I422AlphaToARGBRow_Any_MSA;
  814. if (IS_ALIGNED(width, 8)) {
  815. I422AlphaToARGBRow = I422AlphaToARGBRow_MSA;
  816. }
  817. }
  818. #endif
  819. #if defined(HAS_ARGBATTENUATEROW_SSSE3)
  820. if (TestCpuFlag(kCpuHasSSSE3)) {
  821. ARGBAttenuateRow = ARGBAttenuateRow_Any_SSSE3;
  822. if (IS_ALIGNED(width, 4)) {
  823. ARGBAttenuateRow = ARGBAttenuateRow_SSSE3;
  824. }
  825. }
  826. #endif
  827. #if defined(HAS_ARGBATTENUATEROW_AVX2)
  828. if (TestCpuFlag(kCpuHasAVX2)) {
  829. ARGBAttenuateRow = ARGBAttenuateRow_Any_AVX2;
  830. if (IS_ALIGNED(width, 8)) {
  831. ARGBAttenuateRow = ARGBAttenuateRow_AVX2;
  832. }
  833. }
  834. #endif
  835. #if defined(HAS_ARGBATTENUATEROW_NEON)
  836. if (TestCpuFlag(kCpuHasNEON)) {
  837. ARGBAttenuateRow = ARGBAttenuateRow_Any_NEON;
  838. if (IS_ALIGNED(width, 8)) {
  839. ARGBAttenuateRow = ARGBAttenuateRow_NEON;
  840. }
  841. }
  842. #endif
  843. #if defined(HAS_ARGBATTENUATEROW_MSA)
  844. if (TestCpuFlag(kCpuHasMSA)) {
  845. ARGBAttenuateRow = ARGBAttenuateRow_Any_MSA;
  846. if (IS_ALIGNED(width, 8)) {
  847. ARGBAttenuateRow = ARGBAttenuateRow_MSA;
  848. }
  849. }
  850. #endif
  851. #if defined(HAS_ARGBATTENUATEROW_MMI)
  852. if (TestCpuFlag(kCpuHasMMI)) {
  853. ARGBAttenuateRow = ARGBAttenuateRow_Any_MMI;
  854. if (IS_ALIGNED(width, 2)) {
  855. ARGBAttenuateRow = ARGBAttenuateRow_MMI;
  856. }
  857. }
  858. #endif
  859. for (y = 0; y < height; ++y) {
  860. I422AlphaToARGBRow(src_y, src_u, src_v, src_a, dst_argb, yuvconstants,
  861. width);
  862. if (attenuate) {
  863. ARGBAttenuateRow(dst_argb, dst_argb, width);
  864. }
  865. dst_argb += dst_stride_argb;
  866. src_a += src_stride_a;
  867. src_y += src_stride_y;
  868. if (y & 1) {
  869. src_u += src_stride_u;
  870. src_v += src_stride_v;
  871. }
  872. }
  873. return 0;
  874. }
  875. // Convert I420 with Alpha to ARGB.
  876. LIBYUV_API
  877. int I420AlphaToARGB(const uint8_t* src_y,
  878. int src_stride_y,
  879. const uint8_t* src_u,
  880. int src_stride_u,
  881. const uint8_t* src_v,
  882. int src_stride_v,
  883. const uint8_t* src_a,
  884. int src_stride_a,
  885. uint8_t* dst_argb,
  886. int dst_stride_argb,
  887. int width,
  888. int height,
  889. int attenuate) {
  890. return I420AlphaToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  891. src_stride_v, src_a, src_stride_a, dst_argb,
  892. dst_stride_argb, &kYuvI601Constants, width,
  893. height, attenuate);
  894. }
  895. // Convert I420 with Alpha to ABGR.
  896. LIBYUV_API
  897. int I420AlphaToABGR(const uint8_t* src_y,
  898. int src_stride_y,
  899. const uint8_t* src_u,
  900. int src_stride_u,
  901. const uint8_t* src_v,
  902. int src_stride_v,
  903. const uint8_t* src_a,
  904. int src_stride_a,
  905. uint8_t* dst_abgr,
  906. int dst_stride_abgr,
  907. int width,
  908. int height,
  909. int attenuate) {
  910. return I420AlphaToARGBMatrix(
  911. src_y, src_stride_y, src_v, src_stride_v, // Swap U and V
  912. src_u, src_stride_u, src_a, src_stride_a, dst_abgr, dst_stride_abgr,
  913. &kYvuI601Constants, // Use Yvu matrix
  914. width, height, attenuate);
  915. }
  916. // Convert I400 to ARGB.
  917. LIBYUV_API
  918. int I400ToARGB(const uint8_t* src_y,
  919. int src_stride_y,
  920. uint8_t* dst_argb,
  921. int dst_stride_argb,
  922. int width,
  923. int height) {
  924. int y;
  925. void (*I400ToARGBRow)(const uint8_t* y_buf, uint8_t* rgb_buf, int width) =
  926. I400ToARGBRow_C;
  927. if (!src_y || !dst_argb || width <= 0 || height == 0) {
  928. return -1;
  929. }
  930. // Negative height means invert the image.
  931. if (height < 0) {
  932. height = -height;
  933. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  934. dst_stride_argb = -dst_stride_argb;
  935. }
  936. // Coalesce rows.
  937. if (src_stride_y == width && dst_stride_argb == width * 4) {
  938. width *= height;
  939. height = 1;
  940. src_stride_y = dst_stride_argb = 0;
  941. }
  942. #if defined(HAS_I400TOARGBROW_SSE2)
  943. if (TestCpuFlag(kCpuHasSSE2)) {
  944. I400ToARGBRow = I400ToARGBRow_Any_SSE2;
  945. if (IS_ALIGNED(width, 8)) {
  946. I400ToARGBRow = I400ToARGBRow_SSE2;
  947. }
  948. }
  949. #endif
  950. #if defined(HAS_I400TOARGBROW_AVX2)
  951. if (TestCpuFlag(kCpuHasAVX2)) {
  952. I400ToARGBRow = I400ToARGBRow_Any_AVX2;
  953. if (IS_ALIGNED(width, 16)) {
  954. I400ToARGBRow = I400ToARGBRow_AVX2;
  955. }
  956. }
  957. #endif
  958. #if defined(HAS_I400TOARGBROW_NEON)
  959. if (TestCpuFlag(kCpuHasNEON)) {
  960. I400ToARGBRow = I400ToARGBRow_Any_NEON;
  961. if (IS_ALIGNED(width, 8)) {
  962. I400ToARGBRow = I400ToARGBRow_NEON;
  963. }
  964. }
  965. #endif
  966. #if defined(HAS_I400TOARGBROW_MSA)
  967. if (TestCpuFlag(kCpuHasMSA)) {
  968. I400ToARGBRow = I400ToARGBRow_Any_MSA;
  969. if (IS_ALIGNED(width, 16)) {
  970. I400ToARGBRow = I400ToARGBRow_MSA;
  971. }
  972. }
  973. #endif
  974. #if defined(HAS_I400TOARGBROW_MMI)
  975. if (TestCpuFlag(kCpuHasMMI)) {
  976. I400ToARGBRow = I400ToARGBRow_Any_MMI;
  977. if (IS_ALIGNED(width, 8)) {
  978. I400ToARGBRow = I400ToARGBRow_MMI;
  979. }
  980. }
  981. #endif
  982. for (y = 0; y < height; ++y) {
  983. I400ToARGBRow(src_y, dst_argb, width);
  984. dst_argb += dst_stride_argb;
  985. src_y += src_stride_y;
  986. }
  987. return 0;
  988. }
  989. // Convert J400 to ARGB.
  990. LIBYUV_API
  991. int J400ToARGB(const uint8_t* src_y,
  992. int src_stride_y,
  993. uint8_t* dst_argb,
  994. int dst_stride_argb,
  995. int width,
  996. int height) {
  997. int y;
  998. void (*J400ToARGBRow)(const uint8_t* src_y, uint8_t* dst_argb, int width) =
  999. J400ToARGBRow_C;
  1000. if (!src_y || !dst_argb || width <= 0 || height == 0) {
  1001. return -1;
  1002. }
  1003. // Negative height means invert the image.
  1004. if (height < 0) {
  1005. height = -height;
  1006. src_y = src_y + (height - 1) * src_stride_y;
  1007. src_stride_y = -src_stride_y;
  1008. }
  1009. // Coalesce rows.
  1010. if (src_stride_y == width && dst_stride_argb == width * 4) {
  1011. width *= height;
  1012. height = 1;
  1013. src_stride_y = dst_stride_argb = 0;
  1014. }
  1015. #if defined(HAS_J400TOARGBROW_SSE2)
  1016. if (TestCpuFlag(kCpuHasSSE2)) {
  1017. J400ToARGBRow = J400ToARGBRow_Any_SSE2;
  1018. if (IS_ALIGNED(width, 8)) {
  1019. J400ToARGBRow = J400ToARGBRow_SSE2;
  1020. }
  1021. }
  1022. #endif
  1023. #if defined(HAS_J400TOARGBROW_AVX2)
  1024. if (TestCpuFlag(kCpuHasAVX2)) {
  1025. J400ToARGBRow = J400ToARGBRow_Any_AVX2;
  1026. if (IS_ALIGNED(width, 16)) {
  1027. J400ToARGBRow = J400ToARGBRow_AVX2;
  1028. }
  1029. }
  1030. #endif
  1031. #if defined(HAS_J400TOARGBROW_NEON)
  1032. if (TestCpuFlag(kCpuHasNEON)) {
  1033. J400ToARGBRow = J400ToARGBRow_Any_NEON;
  1034. if (IS_ALIGNED(width, 8)) {
  1035. J400ToARGBRow = J400ToARGBRow_NEON;
  1036. }
  1037. }
  1038. #endif
  1039. #if defined(HAS_J400TOARGBROW_MSA)
  1040. if (TestCpuFlag(kCpuHasMSA)) {
  1041. J400ToARGBRow = J400ToARGBRow_Any_MSA;
  1042. if (IS_ALIGNED(width, 16)) {
  1043. J400ToARGBRow = J400ToARGBRow_MSA;
  1044. }
  1045. }
  1046. #endif
  1047. #if defined(HAS_J400TOARGBROW_MMI)
  1048. if (TestCpuFlag(kCpuHasMMI)) {
  1049. J400ToARGBRow = J400ToARGBRow_Any_MMI;
  1050. if (IS_ALIGNED(width, 4)) {
  1051. J400ToARGBRow = J400ToARGBRow_MMI;
  1052. }
  1053. }
  1054. #endif
  1055. for (y = 0; y < height; ++y) {
  1056. J400ToARGBRow(src_y, dst_argb, width);
  1057. src_y += src_stride_y;
  1058. dst_argb += dst_stride_argb;
  1059. }
  1060. return 0;
  1061. }
  1062. // Shuffle table for converting BGRA to ARGB.
  1063. static const uvec8 kShuffleMaskBGRAToARGB = {
  1064. 3u, 2u, 1u, 0u, 7u, 6u, 5u, 4u, 11u, 10u, 9u, 8u, 15u, 14u, 13u, 12u};
  1065. // Shuffle table for converting ABGR to ARGB.
  1066. static const uvec8 kShuffleMaskABGRToARGB = {
  1067. 2u, 1u, 0u, 3u, 6u, 5u, 4u, 7u, 10u, 9u, 8u, 11u, 14u, 13u, 12u, 15u};
  1068. // Shuffle table for converting RGBA to ARGB.
  1069. static const uvec8 kShuffleMaskRGBAToARGB = {
  1070. 1u, 2u, 3u, 0u, 5u, 6u, 7u, 4u, 9u, 10u, 11u, 8u, 13u, 14u, 15u, 12u};
  1071. // Convert BGRA to ARGB.
  1072. LIBYUV_API
  1073. int BGRAToARGB(const uint8_t* src_bgra,
  1074. int src_stride_bgra,
  1075. uint8_t* dst_argb,
  1076. int dst_stride_argb,
  1077. int width,
  1078. int height) {
  1079. return ARGBShuffle(src_bgra, src_stride_bgra, dst_argb, dst_stride_argb,
  1080. (const uint8_t*)(&kShuffleMaskBGRAToARGB), width, height);
  1081. }
  1082. // Convert ARGB to BGRA (same as BGRAToARGB).
  1083. LIBYUV_API
  1084. int ARGBToBGRA(const uint8_t* src_bgra,
  1085. int src_stride_bgra,
  1086. uint8_t* dst_argb,
  1087. int dst_stride_argb,
  1088. int width,
  1089. int height) {
  1090. return ARGBShuffle(src_bgra, src_stride_bgra, dst_argb, dst_stride_argb,
  1091. (const uint8_t*)(&kShuffleMaskBGRAToARGB), width, height);
  1092. }
  1093. // Convert ABGR to ARGB.
  1094. LIBYUV_API
  1095. int ABGRToARGB(const uint8_t* src_abgr,
  1096. int src_stride_abgr,
  1097. uint8_t* dst_argb,
  1098. int dst_stride_argb,
  1099. int width,
  1100. int height) {
  1101. return ARGBShuffle(src_abgr, src_stride_abgr, dst_argb, dst_stride_argb,
  1102. (const uint8_t*)(&kShuffleMaskABGRToARGB), width, height);
  1103. }
  1104. // Convert ARGB to ABGR to (same as ABGRToARGB).
  1105. LIBYUV_API
  1106. int ARGBToABGR(const uint8_t* src_abgr,
  1107. int src_stride_abgr,
  1108. uint8_t* dst_argb,
  1109. int dst_stride_argb,
  1110. int width,
  1111. int height) {
  1112. return ARGBShuffle(src_abgr, src_stride_abgr, dst_argb, dst_stride_argb,
  1113. (const uint8_t*)(&kShuffleMaskABGRToARGB), width, height);
  1114. }
  1115. // Convert RGBA to ARGB.
  1116. LIBYUV_API
  1117. int RGBAToARGB(const uint8_t* src_rgba,
  1118. int src_stride_rgba,
  1119. uint8_t* dst_argb,
  1120. int dst_stride_argb,
  1121. int width,
  1122. int height) {
  1123. return ARGBShuffle(src_rgba, src_stride_rgba, dst_argb, dst_stride_argb,
  1124. (const uint8_t*)(&kShuffleMaskRGBAToARGB), width, height);
  1125. }
  1126. // Convert RGB24 to ARGB.
  1127. LIBYUV_API
  1128. int RGB24ToARGB(const uint8_t* src_rgb24,
  1129. int src_stride_rgb24,
  1130. uint8_t* dst_argb,
  1131. int dst_stride_argb,
  1132. int width,
  1133. int height) {
  1134. int y;
  1135. void (*RGB24ToARGBRow)(const uint8_t* src_rgb, uint8_t* dst_argb, int width) =
  1136. RGB24ToARGBRow_C;
  1137. if (!src_rgb24 || !dst_argb || width <= 0 || height == 0) {
  1138. return -1;
  1139. }
  1140. // Negative height means invert the image.
  1141. if (height < 0) {
  1142. height = -height;
  1143. src_rgb24 = src_rgb24 + (height - 1) * src_stride_rgb24;
  1144. src_stride_rgb24 = -src_stride_rgb24;
  1145. }
  1146. // Coalesce rows.
  1147. if (src_stride_rgb24 == width * 3 && dst_stride_argb == width * 4) {
  1148. width *= height;
  1149. height = 1;
  1150. src_stride_rgb24 = dst_stride_argb = 0;
  1151. }
  1152. #if defined(HAS_RGB24TOARGBROW_SSSE3)
  1153. if (TestCpuFlag(kCpuHasSSSE3)) {
  1154. RGB24ToARGBRow = RGB24ToARGBRow_Any_SSSE3;
  1155. if (IS_ALIGNED(width, 16)) {
  1156. RGB24ToARGBRow = RGB24ToARGBRow_SSSE3;
  1157. }
  1158. }
  1159. #endif
  1160. #if defined(HAS_RGB24TOARGBROW_NEON)
  1161. if (TestCpuFlag(kCpuHasNEON)) {
  1162. RGB24ToARGBRow = RGB24ToARGBRow_Any_NEON;
  1163. if (IS_ALIGNED(width, 8)) {
  1164. RGB24ToARGBRow = RGB24ToARGBRow_NEON;
  1165. }
  1166. }
  1167. #endif
  1168. #if defined(HAS_RGB24TOARGBROW_MSA)
  1169. if (TestCpuFlag(kCpuHasMSA)) {
  1170. RGB24ToARGBRow = RGB24ToARGBRow_Any_MSA;
  1171. if (IS_ALIGNED(width, 16)) {
  1172. RGB24ToARGBRow = RGB24ToARGBRow_MSA;
  1173. }
  1174. }
  1175. #endif
  1176. #if defined(HAS_RGB24TOARGBROW_MMI)
  1177. if (TestCpuFlag(kCpuHasMMI)) {
  1178. RGB24ToARGBRow = RGB24ToARGBRow_Any_MMI;
  1179. if (IS_ALIGNED(width, 4)) {
  1180. RGB24ToARGBRow = RGB24ToARGBRow_MMI;
  1181. }
  1182. }
  1183. #endif
  1184. for (y = 0; y < height; ++y) {
  1185. RGB24ToARGBRow(src_rgb24, dst_argb, width);
  1186. src_rgb24 += src_stride_rgb24;
  1187. dst_argb += dst_stride_argb;
  1188. }
  1189. return 0;
  1190. }
  1191. // Convert RAW to ARGB.
  1192. LIBYUV_API
  1193. int RAWToARGB(const uint8_t* src_raw,
  1194. int src_stride_raw,
  1195. uint8_t* dst_argb,
  1196. int dst_stride_argb,
  1197. int width,
  1198. int height) {
  1199. int y;
  1200. void (*RAWToARGBRow)(const uint8_t* src_rgb, uint8_t* dst_argb, int width) =
  1201. RAWToARGBRow_C;
  1202. if (!src_raw || !dst_argb || width <= 0 || height == 0) {
  1203. return -1;
  1204. }
  1205. // Negative height means invert the image.
  1206. if (height < 0) {
  1207. height = -height;
  1208. src_raw = src_raw + (height - 1) * src_stride_raw;
  1209. src_stride_raw = -src_stride_raw;
  1210. }
  1211. // Coalesce rows.
  1212. if (src_stride_raw == width * 3 && dst_stride_argb == width * 4) {
  1213. width *= height;
  1214. height = 1;
  1215. src_stride_raw = dst_stride_argb = 0;
  1216. }
  1217. #if defined(HAS_RAWTOARGBROW_SSSE3)
  1218. if (TestCpuFlag(kCpuHasSSSE3)) {
  1219. RAWToARGBRow = RAWToARGBRow_Any_SSSE3;
  1220. if (IS_ALIGNED(width, 16)) {
  1221. RAWToARGBRow = RAWToARGBRow_SSSE3;
  1222. }
  1223. }
  1224. #endif
  1225. #if defined(HAS_RAWTOARGBROW_NEON)
  1226. if (TestCpuFlag(kCpuHasNEON)) {
  1227. RAWToARGBRow = RAWToARGBRow_Any_NEON;
  1228. if (IS_ALIGNED(width, 8)) {
  1229. RAWToARGBRow = RAWToARGBRow_NEON;
  1230. }
  1231. }
  1232. #endif
  1233. #if defined(HAS_RAWTOARGBROW_MSA)
  1234. if (TestCpuFlag(kCpuHasMSA)) {
  1235. RAWToARGBRow = RAWToARGBRow_Any_MSA;
  1236. if (IS_ALIGNED(width, 16)) {
  1237. RAWToARGBRow = RAWToARGBRow_MSA;
  1238. }
  1239. }
  1240. #endif
  1241. #if defined(HAS_RAWTOARGBROW_MMI)
  1242. if (TestCpuFlag(kCpuHasMMI)) {
  1243. RAWToARGBRow = RAWToARGBRow_Any_MMI;
  1244. if (IS_ALIGNED(width, 4)) {
  1245. RAWToARGBRow = RAWToARGBRow_MMI;
  1246. }
  1247. }
  1248. #endif
  1249. for (y = 0; y < height; ++y) {
  1250. RAWToARGBRow(src_raw, dst_argb, width);
  1251. src_raw += src_stride_raw;
  1252. dst_argb += dst_stride_argb;
  1253. }
  1254. return 0;
  1255. }
  1256. // Convert RGB565 to ARGB.
  1257. LIBYUV_API
  1258. int RGB565ToARGB(const uint8_t* src_rgb565,
  1259. int src_stride_rgb565,
  1260. uint8_t* dst_argb,
  1261. int dst_stride_argb,
  1262. int width,
  1263. int height) {
  1264. int y;
  1265. void (*RGB565ToARGBRow)(const uint8_t* src_rgb565, uint8_t* dst_argb,
  1266. int width) = RGB565ToARGBRow_C;
  1267. if (!src_rgb565 || !dst_argb || width <= 0 || height == 0) {
  1268. return -1;
  1269. }
  1270. // Negative height means invert the image.
  1271. if (height < 0) {
  1272. height = -height;
  1273. src_rgb565 = src_rgb565 + (height - 1) * src_stride_rgb565;
  1274. src_stride_rgb565 = -src_stride_rgb565;
  1275. }
  1276. // Coalesce rows.
  1277. if (src_stride_rgb565 == width * 2 && dst_stride_argb == width * 4) {
  1278. width *= height;
  1279. height = 1;
  1280. src_stride_rgb565 = dst_stride_argb = 0;
  1281. }
  1282. #if defined(HAS_RGB565TOARGBROW_SSE2)
  1283. if (TestCpuFlag(kCpuHasSSE2)) {
  1284. RGB565ToARGBRow = RGB565ToARGBRow_Any_SSE2;
  1285. if (IS_ALIGNED(width, 8)) {
  1286. RGB565ToARGBRow = RGB565ToARGBRow_SSE2;
  1287. }
  1288. }
  1289. #endif
  1290. #if defined(HAS_RGB565TOARGBROW_AVX2)
  1291. if (TestCpuFlag(kCpuHasAVX2)) {
  1292. RGB565ToARGBRow = RGB565ToARGBRow_Any_AVX2;
  1293. if (IS_ALIGNED(width, 16)) {
  1294. RGB565ToARGBRow = RGB565ToARGBRow_AVX2;
  1295. }
  1296. }
  1297. #endif
  1298. #if defined(HAS_RGB565TOARGBROW_NEON)
  1299. if (TestCpuFlag(kCpuHasNEON)) {
  1300. RGB565ToARGBRow = RGB565ToARGBRow_Any_NEON;
  1301. if (IS_ALIGNED(width, 8)) {
  1302. RGB565ToARGBRow = RGB565ToARGBRow_NEON;
  1303. }
  1304. }
  1305. #endif
  1306. #if defined(HAS_RGB565TOARGBROW_MSA)
  1307. if (TestCpuFlag(kCpuHasMSA)) {
  1308. RGB565ToARGBRow = RGB565ToARGBRow_Any_MSA;
  1309. if (IS_ALIGNED(width, 16)) {
  1310. RGB565ToARGBRow = RGB565ToARGBRow_MSA;
  1311. }
  1312. }
  1313. #endif
  1314. #if defined(HAS_RGB565TOARGBROW_MMI)
  1315. if (TestCpuFlag(kCpuHasMMI)) {
  1316. RGB565ToARGBRow = RGB565ToARGBRow_Any_MMI;
  1317. if (IS_ALIGNED(width, 4)) {
  1318. RGB565ToARGBRow = RGB565ToARGBRow_MMI;
  1319. }
  1320. }
  1321. #endif
  1322. for (y = 0; y < height; ++y) {
  1323. RGB565ToARGBRow(src_rgb565, dst_argb, width);
  1324. src_rgb565 += src_stride_rgb565;
  1325. dst_argb += dst_stride_argb;
  1326. }
  1327. return 0;
  1328. }
  1329. // Convert ARGB1555 to ARGB.
  1330. LIBYUV_API
  1331. int ARGB1555ToARGB(const uint8_t* src_argb1555,
  1332. int src_stride_argb1555,
  1333. uint8_t* dst_argb,
  1334. int dst_stride_argb,
  1335. int width,
  1336. int height) {
  1337. int y;
  1338. void (*ARGB1555ToARGBRow)(const uint8_t* src_argb1555, uint8_t* dst_argb,
  1339. int width) = ARGB1555ToARGBRow_C;
  1340. if (!src_argb1555 || !dst_argb || width <= 0 || height == 0) {
  1341. return -1;
  1342. }
  1343. // Negative height means invert the image.
  1344. if (height < 0) {
  1345. height = -height;
  1346. src_argb1555 = src_argb1555 + (height - 1) * src_stride_argb1555;
  1347. src_stride_argb1555 = -src_stride_argb1555;
  1348. }
  1349. // Coalesce rows.
  1350. if (src_stride_argb1555 == width * 2 && dst_stride_argb == width * 4) {
  1351. width *= height;
  1352. height = 1;
  1353. src_stride_argb1555 = dst_stride_argb = 0;
  1354. }
  1355. #if defined(HAS_ARGB1555TOARGBROW_SSE2)
  1356. if (TestCpuFlag(kCpuHasSSE2)) {
  1357. ARGB1555ToARGBRow = ARGB1555ToARGBRow_Any_SSE2;
  1358. if (IS_ALIGNED(width, 8)) {
  1359. ARGB1555ToARGBRow = ARGB1555ToARGBRow_SSE2;
  1360. }
  1361. }
  1362. #endif
  1363. #if defined(HAS_ARGB1555TOARGBROW_AVX2)
  1364. if (TestCpuFlag(kCpuHasAVX2)) {
  1365. ARGB1555ToARGBRow = ARGB1555ToARGBRow_Any_AVX2;
  1366. if (IS_ALIGNED(width, 16)) {
  1367. ARGB1555ToARGBRow = ARGB1555ToARGBRow_AVX2;
  1368. }
  1369. }
  1370. #endif
  1371. #if defined(HAS_ARGB1555TOARGBROW_NEON)
  1372. if (TestCpuFlag(kCpuHasNEON)) {
  1373. ARGB1555ToARGBRow = ARGB1555ToARGBRow_Any_NEON;
  1374. if (IS_ALIGNED(width, 8)) {
  1375. ARGB1555ToARGBRow = ARGB1555ToARGBRow_NEON;
  1376. }
  1377. }
  1378. #endif
  1379. #if defined(HAS_ARGB1555TOARGBROW_MSA)
  1380. if (TestCpuFlag(kCpuHasMSA)) {
  1381. ARGB1555ToARGBRow = ARGB1555ToARGBRow_Any_MSA;
  1382. if (IS_ALIGNED(width, 16)) {
  1383. ARGB1555ToARGBRow = ARGB1555ToARGBRow_MSA;
  1384. }
  1385. }
  1386. #endif
  1387. #if defined(HAS_ARGB1555TOARGBROW_MMI)
  1388. if (TestCpuFlag(kCpuHasMMI)) {
  1389. ARGB1555ToARGBRow = ARGB1555ToARGBRow_Any_MMI;
  1390. if (IS_ALIGNED(width, 4)) {
  1391. ARGB1555ToARGBRow = ARGB1555ToARGBRow_MMI;
  1392. }
  1393. }
  1394. #endif
  1395. for (y = 0; y < height; ++y) {
  1396. ARGB1555ToARGBRow(src_argb1555, dst_argb, width);
  1397. src_argb1555 += src_stride_argb1555;
  1398. dst_argb += dst_stride_argb;
  1399. }
  1400. return 0;
  1401. }
  1402. // Convert ARGB4444 to ARGB.
  1403. LIBYUV_API
  1404. int ARGB4444ToARGB(const uint8_t* src_argb4444,
  1405. int src_stride_argb4444,
  1406. uint8_t* dst_argb,
  1407. int dst_stride_argb,
  1408. int width,
  1409. int height) {
  1410. int y;
  1411. void (*ARGB4444ToARGBRow)(const uint8_t* src_argb4444, uint8_t* dst_argb,
  1412. int width) = ARGB4444ToARGBRow_C;
  1413. if (!src_argb4444 || !dst_argb || width <= 0 || height == 0) {
  1414. return -1;
  1415. }
  1416. // Negative height means invert the image.
  1417. if (height < 0) {
  1418. height = -height;
  1419. src_argb4444 = src_argb4444 + (height - 1) * src_stride_argb4444;
  1420. src_stride_argb4444 = -src_stride_argb4444;
  1421. }
  1422. // Coalesce rows.
  1423. if (src_stride_argb4444 == width * 2 && dst_stride_argb == width * 4) {
  1424. width *= height;
  1425. height = 1;
  1426. src_stride_argb4444 = dst_stride_argb = 0;
  1427. }
  1428. #if defined(HAS_ARGB4444TOARGBROW_SSE2)
  1429. if (TestCpuFlag(kCpuHasSSE2)) {
  1430. ARGB4444ToARGBRow = ARGB4444ToARGBRow_Any_SSE2;
  1431. if (IS_ALIGNED(width, 8)) {
  1432. ARGB4444ToARGBRow = ARGB4444ToARGBRow_SSE2;
  1433. }
  1434. }
  1435. #endif
  1436. #if defined(HAS_ARGB4444TOARGBROW_AVX2)
  1437. if (TestCpuFlag(kCpuHasAVX2)) {
  1438. ARGB4444ToARGBRow = ARGB4444ToARGBRow_Any_AVX2;
  1439. if (IS_ALIGNED(width, 16)) {
  1440. ARGB4444ToARGBRow = ARGB4444ToARGBRow_AVX2;
  1441. }
  1442. }
  1443. #endif
  1444. #if defined(HAS_ARGB4444TOARGBROW_NEON)
  1445. if (TestCpuFlag(kCpuHasNEON)) {
  1446. ARGB4444ToARGBRow = ARGB4444ToARGBRow_Any_NEON;
  1447. if (IS_ALIGNED(width, 8)) {
  1448. ARGB4444ToARGBRow = ARGB4444ToARGBRow_NEON;
  1449. }
  1450. }
  1451. #endif
  1452. #if defined(HAS_ARGB4444TOARGBROW_MSA)
  1453. if (TestCpuFlag(kCpuHasMSA)) {
  1454. ARGB4444ToARGBRow = ARGB4444ToARGBRow_Any_MSA;
  1455. if (IS_ALIGNED(width, 16)) {
  1456. ARGB4444ToARGBRow = ARGB4444ToARGBRow_MSA;
  1457. }
  1458. }
  1459. #endif
  1460. #if defined(HAS_ARGB4444TOARGBROW_MMI)
  1461. if (TestCpuFlag(kCpuHasMMI)) {
  1462. ARGB4444ToARGBRow = ARGB4444ToARGBRow_Any_MMI;
  1463. if (IS_ALIGNED(width, 4)) {
  1464. ARGB4444ToARGBRow = ARGB4444ToARGBRow_MMI;
  1465. }
  1466. }
  1467. #endif
  1468. for (y = 0; y < height; ++y) {
  1469. ARGB4444ToARGBRow(src_argb4444, dst_argb, width);
  1470. src_argb4444 += src_stride_argb4444;
  1471. dst_argb += dst_stride_argb;
  1472. }
  1473. return 0;
  1474. }
  1475. // Convert AR30 to ARGB.
  1476. LIBYUV_API
  1477. int AR30ToARGB(const uint8_t* src_ar30,
  1478. int src_stride_ar30,
  1479. uint8_t* dst_argb,
  1480. int dst_stride_argb,
  1481. int width,
  1482. int height) {
  1483. int y;
  1484. if (!src_ar30 || !dst_argb || width <= 0 || height == 0) {
  1485. return -1;
  1486. }
  1487. // Negative height means invert the image.
  1488. if (height < 0) {
  1489. height = -height;
  1490. src_ar30 = src_ar30 + (height - 1) * src_stride_ar30;
  1491. src_stride_ar30 = -src_stride_ar30;
  1492. }
  1493. // Coalesce rows.
  1494. if (src_stride_ar30 == width * 4 && dst_stride_argb == width * 4) {
  1495. width *= height;
  1496. height = 1;
  1497. src_stride_ar30 = dst_stride_argb = 0;
  1498. }
  1499. for (y = 0; y < height; ++y) {
  1500. AR30ToARGBRow_C(src_ar30, dst_argb, width);
  1501. src_ar30 += src_stride_ar30;
  1502. dst_argb += dst_stride_argb;
  1503. }
  1504. return 0;
  1505. }
  1506. // Convert AR30 to ABGR.
  1507. LIBYUV_API
  1508. int AR30ToABGR(const uint8_t* src_ar30,
  1509. int src_stride_ar30,
  1510. uint8_t* dst_abgr,
  1511. int dst_stride_abgr,
  1512. int width,
  1513. int height) {
  1514. int y;
  1515. if (!src_ar30 || !dst_abgr || width <= 0 || height == 0) {
  1516. return -1;
  1517. }
  1518. // Negative height means invert the image.
  1519. if (height < 0) {
  1520. height = -height;
  1521. src_ar30 = src_ar30 + (height - 1) * src_stride_ar30;
  1522. src_stride_ar30 = -src_stride_ar30;
  1523. }
  1524. // Coalesce rows.
  1525. if (src_stride_ar30 == width * 4 && dst_stride_abgr == width * 4) {
  1526. width *= height;
  1527. height = 1;
  1528. src_stride_ar30 = dst_stride_abgr = 0;
  1529. }
  1530. for (y = 0; y < height; ++y) {
  1531. AR30ToABGRRow_C(src_ar30, dst_abgr, width);
  1532. src_ar30 += src_stride_ar30;
  1533. dst_abgr += dst_stride_abgr;
  1534. }
  1535. return 0;
  1536. }
  1537. // Convert AR30 to AB30.
  1538. LIBYUV_API
  1539. int AR30ToAB30(const uint8_t* src_ar30,
  1540. int src_stride_ar30,
  1541. uint8_t* dst_ab30,
  1542. int dst_stride_ab30,
  1543. int width,
  1544. int height) {
  1545. int y;
  1546. if (!src_ar30 || !dst_ab30 || width <= 0 || height == 0) {
  1547. return -1;
  1548. }
  1549. // Negative height means invert the image.
  1550. if (height < 0) {
  1551. height = -height;
  1552. src_ar30 = src_ar30 + (height - 1) * src_stride_ar30;
  1553. src_stride_ar30 = -src_stride_ar30;
  1554. }
  1555. // Coalesce rows.
  1556. if (src_stride_ar30 == width * 4 && dst_stride_ab30 == width * 4) {
  1557. width *= height;
  1558. height = 1;
  1559. src_stride_ar30 = dst_stride_ab30 = 0;
  1560. }
  1561. for (y = 0; y < height; ++y) {
  1562. AR30ToAB30Row_C(src_ar30, dst_ab30, width);
  1563. src_ar30 += src_stride_ar30;
  1564. dst_ab30 += dst_stride_ab30;
  1565. }
  1566. return 0;
  1567. }
  1568. // Convert NV12 to ARGB with matrix
  1569. static int NV12ToARGBMatrix(const uint8_t* src_y,
  1570. int src_stride_y,
  1571. const uint8_t* src_uv,
  1572. int src_stride_uv,
  1573. uint8_t* dst_argb,
  1574. int dst_stride_argb,
  1575. const struct YuvConstants* yuvconstants,
  1576. int width,
  1577. int height) {
  1578. int y;
  1579. void (*NV12ToARGBRow)(
  1580. const uint8_t* y_buf, const uint8_t* uv_buf, uint8_t* rgb_buf,
  1581. const struct YuvConstants* yuvconstants, int width) = NV12ToARGBRow_C;
  1582. if (!src_y || !src_uv || !dst_argb || width <= 0 || height == 0) {
  1583. return -1;
  1584. }
  1585. // Negative height means invert the image.
  1586. if (height < 0) {
  1587. height = -height;
  1588. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  1589. dst_stride_argb = -dst_stride_argb;
  1590. }
  1591. #if defined(HAS_NV12TOARGBROW_SSSE3)
  1592. if (TestCpuFlag(kCpuHasSSSE3)) {
  1593. NV12ToARGBRow = NV12ToARGBRow_Any_SSSE3;
  1594. if (IS_ALIGNED(width, 8)) {
  1595. NV12ToARGBRow = NV12ToARGBRow_SSSE3;
  1596. }
  1597. }
  1598. #endif
  1599. #if defined(HAS_NV12TOARGBROW_AVX2)
  1600. if (TestCpuFlag(kCpuHasAVX2)) {
  1601. NV12ToARGBRow = NV12ToARGBRow_Any_AVX2;
  1602. if (IS_ALIGNED(width, 16)) {
  1603. NV12ToARGBRow = NV12ToARGBRow_AVX2;
  1604. }
  1605. }
  1606. #endif
  1607. #if defined(HAS_NV12TOARGBROW_NEON)
  1608. if (TestCpuFlag(kCpuHasNEON)) {
  1609. NV12ToARGBRow = NV12ToARGBRow_Any_NEON;
  1610. if (IS_ALIGNED(width, 8)) {
  1611. NV12ToARGBRow = NV12ToARGBRow_NEON;
  1612. }
  1613. }
  1614. #endif
  1615. #if defined(HAS_NV12TOARGBROW_MSA)
  1616. if (TestCpuFlag(kCpuHasMSA)) {
  1617. NV12ToARGBRow = NV12ToARGBRow_Any_MSA;
  1618. if (IS_ALIGNED(width, 8)) {
  1619. NV12ToARGBRow = NV12ToARGBRow_MSA;
  1620. }
  1621. }
  1622. #endif
  1623. for (y = 0; y < height; ++y) {
  1624. NV12ToARGBRow(src_y, src_uv, dst_argb, yuvconstants, width);
  1625. dst_argb += dst_stride_argb;
  1626. src_y += src_stride_y;
  1627. if (y & 1) {
  1628. src_uv += src_stride_uv;
  1629. }
  1630. }
  1631. return 0;
  1632. }
  1633. // Convert NV21 to ARGB with matrix
  1634. static int NV21ToARGBMatrix(const uint8_t* src_y,
  1635. int src_stride_y,
  1636. const uint8_t* src_vu,
  1637. int src_stride_vu,
  1638. uint8_t* dst_argb,
  1639. int dst_stride_argb,
  1640. const struct YuvConstants* yuvconstants,
  1641. int width,
  1642. int height) {
  1643. int y;
  1644. void (*NV21ToARGBRow)(
  1645. const uint8_t* y_buf, const uint8_t* uv_buf, uint8_t* rgb_buf,
  1646. const struct YuvConstants* yuvconstants, int width) = NV21ToARGBRow_C;
  1647. if (!src_y || !src_vu || !dst_argb || width <= 0 || height == 0) {
  1648. return -1;
  1649. }
  1650. // Negative height means invert the image.
  1651. if (height < 0) {
  1652. height = -height;
  1653. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  1654. dst_stride_argb = -dst_stride_argb;
  1655. }
  1656. #if defined(HAS_NV21TOARGBROW_SSSE3)
  1657. if (TestCpuFlag(kCpuHasSSSE3)) {
  1658. NV21ToARGBRow = NV21ToARGBRow_Any_SSSE3;
  1659. if (IS_ALIGNED(width, 8)) {
  1660. NV21ToARGBRow = NV21ToARGBRow_SSSE3;
  1661. }
  1662. }
  1663. #endif
  1664. #if defined(HAS_NV21TOARGBROW_AVX2)
  1665. if (TestCpuFlag(kCpuHasAVX2)) {
  1666. NV21ToARGBRow = NV21ToARGBRow_Any_AVX2;
  1667. if (IS_ALIGNED(width, 16)) {
  1668. NV21ToARGBRow = NV21ToARGBRow_AVX2;
  1669. }
  1670. }
  1671. #endif
  1672. #if defined(HAS_NV21TOARGBROW_NEON)
  1673. if (TestCpuFlag(kCpuHasNEON)) {
  1674. NV21ToARGBRow = NV21ToARGBRow_Any_NEON;
  1675. if (IS_ALIGNED(width, 8)) {
  1676. NV21ToARGBRow = NV21ToARGBRow_NEON;
  1677. }
  1678. }
  1679. #endif
  1680. #if defined(HAS_NV21TOARGBROW_MSA)
  1681. if (TestCpuFlag(kCpuHasMSA)) {
  1682. NV21ToARGBRow = NV21ToARGBRow_Any_MSA;
  1683. if (IS_ALIGNED(width, 8)) {
  1684. NV21ToARGBRow = NV21ToARGBRow_MSA;
  1685. }
  1686. }
  1687. #endif
  1688. for (y = 0; y < height; ++y) {
  1689. NV21ToARGBRow(src_y, src_vu, dst_argb, yuvconstants, width);
  1690. dst_argb += dst_stride_argb;
  1691. src_y += src_stride_y;
  1692. if (y & 1) {
  1693. src_vu += src_stride_vu;
  1694. }
  1695. }
  1696. return 0;
  1697. }
  1698. // Convert NV12 to ARGB.
  1699. LIBYUV_API
  1700. int NV12ToARGB(const uint8_t* src_y,
  1701. int src_stride_y,
  1702. const uint8_t* src_uv,
  1703. int src_stride_uv,
  1704. uint8_t* dst_argb,
  1705. int dst_stride_argb,
  1706. int width,
  1707. int height) {
  1708. return NV12ToARGBMatrix(src_y, src_stride_y, src_uv, src_stride_uv, dst_argb,
  1709. dst_stride_argb, &kYuvI601Constants, width, height);
  1710. }
  1711. // Convert NV21 to ARGB.
  1712. LIBYUV_API
  1713. int NV21ToARGB(const uint8_t* src_y,
  1714. int src_stride_y,
  1715. const uint8_t* src_vu,
  1716. int src_stride_vu,
  1717. uint8_t* dst_argb,
  1718. int dst_stride_argb,
  1719. int width,
  1720. int height) {
  1721. return NV21ToARGBMatrix(src_y, src_stride_y, src_vu, src_stride_vu, dst_argb,
  1722. dst_stride_argb, &kYuvI601Constants, width, height);
  1723. }
  1724. // Convert NV12 to ABGR.
  1725. // To output ABGR instead of ARGB swap the UV and use a mirrored yuv matrix.
  1726. // To swap the UV use NV12 instead of NV21.LIBYUV_API
  1727. LIBYUV_API
  1728. int NV12ToABGR(const uint8_t* src_y,
  1729. int src_stride_y,
  1730. const uint8_t* src_uv,
  1731. int src_stride_uv,
  1732. uint8_t* dst_abgr,
  1733. int dst_stride_abgr,
  1734. int width,
  1735. int height) {
  1736. return NV21ToARGBMatrix(src_y, src_stride_y, src_uv, src_stride_uv, dst_abgr,
  1737. dst_stride_abgr, &kYvuI601Constants, width, height);
  1738. }
  1739. // Convert NV21 to ABGR.
  1740. LIBYUV_API
  1741. int NV21ToABGR(const uint8_t* src_y,
  1742. int src_stride_y,
  1743. const uint8_t* src_vu,
  1744. int src_stride_vu,
  1745. uint8_t* dst_abgr,
  1746. int dst_stride_abgr,
  1747. int width,
  1748. int height) {
  1749. return NV12ToARGBMatrix(src_y, src_stride_y, src_vu, src_stride_vu, dst_abgr,
  1750. dst_stride_abgr, &kYvuI601Constants, width, height);
  1751. }
  1752. // TODO(fbarchard): Consider SSSE3 2 step conversion.
  1753. // Convert NV12 to RGB24 with matrix
  1754. static int NV12ToRGB24Matrix(const uint8_t* src_y,
  1755. int src_stride_y,
  1756. const uint8_t* src_uv,
  1757. int src_stride_uv,
  1758. uint8_t* dst_rgb24,
  1759. int dst_stride_rgb24,
  1760. const struct YuvConstants* yuvconstants,
  1761. int width,
  1762. int height) {
  1763. int y;
  1764. void (*NV12ToRGB24Row)(
  1765. const uint8_t* y_buf, const uint8_t* uv_buf, uint8_t* rgb_buf,
  1766. const struct YuvConstants* yuvconstants, int width) = NV12ToRGB24Row_C;
  1767. if (!src_y || !src_uv || !dst_rgb24 || width <= 0 || height == 0) {
  1768. return -1;
  1769. }
  1770. // Negative height means invert the image.
  1771. if (height < 0) {
  1772. height = -height;
  1773. dst_rgb24 = dst_rgb24 + (height - 1) * dst_stride_rgb24;
  1774. dst_stride_rgb24 = -dst_stride_rgb24;
  1775. }
  1776. #if defined(HAS_NV12TORGB24ROW_NEON)
  1777. if (TestCpuFlag(kCpuHasNEON)) {
  1778. NV12ToRGB24Row = NV12ToRGB24Row_Any_NEON;
  1779. if (IS_ALIGNED(width, 8)) {
  1780. NV12ToRGB24Row = NV12ToRGB24Row_NEON;
  1781. }
  1782. }
  1783. #endif
  1784. #if defined(HAS_NV12TORGB24ROW_SSSE3)
  1785. if (TestCpuFlag(kCpuHasSSSE3)) {
  1786. NV12ToRGB24Row = NV12ToRGB24Row_Any_SSSE3;
  1787. if (IS_ALIGNED(width, 16)) {
  1788. NV12ToRGB24Row = NV12ToRGB24Row_SSSE3;
  1789. }
  1790. }
  1791. #endif
  1792. #if defined(HAS_NV12TORGB24ROW_AVX2)
  1793. if (TestCpuFlag(kCpuHasAVX2)) {
  1794. NV12ToRGB24Row = NV12ToRGB24Row_Any_AVX2;
  1795. if (IS_ALIGNED(width, 32)) {
  1796. NV12ToRGB24Row = NV12ToRGB24Row_AVX2;
  1797. }
  1798. }
  1799. #endif
  1800. for (y = 0; y < height; ++y) {
  1801. NV12ToRGB24Row(src_y, src_uv, dst_rgb24, yuvconstants, width);
  1802. dst_rgb24 += dst_stride_rgb24;
  1803. src_y += src_stride_y;
  1804. if (y & 1) {
  1805. src_uv += src_stride_uv;
  1806. }
  1807. }
  1808. return 0;
  1809. }
  1810. // Convert NV21 to RGB24 with matrix
  1811. static int NV21ToRGB24Matrix(const uint8_t* src_y,
  1812. int src_stride_y,
  1813. const uint8_t* src_vu,
  1814. int src_stride_vu,
  1815. uint8_t* dst_rgb24,
  1816. int dst_stride_rgb24,
  1817. const struct YuvConstants* yuvconstants,
  1818. int width,
  1819. int height) {
  1820. int y;
  1821. void (*NV21ToRGB24Row)(
  1822. const uint8_t* y_buf, const uint8_t* uv_buf, uint8_t* rgb_buf,
  1823. const struct YuvConstants* yuvconstants, int width) = NV21ToRGB24Row_C;
  1824. if (!src_y || !src_vu || !dst_rgb24 || width <= 0 || height == 0) {
  1825. return -1;
  1826. }
  1827. // Negative height means invert the image.
  1828. if (height < 0) {
  1829. height = -height;
  1830. dst_rgb24 = dst_rgb24 + (height - 1) * dst_stride_rgb24;
  1831. dst_stride_rgb24 = -dst_stride_rgb24;
  1832. }
  1833. #if defined(HAS_NV21TORGB24ROW_NEON)
  1834. if (TestCpuFlag(kCpuHasNEON)) {
  1835. NV21ToRGB24Row = NV21ToRGB24Row_Any_NEON;
  1836. if (IS_ALIGNED(width, 8)) {
  1837. NV21ToRGB24Row = NV21ToRGB24Row_NEON;
  1838. }
  1839. }
  1840. #endif
  1841. #if defined(HAS_NV21TORGB24ROW_SSSE3)
  1842. if (TestCpuFlag(kCpuHasSSSE3)) {
  1843. NV21ToRGB24Row = NV21ToRGB24Row_Any_SSSE3;
  1844. if (IS_ALIGNED(width, 16)) {
  1845. NV21ToRGB24Row = NV21ToRGB24Row_SSSE3;
  1846. }
  1847. }
  1848. #endif
  1849. #if defined(HAS_NV21TORGB24ROW_AVX2)
  1850. if (TestCpuFlag(kCpuHasAVX2)) {
  1851. NV21ToRGB24Row = NV21ToRGB24Row_Any_AVX2;
  1852. if (IS_ALIGNED(width, 32)) {
  1853. NV21ToRGB24Row = NV21ToRGB24Row_AVX2;
  1854. }
  1855. }
  1856. #endif
  1857. for (y = 0; y < height; ++y) {
  1858. NV21ToRGB24Row(src_y, src_vu, dst_rgb24, yuvconstants, width);
  1859. dst_rgb24 += dst_stride_rgb24;
  1860. src_y += src_stride_y;
  1861. if (y & 1) {
  1862. src_vu += src_stride_vu;
  1863. }
  1864. }
  1865. return 0;
  1866. }
  1867. // Convert NV12 to RGB24.
  1868. LIBYUV_API
  1869. int NV12ToRGB24(const uint8_t* src_y,
  1870. int src_stride_y,
  1871. const uint8_t* src_uv,
  1872. int src_stride_uv,
  1873. uint8_t* dst_rgb24,
  1874. int dst_stride_rgb24,
  1875. int width,
  1876. int height) {
  1877. return NV12ToRGB24Matrix(src_y, src_stride_y, src_uv, src_stride_uv,
  1878. dst_rgb24, dst_stride_rgb24, &kYuvI601Constants,
  1879. width, height);
  1880. }
  1881. // Convert NV21 to RGB24.
  1882. LIBYUV_API
  1883. int NV21ToRGB24(const uint8_t* src_y,
  1884. int src_stride_y,
  1885. const uint8_t* src_vu,
  1886. int src_stride_vu,
  1887. uint8_t* dst_rgb24,
  1888. int dst_stride_rgb24,
  1889. int width,
  1890. int height) {
  1891. return NV21ToRGB24Matrix(src_y, src_stride_y, src_vu, src_stride_vu,
  1892. dst_rgb24, dst_stride_rgb24, &kYuvI601Constants,
  1893. width, height);
  1894. }
  1895. // Convert NV12 to RAW.
  1896. LIBYUV_API
  1897. int NV12ToRAW(const uint8_t* src_y,
  1898. int src_stride_y,
  1899. const uint8_t* src_uv,
  1900. int src_stride_uv,
  1901. uint8_t* dst_raw,
  1902. int dst_stride_raw,
  1903. int width,
  1904. int height) {
  1905. return NV21ToRGB24Matrix(src_y, src_stride_y, src_uv, src_stride_uv, dst_raw,
  1906. dst_stride_raw, &kYvuI601Constants, width, height);
  1907. }
  1908. // Convert NV21 to RAW.
  1909. LIBYUV_API
  1910. int NV21ToRAW(const uint8_t* src_y,
  1911. int src_stride_y,
  1912. const uint8_t* src_vu,
  1913. int src_stride_vu,
  1914. uint8_t* dst_raw,
  1915. int dst_stride_raw,
  1916. int width,
  1917. int height) {
  1918. return NV12ToRGB24Matrix(src_y, src_stride_y, src_vu, src_stride_vu, dst_raw,
  1919. dst_stride_raw, &kYvuI601Constants, width, height);
  1920. }
  1921. // Convert NV21 to YUV24
  1922. int NV21ToYUV24(const uint8_t* src_y,
  1923. int src_stride_y,
  1924. const uint8_t* src_vu,
  1925. int src_stride_vu,
  1926. uint8_t* dst_yuv24,
  1927. int dst_stride_yuv24,
  1928. int width,
  1929. int height) {
  1930. int y;
  1931. void (*NV21ToYUV24Row)(const uint8_t* src_y, const uint8_t* src_vu,
  1932. uint8_t* dst_yuv24, int width) = NV21ToYUV24Row_C;
  1933. if (!src_y || !src_vu || !dst_yuv24 || width <= 0 || height == 0) {
  1934. return -1;
  1935. }
  1936. // Negative height means invert the image.
  1937. if (height < 0) {
  1938. height = -height;
  1939. dst_yuv24 = dst_yuv24 + (height - 1) * dst_stride_yuv24;
  1940. dst_stride_yuv24 = -dst_stride_yuv24;
  1941. }
  1942. #if defined(HAS_NV21TOYUV24ROW_NEON)
  1943. if (TestCpuFlag(kCpuHasNEON)) {
  1944. NV21ToYUV24Row = NV21ToYUV24Row_Any_NEON;
  1945. if (IS_ALIGNED(width, 16)) {
  1946. NV21ToYUV24Row = NV21ToYUV24Row_NEON;
  1947. }
  1948. }
  1949. #endif
  1950. #if defined(HAS_NV21TOYUV24ROW_AVX2)
  1951. if (TestCpuFlag(kCpuHasAVX2)) {
  1952. NV21ToYUV24Row = NV21ToYUV24Row_Any_AVX2;
  1953. if (IS_ALIGNED(width, 32)) {
  1954. NV21ToYUV24Row = NV21ToYUV24Row_AVX2;
  1955. }
  1956. }
  1957. #endif
  1958. for (y = 0; y < height; ++y) {
  1959. NV21ToYUV24Row(src_y, src_vu, dst_yuv24, width);
  1960. dst_yuv24 += dst_stride_yuv24;
  1961. src_y += src_stride_y;
  1962. if (y & 1) {
  1963. src_vu += src_stride_vu;
  1964. }
  1965. }
  1966. return 0;
  1967. }
  1968. // Convert M420 to ARGB.
  1969. LIBYUV_API
  1970. int M420ToARGB(const uint8_t* src_m420,
  1971. int src_stride_m420,
  1972. uint8_t* dst_argb,
  1973. int dst_stride_argb,
  1974. int width,
  1975. int height) {
  1976. int y;
  1977. void (*NV12ToARGBRow)(
  1978. const uint8_t* y_buf, const uint8_t* uv_buf, uint8_t* rgb_buf,
  1979. const struct YuvConstants* yuvconstants, int width) = NV12ToARGBRow_C;
  1980. if (!src_m420 || !dst_argb || width <= 0 || height == 0) {
  1981. return -1;
  1982. }
  1983. // Negative height means invert the image.
  1984. if (height < 0) {
  1985. height = -height;
  1986. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  1987. dst_stride_argb = -dst_stride_argb;
  1988. }
  1989. #if defined(HAS_NV12TOARGBROW_SSSE3)
  1990. if (TestCpuFlag(kCpuHasSSSE3)) {
  1991. NV12ToARGBRow = NV12ToARGBRow_Any_SSSE3;
  1992. if (IS_ALIGNED(width, 8)) {
  1993. NV12ToARGBRow = NV12ToARGBRow_SSSE3;
  1994. }
  1995. }
  1996. #endif
  1997. #if defined(HAS_NV12TOARGBROW_AVX2)
  1998. if (TestCpuFlag(kCpuHasAVX2)) {
  1999. NV12ToARGBRow = NV12ToARGBRow_Any_AVX2;
  2000. if (IS_ALIGNED(width, 16)) {
  2001. NV12ToARGBRow = NV12ToARGBRow_AVX2;
  2002. }
  2003. }
  2004. #endif
  2005. #if defined(HAS_NV12TOARGBROW_NEON)
  2006. if (TestCpuFlag(kCpuHasNEON)) {
  2007. NV12ToARGBRow = NV12ToARGBRow_Any_NEON;
  2008. if (IS_ALIGNED(width, 8)) {
  2009. NV12ToARGBRow = NV12ToARGBRow_NEON;
  2010. }
  2011. }
  2012. #endif
  2013. #if defined(HAS_NV12TOARGBROW_MSA)
  2014. if (TestCpuFlag(kCpuHasMSA)) {
  2015. NV12ToARGBRow = NV12ToARGBRow_Any_MSA;
  2016. if (IS_ALIGNED(width, 8)) {
  2017. NV12ToARGBRow = NV12ToARGBRow_MSA;
  2018. }
  2019. }
  2020. #endif
  2021. for (y = 0; y < height - 1; y += 2) {
  2022. NV12ToARGBRow(src_m420, src_m420 + src_stride_m420 * 2, dst_argb,
  2023. &kYuvI601Constants, width);
  2024. NV12ToARGBRow(src_m420 + src_stride_m420, src_m420 + src_stride_m420 * 2,
  2025. dst_argb + dst_stride_argb, &kYuvI601Constants, width);
  2026. dst_argb += dst_stride_argb * 2;
  2027. src_m420 += src_stride_m420 * 3;
  2028. }
  2029. if (height & 1) {
  2030. NV12ToARGBRow(src_m420, src_m420 + src_stride_m420 * 2, dst_argb,
  2031. &kYuvI601Constants, width);
  2032. }
  2033. return 0;
  2034. }
  2035. // Convert YUY2 to ARGB.
  2036. LIBYUV_API
  2037. int YUY2ToARGB(const uint8_t* src_yuy2,
  2038. int src_stride_yuy2,
  2039. uint8_t* dst_argb,
  2040. int dst_stride_argb,
  2041. int width,
  2042. int height) {
  2043. int y;
  2044. void (*YUY2ToARGBRow)(const uint8_t* src_yuy2, uint8_t* dst_argb,
  2045. const struct YuvConstants* yuvconstants, int width) =
  2046. YUY2ToARGBRow_C;
  2047. if (!src_yuy2 || !dst_argb || width <= 0 || height == 0) {
  2048. return -1;
  2049. }
  2050. // Negative height means invert the image.
  2051. if (height < 0) {
  2052. height = -height;
  2053. src_yuy2 = src_yuy2 + (height - 1) * src_stride_yuy2;
  2054. src_stride_yuy2 = -src_stride_yuy2;
  2055. }
  2056. // Coalesce rows.
  2057. if (src_stride_yuy2 == width * 2 && dst_stride_argb == width * 4) {
  2058. width *= height;
  2059. height = 1;
  2060. src_stride_yuy2 = dst_stride_argb = 0;
  2061. }
  2062. #if defined(HAS_YUY2TOARGBROW_SSSE3)
  2063. if (TestCpuFlag(kCpuHasSSSE3)) {
  2064. YUY2ToARGBRow = YUY2ToARGBRow_Any_SSSE3;
  2065. if (IS_ALIGNED(width, 16)) {
  2066. YUY2ToARGBRow = YUY2ToARGBRow_SSSE3;
  2067. }
  2068. }
  2069. #endif
  2070. #if defined(HAS_YUY2TOARGBROW_AVX2)
  2071. if (TestCpuFlag(kCpuHasAVX2)) {
  2072. YUY2ToARGBRow = YUY2ToARGBRow_Any_AVX2;
  2073. if (IS_ALIGNED(width, 32)) {
  2074. YUY2ToARGBRow = YUY2ToARGBRow_AVX2;
  2075. }
  2076. }
  2077. #endif
  2078. #if defined(HAS_YUY2TOARGBROW_NEON)
  2079. if (TestCpuFlag(kCpuHasNEON)) {
  2080. YUY2ToARGBRow = YUY2ToARGBRow_Any_NEON;
  2081. if (IS_ALIGNED(width, 8)) {
  2082. YUY2ToARGBRow = YUY2ToARGBRow_NEON;
  2083. }
  2084. }
  2085. #endif
  2086. #if defined(HAS_YUY2TOARGBROW_MSA)
  2087. if (TestCpuFlag(kCpuHasMSA)) {
  2088. YUY2ToARGBRow = YUY2ToARGBRow_Any_MSA;
  2089. if (IS_ALIGNED(width, 8)) {
  2090. YUY2ToARGBRow = YUY2ToARGBRow_MSA;
  2091. }
  2092. }
  2093. #endif
  2094. for (y = 0; y < height; ++y) {
  2095. YUY2ToARGBRow(src_yuy2, dst_argb, &kYuvI601Constants, width);
  2096. src_yuy2 += src_stride_yuy2;
  2097. dst_argb += dst_stride_argb;
  2098. }
  2099. return 0;
  2100. }
  2101. // Convert UYVY to ARGB.
  2102. LIBYUV_API
  2103. int UYVYToARGB(const uint8_t* src_uyvy,
  2104. int src_stride_uyvy,
  2105. uint8_t* dst_argb,
  2106. int dst_stride_argb,
  2107. int width,
  2108. int height) {
  2109. int y;
  2110. void (*UYVYToARGBRow)(const uint8_t* src_uyvy, uint8_t* dst_argb,
  2111. const struct YuvConstants* yuvconstants, int width) =
  2112. UYVYToARGBRow_C;
  2113. if (!src_uyvy || !dst_argb || width <= 0 || height == 0) {
  2114. return -1;
  2115. }
  2116. // Negative height means invert the image.
  2117. if (height < 0) {
  2118. height = -height;
  2119. src_uyvy = src_uyvy + (height - 1) * src_stride_uyvy;
  2120. src_stride_uyvy = -src_stride_uyvy;
  2121. }
  2122. // Coalesce rows.
  2123. if (src_stride_uyvy == width * 2 && dst_stride_argb == width * 4) {
  2124. width *= height;
  2125. height = 1;
  2126. src_stride_uyvy = dst_stride_argb = 0;
  2127. }
  2128. #if defined(HAS_UYVYTOARGBROW_SSSE3)
  2129. if (TestCpuFlag(kCpuHasSSSE3)) {
  2130. UYVYToARGBRow = UYVYToARGBRow_Any_SSSE3;
  2131. if (IS_ALIGNED(width, 16)) {
  2132. UYVYToARGBRow = UYVYToARGBRow_SSSE3;
  2133. }
  2134. }
  2135. #endif
  2136. #if defined(HAS_UYVYTOARGBROW_AVX2)
  2137. if (TestCpuFlag(kCpuHasAVX2)) {
  2138. UYVYToARGBRow = UYVYToARGBRow_Any_AVX2;
  2139. if (IS_ALIGNED(width, 32)) {
  2140. UYVYToARGBRow = UYVYToARGBRow_AVX2;
  2141. }
  2142. }
  2143. #endif
  2144. #if defined(HAS_UYVYTOARGBROW_NEON)
  2145. if (TestCpuFlag(kCpuHasNEON)) {
  2146. UYVYToARGBRow = UYVYToARGBRow_Any_NEON;
  2147. if (IS_ALIGNED(width, 8)) {
  2148. UYVYToARGBRow = UYVYToARGBRow_NEON;
  2149. }
  2150. }
  2151. #endif
  2152. #if defined(HAS_UYVYTOARGBROW_MSA)
  2153. if (TestCpuFlag(kCpuHasMSA)) {
  2154. UYVYToARGBRow = UYVYToARGBRow_Any_MSA;
  2155. if (IS_ALIGNED(width, 8)) {
  2156. UYVYToARGBRow = UYVYToARGBRow_MSA;
  2157. }
  2158. }
  2159. #endif
  2160. for (y = 0; y < height; ++y) {
  2161. UYVYToARGBRow(src_uyvy, dst_argb, &kYuvI601Constants, width);
  2162. src_uyvy += src_stride_uyvy;
  2163. dst_argb += dst_stride_argb;
  2164. }
  2165. return 0;
  2166. }
  2167. static void WeavePixels(const uint8_t* src_u,
  2168. const uint8_t* src_v,
  2169. int src_pixel_stride_uv,
  2170. uint8_t* dst_uv,
  2171. int width) {
  2172. int i;
  2173. for (i = 0; i < width; ++i) {
  2174. dst_uv[0] = *src_u;
  2175. dst_uv[1] = *src_v;
  2176. dst_uv += 2;
  2177. src_u += src_pixel_stride_uv;
  2178. src_v += src_pixel_stride_uv;
  2179. }
  2180. }
  2181. // Convert Android420 to ARGB.
  2182. LIBYUV_API
  2183. int Android420ToARGBMatrix(const uint8_t* src_y,
  2184. int src_stride_y,
  2185. const uint8_t* src_u,
  2186. int src_stride_u,
  2187. const uint8_t* src_v,
  2188. int src_stride_v,
  2189. int src_pixel_stride_uv,
  2190. uint8_t* dst_argb,
  2191. int dst_stride_argb,
  2192. const struct YuvConstants* yuvconstants,
  2193. int width,
  2194. int height) {
  2195. int y;
  2196. uint8_t* dst_uv;
  2197. const ptrdiff_t vu_off = src_v - src_u;
  2198. int halfwidth = (width + 1) >> 1;
  2199. int halfheight = (height + 1) >> 1;
  2200. if (!src_y || !src_u || !src_v || !dst_argb || width <= 0 || height == 0) {
  2201. return -1;
  2202. }
  2203. // Negative height means invert the image.
  2204. if (height < 0) {
  2205. height = -height;
  2206. halfheight = (height + 1) >> 1;
  2207. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  2208. dst_stride_argb = -dst_stride_argb;
  2209. }
  2210. // I420
  2211. if (src_pixel_stride_uv == 1) {
  2212. return I420ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  2213. src_stride_v, dst_argb, dst_stride_argb,
  2214. yuvconstants, width, height);
  2215. // NV21
  2216. }
  2217. if (src_pixel_stride_uv == 2 && vu_off == -1 &&
  2218. src_stride_u == src_stride_v) {
  2219. return NV21ToARGBMatrix(src_y, src_stride_y, src_v, src_stride_v, dst_argb,
  2220. dst_stride_argb, yuvconstants, width, height);
  2221. // NV12
  2222. }
  2223. if (src_pixel_stride_uv == 2 && vu_off == 1 && src_stride_u == src_stride_v) {
  2224. return NV12ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, dst_argb,
  2225. dst_stride_argb, yuvconstants, width, height);
  2226. }
  2227. // General case fallback creates NV12
  2228. align_buffer_64(plane_uv, halfwidth * 2 * halfheight);
  2229. dst_uv = plane_uv;
  2230. for (y = 0; y < halfheight; ++y) {
  2231. WeavePixels(src_u, src_v, src_pixel_stride_uv, dst_uv, halfwidth);
  2232. src_u += src_stride_u;
  2233. src_v += src_stride_v;
  2234. dst_uv += halfwidth * 2;
  2235. }
  2236. NV12ToARGBMatrix(src_y, src_stride_y, plane_uv, halfwidth * 2, dst_argb,
  2237. dst_stride_argb, yuvconstants, width, height);
  2238. free_aligned_buffer_64(plane_uv);
  2239. return 0;
  2240. }
  2241. // Convert Android420 to ARGB.
  2242. LIBYUV_API
  2243. int Android420ToARGB(const uint8_t* src_y,
  2244. int src_stride_y,
  2245. const uint8_t* src_u,
  2246. int src_stride_u,
  2247. const uint8_t* src_v,
  2248. int src_stride_v,
  2249. int src_pixel_stride_uv,
  2250. uint8_t* dst_argb,
  2251. int dst_stride_argb,
  2252. int width,
  2253. int height) {
  2254. return Android420ToARGBMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  2255. src_stride_v, src_pixel_stride_uv, dst_argb,
  2256. dst_stride_argb, &kYuvI601Constants, width,
  2257. height);
  2258. }
  2259. // Convert Android420 to ABGR.
  2260. LIBYUV_API
  2261. int Android420ToABGR(const uint8_t* src_y,
  2262. int src_stride_y,
  2263. const uint8_t* src_u,
  2264. int src_stride_u,
  2265. const uint8_t* src_v,
  2266. int src_stride_v,
  2267. int src_pixel_stride_uv,
  2268. uint8_t* dst_abgr,
  2269. int dst_stride_abgr,
  2270. int width,
  2271. int height) {
  2272. return Android420ToARGBMatrix(src_y, src_stride_y, src_v, src_stride_v, src_u,
  2273. src_stride_u, src_pixel_stride_uv, dst_abgr,
  2274. dst_stride_abgr, &kYvuI601Constants, width,
  2275. height);
  2276. }
  2277. #ifdef __cplusplus
  2278. } // extern "C"
  2279. } // namespace libyuv
  2280. #endif