nuclear@20: #include <assert.h>
nuclear@20: #include <string>
nuclear@20: #include <map>
nuclear@20: #include "shader.h"
nuclear@20: #include "sdr.h"
nuclear@20: #include "opengl.h"
nuclear@20: 
nuclear@20: std::map<std::string, unsigned int> sdrdb;
nuclear@20: 
nuclear@21: const SdrProg *SdrProg::active;
nuclear@21: 
nuclear@20: SdrProg::SdrProg()
nuclear@20: {
nuclear@20: 	prog = 0;
nuclear@20: 	valid = false;
nuclear@20: }
nuclear@20: 
nuclear@20: SdrProg::~SdrProg()
nuclear@20: {
nuclear@20: 	destroy();
nuclear@20: }
nuclear@20: 
nuclear@20: void SdrProg::create()
nuclear@20: {
nuclear@20: 	destroy();
nuclear@20: 	prog = glCreateProgram();
nuclear@20: }
nuclear@20: 
nuclear@20: void SdrProg::destroy()
nuclear@20: {
nuclear@20: 	if(prog) {
nuclear@20: 		glDeleteProgram(prog);
nuclear@20: 	}
nuclear@20: 	valid = false;
nuclear@20: 
nuclear@20: 	for(size_t i=0; i<priv_sdr.size(); i++) {
nuclear@20: 		glDeleteShader(priv_sdr[i]);
nuclear@20: 	}
nuclear@20: 	priv_sdr.clear();
nuclear@20: }
nuclear@20: 
nuclear@20: bool SdrProg::attach_shader(unsigned int sdr)
nuclear@20: {
nuclear@20: 	assert(glGetError() == GL_NO_ERROR);
nuclear@20: 
nuclear@20: 	glAttachShader(prog, sdr);
nuclear@20: 	if(glGetError() != GL_NO_ERROR) {
nuclear@20: 		fprintf(stderr, "failed to attach shader %u to program %u\n", sdr, prog);
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@20: 
nuclear@20: 	valid = false;
nuclear@20: 	return true;
nuclear@20: }
nuclear@20: 
nuclear@20: bool SdrProg::create(unsigned int vsdr, unsigned int psdr)
nuclear@20: {
nuclear@20: 	create();
nuclear@20: 
nuclear@20: 	if(!attach_shader(vsdr) || !attach_shader(psdr)) {
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@20: 
nuclear@21: 	bind_default_attribs();
nuclear@21: 	return link();
nuclear@20: }
nuclear@20: 
nuclear@20: bool SdrProg::create(const char *vsrc, const char *psrc)
nuclear@20: {
nuclear@20: 	unsigned int vs = create_shader(vsrc, GL_VERTEX_SHADER);
nuclear@20: 	if(!vs) {
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@20: 
nuclear@20: 	unsigned int ps = create_shader(psrc, GL_FRAGMENT_SHADER);
nuclear@20: 	if(!ps) {
nuclear@20: 		glDeleteShader(vs);
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@20: 
nuclear@20: 	if(!create(vs, ps)) {
nuclear@20: 		glDeleteShader(vs);
nuclear@20: 		glDeleteShader(ps);
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@20: 
nuclear@20: 	priv_sdr.push_back(vs);
nuclear@20: 	priv_sdr.push_back(ps);
nuclear@20: 	return true;
nuclear@20: }
nuclear@20: 
nuclear@20: bool SdrProg::load(const char *vfname, const char *pfname)
nuclear@20: {
nuclear@21: 	unsigned int vs = get_shader(vfname, GL_VERTEX_SHADER);
nuclear@20: 	if(!vs) {
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@20: 
nuclear@21: 	unsigned int ps = get_shader(pfname, GL_FRAGMENT_SHADER);
nuclear@20: 	if(!ps) {
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@20: 
nuclear@20: 	printf("creating shader program (%s, %s)\n", vfname, pfname);
nuclear@20: 	if(!(prog = create_program_link(vs, ps, 0))) {
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@21: 
nuclear@21: 	bind_default_attribs();
nuclear@21: 	return link();
nuclear@20: }
nuclear@20: 
nuclear@20: bool SdrProg::link() const
nuclear@20: {
nuclear@21: 	if(link_program(prog) == -1) {
nuclear@21: 		return false;
nuclear@21: 	}
nuclear@21: 	valid = true;
nuclear@21: 	return true;
nuclear@20: }
nuclear@20: 
nuclear@20: int SdrProg::get_uniform(const char *name) const
nuclear@20: {
nuclear@20: 	if(!bind()) {
nuclear@20: 		return -1;
nuclear@20: 	}
nuclear@20: 	return glGetUniformLocation(prog, name);
nuclear@20: }
nuclear@20: 
nuclear@20: int SdrProg::get_attrib(const char *name) const
nuclear@20: {
nuclear@20: 	if(!bind()) {
nuclear@20: 		return -1;
nuclear@20: 	}
nuclear@20: 	return glGetAttribLocation(prog, name);
nuclear@20: }
nuclear@20: 
nuclear@20: bool SdrProg::bind_attrib(const char *name, int loc) const
nuclear@20: {
nuclear@20: 	if(!prog) {
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@20: 
nuclear@20: 	assert(glGetError() == GL_NO_ERROR);
nuclear@20: 	glBindAttribLocation(prog, loc, name);
nuclear@20: 	if(glGetError() != GL_NO_ERROR) {
nuclear@20: 		fprintf(stderr, "failed to bind attribute %s of program %u to location %d\n",
nuclear@20: 				name, prog, loc);
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@20: 	valid = false;	/* must relink after the glBindAttribLocation call */
nuclear@20: 	return true;
nuclear@20: }
nuclear@20: 
nuclear@21: void SdrProg::bind_default_attribs() const
nuclear@21: {
nuclear@21: 	// XXX must be in sync with SdrDefaultAttrib enums in shader.h
nuclear@21: 	static const char *def_names[] = {
nuclear@21: 		"attr_vertex",
nuclear@21: 		"attr_normal",
nuclear@21: 		"attr_texcoord",
nuclear@21: 		"attr_color",
nuclear@21: 		"attr_tangent",
nuclear@21: 		0
nuclear@21: 	};
nuclear@21: 
nuclear@21: 	for(int i=0; def_names[i]; i++) {
nuclear@21: 		bind_attrib(def_names[i], i);
nuclear@21: 	}
nuclear@21: }
nuclear@21: 
nuclear@21: 
nuclear@20: bool SdrProg::bind() const
nuclear@20: {
nuclear@20: 	if(!prog || (!valid && !link())) {
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@20: 
nuclear@20: 	assert(glGetError() == GL_NO_ERROR);
nuclear@20: 	glUseProgram(prog);
nuclear@20: 	if(glGetError() != GL_NO_ERROR) {
nuclear@20: 		fprintf(stderr, "failed to bind program %d\n", prog);
nuclear@20: 		return false;
nuclear@20: 	}
nuclear@21: 
nuclear@21: 	active = this;
nuclear@20: 	return true;
nuclear@20: }
nuclear@20: 
nuclear@20: unsigned int get_shader(const char *name, unsigned int type)
nuclear@20: {
nuclear@20: 	std::map<std::string, unsigned int>::const_iterator it = sdrdb.find(name);
nuclear@20: 	if(it != sdrdb.end()) {
nuclear@20: 		return it->second;
nuclear@20: 	}
nuclear@20: 
nuclear@20: 	unsigned int sdr = load_shader(name, type);
nuclear@20: 	if(!sdr) {
nuclear@20: 		return 0;
nuclear@20: 	}
nuclear@20: 
nuclear@20: 	sdrdb[name] = sdr;
nuclear@20: 	return sdr;
nuclear@20: }
nuclear@21: 
nuclear@21: SdrProg *get_sdrprog(const char *vfile, const char *pfile)
nuclear@21: {
nuclear@21: 	SdrProg *sp = new SdrProg;
nuclear@21: 	if(!sp->load(vfile, pfile)) {
nuclear@21: 		delete sp;
nuclear@21: 		return 0;
nuclear@21: 	}
nuclear@21: 	return sp;
nuclear@21: }