gfx/thebes/gfxAlphaRecovery.cpp

   1 /* -*- Mode: C++; tab-width: 20; indent-tabs-mode: nil; c-basic-offset: 2 -*-
   2  * This Source Code Form is subject to the terms of the Mozilla Public
   3  * License, v. 2.0. If a copy of the MPL was not distributed with this
   4  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
   5
   6 #include "gfxAlphaRecovery.h"
   7
   8 #include "gfxImageSurface.h"
   9
  10 #define MOZILLA_SSE_INCLUDE_HEADER_FOR_SSE2
  11 #include "mozilla/SSE.h"
  12
  13 #include <xsimd/xsimd.hpp>
  14
  15 /* static */
  16 bool gfxAlphaRecovery::RecoverAlpha(gfxImageSurface* blackSurf,
  17                                     const gfxImageSurface* whiteSurf) {
  18   mozilla::gfx::IntSize size = blackSurf->GetSize();
  19
  20   if (size != whiteSurf->GetSize() ||
  21       (blackSurf->Format() != mozilla::gfx::SurfaceFormat::A8R8G8B8_UINT32 &&
  22        blackSurf->Format() != mozilla::gfx::SurfaceFormat::X8R8G8B8_UINT32) ||
  23       (whiteSurf->Format() != mozilla::gfx::SurfaceFormat::A8R8G8B8_UINT32 &&
  24        whiteSurf->Format() != mozilla::gfx::SurfaceFormat::X8R8G8B8_UINT32))
  25     return false;
  26
  27 #ifdef MOZILLA_MAY_SUPPORT_SSE2
  28   if (mozilla::supports_sse2() &&
  29       RecoverAlphaGeneric<xsimd::sse2>(blackSurf, whiteSurf)) {
  30     return true;
  31   }
  32 #endif
  33 #ifdef MOZILLA_MAY_SUPPORT_NEON
  34   if (mozilla::supports_neon() &&
  35       RecoverAlphaGeneric<xsimd::neon>(blackSurf, whiteSurf)) {
  36     return true;
  37   }
  38 #endif
  39
  40   blackSurf->Flush();
  41   whiteSurf->Flush();
  42
  43   unsigned char* blackData = blackSurf->Data();
  44   unsigned char* whiteData = whiteSurf->Data();
  45
  46   for (int32_t i = 0; i < size.height; ++i) {
  47     uint32_t* blackPixel = reinterpret_cast<uint32_t*>(blackData);
  48     const uint32_t* whitePixel = reinterpret_cast<uint32_t*>(whiteData);
  49     for (int32_t j = 0; j < size.width; ++j) {
  50       uint32_t recovered = RecoverPixel(blackPixel[j], whitePixel[j]);
  51       blackPixel[j] = recovered;
  52     }
  53     blackData += blackSurf->Stride();
  54     whiteData += whiteSurf->Stride();
  55   }
  56
  57   blackSurf->MarkDirty();
  58
  59   return true;
  60 }