blob: aa45fc56e9a21ff02746eea9e55ae9761656c2b6 (
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
|
// vk-command-buffer.h
#pragma once
#include "vk-base.h"
#include "vk-command-encoder.h"
#include "vk-shader-object.h"
#include "vk-transient-heap.h"
namespace gfx
{
using namespace Slang;
namespace vk
{
class CommandBufferImpl : public ICommandBuffer, public ComObject
{
public:
// There are a pair of cyclic references between a `TransientResourceHeap` and
// a `CommandBuffer` created from the heap. We need to break the cycle when
// the public reference count of a command buffer drops to 0.
SLANG_COM_OBJECT_IUNKNOWN_ALL
ICommandBuffer* getInterface(const Guid& guid);
virtual void comFree() override;
public:
VkCommandBuffer m_commandBuffer;
VkCommandBuffer m_preCommandBuffer = VK_NULL_HANDLE;
VkCommandPool m_pool;
DeviceImpl* m_renderer;
BreakableReference<TransientResourceHeapImpl> m_transientHeap;
bool m_isPreCommandBufferEmpty = true;
RootShaderObjectImpl m_rootObject;
RefPtr<MutableRootShaderObjectImpl> m_mutableRootShaderObject;
RefPtr<ResourceCommandEncoder> m_resourceCommandEncoder;
RefPtr<ComputeCommandEncoder> m_computeCommandEncoder;
RefPtr<RenderCommandEncoder> m_renderCommandEncoder;
RefPtr<RayTracingCommandEncoder> m_rayTracingCommandEncoder;
// Command buffers are deallocated by its command pool,
// so no need to free individually.
~CommandBufferImpl() = default;
Result init(DeviceImpl* renderer, VkCommandPool pool, TransientResourceHeapImpl* transientHeap);
void beginCommandBuffer();
Result createPreCommandBuffer();
VkCommandBuffer getPreCommandBuffer();
public:
virtual SLANG_NO_THROW void SLANG_MCALL encodeRenderCommands(
IRenderPassLayout* renderPass,
IFramebuffer* framebuffer,
IRenderCommandEncoder** outEncoder) override;
virtual SLANG_NO_THROW void SLANG_MCALL
encodeComputeCommands(IComputeCommandEncoder** outEncoder) override;
virtual SLANG_NO_THROW void SLANG_MCALL
encodeResourceCommands(IResourceCommandEncoder** outEncoder) override;
virtual SLANG_NO_THROW void SLANG_MCALL
encodeRayTracingCommands(IRayTracingCommandEncoder** outEncoder) override;
virtual SLANG_NO_THROW void SLANG_MCALL close() override;
virtual SLANG_NO_THROW Result SLANG_MCALL getNativeHandle(InteropHandle* outHandle) override;
};
} // namespace vk
} // namespace gfx
|
