support storage descriptors and tracked image aliases

src/graphics/shader/recompiler/ResourceMaterialization.cpp

@@ -91,6 +91,15 @@ bool ValidateResourceSnapshot(const Program& program, const ResourceSnapshot& sn
 			}
 		}
 	}
+	for (uint32_t i = 0; i < program.info.buffers.size(); i++) {
+		const auto alias = program.info.buffers[i].image_alias;
+		if (alias != BufferResource::NoImageAlias && alias >= program.info.images.size()) {
+			if (error != nullptr) {
+				*error = fmt::format("buffer resource {} has invalid image alias {}", i, alias);
+			}
+			return false;
+		}
+	}
 	const auto CheckWidth = [&](const auto& values, uint32_t width, const char* kind) {
 		for (uint32_t i = 0; i < values.size(); i++) {
 			if (values[i].dword_count != width) {

src/graphics/shader/recompiler/ResourceTracking.cpp

@@ -178,6 +178,7 @@ public:
 				}
 			}
 		}
+		LinkImageAliases();
 		m_program.info = std::move(m_info);
 		for (const auto& patch: m_patches) {
 			patch.inst->memory.resource        = patch.resource;
@@ -290,6 +291,24 @@ private:
 		resource->scalar = resource->scalar || inst.memory.kind == ResourceKind::ScalarBuffer;
 	}
 
+	void LinkImageAliases() {
+		for (auto& buffer: m_info.buffers) {
+			const auto* buffer_source = GetDescriptorSource(m_program, buffer.source);
+			if (buffer_source == nullptr || buffer_source->dword_count != 4) {
+				continue;
+			}
+			for (uint32_t i = 0; i < m_info.images.size(); i++) {
+				const auto* image_source = GetDescriptorSource(m_program, m_info.images[i].source);
+				if (image_source != nullptr && image_source->dword_count == 8 &&
+				    std::equal(buffer_source->dwords.begin(), buffer_source->dwords.begin() + 4,
+				               image_source->dwords.begin())) {
+					buffer.image_alias = i;
+					break;
+				}
+			}
+		}
+	}
+
 	uint32_t AddImage(const Instruction& inst) {
 		for (uint32_t i = 0; i < m_info.images.size(); i++) {
 			auto& image = m_info.images[i];

src/graphics/shader/recompiler/ShaderIR.h

@@ -575,11 +575,14 @@ struct SrtPlan {
 };
 
 struct BufferResource {
+	static constexpr uint32_t NoImageAlias = UINT32_MAX;
+
 	uint32_t source            = 0;
 	uint32_t first_use_pc      = 0;
 	uint32_t max_byte_extent   = 0;
 	uint32_t packed_stride     = 0;
 	uint32_t descriptor_format = 0;
+	uint32_t image_alias       = NoImageAlias;
 	bool     read              = false;
 	bool     written           = false;
 	bool     atomic            = false;