vcl/qa/cppunit/jpeg/JpegReaderTest.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  */
  10
  11 #include <unotest/bootstrapfixturebase.hxx>
  12 #include <vcl/graphicfilter.hxx>
  13 #include <bitmapwriteaccess.hxx>
  14 #include <tools/stream.hxx>
  15
  16 static OUString const gaDataUrl("/vcl/qa/cppunit/jpeg/data/");
  17
  18 class JpegReaderTest : public test::BootstrapFixtureBase
  19 {
  20     OUString getFullUrl(const OUString& sFileName)
  21     {
  22         return m_directories.getURLFromSrc(gaDataUrl) + sFileName;
  23     }
  24
  25     Graphic loadJPG(const OUString& aURL);
  26
  27 public:
  28     void testReadRGB();
  29     void testReadGray();
  30     void testReadCMYK();
  31
  32     CPPUNIT_TEST_SUITE(JpegReaderTest);
  33     CPPUNIT_TEST(testReadRGB);
  34     CPPUNIT_TEST(testReadGray);
  35     CPPUNIT_TEST(testReadCMYK);
  36     CPPUNIT_TEST_SUITE_END();
  37 };
  38
  39 static int deltaColor(BitmapColor aColor1, BitmapColor aColor2)
  40 {
  41     int deltaR = std::abs(aColor1.GetRed()   - aColor2.GetRed());
  42     int deltaG = std::abs(aColor1.GetGreen() - aColor2.GetGreen());
  43     int deltaB = std::abs(aColor1.GetBlue()  - aColor2.GetBlue());
  44
  45     return std::max(std::max(deltaR, deltaG), deltaB);
  46 }
  47
  48 static bool checkRect(Bitmap& rBitmap, int aLayerNumber, long nAreaHeight, long nAreaWidth, Color aExpectedColor, int nMaxDelta)
  49 {
  50     BitmapScopedWriteAccess pAccess(rBitmap);
  51
  52     long nWidth  = std::min(nAreaWidth,  pAccess->Width());
  53     long nHeight = std::min(nAreaHeight, pAccess->Height());
  54
  55     long firstX = 0 + aLayerNumber;
  56     long firstY = 0 + aLayerNumber;
  57
  58     long lastX = nWidth  - 1 - aLayerNumber;
  59     long lastY = nHeight - 1 - aLayerNumber;
  60
  61     int delta;
  62
  63     for (long y = firstY; y <= lastY; y++)
  64     {
  65         Color aColorFirst = pAccess->GetPixel(y, firstX);
  66         delta = deltaColor(aColorFirst, aExpectedColor);
  67         if (delta > nMaxDelta)
  68             return false;
  69
  70         Color aColorLast = pAccess->GetPixel(y, lastX);
  71         delta = deltaColor(aColorLast, aExpectedColor);
  72         if (delta > nMaxDelta)
  73             return false;
  74     }
  75     for (long x = firstX; x <= lastX; x++)
  76     {
  77         Color aColorFirst = pAccess->GetPixel(firstY, x);
  78         delta = deltaColor(aColorFirst, aExpectedColor);
  79         if (delta > nMaxDelta)
  80             return false;
  81
  82         Color aColorLast  = pAccess->GetPixel(lastY, x);
  83         delta = deltaColor(aColorLast, aExpectedColor);
  84         if (delta > nMaxDelta)
  85             return false;
  86     }
  87     return true;
  88 }
  89
  90 static int getNumberOfImageComponents(const Graphic& rGraphic)
  91 {
  92     GfxLink aLink = rGraphic.GetGfxLink();
  93     SvMemoryStream aMemoryStream(const_cast<sal_uInt8*>(aLink.GetData()), aLink.GetDataSize(),
  94                                  StreamMode::READ | StreamMode::WRITE);
  95     GraphicDescriptor aDescriptor(aMemoryStream, nullptr);
  96     CPPUNIT_ASSERT(aDescriptor.Detect(true));
  97     return aDescriptor.GetNumberOfImageComponents();
  98 }
  99
 100 Graphic JpegReaderTest::loadJPG(const OUString& aURL)
 101 {
 102     GraphicFilter& rFilter = GraphicFilter::GetGraphicFilter();
 103     Graphic aGraphic;
 104     SvFileStream aFileStream(aURL, StreamMode::READ);
 105     ErrCode bResult = rFilter.ImportGraphic(aGraphic, aURL, aFileStream);
 106     CPPUNIT_ASSERT_EQUAL(ERRCODE_NONE, bResult);
 107     return aGraphic;
 108 }
 109
 110 void JpegReaderTest::testReadRGB()
 111 {
 112     Graphic aGraphic = loadJPG(getFullUrl("JPEGTestRGB.jpeg"));
 113     Bitmap aBitmap = aGraphic.GetBitmapEx().GetBitmap();
 114     Size aSize = aBitmap.GetSizePixel();
 115     CPPUNIT_ASSERT_EQUAL(12L, aSize.Width());
 116     CPPUNIT_ASSERT_EQUAL(12L, aSize.Height());
 117
 118     int nMaxDelta = 1; // still acceptable color error
 119     CPPUNIT_ASSERT(checkRect(aBitmap, 0, 8, 8, Color(0xff, 0xff, 0xff), nMaxDelta));
 120     CPPUNIT_ASSERT(checkRect(aBitmap, 1, 8, 8, Color(0xff, 0x00, 0x00), nMaxDelta));
 121     CPPUNIT_ASSERT(checkRect(aBitmap, 2, 8, 8, Color(0x00, 0xff, 0x00), nMaxDelta));
 122     CPPUNIT_ASSERT(checkRect(aBitmap, 3, 8, 8, Color(0x00, 0x00, 0xff), nMaxDelta));
 123
 124     CPPUNIT_ASSERT_EQUAL(3, getNumberOfImageComponents(aGraphic));
 125 }
 126
 127 void JpegReaderTest::testReadGray()
 128 {
 129     Graphic aGraphic = loadJPG(getFullUrl("JPEGTestGray.jpeg"));
 130     Bitmap aBitmap = aGraphic.GetBitmapEx().GetBitmap();
 131     Size aSize = aBitmap.GetSizePixel();
 132     CPPUNIT_ASSERT_EQUAL(12L, aSize.Width());
 133     CPPUNIT_ASSERT_EQUAL(12L, aSize.Height());
 134
 135     aBitmap.Convert(BmpConversion::N24Bit); // convert to 24bit so we don't need to deal with palette
 136
 137     int nMaxDelta = 1;
 138     CPPUNIT_ASSERT(checkRect(aBitmap, 0, 8, 8, Color(0xff, 0xff, 0xff), nMaxDelta));
 139     CPPUNIT_ASSERT(checkRect(aBitmap, 1, 8, 8, Color(0x36, 0x36, 0x36), nMaxDelta));
 140     CPPUNIT_ASSERT(checkRect(aBitmap, 2, 8, 8, Color(0xb6, 0xb6, 0xb6), nMaxDelta));
 141     CPPUNIT_ASSERT(checkRect(aBitmap, 3, 8, 8, Color(0x12, 0x12, 0x12), nMaxDelta));
 142
 143     CPPUNIT_ASSERT_EQUAL(1, getNumberOfImageComponents(aGraphic));
 144 }
 145
 146 void JpegReaderTest::testReadCMYK()
 147 {
 148     Graphic aGraphic = loadJPG(getFullUrl("JPEGTestCMYK.jpeg"));
 149     Bitmap aBitmap = aGraphic.GetBitmapEx().GetBitmap();
 150     Size aSize = aBitmap.GetSizePixel();
 151     CPPUNIT_ASSERT_EQUAL(12L, aSize.Width());
 152     CPPUNIT_ASSERT_EQUAL(12L, aSize.Height());
 153
 154     int maxDelta = 1;
 155     CPPUNIT_ASSERT(checkRect(aBitmap, 0, 8, 8, Color(0xff, 0xff, 0xff), maxDelta));
 156     CPPUNIT_ASSERT(checkRect(aBitmap, 1, 8, 8, Color(0xff, 0x00, 0x00), maxDelta));
 157     CPPUNIT_ASSERT(checkRect(aBitmap, 2, 8, 8, Color(0x00, 0xff, 0x00), maxDelta));
 158     CPPUNIT_ASSERT(checkRect(aBitmap, 3, 8, 8, Color(0x00, 0x00, 0xff), maxDelta));
 159
 160     CPPUNIT_ASSERT_EQUAL(4, getNumberOfImageComponents(aGraphic));
 161 }
 162
 163 CPPUNIT_TEST_SUITE_REGISTRATION(JpegReaderTest);
 164
 165 CPPUNIT_PLUGIN_IMPLEMENT();
 166
 167 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */