//==---------------- event_info.hpp - SYCL event ---------------------------==//
//
// 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/event_impl.hpp>
#include <detail/ur_info_code.hpp>
#include <sycl/detail/common.hpp>
#include <sycl/detail/info_desc_helpers.hpp>
#include <sycl/info/info_desc.hpp>

namespace sycl {
inline namespace _V1 {
namespace detail {

template <typename Param>
typename Param::return_type
get_event_profiling_info(ur_event_handle_t Event, const AdapterPtr &Adapter) {
  static_assert(is_event_profiling_info_desc<Param>::value,
                "Unexpected event profiling info descriptor");
  typename Param::return_type Result{0};
  // TODO catch an exception and put it to list of asynchronous exceptions
  Adapter->call<UrApiKind::urEventGetProfilingInfo>(
      Event, UrInfoCode<Param>::value, sizeof(Result), &Result, nullptr);
  return Result;
}

template <typename Param>
typename Param::return_type get_event_info(ur_event_handle_t Event,
                                           const AdapterPtr &Adapter) {
  static_assert(is_event_info_desc<Param>::value,
                "Unexpected event info descriptor");
  typename Param::return_type Result{0};
  // TODO catch an exception and put it to list of asynchronous exceptions
  Adapter->call<UrApiKind::urEventGetInfo>(Event, UrInfoCode<Param>::value,
                                           sizeof(Result), &Result, nullptr);

  // If the status is UR_EVENT_STATUS_QUEUED We need to change it since QUEUE is
  // not a valid status in sycl.
  if constexpr (std::is_same<Param,
                             info::event::command_execution_status>::value) {
    Result = static_cast<ur_event_status_t>(Result) == UR_EVENT_STATUS_QUEUED
                 ? sycl::info::event_command_status::submitted
                 : Result;
  }

  return Result;
}

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