// Aseprite Render Library
// Copyright (c) 2020 Igara Studio S.A.
// Copyright (c) 2017 David Capello
//
// This file is released under the terms of the MIT license.
// Read LICENSE.txt for more information.

#ifndef RENDER_DITHERING_MATRIX_H_INCLUDED
#define RENDER_DITHERING_MATRIX_H_INCLUDED
#pragma once

#include <algorithm>
#include <vector>

namespace render {

class DitheringMatrix {
public:
  DitheringMatrix() : m_rows(1), m_cols(1), m_matrix(1, 1), m_maxValue(1) {}

  DitheringMatrix(int rows, int cols)
    : m_rows(rows)
    , m_cols(cols)
    , m_matrix(rows * cols, 0)
    , m_maxValue(1)
  {
  }

  int rows() const { return m_rows; }
  int cols() const { return m_cols; }

  int maxValue() const { return m_maxValue; }
  void calcMaxValue()
  {
    m_maxValue = *std::max_element(m_matrix.begin(), m_matrix.end());
    m_maxValue = std::max(m_maxValue, 1);
  }

  int operator()(int i, int j) const { return m_matrix[(i % m_rows) * m_cols + (j % m_cols)]; }

  int& operator()(int i, int j) { return m_matrix[(i % m_rows) * m_cols + (j % m_cols)]; }

private:
  int m_rows, m_cols;
  std::vector<int> m_matrix;
  int m_maxValue;
};

// Creates a Bayer dither matrix.
class BayerMatrix : public DitheringMatrix {
  static int D2[4];

public:
  BayerMatrix(int n) : DitheringMatrix(n, n)
  {
    for (int i = 0; i < n; ++i)
      for (int j = 0; j < n; ++j)
        operator()(i, j) = Dn(i, j, n);

    calcMaxValue();
  }

private:
  int Dn(int i, int j, int n) const
  {
    ASSERT(i >= 0 && i < n);
    ASSERT(j >= 0 && j < n);

    if (n == 2)
      return D2[i * 2 + j];
    return +4 * Dn(i % (n / 2), j % (n / 2), n / 2) + Dn(i / (n / 2), j / (n / 2), 2);
  }
};

} // namespace render

#endif
