shaderPhong.cpp 7.51 KB
Newer Older
Sylvain Thery's avatar
Sylvain Thery committed
1
2
3
/*******************************************************************************
* CGoGN: Combinatorial and Geometric modeling with Generic N-dimensional Maps  *
* version 0.1                                                                  *
4
* Copyright (C) 2009-2011, IGG Team, LSIIT, University of Strasbourg           *
Sylvain Thery's avatar
Sylvain Thery committed
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
*                                                                              *
* This library is free software; you can redistribute it and/or modify it      *
* under the terms of the GNU Lesser General Public License as published by the *
* Free Software Foundation; either version 2.1 of the License, or (at your     *
* option) any later version.                                                   *
*                                                                              *
* This library is distributed in the hope that it will be useful, but WITHOUT  *
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or        *
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License  *
* for more details.                                                            *
*                                                                              *
* You should have received a copy of the GNU Lesser General Public License     *
* along with this library; if not, write to the Free Software Foundation,      *
* Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301 USA.           *
*                                                                              *
20
* Web site: http://cgogn.u-strasbg.fr/                                         *
Sylvain Thery's avatar
Sylvain Thery committed
21
22
23
24
25
* Contact information: cgogn@unistra.fr                                        *
*                                                                              *
*******************************************************************************/

#include <GL/glew.h>
26
#include "Utils/Shaders/shaderPhong.h"
Sylvain Thery's avatar
Sylvain Thery committed
27
28
29

namespace CGoGN
{
Pierre Kraemer's avatar
merge..    
Pierre Kraemer committed
30

Sylvain Thery's avatar
Sylvain Thery committed
31
32
33
namespace Utils
{

Sylvain Thery's avatar
Sylvain Thery committed
34
std::string ShaderPhong::vertexShaderText =
Sylvain Thery's avatar
Sylvain Thery committed
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
"ATTRIBUTE vec3 VertexPosition, VertexNormal;\n"
"#ifdef WITH_COLOR\n"
"ATTRIBUTE vec3 VertexColor;\n"
"#endif\n"
"uniform mat4 ModelViewProjectionMatrix;\n"
"uniform mat4 ModelViewMatrix;\n"
"uniform mat4 NormalMatrix;\n"
"uniform vec3 lightPosition;\n"
"VARYING_VERT vec3 EyeVector, Normal, LightDir;\n"
"#ifdef WITH_COLOR\n"
"VARYING_VERT vec3 Color;\n"
"#endif\n"
"INVARIANT_POS;\n"
"void main ()\n"
"{\n"
"	Normal = vec3 (NormalMatrix * vec4 (VertexNormal, 0.0));\n"
" 	vec3 Position = vec3 (ModelViewMatrix * vec4 (VertexPosition, 1.0));\n"
"	LightDir = lightPosition - Position;\n"
"	EyeVector = -Position;\n"
"	#ifdef WITH_COLOR\n"
55
"	Color = VertexColor;\n"
Sylvain Thery's avatar
Sylvain Thery committed
56
57
58
59
60
"	#endif\n"
"	gl_Position = ModelViewProjectionMatrix * vec4 (VertexPosition, 1.0);\n"
"}";


Sylvain Thery's avatar
Sylvain Thery committed
61
std::string ShaderPhong::fragmentShaderText =
Sylvain Thery's avatar
Sylvain Thery committed
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
"PRECISON;\n"
"VARYING_FRAG vec3 EyeVector, Normal, LightDir;\n"
"#ifdef WITH_COLOR\n"
"VARYING_FRAG vec3 Color;\n"
"#endif\n"
"uniform vec4 materialDiffuse;\n"
"uniform vec4 materialSpecular;\n"
"uniform vec4 materialAmbient;\n"
"uniform float shininess;\n"
"FRAG_OUT_DEF;\n"
"void main()\n"
"{\n"
"	vec3 N = normalize (Normal);\n"
"	vec3 L = normalize (LightDir);\n"
"	float lambertTerm = dot(N,L);\n"

"	vec4 finalColor = materialAmbient;\n"
"	if(lambertTerm > 0.0)\n"
"	{\n"
"		#ifndef WITH_COLOR\n"
"		finalColor += materialDiffuse * lambertTerm;\n"
"		#else\n"
"		finalColor += vec4((Color*lambertTerm),0.0) ;\n"
"		#endif\n"
"		vec3 E = normalize(EyeVector);\n"
"		vec3 R = reflect(-L, N);\n"
"		float specular = pow( max(dot(R, E), 0.0), shininess );\n"
"		finalColor += materialSpecular * specular;\n"
"	}\n"
"	gl_FragColor=finalColor;\n"
"}";


Sylvain Thery's avatar
Sylvain Thery committed
95
ShaderPhong::ShaderPhong():
Pierre Kraemer's avatar
merge..    
Pierre Kraemer committed
96
97
98
99
100
101
	m_with_color(false),
	m_ambiant(Geom::Vec4f(0.05f,0.05f,0.1f,0.0f)),
	m_diffuse(Geom::Vec4f(0.1f,1.0f,0.1f,0.0f)),
	m_specular(Geom::Vec4f(1.0f,1.0f,1.0f,0.0f)),
	m_shininess(100.0f),
	m_lightPos(Geom::Vec3f(10.0f,10.0f,1000.0f))
Sylvain Thery's avatar
Sylvain Thery committed
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
{
	// get choose GL defines (2 or 3)
	// ans compile shaders
	std::string glxvert(*GLSLShader::DEFINES_GL);
	glxvert.append(vertexShaderText);
	std::string glxfrag(*GLSLShader::DEFINES_GL);
	glxfrag.append(fragmentShaderText);

	loadShadersFromMemory(glxvert.c_str(), glxfrag.c_str());

	// and get and fill uniforms
	bind();
	getLocations();
	sendParams();
}

Sylvain Thery's avatar
Sylvain Thery committed
118
void ShaderPhong::getLocations()
Sylvain Thery's avatar
Sylvain Thery committed
119
{
Pierre Kraemer's avatar
merge..    
Pierre Kraemer committed
120
121
122
123
124
	m_unif_ambiant   = glGetUniformLocation(this->program_handler(), "materialAmbient");
	m_unif_diffuse   = glGetUniformLocation(this->program_handler(), "materialDiffuse");
	m_unif_specular  = glGetUniformLocation(this->program_handler(), "materialSpecular");
	m_unif_shininess = glGetUniformLocation(this->program_handler(), "shininess");
	m_unif_lightPos  = glGetUniformLocation(this->program_handler(), "lightPosition");
Sylvain Thery's avatar
Sylvain Thery committed
125
126
}

Sylvain Thery's avatar
Sylvain Thery committed
127
void ShaderPhong::sendParams()
Sylvain Thery's avatar
Sylvain Thery committed
128
129
130
131
132
133
134
135
{
	glUniform4fv(m_unif_ambiant,  1, m_ambiant.data());
	glUniform4fv(m_unif_diffuse,  1, m_diffuse.data());
	glUniform4fv(m_unif_specular, 1, m_specular.data());
	glUniform1f(m_unif_shininess,    m_shininess);
	glUniform3fv(m_unif_lightPos, 1, m_lightPos.data());
}

Sylvain Thery's avatar
Sylvain Thery committed
136
void ShaderPhong::setAmbiant(const Geom::Vec4f& ambiant)
Sylvain Thery's avatar
Sylvain Thery committed
137
138
139
140
141
142
{
	this->bind();
	glUniform4fv(m_unif_ambiant,1, ambiant.data());
	m_ambiant = ambiant;
}

Sylvain Thery's avatar
Sylvain Thery committed
143
void ShaderPhong::setDiffuse(const Geom::Vec4f& diffuse)
Sylvain Thery's avatar
Sylvain Thery committed
144
145
146
147
148
149
{
	this->bind();
	glUniform4fv(m_unif_diffuse,1, diffuse.data());
	m_diffuse = diffuse;
}

Sylvain Thery's avatar
Sylvain Thery committed
150
void ShaderPhong::setSpecular(const Geom::Vec4f& specular)
Sylvain Thery's avatar
Sylvain Thery committed
151
152
153
154
155
156
{
	this->bind();
	glUniform4fv(m_unif_specular,1,specular.data());
	m_specular = specular;
}

Sylvain Thery's avatar
Sylvain Thery committed
157
void ShaderPhong::setShininess(float shininess)
Sylvain Thery's avatar
Sylvain Thery committed
158
159
160
161
162
163
{
	this->bind();
	glUniform1f (m_unif_shininess, shininess);
	m_shininess = shininess;
}

Sylvain Thery's avatar
Sylvain Thery committed
164
void ShaderPhong::setLightPosition( Geom::Vec3f lightPos)
Sylvain Thery's avatar
Sylvain Thery committed
165
166
167
168
169
170
{
	this->bind();
	glUniform3fv(m_unif_lightPos,1,lightPos.data());
	m_lightPos = lightPos;
}

Sylvain Thery's avatar
Sylvain Thery committed
171
void ShaderPhong::setParams(const Geom::Vec4f& ambiant, const Geom::Vec4f& diffuse, const Geom::Vec4f& specular, float shininess, const Geom::Vec3f& lightPos)
Sylvain Thery's avatar
Sylvain Thery committed
172
173
174
175
176
177
178
179
180
181
{
	m_ambiant = ambiant;
	m_diffuse = diffuse;
	m_specular = specular;
	m_shininess = shininess;
	m_lightPos = lightPos;
	bind();
	sendParams();
}

182
unsigned int ShaderPhong::setAttributeColor(VBO* vbo)
Sylvain Thery's avatar
Sylvain Thery committed
183
{
184
	m_vboColor = vbo;
Sylvain Thery's avatar
Sylvain Thery committed
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
	if (!m_with_color)
	{
		m_with_color=true;
		// set the define and recompile shader
		std::string gl3vert(*GLSLShader::DEFINES_GL);
		gl3vert.append("#define WITH_COLOR 1\n");
		gl3vert.append(vertexShaderText);
		std::string gl3frag(*GLSLShader::DEFINES_GL);
		gl3frag.append("#define WITH_COLOR 1\n");
		gl3frag.append(fragmentShaderText);
		loadShadersFromMemory(gl3vert.c_str(), gl3frag.c_str());

		// and treat uniforms
		bind();
		getLocations();
		sendParams();
	}
	// bind th VA with WBO
203
	return bindVA_VBO("VertexColor", vbo);
Sylvain Thery's avatar
Sylvain Thery committed
204
205
}

Sylvain Thery's avatar
Sylvain Thery committed
206
void ShaderPhong::unsetAttributeColor()
Sylvain Thery's avatar
Sylvain Thery committed
207
{
208
	m_vboColor = NULL;
Sylvain Thery's avatar
Sylvain Thery committed
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
	if (m_with_color)
	{
		m_with_color=false;
		// unbind the VA
		unbindVA("VertexColor");
		// recompile shader
		std::string gl3vert(*GLSLShader::DEFINES_GL);
		gl3vert.append(vertexShaderText);
		std::string gl3frag(*GLSLShader::DEFINES_GL);
		gl3frag.append(fragmentShaderText);
		loadShadersFromMemory(gl3vert.c_str(), gl3frag.c_str());
		// and treat uniforms
		bind();
		getLocations();
		sendParams();
	}
225
226
227
228
229
230
}

void ShaderPhong::restoreUniformsAttribs()
{
	getLocations();
	sendParams();
Sylvain Thery's avatar
Sylvain Thery committed
231

232
233
234
235
	bindVA_VBO("VertexPosition", m_vboPos);
	bindVA_VBO("VertexNormal", m_vboNormal);
	if (m_vboColor)
		bindVA_VBO("VertexColor", m_vboColor);
Sylvain Thery's avatar
Sylvain Thery committed
236
237
}

238
239
unsigned int ShaderPhong::setAttributePosition(VBO* vbo)
{
Pierre Kraemer's avatar
merge..    
Pierre Kraemer committed
240
	m_vboPos = vbo;
241
242
243
244
245
246
247
248
249
	return bindVA_VBO("VertexPosition", vbo);
}

unsigned int ShaderPhong::setAttributeNormal(VBO* vbo)
{
	m_vboNormal = vbo;
	return bindVA_VBO("VertexNormal", vbo);
}

Pierre Kraemer's avatar
merge..    
Pierre Kraemer committed
250
} // namespace Utils
251

Pierre Kraemer's avatar
merge..    
Pierre Kraemer committed
252
} // namespace CGoGN