// Copyright (c) the JPEG XL Project Authors. All rights reserved.

//

// Use of this source code is governed by a BSD-style

// license that can be found in the LICENSE file.

#include "lib/jxl/alpha.h"

#include <algorithm>

#include <cstring>

#include "lib/jxl/base/common.h"

#include "lib/jxl/base/compiler_specific.h"

namespace jxl {

static float Clamp(float x) { return jxl::Clamp1(x, 0.0f, 1.0f); }

void PerformAlphaBlending(const AlphaBlendingInputLayer& bg,

                          const AlphaBlendingInputLayer& fg,

                          const AlphaBlendingOutput& out, size_t num_pixels,

                          bool alpha_is_premultiplied, bool clamp) {

  if (alpha_is_premultiplied) {

    for (size_t x = 0; x < num_pixels; ++x) {

      float fga = clamp ? Clamp(fg.a[x]) : fg.a[x];

      out.r[x] = (fg.r[x] + bg.r[x] * (1.f - fga));

      out.g[x] = (fg.g[x] + bg.g[x] * (1.f - fga));

      out.b[x] = (fg.b[x] + bg.b[x] * (1.f - fga));

      out.a[x] = (1.f - (1.f - fga) * (1.f - bg.a[x]));

    }

  } else {

    for (size_t x = 0; x < num_pixels; ++x) {

      float fga = clamp ? Clamp(fg.a[x]) : fg.a[x];

      const float new_a = 1.f - (1.f - fga) * (1.f - bg.a[x]);

      const float rnew_a = (new_a > 0 ? 1.f / new_a : 0.f);

      out.r[x] = (fg.r[x] * fga + bg.r[x] * bg.a[x] * (1.f - fga)) * rnew_a;

      out.g[x] = (fg.g[x] * fga + bg.g[x] * bg.a[x] * (1.f - fga)) * rnew_a;

      out.b[x] = (fg.b[x] * fga + bg.b[x] * bg.a[x] * (1.f - fga)) * rnew_a;

      out.a[x] = new_a;

    }

  }

}

void PerformAlphaBlending(const float* bg, const float* bga, const float* fg,

                          const float* fga, float* out, size_t num_pixels,

                          bool alpha_is_premultiplied, bool clamp) {

  if (bg == bga && fg == fga) {

    for (size_t x = 0; x < num_pixels; ++x) {

      float fa = clamp ? Clamp(fga[x]) : fga[x];

      out[x] = (1.f - (1.f - fa) * (1.f - bga[x]));

    }

  } else {

    if (alpha_is_premultiplied) {

      for (size_t x = 0; x < num_pixels; ++x) {

        float fa = clamp ? Clamp(fga[x]) : fga[x];

        out[x] = (fg[x] + bg[x] * (1.f - fa));

      }

    } else {

      for (size_t x = 0; x < num_pixels; ++x) {

        float fa = clamp ? Clamp(fga[x]) : fga[x];

        const float new_a = 1.f - (1.f - fa) * (1.f - bga[x]);

        const float rnew_a = (new_a > 0 ? 1.f / new_a : 0.f);

        out[x] = (fg[x] * fa + bg[x] * bga[x] * (1.f - fa)) * rnew_a;

      }

    }

  }

}

void PerformAlphaWeightedAdd(const float* bg, const float* fg, const float* fga,

                             float* out, size_t num_pixels, bool clamp) {

  if (fg == fga) {

    memcpy(out, bg, num_pixels * sizeof(*out));

  } else if (clamp) {

    for (size_t x = 0; x < num_pixels; ++x) {

      out[x] = bg[x] + fg[x] * Clamp(fga[x]);

    }

  } else {

    for (size_t x = 0; x < num_pixels; ++x) {

      out[x] = bg[x] + fg[x] * fga[x];

    }

  }

}

void PerformMulBlending(const float* bg, const float* fg, float* out,

                        size_t num_pixels, bool clamp) {

  if (clamp) {

    for (size_t x = 0; x < num_pixels; ++x) {

      out[x] = bg[x] * Clamp(fg[x]);

    }

  } else {

    for (size_t x = 0; x < num_pixels; ++x) {

      out[x] = bg[x] * fg[x];

    }

  }

}

void PremultiplyAlpha(float* JXL_RESTRICT r, float* JXL_RESTRICT g,

                      float* JXL_RESTRICT b, const float* JXL_RESTRICT a,

                      size_t num_pixels) {

  for (size_t x = 0; x < num_pixels; ++x) {

    const float multiplier = std::max(kSmallAlpha, a[x]);

    r[x] *= multiplier;

    g[x] *= multiplier;

    b[x] *= multiplier;

  }

}

void UnpremultiplyAlpha(float* JXL_RESTRICT r, float* JXL_RESTRICT g,

                        float* JXL_RESTRICT b, const float* JXL_RESTRICT a,

                        size_t num_pixels) {

  for (size_t x = 0; x < num_pixels; ++x) {

    const float multiplier = 1.f / std::max(kSmallAlpha, a[x]);

    r[x] *= multiplier;

    g[x] *= multiplier;

    b[x] *= multiplier;

  }

}

}  // namespace jxl