// Copyright 2019 The Fuchsia Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SRC_UI_TESTING_VIEWS_COLOR_H_
#define SRC_UI_TESTING_VIEWS_COLOR_H_

#include <fuchsia/ui/scenic/cpp/fidl.h>

#include <map>
#include <ostream>
#include <tuple>

namespace scenic {

struct Color {
  // Constructor is idiomatic RGBA, but memory layout is native BGRA.
  constexpr Color(uint8_t r, uint8_t g, uint8_t b, uint8_t a) : b(b), g(g), r(r), a(a) {}

  uint8_t b;
  uint8_t g;
  uint8_t r;
  uint8_t a;
};

inline bool operator==(const Color& a, const Color& b) {
  return a.r == b.r && a.g == b.g && a.b == b.b && a.a == b.a;
}

inline bool operator<(const Color& a, const Color& b) {
  return std::tie(a.r, a.g, a.b, a.a) < std::tie(b.r, b.g, b.b, b.a);
}

// RGBA hex dump. Note that this differs from the internal BGRA memory layout.
std::ostream& operator<<(std::ostream& os, const Color& c);

class Screenshot {
 public:
  Screenshot(const fuchsia::ui::scenic::ScreenshotData& screenshot_data);

  size_t width() const { return width_; }
  size_t height() const { return height_; }
  bool empty() const { return width_ == 0 || height_ == 0; }

  // Notably the indexer behaves like a row-major matrix, whereas the iterator
  // behaves like a flat array.
  const Color* operator[](size_t row) const;

  // Coordinates are in the range [0, 1).
  const Color& ColorAt(float x, float y) const;

  // Notably the iterator behaves like a flat array, whereas the indexer behaves
  // like a row-major matrix.
  const Color* begin() const;
  const Color* end() const;

  // Counts the frequencies of each color in a screenshot.
  std::map<Color, size_t> Histogram() const;

 private:
  const size_t width_, height_;
  std::vector<uint8_t> data_;
};

}  // namespace scenic

#endif  // SRC_UI_TESTING_VIEWS_COLOR_H_