// Copyright 2024 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_WEBNN_PUBLIC_MOJOM_OPERAND_DESCRIPTOR_MOJOM_TRAITS_H_
#define SERVICES_WEBNN_PUBLIC_MOJOM_OPERAND_DESCRIPTOR_MOJOM_TRAITS_H_

#include "base/component_export.h"
#include "mojo/public/cpp/bindings/enum_traits.h"
#include "mojo/public/cpp/bindings/struct_traits.h"
#include "services/webnn/public/cpp/operand_descriptor.h"
#include "services/webnn/public/mojom/webnn_graph.mojom-shared.h"

namespace mojo {

template <>
struct COMPONENT_EXPORT(WEBNN_MOJOM_TRAITS)
    StructTraits<webnn::mojom::OperandDescriptorDataView,
                 webnn::OperandDescriptor> {
  static webnn::OperandDataType data_type(
      const webnn::OperandDescriptor& descriptor);
  static const std::vector<uint32_t>& shape(
      const webnn::OperandDescriptor& descriptor) {
    return descriptor.shape();
  }
  static const std::vector<uint32_t>& pending_permutation(
      const webnn::OperandDescriptor& descriptor) {
    return descriptor.pending_permutation();
  }

  static bool Read(webnn::mojom::OperandDescriptorDataView data,
                   webnn::OperandDescriptor* out);
};

template <>
struct COMPONENT_EXPORT(WEBNN_MOJOM_TRAITS)
    EnumTraits<webnn::mojom::DataType, webnn::OperandDataType> {
  static webnn::mojom::DataType ToMojom(webnn::OperandDataType input);

  static webnn::OperandDataType FromMojom(webnn::mojom::DataType input);
};

}  // namespace mojo

#endif  // SERVICES_WEBNN_PUBLIC_MOJOM_OPERAND_DESCRIPTOR_MOJOM_TRAITS_H_