(root)/iphone/common : /source/ES2Shader.m (revision 96)

//

//  ES2Shader.m

//

//  Created by döme on 08.08.2009.

//

/*

 * Copyright (c) 2009 Doemoetoer Gulyas.

 * All rights reserved.

 *

 * Redistribution and use in source and binary forms, with or without

 * modification, are permitted provided that the following conditions

 * are met:

 * 1. Redistributions of source code must retain the above copyright

 *    notice, this list of conditions and the following disclaimer.

 * 2. Redistributions in binary form must reproduce the above copyright

 *    notice, this list of conditions and the following disclaimer in the

 *    documentation and/or other materials provided with the distribution.

 * 3. The name of the author may not be used to endorse or promote products

 *    derived from this software without specific prior written permission.

 *

 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR

 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES

 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.

 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,

 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT

 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,

 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY

 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT

 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF

 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

 */

#import "ES2Shader.h"

#import <OpenGLES/ES2/gl.h>

@implementation ES2Shader

- (id) initWithVertexShaderString: (NSString*) vs fragmentShaderString: (NSString*) fs

{

	if (!(self = [super init]))

		return nil;

	const char* vsString = [vs UTF8String];

	const char* fsString = [fs UTF8String];

	vsShader = glCreateShader(GL_VERTEX_SHADER);

	fsShader = glCreateShader(GL_FRAGMENT_SHADER);

	assert(vsShader);

	assert(fsShader);

	glShaderSource(vsShader, 1, &vsString, NULL);

	glShaderSource(fsShader, 1, &fsString, NULL);

	glCompileShader(vsShader);

	{

		GLint vsStatus = 0;

		glGetShaderiv(vsShader, GL_COMPILE_STATUS, &vsStatus);

		if (!vsStatus)

		{

			GLint logLength = 0;

			glGetShaderiv(vsShader, GL_INFO_LOG_LENGTH, &logLength);

			char* infoLog = calloc(1, logLength);

			glGetShaderInfoLog(vsShader, logLength, NULL, infoLog);

			NSLog(@"Vertex shader compilation failed: %s", infoLog);

		}

	}

	glCompileShader(fsShader);

	{

		GLint fsStatus = 0;

		glGetShaderiv(fsShader, GL_COMPILE_STATUS, &fsStatus);

		if (!fsStatus)

		{

			GLint logLength = 0;

			glGetShaderiv(fsShader, GL_INFO_LOG_LENGTH, &logLength);

			char* infoLog = calloc(1, logLength);

			glGetShaderInfoLog(fsShader, logLength, NULL, infoLog);

			NSLog(@"Fragment shader compilation failed: %s", infoLog);

		}

	}

	programObj = glCreateProgram();

	assert(programObj);

	glAttachShader(programObj, fsShader);

	glAttachShader(programObj, vsShader);

	glBindAttribLocation(programObj, POSITION_ATTRIB_INDEX, "position");

	glBindAttribLocation(programObj, NORMAL_ATTRIB_INDEX, "normal");

	glBindAttribLocation(programObj, COLOR_ATTRIB_INDEX, "colora");

	glBindAttribLocation(programObj, TEXCOORD_ATTRIB_INDEX, "texcoord");

	glLinkProgram(programObj);

	GLint linkStatus = 0;

	glGetProgramiv(programObj, GL_LINK_STATUS, &linkStatus);

	if (!linkStatus)

	{

		GLint logLength = 0;

		glGetProgramiv(programObj, GL_INFO_LOG_LENGTH, &logLength);

		char* infoLog = calloc(1, logLength);

		glGetProgramInfoLog(programObj, logLength, NULL, infoLog);

		NSLog(@"Shader linking failed: %s", infoLog);

	}

	//sampler0loc = glGetUniformLocation(programObj, "tex0");

	//glUniform1i(sampler0loc, 0);

	uniformDict = [[NSMutableDictionary alloc] init];

	return self;

}

- (id) initWithVertexShaderFile: (NSString*) vs fragmentShaderFile: (NSString*) fs

{

	if (!(self = [super init]))

		return nil;

	assert(GL_NO_ERROR == glGetError());

	vertexShaderName = vs;

	fragmentShaderName = fs;

	NSData* vsData = [NSData dataWithContentsOfFile: [[NSBundle mainBundle] pathForResource: vs ofType: nil]];

	NSData* fsData = [NSData dataWithContentsOfFile: [[NSBundle mainBundle] pathForResource: fs ofType: nil]];

	vsShader = glCreateShader(GL_VERTEX_SHADER);

	fsShader = glCreateShader(GL_FRAGMENT_SHADER);

	assert(vsShader);

	assert(fsShader);

	GLint lvs = [vsData length];

	GLint lfs = [fsData length];

	const char* vsString = [vsData bytes];

	const char* fsString = [fsData bytes];

	glShaderSource(vsShader, 1, &vsString, &lvs);

	glShaderSource(fsShader, 1, &fsString, &lfs);

	glCompileShader(vsShader);

	{

		GLint vsStatus = 0;

		glGetShaderiv(vsShader, GL_COMPILE_STATUS, &vsStatus);

		if (!vsStatus)

		{

			GLint logLength = 0;

			glGetShaderiv(vsShader, GL_INFO_LOG_LENGTH, &logLength);

			char* infoLog = calloc(1, logLength);

			glGetShaderInfoLog(vsShader, logLength, NULL, infoLog);

			NSLog(@"Vertex shader compilation failed: %s", infoLog);

		}

	}

	glCompileShader(fsShader);

	{

		GLint fsStatus = 0;

		glGetShaderiv(fsShader, GL_COMPILE_STATUS, &fsStatus);

		if (GL_FALSE == fsStatus)

		{

			GLint logLength = 0;

			glGetShaderiv(fsShader, GL_INFO_LOG_LENGTH, &logLength);

			char* infoLog = calloc(1, logLength);

			glGetShaderInfoLog(fsShader, logLength, NULL, infoLog);

			NSLog(@"Fragment shader compilation failed: %s", infoLog);

		}

	}

	assert(GL_NO_ERROR == glGetError());

	programObj = glCreateProgram();

	assert(programObj);

	glAttachShader(programObj, fsShader);

	glAttachShader(programObj, vsShader);

	glBindAttribLocation(programObj, POSITION_ATTRIB_INDEX, "position");

	glBindAttribLocation(programObj, NORMAL_ATTRIB_INDEX, "normal");

	glBindAttribLocation(programObj, COLOR_ATTRIB_INDEX, "colora");

	glBindAttribLocation(programObj, TEXCOORD_ATTRIB_INDEX, "texcoord");

	glLinkProgram(programObj);

	GLint linkStatus = 0;

	glGetProgramiv(programObj, GL_LINK_STATUS, &linkStatus);

	if (GL_FALSE == linkStatus)

	{

		GLint logLength = 0;

		glGetProgramiv(programObj, GL_INFO_LOG_LENGTH, &logLength);

		char* infoLog = calloc(1, logLength);

		glGetProgramInfoLog(programObj, logLength, NULL, infoLog);

		NSLog(@"Shader linking failed: %s", infoLog);

	}

//	sampler0loc = glGetUniformLocation(programObj, "tex0");

//	glUniform1i(sampler0loc, 0);

	uniformDict = [[NSMutableDictionary alloc] init];

	assert(GL_NO_ERROR == glGetError());

	return self;

}

- (void) dealloc

{

	glDeleteProgram(programObj);

	glDeleteShader(fsShader);

	glDeleteShader(vsShader);

}

- (void) bindShader

{

	glUseProgram(programObj);

}

- (void) setUniformInteger: (GLint) i named: (NSString*) name

{

	GLint loc = [[uniformDict objectForKey: name] intValue];

	if (!loc)

	{

		loc = glGetUniformLocation(programObj, [name UTF8String]);

		[uniformDict setObject: [NSNumber numberWithUnsignedInt: loc] forKey: name];

	}

	if (loc != -1)

		glUniform1i(loc, i);

	else

		NSLog(@"failed to set uniform %@", name);

}

- (void) setUniformFloat: (float) f named: (NSString*) name

{

	GLint loc = [[uniformDict objectForKey: name] intValue];

	if (!loc)

	{

		loc = glGetUniformLocation(programObj, [name UTF8String]);

		[uniformDict setObject: [NSNumber numberWithUnsignedInt: loc] forKey: name];

	}

	if (loc != -1)

		glUniform1f(loc, f);

	else

		NSLog(@"failed to set uniform %@", name);

}

- (void) setUniformFloat: (float*) f count: (int) count named: (NSString*) name

{

	GLint loc = [[uniformDict objectForKey: name] intValue];

	if (!loc)

	{

		loc = glGetUniformLocation(programObj, [name UTF8String]);

		[uniformDict setObject: [NSNumber numberWithUnsignedInt: loc] forKey: name];

	}

	if (loc != -1)

		glUniform4fv(loc, count, f);

	else

		NSLog(@"failed to set uniform %@", name);

}

- (void) setModelviewMatrix: (m16*) m

{

	m16 nm = mtranspose(minverse(*m));

	if (!modelviewMatrixLoc)

		modelviewMatrixLoc = glGetUniformLocation(programObj, "modelviewMatrix");

	if (!normalMatrixLoc)

		normalMatrixLoc = glGetUniformLocation(programObj, "normalMatrix");

	if (modelviewMatrixLoc != -1)

		glUniformMatrix4fv(modelviewMatrixLoc, 1, GL_FALSE, m->a);

	if (normalMatrixLoc != -1)

		glUniformMatrix4fv(normalMatrixLoc, 1, GL_FALSE, nm.a);

}

- (void) setProjectionMatrix: (m16*) m

{

	if (!projectionMatrixLoc)

		projectionMatrixLoc = glGetUniformLocation(programObj, "projectionMatrix");

	if (projectionMatrixLoc != -1)

		glUniformMatrix4fv(projectionMatrixLoc, 1, GL_FALSE, m->a);

}

- (void) setTextureMatrix: (m16*) m

{

	if (!textureMatrixLoc)

		textureMatrixLoc = glGetUniformLocation(programObj, "textureMatrix");

	if (textureMatrixLoc != -1)

		glUniformMatrix4fv(textureMatrixLoc, 1, GL_FALSE, m->a);

}

- (void) setColorRed: (float) r green: (float) g blue: (float) b alpha: (float) a

{

	if (!colorLoc)

		colorLoc = glGetUniformLocation(programObj, "color");

	if (colorLoc != -1)

		glUniform4f(colorLoc, r, g, b, a);

}

@end