summaryrefslogtreecommitdiff
path: root/honey/shader.lua
blob: 083b260af38d6d2a58b39f6baca7216732281e61 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
local gl = honey.gl

local module = {}
setmetatable(module, {__index=_G})
setfenv(1, module)


Shader = {}
Shader.__index = Shader

local function compileShader(source, type)
	local shader = gl.CreateShader(type)
	gl.ShaderSource(shader, source)
	gl.CompileShader(shader)
	return shader
end

local function readFile(filename)
	local f, err = io.open(filename)
	if not f then error(err) end
	local str = f:read("*a")
	f:close()
	return str
end

function Shader.new(_, sources)
	local self = {}
	self.locations = {}
	self.links = {}

	if sources.vertexFile then
		sources.vertex = readFile(sources.vertexFile)
	end
	if sources.fragmentFile then
		sources.fragment = readFile(sources.fragmentFile)
	end

	local shaders = {}
	if sources.vertex then
		table.insert(shaders, compileShader(sources.vertex, gl.VERTEX_SHADER))
	end
	if sources.fragment then
		table.insert(shaders, compileShader(sources.fragment, gl.FRAGMENT_SHADER))
	end

	self.program = gl.CreateProgram()
	for _, shader in ipairs(shaders) do
		gl.AttachShader(self.program, shader)
	end
	gl.LinkProgram(self.program)
	for _, shader in ipairs(shaders) do
		gl.DeleteShader(shader)
	end

	setmetatable(self, Shader)
	return self
end
setmetatable(Shader, {__call=Shader.new})


function Shader.getLocation(self, name)
	if self.locations[name] then
		return self.locations[name]
	end

	local location = gl.GetUniformLocation(self.program, name)
	self.locations[name] = location
	return location
end


function Shader.use(self)
	gl.UseProgram(self.program)
end


function Shader.setInt(self, name, value)
	local location = self:getLocation(name)
	gl.Uniform1i(location, value)
end
function Shader.setFloat(self, name, value)
	local location = self:getLocation(name)
	gl.Uniform1f(location, value)
end

function Shader.setVec3(self, name, value)
	local location = self:getLocation(name)
	gl.Uniform3f(location, value[1], value[2], value[3])
end
function Shader.setVec4(self, name, value)
	local location = self:getLocation(name)
	gl.Uniform3f(location, value[1], value[2], value[3], value[4])
end

function Shader.setMatrix(self, name, matrix)
	local location = self:getLocation(name)
	gl.UniformMatrix4fv(location, false, matrix.data)
end


function Shader.configure(self, tbl)
	local processKey = function(key, set)
		local subtbl = tbl[key]
		if subtbl then
			for name, value in pairs(subtbl) do 
				self[set](self, name, value) 
			end
		end
	end

	processKey("int", "setInt")
	processKey("float", "setFloat")
	processKey("vec3", "setVec3")
	processKey("vec4", "setVec4")
	processKey("matrix", "setMatrix")
end

return module.Shader