X-Git-Url: https://git.saurik.com/wxWidgets.git/blobdiff_plain/2960bae821456e2adae52936839898e0715b0889..c29c95fe24973b94fd724db767193171ca7c513d:/tests/image/image.cpp diff --git a/tests/image/image.cpp b/tests/image/image.cpp index 7cadc7fb32..6b98ad4f3c 100644 --- a/tests/image/image.cpp +++ b/tests/image/image.cpp @@ -23,7 +23,7 @@ #ifndef WX_PRECOMP #endif // WX_PRECOMP -#include "wx/image.h" +#include "wx/anidecod.h" // wxImageArray #include "wx/palette.h" #include "wx/url.h" #include "wx/log.h" @@ -31,6 +31,8 @@ #include "wx/zstream.h" #include "wx/wfstream.h" +#include "testimage.h" + struct testData { const char* file; wxBitmapType type; @@ -70,6 +72,11 @@ private: CPPUNIT_TEST( SizeImage ); CPPUNIT_TEST( CompareLoadedImage ); CPPUNIT_TEST( CompareSavedImage ); + CPPUNIT_TEST( SavePNG ); + CPPUNIT_TEST( SaveAnimatedGIF ); + CPPUNIT_TEST( ReadCorruptedTGA ); + CPPUNIT_TEST( GIFComment ); + CPPUNIT_TEST( DibPadding ); CPPUNIT_TEST_SUITE_END(); void LoadFromSocketStream(); @@ -78,6 +85,11 @@ private: void SizeImage(); void CompareLoadedImage(); void CompareSavedImage(); + void SavePNG(); + void SaveAnimatedGIF(); + void ReadCorruptedTGA(); + void GIFComment(); + void DibPadding(); DECLARE_NO_COPY_CLASS(ImageTestCase) }; @@ -821,12 +833,10 @@ void ImageTestCase::SizeImage() CPPUNIT_ASSERT_EQUAL( actual.GetSize().x, expected.GetSize().x ); CPPUNIT_ASSERT_EQUAL( actual.GetSize().y, expected.GetSize().y ); - const unsigned data_len = 3 * expected.GetHeight() * expected.GetWidth(); - - WX_ASSERT_MESSAGE + WX_ASSERT_EQUAL_MESSAGE ( ("Resize test #%u: (%d, %d), (%d, %d)", i, st.w, st.h, st.dx, st.dy), - memcmp(actual.GetData(), expected.GetData(), data_len) == 0 + expected, actual ); } } @@ -839,8 +849,6 @@ void ImageTestCase::CompareLoadedImage() wxImage expected24("horse.png"); CPPUNIT_ASSERT( expected24.IsOk() ); - const size_t dataLen = expected8.GetWidth() * expected8.GetHeight() * 3; - for (size_t i=0; iGetSize() ); unsigned bitsPerPixel = testPalette ? 8 : (testAlpha ? 32 : 24); - WX_ASSERT_MESSAGE + WX_ASSERT_EQUAL_MESSAGE ( ("Compare test '%s (%d-bit)' for saving failed", handler.GetExtension(), bitsPerPixel), - - memcmp(actual.GetData(), expected->GetData(), - expected->GetWidth() * expected->GetHeight() * 3) == 0 + *expected, + actual ); #if wxUSE_PALETTE @@ -956,12 +960,11 @@ void CompareImage(const wxImageHandler& handler, const wxImage& image, return; } - WX_ASSERT_MESSAGE + WX_ASSERT_EQUAL_MESSAGE ( ("Compare alpha test '%s' for saving failed", handler.GetExtension()), - - memcmp(actual.GetAlpha(), expected->GetAlpha(), - expected->GetWidth() * expected->GetHeight()) == 0 + *expected, + actual ); } @@ -976,10 +979,9 @@ void ImageTestCase::CompareSavedImage() CPPUNIT_ASSERT( expected24.IsOk() ); CPPUNIT_ASSERT( !expected24.HasAlpha() ); - unsigned long numColours = expected24.CountColours(); wxImage expected8 = expected24.ConvertToGreyscale(); - numColours = expected8.CountColours(); +#if wxUSE_PALETTE unsigned char greys[256]; for (i = 0; i < 256; ++i) { @@ -987,6 +989,8 @@ void ImageTestCase::CompareSavedImage() } wxPalette palette(256, greys, greys, greys); expected8.SetPalette(palette); +#endif // #if wxUSE_PALETTE + expected8.SetOption(wxIMAGE_OPTION_BMP_FORMAT, wxBMP_8BPP_PALETTE); // Create an image with alpha based on the loaded image @@ -1015,74 +1019,247 @@ void ImageTestCase::CompareSavedImage() CompareImage(*handler, expected24); CompareImage(*handler, expected32, wxIMAGE_HAVE_ALPHA); } +} + +void ImageTestCase::SavePNG() +{ + wxImage expected24("horse.png"); + CPPUNIT_ASSERT( expected24.IsOk() ); +#if wxUSE_PALETTE + CPPUNIT_ASSERT( !expected24.HasPalette() ); +#endif // #if wxUSE_PALETTE + wxImage expected8 = expected24.ConvertToGreyscale(); - expected8.LoadFile("horse.gif"); - CPPUNIT_ASSERT( expected8.IsOk() ); + /* + horse.png converted to greyscale should be saved without a palette. + */ + CompareImage(*wxImage::FindHandler(wxBITMAP_TYPE_PNG), expected8); + + /* + But if we explicitly ask for trying to save with a palette, it should work. + */ + expected8.SetOption(wxIMAGE_OPTION_PNG_FORMAT, wxPNG_TYPE_PALETTE); + + CompareImage(*wxImage::FindHandler(wxBITMAP_TYPE_PNG), + expected8, wxIMAGE_HAVE_PALETTE); + + + CPPUNIT_ASSERT( expected8.LoadFile("horse.gif") ); +#if wxUSE_PALETTE CPPUNIT_ASSERT( expected8.HasPalette() ); +#endif // #if wxUSE_PALETTE + + CompareImage(*wxImage::FindHandler(wxBITMAP_TYPE_PNG), + expected8, wxIMAGE_HAVE_PALETTE); + /* + Add alpha to the image in such a way that there will still be a maximum + of 256 unique RGBA combinations. This should result in a saved + PNG image still being palettised and having alpha. + */ expected8.SetAlpha(); - width = expected8.GetWidth(); - height = expected8.GetHeight(); + int x, y; + const int width = expected8.GetWidth(); + const int height = expected8.GetHeight(); for (y = 0; y < height; ++y) { for (x = 0; x < width; ++x) { - expected8.SetAlpha(x, y, (x*y) & wxIMAGE_ALPHA_OPAQUE); + expected8.SetAlpha(x, y, expected8.GetRed(x, y)); } } + CompareImage(*wxImage::FindHandler(wxBITMAP_TYPE_PNG), + expected8, wxIMAGE_HAVE_ALPHA|wxIMAGE_HAVE_PALETTE); + /* - The image contains 256 indexed colours and needs another palette entry - for storing the transparency index. This results in wanting 257 palette - entries but that amount is not supported by PNG, as such this image - should not contain a palette (but still have alpha) and be stored as a - true colour image instead. + Now change the alpha of the first pixel so that we can't save palettised + anymore because there will be 256+1 entries which is beyond PNGs limit + of 256 entries. */ + expected8.SetAlpha(0, 0, 1); + CompareImage(*wxImage::FindHandler(wxBITMAP_TYPE_PNG), expected8, wxIMAGE_HAVE_ALPHA); -#if wxUSE_PALETTE /* - Now do the same test again but remove one (random) palette entry. This - should result in saving the PNG with a palette. + Even if we explicitly ask for saving palettised it should not be done. */ - unsigned char red[256], green[256], blue[256]; - const wxPalette& pal = expected8.GetPalette(); - const int paletteCount = pal.GetColoursCount(); - for (i = 0; i < paletteCount; ++i) + expected8.SetOption(wxIMAGE_OPTION_PNG_FORMAT, wxPNG_TYPE_PALETTE); + CompareImage(*wxImage::FindHandler(wxBITMAP_TYPE_PNG), + expected8, wxIMAGE_HAVE_ALPHA); + +} + +void ImageTestCase::SaveAnimatedGIF() +{ +#if wxUSE_PALETTE + wxImage image("horse.gif"); + CPPUNIT_ASSERT( image.IsOk() ); + + wxImageArray images; + images.Add(image); + int i; + for (i = 0; i < 4-1; ++i) { - expected8.GetPalette().GetRGB(i, &red[i], &green[i], &blue[i]); + images.Add( images[i].Rotate90() ); + + images[i+1].SetPalette(images[0].GetPalette()); } - wxPalette newPal(paletteCount - 1, red, green, blue); - expected8.Replace( - red[paletteCount-1], green[paletteCount-1], blue[paletteCount-1], - red[paletteCount-2], green[paletteCount-2], blue[paletteCount-2]); - expected8.SetPalette(newPal); + wxMemoryOutputStream memOut; + CPPUNIT_ASSERT( wxGIFHandler().SaveAnimation(images, &memOut) ); + + wxGIFHandler handler; + wxMemoryInputStream memIn(memOut); + CPPUNIT_ASSERT(memIn.IsOk()); + const int imageCount = handler.GetImageCount(memIn); + CPPUNIT_ASSERT_EQUAL(4, imageCount); + + for (i = 0; i < imageCount; ++i) + { + wxFileOffset pos = memIn.TellI(); + CPPUNIT_ASSERT( handler.LoadFile(&image, memIn, true, i) ); + memIn.SeekI(pos); + + WX_ASSERT_EQUAL_MESSAGE + ( + ("Compare test for GIF frame number %d failed", i), + images[i], + image + ); + } +#endif // #if wxUSE_PALETTE +} + +void ImageTestCase::ReadCorruptedTGA() +{ + static unsigned char corruptTGA[18+1+3] = + { + 0, + 0, + 10, // RLE compressed image. + 0, 0, + 0, 0, + 0, + 0, 0, + 0, 0, + 1, 0, // Width is 1. + 1, 0, // Height is 1. + 24, // Bits per pixel. + 0, + + 0xff, // Run length (repeat next pixel 127+1 times). + 0xff, 0xff, 0xff // One 24-bit pixel. + }; + + wxMemoryInputStream memIn(corruptTGA, WXSIZEOF(corruptTGA)); + CPPUNIT_ASSERT(memIn.IsOk()); + + wxImage tgaImage; + CPPUNIT_ASSERT( !tgaImage.LoadFile(memIn) ); + + /* - Explicitly make known we want a palettised PNG. If we don't then this - particular image gets saved as a true colour image because there's an - alpha channel present and the PNG saver prefers to keep the alpha over - saving as a palettised image that has alpha converted to a mask. + Instead of repeating a pixel 127+1 times, now tell it there will + follow 127+1 uncompressed pixels (while we only should have 1 in total). */ - expected8.SetOption(wxIMAGE_OPTION_PNG_FORMAT, wxPNG_TYPE_PALETTE); + corruptTGA[18] = 0x7f; + CPPUNIT_ASSERT( !tgaImage.LoadFile(memIn) ); +} + +static void TestGIFComment(const wxString& comment) +{ + wxImage image("horse.gif"); + + image.SetOption(wxIMAGE_OPTION_GIF_COMMENT, comment); + wxMemoryOutputStream memOut; + CPPUNIT_ASSERT(image.SaveFile(memOut, wxBITMAP_TYPE_GIF)); + + wxMemoryInputStream memIn(memOut); + CPPUNIT_ASSERT( image.LoadFile(memIn) ); + + CPPUNIT_ASSERT_EQUAL(comment, + image.GetOption(wxIMAGE_OPTION_GIF_COMMENT)); +} + +void ImageTestCase::GIFComment() +{ + // Test reading a comment. + wxImage image("horse.gif"); + CPPUNIT_ASSERT_EQUAL(" Imported from GRADATION image: gray", + image.GetOption(wxIMAGE_OPTION_GIF_COMMENT)); + + + // Test writing a comment and reading it back. + TestGIFComment("Giving the GIF a gifted giraffe as a gift"); - wxImage ref8 = expected8; + // Test writing and reading a comment again but with a long comment. + TestGIFComment(wxString(wxT('a'), 256) + + wxString(wxT('b'), 256) + + wxString(wxT('c'), 256)); + + + // Test writing comments in an animated GIF and reading them back. + CPPUNIT_ASSERT( image.LoadFile("horse.gif") ); + + wxImageArray images; + int i; + for (i = 0; i < 4; ++i) + { + if (i) + { + images.Add( images[i-1].Rotate90() ); + images[i].SetPalette(images[0].GetPalette()); + } + else + { + images.Add(image); + } + + images[i].SetOption(wxIMAGE_OPTION_GIF_COMMENT, + wxString::Format("GIF comment for frame #%d", i+1)); + + } + + + wxMemoryOutputStream memOut; + CPPUNIT_ASSERT( wxGIFHandler().SaveAnimation(images, &memOut) ); + + wxGIFHandler handler; + wxMemoryInputStream memIn(memOut); + CPPUNIT_ASSERT(memIn.IsOk()); + const int imageCount = handler.GetImageCount(memIn); + for (i = 0; i < imageCount; ++i) + { + wxFileOffset pos = memIn.TellI(); + CPPUNIT_ASSERT( handler.LoadFile(&image, memIn, true /*verbose?*/, i) ); + + CPPUNIT_ASSERT_EQUAL( + wxString::Format("GIF comment for frame #%d", i+1), + image.GetOption(wxIMAGE_OPTION_GIF_COMMENT)); + memIn.SeekI(pos); + } +} + +void ImageTestCase::DibPadding() +{ /* - Convert the alpha channel to a mask like the PNG saver does. Also convert - the colour used for transparency from 1,0,0 to 2,0,0. The latter gets - done by the PNG loader in search of an unused colour to use for - transparency (this should be fixed). + There used to be an error with calculating the DWORD aligned scan line + pitch for a BMP/ICO resulting in buffer overwrites (with at least MSVC9 + Debug this gave a heap corruption assertion when saving the mask of + an ICO). Test for it here. */ - ref8.ConvertAlphaToMask(); - ref8.Replace(1, 0, 0, 2, 0, 0); + wxImage image("horse.gif"); + CPPUNIT_ASSERT( image.IsOk() ); - CompareImage(*wxImage::FindHandler(wxBITMAP_TYPE_PNG), - expected8, wxIMAGE_HAVE_PALETTE, &ref8); -#endif + image = image.Scale(99, 99); + + wxMemoryOutputStream memOut; + CPPUNIT_ASSERT( image.SaveFile(memOut, wxBITMAP_TYPE_ICO) ); } #endif //wxUSE_IMAGE