offload-test-suite/lib/API/DX/DXResources.h at 76635d2fd004a12f8a76fbd8a463be2bd0bdca50 · Traverse-Research/offload-test-suite · GitHub

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
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
//===- DXResources.h - DirectX Resource Helpers ---------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
//
//
//===----------------------------------------------------------------------===//

#ifndef OFFLOADTEST_API_DXRESOURCES_H
#define OFFLOADTEST_API_DXRESOURCES_H

#include "API/Device.h"

#include <d3d12.h>
#include <dxgiformat.h>

namespace offloadtest {

inline D3D12_HEAP_TYPE getDXHeapType(MemoryLocation Location) {
  switch (Location) {
  case MemoryLocation::GpuOnly:
    return D3D12_HEAP_TYPE_DEFAULT;
  case MemoryLocation::CpuToGpu:
    return D3D12_HEAP_TYPE_UPLOAD;
  case MemoryLocation::GpuToCpu:
    return D3D12_HEAP_TYPE_READBACK;
  }
  llvm_unreachable("All MemoryLocation cases handled");
}

inline DXGI_FORMAT getDXGIFormat(Format Fmt) {
  switch (Fmt) {
  case Format::R16Sint:
    return DXGI_FORMAT_R16_SINT;
  case Format::R16Uint:
    return DXGI_FORMAT_R16_UINT;
  case Format::RG16Sint:
    return DXGI_FORMAT_R16G16_SINT;
  case Format::RG16Uint:
    return DXGI_FORMAT_R16G16_UINT;
  case Format::RGBA16Sint:
    return DXGI_FORMAT_R16G16B16A16_SINT;
  case Format::RGBA16Uint:
    return DXGI_FORMAT_R16G16B16A16_UINT;
  case Format::R32Sint:
    return DXGI_FORMAT_R32_SINT;
  case Format::R32Uint:
    return DXGI_FORMAT_R32_UINT;
  case Format::R32Float:
    return DXGI_FORMAT_R32_FLOAT;
  case Format::RG32Sint:
    return DXGI_FORMAT_R32G32_SINT;
  case Format::RG32Uint:
    return DXGI_FORMAT_R32G32_UINT;
  case Format::RG32Float:
    return DXGI_FORMAT_R32G32_FLOAT;
  case Format::RGB32Float:
    return DXGI_FORMAT_R32G32B32_FLOAT;
  case Format::RGBA32Sint:
    return DXGI_FORMAT_R32G32B32A32_SINT;
  case Format::RGBA32Uint:
    return DXGI_FORMAT_R32G32B32A32_UINT;
  case Format::RGBA32Float:
    return DXGI_FORMAT_R32G32B32A32_FLOAT;
  case Format::R64Uint:
    return DXGI_FORMAT_R32G32_UINT;
  case Format::R64Sint:
    return DXGI_FORMAT_R32G32_SINT;
  case Format::RG64Uint:
    return DXGI_FORMAT_R32G32B32A32_UINT;
  case Format::RG64Sint:
    return DXGI_FORMAT_R32G32B32A32_SINT;
  case Format::D32Float:
    return DXGI_FORMAT_D32_FLOAT;
  case Format::D32FloatS8Uint:
    return DXGI_FORMAT_D32_FLOAT_S8X24_UINT;
  }
  llvm_unreachable("All Format cases handled");
}

inline DXGI_FORMAT getDXGIFormatSRV(Format Fmt) {
  switch (Fmt) {
  case Format::D32Float:
    return DXGI_FORMAT_R32_FLOAT;
  case Format::D32FloatS8Uint:
    return DXGI_FORMAT_R32_FLOAT_X8X24_TYPELESS;
  default:
    return getDXGIFormat(Fmt);
  }
}

inline D3D12_RESOURCE_FLAGS getDXResourceFlags(TextureUsage Usage) {
  D3D12_RESOURCE_FLAGS Flags = D3D12_RESOURCE_FLAG_NONE;
  if ((Usage & TextureUsage::Storage) != 0)
    Flags |= D3D12_RESOURCE_FLAG_ALLOW_UNORDERED_ACCESS;
  if ((Usage & TextureUsage::RenderTarget) != 0)
    Flags |= D3D12_RESOURCE_FLAG_ALLOW_RENDER_TARGET;
  if ((Usage & TextureUsage::DepthStencil) != 0)
    Flags |= D3D12_RESOURCE_FLAG_ALLOW_DEPTH_STENCIL;
  return Flags;
}

inline DXGI_FORMAT getDXGIIndexFormat(IndexFormat Fmt) {
  switch (Fmt) {
  case IndexFormat::Uint16:
    return DXGI_FORMAT_R16_UINT;
  case IndexFormat::Uint32:
    return DXGI_FORMAT_R32_UINT;
  }
  llvm_unreachable("All IndexFormat cases handled");
}

} // namespace offloadtest

#endif // OFFLOADTEST_API_DXRESOURCES_H