1 #include "arb_shader_objects.h"
7 #include "programdata.h"
9 #include "uniformblock.h"
17 ProgramData::ProgramData():
22 static Require _req(ARB_shader_objects);
25 // Blocks are intentionally left uncopied
26 ProgramData::ProgramData(const ProgramData &other):
27 uniforms(other.uniforms),
32 for(UniformMap::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
33 i->second = i->second->clone();
36 ProgramData::~ProgramData()
38 for(UniformMap::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
40 for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
41 delete i->second.block;
45 void ProgramData::uniform(const string &name, Uniform *uni)
47 UniformMap::iterator i = uniforms.find(name);
50 /* UniformBlock does not copy the uniforms, so existing blocks will be
51 left with stale pointers. This is not a problem as long as no one stores
52 pointers to the blocks and expects them to stay valid. */
55 changes = VALUES_CHANGED;
60 changes = KEYS_CHANGED;
64 void ProgramData::uniform(const string &name, int v)
66 uniform(name, new Uniform1i(v));
69 void ProgramData::uniform(const string &name, float v)
71 uniform(name, new Uniform1f(v));
74 void ProgramData::uniform(const string &name, float v0, float v1)
76 float va[2] = { v0, v1 };
80 void ProgramData::uniform2(const string &name, const float *v)
82 uniform(name, new Uniform2f(v));
85 void ProgramData::uniform(const string &name, float v0, float v1, float v2)
87 float va[3] = { v0, v1, v2 };
91 void ProgramData::uniform(const string &name, const Vector3 &v)
93 uniform(name, v.x, v.y, v.z);
96 void ProgramData::uniform3(const string &name, const float *v)
98 uniform(name, new Uniform3f(v));
101 void ProgramData::uniform(const string &name, float v0, float v1, float v2, float v3)
103 float va[4] = { v0, v1, v2, v3 };
107 void ProgramData::uniform(const string &name, const Vector4 &v)
109 uniform(name, v.x, v.y, v.z, v.w);
112 void ProgramData::uniform(const string &name, const Color &c)
114 uniform(name, c.r, c.g, c.b, c.a);
117 void ProgramData::uniform4(const string &name, const float *v)
119 uniform(name, new Uniform4f(v));
122 void ProgramData::uniform_matrix4(const string &name, const float *v)
124 uniform(name, new UniformMatrix4x4f(v));
127 void ProgramData::uniform_matrix4(const string &name, const Matrix &m)
130 copy(m.data(), m.data()+16, v);
131 uniform_matrix4(name, v);
134 void ProgramData::uniform1_array(const string &name, unsigned n, const float *v)
136 uniform(name, new UniformArray<Uniform1f>(n, v));
139 void ProgramData::uniform2_array(const string &name, unsigned n, const float *v)
141 uniform(name, new UniformArray<Uniform2f>(n, v));
144 void ProgramData::uniform3_array(const string &name, unsigned n, const float *v)
146 uniform(name, new UniformArray<Uniform3f>(n, v));
149 void ProgramData::uniform4_array(const string &name, unsigned n, const float *v)
151 uniform(name, new UniformArray<Uniform4f>(n, v));
154 void ProgramData::uniform_matrix4_array(const string &name, unsigned n, const float *v)
156 uniform(name, new UniformArray<UniformMatrix4x4f>(n, v));
159 void ProgramData::find_uniforms_for_block(Block &block, const Program::UniformBlockInfo &info) const
161 block.uniforms.clear();
162 for(vector<const Program::UniformInfo *>::const_iterator i=info.uniforms.begin(); i!=info.uniforms.end(); ++i)
164 // XXX individual array elements
165 UniformMap::const_iterator j = uniforms.find((*i)->name);
166 if(j!=uniforms.end())
167 block.uniforms[(*i)->location] = &j->second;
171 UniformBlock *ProgramData::create_block(const Program::UniformBlockInfo &info) const
173 UniformBlock *block = new UniformBlock(info.data_size);
176 buffer = new Buffer(UNIFORM_BUFFER);
177 buffer->set_usage(STREAM_DRAW);
179 block->use_buffer(buffer, last_block);
184 const UniformBlock *ProgramData::get_block(const Program &prog, const Program::UniformBlockInfo *info) const
188 for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
189 if(i->second.changes<changes)
190 i->second.changes = changes;
191 changes = NO_CHANGES;
194 unsigned layout_hash = (info ? info->layout_hash : prog.get_uniform_layout_hash());
196 map<unsigned, Block>::iterator i = blocks.find(layout_hash);
199 i = blocks.insert(BlockMap::value_type(layout_hash, Block())).first;
202 find_uniforms_for_block(i->second, *info);
204 if(!i->second.uniforms.empty())
206 i->second.block = create_block(*info);
207 i->second.changes = VALUES_CHANGED;
212 i->second.block = new UniformBlock;
213 i->second.changes = VALUES_CHANGED;
216 else if(info && i->second.changes==KEYS_CHANGED)
218 find_uniforms_for_block(i->second, *info);
219 if(!i->second.uniforms.empty())
222 i->second.block = create_block(*info);
223 i->second.changes = VALUES_CHANGED;
226 i->second.changes = NO_CHANGES;
232 UniformBlock &block = *i->second.block;
233 if(i->second.changes)
237 vector<const Program::UniformInfo *>::const_iterator j = info->uniforms.begin();
238 map<unsigned, const Uniform *const *>::const_iterator k = i->second.uniforms.begin();
239 while(j!=info->uniforms.end() && k!=i->second.uniforms.end())
241 if(k->first==(*j)->location)
243 block.attach(**j, **k->second);
251 for(UniformMap::const_iterator j=uniforms.begin(); j!=uniforms.end(); ++j)
253 int loc = prog.get_uniform_location(j->first);
255 block.attach(loc, *j->second);
258 i->second.changes = NO_CHANGES;
264 const UniformBlock *ProgramData::get_block(const Program &prog, const string &name) const
267 return get_block(prog, 0);
269 return get_block(prog, &prog.get_uniform_block_info(name));
272 void ProgramData::apply() const
274 const Program *prog = Program::current();
276 throw invalid_operation("ProgramData::apply");
278 const Program::UniformBlockMap &prog_blocks = prog->get_uniform_blocks();
279 if(!prog_blocks.empty())
281 typedef pair<const UniformBlock *, unsigned> ApplyBlock;
282 list<ApplyBlock> apply_blocks;
283 for(Program::UniformBlockMap::const_iterator i=prog_blocks.begin(); i!=prog_blocks.end(); ++i)
284 if(const UniformBlock *block = get_block(*prog, &i->second))
285 apply_blocks.push_back(make_pair(block, i->second.bind_point));
290 for(list<ApplyBlock>::const_iterator i=apply_blocks.begin(); i!=apply_blocks.end(); ++i)
291 i->first->apply(i->second);
294 if(const UniformBlock *block = get_block(*prog, 0))
299 ProgramData::Block::Block():
305 ProgramData::Loader::Loader(ProgramData &pd):
306 DataFile::ObjectLoader<ProgramData>(pd)
308 add("uniform1i", &Loader::uniform1i);
309 add("uniform1f", &Loader::uniform1f);
310 add("uniform2f", &Loader::uniform2f);
311 add("uniform3f", &Loader::uniform3f);
312 add("uniform4f", &Loader::uniform4f);
315 void ProgramData::Loader::uniform1i(const string &n, int v)
320 void ProgramData::Loader::uniform1f(const string &n, float v)
325 void ProgramData::Loader::uniform2f(const string &n, float v0, float v1)
327 obj.uniform(n, v0, v1);
330 void ProgramData::Loader::uniform3f(const string &n, float v0, float v1, float v2)
332 obj.uniform(n, v0, v1, v2);
335 void ProgramData::Loader::uniform4f(const string &n, float v0, float v1, float v2, float v3)
337 obj.uniform(n, v0, v1, v2, v3);