1 // Copyright 2018 The SwiftShader Authors. All Rights Reserved.
2 //
3 // Licensed under the Apache License, Version 2.0 (the "License");
4 // you may not use this file except in compliance with the License.
5 // You may obtain a copy of the License at
6 //
7 //    http://www.apache.org/licenses/LICENSE-2.0
8 //
9 // Unless required by applicable law or agreed to in writing, software
10 // distributed under the License is distributed on an "AS IS" BASIS,
11 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 // See the License for the specific language governing permissions and
13 // limitations under the License.
14 
15 #ifndef VK_SHADER_MODULE_HPP_
16 #define VK_SHADER_MODULE_HPP_
17 
18 #include "VkObject.hpp"
19 #include <vector>
20 
21 namespace rr
22 {
23 	class Routine;
24 }
25 
26 namespace vk
27 {
28 
29 class ShaderModule : public Object<ShaderModule, VkShaderModule>
30 {
31 public:
32 	ShaderModule(const VkShaderModuleCreateInfo* pCreateInfo, void* mem);
33 	~ShaderModule() = delete;
34 	void destroy(const VkAllocationCallbacks* pAllocator);
35 
36 	static size_t ComputeRequiredAllocationSize(const VkShaderModuleCreateInfo* pCreateInfo);
37 	// TODO: reconsider boundary of ShaderModule class; try to avoid 'expose the
38 	// guts' operations, and this copy.
getCode() const39 	std::vector<uint32_t> getCode() const { return std::vector<uint32_t>{ code, code + wordCount };}
40 
41 private:
42 	uint32_t* code = nullptr;
43 	uint32_t wordCount = 0;
44 };
45 
Cast(VkShaderModule object)46 static inline ShaderModule* Cast(VkShaderModule object)
47 {
48 	return reinterpret_cast<ShaderModule*>(object);
49 }
50 
51 } // namespace vk
52 
53 #endif // VK_SHADER_MODULE_HPP_
54