/*
 * Copyright (c) 2014-2021, NVIDIA CORPORATION.  All rights reserved.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *
 * SPDX-FileCopyrightText: Copyright (c) 2014-2021 NVIDIA CORPORATION
 * SPDX-License-Identifier: Apache-2.0
 */


#include "appwindowprofiler_vk.hpp"
#include "context_vk.hpp"
#include "error_vk.hpp"

#include <nvh/misc.hpp>
#include <nvh/nvprint.hpp>

#define GLFW_INCLUDE_NONE
#include <GLFW/glfw3.h>

#ifdef _WIN32
#define GLFW_EXPOSE_NATIVE_WIN32
#include <GLFW/glfw3native.h>
#include <vulkan/vulkan_win32.h>
#else
#define GLFW_EXPOSE_NATIVE_X11
#include <GLFW/glfw3native.h>
#include <xcb/xcb.h>

// Prevent clang format from "organizing" the includes.

#include <vulkan/vulkan_xcb.h>
#endif

namespace nvvk {

void AppWindowProfilerVK::contextInit()
{
  //m_contextWindow.init(&m_deviceInfo, this);
  ContextCreateInfo contextInfo = m_contextInfo;
  m_swapVsync                   = false;

  contextInfo.addInstanceExtension(VK_KHR_SURFACE_EXTENSION_NAME, false);
#ifdef _WIN32
  contextInfo.addInstanceExtension(VK_KHR_WIN32_SURFACE_EXTENSION_NAME, false);
#else
  contextInfo.addInstanceExtension(VK_KHR_XCB_SURFACE_EXTENSION_NAME, false);
#endif
  contextInfo.addDeviceExtension(VK_KHR_SWAPCHAIN_EXTENSION_NAME, false);

  if(!m_context.init(contextInfo))
  {
    LOGE("FATAL ERROR: failed to create Vulkan context\n");
    exit(-1);
    return;
  }

  // Construct the surface description:
  VkResult result;
#ifdef _WIN32
  HWND      hWnd      = glfwGetWin32Window(m_internal);
  HINSTANCE hInstance = GetModuleHandle(NULL);

  VkWin32SurfaceCreateInfoKHR createInfo = {};
  createInfo.sType                       = VK_STRUCTURE_TYPE_WIN32_SURFACE_CREATE_INFO_KHR;
  createInfo.pNext                       = NULL;
  createInfo.hinstance                   = hInstance;
  createInfo.hwnd                        = hWnd;
  result = vkCreateWin32SurfaceKHR(m_context.m_instance, &createInfo, nullptr, &m_surface);
#else  // _WIN32
  result = glfwCreateWindowSurface(m_context.m_instance, m_internal, NULL, &m_surface);
#endif  // _WIN32
  assert(result == VK_SUCCESS);

  m_context.setGCTQueueWithPresent(m_surface);

  m_swapChain.init(m_context.m_device, m_context.m_physicalDevice, m_context.m_queueGCT, m_context.m_queueGCT.familyIndex, m_surface);
  m_swapChain.update(getWidth(), getHeight(), m_swapVsync);
  m_windowState.m_swapSize[0] = m_swapChain.getWidth();
  m_windowState.m_swapSize[1] = m_swapChain.getHeight();

  m_profilerVK.init(m_context.m_device, m_context.m_physicalDevice);
  m_profilerVK.setLabelUsage(m_context.hasInstanceExtension(VK_EXT_DEBUG_UTILS_EXTENSION_NAME));
}

void AppWindowProfilerVK::contextDeinit()
{
  VkResult result = vkDeviceWaitIdle(m_context.m_device);
  if(nvvk::checkResult(result, __FILE__, __LINE__))
  {
    exit(-1);
  }
  m_profilerVK.deinit();
  m_swapChain.deinit();
  vkDestroySurfaceKHR(m_context.m_instance, m_surface, nullptr);
  m_context.deinit();
}

void AppWindowProfilerVK::contextSync()
{
  VkResult result = vkDeviceWaitIdle(m_context.m_device);
  if(nvvk::checkResult(result, __FILE__, __LINE__))
  {
    exit(-1);
  }
}

void AppWindowProfilerVK::swapResize(int width, int height)
{
  if((m_swapChain.getUpdateWidth() != width) || (m_swapChain.getUpdateHeight() != height))
  {
    m_swapChain.update(width, height, m_swapVsync);
    m_windowState.m_swapSize[0] = m_swapChain.getWidth();
    m_windowState.m_swapSize[1] = m_swapChain.getHeight();
  }
}
void AppWindowProfilerVK::swapPrepare()
{
  if(!m_swapChain.acquire())
  {
    LOGE("error: vulkan swapchain acqiure failed, try -vsync 1\n");
    exit(-1);
  }
}

void AppWindowProfilerVK::swapBuffers()
{
  m_swapChain.present(m_context.m_queueGCT);
}

void AppWindowProfilerVK::swapVsync(bool swapVsync)
{
  if(m_swapVsync != swapVsync)
  {
    m_swapChain.update(getWidth(), getHeight(), swapVsync);
    m_swapVsync = swapVsync;
  }
}

const char* AppWindowProfilerVK::contextGetDeviceName()
{
  return m_context.m_physicalInfo.properties10.deviceName;
}

}  // namespace nvvk