/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef MOZILLA_GFX_FILTERPROCESSING_H_
#define MOZILLA_GFX_FILTERPROCESSING_H_

#include "2D.h"
#include "Filters.h"

namespace mozilla::gfx {

const ptrdiff_t B8G8R8A8_COMPONENT_BYTEOFFSET_B =
    static_cast<ptrdiff_t>(mozilla::gfx::SurfaceFormatBit::OS_B) / 8;
const ptrdiff_t B8G8R8A8_COMPONENT_BYTEOFFSET_G =
    static_cast<ptrdiff_t>(mozilla::gfx::SurfaceFormatBit::OS_G) / 8;
const ptrdiff_t B8G8R8A8_COMPONENT_BYTEOFFSET_R =
    static_cast<ptrdiff_t>(mozilla::gfx::SurfaceFormatBit::OS_R) / 8;
const ptrdiff_t B8G8R8A8_COMPONENT_BYTEOFFSET_A =
    static_cast<ptrdiff_t>(mozilla::gfx::SurfaceFormatBit::OS_A) / 8;

class FilterProcessing {
 public:
  // Fast approximate division by 255. It has the property that
  // for all 0 <= v <= 255*255, FastDivideBy255(v) == v/255.
  // But it only uses two adds and two shifts instead of an
  // integer division (which is expensive on many processors).
  template <class B, class A>
  static B FastDivideBy255(A v) {
    return ((v << 8) + v + 255) >> 16;
  }

  static already_AddRefed<DataSourceSurface> ExtractAlpha(
      DataSourceSurface* aSource);
  static already_AddRefed<DataSourceSurface> ConvertToB8G8R8A8(
      SourceSurface* aSurface);
  static already_AddRefed<DataSourceSurface> ApplyBlending(
      DataSourceSurface* aInput1, DataSourceSurface* aInput2,
      BlendMode aBlendMode);
  static void ApplyMorphologyHorizontal(const uint8_t* aSourceData,
                                        int32_t aSourceStride,
                                        uint8_t* aDestData, int32_t aDestStride,
                                        const IntRect& aDestRect,
                                        int32_t aRadius,
                                        MorphologyOperator aOperator);
  static void ApplyMorphologyVertical(const uint8_t* aSourceData,
                                      int32_t aSourceStride, uint8_t* aDestData,
                                      int32_t aDestStride,
                                      const IntRect& aDestRect, int32_t aRadius,
                                      MorphologyOperator aOperator);
  static already_AddRefed<DataSourceSurface> ApplyColorMatrix(
      DataSourceSurface* aInput, const Matrix5x4& aMatrix);
  static void ApplyComposition(DataSourceSurface* aSource,
                               DataSourceSurface* aDest,
                               CompositeOperator aOperator);
  static void DoPremultiplicationCalculation(const IntSize& aSize,
                                             uint8_t* aTargetData,
                                             int32_t aTargetStride,
                                             const uint8_t* aSourceData,
                                             int32_t aSourceStride);
  static void DoUnpremultiplicationCalculation(const IntSize& aSize,
                                               uint8_t* aTargetData,
                                               int32_t aTargetStride,
                                               const uint8_t* aSourceData,
                                               int32_t aSourceStride);
  static void DoOpacityCalculation(const IntSize& aSize, uint8_t* aTargetData,
                                   int32_t aTargetStride,
                                   const uint8_t* aSourceData,
                                   int32_t aSourceStride, Float aValue);
  static void DoOpacityCalculationA8(const IntSize& aSize, uint8_t* aTargetData,
                                     int32_t aTargetStride,
                                     const uint8_t* aSourceData,
                                     int32_t aSourceStride, Float aValue);
  static already_AddRefed<DataSourceSurface> RenderTurbulence(
      const IntSize& aSize, const Point& aOffset, const Size& aBaseFrequency,
      int32_t aSeed, int aNumOctaves, TurbulenceType aType, bool aStitch,
      const Rect& aTileRect);
  static already_AddRefed<DataSourceSurface> ApplyArithmeticCombine(
      DataSourceSurface* aInput1, DataSourceSurface* aInput2, Float aK1,
      Float aK2, Float aK3, Float aK4);

 protected:
  static void ExtractAlpha_Scalar(const IntSize& size,
                                  const uint8_t* sourceData,
                                  int32_t sourceStride, uint8_t* alphaData,
                                  int32_t alphaStride);
  static already_AddRefed<DataSourceSurface> ConvertToB8G8R8A8_Scalar(
      SourceSurface* aSurface);
  static void ApplyMorphologyHorizontal_Scalar(
      const uint8_t* aSourceData, int32_t aSourceStride, uint8_t* aDestData,
      int32_t aDestStride, const IntRect& aDestRect, int32_t aRadius,
      MorphologyOperator aOperator);
  static void ApplyMorphologyVertical_Scalar(
      const uint8_t* aSourceData, int32_t aSourceStride, uint8_t* aDestData,
      int32_t aDestStride, const IntRect& aDestRect, int32_t aRadius,
      MorphologyOperator aOperator);
  static already_AddRefed<DataSourceSurface> ApplyColorMatrix_Scalar(
      DataSourceSurface* aInput, const Matrix5x4& aMatrix);
  static void ApplyComposition_Scalar(DataSourceSurface* aSource,
                                      DataSourceSurface* aDest,
                                      CompositeOperator aOperator);
  static void DoOpacityCalculation_Scalar(const IntSize& aSize,
                                          uint8_t* aTargetData,
                                          int32_t aTargetStride,
                                          const uint8_t* aSourceData,
                                          int32_t aSourceStride, Float aValue);
  static void DoOpacityCalculationA8_Scalar(
      const IntSize& aSize, uint8_t* aTargetData, int32_t aTargetStride,
      const uint8_t* aSourceData, int32_t aSourceStride, Float aValue);
  static already_AddRefed<DataSourceSurface> RenderTurbulence_Scalar(
      const IntSize& aSize, const Point& aOffset, const Size& aBaseFrequency,
      int32_t aSeed, int aNumOctaves, TurbulenceType aType, bool aStitch,
      const Rect& aTileRect);
  static already_AddRefed<DataSourceSurface> ApplyArithmeticCombine_Scalar(
      DataSourceSurface* aInput1, DataSourceSurface* aInput2, Float aK1,
      Float aK2, Float aK3, Float aK4);

#ifdef USE_SSE2
  static void ExtractAlpha_SSE2(const IntSize& size, const uint8_t* sourceData,
                                int32_t sourceStride, uint8_t* alphaData,
                                int32_t alphaStride);
  static already_AddRefed<DataSourceSurface> ConvertToB8G8R8A8_SSE2(
      SourceSurface* aSurface);
  static already_AddRefed<DataSourceSurface> ApplyBlending_SSE2(
      DataSourceSurface* aInput1, DataSourceSurface* aInput2,
      BlendMode aBlendMode);
  static void ApplyMorphologyHorizontal_SSE2(
      const uint8_t* aSourceData, int32_t aSourceStride, uint8_t* aDestData,
      int32_t aDestStride, const IntRect& aDestRect, int32_t aRadius,
      MorphologyOperator aOperator);
  static void ApplyMorphologyVertical_SSE2(
      const uint8_t* aSourceData, int32_t aSourceStride, uint8_t* aDestData,
      int32_t aDestStride, const IntRect& aDestRect, int32_t aRadius,
      MorphologyOperator aOperator);
  static already_AddRefed<DataSourceSurface> ApplyColorMatrix_SSE2(
      DataSourceSurface* aInput, const Matrix5x4& aMatrix);
  static void ApplyComposition_SSE2(DataSourceSurface* aSource,
                                    DataSourceSurface* aDest,
                                    CompositeOperator aOperator);
  static void DoOpacityCalculation_SSE2(const IntSize& aSize,
                                        uint8_t* aTargetData,
                                        int32_t aTargetStride,
                                        const uint8_t* aSourceData,
                                        int32_t aSourceStride, Float aValue);
  static already_AddRefed<DataSourceSurface> RenderTurbulence_SSE2(
      const IntSize& aSize, const Point& aOffset, const Size& aBaseFrequency,
      int32_t aSeed, int aNumOctaves, TurbulenceType aType, bool aStitch,
      const Rect& aTileRect);
  static already_AddRefed<DataSourceSurface> ApplyArithmeticCombine_SSE2(
      DataSourceSurface* aInput1, DataSourceSurface* aInput2, Float aK1,
      Float aK2, Float aK3, Float aK4);
#endif
};

// Constant-time max and min functions for unsigned arguments
static inline unsigned umax(unsigned a, unsigned b) {
  return a - ((a - b) & -(a < b));
}

static inline unsigned umin(unsigned a, unsigned b) {
  return a - ((a - b) & -(a > b));
}

}  // namespace mozilla::gfx

#endif  // MOZILLA_GFX_FILTERPROCESSING_H_