TestImageDecoder.cpp 55 KB

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