planar_functions.cc 100 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517
  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/planar_functions.h"
  11. #include <string.h> // for memset()
  12. #include "libyuv/cpu_id.h"
  13. #ifdef HAVE_JPEG
  14. #include "libyuv/mjpeg_decoder.h"
  15. #endif
  16. #include "libyuv/row.h"
  17. #include "libyuv/scale_row.h" // for ScaleRowDown2
  18. #ifdef __cplusplus
  19. namespace libyuv {
  20. extern "C" {
  21. #endif
  22. // Copy a plane of data
  23. LIBYUV_API
  24. void CopyPlane(const uint8* src_y,
  25. int src_stride_y,
  26. uint8* dst_y,
  27. int dst_stride_y,
  28. int width,
  29. int height) {
  30. int y;
  31. void (*CopyRow)(const uint8* src, uint8* dst, int width) = CopyRow_C;
  32. // Negative height means invert the image.
  33. if (height < 0) {
  34. height = -height;
  35. dst_y = dst_y + (height - 1) * dst_stride_y;
  36. dst_stride_y = -dst_stride_y;
  37. }
  38. // Coalesce rows.
  39. if (src_stride_y == width && dst_stride_y == width) {
  40. width *= height;
  41. height = 1;
  42. src_stride_y = dst_stride_y = 0;
  43. }
  44. // Nothing to do.
  45. if (src_y == dst_y && src_stride_y == dst_stride_y) {
  46. return;
  47. }
  48. #if defined(HAS_COPYROW_SSE2)
  49. if (TestCpuFlag(kCpuHasSSE2)) {
  50. CopyRow = IS_ALIGNED(width, 32) ? CopyRow_SSE2 : CopyRow_Any_SSE2;
  51. }
  52. #endif
  53. #if defined(HAS_COPYROW_AVX)
  54. if (TestCpuFlag(kCpuHasAVX)) {
  55. CopyRow = IS_ALIGNED(width, 64) ? CopyRow_AVX : CopyRow_Any_AVX;
  56. }
  57. #endif
  58. #if defined(HAS_COPYROW_ERMS)
  59. if (TestCpuFlag(kCpuHasERMS)) {
  60. CopyRow = CopyRow_ERMS;
  61. }
  62. #endif
  63. #if defined(HAS_COPYROW_NEON)
  64. if (TestCpuFlag(kCpuHasNEON)) {
  65. CopyRow = IS_ALIGNED(width, 32) ? CopyRow_NEON : CopyRow_Any_NEON;
  66. }
  67. #endif
  68. #if defined(HAS_COPYROW_MIPS)
  69. if (TestCpuFlag(kCpuHasMIPS)) {
  70. CopyRow = CopyRow_MIPS;
  71. }
  72. #endif
  73. // Copy plane
  74. for (y = 0; y < height; ++y) {
  75. CopyRow(src_y, dst_y, width);
  76. src_y += src_stride_y;
  77. dst_y += dst_stride_y;
  78. }
  79. }
  80. // TODO(fbarchard): Consider support for negative height.
  81. // TODO(fbarchard): Consider stride measured in bytes.
  82. LIBYUV_API
  83. void CopyPlane_16(const uint16* src_y,
  84. int src_stride_y,
  85. uint16* dst_y,
  86. int dst_stride_y,
  87. int width,
  88. int height) {
  89. int y;
  90. void (*CopyRow)(const uint16* src, uint16* dst, int width) = CopyRow_16_C;
  91. // Coalesce rows.
  92. if (src_stride_y == width && dst_stride_y == width) {
  93. width *= height;
  94. height = 1;
  95. src_stride_y = dst_stride_y = 0;
  96. }
  97. #if defined(HAS_COPYROW_16_SSE2)
  98. if (TestCpuFlag(kCpuHasSSE2) && IS_ALIGNED(width, 32)) {
  99. CopyRow = CopyRow_16_SSE2;
  100. }
  101. #endif
  102. #if defined(HAS_COPYROW_16_ERMS)
  103. if (TestCpuFlag(kCpuHasERMS)) {
  104. CopyRow = CopyRow_16_ERMS;
  105. }
  106. #endif
  107. #if defined(HAS_COPYROW_16_NEON)
  108. if (TestCpuFlag(kCpuHasNEON) && IS_ALIGNED(width, 32)) {
  109. CopyRow = CopyRow_16_NEON;
  110. }
  111. #endif
  112. #if defined(HAS_COPYROW_16_MIPS)
  113. if (TestCpuFlag(kCpuHasMIPS)) {
  114. CopyRow = CopyRow_16_MIPS;
  115. }
  116. #endif
  117. // Copy plane
  118. for (y = 0; y < height; ++y) {
  119. CopyRow(src_y, dst_y, width);
  120. src_y += src_stride_y;
  121. dst_y += dst_stride_y;
  122. }
  123. }
  124. // Copy I422.
  125. LIBYUV_API
  126. int I422Copy(const uint8* src_y,
  127. int src_stride_y,
  128. const uint8* src_u,
  129. int src_stride_u,
  130. const uint8* src_v,
  131. int src_stride_v,
  132. uint8* dst_y,
  133. int dst_stride_y,
  134. uint8* dst_u,
  135. int dst_stride_u,
  136. uint8* dst_v,
  137. int dst_stride_v,
  138. int width,
  139. int height) {
  140. int halfwidth = (width + 1) >> 1;
  141. if (!src_u || !src_v || !dst_u || !dst_v || width <= 0 || height == 0) {
  142. return -1;
  143. }
  144. // Negative height means invert the image.
  145. if (height < 0) {
  146. height = -height;
  147. src_y = src_y + (height - 1) * src_stride_y;
  148. src_u = src_u + (height - 1) * src_stride_u;
  149. src_v = src_v + (height - 1) * src_stride_v;
  150. src_stride_y = -src_stride_y;
  151. src_stride_u = -src_stride_u;
  152. src_stride_v = -src_stride_v;
  153. }
  154. if (dst_y) {
  155. CopyPlane(src_y, src_stride_y, dst_y, dst_stride_y, width, height);
  156. }
  157. CopyPlane(src_u, src_stride_u, dst_u, dst_stride_u, halfwidth, height);
  158. CopyPlane(src_v, src_stride_v, dst_v, dst_stride_v, halfwidth, height);
  159. return 0;
  160. }
  161. // Copy I444.
  162. LIBYUV_API
  163. int I444Copy(const uint8* src_y,
  164. int src_stride_y,
  165. const uint8* src_u,
  166. int src_stride_u,
  167. const uint8* src_v,
  168. int src_stride_v,
  169. uint8* dst_y,
  170. int dst_stride_y,
  171. uint8* dst_u,
  172. int dst_stride_u,
  173. uint8* dst_v,
  174. int dst_stride_v,
  175. int width,
  176. int height) {
  177. if (!src_u || !src_v || !dst_u || !dst_v || width <= 0 || height == 0) {
  178. return -1;
  179. }
  180. // Negative height means invert the image.
  181. if (height < 0) {
  182. height = -height;
  183. src_y = src_y + (height - 1) * src_stride_y;
  184. src_u = src_u + (height - 1) * src_stride_u;
  185. src_v = src_v + (height - 1) * src_stride_v;
  186. src_stride_y = -src_stride_y;
  187. src_stride_u = -src_stride_u;
  188. src_stride_v = -src_stride_v;
  189. }
  190. if (dst_y) {
  191. CopyPlane(src_y, src_stride_y, dst_y, dst_stride_y, width, height);
  192. }
  193. CopyPlane(src_u, src_stride_u, dst_u, dst_stride_u, width, height);
  194. CopyPlane(src_v, src_stride_v, dst_v, dst_stride_v, width, height);
  195. return 0;
  196. }
  197. // Copy I400.
  198. LIBYUV_API
  199. int I400ToI400(const uint8* src_y,
  200. int src_stride_y,
  201. uint8* dst_y,
  202. int dst_stride_y,
  203. int width,
  204. int height) {
  205. if (!src_y || !dst_y || width <= 0 || height == 0) {
  206. return -1;
  207. }
  208. // Negative height means invert the image.
  209. if (height < 0) {
  210. height = -height;
  211. src_y = src_y + (height - 1) * src_stride_y;
  212. src_stride_y = -src_stride_y;
  213. }
  214. CopyPlane(src_y, src_stride_y, dst_y, dst_stride_y, width, height);
  215. return 0;
  216. }
  217. // Convert I420 to I400.
  218. LIBYUV_API
  219. int I420ToI400(const uint8* src_y,
  220. int src_stride_y,
  221. const uint8* src_u,
  222. int src_stride_u,
  223. const uint8* src_v,
  224. int src_stride_v,
  225. uint8* dst_y,
  226. int dst_stride_y,
  227. int width,
  228. int height) {
  229. (void)src_u;
  230. (void)src_stride_u;
  231. (void)src_v;
  232. (void)src_stride_v;
  233. if (!src_y || !dst_y || width <= 0 || height == 0) {
  234. return -1;
  235. }
  236. // Negative height means invert the image.
  237. if (height < 0) {
  238. height = -height;
  239. src_y = src_y + (height - 1) * src_stride_y;
  240. src_stride_y = -src_stride_y;
  241. }
  242. CopyPlane(src_y, src_stride_y, dst_y, dst_stride_y, width, height);
  243. return 0;
  244. }
  245. // Support function for NV12 etc UV channels.
  246. // Width and height are plane sizes (typically half pixel width).
  247. LIBYUV_API
  248. void SplitUVPlane(const uint8* src_uv,
  249. int src_stride_uv,
  250. uint8* dst_u,
  251. int dst_stride_u,
  252. uint8* dst_v,
  253. int dst_stride_v,
  254. int width,
  255. int height) {
  256. int y;
  257. void (*SplitUVRow)(const uint8* src_uv, uint8* dst_u, uint8* dst_v,
  258. int width) = SplitUVRow_C;
  259. // Negative height means invert the image.
  260. if (height < 0) {
  261. height = -height;
  262. dst_u = dst_u + (height - 1) * dst_stride_u;
  263. dst_v = dst_v + (height - 1) * dst_stride_v;
  264. dst_stride_u = -dst_stride_u;
  265. dst_stride_v = -dst_stride_v;
  266. }
  267. // Coalesce rows.
  268. if (src_stride_uv == width * 2 && dst_stride_u == width &&
  269. dst_stride_v == width) {
  270. width *= height;
  271. height = 1;
  272. src_stride_uv = dst_stride_u = dst_stride_v = 0;
  273. }
  274. #if defined(HAS_SPLITUVROW_SSE2)
  275. if (TestCpuFlag(kCpuHasSSE2)) {
  276. SplitUVRow = SplitUVRow_Any_SSE2;
  277. if (IS_ALIGNED(width, 16)) {
  278. SplitUVRow = SplitUVRow_SSE2;
  279. }
  280. }
  281. #endif
  282. #if defined(HAS_SPLITUVROW_AVX2)
  283. if (TestCpuFlag(kCpuHasAVX2)) {
  284. SplitUVRow = SplitUVRow_Any_AVX2;
  285. if (IS_ALIGNED(width, 32)) {
  286. SplitUVRow = SplitUVRow_AVX2;
  287. }
  288. }
  289. #endif
  290. #if defined(HAS_SPLITUVROW_NEON)
  291. if (TestCpuFlag(kCpuHasNEON)) {
  292. SplitUVRow = SplitUVRow_Any_NEON;
  293. if (IS_ALIGNED(width, 16)) {
  294. SplitUVRow = SplitUVRow_NEON;
  295. }
  296. }
  297. #endif
  298. #if defined(HAS_SPLITUVROW_DSPR2)
  299. if (TestCpuFlag(kCpuHasDSPR2) && IS_ALIGNED(dst_u, 4) &&
  300. IS_ALIGNED(dst_stride_u, 4) && IS_ALIGNED(dst_v, 4) &&
  301. IS_ALIGNED(dst_stride_v, 4)) {
  302. SplitUVRow = SplitUVRow_Any_DSPR2;
  303. if (IS_ALIGNED(width, 16)) {
  304. SplitUVRow = SplitUVRow_DSPR2;
  305. }
  306. }
  307. #endif
  308. #if defined(HAS_SPLITUVROW_MSA)
  309. if (TestCpuFlag(kCpuHasMSA)) {
  310. SplitUVRow = SplitUVRow_Any_MSA;
  311. if (IS_ALIGNED(width, 32)) {
  312. SplitUVRow = SplitUVRow_MSA;
  313. }
  314. }
  315. #endif
  316. for (y = 0; y < height; ++y) {
  317. // Copy a row of UV.
  318. SplitUVRow(src_uv, dst_u, dst_v, width);
  319. dst_u += dst_stride_u;
  320. dst_v += dst_stride_v;
  321. src_uv += src_stride_uv;
  322. }
  323. }
  324. LIBYUV_API
  325. void MergeUVPlane(const uint8* src_u,
  326. int src_stride_u,
  327. const uint8* src_v,
  328. int src_stride_v,
  329. uint8* dst_uv,
  330. int dst_stride_uv,
  331. int width,
  332. int height) {
  333. int y;
  334. void (*MergeUVRow)(const uint8* src_u, const uint8* src_v, uint8* dst_uv,
  335. int width) = MergeUVRow_C;
  336. // Coalesce rows.
  337. // Negative height means invert the image.
  338. if (height < 0) {
  339. height = -height;
  340. dst_uv = dst_uv + (height - 1) * dst_stride_uv;
  341. dst_stride_uv = -dst_stride_uv;
  342. }
  343. // Coalesce rows.
  344. if (src_stride_u == width && src_stride_v == width &&
  345. dst_stride_uv == width * 2) {
  346. width *= height;
  347. height = 1;
  348. src_stride_u = src_stride_v = dst_stride_uv = 0;
  349. }
  350. #if defined(HAS_MERGEUVROW_SSE2)
  351. if (TestCpuFlag(kCpuHasSSE2)) {
  352. MergeUVRow = MergeUVRow_Any_SSE2;
  353. if (IS_ALIGNED(width, 16)) {
  354. MergeUVRow = MergeUVRow_SSE2;
  355. }
  356. }
  357. #endif
  358. #if defined(HAS_MERGEUVROW_AVX2)
  359. if (TestCpuFlag(kCpuHasAVX2)) {
  360. MergeUVRow = MergeUVRow_Any_AVX2;
  361. if (IS_ALIGNED(width, 32)) {
  362. MergeUVRow = MergeUVRow_AVX2;
  363. }
  364. }
  365. #endif
  366. #if defined(HAS_MERGEUVROW_NEON)
  367. if (TestCpuFlag(kCpuHasNEON)) {
  368. MergeUVRow = MergeUVRow_Any_NEON;
  369. if (IS_ALIGNED(width, 16)) {
  370. MergeUVRow = MergeUVRow_NEON;
  371. }
  372. }
  373. #endif
  374. #if defined(HAS_MERGEUVROW_MSA)
  375. if (TestCpuFlag(kCpuHasMSA)) {
  376. MergeUVRow = MergeUVRow_Any_MSA;
  377. if (IS_ALIGNED(width, 16)) {
  378. MergeUVRow = MergeUVRow_MSA;
  379. }
  380. }
  381. #endif
  382. for (y = 0; y < height; ++y) {
  383. // Merge a row of U and V into a row of UV.
  384. MergeUVRow(src_u, src_v, dst_uv, width);
  385. src_u += src_stride_u;
  386. src_v += src_stride_v;
  387. dst_uv += dst_stride_uv;
  388. }
  389. }
  390. // Support function for NV12 etc RGB channels.
  391. // Width and height are plane sizes (typically half pixel width).
  392. LIBYUV_API
  393. void SplitRGBPlane(const uint8* src_rgb,
  394. int src_stride_rgb,
  395. uint8* dst_r,
  396. int dst_stride_r,
  397. uint8* dst_g,
  398. int dst_stride_g,
  399. uint8* dst_b,
  400. int dst_stride_b,
  401. int width,
  402. int height) {
  403. int y;
  404. void (*SplitRGBRow)(const uint8* src_rgb, uint8* dst_r, uint8* dst_g,
  405. uint8* dst_b, int width) = SplitRGBRow_C;
  406. // Negative height means invert the image.
  407. if (height < 0) {
  408. height = -height;
  409. dst_r = dst_r + (height - 1) * dst_stride_r;
  410. dst_g = dst_g + (height - 1) * dst_stride_g;
  411. dst_b = dst_b + (height - 1) * dst_stride_b;
  412. dst_stride_r = -dst_stride_r;
  413. dst_stride_g = -dst_stride_g;
  414. dst_stride_b = -dst_stride_b;
  415. }
  416. // Coalesce rows.
  417. if (src_stride_rgb == width * 3 && dst_stride_r == width &&
  418. dst_stride_g == width && dst_stride_b == width) {
  419. width *= height;
  420. height = 1;
  421. src_stride_rgb = dst_stride_r = dst_stride_g = dst_stride_b = 0;
  422. }
  423. #if defined(HAS_SPLITRGBROW_SSSE3)
  424. if (TestCpuFlag(kCpuHasSSSE3)) {
  425. SplitRGBRow = SplitRGBRow_Any_SSSE3;
  426. if (IS_ALIGNED(width, 16)) {
  427. SplitRGBRow = SplitRGBRow_SSSE3;
  428. }
  429. }
  430. #endif
  431. #if defined(HAS_SPLITRGBROW_NEON)
  432. if (TestCpuFlag(kCpuHasNEON)) {
  433. SplitRGBRow = SplitRGBRow_Any_NEON;
  434. if (IS_ALIGNED(width, 16)) {
  435. SplitRGBRow = SplitRGBRow_NEON;
  436. }
  437. }
  438. #endif
  439. for (y = 0; y < height; ++y) {
  440. // Copy a row of RGB.
  441. SplitRGBRow(src_rgb, dst_r, dst_g, dst_b, width);
  442. dst_r += dst_stride_r;
  443. dst_g += dst_stride_g;
  444. dst_b += dst_stride_b;
  445. src_rgb += src_stride_rgb;
  446. }
  447. }
  448. LIBYUV_API
  449. void MergeRGBPlane(const uint8* src_r,
  450. int src_stride_r,
  451. const uint8* src_g,
  452. int src_stride_g,
  453. const uint8* src_b,
  454. int src_stride_b,
  455. uint8* dst_rgb,
  456. int dst_stride_rgb,
  457. int width,
  458. int height) {
  459. int y;
  460. void (*MergeRGBRow)(const uint8* src_r, const uint8* src_g,
  461. const uint8* src_b, uint8* dst_rgb, int width) =
  462. MergeRGBRow_C;
  463. // Coalesce rows.
  464. // Negative height means invert the image.
  465. if (height < 0) {
  466. height = -height;
  467. dst_rgb = dst_rgb + (height - 1) * dst_stride_rgb;
  468. dst_stride_rgb = -dst_stride_rgb;
  469. }
  470. // Coalesce rows.
  471. if (src_stride_r == width && src_stride_g == width && src_stride_b == width &&
  472. dst_stride_rgb == width * 3) {
  473. width *= height;
  474. height = 1;
  475. src_stride_r = src_stride_g = src_stride_b = dst_stride_rgb = 0;
  476. }
  477. #if defined(HAS_MERGERGBROW_SSSE3)
  478. if (TestCpuFlag(kCpuHasSSSE3)) {
  479. MergeRGBRow = MergeRGBRow_Any_SSSE3;
  480. if (IS_ALIGNED(width, 16)) {
  481. MergeRGBRow = MergeRGBRow_SSSE3;
  482. }
  483. }
  484. #endif
  485. #if defined(HAS_MERGERGBROW_NEON)
  486. if (TestCpuFlag(kCpuHasNEON)) {
  487. MergeRGBRow = MergeRGBRow_Any_NEON;
  488. if (IS_ALIGNED(width, 16)) {
  489. MergeRGBRow = MergeRGBRow_NEON;
  490. }
  491. }
  492. #endif
  493. for (y = 0; y < height; ++y) {
  494. // Merge a row of U and V into a row of RGB.
  495. MergeRGBRow(src_r, src_g, src_b, dst_rgb, width);
  496. src_r += src_stride_r;
  497. src_g += src_stride_g;
  498. src_b += src_stride_b;
  499. dst_rgb += dst_stride_rgb;
  500. }
  501. }
  502. // Mirror a plane of data.
  503. void MirrorPlane(const uint8* src_y,
  504. int src_stride_y,
  505. uint8* dst_y,
  506. int dst_stride_y,
  507. int width,
  508. int height) {
  509. int y;
  510. void (*MirrorRow)(const uint8* src, uint8* dst, int width) = MirrorRow_C;
  511. // Negative height means invert the image.
  512. if (height < 0) {
  513. height = -height;
  514. src_y = src_y + (height - 1) * src_stride_y;
  515. src_stride_y = -src_stride_y;
  516. }
  517. #if defined(HAS_MIRRORROW_NEON)
  518. if (TestCpuFlag(kCpuHasNEON)) {
  519. MirrorRow = MirrorRow_Any_NEON;
  520. if (IS_ALIGNED(width, 16)) {
  521. MirrorRow = MirrorRow_NEON;
  522. }
  523. }
  524. #endif
  525. #if defined(HAS_MIRRORROW_SSSE3)
  526. if (TestCpuFlag(kCpuHasSSSE3)) {
  527. MirrorRow = MirrorRow_Any_SSSE3;
  528. if (IS_ALIGNED(width, 16)) {
  529. MirrorRow = MirrorRow_SSSE3;
  530. }
  531. }
  532. #endif
  533. #if defined(HAS_MIRRORROW_AVX2)
  534. if (TestCpuFlag(kCpuHasAVX2)) {
  535. MirrorRow = MirrorRow_Any_AVX2;
  536. if (IS_ALIGNED(width, 32)) {
  537. MirrorRow = MirrorRow_AVX2;
  538. }
  539. }
  540. #endif
  541. // TODO(fbarchard): Mirror on mips handle unaligned memory.
  542. #if defined(HAS_MIRRORROW_DSPR2)
  543. if (TestCpuFlag(kCpuHasDSPR2) && IS_ALIGNED(src_y, 4) &&
  544. IS_ALIGNED(src_stride_y, 4) && IS_ALIGNED(dst_y, 4) &&
  545. IS_ALIGNED(dst_stride_y, 4)) {
  546. MirrorRow = MirrorRow_DSPR2;
  547. }
  548. #endif
  549. #if defined(HAS_MIRRORROW_MSA)
  550. if (TestCpuFlag(kCpuHasMSA)) {
  551. MirrorRow = MirrorRow_Any_MSA;
  552. if (IS_ALIGNED(width, 64)) {
  553. MirrorRow = MirrorRow_MSA;
  554. }
  555. }
  556. #endif
  557. // Mirror plane
  558. for (y = 0; y < height; ++y) {
  559. MirrorRow(src_y, dst_y, width);
  560. src_y += src_stride_y;
  561. dst_y += dst_stride_y;
  562. }
  563. }
  564. // Convert YUY2 to I422.
  565. LIBYUV_API
  566. int YUY2ToI422(const uint8* src_yuy2,
  567. int src_stride_yuy2,
  568. uint8* dst_y,
  569. int dst_stride_y,
  570. uint8* dst_u,
  571. int dst_stride_u,
  572. uint8* dst_v,
  573. int dst_stride_v,
  574. int width,
  575. int height) {
  576. int y;
  577. void (*YUY2ToUV422Row)(const uint8* src_yuy2, uint8* dst_u, uint8* dst_v,
  578. int width) = YUY2ToUV422Row_C;
  579. void (*YUY2ToYRow)(const uint8* src_yuy2, uint8* dst_y, int width) =
  580. YUY2ToYRow_C;
  581. if (!src_yuy2 || !dst_y || !dst_u || !dst_v || width <= 0 || height == 0) {
  582. return -1;
  583. }
  584. // Negative height means invert the image.
  585. if (height < 0) {
  586. height = -height;
  587. src_yuy2 = src_yuy2 + (height - 1) * src_stride_yuy2;
  588. src_stride_yuy2 = -src_stride_yuy2;
  589. }
  590. // Coalesce rows.
  591. if (src_stride_yuy2 == width * 2 && dst_stride_y == width &&
  592. dst_stride_u * 2 == width && dst_stride_v * 2 == width &&
  593. width * height <= 32768) {
  594. width *= height;
  595. height = 1;
  596. src_stride_yuy2 = dst_stride_y = dst_stride_u = dst_stride_v = 0;
  597. }
  598. #if defined(HAS_YUY2TOYROW_SSE2)
  599. if (TestCpuFlag(kCpuHasSSE2)) {
  600. YUY2ToUV422Row = YUY2ToUV422Row_Any_SSE2;
  601. YUY2ToYRow = YUY2ToYRow_Any_SSE2;
  602. if (IS_ALIGNED(width, 16)) {
  603. YUY2ToUV422Row = YUY2ToUV422Row_SSE2;
  604. YUY2ToYRow = YUY2ToYRow_SSE2;
  605. }
  606. }
  607. #endif
  608. #if defined(HAS_YUY2TOYROW_AVX2)
  609. if (TestCpuFlag(kCpuHasAVX2)) {
  610. YUY2ToUV422Row = YUY2ToUV422Row_Any_AVX2;
  611. YUY2ToYRow = YUY2ToYRow_Any_AVX2;
  612. if (IS_ALIGNED(width, 32)) {
  613. YUY2ToUV422Row = YUY2ToUV422Row_AVX2;
  614. YUY2ToYRow = YUY2ToYRow_AVX2;
  615. }
  616. }
  617. #endif
  618. #if defined(HAS_YUY2TOYROW_NEON)
  619. if (TestCpuFlag(kCpuHasNEON)) {
  620. YUY2ToYRow = YUY2ToYRow_Any_NEON;
  621. YUY2ToUV422Row = YUY2ToUV422Row_Any_NEON;
  622. if (IS_ALIGNED(width, 16)) {
  623. YUY2ToYRow = YUY2ToYRow_NEON;
  624. YUY2ToUV422Row = YUY2ToUV422Row_NEON;
  625. }
  626. }
  627. #endif
  628. #if defined(HAS_YUY2TOYROW_MSA)
  629. if (TestCpuFlag(kCpuHasMSA)) {
  630. YUY2ToYRow = YUY2ToYRow_Any_MSA;
  631. YUY2ToUV422Row = YUY2ToUV422Row_Any_MSA;
  632. if (IS_ALIGNED(width, 32)) {
  633. YUY2ToYRow = YUY2ToYRow_MSA;
  634. YUY2ToUV422Row = YUY2ToUV422Row_MSA;
  635. }
  636. }
  637. #endif
  638. for (y = 0; y < height; ++y) {
  639. YUY2ToUV422Row(src_yuy2, dst_u, dst_v, width);
  640. YUY2ToYRow(src_yuy2, dst_y, width);
  641. src_yuy2 += src_stride_yuy2;
  642. dst_y += dst_stride_y;
  643. dst_u += dst_stride_u;
  644. dst_v += dst_stride_v;
  645. }
  646. return 0;
  647. }
  648. // Convert UYVY to I422.
  649. LIBYUV_API
  650. int UYVYToI422(const uint8* src_uyvy,
  651. int src_stride_uyvy,
  652. uint8* dst_y,
  653. int dst_stride_y,
  654. uint8* dst_u,
  655. int dst_stride_u,
  656. uint8* dst_v,
  657. int dst_stride_v,
  658. int width,
  659. int height) {
  660. int y;
  661. void (*UYVYToUV422Row)(const uint8* src_uyvy, uint8* dst_u, uint8* dst_v,
  662. int width) = UYVYToUV422Row_C;
  663. void (*UYVYToYRow)(const uint8* src_uyvy, uint8* dst_y, int width) =
  664. UYVYToYRow_C;
  665. if (!src_uyvy || !dst_y || !dst_u || !dst_v || width <= 0 || height == 0) {
  666. return -1;
  667. }
  668. // Negative height means invert the image.
  669. if (height < 0) {
  670. height = -height;
  671. src_uyvy = src_uyvy + (height - 1) * src_stride_uyvy;
  672. src_stride_uyvy = -src_stride_uyvy;
  673. }
  674. // Coalesce rows.
  675. if (src_stride_uyvy == width * 2 && dst_stride_y == width &&
  676. dst_stride_u * 2 == width && dst_stride_v * 2 == width &&
  677. width * height <= 32768) {
  678. width *= height;
  679. height = 1;
  680. src_stride_uyvy = dst_stride_y = dst_stride_u = dst_stride_v = 0;
  681. }
  682. #if defined(HAS_UYVYTOYROW_SSE2)
  683. if (TestCpuFlag(kCpuHasSSE2)) {
  684. UYVYToUV422Row = UYVYToUV422Row_Any_SSE2;
  685. UYVYToYRow = UYVYToYRow_Any_SSE2;
  686. if (IS_ALIGNED(width, 16)) {
  687. UYVYToUV422Row = UYVYToUV422Row_SSE2;
  688. UYVYToYRow = UYVYToYRow_SSE2;
  689. }
  690. }
  691. #endif
  692. #if defined(HAS_UYVYTOYROW_AVX2)
  693. if (TestCpuFlag(kCpuHasAVX2)) {
  694. UYVYToUV422Row = UYVYToUV422Row_Any_AVX2;
  695. UYVYToYRow = UYVYToYRow_Any_AVX2;
  696. if (IS_ALIGNED(width, 32)) {
  697. UYVYToUV422Row = UYVYToUV422Row_AVX2;
  698. UYVYToYRow = UYVYToYRow_AVX2;
  699. }
  700. }
  701. #endif
  702. #if defined(HAS_UYVYTOYROW_NEON)
  703. if (TestCpuFlag(kCpuHasNEON)) {
  704. UYVYToYRow = UYVYToYRow_Any_NEON;
  705. UYVYToUV422Row = UYVYToUV422Row_Any_NEON;
  706. if (IS_ALIGNED(width, 16)) {
  707. UYVYToYRow = UYVYToYRow_NEON;
  708. UYVYToUV422Row = UYVYToUV422Row_NEON;
  709. }
  710. }
  711. #endif
  712. #if defined(HAS_UYVYTOYROW_MSA)
  713. if (TestCpuFlag(kCpuHasMSA)) {
  714. UYVYToYRow = UYVYToYRow_Any_MSA;
  715. UYVYToUV422Row = UYVYToUV422Row_Any_MSA;
  716. if (IS_ALIGNED(width, 32)) {
  717. UYVYToYRow = UYVYToYRow_MSA;
  718. UYVYToUV422Row = UYVYToUV422Row_MSA;
  719. }
  720. }
  721. #endif
  722. for (y = 0; y < height; ++y) {
  723. UYVYToUV422Row(src_uyvy, dst_u, dst_v, width);
  724. UYVYToYRow(src_uyvy, dst_y, width);
  725. src_uyvy += src_stride_uyvy;
  726. dst_y += dst_stride_y;
  727. dst_u += dst_stride_u;
  728. dst_v += dst_stride_v;
  729. }
  730. return 0;
  731. }
  732. // Convert YUY2 to Y.
  733. LIBYUV_API
  734. int YUY2ToY(const uint8* src_yuy2,
  735. int src_stride_yuy2,
  736. uint8* dst_y,
  737. int dst_stride_y,
  738. int width,
  739. int height) {
  740. int y;
  741. void (*YUY2ToYRow)(const uint8* src_yuy2, uint8* dst_y, int width) =
  742. YUY2ToYRow_C;
  743. if (!src_yuy2 || !dst_y || width <= 0 || height == 0) {
  744. return -1;
  745. }
  746. // Negative height means invert the image.
  747. if (height < 0) {
  748. height = -height;
  749. src_yuy2 = src_yuy2 + (height - 1) * src_stride_yuy2;
  750. src_stride_yuy2 = -src_stride_yuy2;
  751. }
  752. // Coalesce rows.
  753. if (src_stride_yuy2 == width * 2 && dst_stride_y == width) {
  754. width *= height;
  755. height = 1;
  756. src_stride_yuy2 = dst_stride_y = 0;
  757. }
  758. #if defined(HAS_YUY2TOYROW_SSE2)
  759. if (TestCpuFlag(kCpuHasSSE2)) {
  760. YUY2ToYRow = YUY2ToYRow_Any_SSE2;
  761. if (IS_ALIGNED(width, 16)) {
  762. YUY2ToYRow = YUY2ToYRow_SSE2;
  763. }
  764. }
  765. #endif
  766. #if defined(HAS_YUY2TOYROW_AVX2)
  767. if (TestCpuFlag(kCpuHasAVX2)) {
  768. YUY2ToYRow = YUY2ToYRow_Any_AVX2;
  769. if (IS_ALIGNED(width, 32)) {
  770. YUY2ToYRow = YUY2ToYRow_AVX2;
  771. }
  772. }
  773. #endif
  774. #if defined(HAS_YUY2TOYROW_NEON)
  775. if (TestCpuFlag(kCpuHasNEON)) {
  776. YUY2ToYRow = YUY2ToYRow_Any_NEON;
  777. if (IS_ALIGNED(width, 16)) {
  778. YUY2ToYRow = YUY2ToYRow_NEON;
  779. }
  780. }
  781. #endif
  782. #if defined(HAS_YUY2TOYROW_MSA)
  783. if (TestCpuFlag(kCpuHasMSA)) {
  784. YUY2ToYRow = YUY2ToYRow_Any_MSA;
  785. if (IS_ALIGNED(width, 32)) {
  786. YUY2ToYRow = YUY2ToYRow_MSA;
  787. }
  788. }
  789. #endif
  790. for (y = 0; y < height; ++y) {
  791. YUY2ToYRow(src_yuy2, dst_y, width);
  792. src_yuy2 += src_stride_yuy2;
  793. dst_y += dst_stride_y;
  794. }
  795. return 0;
  796. }
  797. // Mirror I400 with optional flipping
  798. LIBYUV_API
  799. int I400Mirror(const uint8* src_y,
  800. int src_stride_y,
  801. uint8* dst_y,
  802. int dst_stride_y,
  803. int width,
  804. int height) {
  805. if (!src_y || !dst_y || width <= 0 || height == 0) {
  806. return -1;
  807. }
  808. // Negative height means invert the image.
  809. if (height < 0) {
  810. height = -height;
  811. src_y = src_y + (height - 1) * src_stride_y;
  812. src_stride_y = -src_stride_y;
  813. }
  814. MirrorPlane(src_y, src_stride_y, dst_y, dst_stride_y, width, height);
  815. return 0;
  816. }
  817. // Mirror I420 with optional flipping
  818. LIBYUV_API
  819. int I420Mirror(const uint8* src_y,
  820. int src_stride_y,
  821. const uint8* src_u,
  822. int src_stride_u,
  823. const uint8* src_v,
  824. int src_stride_v,
  825. uint8* dst_y,
  826. int dst_stride_y,
  827. uint8* dst_u,
  828. int dst_stride_u,
  829. uint8* dst_v,
  830. int dst_stride_v,
  831. int width,
  832. int height) {
  833. int halfwidth = (width + 1) >> 1;
  834. int halfheight = (height + 1) >> 1;
  835. if (!src_y || !src_u || !src_v || !dst_y || !dst_u || !dst_v || width <= 0 ||
  836. height == 0) {
  837. return -1;
  838. }
  839. // Negative height means invert the image.
  840. if (height < 0) {
  841. height = -height;
  842. halfheight = (height + 1) >> 1;
  843. src_y = src_y + (height - 1) * src_stride_y;
  844. src_u = src_u + (halfheight - 1) * src_stride_u;
  845. src_v = src_v + (halfheight - 1) * src_stride_v;
  846. src_stride_y = -src_stride_y;
  847. src_stride_u = -src_stride_u;
  848. src_stride_v = -src_stride_v;
  849. }
  850. if (dst_y) {
  851. MirrorPlane(src_y, src_stride_y, dst_y, dst_stride_y, width, height);
  852. }
  853. MirrorPlane(src_u, src_stride_u, dst_u, dst_stride_u, halfwidth, halfheight);
  854. MirrorPlane(src_v, src_stride_v, dst_v, dst_stride_v, halfwidth, halfheight);
  855. return 0;
  856. }
  857. // ARGB mirror.
  858. LIBYUV_API
  859. int ARGBMirror(const uint8* src_argb,
  860. int src_stride_argb,
  861. uint8* dst_argb,
  862. int dst_stride_argb,
  863. int width,
  864. int height) {
  865. int y;
  866. void (*ARGBMirrorRow)(const uint8* src, uint8* dst, int width) =
  867. ARGBMirrorRow_C;
  868. if (!src_argb || !dst_argb || width <= 0 || height == 0) {
  869. return -1;
  870. }
  871. // Negative height means invert the image.
  872. if (height < 0) {
  873. height = -height;
  874. src_argb = src_argb + (height - 1) * src_stride_argb;
  875. src_stride_argb = -src_stride_argb;
  876. }
  877. #if defined(HAS_ARGBMIRRORROW_NEON)
  878. if (TestCpuFlag(kCpuHasNEON)) {
  879. ARGBMirrorRow = ARGBMirrorRow_Any_NEON;
  880. if (IS_ALIGNED(width, 4)) {
  881. ARGBMirrorRow = ARGBMirrorRow_NEON;
  882. }
  883. }
  884. #endif
  885. #if defined(HAS_ARGBMIRRORROW_SSE2)
  886. if (TestCpuFlag(kCpuHasSSE2)) {
  887. ARGBMirrorRow = ARGBMirrorRow_Any_SSE2;
  888. if (IS_ALIGNED(width, 4)) {
  889. ARGBMirrorRow = ARGBMirrorRow_SSE2;
  890. }
  891. }
  892. #endif
  893. #if defined(HAS_ARGBMIRRORROW_AVX2)
  894. if (TestCpuFlag(kCpuHasAVX2)) {
  895. ARGBMirrorRow = ARGBMirrorRow_Any_AVX2;
  896. if (IS_ALIGNED(width, 8)) {
  897. ARGBMirrorRow = ARGBMirrorRow_AVX2;
  898. }
  899. }
  900. #endif
  901. #if defined(HAS_ARGBMIRRORROW_MSA)
  902. if (TestCpuFlag(kCpuHasMSA)) {
  903. ARGBMirrorRow = ARGBMirrorRow_Any_MSA;
  904. if (IS_ALIGNED(width, 16)) {
  905. ARGBMirrorRow = ARGBMirrorRow_MSA;
  906. }
  907. }
  908. #endif
  909. // Mirror plane
  910. for (y = 0; y < height; ++y) {
  911. ARGBMirrorRow(src_argb, dst_argb, width);
  912. src_argb += src_stride_argb;
  913. dst_argb += dst_stride_argb;
  914. }
  915. return 0;
  916. }
  917. // Get a blender that optimized for the CPU and pixel count.
  918. // As there are 6 blenders to choose from, the caller should try to use
  919. // the same blend function for all pixels if possible.
  920. LIBYUV_API
  921. ARGBBlendRow GetARGBBlend() {
  922. void (*ARGBBlendRow)(const uint8* src_argb, const uint8* src_argb1,
  923. uint8* dst_argb, int width) = ARGBBlendRow_C;
  924. #if defined(HAS_ARGBBLENDROW_SSSE3)
  925. if (TestCpuFlag(kCpuHasSSSE3)) {
  926. ARGBBlendRow = ARGBBlendRow_SSSE3;
  927. return ARGBBlendRow;
  928. }
  929. #endif
  930. #if defined(HAS_ARGBBLENDROW_NEON)
  931. if (TestCpuFlag(kCpuHasNEON)) {
  932. ARGBBlendRow = ARGBBlendRow_NEON;
  933. }
  934. #endif
  935. #if defined(HAS_ARGBBLENDROW_MSA)
  936. if (TestCpuFlag(kCpuHasMSA)) {
  937. ARGBBlendRow = ARGBBlendRow_MSA;
  938. }
  939. #endif
  940. return ARGBBlendRow;
  941. }
  942. // Alpha Blend 2 ARGB images and store to destination.
  943. LIBYUV_API
  944. int ARGBBlend(const uint8* src_argb0,
  945. int src_stride_argb0,
  946. const uint8* src_argb1,
  947. int src_stride_argb1,
  948. uint8* dst_argb,
  949. int dst_stride_argb,
  950. int width,
  951. int height) {
  952. int y;
  953. void (*ARGBBlendRow)(const uint8* src_argb, const uint8* src_argb1,
  954. uint8* dst_argb, int width) = GetARGBBlend();
  955. if (!src_argb0 || !src_argb1 || !dst_argb || width <= 0 || height == 0) {
  956. return -1;
  957. }
  958. // Negative height means invert the image.
  959. if (height < 0) {
  960. height = -height;
  961. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  962. dst_stride_argb = -dst_stride_argb;
  963. }
  964. // Coalesce rows.
  965. if (src_stride_argb0 == width * 4 && src_stride_argb1 == width * 4 &&
  966. dst_stride_argb == width * 4) {
  967. width *= height;
  968. height = 1;
  969. src_stride_argb0 = src_stride_argb1 = dst_stride_argb = 0;
  970. }
  971. for (y = 0; y < height; ++y) {
  972. ARGBBlendRow(src_argb0, src_argb1, dst_argb, width);
  973. src_argb0 += src_stride_argb0;
  974. src_argb1 += src_stride_argb1;
  975. dst_argb += dst_stride_argb;
  976. }
  977. return 0;
  978. }
  979. // Alpha Blend plane and store to destination.
  980. LIBYUV_API
  981. int BlendPlane(const uint8* src_y0,
  982. int src_stride_y0,
  983. const uint8* src_y1,
  984. int src_stride_y1,
  985. const uint8* alpha,
  986. int alpha_stride,
  987. uint8* dst_y,
  988. int dst_stride_y,
  989. int width,
  990. int height) {
  991. int y;
  992. void (*BlendPlaneRow)(const uint8* src0, const uint8* src1,
  993. const uint8* alpha, uint8* dst, int width) =
  994. BlendPlaneRow_C;
  995. if (!src_y0 || !src_y1 || !alpha || !dst_y || width <= 0 || height == 0) {
  996. return -1;
  997. }
  998. // Negative height means invert the image.
  999. if (height < 0) {
  1000. height = -height;
  1001. dst_y = dst_y + (height - 1) * dst_stride_y;
  1002. dst_stride_y = -dst_stride_y;
  1003. }
  1004. // Coalesce rows for Y plane.
  1005. if (src_stride_y0 == width && src_stride_y1 == width &&
  1006. alpha_stride == width && dst_stride_y == width) {
  1007. width *= height;
  1008. height = 1;
  1009. src_stride_y0 = src_stride_y1 = alpha_stride = dst_stride_y = 0;
  1010. }
  1011. #if defined(HAS_BLENDPLANEROW_SSSE3)
  1012. if (TestCpuFlag(kCpuHasSSSE3)) {
  1013. BlendPlaneRow = BlendPlaneRow_Any_SSSE3;
  1014. if (IS_ALIGNED(width, 8)) {
  1015. BlendPlaneRow = BlendPlaneRow_SSSE3;
  1016. }
  1017. }
  1018. #endif
  1019. #if defined(HAS_BLENDPLANEROW_AVX2)
  1020. if (TestCpuFlag(kCpuHasAVX2)) {
  1021. BlendPlaneRow = BlendPlaneRow_Any_AVX2;
  1022. if (IS_ALIGNED(width, 32)) {
  1023. BlendPlaneRow = BlendPlaneRow_AVX2;
  1024. }
  1025. }
  1026. #endif
  1027. for (y = 0; y < height; ++y) {
  1028. BlendPlaneRow(src_y0, src_y1, alpha, dst_y, width);
  1029. src_y0 += src_stride_y0;
  1030. src_y1 += src_stride_y1;
  1031. alpha += alpha_stride;
  1032. dst_y += dst_stride_y;
  1033. }
  1034. return 0;
  1035. }
  1036. #define MAXTWIDTH 2048
  1037. // Alpha Blend YUV images and store to destination.
  1038. LIBYUV_API
  1039. int I420Blend(const uint8* src_y0,
  1040. int src_stride_y0,
  1041. const uint8* src_u0,
  1042. int src_stride_u0,
  1043. const uint8* src_v0,
  1044. int src_stride_v0,
  1045. const uint8* src_y1,
  1046. int src_stride_y1,
  1047. const uint8* src_u1,
  1048. int src_stride_u1,
  1049. const uint8* src_v1,
  1050. int src_stride_v1,
  1051. const uint8* alpha,
  1052. int alpha_stride,
  1053. uint8* dst_y,
  1054. int dst_stride_y,
  1055. uint8* dst_u,
  1056. int dst_stride_u,
  1057. uint8* dst_v,
  1058. int dst_stride_v,
  1059. int width,
  1060. int height) {
  1061. int y;
  1062. // Half width/height for UV.
  1063. int halfwidth = (width + 1) >> 1;
  1064. void (*BlendPlaneRow)(const uint8* src0, const uint8* src1,
  1065. const uint8* alpha, uint8* dst, int width) =
  1066. BlendPlaneRow_C;
  1067. void (*ScaleRowDown2)(const uint8* src_ptr, ptrdiff_t src_stride,
  1068. uint8* dst_ptr, int dst_width) = ScaleRowDown2Box_C;
  1069. if (!src_y0 || !src_u0 || !src_v0 || !src_y1 || !src_u1 || !src_v1 ||
  1070. !alpha || !dst_y || !dst_u || !dst_v || width <= 0 || height == 0) {
  1071. return -1;
  1072. }
  1073. // Negative height means invert the image.
  1074. if (height < 0) {
  1075. height = -height;
  1076. dst_y = dst_y + (height - 1) * dst_stride_y;
  1077. dst_stride_y = -dst_stride_y;
  1078. }
  1079. // Blend Y plane.
  1080. BlendPlane(src_y0, src_stride_y0, src_y1, src_stride_y1, alpha, alpha_stride,
  1081. dst_y, dst_stride_y, width, height);
  1082. #if defined(HAS_BLENDPLANEROW_SSSE3)
  1083. if (TestCpuFlag(kCpuHasSSSE3)) {
  1084. BlendPlaneRow = BlendPlaneRow_Any_SSSE3;
  1085. if (IS_ALIGNED(halfwidth, 8)) {
  1086. BlendPlaneRow = BlendPlaneRow_SSSE3;
  1087. }
  1088. }
  1089. #endif
  1090. #if defined(HAS_BLENDPLANEROW_AVX2)
  1091. if (TestCpuFlag(kCpuHasAVX2)) {
  1092. BlendPlaneRow = BlendPlaneRow_Any_AVX2;
  1093. if (IS_ALIGNED(halfwidth, 32)) {
  1094. BlendPlaneRow = BlendPlaneRow_AVX2;
  1095. }
  1096. }
  1097. #endif
  1098. if (!IS_ALIGNED(width, 2)) {
  1099. ScaleRowDown2 = ScaleRowDown2Box_Odd_C;
  1100. }
  1101. #if defined(HAS_SCALEROWDOWN2_NEON)
  1102. if (TestCpuFlag(kCpuHasNEON)) {
  1103. ScaleRowDown2 = ScaleRowDown2Box_Odd_NEON;
  1104. if (IS_ALIGNED(width, 2)) {
  1105. ScaleRowDown2 = ScaleRowDown2Box_Any_NEON;
  1106. if (IS_ALIGNED(halfwidth, 16)) {
  1107. ScaleRowDown2 = ScaleRowDown2Box_NEON;
  1108. }
  1109. }
  1110. }
  1111. #endif
  1112. #if defined(HAS_SCALEROWDOWN2_SSSE3)
  1113. if (TestCpuFlag(kCpuHasSSSE3)) {
  1114. ScaleRowDown2 = ScaleRowDown2Box_Odd_SSSE3;
  1115. if (IS_ALIGNED(width, 2)) {
  1116. ScaleRowDown2 = ScaleRowDown2Box_Any_SSSE3;
  1117. if (IS_ALIGNED(halfwidth, 16)) {
  1118. ScaleRowDown2 = ScaleRowDown2Box_SSSE3;
  1119. }
  1120. }
  1121. }
  1122. #endif
  1123. #if defined(HAS_SCALEROWDOWN2_AVX2)
  1124. if (TestCpuFlag(kCpuHasAVX2)) {
  1125. ScaleRowDown2 = ScaleRowDown2Box_Odd_AVX2;
  1126. if (IS_ALIGNED(width, 2)) {
  1127. ScaleRowDown2 = ScaleRowDown2Box_Any_AVX2;
  1128. if (IS_ALIGNED(halfwidth, 32)) {
  1129. ScaleRowDown2 = ScaleRowDown2Box_AVX2;
  1130. }
  1131. }
  1132. }
  1133. #endif
  1134. // Row buffer for intermediate alpha pixels.
  1135. align_buffer_64(halfalpha, halfwidth);
  1136. for (y = 0; y < height; y += 2) {
  1137. // last row of odd height image use 1 row of alpha instead of 2.
  1138. if (y == (height - 1)) {
  1139. alpha_stride = 0;
  1140. }
  1141. // Subsample 2 rows of UV to half width and half height.
  1142. ScaleRowDown2(alpha, alpha_stride, halfalpha, halfwidth);
  1143. alpha += alpha_stride * 2;
  1144. BlendPlaneRow(src_u0, src_u1, halfalpha, dst_u, halfwidth);
  1145. BlendPlaneRow(src_v0, src_v1, halfalpha, dst_v, halfwidth);
  1146. src_u0 += src_stride_u0;
  1147. src_u1 += src_stride_u1;
  1148. dst_u += dst_stride_u;
  1149. src_v0 += src_stride_v0;
  1150. src_v1 += src_stride_v1;
  1151. dst_v += dst_stride_v;
  1152. }
  1153. free_aligned_buffer_64(halfalpha);
  1154. return 0;
  1155. }
  1156. // Multiply 2 ARGB images and store to destination.
  1157. LIBYUV_API
  1158. int ARGBMultiply(const uint8* src_argb0,
  1159. int src_stride_argb0,
  1160. const uint8* src_argb1,
  1161. int src_stride_argb1,
  1162. uint8* dst_argb,
  1163. int dst_stride_argb,
  1164. int width,
  1165. int height) {
  1166. int y;
  1167. void (*ARGBMultiplyRow)(const uint8* src0, const uint8* src1, uint8* dst,
  1168. int width) = ARGBMultiplyRow_C;
  1169. if (!src_argb0 || !src_argb1 || !dst_argb || width <= 0 || height == 0) {
  1170. return -1;
  1171. }
  1172. // Negative height means invert the image.
  1173. if (height < 0) {
  1174. height = -height;
  1175. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  1176. dst_stride_argb = -dst_stride_argb;
  1177. }
  1178. // Coalesce rows.
  1179. if (src_stride_argb0 == width * 4 && src_stride_argb1 == width * 4 &&
  1180. dst_stride_argb == width * 4) {
  1181. width *= height;
  1182. height = 1;
  1183. src_stride_argb0 = src_stride_argb1 = dst_stride_argb = 0;
  1184. }
  1185. #if defined(HAS_ARGBMULTIPLYROW_SSE2)
  1186. if (TestCpuFlag(kCpuHasSSE2)) {
  1187. ARGBMultiplyRow = ARGBMultiplyRow_Any_SSE2;
  1188. if (IS_ALIGNED(width, 4)) {
  1189. ARGBMultiplyRow = ARGBMultiplyRow_SSE2;
  1190. }
  1191. }
  1192. #endif
  1193. #if defined(HAS_ARGBMULTIPLYROW_AVX2)
  1194. if (TestCpuFlag(kCpuHasAVX2)) {
  1195. ARGBMultiplyRow = ARGBMultiplyRow_Any_AVX2;
  1196. if (IS_ALIGNED(width, 8)) {
  1197. ARGBMultiplyRow = ARGBMultiplyRow_AVX2;
  1198. }
  1199. }
  1200. #endif
  1201. #if defined(HAS_ARGBMULTIPLYROW_NEON)
  1202. if (TestCpuFlag(kCpuHasNEON)) {
  1203. ARGBMultiplyRow = ARGBMultiplyRow_Any_NEON;
  1204. if (IS_ALIGNED(width, 8)) {
  1205. ARGBMultiplyRow = ARGBMultiplyRow_NEON;
  1206. }
  1207. }
  1208. #endif
  1209. #if defined(HAS_ARGBMULTIPLYROW_MSA)
  1210. if (TestCpuFlag(kCpuHasMSA)) {
  1211. ARGBMultiplyRow = ARGBMultiplyRow_Any_MSA;
  1212. if (IS_ALIGNED(width, 4)) {
  1213. ARGBMultiplyRow = ARGBMultiplyRow_MSA;
  1214. }
  1215. }
  1216. #endif
  1217. // Multiply plane
  1218. for (y = 0; y < height; ++y) {
  1219. ARGBMultiplyRow(src_argb0, src_argb1, dst_argb, width);
  1220. src_argb0 += src_stride_argb0;
  1221. src_argb1 += src_stride_argb1;
  1222. dst_argb += dst_stride_argb;
  1223. }
  1224. return 0;
  1225. }
  1226. // Add 2 ARGB images and store to destination.
  1227. LIBYUV_API
  1228. int ARGBAdd(const uint8* src_argb0,
  1229. int src_stride_argb0,
  1230. const uint8* src_argb1,
  1231. int src_stride_argb1,
  1232. uint8* dst_argb,
  1233. int dst_stride_argb,
  1234. int width,
  1235. int height) {
  1236. int y;
  1237. void (*ARGBAddRow)(const uint8* src0, const uint8* src1, uint8* dst,
  1238. int width) = ARGBAddRow_C;
  1239. if (!src_argb0 || !src_argb1 || !dst_argb || width <= 0 || height == 0) {
  1240. return -1;
  1241. }
  1242. // Negative height means invert the image.
  1243. if (height < 0) {
  1244. height = -height;
  1245. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  1246. dst_stride_argb = -dst_stride_argb;
  1247. }
  1248. // Coalesce rows.
  1249. if (src_stride_argb0 == width * 4 && src_stride_argb1 == width * 4 &&
  1250. dst_stride_argb == width * 4) {
  1251. width *= height;
  1252. height = 1;
  1253. src_stride_argb0 = src_stride_argb1 = dst_stride_argb = 0;
  1254. }
  1255. #if defined(HAS_ARGBADDROW_SSE2) && (defined(_MSC_VER) && !defined(__clang__))
  1256. if (TestCpuFlag(kCpuHasSSE2)) {
  1257. ARGBAddRow = ARGBAddRow_SSE2;
  1258. }
  1259. #endif
  1260. #if defined(HAS_ARGBADDROW_SSE2) && !(defined(_MSC_VER) && !defined(__clang__))
  1261. if (TestCpuFlag(kCpuHasSSE2)) {
  1262. ARGBAddRow = ARGBAddRow_Any_SSE2;
  1263. if (IS_ALIGNED(width, 4)) {
  1264. ARGBAddRow = ARGBAddRow_SSE2;
  1265. }
  1266. }
  1267. #endif
  1268. #if defined(HAS_ARGBADDROW_AVX2)
  1269. if (TestCpuFlag(kCpuHasAVX2)) {
  1270. ARGBAddRow = ARGBAddRow_Any_AVX2;
  1271. if (IS_ALIGNED(width, 8)) {
  1272. ARGBAddRow = ARGBAddRow_AVX2;
  1273. }
  1274. }
  1275. #endif
  1276. #if defined(HAS_ARGBADDROW_NEON)
  1277. if (TestCpuFlag(kCpuHasNEON)) {
  1278. ARGBAddRow = ARGBAddRow_Any_NEON;
  1279. if (IS_ALIGNED(width, 8)) {
  1280. ARGBAddRow = ARGBAddRow_NEON;
  1281. }
  1282. }
  1283. #endif
  1284. #if defined(HAS_ARGBADDROW_MSA)
  1285. if (TestCpuFlag(kCpuHasMSA)) {
  1286. ARGBAddRow = ARGBAddRow_Any_MSA;
  1287. if (IS_ALIGNED(width, 8)) {
  1288. ARGBAddRow = ARGBAddRow_MSA;
  1289. }
  1290. }
  1291. #endif
  1292. // Add plane
  1293. for (y = 0; y < height; ++y) {
  1294. ARGBAddRow(src_argb0, src_argb1, dst_argb, width);
  1295. src_argb0 += src_stride_argb0;
  1296. src_argb1 += src_stride_argb1;
  1297. dst_argb += dst_stride_argb;
  1298. }
  1299. return 0;
  1300. }
  1301. // Subtract 2 ARGB images and store to destination.
  1302. LIBYUV_API
  1303. int ARGBSubtract(const uint8* src_argb0,
  1304. int src_stride_argb0,
  1305. const uint8* src_argb1,
  1306. int src_stride_argb1,
  1307. uint8* dst_argb,
  1308. int dst_stride_argb,
  1309. int width,
  1310. int height) {
  1311. int y;
  1312. void (*ARGBSubtractRow)(const uint8* src0, const uint8* src1, uint8* dst,
  1313. int width) = ARGBSubtractRow_C;
  1314. if (!src_argb0 || !src_argb1 || !dst_argb || width <= 0 || height == 0) {
  1315. return -1;
  1316. }
  1317. // Negative height means invert the image.
  1318. if (height < 0) {
  1319. height = -height;
  1320. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  1321. dst_stride_argb = -dst_stride_argb;
  1322. }
  1323. // Coalesce rows.
  1324. if (src_stride_argb0 == width * 4 && src_stride_argb1 == width * 4 &&
  1325. dst_stride_argb == width * 4) {
  1326. width *= height;
  1327. height = 1;
  1328. src_stride_argb0 = src_stride_argb1 = dst_stride_argb = 0;
  1329. }
  1330. #if defined(HAS_ARGBSUBTRACTROW_SSE2)
  1331. if (TestCpuFlag(kCpuHasSSE2)) {
  1332. ARGBSubtractRow = ARGBSubtractRow_Any_SSE2;
  1333. if (IS_ALIGNED(width, 4)) {
  1334. ARGBSubtractRow = ARGBSubtractRow_SSE2;
  1335. }
  1336. }
  1337. #endif
  1338. #if defined(HAS_ARGBSUBTRACTROW_AVX2)
  1339. if (TestCpuFlag(kCpuHasAVX2)) {
  1340. ARGBSubtractRow = ARGBSubtractRow_Any_AVX2;
  1341. if (IS_ALIGNED(width, 8)) {
  1342. ARGBSubtractRow = ARGBSubtractRow_AVX2;
  1343. }
  1344. }
  1345. #endif
  1346. #if defined(HAS_ARGBSUBTRACTROW_NEON)
  1347. if (TestCpuFlag(kCpuHasNEON)) {
  1348. ARGBSubtractRow = ARGBSubtractRow_Any_NEON;
  1349. if (IS_ALIGNED(width, 8)) {
  1350. ARGBSubtractRow = ARGBSubtractRow_NEON;
  1351. }
  1352. }
  1353. #endif
  1354. #if defined(HAS_ARGBSUBTRACTROW_MSA)
  1355. if (TestCpuFlag(kCpuHasMSA)) {
  1356. ARGBSubtractRow = ARGBSubtractRow_Any_MSA;
  1357. if (IS_ALIGNED(width, 8)) {
  1358. ARGBSubtractRow = ARGBSubtractRow_MSA;
  1359. }
  1360. }
  1361. #endif
  1362. // Subtract plane
  1363. for (y = 0; y < height; ++y) {
  1364. ARGBSubtractRow(src_argb0, src_argb1, dst_argb, width);
  1365. src_argb0 += src_stride_argb0;
  1366. src_argb1 += src_stride_argb1;
  1367. dst_argb += dst_stride_argb;
  1368. }
  1369. return 0;
  1370. }
  1371. // Convert I422 to RGBA with matrix
  1372. static int I422ToRGBAMatrix(const uint8* src_y,
  1373. int src_stride_y,
  1374. const uint8* src_u,
  1375. int src_stride_u,
  1376. const uint8* src_v,
  1377. int src_stride_v,
  1378. uint8* dst_rgba,
  1379. int dst_stride_rgba,
  1380. const struct YuvConstants* yuvconstants,
  1381. int width,
  1382. int height) {
  1383. int y;
  1384. void (*I422ToRGBARow)(const uint8* y_buf, const uint8* u_buf,
  1385. const uint8* v_buf, uint8* rgb_buf,
  1386. const struct YuvConstants* yuvconstants, int width) =
  1387. I422ToRGBARow_C;
  1388. if (!src_y || !src_u || !src_v || !dst_rgba || width <= 0 || height == 0) {
  1389. return -1;
  1390. }
  1391. // Negative height means invert the image.
  1392. if (height < 0) {
  1393. height = -height;
  1394. dst_rgba = dst_rgba + (height - 1) * dst_stride_rgba;
  1395. dst_stride_rgba = -dst_stride_rgba;
  1396. }
  1397. #if defined(HAS_I422TORGBAROW_SSSE3)
  1398. if (TestCpuFlag(kCpuHasSSSE3)) {
  1399. I422ToRGBARow = I422ToRGBARow_Any_SSSE3;
  1400. if (IS_ALIGNED(width, 8)) {
  1401. I422ToRGBARow = I422ToRGBARow_SSSE3;
  1402. }
  1403. }
  1404. #endif
  1405. #if defined(HAS_I422TORGBAROW_AVX2)
  1406. if (TestCpuFlag(kCpuHasAVX2)) {
  1407. I422ToRGBARow = I422ToRGBARow_Any_AVX2;
  1408. if (IS_ALIGNED(width, 16)) {
  1409. I422ToRGBARow = I422ToRGBARow_AVX2;
  1410. }
  1411. }
  1412. #endif
  1413. #if defined(HAS_I422TORGBAROW_NEON)
  1414. if (TestCpuFlag(kCpuHasNEON)) {
  1415. I422ToRGBARow = I422ToRGBARow_Any_NEON;
  1416. if (IS_ALIGNED(width, 8)) {
  1417. I422ToRGBARow = I422ToRGBARow_NEON;
  1418. }
  1419. }
  1420. #endif
  1421. #if defined(HAS_I422TORGBAROW_DSPR2)
  1422. if (TestCpuFlag(kCpuHasDSPR2) && IS_ALIGNED(width, 4) &&
  1423. IS_ALIGNED(src_y, 4) && IS_ALIGNED(src_stride_y, 4) &&
  1424. IS_ALIGNED(src_u, 2) && IS_ALIGNED(src_stride_u, 2) &&
  1425. IS_ALIGNED(src_v, 2) && IS_ALIGNED(src_stride_v, 2) &&
  1426. IS_ALIGNED(dst_rgba, 4) && IS_ALIGNED(dst_stride_rgba, 4)) {
  1427. I422ToRGBARow = I422ToRGBARow_DSPR2;
  1428. }
  1429. #endif
  1430. #if defined(HAS_I422TORGBAROW_MSA)
  1431. if (TestCpuFlag(kCpuHasMSA)) {
  1432. I422ToRGBARow = I422ToRGBARow_Any_MSA;
  1433. if (IS_ALIGNED(width, 8)) {
  1434. I422ToRGBARow = I422ToRGBARow_MSA;
  1435. }
  1436. }
  1437. #endif
  1438. for (y = 0; y < height; ++y) {
  1439. I422ToRGBARow(src_y, src_u, src_v, dst_rgba, yuvconstants, width);
  1440. dst_rgba += dst_stride_rgba;
  1441. src_y += src_stride_y;
  1442. src_u += src_stride_u;
  1443. src_v += src_stride_v;
  1444. }
  1445. return 0;
  1446. }
  1447. // Convert I422 to RGBA.
  1448. LIBYUV_API
  1449. int I422ToRGBA(const uint8* src_y,
  1450. int src_stride_y,
  1451. const uint8* src_u,
  1452. int src_stride_u,
  1453. const uint8* src_v,
  1454. int src_stride_v,
  1455. uint8* dst_rgba,
  1456. int dst_stride_rgba,
  1457. int width,
  1458. int height) {
  1459. return I422ToRGBAMatrix(src_y, src_stride_y, src_u, src_stride_u, src_v,
  1460. src_stride_v, dst_rgba, dst_stride_rgba,
  1461. &kYuvI601Constants, width, height);
  1462. }
  1463. // Convert I422 to BGRA.
  1464. LIBYUV_API
  1465. int I422ToBGRA(const uint8* src_y,
  1466. int src_stride_y,
  1467. const uint8* src_u,
  1468. int src_stride_u,
  1469. const uint8* src_v,
  1470. int src_stride_v,
  1471. uint8* dst_bgra,
  1472. int dst_stride_bgra,
  1473. int width,
  1474. int height) {
  1475. return I422ToRGBAMatrix(src_y, src_stride_y, src_v,
  1476. src_stride_v, // Swap U and V
  1477. src_u, src_stride_u, dst_bgra, dst_stride_bgra,
  1478. &kYvuI601Constants, // Use Yvu matrix
  1479. width, height);
  1480. }
  1481. // Convert NV12 to RGB565.
  1482. LIBYUV_API
  1483. int NV12ToRGB565(const uint8* src_y,
  1484. int src_stride_y,
  1485. const uint8* src_uv,
  1486. int src_stride_uv,
  1487. uint8* dst_rgb565,
  1488. int dst_stride_rgb565,
  1489. int width,
  1490. int height) {
  1491. int y;
  1492. void (*NV12ToRGB565Row)(
  1493. const uint8* y_buf, const uint8* uv_buf, uint8* rgb_buf,
  1494. const struct YuvConstants* yuvconstants, int width) = NV12ToRGB565Row_C;
  1495. if (!src_y || !src_uv || !dst_rgb565 || width <= 0 || height == 0) {
  1496. return -1;
  1497. }
  1498. // Negative height means invert the image.
  1499. if (height < 0) {
  1500. height = -height;
  1501. dst_rgb565 = dst_rgb565 + (height - 1) * dst_stride_rgb565;
  1502. dst_stride_rgb565 = -dst_stride_rgb565;
  1503. }
  1504. #if defined(HAS_NV12TORGB565ROW_SSSE3)
  1505. if (TestCpuFlag(kCpuHasSSSE3)) {
  1506. NV12ToRGB565Row = NV12ToRGB565Row_Any_SSSE3;
  1507. if (IS_ALIGNED(width, 8)) {
  1508. NV12ToRGB565Row = NV12ToRGB565Row_SSSE3;
  1509. }
  1510. }
  1511. #endif
  1512. #if defined(HAS_NV12TORGB565ROW_AVX2)
  1513. if (TestCpuFlag(kCpuHasAVX2)) {
  1514. NV12ToRGB565Row = NV12ToRGB565Row_Any_AVX2;
  1515. if (IS_ALIGNED(width, 16)) {
  1516. NV12ToRGB565Row = NV12ToRGB565Row_AVX2;
  1517. }
  1518. }
  1519. #endif
  1520. #if defined(HAS_NV12TORGB565ROW_NEON)
  1521. if (TestCpuFlag(kCpuHasNEON)) {
  1522. NV12ToRGB565Row = NV12ToRGB565Row_Any_NEON;
  1523. if (IS_ALIGNED(width, 8)) {
  1524. NV12ToRGB565Row = NV12ToRGB565Row_NEON;
  1525. }
  1526. }
  1527. #endif
  1528. #if defined(HAS_NV12TORGB565ROW_MSA)
  1529. if (TestCpuFlag(kCpuHasMSA)) {
  1530. NV12ToRGB565Row = NV12ToRGB565Row_Any_MSA;
  1531. if (IS_ALIGNED(width, 8)) {
  1532. NV12ToRGB565Row = NV12ToRGB565Row_MSA;
  1533. }
  1534. }
  1535. #endif
  1536. for (y = 0; y < height; ++y) {
  1537. NV12ToRGB565Row(src_y, src_uv, dst_rgb565, &kYuvI601Constants, width);
  1538. dst_rgb565 += dst_stride_rgb565;
  1539. src_y += src_stride_y;
  1540. if (y & 1) {
  1541. src_uv += src_stride_uv;
  1542. }
  1543. }
  1544. return 0;
  1545. }
  1546. // Convert RAW to RGB24.
  1547. LIBYUV_API
  1548. int RAWToRGB24(const uint8* src_raw,
  1549. int src_stride_raw,
  1550. uint8* dst_rgb24,
  1551. int dst_stride_rgb24,
  1552. int width,
  1553. int height) {
  1554. int y;
  1555. void (*RAWToRGB24Row)(const uint8* src_rgb, uint8* dst_rgb24, int width) =
  1556. RAWToRGB24Row_C;
  1557. if (!src_raw || !dst_rgb24 || width <= 0 || height == 0) {
  1558. return -1;
  1559. }
  1560. // Negative height means invert the image.
  1561. if (height < 0) {
  1562. height = -height;
  1563. src_raw = src_raw + (height - 1) * src_stride_raw;
  1564. src_stride_raw = -src_stride_raw;
  1565. }
  1566. // Coalesce rows.
  1567. if (src_stride_raw == width * 3 && dst_stride_rgb24 == width * 3) {
  1568. width *= height;
  1569. height = 1;
  1570. src_stride_raw = dst_stride_rgb24 = 0;
  1571. }
  1572. #if defined(HAS_RAWTORGB24ROW_SSSE3)
  1573. if (TestCpuFlag(kCpuHasSSSE3)) {
  1574. RAWToRGB24Row = RAWToRGB24Row_Any_SSSE3;
  1575. if (IS_ALIGNED(width, 8)) {
  1576. RAWToRGB24Row = RAWToRGB24Row_SSSE3;
  1577. }
  1578. }
  1579. #endif
  1580. #if defined(HAS_RAWTORGB24ROW_NEON)
  1581. if (TestCpuFlag(kCpuHasNEON)) {
  1582. RAWToRGB24Row = RAWToRGB24Row_Any_NEON;
  1583. if (IS_ALIGNED(width, 8)) {
  1584. RAWToRGB24Row = RAWToRGB24Row_NEON;
  1585. }
  1586. }
  1587. #endif
  1588. #if defined(HAS_RAWTORGB24ROW_MSA)
  1589. if (TestCpuFlag(kCpuHasMSA)) {
  1590. RAWToRGB24Row = RAWToRGB24Row_Any_MSA;
  1591. if (IS_ALIGNED(width, 16)) {
  1592. RAWToRGB24Row = RAWToRGB24Row_MSA;
  1593. }
  1594. }
  1595. #endif
  1596. for (y = 0; y < height; ++y) {
  1597. RAWToRGB24Row(src_raw, dst_rgb24, width);
  1598. src_raw += src_stride_raw;
  1599. dst_rgb24 += dst_stride_rgb24;
  1600. }
  1601. return 0;
  1602. }
  1603. LIBYUV_API
  1604. void SetPlane(uint8* dst_y,
  1605. int dst_stride_y,
  1606. int width,
  1607. int height,
  1608. uint32 value) {
  1609. int y;
  1610. void (*SetRow)(uint8 * dst, uint8 value, int width) = SetRow_C;
  1611. if (height < 0) {
  1612. height = -height;
  1613. dst_y = dst_y + (height - 1) * dst_stride_y;
  1614. dst_stride_y = -dst_stride_y;
  1615. }
  1616. // Coalesce rows.
  1617. if (dst_stride_y == width) {
  1618. width *= height;
  1619. height = 1;
  1620. dst_stride_y = 0;
  1621. }
  1622. #if defined(HAS_SETROW_NEON)
  1623. if (TestCpuFlag(kCpuHasNEON)) {
  1624. SetRow = SetRow_Any_NEON;
  1625. if (IS_ALIGNED(width, 16)) {
  1626. SetRow = SetRow_NEON;
  1627. }
  1628. }
  1629. #endif
  1630. #if defined(HAS_SETROW_X86)
  1631. if (TestCpuFlag(kCpuHasX86)) {
  1632. SetRow = SetRow_Any_X86;
  1633. if (IS_ALIGNED(width, 4)) {
  1634. SetRow = SetRow_X86;
  1635. }
  1636. }
  1637. #endif
  1638. #if defined(HAS_SETROW_ERMS)
  1639. if (TestCpuFlag(kCpuHasERMS)) {
  1640. SetRow = SetRow_ERMS;
  1641. }
  1642. #endif
  1643. #if defined(HAS_SETROW_MSA)
  1644. if (TestCpuFlag(kCpuHasMSA) && IS_ALIGNED(width, 16)) {
  1645. SetRow = SetRow_MSA;
  1646. }
  1647. #endif
  1648. // Set plane
  1649. for (y = 0; y < height; ++y) {
  1650. SetRow(dst_y, value, width);
  1651. dst_y += dst_stride_y;
  1652. }
  1653. }
  1654. // Draw a rectangle into I420
  1655. LIBYUV_API
  1656. int I420Rect(uint8* dst_y,
  1657. int dst_stride_y,
  1658. uint8* dst_u,
  1659. int dst_stride_u,
  1660. uint8* dst_v,
  1661. int dst_stride_v,
  1662. int x,
  1663. int y,
  1664. int width,
  1665. int height,
  1666. int value_y,
  1667. int value_u,
  1668. int value_v) {
  1669. int halfwidth = (width + 1) >> 1;
  1670. int halfheight = (height + 1) >> 1;
  1671. uint8* start_y = dst_y + y * dst_stride_y + x;
  1672. uint8* start_u = dst_u + (y / 2) * dst_stride_u + (x / 2);
  1673. uint8* start_v = dst_v + (y / 2) * dst_stride_v + (x / 2);
  1674. if (!dst_y || !dst_u || !dst_v || width <= 0 || height == 0 || x < 0 ||
  1675. y < 0 || value_y < 0 || value_y > 255 || value_u < 0 || value_u > 255 ||
  1676. value_v < 0 || value_v > 255) {
  1677. return -1;
  1678. }
  1679. SetPlane(start_y, dst_stride_y, width, height, value_y);
  1680. SetPlane(start_u, dst_stride_u, halfwidth, halfheight, value_u);
  1681. SetPlane(start_v, dst_stride_v, halfwidth, halfheight, value_v);
  1682. return 0;
  1683. }
  1684. // Draw a rectangle into ARGB
  1685. LIBYUV_API
  1686. int ARGBRect(uint8* dst_argb,
  1687. int dst_stride_argb,
  1688. int dst_x,
  1689. int dst_y,
  1690. int width,
  1691. int height,
  1692. uint32 value) {
  1693. int y;
  1694. void (*ARGBSetRow)(uint8 * dst_argb, uint32 value, int width) = ARGBSetRow_C;
  1695. if (!dst_argb || width <= 0 || height == 0 || dst_x < 0 || dst_y < 0) {
  1696. return -1;
  1697. }
  1698. if (height < 0) {
  1699. height = -height;
  1700. dst_argb = dst_argb + (height - 1) * dst_stride_argb;
  1701. dst_stride_argb = -dst_stride_argb;
  1702. }
  1703. dst_argb += dst_y * dst_stride_argb + dst_x * 4;
  1704. // Coalesce rows.
  1705. if (dst_stride_argb == width * 4) {
  1706. width *= height;
  1707. height = 1;
  1708. dst_stride_argb = 0;
  1709. }
  1710. #if defined(HAS_ARGBSETROW_NEON)
  1711. if (TestCpuFlag(kCpuHasNEON)) {
  1712. ARGBSetRow = ARGBSetRow_Any_NEON;
  1713. if (IS_ALIGNED(width, 4)) {
  1714. ARGBSetRow = ARGBSetRow_NEON;
  1715. }
  1716. }
  1717. #endif
  1718. #if defined(HAS_ARGBSETROW_X86)
  1719. if (TestCpuFlag(kCpuHasX86)) {
  1720. ARGBSetRow = ARGBSetRow_X86;
  1721. }
  1722. #endif
  1723. #if defined(HAS_ARGBSETROW_MSA)
  1724. if (TestCpuFlag(kCpuHasMSA)) {
  1725. ARGBSetRow = ARGBSetRow_Any_MSA;
  1726. if (IS_ALIGNED(width, 4)) {
  1727. ARGBSetRow = ARGBSetRow_MSA;
  1728. }
  1729. }
  1730. #endif
  1731. // Set plane
  1732. for (y = 0; y < height; ++y) {
  1733. ARGBSetRow(dst_argb, value, width);
  1734. dst_argb += dst_stride_argb;
  1735. }
  1736. return 0;
  1737. }
  1738. // Convert unattentuated ARGB to preattenuated ARGB.
  1739. // An unattenutated ARGB alpha blend uses the formula
  1740. // p = a * f + (1 - a) * b
  1741. // where
  1742. // p is output pixel
  1743. // f is foreground pixel
  1744. // b is background pixel
  1745. // a is alpha value from foreground pixel
  1746. // An preattenutated ARGB alpha blend uses the formula
  1747. // p = f + (1 - a) * b
  1748. // where
  1749. // f is foreground pixel premultiplied by alpha
  1750. LIBYUV_API
  1751. int ARGBAttenuate(const uint8* src_argb,
  1752. int src_stride_argb,
  1753. uint8* dst_argb,
  1754. int dst_stride_argb,
  1755. int width,
  1756. int height) {
  1757. int y;
  1758. void (*ARGBAttenuateRow)(const uint8* src_argb, uint8* dst_argb, int width) =
  1759. ARGBAttenuateRow_C;
  1760. if (!src_argb || !dst_argb || width <= 0 || height == 0) {
  1761. return -1;
  1762. }
  1763. if (height < 0) {
  1764. height = -height;
  1765. src_argb = src_argb + (height - 1) * src_stride_argb;
  1766. src_stride_argb = -src_stride_argb;
  1767. }
  1768. // Coalesce rows.
  1769. if (src_stride_argb == width * 4 && dst_stride_argb == width * 4) {
  1770. width *= height;
  1771. height = 1;
  1772. src_stride_argb = dst_stride_argb = 0;
  1773. }
  1774. #if defined(HAS_ARGBATTENUATEROW_SSSE3)
  1775. if (TestCpuFlag(kCpuHasSSSE3)) {
  1776. ARGBAttenuateRow = ARGBAttenuateRow_Any_SSSE3;
  1777. if (IS_ALIGNED(width, 4)) {
  1778. ARGBAttenuateRow = ARGBAttenuateRow_SSSE3;
  1779. }
  1780. }
  1781. #endif
  1782. #if defined(HAS_ARGBATTENUATEROW_AVX2)
  1783. if (TestCpuFlag(kCpuHasAVX2)) {
  1784. ARGBAttenuateRow = ARGBAttenuateRow_Any_AVX2;
  1785. if (IS_ALIGNED(width, 8)) {
  1786. ARGBAttenuateRow = ARGBAttenuateRow_AVX2;
  1787. }
  1788. }
  1789. #endif
  1790. #if defined(HAS_ARGBATTENUATEROW_NEON)
  1791. if (TestCpuFlag(kCpuHasNEON)) {
  1792. ARGBAttenuateRow = ARGBAttenuateRow_Any_NEON;
  1793. if (IS_ALIGNED(width, 8)) {
  1794. ARGBAttenuateRow = ARGBAttenuateRow_NEON;
  1795. }
  1796. }
  1797. #endif
  1798. #if defined(HAS_ARGBATTENUATEROW_MSA)
  1799. if (TestCpuFlag(kCpuHasMSA)) {
  1800. ARGBAttenuateRow = ARGBAttenuateRow_Any_MSA;
  1801. if (IS_ALIGNED(width, 8)) {
  1802. ARGBAttenuateRow = ARGBAttenuateRow_MSA;
  1803. }
  1804. }
  1805. #endif
  1806. for (y = 0; y < height; ++y) {
  1807. ARGBAttenuateRow(src_argb, dst_argb, width);
  1808. src_argb += src_stride_argb;
  1809. dst_argb += dst_stride_argb;
  1810. }
  1811. return 0;
  1812. }
  1813. // Convert preattentuated ARGB to unattenuated ARGB.
  1814. LIBYUV_API
  1815. int ARGBUnattenuate(const uint8* src_argb,
  1816. int src_stride_argb,
  1817. uint8* dst_argb,
  1818. int dst_stride_argb,
  1819. int width,
  1820. int height) {
  1821. int y;
  1822. void (*ARGBUnattenuateRow)(const uint8* src_argb, uint8* dst_argb,
  1823. int width) = ARGBUnattenuateRow_C;
  1824. if (!src_argb || !dst_argb || width <= 0 || height == 0) {
  1825. return -1;
  1826. }
  1827. if (height < 0) {
  1828. height = -height;
  1829. src_argb = src_argb + (height - 1) * src_stride_argb;
  1830. src_stride_argb = -src_stride_argb;
  1831. }
  1832. // Coalesce rows.
  1833. if (src_stride_argb == width * 4 && dst_stride_argb == width * 4) {
  1834. width *= height;
  1835. height = 1;
  1836. src_stride_argb = dst_stride_argb = 0;
  1837. }
  1838. #if defined(HAS_ARGBUNATTENUATEROW_SSE2)
  1839. if (TestCpuFlag(kCpuHasSSE2)) {
  1840. ARGBUnattenuateRow = ARGBUnattenuateRow_Any_SSE2;
  1841. if (IS_ALIGNED(width, 4)) {
  1842. ARGBUnattenuateRow = ARGBUnattenuateRow_SSE2;
  1843. }
  1844. }
  1845. #endif
  1846. #if defined(HAS_ARGBUNATTENUATEROW_AVX2)
  1847. if (TestCpuFlag(kCpuHasAVX2)) {
  1848. ARGBUnattenuateRow = ARGBUnattenuateRow_Any_AVX2;
  1849. if (IS_ALIGNED(width, 8)) {
  1850. ARGBUnattenuateRow = ARGBUnattenuateRow_AVX2;
  1851. }
  1852. }
  1853. #endif
  1854. // TODO(fbarchard): Neon version.
  1855. for (y = 0; y < height; ++y) {
  1856. ARGBUnattenuateRow(src_argb, dst_argb, width);
  1857. src_argb += src_stride_argb;
  1858. dst_argb += dst_stride_argb;
  1859. }
  1860. return 0;
  1861. }
  1862. // Convert ARGB to Grayed ARGB.
  1863. LIBYUV_API
  1864. int ARGBGrayTo(const uint8* src_argb,
  1865. int src_stride_argb,
  1866. uint8* dst_argb,
  1867. int dst_stride_argb,
  1868. int width,
  1869. int height) {
  1870. int y;
  1871. void (*ARGBGrayRow)(const uint8* src_argb, uint8* dst_argb, int width) =
  1872. ARGBGrayRow_C;
  1873. if (!src_argb || !dst_argb || width <= 0 || height == 0) {
  1874. return -1;
  1875. }
  1876. if (height < 0) {
  1877. height = -height;
  1878. src_argb = src_argb + (height - 1) * src_stride_argb;
  1879. src_stride_argb = -src_stride_argb;
  1880. }
  1881. // Coalesce rows.
  1882. if (src_stride_argb == width * 4 && dst_stride_argb == width * 4) {
  1883. width *= height;
  1884. height = 1;
  1885. src_stride_argb = dst_stride_argb = 0;
  1886. }
  1887. #if defined(HAS_ARGBGRAYROW_SSSE3)
  1888. if (TestCpuFlag(kCpuHasSSSE3) && IS_ALIGNED(width, 8)) {
  1889. ARGBGrayRow = ARGBGrayRow_SSSE3;
  1890. }
  1891. #endif
  1892. #if defined(HAS_ARGBGRAYROW_NEON)
  1893. if (TestCpuFlag(kCpuHasNEON) && IS_ALIGNED(width, 8)) {
  1894. ARGBGrayRow = ARGBGrayRow_NEON;
  1895. }
  1896. #endif
  1897. #if defined(HAS_ARGBGRAYROW_MSA)
  1898. if (TestCpuFlag(kCpuHasMSA) && IS_ALIGNED(width, 8)) {
  1899. ARGBGrayRow = ARGBGrayRow_MSA;
  1900. }
  1901. #endif
  1902. for (y = 0; y < height; ++y) {
  1903. ARGBGrayRow(src_argb, dst_argb, width);
  1904. src_argb += src_stride_argb;
  1905. dst_argb += dst_stride_argb;
  1906. }
  1907. return 0;
  1908. }
  1909. // Make a rectangle of ARGB gray scale.
  1910. LIBYUV_API
  1911. int ARGBGray(uint8* dst_argb,
  1912. int dst_stride_argb,
  1913. int dst_x,
  1914. int dst_y,
  1915. int width,
  1916. int height) {
  1917. int y;
  1918. void (*ARGBGrayRow)(const uint8* src_argb, uint8* dst_argb, int width) =
  1919. ARGBGrayRow_C;
  1920. uint8* dst = dst_argb + dst_y * dst_stride_argb + dst_x * 4;
  1921. if (!dst_argb || width <= 0 || height <= 0 || dst_x < 0 || dst_y < 0) {
  1922. return -1;
  1923. }
  1924. // Coalesce rows.
  1925. if (dst_stride_argb == width * 4) {
  1926. width *= height;
  1927. height = 1;
  1928. dst_stride_argb = 0;
  1929. }
  1930. #if defined(HAS_ARGBGRAYROW_SSSE3)
  1931. if (TestCpuFlag(kCpuHasSSSE3) && IS_ALIGNED(width, 8)) {
  1932. ARGBGrayRow = ARGBGrayRow_SSSE3;
  1933. }
  1934. #endif
  1935. #if defined(HAS_ARGBGRAYROW_NEON)
  1936. if (TestCpuFlag(kCpuHasNEON) && IS_ALIGNED(width, 8)) {
  1937. ARGBGrayRow = ARGBGrayRow_NEON;
  1938. }
  1939. #endif
  1940. #if defined(HAS_ARGBGRAYROW_MSA)
  1941. if (TestCpuFlag(kCpuHasMSA) && IS_ALIGNED(width, 8)) {
  1942. ARGBGrayRow = ARGBGrayRow_MSA;
  1943. }
  1944. #endif
  1945. for (y = 0; y < height; ++y) {
  1946. ARGBGrayRow(dst, dst, width);
  1947. dst += dst_stride_argb;
  1948. }
  1949. return 0;
  1950. }
  1951. // Make a rectangle of ARGB Sepia tone.
  1952. LIBYUV_API
  1953. int ARGBSepia(uint8* dst_argb,
  1954. int dst_stride_argb,
  1955. int dst_x,
  1956. int dst_y,
  1957. int width,
  1958. int height) {
  1959. int y;
  1960. void (*ARGBSepiaRow)(uint8 * dst_argb, int width) = ARGBSepiaRow_C;
  1961. uint8* dst = dst_argb + dst_y * dst_stride_argb + dst_x * 4;
  1962. if (!dst_argb || width <= 0 || height <= 0 || dst_x < 0 || dst_y < 0) {
  1963. return -1;
  1964. }
  1965. // Coalesce rows.
  1966. if (dst_stride_argb == width * 4) {
  1967. width *= height;
  1968. height = 1;
  1969. dst_stride_argb = 0;
  1970. }
  1971. #if defined(HAS_ARGBSEPIAROW_SSSE3)
  1972. if (TestCpuFlag(kCpuHasSSSE3) && IS_ALIGNED(width, 8)) {
  1973. ARGBSepiaRow = ARGBSepiaRow_SSSE3;
  1974. }
  1975. #endif
  1976. #if defined(HAS_ARGBSEPIAROW_NEON)
  1977. if (TestCpuFlag(kCpuHasNEON) && IS_ALIGNED(width, 8)) {
  1978. ARGBSepiaRow = ARGBSepiaRow_NEON;
  1979. }
  1980. #endif
  1981. #if defined(HAS_ARGBSEPIAROW_MSA)
  1982. if (TestCpuFlag(kCpuHasMSA) && IS_ALIGNED(width, 8)) {
  1983. ARGBSepiaRow = ARGBSepiaRow_MSA;
  1984. }
  1985. #endif
  1986. for (y = 0; y < height; ++y) {
  1987. ARGBSepiaRow(dst, width);
  1988. dst += dst_stride_argb;
  1989. }
  1990. return 0;
  1991. }
  1992. // Apply a 4x4 matrix to each ARGB pixel.
  1993. // Note: Normally for shading, but can be used to swizzle or invert.
  1994. LIBYUV_API
  1995. int ARGBColorMatrix(const uint8* src_argb,
  1996. int src_stride_argb,
  1997. uint8* dst_argb,
  1998. int dst_stride_argb,
  1999. const int8* matrix_argb,
  2000. int width,
  2001. int height) {
  2002. int y;
  2003. void (*ARGBColorMatrixRow)(const uint8* src_argb, uint8* dst_argb,
  2004. const int8* matrix_argb, int width) =
  2005. ARGBColorMatrixRow_C;
  2006. if (!src_argb || !dst_argb || !matrix_argb || width <= 0 || height == 0) {
  2007. return -1;
  2008. }
  2009. if (height < 0) {
  2010. height = -height;
  2011. src_argb = src_argb + (height - 1) * src_stride_argb;
  2012. src_stride_argb = -src_stride_argb;
  2013. }
  2014. // Coalesce rows.
  2015. if (src_stride_argb == width * 4 && dst_stride_argb == width * 4) {
  2016. width *= height;
  2017. height = 1;
  2018. src_stride_argb = dst_stride_argb = 0;
  2019. }
  2020. #if defined(HAS_ARGBCOLORMATRIXROW_SSSE3)
  2021. if (TestCpuFlag(kCpuHasSSSE3) && IS_ALIGNED(width, 8)) {
  2022. ARGBColorMatrixRow = ARGBColorMatrixRow_SSSE3;
  2023. }
  2024. #endif
  2025. #if defined(HAS_ARGBCOLORMATRIXROW_NEON)
  2026. if (TestCpuFlag(kCpuHasNEON) && IS_ALIGNED(width, 8)) {
  2027. ARGBColorMatrixRow = ARGBColorMatrixRow_NEON;
  2028. }
  2029. #endif
  2030. #if defined(HAS_ARGBCOLORMATRIXROW_MSA)
  2031. if (TestCpuFlag(kCpuHasMSA) && IS_ALIGNED(width, 8)) {
  2032. ARGBColorMatrixRow = ARGBColorMatrixRow_MSA;
  2033. }
  2034. #endif
  2035. for (y = 0; y < height; ++y) {
  2036. ARGBColorMatrixRow(src_argb, dst_argb, matrix_argb, width);
  2037. src_argb += src_stride_argb;
  2038. dst_argb += dst_stride_argb;
  2039. }
  2040. return 0;
  2041. }
  2042. // Apply a 4x3 matrix to each ARGB pixel.
  2043. // Deprecated.
  2044. LIBYUV_API
  2045. int RGBColorMatrix(uint8* dst_argb,
  2046. int dst_stride_argb,
  2047. const int8* matrix_rgb,
  2048. int dst_x,
  2049. int dst_y,
  2050. int width,
  2051. int height) {
  2052. SIMD_ALIGNED(int8 matrix_argb[16]);
  2053. uint8* dst = dst_argb + dst_y * dst_stride_argb + dst_x * 4;
  2054. if (!dst_argb || !matrix_rgb || width <= 0 || height <= 0 || dst_x < 0 ||
  2055. dst_y < 0) {
  2056. return -1;
  2057. }
  2058. // Convert 4x3 7 bit matrix to 4x4 6 bit matrix.
  2059. matrix_argb[0] = matrix_rgb[0] / 2;
  2060. matrix_argb[1] = matrix_rgb[1] / 2;
  2061. matrix_argb[2] = matrix_rgb[2] / 2;
  2062. matrix_argb[3] = matrix_rgb[3] / 2;
  2063. matrix_argb[4] = matrix_rgb[4] / 2;
  2064. matrix_argb[5] = matrix_rgb[5] / 2;
  2065. matrix_argb[6] = matrix_rgb[6] / 2;
  2066. matrix_argb[7] = matrix_rgb[7] / 2;
  2067. matrix_argb[8] = matrix_rgb[8] / 2;
  2068. matrix_argb[9] = matrix_rgb[9] / 2;
  2069. matrix_argb[10] = matrix_rgb[10] / 2;
  2070. matrix_argb[11] = matrix_rgb[11] / 2;
  2071. matrix_argb[14] = matrix_argb[13] = matrix_argb[12] = 0;
  2072. matrix_argb[15] = 64; // 1.0
  2073. return ARGBColorMatrix((const uint8*)(dst), dst_stride_argb, dst,
  2074. dst_stride_argb, &matrix_argb[0], width, height);
  2075. }
  2076. // Apply a color table each ARGB pixel.
  2077. // Table contains 256 ARGB values.
  2078. LIBYUV_API
  2079. int ARGBColorTable(uint8* dst_argb,
  2080. int dst_stride_argb,
  2081. const uint8* table_argb,
  2082. int dst_x,
  2083. int dst_y,
  2084. int width,
  2085. int height) {
  2086. int y;
  2087. void (*ARGBColorTableRow)(uint8 * dst_argb, const uint8* table_argb,
  2088. int width) = ARGBColorTableRow_C;
  2089. uint8* dst = dst_argb + dst_y * dst_stride_argb + dst_x * 4;
  2090. if (!dst_argb || !table_argb || width <= 0 || height <= 0 || dst_x < 0 ||
  2091. dst_y < 0) {
  2092. return -1;
  2093. }
  2094. // Coalesce rows.
  2095. if (dst_stride_argb == width * 4) {
  2096. width *= height;
  2097. height = 1;
  2098. dst_stride_argb = 0;
  2099. }
  2100. #if defined(HAS_ARGBCOLORTABLEROW_X86)
  2101. if (TestCpuFlag(kCpuHasX86)) {
  2102. ARGBColorTableRow = ARGBColorTableRow_X86;
  2103. }
  2104. #endif
  2105. for (y = 0; y < height; ++y) {
  2106. ARGBColorTableRow(dst, table_argb, width);
  2107. dst += dst_stride_argb;
  2108. }
  2109. return 0;
  2110. }
  2111. // Apply a color table each ARGB pixel but preserve destination alpha.
  2112. // Table contains 256 ARGB values.
  2113. LIBYUV_API
  2114. int RGBColorTable(uint8* dst_argb,
  2115. int dst_stride_argb,
  2116. const uint8* table_argb,
  2117. int dst_x,
  2118. int dst_y,
  2119. int width,
  2120. int height) {
  2121. int y;
  2122. void (*RGBColorTableRow)(uint8 * dst_argb, const uint8* table_argb,
  2123. int width) = RGBColorTableRow_C;
  2124. uint8* dst = dst_argb + dst_y * dst_stride_argb + dst_x * 4;
  2125. if (!dst_argb || !table_argb || width <= 0 || height <= 0 || dst_x < 0 ||
  2126. dst_y < 0) {
  2127. return -1;
  2128. }
  2129. // Coalesce rows.
  2130. if (dst_stride_argb == width * 4) {
  2131. width *= height;
  2132. height = 1;
  2133. dst_stride_argb = 0;
  2134. }
  2135. #if defined(HAS_RGBCOLORTABLEROW_X86)
  2136. if (TestCpuFlag(kCpuHasX86)) {
  2137. RGBColorTableRow = RGBColorTableRow_X86;
  2138. }
  2139. #endif
  2140. for (y = 0; y < height; ++y) {
  2141. RGBColorTableRow(dst, table_argb, width);
  2142. dst += dst_stride_argb;
  2143. }
  2144. return 0;
  2145. }
  2146. // ARGBQuantize is used to posterize art.
  2147. // e.g. rgb / qvalue * qvalue + qvalue / 2
  2148. // But the low levels implement efficiently with 3 parameters, and could be
  2149. // used for other high level operations.
  2150. // dst_argb[0] = (b * scale >> 16) * interval_size + interval_offset;
  2151. // where scale is 1 / interval_size as a fixed point value.
  2152. // The divide is replaces with a multiply by reciprocal fixed point multiply.
  2153. // Caveat - although SSE2 saturates, the C function does not and should be used
  2154. // with care if doing anything but quantization.
  2155. LIBYUV_API
  2156. int ARGBQuantize(uint8* dst_argb,
  2157. int dst_stride_argb,
  2158. int scale,
  2159. int interval_size,
  2160. int interval_offset,
  2161. int dst_x,
  2162. int dst_y,
  2163. int width,
  2164. int height) {
  2165. int y;
  2166. void (*ARGBQuantizeRow)(uint8 * dst_argb, int scale, int interval_size,
  2167. int interval_offset, int width) = ARGBQuantizeRow_C;
  2168. uint8* dst = dst_argb + dst_y * dst_stride_argb + dst_x * 4;
  2169. if (!dst_argb || width <= 0 || height <= 0 || dst_x < 0 || dst_y < 0 ||
  2170. interval_size < 1 || interval_size > 255) {
  2171. return -1;
  2172. }
  2173. // Coalesce rows.
  2174. if (dst_stride_argb == width * 4) {
  2175. width *= height;
  2176. height = 1;
  2177. dst_stride_argb = 0;
  2178. }
  2179. #if defined(HAS_ARGBQUANTIZEROW_SSE2)
  2180. if (TestCpuFlag(kCpuHasSSE2) && IS_ALIGNED(width, 4)) {
  2181. ARGBQuantizeRow = ARGBQuantizeRow_SSE2;
  2182. }
  2183. #endif
  2184. #if defined(HAS_ARGBQUANTIZEROW_NEON)
  2185. if (TestCpuFlag(kCpuHasNEON) && IS_ALIGNED(width, 8)) {
  2186. ARGBQuantizeRow = ARGBQuantizeRow_NEON;
  2187. }
  2188. #endif
  2189. #if defined(HAS_ARGBQUANTIZEROW_MSA)
  2190. if (TestCpuFlag(kCpuHasMSA) && IS_ALIGNED(width, 8)) {
  2191. ARGBQuantizeRow = ARGBQuantizeRow_MSA;
  2192. }
  2193. #endif
  2194. for (y = 0; y < height; ++y) {
  2195. ARGBQuantizeRow(dst, scale, interval_size, interval_offset, width);
  2196. dst += dst_stride_argb;
  2197. }
  2198. return 0;
  2199. }
  2200. // Computes table of cumulative sum for image where the value is the sum
  2201. // of all values above and to the left of the entry. Used by ARGBBlur.
  2202. LIBYUV_API
  2203. int ARGBComputeCumulativeSum(const uint8* src_argb,
  2204. int src_stride_argb,
  2205. int32* dst_cumsum,
  2206. int dst_stride32_cumsum,
  2207. int width,
  2208. int height) {
  2209. int y;
  2210. void (*ComputeCumulativeSumRow)(const uint8* row, int32* cumsum,
  2211. const int32* previous_cumsum, int width) =
  2212. ComputeCumulativeSumRow_C;
  2213. int32* previous_cumsum = dst_cumsum;
  2214. if (!dst_cumsum || !src_argb || width <= 0 || height <= 0) {
  2215. return -1;
  2216. }
  2217. #if defined(HAS_CUMULATIVESUMTOAVERAGEROW_SSE2)
  2218. if (TestCpuFlag(kCpuHasSSE2)) {
  2219. ComputeCumulativeSumRow = ComputeCumulativeSumRow_SSE2;
  2220. }
  2221. #endif
  2222. memset(dst_cumsum, 0, width * sizeof(dst_cumsum[0]) * 4); // 4 int per pixel.
  2223. for (y = 0; y < height; ++y) {
  2224. ComputeCumulativeSumRow(src_argb, dst_cumsum, previous_cumsum, width);
  2225. previous_cumsum = dst_cumsum;
  2226. dst_cumsum += dst_stride32_cumsum;
  2227. src_argb += src_stride_argb;
  2228. }
  2229. return 0;
  2230. }
  2231. // Blur ARGB image.
  2232. // Caller should allocate CumulativeSum table of width * height * 16 bytes
  2233. // aligned to 16 byte boundary. height can be radius * 2 + 2 to save memory
  2234. // as the buffer is treated as circular.
  2235. LIBYUV_API
  2236. int ARGBBlur(const uint8* src_argb,
  2237. int src_stride_argb,
  2238. uint8* dst_argb,
  2239. int dst_stride_argb,
  2240. int32* dst_cumsum,
  2241. int dst_stride32_cumsum,
  2242. int width,
  2243. int height,
  2244. int radius) {
  2245. int y;
  2246. void (*ComputeCumulativeSumRow)(const uint8* row, int32* cumsum,
  2247. const int32* previous_cumsum, int width) =
  2248. ComputeCumulativeSumRow_C;
  2249. void (*CumulativeSumToAverageRow)(const int32* topleft, const int32* botleft,
  2250. int width, int area, uint8* dst,
  2251. int count) = CumulativeSumToAverageRow_C;
  2252. int32* cumsum_bot_row;
  2253. int32* max_cumsum_bot_row;
  2254. int32* cumsum_top_row;
  2255. if (!src_argb || !dst_argb || width <= 0 || height == 0) {
  2256. return -1;
  2257. }
  2258. if (height < 0) {
  2259. height = -height;
  2260. src_argb = src_argb + (height - 1) * src_stride_argb;
  2261. src_stride_argb = -src_stride_argb;
  2262. }
  2263. if (radius > height) {
  2264. radius = height;
  2265. }
  2266. if (radius > (width / 2 - 1)) {
  2267. radius = width / 2 - 1;
  2268. }
  2269. if (radius <= 0) {
  2270. return -1;
  2271. }
  2272. #if defined(HAS_CUMULATIVESUMTOAVERAGEROW_SSE2)
  2273. if (TestCpuFlag(kCpuHasSSE2)) {
  2274. ComputeCumulativeSumRow = ComputeCumulativeSumRow_SSE2;
  2275. CumulativeSumToAverageRow = CumulativeSumToAverageRow_SSE2;
  2276. }
  2277. #endif
  2278. // Compute enough CumulativeSum for first row to be blurred. After this
  2279. // one row of CumulativeSum is updated at a time.
  2280. ARGBComputeCumulativeSum(src_argb, src_stride_argb, dst_cumsum,
  2281. dst_stride32_cumsum, width, radius);
  2282. src_argb = src_argb + radius * src_stride_argb;
  2283. cumsum_bot_row = &dst_cumsum[(radius - 1) * dst_stride32_cumsum];
  2284. max_cumsum_bot_row = &dst_cumsum[(radius * 2 + 2) * dst_stride32_cumsum];
  2285. cumsum_top_row = &dst_cumsum[0];
  2286. for (y = 0; y < height; ++y) {
  2287. int top_y = ((y - radius - 1) >= 0) ? (y - radius - 1) : 0;
  2288. int bot_y = ((y + radius) < height) ? (y + radius) : (height - 1);
  2289. int area = radius * (bot_y - top_y);
  2290. int boxwidth = radius * 4;
  2291. int x;
  2292. int n;
  2293. // Increment cumsum_top_row pointer with circular buffer wrap around.
  2294. if (top_y) {
  2295. cumsum_top_row += dst_stride32_cumsum;
  2296. if (cumsum_top_row >= max_cumsum_bot_row) {
  2297. cumsum_top_row = dst_cumsum;
  2298. }
  2299. }
  2300. // Increment cumsum_bot_row pointer with circular buffer wrap around and
  2301. // then fill in a row of CumulativeSum.
  2302. if ((y + radius) < height) {
  2303. const int32* prev_cumsum_bot_row = cumsum_bot_row;
  2304. cumsum_bot_row += dst_stride32_cumsum;
  2305. if (cumsum_bot_row >= max_cumsum_bot_row) {
  2306. cumsum_bot_row = dst_cumsum;
  2307. }
  2308. ComputeCumulativeSumRow(src_argb, cumsum_bot_row, prev_cumsum_bot_row,
  2309. width);
  2310. src_argb += src_stride_argb;
  2311. }
  2312. // Left clipped.
  2313. for (x = 0; x < radius + 1; ++x) {
  2314. CumulativeSumToAverageRow(cumsum_top_row, cumsum_bot_row, boxwidth, area,
  2315. &dst_argb[x * 4], 1);
  2316. area += (bot_y - top_y);
  2317. boxwidth += 4;
  2318. }
  2319. // Middle unclipped.
  2320. n = (width - 1) - radius - x + 1;
  2321. CumulativeSumToAverageRow(cumsum_top_row, cumsum_bot_row, boxwidth, area,
  2322. &dst_argb[x * 4], n);
  2323. // Right clipped.
  2324. for (x += n; x <= width - 1; ++x) {
  2325. area -= (bot_y - top_y);
  2326. boxwidth -= 4;
  2327. CumulativeSumToAverageRow(cumsum_top_row + (x - radius - 1) * 4,
  2328. cumsum_bot_row + (x - radius - 1) * 4, boxwidth,
  2329. area, &dst_argb[x * 4], 1);
  2330. }
  2331. dst_argb += dst_stride_argb;
  2332. }
  2333. return 0;
  2334. }
  2335. // Multiply ARGB image by a specified ARGB value.
  2336. LIBYUV_API
  2337. int ARGBShade(const uint8* src_argb,
  2338. int src_stride_argb,
  2339. uint8* dst_argb,
  2340. int dst_stride_argb,
  2341. int width,
  2342. int height,
  2343. uint32 value) {
  2344. int y;
  2345. void (*ARGBShadeRow)(const uint8* src_argb, uint8* dst_argb, int width,
  2346. uint32 value) = ARGBShadeRow_C;
  2347. if (!src_argb || !dst_argb || width <= 0 || height == 0 || value == 0u) {
  2348. return -1;
  2349. }
  2350. if (height < 0) {
  2351. height = -height;
  2352. src_argb = src_argb + (height - 1) * src_stride_argb;
  2353. src_stride_argb = -src_stride_argb;
  2354. }
  2355. // Coalesce rows.
  2356. if (src_stride_argb == width * 4 && dst_stride_argb == width * 4) {
  2357. width *= height;
  2358. height = 1;
  2359. src_stride_argb = dst_stride_argb = 0;
  2360. }
  2361. #if defined(HAS_ARGBSHADEROW_SSE2)
  2362. if (TestCpuFlag(kCpuHasSSE2) && IS_ALIGNED(width, 4)) {
  2363. ARGBShadeRow = ARGBShadeRow_SSE2;
  2364. }
  2365. #endif
  2366. #if defined(HAS_ARGBSHADEROW_NEON)
  2367. if (TestCpuFlag(kCpuHasNEON) && IS_ALIGNED(width, 8)) {
  2368. ARGBShadeRow = ARGBShadeRow_NEON;
  2369. }
  2370. #endif
  2371. #if defined(HAS_ARGBSHADEROW_MSA)
  2372. if (TestCpuFlag(kCpuHasMSA) && IS_ALIGNED(width, 4)) {
  2373. ARGBShadeRow = ARGBShadeRow_MSA;
  2374. }
  2375. #endif
  2376. for (y = 0; y < height; ++y) {
  2377. ARGBShadeRow(src_argb, dst_argb, width, value);
  2378. src_argb += src_stride_argb;
  2379. dst_argb += dst_stride_argb;
  2380. }
  2381. return 0;
  2382. }
  2383. // Interpolate 2 planes by specified amount (0 to 255).
  2384. LIBYUV_API
  2385. int InterpolatePlane(const uint8* src0,
  2386. int src_stride0,
  2387. const uint8* src1,
  2388. int src_stride1,
  2389. uint8* dst,
  2390. int dst_stride,
  2391. int width,
  2392. int height,
  2393. int interpolation) {
  2394. int y;
  2395. void (*InterpolateRow)(uint8 * dst_ptr, const uint8* src_ptr,
  2396. ptrdiff_t src_stride, int dst_width,
  2397. int source_y_fraction) = InterpolateRow_C;
  2398. if (!src0 || !src1 || !dst || width <= 0 || height == 0) {
  2399. return -1;
  2400. }
  2401. // Negative height means invert the image.
  2402. if (height < 0) {
  2403. height = -height;
  2404. dst = dst + (height - 1) * dst_stride;
  2405. dst_stride = -dst_stride;
  2406. }
  2407. // Coalesce rows.
  2408. if (src_stride0 == width && src_stride1 == width && dst_stride == width) {
  2409. width *= height;
  2410. height = 1;
  2411. src_stride0 = src_stride1 = dst_stride = 0;
  2412. }
  2413. #if defined(HAS_INTERPOLATEROW_SSSE3)
  2414. if (TestCpuFlag(kCpuHasSSSE3)) {
  2415. InterpolateRow = InterpolateRow_Any_SSSE3;
  2416. if (IS_ALIGNED(width, 16)) {
  2417. InterpolateRow = InterpolateRow_SSSE3;
  2418. }
  2419. }
  2420. #endif
  2421. #if defined(HAS_INTERPOLATEROW_AVX2)
  2422. if (TestCpuFlag(kCpuHasAVX2)) {
  2423. InterpolateRow = InterpolateRow_Any_AVX2;
  2424. if (IS_ALIGNED(width, 32)) {
  2425. InterpolateRow = InterpolateRow_AVX2;
  2426. }
  2427. }
  2428. #endif
  2429. #if defined(HAS_INTERPOLATEROW_NEON)
  2430. if (TestCpuFlag(kCpuHasNEON)) {
  2431. InterpolateRow = InterpolateRow_Any_NEON;
  2432. if (IS_ALIGNED(width, 16)) {
  2433. InterpolateRow = InterpolateRow_NEON;
  2434. }
  2435. }
  2436. #endif
  2437. #if defined(HAS_INTERPOLATEROW_DSPR2)
  2438. if (TestCpuFlag(kCpuHasDSPR2) && IS_ALIGNED(src0, 4) &&
  2439. IS_ALIGNED(src_stride0, 4) && IS_ALIGNED(src1, 4) &&
  2440. IS_ALIGNED(src_stride1, 4) && IS_ALIGNED(dst, 4) &&
  2441. IS_ALIGNED(dst_stride, 4) && IS_ALIGNED(width, 4)) {
  2442. InterpolateRow = InterpolateRow_DSPR2;
  2443. }
  2444. #endif
  2445. #if defined(HAS_INTERPOLATEROW_MSA)
  2446. if (TestCpuFlag(kCpuHasMSA)) {
  2447. InterpolateRow = InterpolateRow_Any_MSA;
  2448. if (IS_ALIGNED(width, 32)) {
  2449. InterpolateRow = InterpolateRow_MSA;
  2450. }
  2451. }
  2452. #endif
  2453. for (y = 0; y < height; ++y) {
  2454. InterpolateRow(dst, src0, src1 - src0, width, interpolation);
  2455. src0 += src_stride0;
  2456. src1 += src_stride1;
  2457. dst += dst_stride;
  2458. }
  2459. return 0;
  2460. }
  2461. // Interpolate 2 ARGB images by specified amount (0 to 255).
  2462. LIBYUV_API
  2463. int ARGBInterpolate(const uint8* src_argb0,
  2464. int src_stride_argb0,
  2465. const uint8* src_argb1,
  2466. int src_stride_argb1,
  2467. uint8* dst_argb,
  2468. int dst_stride_argb,
  2469. int width,
  2470. int height,
  2471. int interpolation) {
  2472. return InterpolatePlane(src_argb0, src_stride_argb0, src_argb1,
  2473. src_stride_argb1, dst_argb, dst_stride_argb,
  2474. width * 4, height, interpolation);
  2475. }
  2476. // Interpolate 2 YUV images by specified amount (0 to 255).
  2477. LIBYUV_API
  2478. int I420Interpolate(const uint8* src0_y,
  2479. int src0_stride_y,
  2480. const uint8* src0_u,
  2481. int src0_stride_u,
  2482. const uint8* src0_v,
  2483. int src0_stride_v,
  2484. const uint8* src1_y,
  2485. int src1_stride_y,
  2486. const uint8* src1_u,
  2487. int src1_stride_u,
  2488. const uint8* src1_v,
  2489. int src1_stride_v,
  2490. uint8* dst_y,
  2491. int dst_stride_y,
  2492. uint8* dst_u,
  2493. int dst_stride_u,
  2494. uint8* dst_v,
  2495. int dst_stride_v,
  2496. int width,
  2497. int height,
  2498. int interpolation) {
  2499. int halfwidth = (width + 1) >> 1;
  2500. int halfheight = (height + 1) >> 1;
  2501. if (!src0_y || !src0_u || !src0_v || !src1_y || !src1_u || !src1_v ||
  2502. !dst_y || !dst_u || !dst_v || width <= 0 || height == 0) {
  2503. return -1;
  2504. }
  2505. InterpolatePlane(src0_y, src0_stride_y, src1_y, src1_stride_y, dst_y,
  2506. dst_stride_y, width, height, interpolation);
  2507. InterpolatePlane(src0_u, src0_stride_u, src1_u, src1_stride_u, dst_u,
  2508. dst_stride_u, halfwidth, halfheight, interpolation);
  2509. InterpolatePlane(src0_v, src0_stride_v, src1_v, src1_stride_v, dst_v,
  2510. dst_stride_v, halfwidth, halfheight, interpolation);
  2511. return 0;
  2512. }
  2513. // Shuffle ARGB channel order. e.g. BGRA to ARGB.
  2514. LIBYUV_API
  2515. int ARGBShuffle(const uint8* src_bgra,
  2516. int src_stride_bgra,
  2517. uint8* dst_argb,
  2518. int dst_stride_argb,
  2519. const uint8* shuffler,
  2520. int width,
  2521. int height) {
  2522. int y;
  2523. void (*ARGBShuffleRow)(const uint8* src_bgra, uint8* dst_argb,
  2524. const uint8* shuffler, int width) = ARGBShuffleRow_C;
  2525. if (!src_bgra || !dst_argb || width <= 0 || height == 0) {
  2526. return -1;
  2527. }
  2528. // Negative height means invert the image.
  2529. if (height < 0) {
  2530. height = -height;
  2531. src_bgra = src_bgra + (height - 1) * src_stride_bgra;
  2532. src_stride_bgra = -src_stride_bgra;
  2533. }
  2534. // Coalesce rows.
  2535. if (src_stride_bgra == width * 4 && dst_stride_argb == width * 4) {
  2536. width *= height;
  2537. height = 1;
  2538. src_stride_bgra = dst_stride_argb = 0;
  2539. }
  2540. #if defined(HAS_ARGBSHUFFLEROW_SSE2)
  2541. if (TestCpuFlag(kCpuHasSSE2)) {
  2542. ARGBShuffleRow = ARGBShuffleRow_Any_SSE2;
  2543. if (IS_ALIGNED(width, 4)) {
  2544. ARGBShuffleRow = ARGBShuffleRow_SSE2;
  2545. }
  2546. }
  2547. #endif
  2548. #if defined(HAS_ARGBSHUFFLEROW_SSSE3)
  2549. if (TestCpuFlag(kCpuHasSSSE3)) {
  2550. ARGBShuffleRow = ARGBShuffleRow_Any_SSSE3;
  2551. if (IS_ALIGNED(width, 8)) {
  2552. ARGBShuffleRow = ARGBShuffleRow_SSSE3;
  2553. }
  2554. }
  2555. #endif
  2556. #if defined(HAS_ARGBSHUFFLEROW_AVX2)
  2557. if (TestCpuFlag(kCpuHasAVX2)) {
  2558. ARGBShuffleRow = ARGBShuffleRow_Any_AVX2;
  2559. if (IS_ALIGNED(width, 16)) {
  2560. ARGBShuffleRow = ARGBShuffleRow_AVX2;
  2561. }
  2562. }
  2563. #endif
  2564. #if defined(HAS_ARGBSHUFFLEROW_NEON)
  2565. if (TestCpuFlag(kCpuHasNEON)) {
  2566. ARGBShuffleRow = ARGBShuffleRow_Any_NEON;
  2567. if (IS_ALIGNED(width, 4)) {
  2568. ARGBShuffleRow = ARGBShuffleRow_NEON;
  2569. }
  2570. }
  2571. #endif
  2572. #if defined(HAS_ARGBSHUFFLEROW_MSA)
  2573. if (TestCpuFlag(kCpuHasMSA)) {
  2574. ARGBShuffleRow = ARGBShuffleRow_Any_MSA;
  2575. if (IS_ALIGNED(width, 8)) {
  2576. ARGBShuffleRow = ARGBShuffleRow_MSA;
  2577. }
  2578. }
  2579. #endif
  2580. for (y = 0; y < height; ++y) {
  2581. ARGBShuffleRow(src_bgra, dst_argb, shuffler, width);
  2582. src_bgra += src_stride_bgra;
  2583. dst_argb += dst_stride_argb;
  2584. }
  2585. return 0;
  2586. }
  2587. // Sobel ARGB effect.
  2588. static int ARGBSobelize(const uint8* src_argb,
  2589. int src_stride_argb,
  2590. uint8* dst_argb,
  2591. int dst_stride_argb,
  2592. int width,
  2593. int height,
  2594. void (*SobelRow)(const uint8* src_sobelx,
  2595. const uint8* src_sobely,
  2596. uint8* dst,
  2597. int width)) {
  2598. int y;
  2599. void (*ARGBToYJRow)(const uint8* src_argb, uint8* dst_g, int width) =
  2600. ARGBToYJRow_C;
  2601. void (*SobelYRow)(const uint8* src_y0, const uint8* src_y1, uint8* dst_sobely,
  2602. int width) = SobelYRow_C;
  2603. void (*SobelXRow)(const uint8* src_y0, const uint8* src_y1,
  2604. const uint8* src_y2, uint8* dst_sobely, int width) =
  2605. SobelXRow_C;
  2606. const int kEdge = 16; // Extra pixels at start of row for extrude/align.
  2607. if (!src_argb || !dst_argb || width <= 0 || height == 0) {
  2608. return -1;
  2609. }
  2610. // Negative height means invert the image.
  2611. if (height < 0) {
  2612. height = -height;
  2613. src_argb = src_argb + (height - 1) * src_stride_argb;
  2614. src_stride_argb = -src_stride_argb;
  2615. }
  2616. #if defined(HAS_ARGBTOYJROW_SSSE3)
  2617. if (TestCpuFlag(kCpuHasSSSE3)) {
  2618. ARGBToYJRow = ARGBToYJRow_Any_SSSE3;
  2619. if (IS_ALIGNED(width, 16)) {
  2620. ARGBToYJRow = ARGBToYJRow_SSSE3;
  2621. }
  2622. }
  2623. #endif
  2624. #if defined(HAS_ARGBTOYJROW_AVX2)
  2625. if (TestCpuFlag(kCpuHasAVX2)) {
  2626. ARGBToYJRow = ARGBToYJRow_Any_AVX2;
  2627. if (IS_ALIGNED(width, 32)) {
  2628. ARGBToYJRow = ARGBToYJRow_AVX2;
  2629. }
  2630. }
  2631. #endif
  2632. #if defined(HAS_ARGBTOYJROW_NEON)
  2633. if (TestCpuFlag(kCpuHasNEON)) {
  2634. ARGBToYJRow = ARGBToYJRow_Any_NEON;
  2635. if (IS_ALIGNED(width, 8)) {
  2636. ARGBToYJRow = ARGBToYJRow_NEON;
  2637. }
  2638. }
  2639. #endif
  2640. #if defined(HAS_ARGBTOYJROW_MSA)
  2641. if (TestCpuFlag(kCpuHasMSA)) {
  2642. ARGBToYJRow = ARGBToYJRow_Any_MSA;
  2643. if (IS_ALIGNED(width, 16)) {
  2644. ARGBToYJRow = ARGBToYJRow_MSA;
  2645. }
  2646. }
  2647. #endif
  2648. #if defined(HAS_SOBELYROW_SSE2)
  2649. if (TestCpuFlag(kCpuHasSSE2)) {
  2650. SobelYRow = SobelYRow_SSE2;
  2651. }
  2652. #endif
  2653. #if defined(HAS_SOBELYROW_NEON)
  2654. if (TestCpuFlag(kCpuHasNEON)) {
  2655. SobelYRow = SobelYRow_NEON;
  2656. }
  2657. #endif
  2658. #if defined(HAS_SOBELYROW_MSA)
  2659. if (TestCpuFlag(kCpuHasMSA)) {
  2660. SobelYRow = SobelYRow_MSA;
  2661. }
  2662. #endif
  2663. #if defined(HAS_SOBELXROW_SSE2)
  2664. if (TestCpuFlag(kCpuHasSSE2)) {
  2665. SobelXRow = SobelXRow_SSE2;
  2666. }
  2667. #endif
  2668. #if defined(HAS_SOBELXROW_NEON)
  2669. if (TestCpuFlag(kCpuHasNEON)) {
  2670. SobelXRow = SobelXRow_NEON;
  2671. }
  2672. #endif
  2673. #if defined(HAS_SOBELXROW_MSA)
  2674. if (TestCpuFlag(kCpuHasMSA)) {
  2675. SobelXRow = SobelXRow_MSA;
  2676. }
  2677. #endif
  2678. {
  2679. // 3 rows with edges before/after.
  2680. const int kRowSize = (width + kEdge + 31) & ~31;
  2681. align_buffer_64(rows, kRowSize * 2 + (kEdge + kRowSize * 3 + kEdge));
  2682. uint8* row_sobelx = rows;
  2683. uint8* row_sobely = rows + kRowSize;
  2684. uint8* row_y = rows + kRowSize * 2;
  2685. // Convert first row.
  2686. uint8* row_y0 = row_y + kEdge;
  2687. uint8* row_y1 = row_y0 + kRowSize;
  2688. uint8* row_y2 = row_y1 + kRowSize;
  2689. ARGBToYJRow(src_argb, row_y0, width);
  2690. row_y0[-1] = row_y0[0];
  2691. memset(row_y0 + width, row_y0[width - 1], 16); // Extrude 16 for valgrind.
  2692. ARGBToYJRow(src_argb, row_y1, width);
  2693. row_y1[-1] = row_y1[0];
  2694. memset(row_y1 + width, row_y1[width - 1], 16);
  2695. memset(row_y2 + width, 0, 16);
  2696. for (y = 0; y < height; ++y) {
  2697. // Convert next row of ARGB to G.
  2698. if (y < (height - 1)) {
  2699. src_argb += src_stride_argb;
  2700. }
  2701. ARGBToYJRow(src_argb, row_y2, width);
  2702. row_y2[-1] = row_y2[0];
  2703. row_y2[width] = row_y2[width - 1];
  2704. SobelXRow(row_y0 - 1, row_y1 - 1, row_y2 - 1, row_sobelx, width);
  2705. SobelYRow(row_y0 - 1, row_y2 - 1, row_sobely, width);
  2706. SobelRow(row_sobelx, row_sobely, dst_argb, width);
  2707. // Cycle thru circular queue of 3 row_y buffers.
  2708. {
  2709. uint8* row_yt = row_y0;
  2710. row_y0 = row_y1;
  2711. row_y1 = row_y2;
  2712. row_y2 = row_yt;
  2713. }
  2714. dst_argb += dst_stride_argb;
  2715. }
  2716. free_aligned_buffer_64(rows);
  2717. }
  2718. return 0;
  2719. }
  2720. // Sobel ARGB effect.
  2721. LIBYUV_API
  2722. int ARGBSobel(const uint8* src_argb,
  2723. int src_stride_argb,
  2724. uint8* dst_argb,
  2725. int dst_stride_argb,
  2726. int width,
  2727. int height) {
  2728. void (*SobelRow)(const uint8* src_sobelx, const uint8* src_sobely,
  2729. uint8* dst_argb, int width) = SobelRow_C;
  2730. #if defined(HAS_SOBELROW_SSE2)
  2731. if (TestCpuFlag(kCpuHasSSE2)) {
  2732. SobelRow = SobelRow_Any_SSE2;
  2733. if (IS_ALIGNED(width, 16)) {
  2734. SobelRow = SobelRow_SSE2;
  2735. }
  2736. }
  2737. #endif
  2738. #if defined(HAS_SOBELROW_NEON)
  2739. if (TestCpuFlag(kCpuHasNEON)) {
  2740. SobelRow = SobelRow_Any_NEON;
  2741. if (IS_ALIGNED(width, 8)) {
  2742. SobelRow = SobelRow_NEON;
  2743. }
  2744. }
  2745. #endif
  2746. #if defined(HAS_SOBELROW_MSA)
  2747. if (TestCpuFlag(kCpuHasMSA)) {
  2748. SobelRow = SobelRow_Any_MSA;
  2749. if (IS_ALIGNED(width, 16)) {
  2750. SobelRow = SobelRow_MSA;
  2751. }
  2752. }
  2753. #endif
  2754. return ARGBSobelize(src_argb, src_stride_argb, dst_argb, dst_stride_argb,
  2755. width, height, SobelRow);
  2756. }
  2757. // Sobel ARGB effect with planar output.
  2758. LIBYUV_API
  2759. int ARGBSobelToPlane(const uint8* src_argb,
  2760. int src_stride_argb,
  2761. uint8* dst_y,
  2762. int dst_stride_y,
  2763. int width,
  2764. int height) {
  2765. void (*SobelToPlaneRow)(const uint8* src_sobelx, const uint8* src_sobely,
  2766. uint8* dst_, int width) = SobelToPlaneRow_C;
  2767. #if defined(HAS_SOBELTOPLANEROW_SSE2)
  2768. if (TestCpuFlag(kCpuHasSSE2)) {
  2769. SobelToPlaneRow = SobelToPlaneRow_Any_SSE2;
  2770. if (IS_ALIGNED(width, 16)) {
  2771. SobelToPlaneRow = SobelToPlaneRow_SSE2;
  2772. }
  2773. }
  2774. #endif
  2775. #if defined(HAS_SOBELTOPLANEROW_NEON)
  2776. if (TestCpuFlag(kCpuHasNEON)) {
  2777. SobelToPlaneRow = SobelToPlaneRow_Any_NEON;
  2778. if (IS_ALIGNED(width, 16)) {
  2779. SobelToPlaneRow = SobelToPlaneRow_NEON;
  2780. }
  2781. }
  2782. #endif
  2783. #if defined(HAS_SOBELTOPLANEROW_MSA)
  2784. if (TestCpuFlag(kCpuHasMSA)) {
  2785. SobelToPlaneRow = SobelToPlaneRow_Any_MSA;
  2786. if (IS_ALIGNED(width, 32)) {
  2787. SobelToPlaneRow = SobelToPlaneRow_MSA;
  2788. }
  2789. }
  2790. #endif
  2791. return ARGBSobelize(src_argb, src_stride_argb, dst_y, dst_stride_y, width,
  2792. height, SobelToPlaneRow);
  2793. }
  2794. // SobelXY ARGB effect.
  2795. // Similar to Sobel, but also stores Sobel X in R and Sobel Y in B. G = Sobel.
  2796. LIBYUV_API
  2797. int ARGBSobelXY(const uint8* src_argb,
  2798. int src_stride_argb,
  2799. uint8* dst_argb,
  2800. int dst_stride_argb,
  2801. int width,
  2802. int height) {
  2803. void (*SobelXYRow)(const uint8* src_sobelx, const uint8* src_sobely,
  2804. uint8* dst_argb, int width) = SobelXYRow_C;
  2805. #if defined(HAS_SOBELXYROW_SSE2)
  2806. if (TestCpuFlag(kCpuHasSSE2)) {
  2807. SobelXYRow = SobelXYRow_Any_SSE2;
  2808. if (IS_ALIGNED(width, 16)) {
  2809. SobelXYRow = SobelXYRow_SSE2;
  2810. }
  2811. }
  2812. #endif
  2813. #if defined(HAS_SOBELXYROW_NEON)
  2814. if (TestCpuFlag(kCpuHasNEON)) {
  2815. SobelXYRow = SobelXYRow_Any_NEON;
  2816. if (IS_ALIGNED(width, 8)) {
  2817. SobelXYRow = SobelXYRow_NEON;
  2818. }
  2819. }
  2820. #endif
  2821. #if defined(HAS_SOBELXYROW_MSA)
  2822. if (TestCpuFlag(kCpuHasMSA)) {
  2823. SobelXYRow = SobelXYRow_Any_MSA;
  2824. if (IS_ALIGNED(width, 16)) {
  2825. SobelXYRow = SobelXYRow_MSA;
  2826. }
  2827. }
  2828. #endif
  2829. return ARGBSobelize(src_argb, src_stride_argb, dst_argb, dst_stride_argb,
  2830. width, height, SobelXYRow);
  2831. }
  2832. // Apply a 4x4 polynomial to each ARGB pixel.
  2833. LIBYUV_API
  2834. int ARGBPolynomial(const uint8* src_argb,
  2835. int src_stride_argb,
  2836. uint8* dst_argb,
  2837. int dst_stride_argb,
  2838. const float* poly,
  2839. int width,
  2840. int height) {
  2841. int y;
  2842. void (*ARGBPolynomialRow)(const uint8* src_argb, uint8* dst_argb,
  2843. const float* poly, int width) = ARGBPolynomialRow_C;
  2844. if (!src_argb || !dst_argb || !poly || width <= 0 || height == 0) {
  2845. return -1;
  2846. }
  2847. // Negative height means invert the image.
  2848. if (height < 0) {
  2849. height = -height;
  2850. src_argb = src_argb + (height - 1) * src_stride_argb;
  2851. src_stride_argb = -src_stride_argb;
  2852. }
  2853. // Coalesce rows.
  2854. if (src_stride_argb == width * 4 && dst_stride_argb == width * 4) {
  2855. width *= height;
  2856. height = 1;
  2857. src_stride_argb = dst_stride_argb = 0;
  2858. }
  2859. #if defined(HAS_ARGBPOLYNOMIALROW_SSE2)
  2860. if (TestCpuFlag(kCpuHasSSE2) && IS_ALIGNED(width, 2)) {
  2861. ARGBPolynomialRow = ARGBPolynomialRow_SSE2;
  2862. }
  2863. #endif
  2864. #if defined(HAS_ARGBPOLYNOMIALROW_AVX2)
  2865. if (TestCpuFlag(kCpuHasAVX2) && TestCpuFlag(kCpuHasFMA3) &&
  2866. IS_ALIGNED(width, 2)) {
  2867. ARGBPolynomialRow = ARGBPolynomialRow_AVX2;
  2868. }
  2869. #endif
  2870. for (y = 0; y < height; ++y) {
  2871. ARGBPolynomialRow(src_argb, dst_argb, poly, width);
  2872. src_argb += src_stride_argb;
  2873. dst_argb += dst_stride_argb;
  2874. }
  2875. return 0;
  2876. }
  2877. // Convert plane of 16 bit shorts to half floats.
  2878. // Source values are multiplied by scale before storing as half float.
  2879. LIBYUV_API
  2880. int HalfFloatPlane(const uint16* src_y,
  2881. int src_stride_y,
  2882. uint16* dst_y,
  2883. int dst_stride_y,
  2884. float scale,
  2885. int width,
  2886. int height) {
  2887. int y;
  2888. void (*HalfFloatRow)(const uint16* src, uint16* dst, float scale, int width) =
  2889. HalfFloatRow_C;
  2890. if (!src_y || !dst_y || width <= 0 || height == 0) {
  2891. return -1;
  2892. }
  2893. src_stride_y >>= 1;
  2894. dst_stride_y >>= 1;
  2895. // Negative height means invert the image.
  2896. if (height < 0) {
  2897. height = -height;
  2898. src_y = src_y + (height - 1) * src_stride_y;
  2899. src_stride_y = -src_stride_y;
  2900. }
  2901. // Coalesce rows.
  2902. if (src_stride_y == width && dst_stride_y == width) {
  2903. width *= height;
  2904. height = 1;
  2905. src_stride_y = dst_stride_y = 0;
  2906. }
  2907. #if defined(HAS_HALFFLOATROW_SSE2)
  2908. if (TestCpuFlag(kCpuHasSSE2)) {
  2909. HalfFloatRow = HalfFloatRow_Any_SSE2;
  2910. if (IS_ALIGNED(width, 8)) {
  2911. HalfFloatRow = HalfFloatRow_SSE2;
  2912. }
  2913. }
  2914. #endif
  2915. #if defined(HAS_HALFFLOATROW_AVX2)
  2916. if (TestCpuFlag(kCpuHasAVX2)) {
  2917. HalfFloatRow = HalfFloatRow_Any_AVX2;
  2918. if (IS_ALIGNED(width, 16)) {
  2919. HalfFloatRow = HalfFloatRow_AVX2;
  2920. }
  2921. }
  2922. #endif
  2923. #if defined(HAS_HALFFLOATROW_F16C)
  2924. if (TestCpuFlag(kCpuHasAVX2) && TestCpuFlag(kCpuHasF16C)) {
  2925. HalfFloatRow =
  2926. (scale == 1.0f) ? HalfFloat1Row_Any_F16C : HalfFloatRow_Any_F16C;
  2927. if (IS_ALIGNED(width, 16)) {
  2928. HalfFloatRow = (scale == 1.0f) ? HalfFloat1Row_F16C : HalfFloatRow_F16C;
  2929. }
  2930. }
  2931. #endif
  2932. #if defined(HAS_HALFFLOATROW_NEON)
  2933. if (TestCpuFlag(kCpuHasNEON)) {
  2934. HalfFloatRow =
  2935. (scale == 1.0f) ? HalfFloat1Row_Any_NEON : HalfFloatRow_Any_NEON;
  2936. if (IS_ALIGNED(width, 8)) {
  2937. HalfFloatRow = (scale == 1.0f) ? HalfFloat1Row_NEON : HalfFloatRow_NEON;
  2938. }
  2939. }
  2940. #endif
  2941. #if defined(HAS_HALFFLOATROW_MSA)
  2942. if (TestCpuFlag(kCpuHasMSA)) {
  2943. HalfFloatRow = HalfFloatRow_Any_MSA;
  2944. if (IS_ALIGNED(width, 32)) {
  2945. HalfFloatRow = HalfFloatRow_MSA;
  2946. }
  2947. }
  2948. #endif
  2949. for (y = 0; y < height; ++y) {
  2950. HalfFloatRow(src_y, dst_y, scale, width);
  2951. src_y += src_stride_y;
  2952. dst_y += dst_stride_y;
  2953. }
  2954. return 0;
  2955. }
  2956. // Apply a lumacolortable to each ARGB pixel.
  2957. LIBYUV_API
  2958. int ARGBLumaColorTable(const uint8* src_argb,
  2959. int src_stride_argb,
  2960. uint8* dst_argb,
  2961. int dst_stride_argb,
  2962. const uint8* luma,
  2963. int width,
  2964. int height) {
  2965. int y;
  2966. void (*ARGBLumaColorTableRow)(
  2967. const uint8* src_argb, uint8* dst_argb, int width, const uint8* luma,
  2968. const uint32 lumacoeff) = ARGBLumaColorTableRow_C;
  2969. if (!src_argb || !dst_argb || !luma || width <= 0 || height == 0) {
  2970. return -1;
  2971. }
  2972. // Negative height means invert the image.
  2973. if (height < 0) {
  2974. height = -height;
  2975. src_argb = src_argb + (height - 1) * src_stride_argb;
  2976. src_stride_argb = -src_stride_argb;
  2977. }
  2978. // Coalesce rows.
  2979. if (src_stride_argb == width * 4 && dst_stride_argb == width * 4) {
  2980. width *= height;
  2981. height = 1;
  2982. src_stride_argb = dst_stride_argb = 0;
  2983. }
  2984. #if defined(HAS_ARGBLUMACOLORTABLEROW_SSSE3)
  2985. if (TestCpuFlag(kCpuHasSSSE3) && IS_ALIGNED(width, 4)) {
  2986. ARGBLumaColorTableRow = ARGBLumaColorTableRow_SSSE3;
  2987. }
  2988. #endif
  2989. for (y = 0; y < height; ++y) {
  2990. ARGBLumaColorTableRow(src_argb, dst_argb, width, luma, 0x00264b0f);
  2991. src_argb += src_stride_argb;
  2992. dst_argb += dst_stride_argb;
  2993. }
  2994. return 0;
  2995. }
  2996. // Copy Alpha from one ARGB image to another.
  2997. LIBYUV_API
  2998. int ARGBCopyAlpha(const uint8* src_argb,
  2999. int src_stride_argb,
  3000. uint8* dst_argb,
  3001. int dst_stride_argb,
  3002. int width,
  3003. int height) {
  3004. int y;
  3005. void (*ARGBCopyAlphaRow)(const uint8* src_argb, uint8* dst_argb, int width) =
  3006. ARGBCopyAlphaRow_C;
  3007. if (!src_argb || !dst_argb || width <= 0 || height == 0) {
  3008. return -1;
  3009. }
  3010. // Negative height means invert the image.
  3011. if (height < 0) {
  3012. height = -height;
  3013. src_argb = src_argb + (height - 1) * src_stride_argb;
  3014. src_stride_argb = -src_stride_argb;
  3015. }
  3016. // Coalesce rows.
  3017. if (src_stride_argb == width * 4 && dst_stride_argb == width * 4) {
  3018. width *= height;
  3019. height = 1;
  3020. src_stride_argb = dst_stride_argb = 0;
  3021. }
  3022. #if defined(HAS_ARGBCOPYALPHAROW_SSE2)
  3023. if (TestCpuFlag(kCpuHasSSE2)) {
  3024. ARGBCopyAlphaRow = ARGBCopyAlphaRow_Any_SSE2;
  3025. if (IS_ALIGNED(width, 8)) {
  3026. ARGBCopyAlphaRow = ARGBCopyAlphaRow_SSE2;
  3027. }
  3028. }
  3029. #endif
  3030. #if defined(HAS_ARGBCOPYALPHAROW_AVX2)
  3031. if (TestCpuFlag(kCpuHasAVX2)) {
  3032. ARGBCopyAlphaRow = ARGBCopyAlphaRow_Any_AVX2;
  3033. if (IS_ALIGNED(width, 16)) {
  3034. ARGBCopyAlphaRow = ARGBCopyAlphaRow_AVX2;
  3035. }
  3036. }
  3037. #endif
  3038. for (y = 0; y < height; ++y) {
  3039. ARGBCopyAlphaRow(src_argb, dst_argb, width);
  3040. src_argb += src_stride_argb;
  3041. dst_argb += dst_stride_argb;
  3042. }
  3043. return 0;
  3044. }
  3045. // Extract just the alpha channel from ARGB.
  3046. LIBYUV_API
  3047. int ARGBExtractAlpha(const uint8* src_argb,
  3048. int src_stride,
  3049. uint8* dst_a,
  3050. int dst_stride,
  3051. int width,
  3052. int height) {
  3053. if (!src_argb || !dst_a || width <= 0 || height == 0) {
  3054. return -1;
  3055. }
  3056. // Negative height means invert the image.
  3057. if (height < 0) {
  3058. height = -height;
  3059. src_argb += (height - 1) * src_stride;
  3060. src_stride = -src_stride;
  3061. }
  3062. // Coalesce rows.
  3063. if (src_stride == width * 4 && dst_stride == width) {
  3064. width *= height;
  3065. height = 1;
  3066. src_stride = dst_stride = 0;
  3067. }
  3068. void (*ARGBExtractAlphaRow)(const uint8* src_argb, uint8* dst_a, int width) =
  3069. ARGBExtractAlphaRow_C;
  3070. #if defined(HAS_ARGBEXTRACTALPHAROW_SSE2)
  3071. if (TestCpuFlag(kCpuHasSSE2)) {
  3072. ARGBExtractAlphaRow = IS_ALIGNED(width, 8) ? ARGBExtractAlphaRow_SSE2
  3073. : ARGBExtractAlphaRow_Any_SSE2;
  3074. }
  3075. #endif
  3076. #if defined(HAS_ARGBEXTRACTALPHAROW_AVX2)
  3077. if (TestCpuFlag(kCpuHasAVX2)) {
  3078. ARGBExtractAlphaRow = IS_ALIGNED(width, 32) ? ARGBExtractAlphaRow_AVX2
  3079. : ARGBExtractAlphaRow_Any_AVX2;
  3080. }
  3081. #endif
  3082. #if defined(HAS_ARGBEXTRACTALPHAROW_NEON)
  3083. if (TestCpuFlag(kCpuHasNEON)) {
  3084. ARGBExtractAlphaRow = IS_ALIGNED(width, 16) ? ARGBExtractAlphaRow_NEON
  3085. : ARGBExtractAlphaRow_Any_NEON;
  3086. }
  3087. #endif
  3088. #if defined(HAS_ARGBEXTRACTALPHAROW_MSA)
  3089. if (TestCpuFlag(kCpuHasMSA)) {
  3090. ARGBExtractAlphaRow = IS_ALIGNED(width, 16) ? ARGBExtractAlphaRow_MSA
  3091. : ARGBExtractAlphaRow_Any_MSA;
  3092. }
  3093. #endif
  3094. for (int y = 0; y < height; ++y) {
  3095. ARGBExtractAlphaRow(src_argb, dst_a, width);
  3096. src_argb += src_stride;
  3097. dst_a += dst_stride;
  3098. }
  3099. return 0;
  3100. }
  3101. // Copy a planar Y channel to the alpha channel of a destination ARGB image.
  3102. LIBYUV_API
  3103. int ARGBCopyYToAlpha(const uint8* src_y,
  3104. int src_stride_y,
  3105. uint8* dst_argb,
  3106. int dst_stride_argb,
  3107. int width,
  3108. int height) {
  3109. int y;
  3110. void (*ARGBCopyYToAlphaRow)(const uint8* src_y, uint8* dst_argb, int width) =
  3111. ARGBCopyYToAlphaRow_C;
  3112. if (!src_y || !dst_argb || width <= 0 || height == 0) {
  3113. return -1;
  3114. }
  3115. // Negative height means invert the image.
  3116. if (height < 0) {
  3117. height = -height;
  3118. src_y = src_y + (height - 1) * src_stride_y;
  3119. src_stride_y = -src_stride_y;
  3120. }
  3121. // Coalesce rows.
  3122. if (src_stride_y == width && dst_stride_argb == width * 4) {
  3123. width *= height;
  3124. height = 1;
  3125. src_stride_y = dst_stride_argb = 0;
  3126. }
  3127. #if defined(HAS_ARGBCOPYYTOALPHAROW_SSE2)
  3128. if (TestCpuFlag(kCpuHasSSE2)) {
  3129. ARGBCopyYToAlphaRow = ARGBCopyYToAlphaRow_Any_SSE2;
  3130. if (IS_ALIGNED(width, 8)) {
  3131. ARGBCopyYToAlphaRow = ARGBCopyYToAlphaRow_SSE2;
  3132. }
  3133. }
  3134. #endif
  3135. #if defined(HAS_ARGBCOPYYTOALPHAROW_AVX2)
  3136. if (TestCpuFlag(kCpuHasAVX2)) {
  3137. ARGBCopyYToAlphaRow = ARGBCopyYToAlphaRow_Any_AVX2;
  3138. if (IS_ALIGNED(width, 16)) {
  3139. ARGBCopyYToAlphaRow = ARGBCopyYToAlphaRow_AVX2;
  3140. }
  3141. }
  3142. #endif
  3143. for (y = 0; y < height; ++y) {
  3144. ARGBCopyYToAlphaRow(src_y, dst_argb, width);
  3145. src_y += src_stride_y;
  3146. dst_argb += dst_stride_argb;
  3147. }
  3148. return 0;
  3149. }
  3150. // TODO(fbarchard): Consider if width is even Y channel can be split
  3151. // directly. A SplitUVRow_Odd function could copy the remaining chroma.
  3152. LIBYUV_API
  3153. int YUY2ToNV12(const uint8* src_yuy2,
  3154. int src_stride_yuy2,
  3155. uint8* dst_y,
  3156. int dst_stride_y,
  3157. uint8* dst_uv,
  3158. int dst_stride_uv,
  3159. int width,
  3160. int height) {
  3161. int y;
  3162. int halfwidth = (width + 1) >> 1;
  3163. void (*SplitUVRow)(const uint8* src_uv, uint8* dst_u, uint8* dst_v,
  3164. int width) = SplitUVRow_C;
  3165. void (*InterpolateRow)(uint8 * dst_ptr, const uint8* src_ptr,
  3166. ptrdiff_t src_stride, int dst_width,
  3167. int source_y_fraction) = InterpolateRow_C;
  3168. if (!src_yuy2 || !dst_y || !dst_uv || width <= 0 || height == 0) {
  3169. return -1;
  3170. }
  3171. // Negative height means invert the image.
  3172. if (height < 0) {
  3173. height = -height;
  3174. src_yuy2 = src_yuy2 + (height - 1) * src_stride_yuy2;
  3175. src_stride_yuy2 = -src_stride_yuy2;
  3176. }
  3177. #if defined(HAS_SPLITUVROW_SSE2)
  3178. if (TestCpuFlag(kCpuHasSSE2)) {
  3179. SplitUVRow = SplitUVRow_Any_SSE2;
  3180. if (IS_ALIGNED(width, 16)) {
  3181. SplitUVRow = SplitUVRow_SSE2;
  3182. }
  3183. }
  3184. #endif
  3185. #if defined(HAS_SPLITUVROW_AVX2)
  3186. if (TestCpuFlag(kCpuHasAVX2)) {
  3187. SplitUVRow = SplitUVRow_Any_AVX2;
  3188. if (IS_ALIGNED(width, 32)) {
  3189. SplitUVRow = SplitUVRow_AVX2;
  3190. }
  3191. }
  3192. #endif
  3193. #if defined(HAS_SPLITUVROW_NEON)
  3194. if (TestCpuFlag(kCpuHasNEON)) {
  3195. SplitUVRow = SplitUVRow_Any_NEON;
  3196. if (IS_ALIGNED(width, 16)) {
  3197. SplitUVRow = SplitUVRow_NEON;
  3198. }
  3199. }
  3200. #endif
  3201. #if defined(HAS_SPLITUVROW_MSA)
  3202. if (TestCpuFlag(kCpuHasMSA)) {
  3203. SplitUVRow = SplitUVRow_Any_MSA;
  3204. if (IS_ALIGNED(width, 32)) {
  3205. SplitUVRow = SplitUVRow_MSA;
  3206. }
  3207. }
  3208. #endif
  3209. #if defined(HAS_INTERPOLATEROW_SSSE3)
  3210. if (TestCpuFlag(kCpuHasSSSE3)) {
  3211. InterpolateRow = InterpolateRow_Any_SSSE3;
  3212. if (IS_ALIGNED(width, 16)) {
  3213. InterpolateRow = InterpolateRow_SSSE3;
  3214. }
  3215. }
  3216. #endif
  3217. #if defined(HAS_INTERPOLATEROW_AVX2)
  3218. if (TestCpuFlag(kCpuHasAVX2)) {
  3219. InterpolateRow = InterpolateRow_Any_AVX2;
  3220. if (IS_ALIGNED(width, 32)) {
  3221. InterpolateRow = InterpolateRow_AVX2;
  3222. }
  3223. }
  3224. #endif
  3225. #if defined(HAS_INTERPOLATEROW_NEON)
  3226. if (TestCpuFlag(kCpuHasNEON)) {
  3227. InterpolateRow = InterpolateRow_Any_NEON;
  3228. if (IS_ALIGNED(width, 16)) {
  3229. InterpolateRow = InterpolateRow_NEON;
  3230. }
  3231. }
  3232. #endif
  3233. #if defined(HAS_INTERPOLATEROW_MSA)
  3234. if (TestCpuFlag(kCpuHasMSA)) {
  3235. InterpolateRow = InterpolateRow_Any_MSA;
  3236. if (IS_ALIGNED(width, 32)) {
  3237. InterpolateRow = InterpolateRow_MSA;
  3238. }
  3239. }
  3240. #endif
  3241. {
  3242. int awidth = halfwidth * 2;
  3243. // row of y and 2 rows of uv
  3244. align_buffer_64(rows, awidth * 3);
  3245. for (y = 0; y < height - 1; y += 2) {
  3246. // Split Y from UV.
  3247. SplitUVRow(src_yuy2, rows, rows + awidth, awidth);
  3248. memcpy(dst_y, rows, width);
  3249. SplitUVRow(src_yuy2 + src_stride_yuy2, rows, rows + awidth * 2, awidth);
  3250. memcpy(dst_y + dst_stride_y, rows, width);
  3251. InterpolateRow(dst_uv, rows + awidth, awidth, awidth, 128);
  3252. src_yuy2 += src_stride_yuy2 * 2;
  3253. dst_y += dst_stride_y * 2;
  3254. dst_uv += dst_stride_uv;
  3255. }
  3256. if (height & 1) {
  3257. // Split Y from UV.
  3258. SplitUVRow(src_yuy2, rows, dst_uv, awidth);
  3259. memcpy(dst_y, rows, width);
  3260. }
  3261. free_aligned_buffer_64(rows);
  3262. }
  3263. return 0;
  3264. }
  3265. LIBYUV_API
  3266. int UYVYToNV12(const uint8* src_uyvy,
  3267. int src_stride_uyvy,
  3268. uint8* dst_y,
  3269. int dst_stride_y,
  3270. uint8* dst_uv,
  3271. int dst_stride_uv,
  3272. int width,
  3273. int height) {
  3274. int y;
  3275. int halfwidth = (width + 1) >> 1;
  3276. void (*SplitUVRow)(const uint8* src_uv, uint8* dst_u, uint8* dst_v,
  3277. int width) = SplitUVRow_C;
  3278. void (*InterpolateRow)(uint8 * dst_ptr, const uint8* src_ptr,
  3279. ptrdiff_t src_stride, int dst_width,
  3280. int source_y_fraction) = InterpolateRow_C;
  3281. if (!src_uyvy || !dst_y || !dst_uv || width <= 0 || height == 0) {
  3282. return -1;
  3283. }
  3284. // Negative height means invert the image.
  3285. if (height < 0) {
  3286. height = -height;
  3287. src_uyvy = src_uyvy + (height - 1) * src_stride_uyvy;
  3288. src_stride_uyvy = -src_stride_uyvy;
  3289. }
  3290. #if defined(HAS_SPLITUVROW_SSE2)
  3291. if (TestCpuFlag(kCpuHasSSE2)) {
  3292. SplitUVRow = SplitUVRow_Any_SSE2;
  3293. if (IS_ALIGNED(width, 16)) {
  3294. SplitUVRow = SplitUVRow_SSE2;
  3295. }
  3296. }
  3297. #endif
  3298. #if defined(HAS_SPLITUVROW_AVX2)
  3299. if (TestCpuFlag(kCpuHasAVX2)) {
  3300. SplitUVRow = SplitUVRow_Any_AVX2;
  3301. if (IS_ALIGNED(width, 32)) {
  3302. SplitUVRow = SplitUVRow_AVX2;
  3303. }
  3304. }
  3305. #endif
  3306. #if defined(HAS_SPLITUVROW_NEON)
  3307. if (TestCpuFlag(kCpuHasNEON)) {
  3308. SplitUVRow = SplitUVRow_Any_NEON;
  3309. if (IS_ALIGNED(width, 16)) {
  3310. SplitUVRow = SplitUVRow_NEON;
  3311. }
  3312. }
  3313. #endif
  3314. #if defined(HAS_SPLITUVROW_MSA)
  3315. if (TestCpuFlag(kCpuHasMSA)) {
  3316. SplitUVRow = SplitUVRow_Any_MSA;
  3317. if (IS_ALIGNED(width, 32)) {
  3318. SplitUVRow = SplitUVRow_MSA;
  3319. }
  3320. }
  3321. #endif
  3322. #if defined(HAS_INTERPOLATEROW_SSSE3)
  3323. if (TestCpuFlag(kCpuHasSSSE3)) {
  3324. InterpolateRow = InterpolateRow_Any_SSSE3;
  3325. if (IS_ALIGNED(width, 16)) {
  3326. InterpolateRow = InterpolateRow_SSSE3;
  3327. }
  3328. }
  3329. #endif
  3330. #if defined(HAS_INTERPOLATEROW_AVX2)
  3331. if (TestCpuFlag(kCpuHasAVX2)) {
  3332. InterpolateRow = InterpolateRow_Any_AVX2;
  3333. if (IS_ALIGNED(width, 32)) {
  3334. InterpolateRow = InterpolateRow_AVX2;
  3335. }
  3336. }
  3337. #endif
  3338. #if defined(HAS_INTERPOLATEROW_NEON)
  3339. if (TestCpuFlag(kCpuHasNEON)) {
  3340. InterpolateRow = InterpolateRow_Any_NEON;
  3341. if (IS_ALIGNED(width, 16)) {
  3342. InterpolateRow = InterpolateRow_NEON;
  3343. }
  3344. }
  3345. #endif
  3346. #if defined(HAS_INTERPOLATEROW_MSA)
  3347. if (TestCpuFlag(kCpuHasMSA)) {
  3348. InterpolateRow = InterpolateRow_Any_MSA;
  3349. if (IS_ALIGNED(width, 32)) {
  3350. InterpolateRow = InterpolateRow_MSA;
  3351. }
  3352. }
  3353. #endif
  3354. {
  3355. int awidth = halfwidth * 2;
  3356. // row of y and 2 rows of uv
  3357. align_buffer_64(rows, awidth * 3);
  3358. for (y = 0; y < height - 1; y += 2) {
  3359. // Split Y from UV.
  3360. SplitUVRow(src_uyvy, rows + awidth, rows, awidth);
  3361. memcpy(dst_y, rows, width);
  3362. SplitUVRow(src_uyvy + src_stride_uyvy, rows + awidth * 2, rows, awidth);
  3363. memcpy(dst_y + dst_stride_y, rows, width);
  3364. InterpolateRow(dst_uv, rows + awidth, awidth, awidth, 128);
  3365. src_uyvy += src_stride_uyvy * 2;
  3366. dst_y += dst_stride_y * 2;
  3367. dst_uv += dst_stride_uv;
  3368. }
  3369. if (height & 1) {
  3370. // Split Y from UV.
  3371. SplitUVRow(src_uyvy, dst_uv, rows, awidth);
  3372. memcpy(dst_y, rows, width);
  3373. }
  3374. free_aligned_buffer_64(rows);
  3375. }
  3376. return 0;
  3377. }
  3378. #ifdef __cplusplus
  3379. } // extern "C"
  3380. } // namespace libyuv
  3381. #endif