1 files changed, 105 insertions, 0 deletions
diff --git a/src/vulkan/descriptor.h b/src/vulkan/descriptor.h
new file mode 100644
index 0000000..ba702ff
--- /dev/null
+++ b/src/vulkan/descriptor.h
@@ -0,0 +1,105 @@
+// 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.
+
+#ifndef SRC_VULKAN_DESCRIPTOR_H_
+#define SRC_VULKAN_DESCRIPTOR_H_
+
+#include <memory>
+#include <vector>
+
+#include "amber/result.h"
+#include "amber/value.h"
+#include "amber/vulkan_header.h"
+#include "src/buffer.h"
+#include "src/engine.h"
+#include "src/vulkan/resource.h"
+
+namespace amber {
+namespace vulkan {
+
+class CommandBuffer;
+class Device;
+class BufferDescriptor;
+class BufferBackedDescriptor;
+class SamplerDescriptor;
+
+enum class DescriptorType : uint8_t {
+  kStorageBuffer = 0,
+  kStorageBufferDynamic,
+  kUniformBuffer,
+  kUniformBufferDynamic,
+  kStorageImage,
+  kSampledImage,
+  kCombinedImageSampler,
+  kUniformTexelBuffer,
+  kStorageTexelBuffer,
+  kSampler
+};
+
+class Descriptor {
+ public:
+  Descriptor(DescriptorType type,
+             Device* device,
+             uint32_t desc_set,
+             uint32_t binding);
+  virtual ~Descriptor();
+
+  virtual void UpdateDescriptorSetIfNeeded(VkDescriptorSet descriptor_set) = 0;
+  virtual Result CreateResourceIfNeeded() = 0;
+  virtual Result RecordCopyDataToResourceIfNeeded(CommandBuffer*) { return {}; }
+  virtual Result RecordCopyDataToHost(CommandBuffer*) { return {}; }
+  virtual Result MoveResourceToBufferOutput() { return {}; }
+  virtual Result SetSizeInElements(uint32_t) { return {}; }
+  virtual Result AddToBuffer(const std::vector<Value>&, uint32_t) { return {}; }
+  virtual uint32_t GetDescriptorCount() { return 1; }
+  virtual std::vector<uint32_t> GetDynamicOffsets() { return {}; }
+  virtual BufferDescriptor* AsBufferDescriptor() { return nullptr; }
+  virtual BufferBackedDescriptor* AsBufferBackedDescriptor() { return nullptr; }
+  virtual SamplerDescriptor* AsSamplerDescriptor() { return nullptr; }
+  uint32_t GetDescriptorSet() const { return descriptor_set_; }
+  uint32_t GetBinding() const { return binding_; }
+  VkDescriptorType GetVkDescriptorType() const;
+  DescriptorType GetDescriptorType() const { return type_; }
+
+  bool IsStorageBuffer() const {
+    return type_ == DescriptorType::kStorageBuffer;
+  }
+  bool IsStorageBufferDynamic() const {
+    return type_ == DescriptorType::kStorageBufferDynamic;
+  }
+  bool IsUniformBuffer() const {
+    return type_ == DescriptorType::kUniformBuffer;
+  }
+  bool IsUniformBufferDynamic() const {
+    return type_ == DescriptorType::kUniformBufferDynamic;
+  }
+  bool IsUniformTexelBuffer() const {
+    return type_ == DescriptorType::kUniformTexelBuffer;
+  }
+  bool IsStorageTexelBuffer() const {
+    return type_ == DescriptorType::kStorageTexelBuffer;
+  }
+
+ protected:
+  Device* device_ = nullptr;
+  DescriptorType type_ = DescriptorType::kStorageBuffer;
+  uint32_t descriptor_set_ = 0;
+  uint32_t binding_ = 0;
+  bool is_descriptor_set_update_needed_ = false;
+};
+
+}  // namespace vulkan
+}  // namespace amber
+
+#endif  // SRC_VULKAN_DESCRIPTOR_H_