// Copyright 2019 The Dawn Authors
//
// 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.

// VulkanBackend.cpp: contains the definition of symbols exported by VulkanBackend.h so that they
// can be compiled twice: once export (shared library), once not exported (static library)

// Include vulkan_platform.h before VulkanBackend.h includes vulkan.h so that we use our version
// of the non-dispatchable handles.
#include "common/vulkan_platform.h"

#include "dawn_native/VulkanBackend.h"

#include "common/SwapChainUtils.h"
#include "dawn_native/vulkan/DeviceVk.h"
#include "dawn_native/vulkan/NativeSwapChainImplVk.h"
#include "dawn_native/vulkan/TextureVk.h"

namespace dawn_native { namespace vulkan {

    VkInstance GetInstance(WGPUDevice device) {
        Device* backendDevice = reinterpret_cast<Device*>(device);
        return backendDevice->GetVkInstance();
    }

    DAWN_NATIVE_EXPORT PFN_vkVoidFunction GetInstanceProcAddr(WGPUDevice device,
                                                              const char* pName) {
        Device* backendDevice = reinterpret_cast<Device*>(device);
        return (*backendDevice->fn.GetInstanceProcAddr)(backendDevice->GetVkInstance(), pName);
    }

    // Explicitly export this function because it uses the "native" type for surfaces while the
    // header as seen in this file uses the wrapped type.
    DAWN_NATIVE_EXPORT DawnSwapChainImplementation
    CreateNativeSwapChainImpl(WGPUDevice device, ::VkSurfaceKHR surfaceNative) {
        Device* backendDevice = reinterpret_cast<Device*>(device);
        VkSurfaceKHR surface = VkSurfaceKHR::CreateFromHandle(surfaceNative);

        DawnSwapChainImplementation impl;
        impl = CreateSwapChainImplementation(new NativeSwapChainImpl(backendDevice, surface));
        impl.textureUsage = WGPUTextureUsage_Present;

        return impl;
    }

    WGPUTextureFormat GetNativeSwapChainPreferredFormat(
        const DawnSwapChainImplementation* swapChain) {
        NativeSwapChainImpl* impl = reinterpret_cast<NativeSwapChainImpl*>(swapChain->userData);
        return static_cast<WGPUTextureFormat>(impl->GetPreferredFormat());
    }

#if defined(DAWN_PLATFORM_LINUX)
    ExternalImageDescriptorOpaqueFD::ExternalImageDescriptorOpaqueFD()
        : ExternalImageDescriptorFD(ExternalImageType::OpaqueFD) {
    }

    ExternalImageDescriptorDmaBuf::ExternalImageDescriptorDmaBuf()
        : ExternalImageDescriptorFD(ExternalImageType::DmaBuf) {
    }

    ExternalImageExportInfoOpaqueFD::ExternalImageExportInfoOpaqueFD()
        : ExternalImageExportInfoFD(ExternalImageType::OpaqueFD) {
    }

    ExternalImageExportInfoDmaBuf::ExternalImageExportInfoDmaBuf()
        : ExternalImageExportInfoFD(ExternalImageType::DmaBuf) {
    }
#endif  // DAWN_PLATFORM_LINUX

    WGPUTexture WrapVulkanImage(WGPUDevice cDevice, const ExternalImageDescriptorVk* descriptor) {
#if defined(DAWN_PLATFORM_LINUX)
        switch (descriptor->type) {
            case ExternalImageType::OpaqueFD:
            case ExternalImageType::DmaBuf: {
                const ExternalImageDescriptorFD* fdDescriptor =
                    static_cast<const ExternalImageDescriptorFD*>(descriptor);
                Device* device = reinterpret_cast<Device*>(cDevice);
                TextureBase* texture = device->CreateTextureWrappingVulkanImage(
                    fdDescriptor, fdDescriptor->memoryFD, fdDescriptor->waitFDs);
                return reinterpret_cast<WGPUTexture>(texture);
            }
            default:
                return nullptr;
        }
#else
        return nullptr;
#endif  // DAWN_PLATFORM_LINUX
    }

    bool ExportVulkanImage(WGPUTexture cTexture,
                           VkImageLayout desiredLayout,
                           ExternalImageExportInfoVk* info) {
        if (cTexture == nullptr) {
            return false;
        }
#if defined(DAWN_PLATFORM_LINUX)
        switch (info->type) {
            case ExternalImageType::OpaqueFD:
            case ExternalImageType::DmaBuf: {
                Texture* texture = reinterpret_cast<Texture*>(cTexture);
                Device* device = ToBackend(texture->GetDevice());
                ExternalImageExportInfoFD* fdInfo = static_cast<ExternalImageExportInfoFD*>(info);
                return device->SignalAndExportExternalTexture(texture, desiredLayout, fdInfo,
                                                              &fdInfo->semaphoreHandles);
            }
            default:
                return false;
        }
#else
        return false;
#endif  // DAWN_PLATFORM_LINUX
    }

}}  // namespace dawn_native::vulkan