// Copyright 2018 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.

#include "dawn/native/vulkan/QueueVk.h"

#include "dawn/common/Math.h"
#include "dawn/native/Buffer.h"
#include "dawn/native/CommandValidation.h"
#include "dawn/native/Commands.h"
#include "dawn/native/DynamicUploader.h"
#include "dawn/native/vulkan/CommandBufferVk.h"
#include "dawn/native/vulkan/CommandRecordingContext.h"
#include "dawn/native/vulkan/DeviceVk.h"
#include "dawn/native/vulkan/UtilsVulkan.h"
#include "dawn/platform/DawnPlatform.h"
#include "dawn/platform/tracing/TraceEvent.h"

namespace dawn::native::vulkan {

    // static
    Ref<Queue> Queue::Create(Device* device, const QueueDescriptor* descriptor) {
        Ref<Queue> queue = AcquireRef(new Queue(device, descriptor));
        queue->Initialize();
        return queue;
    }

    Queue::Queue(Device* device, const QueueDescriptor* descriptor)
        : QueueBase(device, descriptor) {
    }

    Queue::~Queue() {
    }

    void Queue::Initialize() {
        SetLabelImpl();
    }

    MaybeError Queue::SubmitImpl(uint32_t commandCount, CommandBufferBase* const* commands) {
        Device* device = ToBackend(GetDevice());

        DAWN_TRY(device->Tick());

        TRACE_EVENT_BEGIN0(GetDevice()->GetPlatform(), Recording,
                           "CommandBufferVk::RecordCommands");
        CommandRecordingContext* recordingContext = device->GetPendingRecordingContext();
        for (uint32_t i = 0; i < commandCount; ++i) {
            DAWN_TRY(ToBackend(commands[i])->RecordCommands(recordingContext));
        }
        TRACE_EVENT_END0(GetDevice()->GetPlatform(), Recording, "CommandBufferVk::RecordCommands");

        DAWN_TRY(device->SubmitPendingCommands());

        return {};
    }

    void Queue::SetLabelImpl() {
        Device* device = ToBackend(GetDevice());
        // VKDevice reinterpret_casts to a uint64_t rather than a uint64_t& like most other types
        // because it's a dispatchable handle, and thus doesn't have the VkHandle wrapper that
        // Dawn creates for anything defined with VK_DEFINE_NON_DISPATCHABLE_HANDLE.

        // TODO(crbug.com/dawn/1344): When we start using multiple queues this needs to be adjusted
        // so it doesn't always change the default queue's label.
        SetDebugName(device, VK_OBJECT_TYPE_QUEUE, reinterpret_cast<uint64_t>(device->GetQueue()),
                     "Dawn_Queue", GetLabel());
    }

}  // namespace dawn::native::vulkan