1 #include <msp/core/algorithm.h>
10 Program::Program(const Module &mod, const map<string, int> &spec_values)
12 add_stages(mod, spec_values);
15 void Program::add_stages(const Module &mod, const map<string, int> &spec_values)
18 throw invalid_operation("Program::add_stages");
20 reflect_data = ReflectData();
22 TransientData transient;
23 switch(mod.get_format())
26 add_glsl_stages(static_cast<const GlslModule &>(mod), spec_values, transient);
29 add_spirv_stages(static_cast<const SpirVModule &>(mod), spec_values, transient);
32 throw invalid_argument("Program::add_stages");
37 if(mod.get_format()==Module::GLSL)
41 apply_bindings(transient);
43 else if(mod.get_format()==Module::SPIR_V)
45 collect_uniforms(static_cast<const SpirVModule &>(mod), transient.spec_values);
46 collect_attributes(static_cast<const SpirVModule &>(mod));
49 for(ReflectData::UniformBlockInfo &b: reflect_data.uniform_blocks)
50 if(!b.data_size && !b.uniforms.empty())
52 const ReflectData::UniformInfo &uni = *b.uniforms.back();
53 b.data_size = uni.location*16+uni.array_size*get_type_size(uni.type);
56 for(const ReflectData::UniformInfo &u: reflect_data.uniforms)
58 for(const ReflectData::AttributeInfo &a: reflect_data.attributes)
62 void Program::collect_uniforms(const SpirVModule &mod, const map<unsigned, int> &spec_values)
64 // Prepare the default block
65 reflect_data.uniform_blocks.push_back(ReflectData::UniformBlockInfo());
66 vector<vector<string> > block_uniform_names(1);
68 for(const SpirVModule::Variable &v: mod.get_variables())
70 if(v.storage==SpirVModule::UNIFORM && v.struct_type)
72 reflect_data.uniform_blocks.push_back(ReflectData::UniformBlockInfo());
73 ReflectData::UniformBlockInfo &info = reflect_data.uniform_blocks.back();
74 info.name = v.struct_type->name;
75 info.bind_point = v.binding;
76 info.data_size = v.struct_type->size;
80 prefix = v.struct_type->name+".";
81 block_uniform_names.push_back(vector<string>());
82 collect_block_uniforms(*v.struct_type, prefix, 0, spec_values, block_uniform_names.back());
84 else if(v.storage==SpirVModule::UNIFORM_CONSTANT && v.location>=0)
86 block_uniform_names[0].push_back(v.name);
87 reflect_data.uniforms.push_back(ReflectData::UniformInfo());
88 ReflectData::UniformInfo &info = reflect_data.uniforms.back();
91 info.location = v.location;
92 info.binding = v.binding;
93 info.array_size = v.array_size;
98 sort_member(reflect_data.uniforms, &ReflectData::UniformInfo::tag);
100 for(unsigned i=0; i<reflect_data.uniform_blocks.size(); ++i)
102 ReflectData::UniformBlockInfo &block = reflect_data.uniform_blocks[i];
103 for(const string &n: block_uniform_names[i])
105 // The element is already known to be present
106 ReflectData::UniformInfo &uni = *lower_bound_member(reflect_data.uniforms, Tag(n), &ReflectData::UniformInfo::tag);
107 block.uniforms.push_back(&uni);
110 block.sort_uniforms();
111 block.update_layout_hash();
114 reflect_data.update_layout_hash();
117 void Program::collect_block_uniforms(const SpirVModule::Structure &strct, const string &prefix, unsigned base_offset, const map<unsigned, int> &spec_values, vector<string> &uniform_names)
119 for(const SpirVModule::StructMember &m: strct.members)
121 unsigned offset = base_offset+m.offset;
124 unsigned array_size = m.array_size;
125 if(m.array_size_spec)
127 array_size = m.array_size_spec->i_value;
128 auto j = spec_values.find(m.array_size_spec->constant_id);
129 if(j!=spec_values.end())
130 array_size = j->second;
135 for(unsigned j=0; j<array_size; ++j, offset+=m.array_stride)
136 collect_block_uniforms(*m.struct_type, format("%s%s[%d].", prefix, m.name, j), offset, spec_values, uniform_names);
139 collect_block_uniforms(*m.struct_type, prefix+m.name+".", offset, spec_values, uniform_names);
143 string name = prefix+m.name;
144 uniform_names.push_back(name);
145 reflect_data.uniforms.push_back(ReflectData::UniformInfo());
146 ReflectData::UniformInfo &info = reflect_data.uniforms.back();
149 info.offset = offset;
150 info.array_size = m.array_size;
151 info.array_stride = m.array_stride;
152 info.matrix_stride = m.matrix_stride;
158 void Program::collect_attributes(const SpirVModule &mod)
160 for(const SpirVModule::EntryPoint &e: mod.get_entry_points())
161 if(e.stage==SpirVModule::VERTEX && e.name=="main")
163 for(const SpirVModule::Variable *v: e.globals)
164 if(v->storage==SpirVModule::INPUT)
166 reflect_data.attributes.push_back(ReflectData::AttributeInfo());
167 ReflectData::AttributeInfo &info = reflect_data.attributes.back();
169 info.location = v->location;
170 info.array_size = v->array_size;
176 void Program::collect_builtins(const SpirVModule &mod)
178 for(const SpirVModule::Variable &v: mod.get_variables())
179 if(v.storage==SpirVModule::OUTPUT && v.struct_type)
180 collect_builtins(*v.struct_type);
183 void Program::collect_builtins(const SpirVModule::Structure &strct)
185 for(const SpirVModule::StructMember &m: strct.members)
186 if(m.builtin==SpirVModule::CLIP_DISTANCE)
187 reflect_data.n_clip_distances = m.array_size;
190 const ReflectData::UniformBlockInfo &Program::get_uniform_block_info(const string &name) const
192 auto i = find_member(reflect_data.uniform_blocks, name, &ReflectData::UniformBlockInfo::name);
193 if(i==reflect_data.uniform_blocks.end())
194 throw key_error(name);
198 const ReflectData::UniformInfo &Program::get_uniform_info(const string &name) const
200 auto i = lower_bound_member(reflect_data.uniforms, Tag(name), &ReflectData::UniformInfo::tag);
201 if(i==reflect_data.uniforms.end() || i->name!=name)
202 throw key_error(name);
206 const ReflectData::UniformInfo &Program::get_uniform_info(Tag tag) const
208 auto i = lower_bound_member(reflect_data.uniforms, tag, &ReflectData::UniformInfo::tag);
209 if(i==reflect_data.uniforms.end() || i->tag!=tag)
210 throw key_error(tag);
214 int Program::get_uniform_location(const string &name) const
216 if(name[name.size()-1]==']')
217 throw invalid_argument("Program::get_uniform_location");
219 auto i = lower_bound_member(reflect_data.uniforms, Tag(name), &ReflectData::UniformInfo::tag);
220 return i!=reflect_data.uniforms.end() && i->name==name && i->block->bind_point<0 ? i->location : -1;
223 int Program::get_uniform_location(Tag tag) const
225 auto i = lower_bound_member(reflect_data.uniforms, tag, &ReflectData::UniformInfo::tag);
226 return i!=reflect_data.uniforms.end() && i->tag==tag && i->block->bind_point<0 ? i->location : -1;
229 int Program::get_uniform_binding(Tag tag) const
231 auto i = lower_bound_member(reflect_data.uniforms, tag, &ReflectData::UniformInfo::tag);
232 return i!=reflect_data.uniforms.end() && i->tag==tag ? i->binding : -1;
235 const ReflectData::AttributeInfo &Program::get_attribute_info(const string &name) const
237 auto i = lower_bound_member(reflect_data.attributes, name, &ReflectData::AttributeInfo::name);
238 if(i==reflect_data.attributes.end() || i->name!=name)
239 throw key_error(name);
243 int Program::get_attribute_location(const string &name) const
245 if(name[name.size()-1]==']')
246 throw invalid_argument("Program::get_attribute_location");
248 auto i = lower_bound_member(reflect_data.attributes, name, &ReflectData::AttributeInfo::name);
249 return i!=reflect_data.attributes.end() && i->name==name ? i->location : -1;
253 Program::Loader::Loader(Program &p, Collection &c):
254 DataFile::CollectionObjectLoader<Program>(p, &c)
256 add("module", &Loader::module);
259 void Program::Loader::module(const string &n)
261 map<string, int> spec_values;
262 SpecializationLoader ldr(spec_values);
264 obj.add_stages(get_collection().get<Module>(n), spec_values);
268 DataFile::Loader::ActionMap Program::SpecializationLoader::shared_actions;
270 Program::SpecializationLoader::SpecializationLoader(map<string, int> &sv):
273 set_actions(shared_actions);
276 void Program::SpecializationLoader::init_actions()
278 add("specialize", &SpecializationLoader::specialize_bool);
279 add("specialize", &SpecializationLoader::specialize_int);
282 void Program::SpecializationLoader::specialize_bool(const string &name, bool value)
284 spec_values[name] = value;
287 void Program::SpecializationLoader::specialize_int(const string &name, int value)
289 spec_values[name] = value;