//==------ platform_info.hpp - SYCL platform info methods ------------------==//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#pragma once
#include <detail/adapter.hpp>
#include <detail/ur_info_code.hpp>
#include <sycl/detail/common.hpp>
#include <sycl/detail/info_desc_helpers.hpp>
#include <sycl/detail/ur.hpp>
#include <sycl/info/info_desc.hpp>

#include "split_string.hpp"

namespace sycl {
inline namespace _V1 {
namespace detail {

inline std::string get_platform_info_string_impl(ur_platform_handle_t Plt,
                                                 const AdapterPtr &Adapter,
                                                 ur_platform_info_t UrCode) {
  size_t ResultSize = 0;
  // TODO catch an exception and put it to list of asynchronous exceptions
  Adapter->call<UrApiKind::urPlatformGetInfo>(Plt, UrCode, 0, nullptr,
                                              &ResultSize);
  if (ResultSize == 0) {
    return "";
  }
  std::unique_ptr<char[]> Result(new char[ResultSize]);
  // TODO catch an exception and put it to list of asynchronous exceptions
  Adapter->call<UrApiKind::urPlatformGetInfo>(Plt, UrCode, ResultSize,
                                              Result.get(), nullptr);
  return Result.get();
}
// The platform information methods
template <typename Param>
typename std::enable_if<
    std::is_same<typename Param::return_type, std::string>::value,
    std::string>::type
get_platform_info(ur_platform_handle_t Plt, const AdapterPtr &Adapter) {
  static_assert(is_platform_info_desc<Param>::value,
                "Invalid platform information descriptor");
  return get_platform_info_string_impl(Plt, Adapter,
                                       detail::UrInfoCode<Param>::value);
}

template <typename Param>
typename std::enable_if<std::is_same<Param, info::platform::extensions>::value,
                        std::vector<std::string>>::type
get_platform_info(ur_platform_handle_t Plt, const AdapterPtr &Adapter) {
  static_assert(is_platform_info_desc<Param>::value,
                "Invalid platform information descriptor");
  std::string Result = get_platform_info_string_impl(
      Plt, Adapter, detail::UrInfoCode<info::platform::extensions>::value);
  return split_string(Result, ' ');
}

} // namespace detail
} // namespace _V1
} // namespace sycl