1 /*
2 * Copyright 2024 Google LLC
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7 #include "include/core/SkString.h"
8 #include "include/gpu/graphite/vk/VulkanGraphiteTypes.h"
9 #include "include/gpu/vk/VulkanMutableTextureState.h"
10 #include "src/gpu/graphite/TextureInfoPriv.h"
11 #include "src/gpu/graphite/vk/VulkanGraphiteTypesPriv.h"
12 #include "src/gpu/vk/VulkanUtilsPriv.h"
13
14 #include <cstdint>
15
16 namespace skgpu::graphite {
17
18 class VulkanTextureInfoData final : public TextureInfoData {
19 public:
VulkanTextureInfoData(VulkanTextureSpec v)20 VulkanTextureInfoData(VulkanTextureSpec v) : fVkSpec(v) {}
21
22 #if defined(SK_DEBUG)
type() const23 skgpu::BackendApi type() const override { return skgpu::BackendApi::kVulkan; }
24 #endif
25
spec() const26 VulkanTextureSpec spec() const { return fVkSpec; }
27
28 private:
29 VulkanTextureSpec fVkSpec;
30
bytesPerPixel() const31 size_t bytesPerPixel() const override { return VkFormatBytesPerBlock(fVkSpec.fFormat); }
32
compressionType() const33 SkTextureCompressionType compressionType() const override {
34 return VkFormatToCompressionType(fVkSpec.fFormat);
35 }
36
isMemoryless() const37 bool isMemoryless() const override { return false; }
38
toString() const39 SkString toString() const override {
40 return SkStringPrintf("Vulkan(%s,", fVkSpec.toString().c_str());
41 }
42
toRPAttachmentString(uint32_t sampleCount) const43 SkString toRPAttachmentString(uint32_t sampleCount) const override {
44 return SkStringPrintf(
45 "Vulkan(f=%u,s=%u)", static_cast<unsigned int>(fVkSpec.fFormat), sampleCount);
46 }
47
copyTo(AnyTextureInfoData & dstData) const48 void copyTo(AnyTextureInfoData& dstData) const override {
49 // Don't assert that dstData has a Vulkan type() because it could be
50 // uninitialized and that assert would fail.
51 dstData.emplace<VulkanTextureInfoData>(fVkSpec);
52 }
53
equal(const TextureInfoData * that) const54 bool equal(const TextureInfoData* that) const override {
55 SkASSERT(!that || that->type() == skgpu::BackendApi::kVulkan);
56 if (auto otherVk = static_cast<const VulkanTextureInfoData*>(that)) {
57 return fVkSpec == otherVk->fVkSpec;
58 }
59 return false;
60 }
61
isCompatible(const TextureInfoData * that) const62 bool isCompatible(const TextureInfoData* that) const override {
63 SkASSERT(!that || that->type() == skgpu::BackendApi::kVulkan);
64 if (auto otherVk = static_cast<const VulkanTextureInfoData*>(that)) {
65 return fVkSpec.isCompatible(otherVk->fVkSpec);
66 }
67 return false;
68 }
69 };
70
get_and_cast_data(const TextureInfo & info)71 static const VulkanTextureInfoData* get_and_cast_data(const TextureInfo& info) {
72 auto data = TextureInfoPriv::GetData(info);
73 SkASSERT(!data || data->type() == skgpu::BackendApi::kVulkan);
74 return static_cast<const VulkanTextureInfoData*>(data);
75 }
76
77 namespace TextureInfos {
MakeVulkan(const VulkanTextureInfo & vkInfo)78 TextureInfo MakeVulkan(const VulkanTextureInfo& vkInfo) {
79 skgpu::Protected p = Protected::kNo;
80 if (vkInfo.fFlags & VK_IMAGE_CREATE_PROTECTED_BIT) {
81 p = Protected::kYes;
82 }
83 return TextureInfoPriv::Make(skgpu::BackendApi::kVulkan,
84 vkInfo.fSampleCount,
85 vkInfo.fMipmapped,
86 p,
87 VulkanTextureInfoData(vkInfo));
88 }
89
GetVulkanTextureInfo(const TextureInfo & info,VulkanTextureInfo * out)90 bool GetVulkanTextureInfo(const TextureInfo& info, VulkanTextureInfo* out) {
91 if (!info.isValid() || info.backend() != skgpu::BackendApi::kVulkan) {
92 return false;
93 }
94 SkASSERT(out);
95 const VulkanTextureInfoData* vkData = get_and_cast_data(info);
96 SkASSERT(vkData);
97 *out = VulkanTextureSpecToTextureInfo(vkData->spec(), info.numSamples(), info.mipmapped());
98 return true;
99 }
100
101 // This cannot return a const reference or we get a warning about returning
102 // a reference to a temporary local variable.
GetVulkanTextureSpec(const TextureInfo & info)103 VulkanTextureSpec GetVulkanTextureSpec(const TextureInfo& info) {
104 SkASSERT(info.isValid() && info.backend() == skgpu::BackendApi::kVulkan);
105 const VulkanTextureInfoData* vkData = get_and_cast_data(info);
106 SkASSERT(vkData);
107 return vkData->spec();
108 }
109
GetVkFormat(const TextureInfo & info)110 VkFormat GetVkFormat(const TextureInfo& info) {
111 SkASSERT(info.isValid() && info.backend() == skgpu::BackendApi::kVulkan);
112 const VulkanTextureInfoData* vkData = get_and_cast_data(info);
113 SkASSERT(vkData);
114 return vkData->spec().fFormat;
115 }
116
GetVkUsageFlags(const TextureInfo & info)117 VkImageUsageFlags GetVkUsageFlags(const TextureInfo& info) {
118 SkASSERT(info.isValid() && info.backend() == skgpu::BackendApi::kVulkan);
119 const VulkanTextureInfoData* vkData = get_and_cast_data(info);
120 SkASSERT(vkData);
121 return vkData->spec().fImageUsageFlags;
122 }
123
GetVulkanYcbcrConversionInfo(const TextureInfo & info)124 VulkanYcbcrConversionInfo GetVulkanYcbcrConversionInfo(const TextureInfo& info) {
125 SkASSERT(info.isValid() && info.backend() == skgpu::BackendApi::kVulkan);
126 const VulkanTextureInfoData* vkData = get_and_cast_data(info);
127 SkASSERT(vkData);
128 return vkData->spec().fYcbcrConversionInfo;
129 }
130
131 } // namespace TextureInfos
132
133 } // namespace skgpu::graphite
134