com.domsplace.FaultEngine.Shader.ShaderProgram.java Source code

Introduction

Here is the source code for com.domsplace.FaultEngine.Shader.ShaderProgram.java

Source

/*
 * Copyright 2014 Dominic Masters.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.domsplace.FaultEngine.Shader;

import com.domsplace.FaultEngine.Utilities.FileUtilities;
import java.nio.FloatBuffer;
import java.util.ArrayList;
import java.util.List;
import org.lwjgl.BufferUtils;
import org.lwjgl.opengl.GL20;
import static org.lwjgl.opengl.GL20.*;

/**
 *
 * @author Dominic Masters
 */
public class ShaderProgram {
    public static void unbindProgram() {
        glUseProgram(0);
    }

    //Predefined Shaders
    private static ShaderProgram toonShader = null;

    public static ShaderProgram getToonShader() {
        if (toonShader != null)
            return toonShader;
        Shader toonFragShader = new Shader();
        Shader toonVertShader = new Shader();
        toonShader = new ShaderProgram();
        try {
            toonFragShader.compile(FileUtilities.getResourceAsString("shaders/ToonShader/toon_shader.frag"),
                    GL20.GL_FRAGMENT_SHADER);
            toonVertShader.compile(FileUtilities.getResourceAsString("shaders/ToonShader/toon_shader.vert"),
                    GL20.GL_VERTEX_SHADER);
            toonShader.addShader(toonFragShader).addShader(toonVertShader);
            toonShader.compile();
        } catch (Exception e) {
        }
        return toonShader;
    }

    //Instance
    private int program = -1;
    private List<Shader> shaders = new ArrayList<Shader>();

    public ShaderProgram() {
    }

    public int getProgramID() {
        return this.program;
    }

    public ShaderProgram addShader(Shader shader) {
        this.shaders.add(shader);
        return this;
    }

    public void compile() throws Exception {
        this.program = glCreateProgram();
        if (this.program == 0) {
            this.program = -1;
            throw new Exception("Failed to create ShaderProgram (Failed to Gen ID)");
        }

        for (Shader shader : this.shaders) {
            glAttachShader(this.program, shader.getID());
        }
        glLinkProgram(this.program);
        glValidateProgram(this.program);
    }

    public void bind() {
        glUseProgram(this.program);
    }

    public int getVariableID(String variable) {
        return glGetUniformLocation(this.program, variable);
    }

    public void setVariable(String variable, float[] data) {
        FloatBuffer fb = BufferUtils.createFloatBuffer(data.length);
        fb.put(data);
        fb.flip();
        setVariable(variable, fb);
    }

    public void setVariable(String variable, int data) {
        setVariableByID(getVariableID(variable), data);
    }

    public void setVariable(String variable, boolean data) {
        setVariableByID(getVariableID(variable), data ? 1 : 0);
    }

    public void setVariable(String variable, FloatBuffer stuff) {
        setVariableByID(getVariableID(variable), stuff);
    }

    public void setVariableByID(int id, FloatBuffer stuff) {
        int size = stuff.capacity();
        if (size == 1) {
            GL20.glUniform1(id, stuff);
        } else if (size == 2) {
            GL20.glUniform2(id, stuff);
        } else if (size == 3) {
            GL20.glUniform3(id, stuff);
        } else if (size == 4) {
            GL20.glUniform4(id, stuff);
        }
    }

    public void setVariableByID(int id, int stuff) {
        GL20.glUniform1i(id, stuff);
    }
}