xenocara/dist/Mesa/progs/glsl/shtest.c

712 lines
16 KiB
C

/*
* Simple shader test harness.
* Brian Paul
* 13 Aug 2009
*
* Usage:
* shtest --vs vertShaderFile --fs fragShaderFile
*
* In this case the given vertex/frag shaders are read and compiled.
* Random values are assigned to the uniforms.
*
* or:
* shtest configFile
*
* In this case a config file is read that specifies the file names
* of the shaders plus initial values for uniforms.
*
* Example config file:
*
* vs shader.vert
* fs shader.frag
* uniform pi 3.14159
* uniform v1 1.0 0.5 0.2 0.3
* texture 0 2D texture0.rgb
* texture 1 CUBE texture1.rgb
* texture 2 RECT texture2.rgb
*
*/
#include <assert.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <math.h>
#include <GL/glew.h>
#include <GL/glu.h>
#include <GL/glut.h>
#include "shaderutil.h"
#include "readtex.h"
typedef enum
{
SPHERE,
CUBE,
NUM_SHAPES
} shape;
static char *FragShaderFile = NULL;
static char *VertShaderFile = NULL;
static char *ConfigFile = NULL;
/* program/shader objects */
static GLuint fragShader;
static GLuint vertShader;
static GLuint Program;
#define MAX_UNIFORMS 100
static struct uniform_info Uniforms[MAX_UNIFORMS];
static GLuint NumUniforms = 0;
#define MAX_ATTRIBS 100
static struct attrib_info Attribs[MAX_ATTRIBS];
static GLuint NumAttribs = 0;
/**
* Config file info.
*/
struct config_file
{
struct name_value
{
char name[100];
float value[4];
int type;
} uniforms[100];
int num_uniforms;
};
static GLint win = 0;
static GLboolean Anim = GL_FALSE;
static GLfloat TexRot = 0.0;
static GLfloat xRot = 0.0f, yRot = 0.0f, zRot = 0.0f;
static shape Object = SPHERE;
static float
RandomFloat(float min, float max)
{
int k = rand() % 10000;
float x = min + (max - min) * k / 10000.0;
return x;
}
/** Set new random values for uniforms */
static void
RandomUniformValues(void)
{
GLuint i;
for (i = 0; i < NumUniforms; i++) {
switch (Uniforms[i].type) {
case GL_FLOAT:
Uniforms[i].value[0] = RandomFloat(0.0, 1.0);
break;
case GL_SAMPLER_1D:
case GL_SAMPLER_2D:
case GL_SAMPLER_3D:
case GL_SAMPLER_CUBE:
case GL_SAMPLER_2D_RECT_ARB:
/* don't change sampler values - random values are bad */
break;
default:
Uniforms[i].value[0] = RandomFloat(-1.0, 2.0);
Uniforms[i].value[1] = RandomFloat(-1.0, 2.0);
Uniforms[i].value[2] = RandomFloat(-1.0, 2.0);
Uniforms[i].value[3] = RandomFloat(-1.0, 2.0);
}
}
}
static void
Idle(void)
{
yRot += 2.0;
if (yRot > 360.0)
yRot -= 360.0;
glutPostRedisplay();
}
static void
SquareVertex(GLfloat s, GLfloat t, GLfloat size)
{
GLfloat x = -size + s * 2.0 * size;
GLfloat y = -size + t * 2.0 * size;
GLuint i;
glMultiTexCoord2f(GL_TEXTURE0, s, t);
glMultiTexCoord2f(GL_TEXTURE1, s, t);
glMultiTexCoord2f(GL_TEXTURE2, s, t);
glMultiTexCoord2f(GL_TEXTURE3, s, t);
/* assign (s,t) to the generic attributes */
for (i = 0; i < NumAttribs; i++) {
if (Attribs[i].location >= 0) {
glVertexAttrib2f(Attribs[i].location, s, t);
}
}
glVertex2f(x, y);
}
/*
* Draw a square, specifying normal and tangent vectors.
*/
static void
Square(GLfloat size)
{
GLint tangentAttrib = 1;
glNormal3f(0, 0, 1);
glVertexAttrib3f(tangentAttrib, 1, 0, 0);
glBegin(GL_POLYGON);
#if 1
SquareVertex(0, 0, size);
SquareVertex(1, 0, size);
SquareVertex(1, 1, size);
SquareVertex(0, 1, size);
#else
glTexCoord2f(0, 0); glVertex2f(-size, -size);
glTexCoord2f(1, 0); glVertex2f( size, -size);
glTexCoord2f(1, 1); glVertex2f( size, size);
glTexCoord2f(0, 1); glVertex2f(-size, size);
#endif
glEnd();
}
static void
Cube(GLfloat size)
{
/* +X */
glPushMatrix();
glRotatef(90, 0, 1, 0);
glTranslatef(0, 0, size);
Square(size);
glPopMatrix();
/* -X */
glPushMatrix();
glRotatef(-90, 0, 1, 0);
glTranslatef(0, 0, size);
Square(size);
glPopMatrix();
/* +Y */
glPushMatrix();
glRotatef(90, 1, 0, 0);
glTranslatef(0, 0, size);
Square(size);
glPopMatrix();
/* -Y */
glPushMatrix();
glRotatef(-90, 1, 0, 0);
glTranslatef(0, 0, size);
Square(size);
glPopMatrix();
/* +Z */
glPushMatrix();
glTranslatef(0, 0, size);
Square(size);
glPopMatrix();
/* -Z */
glPushMatrix();
glRotatef(180, 0, 1, 0);
glTranslatef(0, 0, size);
Square(size);
glPopMatrix();
}
static void
Sphere(GLfloat radius, GLint slices, GLint stacks)
{
static GLUquadricObj *q = NULL;
if (!q) {
q = gluNewQuadric();
gluQuadricDrawStyle(q, GLU_FILL);
gluQuadricNormals(q, GLU_SMOOTH);
gluQuadricTexture(q, GL_TRUE);
}
gluSphere(q, radius, slices, stacks);
}
static void
Redisplay(void)
{
glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
glPushMatrix();
glRotatef(xRot, 1.0f, 0.0f, 0.0f);
glRotatef(yRot, 0.0f, 1.0f, 0.0f);
glRotatef(zRot, 0.0f, 0.0f, 1.0f);
glMatrixMode(GL_TEXTURE);
glLoadIdentity();
glRotatef(TexRot, 0.0f, 1.0f, 0.0f);
glMatrixMode(GL_MODELVIEW);
if (Object == SPHERE) {
Sphere(2.5, 20, 10);
}
else if (Object == CUBE) {
Cube(2.0);
}
glPopMatrix();
glutSwapBuffers();
}
static void
Reshape(int width, int height)
{
glViewport(0, 0, width, height);
glMatrixMode(GL_PROJECTION);
glLoadIdentity();
glFrustum(-1.0, 1.0, -1.0, 1.0, 5.0, 25.0);
glMatrixMode(GL_MODELVIEW);
glLoadIdentity();
glTranslatef(0.0f, 0.0f, -15.0f);
}
static void
CleanUp(void)
{
glDeleteShader(fragShader);
glDeleteShader(vertShader);
glDeleteProgram(Program);
glutDestroyWindow(win);
}
static void
Key(unsigned char key, int x, int y)
{
const GLfloat step = 2.0;
(void) x;
(void) y;
switch(key) {
case 'a':
Anim = !Anim;
if (Anim)
glutIdleFunc(Idle);
else
glutIdleFunc(NULL);
break;
case 'z':
zRot += step;
break;
case 'Z':
zRot -= step;
break;
case 'o':
Object = (Object + 1) % NUM_SHAPES;
break;
case 'r':
RandomUniformValues();
SetUniformValues(Program, Uniforms);
PrintUniforms(Uniforms);
break;
case 27:
CleanUp();
exit(0);
break;
}
glutPostRedisplay();
}
static void
SpecialKey(int key, int x, int y)
{
const GLfloat step = 2.0;
(void) x;
(void) y;
switch(key) {
case GLUT_KEY_UP:
xRot += step;
break;
case GLUT_KEY_DOWN:
xRot -= step;
break;
case GLUT_KEY_LEFT:
yRot -= step;
break;
case GLUT_KEY_RIGHT:
yRot += step;
break;
}
glutPostRedisplay();
}
static void
InitUniforms(const struct config_file *conf,
struct uniform_info uniforms[])
{
int i;
for (i = 0; i < conf->num_uniforms; i++) {
int j;
for (j = 0; uniforms[j].name; j++) {
if (strcmp(uniforms[j].name, conf->uniforms[i].name) == 0) {
uniforms[j].type = conf->uniforms[i].type;
uniforms[j].value[0] = conf->uniforms[i].value[0];
uniforms[j].value[1] = conf->uniforms[i].value[1];
uniforms[j].value[2] = conf->uniforms[i].value[2];
uniforms[j].value[3] = conf->uniforms[i].value[3];
}
}
}
}
static void
LoadTexture(GLint unit, GLenum target, const char *texFileName)
{
GLint imgWidth, imgHeight;
GLenum imgFormat;
GLubyte *image = NULL;
GLuint tex;
GLenum filter = GL_LINEAR;
GLenum objTarget;
image = LoadRGBImage(texFileName, &imgWidth, &imgHeight, &imgFormat);
if (!image) {
printf("Couldn't read %s\n", texFileName);
exit(1);
}
printf("Load Texture: unit %d, target 0x%x: %s %d x %d\n",
unit, target, texFileName, imgWidth, imgHeight);
if (target >= GL_TEXTURE_CUBE_MAP_POSITIVE_X &&
target <= GL_TEXTURE_CUBE_MAP_NEGATIVE_Z) {
objTarget = GL_TEXTURE_CUBE_MAP;
}
else {
objTarget = target;
}
glActiveTexture(GL_TEXTURE0 + unit);
glGenTextures(1, &tex);
glBindTexture(objTarget, tex);
if (target == GL_TEXTURE_3D) {
/* depth=1 */
gluBuild3DMipmaps(target, 4, imgWidth, imgHeight, 1,
imgFormat, GL_UNSIGNED_BYTE, image);
}
else if (target == GL_TEXTURE_1D) {
gluBuild1DMipmaps(target, 4, imgWidth,
imgFormat, GL_UNSIGNED_BYTE, image);
}
else {
gluBuild2DMipmaps(target, 4, imgWidth, imgHeight,
imgFormat, GL_UNSIGNED_BYTE, image);
}
free(image);
glTexParameteri(objTarget, GL_TEXTURE_WRAP_S, GL_REPEAT);
glTexParameteri(objTarget, GL_TEXTURE_WRAP_T, GL_REPEAT);
glTexParameteri(objTarget, GL_TEXTURE_MIN_FILTER, filter);
glTexParameteri(objTarget, GL_TEXTURE_MAG_FILTER, filter);
}
static GLenum
TypeFromName(const char *n)
{
static const struct {
const char *name;
GLenum type;
} types[] = {
{ "GL_FLOAT", GL_FLOAT },
{ "GL_FLOAT_VEC2", GL_FLOAT_VEC2 },
{ "GL_FLOAT_VEC3", GL_FLOAT_VEC3 },
{ "GL_FLOAT_VEC4", GL_FLOAT_VEC4 },
{ "GL_INT", GL_INT },
{ "GL_INT_VEC2", GL_INT_VEC2 },
{ "GL_INT_VEC3", GL_INT_VEC3 },
{ "GL_INT_VEC4", GL_INT_VEC4 },
{ "GL_SAMPLER_1D", GL_SAMPLER_1D },
{ "GL_SAMPLER_2D", GL_SAMPLER_2D },
{ "GL_SAMPLER_3D", GL_SAMPLER_3D },
{ "GL_SAMPLER_CUBE", GL_SAMPLER_CUBE },
{ "GL_SAMPLER_2D_RECT", GL_SAMPLER_2D_RECT_ARB },
{ NULL, 0 }
};
GLuint i;
for (i = 0; types[i].name; i++) {
if (strcmp(types[i].name, n) == 0)
return types[i].type;
}
abort();
return GL_NONE;
}
/**
* Read a config file.
*/
static void
ReadConfigFile(const char *filename, struct config_file *conf)
{
char line[1000];
FILE *f;
f = fopen(filename, "r");
if (!f) {
fprintf(stderr, "Unable to open config file %s\n", filename);
exit(1);
}
conf->num_uniforms = 0;
/* ugly but functional parser */
while (fgets(line, sizeof(line), f) != NULL) {
if (line[0]) {
if (strncmp(line, "vs ", 3) == 0) {
VertShaderFile = strdup(line + 3);
VertShaderFile[strlen(VertShaderFile) - 1] = 0;
}
else if (strncmp(line, "fs ", 3) == 0) {
FragShaderFile = strdup(line + 3);
FragShaderFile[strlen(FragShaderFile) - 1] = 0;
}
else if (strncmp(line, "texture ", 8) == 0) {
char target[100], texFileName[100];
int unit, k;
k = sscanf(line + 8, "%d %s %s", &unit, target, texFileName);
assert(k == 3 || k == 8);
if (strcmp(target, "CUBE") == 0) {
char texFileNames[6][100];
k = sscanf(line + 8, "%d %s %s %s %s %s %s %s",
&unit, target,
texFileNames[0],
texFileNames[1],
texFileNames[2],
texFileNames[3],
texFileNames[4],
texFileNames[5]);
LoadTexture(unit, GL_TEXTURE_CUBE_MAP_POSITIVE_X, texFileNames[0]);
LoadTexture(unit, GL_TEXTURE_CUBE_MAP_NEGATIVE_X, texFileNames[1]);
LoadTexture(unit, GL_TEXTURE_CUBE_MAP_POSITIVE_Y, texFileNames[2]);
LoadTexture(unit, GL_TEXTURE_CUBE_MAP_NEGATIVE_Y, texFileNames[3]);
LoadTexture(unit, GL_TEXTURE_CUBE_MAP_POSITIVE_Z, texFileNames[4]);
LoadTexture(unit, GL_TEXTURE_CUBE_MAP_NEGATIVE_Z, texFileNames[5]);
}
else if (!strcmp(target, "2D")) {
LoadTexture(unit, GL_TEXTURE_2D, texFileName);
}
else if (!strcmp(target, "3D")) {
LoadTexture(unit, GL_TEXTURE_3D, texFileName);
}
else if (!strcmp(target, "RECT")) {
LoadTexture(unit, GL_TEXTURE_RECTANGLE_ARB, texFileName);
}
else {
printf("Bad texture target: %s\n", target);
exit(1);
}
}
else if (strncmp(line, "uniform ", 8) == 0) {
char name[1000], typeName[100];
int k;
float v1 = 0.0F, v2 = 0.0F, v3 = 0.0F, v4 = 0.0F;
GLenum type;
k = sscanf(line + 8, "%s %s %f %f %f %f", name, typeName,
&v1, &v2, &v3, &v4);
type = TypeFromName(typeName);
if (strlen(name) + 1 > sizeof(conf->uniforms[conf->num_uniforms].name)) {
fprintf(stderr, "string overflow\n");
exit(1);
}
strcpy(conf->uniforms[conf->num_uniforms].name, name);
conf->uniforms[conf->num_uniforms].value[0] = v1;
conf->uniforms[conf->num_uniforms].value[1] = v2;
conf->uniforms[conf->num_uniforms].value[2] = v3;
conf->uniforms[conf->num_uniforms].value[3] = v4;
conf->uniforms[conf->num_uniforms].type = type;
conf->num_uniforms++;
}
else {
if (strlen(line) > 1) {
fprintf(stderr, "syntax error in: %s\n", line);
break;
}
}
}
}
fclose(f);
}
static void
Init(void)
{
GLdouble vertTime, fragTime, linkTime;
struct config_file config;
memset(&config, 0, sizeof(config));
if (ConfigFile)
ReadConfigFile(ConfigFile, &config);
if (!VertShaderFile) {
fprintf(stderr, "Error: no vertex shader\n");
exit(1);
}
if (!FragShaderFile) {
fprintf(stderr, "Error: no fragment shader\n");
exit(1);
}
if (!ShadersSupported())
exit(1);
vertShader = CompileShaderFile(GL_VERTEX_SHADER, VertShaderFile);
vertTime = GetShaderCompileTime();
fragShader = CompileShaderFile(GL_FRAGMENT_SHADER, FragShaderFile);
fragTime = GetShaderCompileTime();
Program = LinkShaders(vertShader, fragShader);
linkTime = GetShaderLinkTime();
printf("Read vert shader %s\n", VertShaderFile);
printf("Read frag shader %s\n", FragShaderFile);
printf("Time to compile vertex shader: %fs\n", vertTime);
printf("Time to compile fragment shader: %fs\n", fragTime);
printf("Time to link shaders: %fs\n", linkTime);
assert(ValidateShaderProgram(Program));
glUseProgram(Program);
NumUniforms = GetUniforms(Program, Uniforms);
if (config.num_uniforms) {
InitUniforms(&config, Uniforms);
}
else {
RandomUniformValues();
}
SetUniformValues(Program, Uniforms);
PrintUniforms(Uniforms);
NumAttribs = GetAttribs(Program, Attribs);
PrintAttribs(Attribs);
/* assert(glGetError() == 0); */
glClearColor(0.4f, 0.4f, 0.8f, 0.0f);
glEnable(GL_DEPTH_TEST);
glColor3f(1, 0, 0);
}
static void
Keys(void)
{
printf("Keyboard:\n");
printf(" a Animation toggle\n");
printf(" r Randomize uniform values\n");
printf(" o Change object\n");
printf(" arrows Rotate object\n");
printf(" ESC Exit\n");
}
static void
Usage(void)
{
printf("Usage:\n");
printf(" shtest config.shtest\n");
printf(" Run w/ given config file.\n");
printf(" shtest --vs vertShader --fs fragShader\n");
printf(" Load/compile given shaders.\n");
}
static void
ParseOptions(int argc, char *argv[])
{
int i;
if (argc == 1) {
Usage();
exit(1);
}
for (i = 1; i < argc; i++) {
if (strcmp(argv[i], "--fs") == 0) {
FragShaderFile = argv[i+1];
i++;
}
else if (strcmp(argv[i], "--vs") == 0) {
VertShaderFile = argv[i+1];
i++;
}
else {
/* assume the arg is a config file */
ConfigFile = argv[i];
break;
}
}
}
int
main(int argc, char *argv[])
{
glutInitWindowSize(400, 400);
glutInit(&argc, argv);
glutInitDisplayMode(GLUT_RGB | GLUT_DOUBLE | GLUT_DEPTH);
win = glutCreateWindow(argv[0]);
glewInit();
glutReshapeFunc(Reshape);
glutKeyboardFunc(Key);
glutSpecialFunc(SpecialKey);
glutDisplayFunc(Redisplay);
ParseOptions(argc, argv);
Init();
Keys();
glutMainLoop();
return 0;
}