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

#ifndef SERVICES_VIZ_PUBLIC_CPP_COMPOSITING_TRANSFERABLE_RESOURCE_MOJOM_TRAITS_H_
#define SERVICES_VIZ_PUBLIC_CPP_COMPOSITING_TRANSFERABLE_RESOURCE_MOJOM_TRAITS_H_

#include <optional>

#include "base/types/expected.h"
#include "build/build_config.h"
#include "components/viz/common/resources/resource_id.h"
#include "components/viz/common/resources/transferable_resource.h"
#include "mojo/public/cpp/bindings/deserialization_error.h"
#include "services/viz/public/mojom/compositing/transferable_resource.mojom-shared.h"
#include "skia/public/mojom/image_info_mojom_traits.h"

#if BUILDFLAG(IS_ANDROID)
#include "gpu/ipc/common/vulkan_ycbcr_info_mojom_traits.h"
#include "gpu/vulkan/vulkan_ycbcr_info.h"
#endif

namespace mojo {

template <>
struct StructTraits<viz::mojom::MetadataOverrideDataView,
                    viz::TransferableResource::MetadataOverride> {
  static const std::optional<bool>& is_overlay_candidate(
      const viz::TransferableResource::MetadataOverride& input) {
    return input.is_overlay_candidate;
  }

  static const std::optional<gfx::ColorSpace>& color_space(
      const viz::TransferableResource::MetadataOverride& input) {
    return input.color_space;
  }

  static const std::optional<SkAlphaType>& alpha_type(
      const viz::TransferableResource::MetadataOverride& input) {
    return input.alpha_type;
  }

  static base::expected<void, DeserializationError> Read(
      viz::mojom::MetadataOverrideDataView data,
      viz::TransferableResource::MetadataOverride* out);
};

template <>
struct EnumTraits<viz::mojom::SynchronizationType,
                  viz::TransferableResource::SynchronizationType> {
  static viz::mojom::SynchronizationType ToMojom(
      viz::TransferableResource::SynchronizationType type);

  static viz::TransferableResource::SynchronizationType FromMojom(
      viz::mojom::SynchronizationType input);
};

template <>
struct EnumTraits<viz::mojom::ResourceSource,
                  viz::TransferableResource::ResourceSource> {
  static viz::mojom::ResourceSource ToMojom(
      viz::TransferableResource::ResourceSource source);

  static viz::TransferableResource::ResourceSource FromMojom(
      viz::mojom::ResourceSource input);
};

template <>
struct StructTraits<viz::mojom::TransferableResourceDataView,
                    viz::TransferableResource> {
  static const viz::ResourceId& id(const viz::TransferableResource& resource) {
    return resource.id;
  }

  static gpu::ExportedSharedImage shared_image(
      const viz::TransferableResource& resource) {
    return resource.shared_image()->Export();
  }

  static const gpu::SyncToken& sync_token(
      const viz::TransferableResource& resource) {
    return resource.sync_token();
  }

  static viz::TransferableResource::SynchronizationType synchronization_type(
      const viz::TransferableResource& resource) {
    return resource.synchronization_type;
  }

#if BUILDFLAG(IS_ANDROID)
  static bool is_backed_by_surface_view(
      const viz::TransferableResource& resource) {
    return resource.is_backed_by_surface_view;
  }
#endif

#if BUILDFLAG(IS_ANDROID) || BUILDFLAG(IS_WIN)
  static bool wants_promotion_hint(const viz::TransferableResource& resource) {
    return resource.wants_promotion_hint;
  }
#endif

  static const gfx::HDRMetadata& hdr_metadata(
      const viz::TransferableResource& resource) {
    return resource.hdr_metadata;
  }

  static bool needs_detiling(const viz::TransferableResource& resource) {
    return resource.needs_detiling;
  }

#if BUILDFLAG(IS_ANDROID)
  static const std::optional<gpu::VulkanYCbCrInfo>& ycbcr_info(
      const viz::TransferableResource& resource) {
    return resource.ycbcr_info;
  }
#endif

  static viz::TransferableResource::ResourceSource resource_source(
      const viz::TransferableResource& resource) {
    return resource.resource_source;
  }

  static viz::TransferableResource::MetadataOverride metadata_override(
      const viz::TransferableResource& resource) {
    return resource.metadata_override();
  }

  static base::expected<void, DeserializationError> Read(
      viz::mojom::TransferableResourceDataView data,
      viz::TransferableResource* out);
};

}  // namespace mojo

#endif  // SERVICES_VIZ_PUBLIC_CPP_COMPOSITING_TRANSFERABLE_RESOURCE_MOJOM_TRAITS_H_