TestImageDecoder.cpp 58 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372
  1. /*
  2. * Copyright (c) 2021, the SerenityOS developers.
  3. * Copyright (c) 2021, Brian Gianforcaro <bgianf@serenityos.org>
  4. *
  5. * SPDX-License-Identifier: BSD-2-Clause
  6. */
  7. #include <AK/ByteString.h>
  8. #include <LibCore/MappedFile.h>
  9. #include <LibGfx/ImageFormats/BMPLoader.h>
  10. #include <LibGfx/ImageFormats/DDSLoader.h>
  11. #include <LibGfx/ImageFormats/GIFLoader.h>
  12. #include <LibGfx/ImageFormats/ICOLoader.h>
  13. #include <LibGfx/ImageFormats/ILBMLoader.h>
  14. #include <LibGfx/ImageFormats/ImageDecoder.h>
  15. #include <LibGfx/ImageFormats/JBIG2Loader.h>
  16. #include <LibGfx/ImageFormats/JPEG2000Loader.h>
  17. #include <LibGfx/ImageFormats/JPEGLoader.h>
  18. #include <LibGfx/ImageFormats/JPEGXLLoader.h>
  19. #include <LibGfx/ImageFormats/PAMLoader.h>
  20. #include <LibGfx/ImageFormats/PBMLoader.h>
  21. #include <LibGfx/ImageFormats/PGMLoader.h>
  22. #include <LibGfx/ImageFormats/PNGLoader.h>
  23. #include <LibGfx/ImageFormats/PPMLoader.h>
  24. #include <LibGfx/ImageFormats/TGALoader.h>
  25. #include <LibGfx/ImageFormats/TIFFLoader.h>
  26. #include <LibGfx/ImageFormats/TinyVGLoader.h>
  27. #include <LibGfx/ImageFormats/WebPLoader.h>
  28. #include <LibTest/TestCase.h>
  29. #include <stdio.h>
  30. #include <string.h>
  31. #ifdef AK_OS_SERENITY
  32. # define TEST_INPUT(x) ("/usr/Tests/LibGfx/test-inputs/" x)
  33. #else
  34. # define TEST_INPUT(x) ("test-inputs/" x)
  35. #endif
  36. static ErrorOr<Gfx::ImageFrameDescriptor> expect_single_frame(Gfx::ImageDecoderPlugin& plugin_decoder)
  37. {
  38. EXPECT_EQ(plugin_decoder.frame_count(), 1u);
  39. EXPECT(!plugin_decoder.is_animated());
  40. EXPECT(!plugin_decoder.loop_count());
  41. auto frame = TRY(plugin_decoder.frame(0));
  42. EXPECT_EQ(frame.duration, 0);
  43. return frame;
  44. }
  45. static ErrorOr<Gfx::ImageFrameDescriptor> expect_single_frame_of_size(Gfx::ImageDecoderPlugin& plugin_decoder, Gfx::IntSize size)
  46. {
  47. EXPECT_EQ(plugin_decoder.size(), size);
  48. auto frame = TRY(expect_single_frame(plugin_decoder));
  49. EXPECT_EQ(frame.image->size(), size);
  50. return frame;
  51. }
  52. TEST_CASE(test_bmp)
  53. {
  54. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("bmp/rgba32-1.bmp"sv)));
  55. EXPECT(Gfx::BMPImageDecoderPlugin::sniff(file->bytes()));
  56. auto plugin_decoder = TRY_OR_FAIL(Gfx::BMPImageDecoderPlugin::create(file->bytes()));
  57. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  58. }
  59. TEST_CASE(test_bmp_top_down)
  60. {
  61. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("bmp/top-down.bmp"sv)));
  62. EXPECT(Gfx::BMPImageDecoderPlugin::sniff(file->bytes()));
  63. auto plugin_decoder = TRY_OR_FAIL(Gfx::BMPImageDecoderPlugin::create(file->bytes()));
  64. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  65. }
  66. TEST_CASE(test_bmp_1bpp)
  67. {
  68. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("bmp/bitmap.bmp"sv)));
  69. EXPECT(Gfx::BMPImageDecoderPlugin::sniff(file->bytes()));
  70. auto plugin_decoder = TRY_OR_FAIL(Gfx::BMPImageDecoderPlugin::create(file->bytes()));
  71. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 399, 400 }));
  72. EXPECT_EQ(frame.image->begin()[0], 0xff'ff'ff'ff);
  73. }
  74. TEST_CASE(test_ico_malformed_frame)
  75. {
  76. Array test_inputs = {
  77. TEST_INPUT("ico/oss-fuzz-testcase-62541.ico"sv),
  78. TEST_INPUT("ico/oss-fuzz-testcase-63177.ico"sv),
  79. TEST_INPUT("ico/oss-fuzz-testcase-63357.ico"sv)
  80. };
  81. for (auto test_input : test_inputs) {
  82. auto file = TRY_OR_FAIL(Core::MappedFile::map(test_input));
  83. auto plugin_decoder = TRY_OR_FAIL(Gfx::ICOImageDecoderPlugin::create(file->bytes()));
  84. auto frame_or_error = plugin_decoder->frame(0);
  85. EXPECT(frame_or_error.is_error());
  86. }
  87. }
  88. TEST_CASE(test_gif)
  89. {
  90. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("download-animation.gif"sv)));
  91. EXPECT(Gfx::GIFImageDecoderPlugin::sniff(file->bytes()));
  92. auto plugin_decoder = TRY_OR_FAIL(Gfx::GIFImageDecoderPlugin::create(file->bytes()));
  93. EXPECT(plugin_decoder->frame_count());
  94. EXPECT(plugin_decoder->is_animated());
  95. EXPECT(!plugin_decoder->loop_count());
  96. auto frame = TRY_OR_FAIL(plugin_decoder->frame(1));
  97. EXPECT(frame.duration == 400);
  98. }
  99. TEST_CASE(test_gif_without_global_color_table)
  100. {
  101. Array<u8, 35> gif_data {
  102. // Header (6 bytes): "GIF89a"
  103. 0x47,
  104. 0x49,
  105. 0x46,
  106. 0x38,
  107. 0x39,
  108. 0x61,
  109. // Logical Screen Descriptor (7 bytes)
  110. 0x01,
  111. 0x00, // Width (1)
  112. 0x01,
  113. 0x00, // Height (1)
  114. 0x00, // Packed fields (NOTE: the MSB here is the Global Color Table flag!)
  115. 0x00, // Background Color Index
  116. 0x00, // Pixel Aspect Ratio
  117. // Image Descriptor (10 bytes)
  118. 0x2C,
  119. 0x00,
  120. 0x00,
  121. 0x00,
  122. 0x00,
  123. 0x01,
  124. 0x00,
  125. 0x01,
  126. 0x00,
  127. 0x80,
  128. // Local Color Table (6 bytes: 2 colors, 3 bytes per color)
  129. 0x00,
  130. 0x00,
  131. 0x00, // Color 1: Black (RGB: 0, 0, 0)
  132. 0xff,
  133. 0x00,
  134. 0x00, // Color 2: Red (RGB: 255, 0, 0)
  135. // Image Data (8 bytes)
  136. 0x02, // LZW Minimum Code Size
  137. 0x02, // Data Sub-block size (2 bytes)
  138. 0x4C,
  139. 0x01, // Image Data
  140. 0x00, // Data Sub-block Terminator
  141. // Trailer (1 byte)
  142. 0x3B,
  143. };
  144. auto plugin_decoder = TRY_OR_FAIL(Gfx::GIFImageDecoderPlugin::create(gif_data));
  145. EXPECT_EQ(plugin_decoder->frame_count(), 1u);
  146. auto frame = TRY_OR_FAIL(plugin_decoder->frame(0));
  147. EXPECT(frame.image);
  148. EXPECT_EQ(frame.image->size(), Gfx::IntSize(1, 1));
  149. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::Red);
  150. }
  151. TEST_CASE(test_not_ico)
  152. {
  153. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("png/buggie.png"sv)));
  154. EXPECT(!Gfx::ICOImageDecoderPlugin::sniff(file->bytes()));
  155. EXPECT(Gfx::ICOImageDecoderPlugin::create(file->bytes()).is_error());
  156. }
  157. TEST_CASE(test_bmp_embedded_in_ico)
  158. {
  159. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("ico/serenity.ico"sv)));
  160. EXPECT(Gfx::ICOImageDecoderPlugin::sniff(file->bytes()));
  161. auto plugin_decoder = TRY_OR_FAIL(Gfx::ICOImageDecoderPlugin::create(file->bytes()));
  162. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 16, 16 }));
  163. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::Transparent);
  164. EXPECT_EQ(frame.image->get_pixel(7, 4), Gfx::Color(161, 0, 0));
  165. }
  166. TEST_CASE(test_malformed_maskless_ico)
  167. {
  168. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("ico/malformed_maskless.ico"sv)));
  169. EXPECT(Gfx::ICOImageDecoderPlugin::sniff(file->bytes()));
  170. auto plugin_decoder = TRY_OR_FAIL(Gfx::ICOImageDecoderPlugin::create(file->bytes()));
  171. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 16, 16 }));
  172. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::Transparent);
  173. EXPECT_EQ(frame.image->get_pixel(7, 4), Gfx::Color(161, 0, 0));
  174. }
  175. TEST_CASE(test_ilbm)
  176. {
  177. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("ilbm/gradient.iff"sv)));
  178. EXPECT(Gfx::ILBMImageDecoderPlugin::sniff(file->bytes()));
  179. auto plugin_decoder = TRY_OR_FAIL(Gfx::ILBMImageDecoderPlugin::create(file->bytes()));
  180. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 320, 200 }));
  181. EXPECT_EQ(frame.image->get_pixel(8, 0), Gfx::Color(0xee, 0xbb, 0, 255));
  182. }
  183. TEST_CASE(test_ilbm_uncompressed)
  184. {
  185. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("ilbm/gradient-uncompressed.iff"sv)));
  186. EXPECT(Gfx::ILBMImageDecoderPlugin::sniff(file->bytes()));
  187. auto plugin_decoder = TRY_OR_FAIL(Gfx::ILBMImageDecoderPlugin::create(file->bytes()));
  188. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 320, 200 }));
  189. EXPECT_EQ(frame.image->get_pixel(8, 0), Gfx::Color(0xee, 0xbb, 0, 255));
  190. }
  191. TEST_CASE(test_ilbm_ham6)
  192. {
  193. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("ilbm/ham6.iff"sv)));
  194. EXPECT(Gfx::ILBMImageDecoderPlugin::sniff(file->bytes()));
  195. auto plugin_decoder = TRY_OR_FAIL(Gfx::ILBMImageDecoderPlugin::create(file->bytes()));
  196. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 256, 256 }));
  197. EXPECT_EQ(frame.image->get_pixel(77, 107), Gfx::Color(0xf0, 0x40, 0x40, 0xff));
  198. }
  199. TEST_CASE(test_ilbm_dos)
  200. {
  201. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("ilbm/serenity.lbm"sv)));
  202. EXPECT(Gfx::ILBMImageDecoderPlugin::sniff(file->bytes()));
  203. auto plugin_decoder = TRY_OR_FAIL(Gfx::ILBMImageDecoderPlugin::create(file->bytes()));
  204. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 640, 480 }));
  205. EXPECT_EQ(frame.image->get_pixel(315, 134), Gfx::Color::NamedColor::Red);
  206. }
  207. TEST_CASE(test_24bit)
  208. {
  209. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("ilbm/serenity-24bit.iff"sv)));
  210. EXPECT(Gfx::ILBMImageDecoderPlugin::sniff(file->bytes()));
  211. auto plugin_decoder = TRY_OR_FAIL(Gfx::ILBMImageDecoderPlugin::create(file->bytes()));
  212. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 640, 640 }));
  213. EXPECT_EQ(frame.image->get_pixel(158, 270), Gfx::Color(0xee, 0x3d, 0x3c, 255));
  214. }
  215. TEST_CASE(test_brush_transparent_color)
  216. {
  217. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("ilbm/brush-transparent-color.iff"sv)));
  218. EXPECT(Gfx::ILBMImageDecoderPlugin::sniff(file->bytes()));
  219. auto plugin_decoder = TRY_OR_FAIL(Gfx::ILBMImageDecoderPlugin::create(file->bytes()));
  220. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 266, 309 }));
  221. EXPECT_EQ(frame.image->get_pixel(114, 103), Gfx::Color::NamedColor::Black);
  222. }
  223. TEST_CASE(test_small_24bit)
  224. {
  225. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("ilbm/small-24bit.iff"sv)));
  226. EXPECT(Gfx::ILBMImageDecoderPlugin::sniff(file->bytes()));
  227. auto plugin_decoder = TRY_OR_FAIL(Gfx::ILBMImageDecoderPlugin::create(file->bytes()));
  228. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 10, 10 }));
  229. EXPECT_EQ(frame.image->get_pixel(0, 4), Gfx::Color(1, 0, 1, 255));
  230. }
  231. TEST_CASE(test_stencil_mask)
  232. {
  233. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("ilbm/test-stencil.iff"sv)));
  234. EXPECT(Gfx::ILBMImageDecoderPlugin::sniff(file->bytes()));
  235. auto plugin_decoder = TRY_OR_FAIL(Gfx::ILBMImageDecoderPlugin::create(file->bytes()));
  236. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 320, 200 }));
  237. EXPECT_EQ(frame.image->get_pixel(0, 4), Gfx::Color(0, 0, 0, 255));
  238. }
  239. TEST_CASE(test_ilbm_malformed_header)
  240. {
  241. Array test_inputs = {
  242. TEST_INPUT("ilbm/truncated-bmhd-chunk.iff"sv)
  243. };
  244. for (auto test_input : test_inputs) {
  245. auto file = TRY_OR_FAIL(Core::MappedFile::map(test_input));
  246. auto plugin_decoder_or_error = Gfx::ILBMImageDecoderPlugin::create(file->bytes());
  247. EXPECT(plugin_decoder_or_error.is_error());
  248. }
  249. }
  250. TEST_CASE(test_ilbm_malformed_frame)
  251. {
  252. Array test_inputs = {
  253. TEST_INPUT("ilbm/incorrect-cmap-size.iff"sv),
  254. TEST_INPUT("ilbm/incorrect-uncompressed-size.iff"sv),
  255. TEST_INPUT("ilbm/missing-body-chunk.iff"sv)
  256. };
  257. for (auto test_input : test_inputs) {
  258. auto file = TRY_OR_FAIL(Core::MappedFile::map(test_input));
  259. auto plugin_decoder = TRY_OR_FAIL(Gfx::ILBMImageDecoderPlugin::create(file->bytes()));
  260. auto frame_or_error = plugin_decoder->frame(0);
  261. EXPECT(frame_or_error.is_error());
  262. }
  263. }
  264. TEST_CASE(test_jbig2_size)
  265. {
  266. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jbig2/bitmap.jbig2"sv)));
  267. EXPECT(Gfx::JBIG2ImageDecoderPlugin::sniff(file->bytes()));
  268. auto plugin_decoder = TRY_OR_FAIL(Gfx::JBIG2ImageDecoderPlugin::create(file->bytes()));
  269. EXPECT_EQ(plugin_decoder->size(), Gfx::IntSize(399, 400));
  270. }
  271. TEST_CASE(test_jbig2_black_47x23)
  272. {
  273. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jbig2/black_47x23.jbig2"sv)));
  274. EXPECT(Gfx::JBIG2ImageDecoderPlugin::sniff(file->bytes()));
  275. auto plugin_decoder = TRY_OR_FAIL(Gfx::JBIG2ImageDecoderPlugin::create(file->bytes()));
  276. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 47, 23 }));
  277. for (auto pixel : *frame.image)
  278. EXPECT_EQ(pixel, Gfx::Color(Gfx::Color::Black).value());
  279. }
  280. TEST_CASE(test_jbig2_white_47x23)
  281. {
  282. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jbig2/white_47x23.jbig2"sv)));
  283. EXPECT(Gfx::JBIG2ImageDecoderPlugin::sniff(file->bytes()));
  284. auto plugin_decoder = TRY_OR_FAIL(Gfx::JBIG2ImageDecoderPlugin::create(file->bytes()));
  285. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 47, 23 }));
  286. for (auto pixel : *frame.image)
  287. EXPECT_EQ(pixel, Gfx::Color(Gfx::Color::White).value());
  288. }
  289. TEST_CASE(test_jbig2_decode)
  290. {
  291. auto bmp_file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("bmp/bitmap.bmp"sv)));
  292. auto bmp_plugin_decoder = TRY_OR_FAIL(Gfx::BMPImageDecoderPlugin::create(bmp_file->bytes()));
  293. auto bmp_frame = TRY_OR_FAIL(expect_single_frame_of_size(*bmp_plugin_decoder, { 399, 400 }));
  294. Array test_inputs = {
  295. TEST_INPUT("jbig2/bitmap.jbig2"sv),
  296. TEST_INPUT("jbig2/bitmap-customat.jbig2"sv),
  297. TEST_INPUT("jbig2/bitmap-tpgdon.jbig2"sv),
  298. TEST_INPUT("jbig2/bitmap-customat-tpgdon.jbig2"sv),
  299. TEST_INPUT("jbig2/bitmap-template1.jbig2"sv),
  300. TEST_INPUT("jbig2/bitmap-template1-customat.jbig2"sv),
  301. TEST_INPUT("jbig2/bitmap-template1-tpgdon.jbig2"sv),
  302. TEST_INPUT("jbig2/bitmap-template1-customat-tpgdon.jbig2"sv),
  303. TEST_INPUT("jbig2/bitmap-template2.jbig2"sv),
  304. TEST_INPUT("jbig2/bitmap-template2-customat.jbig2"sv),
  305. TEST_INPUT("jbig2/bitmap-template2-tpgdon.jbig2"sv),
  306. TEST_INPUT("jbig2/bitmap-template2-customat-tpgdon.jbig2"sv),
  307. TEST_INPUT("jbig2/bitmap-template3.jbig2"sv),
  308. TEST_INPUT("jbig2/bitmap-template3-customat.jbig2"sv),
  309. TEST_INPUT("jbig2/bitmap-template3-tpgdon.jbig2"sv),
  310. TEST_INPUT("jbig2/bitmap-template3-customat-tpgdon.jbig2"sv),
  311. TEST_INPUT("jbig2/bitmap-symbol.jbig2"sv),
  312. TEST_INPUT("jbig2/bitmap-symbol-textrefine.jbig2"sv),
  313. TEST_INPUT("jbig2/bitmap-symbol-symbolrefine.jbig2"sv),
  314. TEST_INPUT("jbig2/bitmap-symbol-textbottomleft.jbig2"sv),
  315. TEST_INPUT("jbig2/bitmap-symbol-textbottomlefttranspose.jbig2"sv),
  316. TEST_INPUT("jbig2/bitmap-symbol-textbottomright.jbig2"sv),
  317. TEST_INPUT("jbig2/bitmap-symbol-textbottomrighttranspose.jbig2"sv),
  318. TEST_INPUT("jbig2/bitmap-symbol-texttopright.jbig2"sv),
  319. TEST_INPUT("jbig2/bitmap-symbol-texttoprighttranspose.jbig2"sv),
  320. TEST_INPUT("jbig2/bitmap-symbol-texttranspose.jbig2"sv),
  321. };
  322. for (auto test_input : test_inputs) {
  323. auto file = TRY_OR_FAIL(Core::MappedFile::map(test_input));
  324. EXPECT(Gfx::JBIG2ImageDecoderPlugin::sniff(file->bytes()));
  325. auto plugin_decoder = TRY_OR_FAIL(Gfx::JBIG2ImageDecoderPlugin::create(file->bytes()));
  326. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 399, 400 }));
  327. for (int y = 0; y < frame.image->height(); ++y)
  328. for (int x = 0; x < frame.image->width(); ++x)
  329. EXPECT_EQ(frame.image->get_pixel(x, y), bmp_frame.image->get_pixel(x, y));
  330. }
  331. }
  332. TEST_CASE(test_jbig2_arithmetic_decoder)
  333. {
  334. // https://www.itu.int/rec/T-REC-T.88-201808-I
  335. // H.2 Test sequence for arithmetic coder
  336. // clang-format off
  337. constexpr auto input = to_array<u8>({
  338. 0x84, 0xC7, 0x3B, 0xFC, 0xE1, 0xA1, 0x43, 0x04,
  339. 0x02, 0x20, 0x00, 0x00, 0x41, 0x0D, 0xBB, 0x86,
  340. 0xF4, 0x31, 0x7F, 0xFF, 0x88, 0xFF, 0x37, 0x47,
  341. 0x1A, 0xDB, 0x6A, 0xDF, 0xFF, 0xAC
  342. });
  343. constexpr auto output = to_array<u8>({
  344. 0x00, 0x02, 0x00, 0x51, 0x00, 0x00, 0x00, 0xC0,
  345. 0x03, 0x52, 0x87, 0x2A, 0xAA, 0xAA, 0xAA, 0xAA,
  346. 0x82, 0xC0, 0x20, 0x00, 0xFC, 0xD7, 0x9E, 0xF6,
  347. 0xBF, 0x7F, 0xED, 0x90, 0x4F, 0x46, 0xA3, 0xBF
  348. });
  349. // clang-format on
  350. // "For this entire test, a single value of CX is used. I(CX) is initially 0 and MPS(CX) is initially 0."
  351. Gfx::JBIG2::ArithmeticDecoder::Context context { 0, 0 };
  352. auto decoder = MUST(Gfx::JBIG2::ArithmeticDecoder::initialize(input));
  353. for (auto expected : output) {
  354. u8 actual = 0;
  355. for (size_t i = 0; i < 8; ++i) {
  356. actual <<= 1;
  357. actual |= static_cast<u8>(decoder.get_next_bit(context));
  358. }
  359. EXPECT_EQ(actual, expected);
  360. }
  361. }
  362. TEST_CASE(test_jpeg_sof0_one_scan)
  363. {
  364. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/rgb24.jpg"sv)));
  365. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  366. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  367. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  368. }
  369. TEST_CASE(test_jpeg_sof0_several_scans)
  370. {
  371. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/several_scans.jpg"sv)));
  372. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  373. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  374. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 592, 800 }));
  375. }
  376. TEST_CASE(test_odd_mcu_restart_interval)
  377. {
  378. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/odd-restart.jpg"sv)));
  379. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  380. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  381. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 102, 77 }));
  382. }
  383. TEST_CASE(test_jpeg_rgb_components)
  384. {
  385. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/rgb_components.jpg"sv)));
  386. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  387. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  388. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 592, 800 }));
  389. }
  390. TEST_CASE(test_jpeg_ycck)
  391. {
  392. Array test_inputs = {
  393. TEST_INPUT("jpg/ycck-1111.jpg"sv),
  394. TEST_INPUT("jpg/ycck-2111.jpg"sv),
  395. TEST_INPUT("jpg/ycck-2112.jpg"sv),
  396. };
  397. for (auto test_input : test_inputs) {
  398. auto file = TRY_OR_FAIL(Core::MappedFile::map(test_input));
  399. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  400. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  401. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 592, 800 }));
  402. // Compare difference between pixels so we don't depend on exact CMYK->RGB conversion behavior.
  403. // These two pixels are currently off by one in R.
  404. // FIXME: For 2111, they're off by way more.
  405. EXPECT(frame.image->get_pixel(6, 319).distance_squared_to(frame.image->get_pixel(6, 320)) < 1.0f / 255.0f);
  406. }
  407. }
  408. TEST_CASE(test_jpeg_sof2_spectral_selection)
  409. {
  410. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/spectral_selection.jpg"sv)));
  411. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  412. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  413. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 592, 800 }));
  414. }
  415. TEST_CASE(test_jpeg_sof0_several_scans_odd_number_mcu)
  416. {
  417. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/several_scans_odd_number_mcu.jpg"sv)));
  418. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  419. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  420. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 600, 600 }));
  421. }
  422. TEST_CASE(test_jpeg_sof2_successive_aproximation)
  423. {
  424. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/successive_approximation.jpg"sv)));
  425. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  426. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  427. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 600, 800 }));
  428. }
  429. TEST_CASE(test_jpeg_sof1_12bits)
  430. {
  431. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/12-bit.jpg"sv)));
  432. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  433. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  434. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 320, 240 }));
  435. }
  436. TEST_CASE(test_jpeg_sof2_12bits)
  437. {
  438. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/12-bit-progressive.jpg"sv)));
  439. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  440. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  441. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 320, 240 }));
  442. }
  443. TEST_CASE(test_jpeg_empty_icc)
  444. {
  445. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/gradient_empty_icc.jpg"sv)));
  446. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  447. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  448. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 80, 80 }));
  449. }
  450. TEST_CASE(test_jpeg_grayscale_with_app14)
  451. {
  452. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/grayscale_app14.jpg"sv)));
  453. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  454. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  455. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 80, 80 }));
  456. }
  457. TEST_CASE(test_jpeg_grayscale_with_weird_mcu_and_reset_marker)
  458. {
  459. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpg/grayscale_mcu.jpg"sv)));
  460. EXPECT(Gfx::JPEGImageDecoderPlugin::sniff(file->bytes()));
  461. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  462. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 320, 240 }));
  463. }
  464. TEST_CASE(test_jpeg_malformed_header)
  465. {
  466. Array test_inputs = {
  467. TEST_INPUT("jpg/oss-fuzz-testcase-59785.jpg"sv)
  468. };
  469. for (auto test_input : test_inputs) {
  470. auto file = TRY_OR_FAIL(Core::MappedFile::map(test_input));
  471. auto plugin_decoder_or_error = Gfx::JPEGImageDecoderPlugin::create(file->bytes());
  472. EXPECT(plugin_decoder_or_error.is_error());
  473. }
  474. }
  475. TEST_CASE(test_jpeg_malformed_frame)
  476. {
  477. Array test_inputs = {
  478. TEST_INPUT("jpg/oss-fuzz-testcase-62584.jpg"sv),
  479. TEST_INPUT("jpg/oss-fuzz-testcase-63815.jpg"sv)
  480. };
  481. for (auto test_input : test_inputs) {
  482. auto file = TRY_OR_FAIL(Core::MappedFile::map(test_input));
  483. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGImageDecoderPlugin::create(file->bytes()));
  484. auto frame_or_error = plugin_decoder->frame(0);
  485. EXPECT(frame_or_error.is_error());
  486. }
  487. }
  488. TEST_CASE(test_jpeg2000_simple)
  489. {
  490. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpeg2000/simple.jp2"sv)));
  491. EXPECT(Gfx::JPEG2000ImageDecoderPlugin::sniff(file->bytes()));
  492. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEG2000ImageDecoderPlugin::create(file->bytes()));
  493. EXPECT_EQ(plugin_decoder->size(), Gfx::IntSize(119, 101));
  494. auto icc_bytes = MUST(plugin_decoder->icc_data());
  495. EXPECT(icc_bytes.has_value());
  496. EXPECT_EQ(icc_bytes->size(), 3144u);
  497. }
  498. TEST_CASE(test_jpeg2000_gray)
  499. {
  500. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jpeg2000/buggie-gray.jpf"sv)));
  501. EXPECT(Gfx::JPEG2000ImageDecoderPlugin::sniff(file->bytes()));
  502. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEG2000ImageDecoderPlugin::create(file->bytes()));
  503. EXPECT_EQ(plugin_decoder->size(), Gfx::IntSize(64, 138));
  504. // The file contains both a simple and a real profile. Make sure we get the bigger one.
  505. auto icc_bytes = MUST(plugin_decoder->icc_data());
  506. EXPECT(icc_bytes.has_value());
  507. EXPECT_EQ(icc_bytes->size(), 912u);
  508. }
  509. TEST_CASE(test_pam_rgb)
  510. {
  511. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("pnm/2x1.pam"sv)));
  512. EXPECT(Gfx::PAMImageDecoderPlugin::sniff(file->bytes()));
  513. auto plugin_decoder = TRY_OR_FAIL(Gfx::PAMImageDecoderPlugin::create(file->bytes()));
  514. auto frame = TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  515. EXPECT_EQ(frame.image->size(), Gfx::IntSize(2, 1));
  516. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color('0', 'z', '0'));
  517. EXPECT_EQ(frame.image->get_pixel(1, 0), Gfx::Color('0', '0', 'z'));
  518. }
  519. TEST_CASE(test_pam_cmyk)
  520. {
  521. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("pnm/2x1-cmyk.pam"sv)));
  522. EXPECT(Gfx::PAMImageDecoderPlugin::sniff(file->bytes()));
  523. auto plugin_decoder = TRY_OR_FAIL(Gfx::PAMImageDecoderPlugin::create(file->bytes()));
  524. EXPECT_EQ(plugin_decoder->natural_frame_format(), Gfx::NaturalFrameFormat::CMYK);
  525. auto cmyk_frame = TRY_OR_FAIL(plugin_decoder->cmyk_frame());
  526. EXPECT_EQ(cmyk_frame->size(), Gfx::IntSize(2, 1));
  527. EXPECT_EQ(cmyk_frame->begin()[0], (Gfx::CMYK { '0', 'z', '0', 'y' }));
  528. EXPECT_EQ(cmyk_frame->begin()[1], (Gfx::CMYK { '0', '0', 'z', 'y' }));
  529. auto frame = TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  530. EXPECT_EQ(frame.image->size(), Gfx::IntSize(2, 1));
  531. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color('l', 'E', 'l'));
  532. EXPECT_EQ(frame.image->get_pixel(1, 0), Gfx::Color('l', 'l', 'E'));
  533. }
  534. TEST_CASE(test_pbm)
  535. {
  536. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("pnm/buggie-raw.pbm"sv)));
  537. EXPECT(Gfx::PBMImageDecoderPlugin::sniff(file->bytes()));
  538. auto plugin_decoder = TRY_OR_FAIL(Gfx::PBMImageDecoderPlugin::create(file->bytes()));
  539. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  540. }
  541. TEST_CASE(test_pgm)
  542. {
  543. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("pnm/buggie-raw.pgm"sv)));
  544. EXPECT(Gfx::PGMImageDecoderPlugin::sniff(file->bytes()));
  545. auto plugin_decoder = TRY_OR_FAIL(Gfx::PGMImageDecoderPlugin::create(file->bytes()));
  546. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  547. }
  548. TEST_CASE(test_png)
  549. {
  550. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("png/buggie.png"sv)));
  551. EXPECT(Gfx::PNGImageDecoderPlugin::sniff(file->bytes()));
  552. auto plugin_decoder = TRY_OR_FAIL(Gfx::PNGImageDecoderPlugin::create(file->bytes()));
  553. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  554. }
  555. TEST_CASE(test_png_malformed_frame)
  556. {
  557. Array test_inputs = {
  558. TEST_INPUT("png/oss-fuzz-testcase-62371.png"sv),
  559. TEST_INPUT("png/oss-fuzz-testcase-63052.png"sv)
  560. };
  561. for (auto test_input : test_inputs) {
  562. auto file = TRY_OR_FAIL(Core::MappedFile::map(test_input));
  563. auto plugin_decoder = TRY_OR_FAIL(Gfx::PNGImageDecoderPlugin::create(file->bytes()));
  564. auto frame_or_error = plugin_decoder->frame(0);
  565. EXPECT(frame_or_error.is_error());
  566. }
  567. }
  568. TEST_CASE(test_ppm)
  569. {
  570. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("pnm/buggie-raw.ppm"sv)));
  571. EXPECT(Gfx::PPMImageDecoderPlugin::sniff(file->bytes()));
  572. auto plugin_decoder = TRY_OR_FAIL(Gfx::PPMImageDecoderPlugin::create(file->bytes()));
  573. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  574. }
  575. TEST_CASE(test_targa_bottom_left)
  576. {
  577. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tga/buggie-bottom-left-uncompressed.tga"sv)));
  578. EXPECT(Gfx::TGAImageDecoderPlugin::validate_before_create(file->bytes()));
  579. auto plugin_decoder = TRY_OR_FAIL(Gfx::TGAImageDecoderPlugin::create(file->bytes()));
  580. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  581. }
  582. TEST_CASE(test_targa_top_left)
  583. {
  584. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tga/buggie-top-left-uncompressed.tga"sv)));
  585. EXPECT(Gfx::TGAImageDecoderPlugin::validate_before_create(file->bytes()));
  586. auto plugin_decoder = TRY_OR_FAIL(Gfx::TGAImageDecoderPlugin::create(file->bytes()));
  587. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  588. }
  589. TEST_CASE(test_targa_bottom_left_compressed)
  590. {
  591. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tga/buggie-bottom-left-compressed.tga"sv)));
  592. EXPECT(Gfx::TGAImageDecoderPlugin::validate_before_create(file->bytes()));
  593. auto plugin_decoder = TRY_OR_FAIL(Gfx::TGAImageDecoderPlugin::create(file->bytes()));
  594. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  595. }
  596. TEST_CASE(test_targa_top_left_compressed)
  597. {
  598. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tga/buggie-top-left-compressed.tga"sv)));
  599. EXPECT(Gfx::TGAImageDecoderPlugin::validate_before_create(file->bytes()));
  600. auto plugin_decoder = TRY_OR_FAIL(Gfx::TGAImageDecoderPlugin::create(file->bytes()));
  601. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  602. }
  603. TEST_CASE(test_tiff_uncompressed)
  604. {
  605. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/uncompressed.tiff"sv)));
  606. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  607. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  608. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  609. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  610. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Red);
  611. }
  612. TEST_CASE(test_tiff_ccitt_rle)
  613. {
  614. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/ccitt_rle.tiff"sv)));
  615. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  616. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  617. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  618. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  619. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Black);
  620. }
  621. TEST_CASE(test_tiff_ccitt3)
  622. {
  623. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/ccitt3.tiff"sv)));
  624. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  625. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  626. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  627. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  628. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Black);
  629. }
  630. TEST_CASE(test_tiff_ccitt3_fill)
  631. {
  632. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/ccitt3_1d_fill.tiff"sv)));
  633. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  634. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  635. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 6, 4 }));
  636. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  637. EXPECT_EQ(frame.image->get_pixel(3, 0), Gfx::Color::NamedColor::Black);
  638. EXPECT_EQ(frame.image->get_pixel(2, 2), Gfx::Color::NamedColor::White);
  639. EXPECT_EQ(frame.image->get_pixel(5, 3), Gfx::Color::NamedColor::White);
  640. }
  641. TEST_CASE(test_tiff_ccitt3_2d)
  642. {
  643. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/ccitt3_2d.tiff"sv)));
  644. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  645. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  646. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  647. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  648. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Black);
  649. }
  650. TEST_CASE(test_tiff_ccitt3_2d_fill)
  651. {
  652. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/ccitt3_2d_fill.tiff"sv)));
  653. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  654. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  655. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  656. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  657. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Black);
  658. }
  659. TEST_CASE(test_tiff_ccitt4)
  660. {
  661. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/ccitt4.tiff"sv)));
  662. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  663. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  664. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  665. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  666. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Black);
  667. }
  668. TEST_CASE(test_tiff_lzw)
  669. {
  670. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/lzw.tiff"sv)));
  671. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  672. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  673. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  674. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  675. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Red);
  676. }
  677. TEST_CASE(test_tiff_deflate)
  678. {
  679. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/deflate.tiff"sv)));
  680. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  681. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  682. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  683. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  684. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Red);
  685. }
  686. TEST_CASE(test_tiff_krita)
  687. {
  688. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/krita.tif"sv)));
  689. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  690. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  691. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  692. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  693. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Red);
  694. }
  695. TEST_CASE(test_tiff_orientation)
  696. {
  697. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/orientation.tiff"sv)));
  698. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  699. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  700. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 300, 400 }));
  701. // Orientation is Rotate90Clockwise
  702. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  703. EXPECT_EQ(frame.image->get_pixel(300 - 75, 60), Gfx::Color::NamedColor::Red);
  704. }
  705. TEST_CASE(test_tiff_packed_bits)
  706. {
  707. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/packed_bits.tiff"sv)));
  708. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  709. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  710. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  711. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  712. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Red);
  713. }
  714. TEST_CASE(test_tiff_grayscale)
  715. {
  716. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/grayscale.tiff"sv)));
  717. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  718. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  719. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  720. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  721. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color(130, 130, 130));
  722. }
  723. TEST_CASE(test_tiff_grayscale_alpha)
  724. {
  725. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/grayscale_alpha.tiff"sv)));
  726. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  727. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  728. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  729. EXPECT_EQ(frame.image->get_pixel(0, 0).alpha(), 0);
  730. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color(130, 130, 130));
  731. }
  732. TEST_CASE(test_tiff_rgb_alpha)
  733. {
  734. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/rgb_alpha.tiff"sv)));
  735. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  736. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  737. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  738. EXPECT_EQ(frame.image->get_pixel(0, 0).alpha(), 0);
  739. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Red);
  740. }
  741. TEST_CASE(test_tiff_palette_alpha)
  742. {
  743. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/rgb_palette_alpha.tiff"sv)));
  744. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  745. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  746. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  747. EXPECT_EQ(frame.image->get_pixel(0, 0).alpha(), 0);
  748. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Red);
  749. }
  750. TEST_CASE(test_tiff_alpha_predictor)
  751. {
  752. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/alpha_predictor.tiff"sv)));
  753. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  754. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  755. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  756. EXPECT_EQ(frame.image->get_pixel(0, 0).alpha(), 255);
  757. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Red);
  758. }
  759. TEST_CASE(test_tiff_16_bits)
  760. {
  761. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/16_bits.tiff"sv)));
  762. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  763. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  764. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  765. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  766. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Red);
  767. }
  768. TEST_CASE(test_tiff_cmyk)
  769. {
  770. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/cmyk.tiff"sv)));
  771. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  772. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  773. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  774. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  775. // I stripped the ICC profile from the image, so we can't test for equality with Red here.
  776. EXPECT_NE(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::White);
  777. }
  778. TEST_CASE(test_tiff_tiled)
  779. {
  780. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/tiled.tiff"sv)));
  781. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  782. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  783. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 300 }));
  784. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::White);
  785. EXPECT_EQ(frame.image->get_pixel(60, 75), Gfx::Color::NamedColor::Red);
  786. }
  787. TEST_CASE(test_tiff_invalid_tag)
  788. {
  789. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tiff/invalid_tag.tiff"sv)));
  790. EXPECT(Gfx::TIFFImageDecoderPlugin::sniff(file->bytes()));
  791. auto plugin_decoder = TRY_OR_FAIL(Gfx::TIFFImageDecoderPlugin::create(file->bytes()));
  792. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 10, 10 }));
  793. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color::NamedColor::Black);
  794. EXPECT_EQ(frame.image->get_pixel(0, 9), Gfx::Color::NamedColor::White);
  795. }
  796. TEST_CASE(test_webp_simple_lossy)
  797. {
  798. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/simple-vp8.webp"sv)));
  799. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  800. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  801. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 240, 240 }));
  802. // While VP8 YUV contents are defined bit-exact, the YUV->RGB conversion isn't.
  803. // So pixels changing by 1 or so below is fine if you change code.
  804. EXPECT_EQ(frame.image->get_pixel(120, 232), Gfx::Color(0xf2, 0xef, 0xf0, 255));
  805. EXPECT_EQ(frame.image->get_pixel(198, 202), Gfx::Color(0x7b, 0xaa, 0xd5, 255));
  806. }
  807. TEST_CASE(test_webp_simple_lossless)
  808. {
  809. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/simple-vp8l.webp"sv)));
  810. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  811. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  812. // Ironically, simple-vp8l.webp is a much more complex file than extended-lossless.webp tested below.
  813. // extended-lossless.webp tests the decoding basics.
  814. // This here tests the predictor, color, and subtract green transforms,
  815. // as well as meta prefix images, one-element canonical code handling,
  816. // and handling of canonical codes with more than 288 elements.
  817. // This image uses all 13 predictor modes of the predictor transform.
  818. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 386, 395 }));
  819. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color(0, 0, 0, 0));
  820. // This pixel tests all predictor modes except 5, 7, 8, 9, and 13.
  821. EXPECT_EQ(frame.image->get_pixel(289, 332), Gfx::Color(0xf2, 0xee, 0xd3, 255));
  822. }
  823. TEST_CASE(test_webp_simple_lossless_alpha_used_false)
  824. {
  825. // This file is identical to simple-vp8l.webp, but the `is_alpha_used` used bit is false.
  826. // The file still contains alpha data. This tests that the decoder replaces the stored alpha data with 0xff if `is_alpha_used` is false.
  827. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/simple-vp8l-alpha-used-false.webp"sv)));
  828. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  829. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  830. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 386, 395 }));
  831. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color(0, 0, 0, 0xff));
  832. }
  833. TEST_CASE(test_webp_extended_lossy)
  834. {
  835. // This extended lossy image has an ALPH chunk for (losslessly compressed) alpha data.
  836. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/extended-lossy.webp"sv)));
  837. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  838. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  839. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 417, 223 }));
  840. // While VP8 YUV contents are defined bit-exact, the YUV->RGB conversion isn't.
  841. // So pixels changing by 1 or so below is fine if you change code.
  842. EXPECT_EQ(frame.image->get_pixel(89, 72), Gfx::Color(255, 1, 0, 255));
  843. EXPECT_EQ(frame.image->get_pixel(174, 69), Gfx::Color(0, 255, 0, 255));
  844. EXPECT_EQ(frame.image->get_pixel(245, 84), Gfx::Color(0, 0, 255, 255));
  845. EXPECT_EQ(frame.image->get_pixel(352, 125), Gfx::Color(0, 0, 0, 128));
  846. EXPECT_EQ(frame.image->get_pixel(355, 106), Gfx::Color(0, 0, 0, 0));
  847. // Check same basic pixels as in test_webp_extended_lossless too.
  848. // (The top-left pixel in the lossy version is fully transparent white, compared to fully transparent black in the lossless version).
  849. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color(255, 255, 255, 0));
  850. EXPECT_EQ(frame.image->get_pixel(43, 75), Gfx::Color(255, 0, 2, 255));
  851. EXPECT_EQ(frame.image->get_pixel(141, 75), Gfx::Color(0, 255, 3, 255));
  852. EXPECT_EQ(frame.image->get_pixel(235, 75), Gfx::Color(0, 0, 255, 255));
  853. EXPECT_EQ(frame.image->get_pixel(341, 75), Gfx::Color(0, 0, 0, 128));
  854. }
  855. TEST_CASE(test_webp_extended_lossy_alpha_horizontal_filter)
  856. {
  857. // Also lossy rgb + lossless alpha, but with a horizontal alpha filtering method.
  858. // The image should look like smolkling.webp, but with a horizontal alpha gradient.
  859. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/smolkling-horizontal-alpha.webp"sv)));
  860. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  861. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  862. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 264, 264 }));
  863. // While VP8 YUV contents are defined bit-exact, the YUV->RGB conversion isn't.
  864. // So pixels changing by 1 or so below is fine if you change code.
  865. // The important component in this test is alpha, and that shouldn't change even by 1 as it's losslessly compressed and doesn't use YUV.
  866. EXPECT_EQ(frame.image->get_pixel(131, 131), Gfx::Color(0x8f, 0x51, 0x2f, 0x4b));
  867. }
  868. TEST_CASE(test_webp_extended_lossy_alpha_vertical_filter)
  869. {
  870. // Also lossy rgb + lossless alpha, but with a vertical alpha filtering method.
  871. // The image should look like smolkling.webp, but with a vertical alpha gradient, and with a fully transparent first column.
  872. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/smolkling-vertical-alpha.webp"sv)));
  873. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  874. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  875. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 264, 264 }));
  876. // While VP8 YUV contents are defined bit-exact, the YUV->RGB conversion isn't.
  877. // So pixels changing by 1 or so below is fine if you change code.
  878. // The important component in this test is alpha, and that shouldn't change even by 1 as it's losslessly compressed and doesn't use YUV.
  879. EXPECT_EQ(frame.image->get_pixel(131, 131), Gfx::Color(0x94, 0x50, 0x32, 0x4c));
  880. }
  881. TEST_CASE(test_webp_extended_lossy_alpha_gradient_filter)
  882. {
  883. // Also lossy rgb + lossless alpha, but with a gradient alpha filtering method.
  884. // The image should look like smolkling.webp, but with a few transparent pixels in the shape of a C on it. Most of the image should not be transparent.
  885. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/smolkling-gradient-alpha.webp"sv)));
  886. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  887. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  888. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 264, 264 }));
  889. // While VP8 YUV contents are defined bit-exact, the YUV->RGB conversion isn't.
  890. // So pixels changing by 1 or so below is fine if you change code.
  891. // The important component in this test is alpha, and that shouldn't change even by 1 as it's losslessly compressed and doesn't use YUV.
  892. // In particular, the center of the image should be fully opaque, not fully transparent.
  893. EXPECT_EQ(frame.image->get_pixel(131, 131), Gfx::Color(0x8c, 0x47, 0x2e, 255));
  894. }
  895. TEST_CASE(test_webp_extended_lossy_uncompressed_alpha)
  896. {
  897. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/extended-lossy-uncompressed-alpha.webp"sv)));
  898. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  899. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  900. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 417, 223 }));
  901. // While VP8 YUV contents are defined bit-exact, the YUV->RGB conversion isn't.
  902. // So pixels changing by 1 or so below is fine if you change code.
  903. EXPECT_EQ(frame.image->get_pixel(89, 72), Gfx::Color(255, 0, 4, 255));
  904. EXPECT_EQ(frame.image->get_pixel(174, 69), Gfx::Color(4, 255, 0, 255));
  905. EXPECT_EQ(frame.image->get_pixel(245, 84), Gfx::Color(0, 0, 255, 255));
  906. EXPECT_EQ(frame.image->get_pixel(352, 125), Gfx::Color(0, 0, 0, 128));
  907. EXPECT_EQ(frame.image->get_pixel(355, 106), Gfx::Color(0, 0, 0, 0));
  908. }
  909. TEST_CASE(test_webp_extended_lossy_negative_quantization_offset)
  910. {
  911. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/smolkling.webp"sv)));
  912. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  913. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  914. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 264, 264 }));
  915. // While VP8 YUV contents are defined bit-exact, the YUV->RGB conversion isn't.
  916. // So pixels changing by 1 or so below is fine if you change code.
  917. EXPECT_EQ(frame.image->get_pixel(16, 16), Gfx::Color(0x3c, 0x24, 0x1a, 255));
  918. }
  919. TEST_CASE(test_webp_lossy_4)
  920. {
  921. // This is https://commons.wikimedia.org/wiki/File:Fr%C3%BChling_bl%C3%BChender_Kirschenbaum.jpg,
  922. // under the Creative Commons Attribution-Share Alike 3.0 Unported license. The image was re-encoded
  923. // as webp at https://developers.google.com/speed/webp/gallery1 and the webp version is from there.
  924. // No other changes have been made.
  925. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/4.webp"sv)));
  926. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  927. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  928. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 1024, 772 }));
  929. // This image tests macroblocks that have `skip_coefficients` set to true, and it test a boolean entropy decoder edge case.
  930. EXPECT_EQ(frame.image->get_pixel(780, 570), Gfx::Color(0x72, 0xc8, 0xf6, 255));
  931. }
  932. TEST_CASE(test_webp_lossy_4_with_partitions)
  933. {
  934. // Same input file as in the previous test, but re-encoded to use 8 secondary partitions.
  935. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/4-with-8-partitions.webp"sv)));
  936. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  937. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  938. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 1024, 772 }));
  939. EXPECT_EQ(frame.image->get_pixel(780, 570), Gfx::Color(0x73, 0xc9, 0xf9, 255));
  940. }
  941. TEST_CASE(test_webp_extended_lossless)
  942. {
  943. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/extended-lossless.webp"sv)));
  944. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  945. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  946. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 417, 223 }));
  947. // Check some basic pixels.
  948. EXPECT_EQ(frame.image->get_pixel(0, 0), Gfx::Color(0, 0, 0, 0));
  949. EXPECT_EQ(frame.image->get_pixel(43, 75), Gfx::Color(255, 0, 0, 255));
  950. EXPECT_EQ(frame.image->get_pixel(141, 75), Gfx::Color(0, 255, 0, 255));
  951. EXPECT_EQ(frame.image->get_pixel(235, 75), Gfx::Color(0, 0, 255, 255));
  952. EXPECT_EQ(frame.image->get_pixel(341, 75), Gfx::Color(0, 0, 0, 128));
  953. // Check pixels using the color cache.
  954. EXPECT_EQ(frame.image->get_pixel(94, 73), Gfx::Color(255, 0, 0, 255));
  955. EXPECT_EQ(frame.image->get_pixel(176, 115), Gfx::Color(0, 255, 0, 255));
  956. EXPECT_EQ(frame.image->get_pixel(290, 89), Gfx::Color(0, 0, 255, 255));
  957. EXPECT_EQ(frame.image->get_pixel(359, 73), Gfx::Color(0, 0, 0, 128));
  958. }
  959. TEST_CASE(test_webp_simple_lossless_color_index_transform)
  960. {
  961. // In addition to testing the index transform, this file also tests handling of explicity setting max_symbol.
  962. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/Qpalette.webp"sv)));
  963. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  964. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  965. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 256, 256 }));
  966. EXPECT_EQ(frame.image->get_pixel(100, 100), Gfx::Color(0x73, 0x37, 0x23, 0xff));
  967. }
  968. TEST_CASE(test_webp_simple_lossless_color_index_transform_pixel_bundling)
  969. {
  970. struct TestCase {
  971. StringView file_name;
  972. Gfx::Color line_color;
  973. Gfx::Color background_color;
  974. };
  975. // The number after the dash is the number of colors in each file's color index bitmap.
  976. // catdog-alert-2 tests the 1-bit-per-pixel case,
  977. // catdog-alert-3 tests the 2-bit-per-pixel case,
  978. // catdog-alert-8 and catdog-alert-13 both test the 4-bits-per-pixel case.
  979. // catdog-alert-13-alpha-used-false is like catdog-alert-13, but with is_alpha_used set to false in the header
  980. // (which has the effect of ignoring the alpha information in the palette and instead always setting alpha to 0xff).
  981. TestCase test_cases[] = {
  982. { "webp/catdog-alert-2.webp"sv, Gfx::Color(0x35, 0x12, 0x0a, 0xff), Gfx::Color(0xf3, 0xe6, 0xd8, 0xff) },
  983. { "webp/catdog-alert-3.webp"sv, Gfx::Color(0x35, 0x12, 0x0a, 0xff), Gfx::Color(0, 0, 0, 0) },
  984. { "webp/catdog-alert-8.webp"sv, Gfx::Color(0, 0, 0, 255), Gfx::Color(0, 0, 0, 0) },
  985. { "webp/catdog-alert-13.webp"sv, Gfx::Color(0, 0, 0, 255), Gfx::Color(0, 0, 0, 0) },
  986. { "webp/catdog-alert-13-alpha-used-false.webp"sv, Gfx::Color(0, 0, 0, 255), Gfx::Color(0, 0, 0, 255) },
  987. };
  988. for (auto test_case : test_cases) {
  989. auto file = TRY_OR_FAIL(Core::MappedFile::map(MUST(String::formatted("{}{}", TEST_INPUT(""), test_case.file_name))));
  990. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  991. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  992. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 32, 32 }));
  993. EXPECT_EQ(frame.image->get_pixel(4, 0), test_case.background_color);
  994. EXPECT_EQ(frame.image->get_pixel(5, 0), test_case.line_color);
  995. EXPECT_EQ(frame.image->get_pixel(9, 5), test_case.background_color);
  996. EXPECT_EQ(frame.image->get_pixel(10, 5), test_case.line_color);
  997. EXPECT_EQ(frame.image->get_pixel(11, 5), test_case.background_color);
  998. }
  999. }
  1000. TEST_CASE(test_webp_simple_lossless_color_index_transform_pixel_bundling_odd_width)
  1001. {
  1002. StringView file_names[] = {
  1003. "webp/width11-height11-colors2.webp"sv,
  1004. "webp/width11-height11-colors3.webp"sv,
  1005. "webp/width11-height11-colors15.webp"sv,
  1006. };
  1007. for (auto file_name : file_names) {
  1008. auto file = TRY_OR_FAIL(Core::MappedFile::map(MUST(String::formatted("{}{}", TEST_INPUT(""), file_name))));
  1009. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  1010. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 11, 11 }));
  1011. }
  1012. }
  1013. TEST_CASE(test_webp_extended_lossless_animated)
  1014. {
  1015. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("webp/extended-lossless-animated.webp"sv)));
  1016. EXPECT(Gfx::WebPImageDecoderPlugin::sniff(file->bytes()));
  1017. auto plugin_decoder = TRY_OR_FAIL(Gfx::WebPImageDecoderPlugin::create(file->bytes()));
  1018. EXPECT_EQ(plugin_decoder->frame_count(), 8u);
  1019. EXPECT(plugin_decoder->is_animated());
  1020. EXPECT_EQ(plugin_decoder->loop_count(), 42u);
  1021. EXPECT_EQ(plugin_decoder->size(), Gfx::IntSize(990, 1050));
  1022. for (size_t frame_index = 0; frame_index < plugin_decoder->frame_count(); ++frame_index) {
  1023. auto frame = TRY_OR_FAIL(plugin_decoder->frame(frame_index));
  1024. EXPECT_EQ(frame.image->size(), Gfx::IntSize(990, 1050));
  1025. // This pixel happens to be the same color in all frames.
  1026. EXPECT_EQ(frame.image->get_pixel(500, 700), Gfx::Color::Yellow);
  1027. // This one isn't the same in all frames.
  1028. EXPECT_EQ(frame.image->get_pixel(500, 0), (frame_index == 2 || frame_index == 6) ? Gfx::Color::Black : Gfx::Color(255, 255, 255, 0));
  1029. }
  1030. }
  1031. TEST_CASE(test_tvg)
  1032. {
  1033. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tvg/yak.tvg"sv)));
  1034. EXPECT(Gfx::TinyVGImageDecoderPlugin::sniff(file->bytes()));
  1035. auto plugin_decoder = TRY_OR_FAIL(Gfx::TinyVGImageDecoderPlugin::create(file->bytes()));
  1036. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 1024, 1024 }));
  1037. }
  1038. TEST_CASE(test_everything_tvg)
  1039. {
  1040. Array file_names {
  1041. TEST_INPUT("tvg/everything.tvg"sv),
  1042. TEST_INPUT("tvg/everything-32.tvg"sv)
  1043. };
  1044. for (auto file_name : file_names) {
  1045. auto file = TRY_OR_FAIL(Core::MappedFile::map(file_name));
  1046. EXPECT(Gfx::TinyVGImageDecoderPlugin::sniff(file->bytes()));
  1047. auto plugin_decoder = TRY_OR_FAIL(Gfx::TinyVGImageDecoderPlugin::create(file->bytes()));
  1048. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 400, 768 }));
  1049. }
  1050. }
  1051. TEST_CASE(test_tvg_malformed)
  1052. {
  1053. Array test_inputs = {
  1054. TEST_INPUT("tvg/bogus-color-table-size.tvg"sv)
  1055. };
  1056. for (auto test_input : test_inputs) {
  1057. auto file = TRY_OR_FAIL(Core::MappedFile::map(test_input));
  1058. auto plugin_decoder = TRY_OR_FAIL(Gfx::TinyVGImageDecoderPlugin::create(file->bytes()));
  1059. auto frame_or_error = plugin_decoder->frame(0);
  1060. EXPECT(frame_or_error.is_error());
  1061. }
  1062. }
  1063. TEST_CASE(test_tvg_rgb565)
  1064. {
  1065. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("tvg/green-rgb565.tvg"sv)));
  1066. EXPECT(Gfx::TinyVGImageDecoderPlugin::sniff(file->bytes()));
  1067. auto plugin_decoder = TRY_OR_FAIL(Gfx::TinyVGImageDecoderPlugin::create(file->bytes()));
  1068. auto frame = TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 100, 100 }));
  1069. // Should be a solid dark green:
  1070. EXPECT_EQ(frame.image->get_pixel(50, 50), Gfx::Color(0, 130, 0));
  1071. }
  1072. TEST_CASE(test_jxl_modular_simple_tree_upsample2_10bits)
  1073. {
  1074. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jxl/modular_simple_tree_upsample2_10bits_rct.jxl"sv)));
  1075. EXPECT(Gfx::JPEGXLImageDecoderPlugin::sniff(file->bytes()));
  1076. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGXLImageDecoderPlugin::create(file->bytes()));
  1077. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 128, 128 }));
  1078. auto frame = TRY_OR_FAIL(plugin_decoder->frame(0));
  1079. EXPECT_EQ(frame.image->get_pixel(42, 57), Gfx::Color::from_string("#4c0072"sv));
  1080. }
  1081. TEST_CASE(test_jxl_modular_property_8)
  1082. {
  1083. auto file = TRY_OR_FAIL(Core::MappedFile::map(TEST_INPUT("jxl/modular_property_8.jxl"sv)));
  1084. EXPECT(Gfx::JPEGXLImageDecoderPlugin::sniff(file->bytes()));
  1085. auto plugin_decoder = TRY_OR_FAIL(Gfx::JPEGXLImageDecoderPlugin::create(file->bytes()));
  1086. TRY_OR_FAIL(expect_single_frame_of_size(*plugin_decoder, { 32, 32 }));
  1087. auto frame = TRY_OR_FAIL(plugin_decoder->frame(0));
  1088. for (u8 i = 0; i < 32; ++i) {
  1089. for (u8 j = 0; j < 32; ++j) {
  1090. auto const color = frame.image->get_pixel(i, j);
  1091. if ((i + j) % 2 == 0)
  1092. EXPECT_EQ(color, Gfx::Color::Black);
  1093. else
  1094. EXPECT_EQ(color, Gfx::Color::Yellow);
  1095. }
  1096. }
  1097. }
  1098. TEST_CASE(test_dds)
  1099. {
  1100. Array file_names = {
  1101. TEST_INPUT("dds/catdog-alert-29x29.dds"sv),
  1102. TEST_INPUT("dds/catdog-alert-32x32.dds"sv)
  1103. };
  1104. for (auto file_name : file_names) {
  1105. auto file = TRY_OR_FAIL(Core::MappedFile::map(file_name));
  1106. EXPECT(Gfx::DDSImageDecoderPlugin::sniff(file->bytes()));
  1107. auto plugin_decoder = TRY_OR_FAIL(Gfx::DDSImageDecoderPlugin::create(file->bytes()));
  1108. TRY_OR_FAIL(expect_single_frame(*plugin_decoder));
  1109. }
  1110. }