// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef SKIA_PUBLIC_MOJOM_IMAGE_INFO_MOJOM_TRAITS_H_
#define SKIA_PUBLIC_MOJOM_IMAGE_INFO_MOJOM_TRAITS_H_

#include <optional>
#include <vector>

#include "base/component_export.h"
#include "skia/public/mojom/image_info.mojom-shared.h"
#include "third_party/skia/include/core/SkImageInfo.h"

namespace mojo {

template <>
struct COMPONENT_EXPORT(SKIA_SHARED_TRAITS)
    EnumTraits<skia::mojom::AlphaType, SkAlphaType> {
  static skia::mojom::AlphaType ToMojom(SkAlphaType type);
  static std::optional<SkAlphaType> FromMojom(skia::mojom::AlphaType in);
};

template <>
struct COMPONENT_EXPORT(SKIA_SHARED_TRAITS)
    EnumTraits<skia::mojom::ColorType, SkColorType> {
  static skia::mojom::ColorType ToMojom(SkColorType type);
  static std::optional<SkColorType> FromMojom(skia::mojom::ColorType in);
};

template <>
struct COMPONENT_EXPORT(SKIA_SHARED_TRAITS)
    StructTraits<skia::mojom::ImageInfoDataView, SkImageInfo> {
  static SkColorType color_type(const SkImageInfo& info) {
    return info.colorType();
  }
  static SkAlphaType alpha_type(const SkImageInfo& info) {
    return info.alphaType();
  }
  static uint32_t width(const SkImageInfo& info);
  static uint32_t height(const SkImageInfo& info);
  static std::optional<std::vector<float>> color_transfer_function(
      const SkImageInfo& info);
  static std::optional<std::vector<float>> color_to_xyz_matrix(
      const SkImageInfo& info);

  static bool Read(skia::mojom::ImageInfoDataView data, SkImageInfo* info);
};

template <>
struct COMPONENT_EXPORT(SKIA_SHARED_TRAITS)
    StructTraits<skia::mojom::BitmapN32ImageInfoDataView, SkImageInfo> {
  static SkAlphaType alpha_type(const SkImageInfo& info) {
    // BitmapN32ImageInfo only allows N32 SkImageInfos.
    CHECK_EQ(info.colorType(), kN32_SkColorType);
    return info.alphaType();
  }
  static uint32_t width(const SkImageInfo& info) {
    return ImageInfoStructTraits::width(info);
  }
  static uint32_t height(const SkImageInfo& info) {
    return ImageInfoStructTraits::height(info);
  }
  static std::optional<std::vector<float>> color_transfer_function(
      const SkImageInfo& info) {
    return ImageInfoStructTraits::color_transfer_function(info);
  }
  static std::optional<std::vector<float>> color_to_xyz_matrix(
      const SkImageInfo& info) {
    return ImageInfoStructTraits::color_to_xyz_matrix(info);
  }

  static bool Read(skia::mojom::BitmapN32ImageInfoDataView data,
                   SkImageInfo* info);

 private:
  using ImageInfoStructTraits =
      StructTraits<skia::mojom::ImageInfoDataView, SkImageInfo>;
};

}  // namespace mojo

#endif  // SKIA_PUBLIC_MOJOM_IMAGE_INFO_MOJOM_TRAITS_H_