Cloth/main.cpp at master · bailus/Cloth · GitHub

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
119
120
121
122
123
124
125
126
127
128
129
//============================================================================
// Author      : Samuel Bailey
// Description :
//============================================================================

#include "Includes.h"
#include "Shader.h"
#include "Cloth.h"
#include "Sphere.h"
#include "Floor.h"

GLuint mainWnd;
GLuint winWidth = 640;
GLuint winHeight = 480;
GLuint timeStep = 1000/60; // 1000/fps

Shader *shader;
Cloth *cloth;
#define numThings 1
Thing *things[numThings];

void timer(int t) {
	cloth->step(timeStep);
	for (int i = 0; i < numThings; i++)
		things[i]->collision(&(cloth->particles));
	glutPostRedisplay();  //calls display()
	glutTimerFunc(timeStep, timer, t);
}

void reshape(int w, int h) {
	winWidth = w;
	winHeight = h;
    glViewport(0, 0, w, h);
}

void setCamera() {
	//projection matrix
	glm::mat4 projection = glm::perspective((float)(20*M_PI/180), (float)winWidth/(float)winHeight, 1.f, 1000.f);
	shader->setProjectionMatrix(projection);

	//view matrix
	glm::mat4 view = glm::lookAt(
			glm::vec3(-100, 50, 300), //eye
			glm::vec3(0, -20, 0), //center
			glm::vec3(0, 1, 0) //up
	);
	shader->setViewMatrix(view);
}

void setLight() {
	float direction[] = { 0.5f, 1.0f, 1.0f, 0.0f };
	float diffintensity[] = { 0.6f, 0.6f, 0.6f, 1.0f };
	float ambient[] = { 0.1f, 0.1f, 0.1f, 1.0f };

	glLightfv(GL_LIGHT0, GL_POSITION, direction);
	glLightfv(GL_LIGHT0, GL_DIFFUSE, diffintensity);
	glLightfv(GL_LIGHT0, GL_AMBIENT, ambient);

	glEnable(GL_LIGHT0);
}

void display() {
	glClear(GL_DEPTH_BUFFER_BIT | GL_COLOR_BUFFER_BIT);
	glEnable(GL_DEPTH_TEST);
	glEnable(GL_LIGHTING);
	glEnable(GL_COLOR_MATERIAL);
	glColorMaterial(GL_FRONT_AND_BACK, GL_AMBIENT_AND_DIFFUSE);
	glShadeModel(GL_SMOOTH);

	GLenum err = glGetError();
	if (err != GL_NO_ERROR) {
		printf("%s\n", gluErrorString(err));
	}

	setCamera();
	setLight();
	shader->setModelMatrix(glm::mat4());
	glColor3f(1,1,1);
	if (cloth != NULL)
		cloth->display(GL_TRIANGLES);
		//cloth->display(GL_LINE_LOOP);
		//cloth->displayConstraints();
	for (int i = 0; i < numThings; i++)
		things[i]->display(shader);

	glDisable(GL_DEPTH_TEST);
	glDisable(GL_LIGHTING);
	glDisable(GL_COLOR_MATERIAL);

	glutSwapBuffers();
}

void keyboardHandler(unsigned char key, int x, int y) {
	if (key == 'w') things[0]->move(glm::vec3(0,0,-1));
	if (key == 's') things[0]->move(glm::vec3(0,0,1));
	if (key == 'a') things[0]->move(glm::vec3(-1,0,0));
	if (key == 'd') things[0]->move(glm::vec3(1,0,0));
	if (key == 'r') things[0]->move(glm::vec3(0,-1,0));
	if (key == 'f') things[0]->move(glm::vec3(0,1,0));
}

int main(int argc, char** argv) {
	glutInit(&argc, argv);

    glutInitDisplayMode(GLUT_DOUBLE | GLUT_RGBA | GLUT_DEPTH);
    glutInitWindowSize(winWidth, winHeight);

    mainWnd = glutCreateWindow("Cloth");

    glewInit();
    shader = new Shader((char*)"phong.vert", (char*)"phong.frag");
	shader->use();

	glm::mat4 transform = glm::translate(glm::vec3(-50,20,25)) * glm::rotate(30.f, glm::vec3(1,0,0));
	cloth = new Cloth(100, 50, 100, 50, 100, 0.01, transform);
	cloth->loadTexture((char*)"nzFlag.png");
	things[0] = new Sphere(glm::vec3(0,0,0), 20, glm::vec4(0.2, 0.8, 0.2, 1));
	//things[1] = new Floor(glm::vec3(0,-50,0), 400, 400, glm::vec4(1, 0, 0, 1));

    glutDisplayFunc(display);
    glutReshapeFunc(reshape);
    glutKeyboardFunc(keyboardHandler);

	timer(0);

	glutMainLoop();

    return EXIT_SUCCESS;
}