// lighting.c
// By M.Szwoch, Based on
// OpenGL SuperBible, Chapter 17
// Demonstrates vertex and fragment shaders together
// Program by Benjamin Lipchak

#include "../shared/gltools.h"   // OpenGL toolkit
#include "../shared/glFrame.h"   // Frame toolkit

#include <stdio.h>

// 3. Dodaj swoje imię i nazwisko do belki tytułowej
// ...
const char *authorName = "Bartłomiej Gągol";

enum ShaderTypes { SIMPLE, DIFFUSE, SPECULAR, THREELIGHTS, FOG, PARAMS, SHADERS_NUM };

GLuint vShader[ SHADERS_NUM ]; // identyfikatory programów cieniowania wierzchołków (VS)
GLuint fShader[ SHADERS_NUM ]; // identyfikatory programów cieniowania fragmentów (FS)
GLuint progObj[ SHADERS_NUM ]; // obiekty programów cieniowania (VS+FS)
GLboolean needsValidation[ SHADERS_NUM ]; // informacja o aktualności programu cieniownaia
char *shaderNames[ SHADERS_NUM ] =
{ // nazwy programów cieniowania
	"simple",
	"diffuse",
	"specular",
	"3lights",
	"fog",
	"params"
};
// aktualny program cieniowania
ShaderTypes actualShader = SIMPLE;
// rozmiary okna
GLint windowWidth = 1024;
GLint windowHeight = 768;

GLint mainMenu, shaderMenu;             // menu handles

GLfloat cameraPos[] = { 100.0f, 75.0f, 150.0f, 1.0f };
GLdouble cameraZoom = 0.4;

GLfloat lightPos0[] = { 140.0f, 250.0f, 140.0f, 1.0f };
GLfloat lightPos1[] = { -140.0f, 250.0f, 140.0f, 1.0f };
GLfloat lightPos2[] = { 0.0f, 250.0f, -200.0f, 1.0f };

GLfloat lightRotation = 0.0f;

#define MAX_INFO_LOG_SIZE 2048
GLchar infoLog[ MAX_INFO_LOG_SIZE ]; // bufor na błędy obsługi programów cieniowania
void FatalError( char *errorKind, char *errorDescription, char *errorDetails )
{ // wyświetlenie informacji o błędzie i zakończenie programu ("radykalna" obsługa błędów)
	fprintf( stderr, "!!! %s ERROR !!!\n", errorKind );
	fprintf( stderr, " %s \"%s\"\n", errorDescription, errorDetails );
	Sleep( 5000 );
	exit( 0 );
}
GLchar *LoadShaderText( const char *fileName )
{ // Załadowanie z pliku kodu źródłowego programu cieniowania jako tekstu w formacie ASCIIZ
	GLchar *shaderText = NULL;
	GLint shaderLength = 0;
	FILE *fp;
	fp = fopen( fileName, "r" );
	if ( fp != NULL )
	{
		while ( fgetc( fp ) != EOF )
			shaderLength++;
		rewind( fp );
		shaderText = (GLchar *)malloc( shaderLength + 1 );
		if ( shaderText != NULL )
			fread( shaderText, 1, shaderLength, fp );
		shaderText[ shaderLength ] = '\0';
		fclose( fp );
	}
	return shaderText;
}
void LoadAndCompileShader( char * fullFileName, GLuint shaderKind, GLuint *shaderNum )
{ // załadowanie programu cieniowania z dysku i jego kompilacja
	// załadowanie programu cieniowania z dysku
	GLint success;
	GLchar *shString = LoadShaderText( fullFileName ); // załadowanie kodu źródłowego
	fprintf( stdout, "Loading '%s' shader\n", fullFileName );

	if ( !shString )
		FatalError( "LOADING", "Unable to load", fullFileName );

	// 5. Utworzenie obiektu cieniowania funkcją glCreateShader()
	// Parametrem jest rodzaj obiektu cieniowania (shaderKind)
	// Wynik wywołania należy podstawić do identyfikator obiektu (*shaderNum)
	// ...
	*shaderNum = glCreateShader(shaderKind);

	// 6. Wskazanie kodu źródłowego obiektu cieniowania funkcją glShaderSource()
	// Pierwszym parametrem jest identyfikator obiektu cieniowania,
	// drugim liczba obiektów (1),
	// trzecim wskaźnik na wskaźnik (const GLchar **)& na kod źródłowy
	// czwartym NULL
	// ...
	glShaderSource(*shaderNum, 1, (const GLchar **)& shString, NULL);

	free( shString );

	// 7. Kompilacja obiektu cieniowania funkcją glCompileShader()
	// Parametrem jest identyfikator obiektu cieniowania,
	// ...
	glCompileShader(*shaderNum);


	// 8. Sprawdzenie statusu kompilacji funkcją glGetShaderiv()
	// Pierwszym parametrem jest identyfikator obiektu cieniowania,
	// drugim rodzaj sprawdzanego błędu (GL_COMPILE_STATUS)
	// trzecim adres zmiennej na zwracany kod błędu (success)
	// ...
	glGetShaderiv(*shaderNum, GL_COMPILE_STATUS, &success);

	// 9. Informacja o ewentualnym błędzie w przypadku, gdy success == 0
	// Tekst komunikatu pobierany jest funkcją glGetShaderInfoLog()
	//  Pierwszym parametrem jest identyfikator obiektu cieniowania,
	//  drugim rozmiar bufora (MAX_INFO_LOG_SIZE)
	//  trzecim NULL
	//  czwartym bufor (infoLog)
	// Tekst błędu należy wypisać funkcją FatalError()
	// ...
	if(success == 0){
		glGetShaderInfoLog(*shaderNum, MAX_INFO_LOG_SIZE, NULL, infoLog);
		FatalError(infoLog, NULL, NULL);
	}




}
void PrepareShader( GLint shaderNum )
{ // Przygotowanie programu cieniowania
	char fullFileName[ 255 ];
	GLint success;

	// Określenie w fullFileName nazwy względnej pliku ze wskazanym przez shaderNum programem cieniowania wierzchołków
	sprintf( fullFileName, "./shaders/%s.vs", shaderNames[ shaderNum ] );

	// 11. Załadowanie i kompilacja programu cieniowania wierzchołków funkcją LoadAndCompileShader()
	// Pierwszym parametrem jest nazwa pliku,
	// drugim rodzaj programu cieniowania (GL_VERTEX_SHADER)
	// trzecim adres odpowiedniego pola w tablicy identyfikatorów obiektów cieniowania vShader
	// ...
	LoadAndCompileShader(fullFileName, GL_VERTEX_SHADER, &vShader[shaderNum]);

	// 12. Określenie nazwy względnej pliku ze wskazanym (shaderNum) programem cieniowania fragmentów
	// Pliki mają rozszerzenie "fs"
	// ...
	sprintf( fullFileName, "./shaders/%s.fs", shaderNames[ shaderNum ] );


	// 13. Załadowanie i kompilacja programu cieniowania fragmentów: (GL_FRAGMENT_SHADER, fShader)
	// ...
	LoadAndCompileShader(fullFileName, GL_FRAGMENT_SHADER, &fShader[shaderNum]);


	// 14. Utworzenie obiektu programu cieniowania funkcją glCreateProgram()
	// Identyfikator obiektu programu cieniowania należy podstawić do odpowiedniego miejsca tablicy progObj
	// ...
	progObj[shaderNum] = glCreateProgram();

	// 15. Dołączenie do obiektu programu cieniowania skompilowanego kodu VS funkcją glAttachShader()
	// Pierwszym parametrem jest identyfikator obiektu cieniowania,
	// drugim parametrem identyfikator obiektu VS
	// ...
	glAttachShader(progObj[shaderNum], vShader[shaderNum]);

	// 16. Dołączenie do obiektu programu cieniowania skompilowanego kodu FS
	// ...
	glAttachShader(progObj[shaderNum], fShader[shaderNum]);


	// 17. Łączenie programu cieniowania funkcją glLinkProgram()
	// Parametrem jest identyfikator obiektu programu cieniowania
	// ...
	glLinkProgram(progObj[shaderNum]);

	// 18. Sprawdzenie statusu łączenia funkcją glGetProgramiv()
	// Pierwszym parametrem jest identyfikator obiektu programu cieniowania,
	// drugim rodzaj sprawdzanego błędu (GL_LINK_STATUS)
	// trzecim adres zmiennej na zwracany kod błędu (success)
	// ...
	glGetProgramiv(progObj[shaderNum], GL_LINK_STATUS, &success);

	// 19. Informacja o ewentualnym błędzie w przypadku, gdy success == 0
	// SHADER LINKING", "Unable to link"
	// ...

	if(success == 0){
		FatalError("SHADER LINKING", "Unable to link", NULL);
	}





	// 20. Ustawienie w odpowiednim miejscu tablicy needsValidation[]
	// informacji o konieczności walidacji programu cieniowania (GL_TRUE)
	// ...
	needsValidation[shaderNum] = GL_TRUE;

}
// ------------------------------------------------------------------------
void DrawModels(void)
{ // narysowanie obiektów na scenie, ustawienie parametrów Uniform dla jednostek cieniowania
	M3DVector3f lightPos0Eye, lightPos1Eye, lightPos2Eye;
	M3DMatrix44f mv;

	// transformacja światła do przestrzeni kamery (eye space)
	glPushMatrix();
	glRotatef( lightRotation, 0.0, 1.0, 0.0 );
	glGetFloatv( GL_MODELVIEW_MATRIX, mv );
	m3dTransformVector3( lightPos0Eye, lightPos0, mv );


	if ( actualShader >= THREELIGHTS )
	{
		m3dTransformVector3( lightPos1Eye, lightPos1, mv );
		m3dTransformVector3( lightPos2Eye, lightPos2, mv );
	}

	glPopMatrix();
	GLint uniformLoc;

	// 29. Pobranie identyfikatora zmiennej Uniform dla źródła światła nr 0 funkcją glGetUniformLocation()
	// Pierwszym parametrem jest aktualny identyfikator obiektu programu cieniowania
	// drugim parametrem jest nazwa zmiennej Uniform "lightPos[0]"
	// Wynik należy podstawić do zmiennej uniformLoc
	// ...
	uniformLoc = glGetUniformLocation(progObj[actualShader], "lightPos[0]");


	// 30. Jeżeli zwrócony identyfikator jest różny od -1, ustawienie aktualnej pozycji światła 0 funkcją glUniform3fv()
	// Pierwszym parametrem jest identyfikator zmiennej,
	// drugim parametrem jest liczba argumentów (1)
	// trzecim parametrem wektor współrzędnych oświetlenia o (lightPos0Eye)
	// ...
	if(uniformLoc != -1){
		glUniform3fv(uniformLoc, 1, lightPos0Eye);
	}


	// 37. Analogiczne ustawienie parametru dla światła nr 1
	// ...
	uniformLoc = glGetUniformLocation(progObj[actualShader], "lightPos[1]");

	if(uniformLoc != -1){
		glUniform3fv(uniformLoc, 1, lightPos1Eye);
	}



	// 37. Analogiczne ustawienie parametru dla światła nr 2
	// ...
	uniformLoc = glGetUniformLocation(progObj[actualShader], "lightPos[2]");

	if(uniformLoc != -1){
		glUniform3fv(uniformLoc, 1, lightPos2Eye);
	}




	// Narysowanie niebieskiej podstawy obiektów
	glColor3f( 0.0f, 0.0f, 0.90f );
	glNormal3f( 0.0f, 1.0f, 0.0f );
	glBegin( GL_QUADS );
		glVertex3f( -100.0f, -25.0f, -100.0f );
		glVertex3f( -100.0f, -25.0f, 100.0f );
		glVertex3f( 100.0f,  -25.0f, 100.0f );
		glVertex3f( 100.0f,  -25.0f, -100.0f );
	glEnd();
	// czerwony sześcian
	glColor3f( 1.0f, 0.0f, 0.0f );
	glutSolidCube( 48.0f );
	// zielona sfera
	glColor3f( 0.0f, 1.0f, 0.0f );
	glPushMatrix();
	glTranslatef( -60.0f, 0.0f, 0.0f );
	glutSolidSphere( 25.0f, 50, 50 );
	glPopMatrix();
	// bordowy torus
	glColor3f( 1.0f, 0.0f, 1.0f );
	glPushMatrix();
	glTranslatef( 0.0f, 0.0f, 60.0f );
	glutSolidTorus( 8.0f, 16.0f, 50, 50 );
	glPopMatrix();
	// żółty stożek
	glColor3f( 1.0f, 1.0f, 0.0f );
	glPushMatrix();
	glRotatef( -90.0f, 1.0f, 0.0f, 0.0f );
	glTranslatef( 60.0f, 0.0f, -24.0f );
	glutSolidCone( 25.0f, 50.0f, 50, 50 );
	glPopMatrix();
	// morski czajniczek
	glColor3f( 0.0f, 1.0f, 1.0f );
	glPushMatrix();
	glTranslatef( 0.0f, 0.0f, -60.0f );
	glutSolidTeapot( 25.0f );
	glPopMatrix();
}
void RenderScene( void )
{ // rysowanie sceny
	// ustawienie perspektywy
	glMatrixMode( GL_PROJECTION );
	glLoadIdentity();
	if ( windowWidth > windowHeight )
	{
		GLdouble ar = (GLdouble)windowWidth / (GLdouble)windowHeight;
		glFrustum( -ar * cameraZoom, ar * cameraZoom, -cameraZoom, cameraZoom, 1.0, 1000.0 );
	}
	else
	{
		GLdouble ar = (GLdouble)windowHeight / (GLdouble)windowWidth;
		glFrustum( -cameraZoom, cameraZoom, -ar * cameraZoom, ar * cameraZoom, 1.0, 1000.0 );
	}
	glMatrixMode( GL_MODELVIEW );
	glLoadIdentity();
	gluLookAt( cameraPos[ 0 ], cameraPos[ 1 ], cameraPos[ 2 ], 0.0f, 0.0f, 0.0f, 0.0f, 1.0f, 0.0f );
	glViewport( 0, 0, windowWidth, windowHeight );
	// wyczyszczenie okna
	glClear( GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT );

	GLint success;
	// 20. Walidacja aktualnego (actualShader) programu cieniowania.
	// WYŁĄCZNIE w przypadku, gdy potrzebuje walidacji needsValidation[]
	// UWAGA! actualShader jest indeksem!!!!!
	// ...
	if ( needsValidation[ actualShader ] )
	{

		// 21. Walidacja aktualnego (actualShader) programu cieniowania funkcją glValidateProgram()
		// Parametrem jest aktualny identyfikator obiektu programu cieniowania
		// ...
		glValidateProgram(progObj[actualShader]);

		// 22. Sprawdzenie statusu walidacji funkcją glGetProgramiv()
		// Pierwszym parametrem jest identyfikator obiektu programu cieniowania,
		// drugim rodzaj sprawdzanego błędu (GL_VALIDATE_STATUS)
		// trzecim adres zmiennej na zwracany kod błędu (success)
		// ...
		glGetProgramiv(progObj[actualShader], GL_VALIDATE_STATUS, &success);

		// 23. Informacja o ewentualnym błędzie w przypadku, gdy success == 0
		// "SHADER VALIDATION", "Unable to validate"
		// ...

		if(success == 0){
			FatalError("SHADER VALIDATION", "Unable to validate", NULL);
		}





		// 24. Wyzerowanie informacji o potrzebie walidacji aktualnego programu cieniowania
		// ...
		needsValidation[actualShader] = GL_FALSE;

	}
	// Narysowanie modeli
	DrawModels();
	if ( glGetError() != GL_NO_ERROR )
		fprintf( stderr, "GL Error!\n" );
	// narysowanie sceny
	glutSwapBuffers();
}

// This function does any needed initialization on the rendering
// context.
void SetupRC()
{ // inicjalizacja sceny
	fprintf( stdout, "Masters of GLSL shaders (2016)\n\n" );
	// 4. Sprawdzenie spełnienia minimalnych wymogów wersji cieniowania SM (shadermodel)
	if ( !GLEE_VERSION_2_0 && ( !GLEE_ARB_fragment_shader || !GLEE_ARB_vertex_shader ||
				!GLEE_ARB_shader_objects || !GLEE_ARB_shading_language_100 ) )
		FatalError( "OpenGL version", "OpenGL version", "earlier then 2.0" );

	fprintf( stdout, "Controls:\n" );
	fprintf( stdout, "\tR/L arrows\t+/- rotate lights\n\n" );
	fprintf( stdout, "\tx/X\t\tMove +/- in x direction\n" );
	fprintf( stdout, "\ty/Y\t\tMove +/- in y direction\n" );
	fprintf( stdout, "\tz/Z\t\tMove +/- in z direction\n\n" );
	fprintf( stdout, "\tq, ESC\t\tExit demo\n\n" );

	// ustawienia OpenGL
	glClearColor( 0.0f, 0.0f, 0.0f, 1.0f ); // czarne tło
	glEnable( GL_DEPTH_TEST );
	glDepthFunc( GL_LEQUAL );
	glShadeModel( GL_SMOOTH );

	// 25. Wczytanie i przygotowanie wszystkich SHADERS_NUM programów cieniowania funkcją PrepareShader()
	// ...
	for(int i = 0; i < SHADERS_NUM; i++){
		PrepareShader(i);
	}


	// 26. Ustawienie aktualnego (actualShader) programu cieniowania funkcją glUseProgram()
	// Parametrem jest identyfikator aktualnego programu cieniowania
	// UWAGA! actualShader jest indeksem!!!!!
	// ...
	glUseProgram(progObj[actualShader]);

}
char *SetTitle( void )
{
	static char buffer[ 200 ];
	sprintf( buffer, "Master of GLSL shaders by %s [%s]", authorName, shaderNames[ actualShader ] );
	return( buffer );
}
void KeyPressFunc( unsigned char key, int x, int y )
{
	switch ( key )
	{
		case ' ': // spacja
			actualShader = (ShaderTypes)( ( actualShader + 1 ) % SHADERS_NUM ); // naprzemienne włączenie kolejnych programów cieniowania
			glutSetWindowTitle( SetTitle() );

			// 27. Ustawienie nowego (actualShader) programu cieniowania funkcją glUseProgram()
			// Parametrem jest identyfikator aktualnego programu cieniowania
			// ...
			glUseProgram(progObj[actualShader]);

			break;
		case 'x':
			cameraPos[ 0 ] += 5.0f;
			break;
		case 'X':
         cameraPos[0] -= 5.0f;
			break;
		case 'y':
			cameraPos[ 1 ] += 5.0f;
			break;
		case 'Y':
			cameraPos[ 1 ] -= 5.0f;
			break;
		case 'z':
			cameraPos[ 2 ] += 5.0f;
			break;
		case 'Z':
			cameraPos[2] -= 5.0f;
			break;
		case 'q':
		case 'Q':
		case 27 : /* ESC */
			glutDestroyWindow( glutGetWindow() );
			break;
	}
	glutPostRedisplay(); // odświeżenie okna
}
void SpecialKeys(int key, int x, int y)
{
	switch ( key )
	{
		case GLUT_KEY_LEFT:
			lightRotation -= 5.0f;
			break;
		case GLUT_KEY_RIGHT:
			lightRotation += 5.0f;
			break;
		case GLUT_KEY_UP:
			cameraPos[ 1 ] += 5.0f;
			break;
		case GLUT_KEY_DOWN:
			cameraPos[1] -= 5.0f;
			break;
	}
	glutPostRedisplay(); // odświeżenie okna
}
void ChangeSize( int w, int h )
{
	windowWidth = w;
	windowHeight = h;
}

int main( int argc, char* argv[] )
{
	GLint i;
	glutInit( &argc, argv );
	glutInitDisplayMode( GLUT_DOUBLE | GLUT_RGB | GLUT_DEPTH );
	glutInitWindowSize( windowWidth, windowHeight );
	glutCreateWindow( SetTitle() ); // utworzenie okna
	glutReshapeFunc( ChangeSize );
	glutKeyboardFunc( KeyPressFunc );
	glutSpecialFunc( SpecialKeys );
	glutDisplayFunc( RenderScene );
	SetupRC();
	glutMainLoop();

	if ( glDeleteProgram && glDeleteShader )
	{
		// 26. Zwolnienie pamięci dla wszystkich (SHADERS_NUM) obiektów programu cieniowania
		// oraz jednostek cieniowania VS i FS funkcjami glDeleteProgram() oraz glDeleteShader()
		// ...
		for ( i = 0; i < SHADERS_NUM; i++ )
		{
			glDeleteProgram( progObj[ i ] );
			glDeleteShader( vShader[ i ] );
			glDeleteShader( fShader[ i ] );
		}
	}
	return 0;
}