X-Git-Url: http://git.tdb.fi/?a=blobdiff_plain;f=source%2Fprogramdata.cpp;h=1287914005055ff91bb2102260daf4b28d50e710;hb=fa5b232a16e11d7950e80764497f0167ec9e3b41;hp=43767bf9526a4449cde0a6d5d3f076bba0460b4b;hpb=f9c15dc04462b2f1eea1d6bdd71e3ba967b1bd8c;p=libs%2Fgl.git

diff --git a/source/programdata.cpp b/source/programdata.cpp
index 43767bf9..12879140 100644
--- a/source/programdata.cpp
+++ b/source/programdata.cpp
@@ -1,6 +1,7 @@
+#include <msp/gl/extensions/arb_shader_objects.h>
+#include "buffer.h"
 #include "color.h"
 #include "error.h"
-#include "extension.h"
 #include "matrix.h"
 #include "program.h"
 #include "programdata.h"
@@ -14,26 +15,49 @@ namespace Msp {
 namespace GL {
 
 ProgramData::ProgramData():
-	modified(false)
+	last_block(0),
+	buffer(0),
+	changes(NO_CHANGES)
 {
-	static RequireExtension _ext("GL_ARB_shader_objects");
+	static Require _req(ARB_shader_objects);
 }
 
 // Blocks are intentionally left uncopied
 ProgramData::ProgramData(const ProgramData &other):
 	uniforms(other.uniforms),
-	modified(false)
+	last_block(0),
+	buffer(0),
+	changes(NO_CHANGES)
 {
 	for(UniformMap::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
 		i->second = i->second->clone();
 }
 
+ProgramData &ProgramData::operator=(const ProgramData &other)
+{
+	for(UniformMap::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
+		delete i->second;
+	uniforms.clear();
+
+	for(UniformMap::const_iterator i=other.uniforms.begin(); i!=other.uniforms.end(); ++i)
+		uniforms.insert(uniforms.end(), UniformMap::value_type(i->first, i->second->clone()));
+
+	for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
+		delete i->second.block;
+	blocks.clear();
+
+	changes = NO_CHANGES;
+
+	return *this;
+}
+
 ProgramData::~ProgramData()
 {
 	for(UniformMap::iterator i=uniforms.begin(); i!=uniforms.end(); ++i)
 		delete i->second;
 	for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
 		delete i->second.block;
+	delete buffer;
 }
 
 void ProgramData::uniform(const string &name, Uniform *uni)
@@ -41,13 +65,18 @@ void ProgramData::uniform(const string &name, Uniform *uni)
 	UniformMap::iterator i = uniforms.find(name);
 	if(i!=uniforms.end())
 	{
+		/* UniformBlock does not copy the uniforms, so existing blocks will be
+		left with stale pointers.  This is not a problem as long as no one stores
+		pointers to the blocks and expects them to stay valid. */
 		delete i->second;
 		i->second = uni;
+		changes = VALUES_CHANGED;
 	}
 	else
+	{
 		uniforms[name] = uni;
-
-	modified = true;
+		changes = KEYS_CHANGED;
+	}
 }
 
 void ProgramData::uniform(const string &name, int v)
@@ -108,49 +137,162 @@ void ProgramData::uniform4(const string &name, const float *v)
 	uniform(name, new Uniform4f(v));
 }
 
+void ProgramData::uniform_matrix2(const string &name, const float *v)
+{
+	uniform(name, new UniformMatrix2x2f(v));
+}
+
+void ProgramData::uniform_matrix3(const string &name, const float *v)
+{
+	uniform(name, new UniformMatrix3x3f(v));
+}
+
+void ProgramData::uniform(const string &name, const Matrix &m)
+{
+	uniform_matrix4(name, m.data());
+}
+
 void ProgramData::uniform_matrix4(const string &name, const float *v)
 {
 	uniform(name, new UniformMatrix4x4f(v));
 }
 
-void ProgramData::uniform_matrix4(const string &name, const Matrix &m)
+void ProgramData::uniform1_array(const string &name, unsigned n, const float *v)
+{
+	uniform(name, new UniformArray<Uniform1f>(n, v));
+}
+
+void ProgramData::uniform2_array(const string &name, unsigned n, const float *v)
+{
+	uniform(name, new UniformArray<Uniform2f>(n, v));
+}
+
+void ProgramData::uniform3_array(const string &name, unsigned n, const float *v)
+{
+	uniform(name, new UniformArray<Uniform3f>(n, v));
+}
+
+void ProgramData::uniform4_array(const string &name, unsigned n, const float *v)
+{
+	uniform(name, new UniformArray<Uniform4f>(n, v));
+}
+
+void ProgramData::uniform_matrix4_array(const string &name, unsigned n, const float *v)
 {
-	float v[16];
-	copy(m.data(), m.data()+16, v);
-	uniform_matrix4(name, v);
+	uniform(name, new UniformArray<UniformMatrix4x4f>(n, v));
 }
 
-const UniformBlock &ProgramData::get_block(const Program &prog) const
+void ProgramData::find_uniforms_for_block(Block &block, const Program::UniformBlockInfo &info) const
 {
-	if(modified)
+	block.uniforms.clear();
+	for(vector<const Program::UniformInfo *>::const_iterator i=info.uniforms.begin(); i!=info.uniforms.end(); ++i)
+	{
+		// XXX individual array elements
+		UniformMap::const_iterator j = uniforms.find((*i)->name);
+		if(j!=uniforms.end())
+			block.uniforms[(*i)->location] = &j->second;
+	}
+}
+
+UniformBlock *ProgramData::create_block(const Program::UniformBlockInfo &info) const
+{
+	UniformBlock *block = new UniformBlock(info.data_size);
+	if(!buffer)
+	{
+		buffer = new Buffer(UNIFORM_BUFFER);
+		buffer->set_usage(STREAM_DRAW);
+	}
+	block->use_buffer(buffer, last_block);
+	last_block = block;
+	return block;
+}
+
+const UniformBlock *ProgramData::get_block(const Program &prog, const Program::UniformBlockInfo *info) const
+{
+	if(changes)
 	{
 		for(BlockMap::iterator i=blocks.begin(); i!=blocks.end(); ++i)
-			i->second.dirty = true;
-		modified = false;
+			if(i->second.changes<changes)
+				i->second.changes = changes;
+		changes = NO_CHANGES;
 	}
 
-	unsigned layout_hash = prog.get_uniform_layout_hash();
+	unsigned layout_hash = (info ? info->layout_hash : prog.get_uniform_layout_hash());
+
 	map<unsigned, Block>::iterator i = blocks.find(layout_hash);
 	if(i==blocks.end())
 	{
 		i = blocks.insert(BlockMap::value_type(layout_hash, Block())).first;
-		i->second.dirty = true;
-		i->second.block = new UniformBlock;
+		if(info)
+		{
+			find_uniforms_for_block(i->second, *info);
+
+			if(!i->second.uniforms.empty())
+			{
+				i->second.block = create_block(*info);
+				i->second.changes = VALUES_CHANGED;
+			}
+		}
+		else
+		{
+			i->second.block = new UniformBlock;
+			i->second.changes = VALUES_CHANGED;
+		}
 	}
+	else if(info && i->second.changes==KEYS_CHANGED)
+	{
+		find_uniforms_for_block(i->second, *info);
+		if(!i->second.uniforms.empty())
+		{
+			if(!i->second.block)
+				i->second.block = create_block(*info);
+			i->second.changes = VALUES_CHANGED;
+		}
+		else
+			i->second.changes = NO_CHANGES;
+	}
+
+	if(!i->second.block)
+		return 0;
 
 	UniformBlock &block = *i->second.block;
-	if(i->second.dirty)
+	if(i->second.changes)
 	{
-		for(UniformMap::const_iterator j=uniforms.begin(); j!=uniforms.end(); ++j)
+		if(info)
 		{
-			int loc = prog.get_uniform_location(j->first);
-			if(loc>=0)
-				block.uniform(loc, *j->second);
+			vector<const Program::UniformInfo *>::const_iterator j = info->uniforms.begin();
+			map<unsigned, const Uniform *const *>::const_iterator k = i->second.uniforms.begin();
+			while(j!=info->uniforms.end() && k!=i->second.uniforms.end())
+			{
+				if(k->first==(*j)->location)
+				{
+					block.attach(**j, **k->second);
+					++k;
+				}
+				++j;
+			}
 		}
-		i->second.dirty = false;
+		else
+		{
+			for(UniformMap::const_iterator j=uniforms.begin(); j!=uniforms.end(); ++j)
+			{
+				int loc = prog.get_uniform_location(j->first);
+				if(loc>=0)
+					block.attach(loc, *j->second);
+			}
+		}
+		i->second.changes = NO_CHANGES;
 	}
 
-	return block;
+	return &block;
+}
+
+const UniformBlock *ProgramData::get_block(const Program &prog, const string &name) const
+{
+	if(name.empty())
+		return get_block(prog, 0);
+	else
+		return get_block(prog, &prog.get_uniform_block_info(name));
 }
 
 void ProgramData::apply() const
@@ -159,13 +301,29 @@ void ProgramData::apply() const
 	if(!prog)
 		throw invalid_operation("ProgramData::apply");
 
-	const UniformBlock &block = get_block(*prog);
-	block.apply(-1);
+	const Program::UniformBlockMap &prog_blocks = prog->get_uniform_blocks();
+	if(!prog_blocks.empty())
+	{
+		typedef pair<const UniformBlock *, unsigned> ApplyBlock;
+		list<ApplyBlock> apply_blocks;
+		for(Program::UniformBlockMap::const_iterator i=prog_blocks.begin(); i!=prog_blocks.end(); ++i)
+			if(const UniformBlock *block = get_block(*prog, &i->second))
+				apply_blocks.push_back(make_pair(block, i->second.bind_point));
+
+		if(buffer)
+			buffer->bind();
+
+		for(list<ApplyBlock>::const_iterator i=apply_blocks.begin(); i!=apply_blocks.end(); ++i)
+			i->first->apply(i->second);
+	}
+
+	if(const UniformBlock *block = get_block(*prog, 0))
+		block->apply(-1);
 }
 
 
 ProgramData::Block::Block():
-	dirty(false),
+	changes(NO_CHANGES),
 	block(0)
 { }