aboutsummaryrefslogtreecommitdiff
path: root/src/vulkan/index_buffer.cc
blob: 30b223d9bc5184c34326c25a3a9c54b6d84cb6bc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
// Copyright 2019 The Amber 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 "src/vulkan/index_buffer.h"

#include <cstring>

#include "src/make_unique.h"
#include "src/vulkan/command_buffer.h"
#include "src/vulkan/device.h"

namespace amber {
namespace vulkan {

IndexBuffer::IndexBuffer(Device* device) : device_(device) {}

IndexBuffer::~IndexBuffer() = default;

Result IndexBuffer::SendIndexData(CommandBuffer* command, Buffer* buffer) {
  if (transfer_buffer_) {
    return Result(
        "IndexBuffer::SendIndexData must be called once when it is created");
  }

  if (buffer->ElementCount() == 0)
    return Result("IndexBuffer::SendIndexData |buffer| is empty");

  if (buffer->GetFormat()->IsUint32())
    index_type_ = VK_INDEX_TYPE_UINT32;
  else if (buffer->GetFormat()->IsUint16())
    index_type_ = VK_INDEX_TYPE_UINT16;
  else if (buffer->GetFormat()->IsUint8())
    index_type_ = VK_INDEX_TYPE_UINT8_EXT;
  else
    return Result("IndexBuffer::SendIndexData unexpected index buffer format");

  transfer_buffer_ =
      MakeUnique<TransferBuffer>(device_, buffer->GetSizeInBytes(), nullptr);
  Result r = transfer_buffer_->AddUsageFlags(VK_BUFFER_USAGE_INDEX_BUFFER_BIT |
                                             VK_BUFFER_USAGE_TRANSFER_DST_BIT);
  if (!r.IsSuccess())
    return r;

  r = transfer_buffer_->Initialize();
  if (!r.IsSuccess())
    return r;

  std::memcpy(transfer_buffer_->HostAccessibleMemoryPtr(),
              buffer->ValuePtr()->data(), buffer->GetSizeInBytes());

  transfer_buffer_->CopyToDevice(command);
  return {};
}

Result IndexBuffer::BindToCommandBuffer(CommandBuffer* command) {
  if (!transfer_buffer_) {
    return Result(
        "IndexBuffer::BindToCommandBuffer |transfer_buffer_| is nullptr");
  }

  device_->GetPtrs()->vkCmdBindIndexBuffer(command->GetVkCommandBuffer(),
                                           transfer_buffer_->GetVkBuffer(), 0,
                                           index_type_);
  return {};
}

}  // namespace vulkan
}  // namespace amber