#include <GL/freeglut.h>
#include <GL/freeglut_std.h>
#include <GL/gl.h>
#include <GL/glext.h>
#include <GL/glu.h>
#include <stdio.h>
#include <math.h>
#include <stdlib.h>
#include "ui.h"
#include "grass_texture_color.h"
#include "help.h"
#include "cow.h"
// close enough
#define PI 3.14

// Camera rotation and stuff
float camRadius = 40.0;
vec3 camCenter = { 0.0, 0.0, 0.0 };
vec3 camRot = { -PI * 0.25, PI, 0.0 };
struct { 
    unsigned char lockYMovement: 1;
    unsigned char showHelp: 1;
    unsigned char cowsPOV: 1;
    GLenum textureFilter;
    
} settings = {
    0, 0, 0, GL_NEAREST, 
};
struct {
    int main;
    int cowControl;
    int filters;
} menus; // to simply hold the menues in a labed manner
GLuint grassList = -1;


void generateGrassVertexList() {
    grassList = glGenLists(1);
    glNewList(grassList, GL_COMPILE);
        glDisable(GL_CULL_FACE);
            for(float x = -50.0; x < 50.0; x += 1.0) {
                for(float z = -50.0; z < 50.0; z += 1.0) {
                    glPushMatrix();
                    glTranslatef(x, 0.0, z);
                    glRotatef(rand() % 360, 0.0, 1.0, 0.0);
                    glBegin(GL_TRIANGLES);
                    glVertex3f(0.0, 0.0, 0.0);
                    glVertex3f(0.5, 0.0, 0.0);
                    glVertex3f(0.0, 1.5, 0.0);
                    glVertex3f(0.0, 0.0, 0.0);
                    glVertex3f(0.0, 0.0, 0.5);
                    glVertex3f(0.0, 1.5, 0.0);
                    glEnd();
                    glPopMatrix();
                }
            }
        glEnable(GL_CULL_FACE);
    glEndList();
}

vec3 rotateByCameraRotation(vec3 v) {
    vec3 r;
    // rotate along the x axis(yz plane)
    r.z = cosf(camRot.x) * v.z + sinf(camRot.x) * v.y;
    r.y = -sinf(camRot.x) * v.z + cosf(camRot.x) * v.y;
    v = r;
    // rotate along the y axis(xy plane)
    r.z = cosf(camRot.y) * v.z + sinf(camRot.y) * v.x;
    r.x = -sinf(camRot.y) * v.z + cosf(camRot.y) * v.x;
    return r;
}

// Ui stuff
int lastMouseButton = GLUT_LEFT_BUTTON;
int lastMouseX = 0, lastMouseY = 0;
ui_slider* draggedSlider = NULL;
ui_button buttons[] = {}; // TODO: maybe delete? the menu thing is easier to work with
ui_slider sliders[] = {
    { // ambient red
        /* pos */ { { 0.9, 0.9, 0.0 }, { -50.0, -10.0, 0.0 } },
        /* size */ { 100.0, 10.0, 10.0 },
        0.0
    },
    { // ambient green
        /* pos */ { { 0.9, 0.9, 0.0 }, { -50.0, -40.0, 0.0 } },
        /* size */ { 100.0, 10.0, 10.0 },
        0.0
    },
    { // ambient blue
        /* pos */ { { 0.9, 0.9, 0.0 }, { -50.0, -70.0, 0.0 } },
        /* size */ { 100.0, 10.0, 10.0 },
        0.3
    }
    
};

void drawUi(void) {
    glDisable(GL_LIGHTING);
    glDisable(GL_DEPTH_TEST);
    
    glMatrixMode(GL_MODELVIEW);
    glLoadIdentity();
    // set ui camera
    int wWidth = glutGet(GLUT_WINDOW_WIDTH);
    int wHeight = glutGet(GLUT_WINDOW_HEIGHT);
    glMatrixMode(GL_PROJECTION);
    glLoadIdentity();
    gluOrtho2D(0.0, wWidth, 0.0, wHeight);
    
    // Draw buttons
    for(ui_button* b = buttons; b < buttons + sizeof buttons / sizeof(ui_button); b += 1) {
        ui_button_draw(b);
    }
    for(ui_slider* s = sliders; s < sliders + sizeof sliders / sizeof(ui_slider); s += 1) {
        ui_slider_draw(s);
    }
    // sliders labels
    glColor3f(1.0, 1.0, 1.0);
    unsigned char t[] = "Ambient light";
    glRasterPos2f(wWidth * 0.9 - glutBitmapLength(GLUT_BITMAP_TIMES_ROMAN_24, t) * 0.5, wHeight * 0.90 + 20.0);
    glutBitmapString(GLUT_BITMAP_TIMES_ROMAN_24, t);
    glRasterPos2f(wWidth * 0.9 - 75.0, wHeight * 0.9 - 10.0);
    glutBitmapString(GLUT_BITMAP_TIMES_ROMAN_24, (unsigned char*)"r\ng\nb");
    
    if(settings.showHelp) {
        glColor3f(0.3, 0.3, 0.3);
        glBegin(GL_QUADS);
            glVertex3f(wWidth * 0.1, wHeight * 0.1, 0.0);
            glVertex3f(wWidth * 0.9, wHeight * 0.1, 0.0);
            glVertex3f(wWidth * 0.9, wHeight * 0.9, 0.0);
            glVertex3f(wWidth * 0.1, wHeight * 0.9, 0.0);
        glEnd();
        glColor3f(1.0,1.0,1.0);
        glRasterPos2f(wWidth * 0.12, wHeight * 0.8);
        glutBitmapString(GLUT_BITMAP_TIMES_ROMAN_24, helpString);
    }
    
    int err = glGetError();
    if(err != 0) {
        printf("opengl error %d: %s\n", err, gluErrorString(err));
    }
}

void drawWorld(void) {
    glEnable(GL_DEPTH_TEST);
    // set perspective camera
    GLdouble wWidth = (GLdouble) glutGet(GLUT_WINDOW_WIDTH);
    GLdouble wHeight = (GLdouble) glutGet(GLUT_WINDOW_HEIGHT);
    // Calculate current camera position from rotation
    
    glMatrixMode(GL_MODELVIEW);
    glLoadIdentity();
    if(settings.cowsPOV) {
        setCameraToCowPOV();
    }
    else {
        vec3 cp = vec3_new(0.0, 0.0, camRadius);
        cp = rotateByCameraRotation(cp);
        cp = vec3_add(cp, camCenter);
        vec3 up = rotateByCameraRotation(vec3_new(0.0, 1.0, 0.0));
        gluLookAt(cp.x, cp.y, cp.z, camCenter.x, camCenter.y, camCenter.z, up.x, up.y, up.z);
    }
    glMatrixMode(GL_PROJECTION);
    glLoadIdentity();
    gluPerspective(90, wWidth / wHeight  , 0.5, 500.0);
    // Lights
    glEnable(GL_LIGHTING);
    // Ambient light
    GLfloat ambient[] = { sliders[0].value, sliders[1].value, sliders[2].value, 1.0 };
    glLightModelfv(GL_LIGHT_MODEL_AMBIENT, ambient);
    glEnable(GL_LIGHT0);
    // Directional light - aka light
    GLfloat light1[] = { 0.0, 1.0, 0.0, 0.0 };
    glLightfv(GL_LIGHT0, GL_POSITION, light1);
    GLfloat lightColor[] = {0.5, 0.4, 0.1, 1.0};
    glLightfv(GL_LIGHT0, GL_AMBIENT, lightColor);
    glLightfv(GL_LIGHT0, GL_DIFFUSE, lightColor);
    glLightfv(GL_LIGHT0, GL_SPECULAR, lightColor);
    // World
    drawCow(settings.cowsPOV);
    // Ground
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, settings.textureFilter);
    glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, settings.textureFilter);
    glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, 128, 128, 0, GL_RGBA, GL_UNSIGNED_BYTE, grass_color_128);
    glEnable(GL_TEXTURE_2D);
    glBegin(GL_QUADS);
        glColor3f(1.0, 1.0, 0.0);
        glNormal3f(0.0, 5.0, 0.0);
        glTexCoord2f(0.0, 0.0); glVertex3f(-50.0, 0.0, -50.0);
        glTexCoord2f(0.0, 1.0); glVertex3f(-50.0, 0.0, 50.0);
        glTexCoord2f(1.0, 1.0); glVertex3f(50.0, 0.0, 50.0);
        glTexCoord2f(1.0, 0.0); glVertex3f(50.0, 0.0, -50.0);
    glEnd();
    glDisable(GL_TEXTURE_2D);
    // Add a simple cube to see where the cow was originally facing
    glPushMatrix();
    glTranslatef(0.0, 10.0, 50.0);
    glEnable(GL_COLOR_MATERIAL);
    glColor3f(1.0, 1.0, 1.0);
    glutSolidCube(20.0);
    glPopMatrix();
    glColor3f(0.1, 0.3, 0.0);
    glCallList(grassList);
    glDisable(GL_COLOR_MATERIAL);
}

void display(void) {
    // Clear screen
    glClearColor(0.0, 0.0, 0.0, 1.0); // Dark theme :)
    glClear(GL_COLOR_BUFFER_BIT);
    glClear(GL_DEPTH_BUFFER_BIT);

    drawWorld();
    drawUi();

    glutSwapBuffers();
    glFlush();
}

void mouseEvent(int button, int state, int x, int y) {
    lastMouseX = x; lastMouseY = y;
    if(state == 0) lastMouseButton = button;
    if(state == 0 && button == GLUT_LEFT_BUTTON) {
        for(ui_button* b = buttons; b < buttons + sizeof buttons / sizeof(ui_button); b += 1) {
            if(ui_button_mouse_over(b, x, y) && b->onClick != NULL) {
                b->clicked = 1;
            }
        }
        for(ui_slider* s = sliders; s < sliders + sizeof sliders / sizeof(ui_slider); s += 1) {
            if(ui_slider_mouse_over(s, x, y)) {
                ui_slider_onclick(s, x, y);
                draggedSlider = s;
            }
        }
    }
    else if(state == 1 && button == GLUT_LEFT_BUTTON) {
        draggedSlider = NULL;
        for(ui_button* b = buttons; b < buttons + sizeof buttons / sizeof(ui_button); b += 1) {
            b->clicked = 0;
            if(ui_button_mouse_over(b, x, y) && b->onClick != NULL) {
                b->onClick();
            }
        }
    }
    glutPostRedisplay();
}

void mouseMotionEvent(int x, int y) {
    if(lastMouseButton == GLUT_LEFT_BUTTON) {
        if(draggedSlider != NULL) {
            if(ui_slider_mouse_over(draggedSlider, x, y)) {
                ui_slider_onclick(draggedSlider, x, y);
            }
        }
        else {
            // rotate the camera
            camRot.y -= (x - lastMouseX) * PI * 0.01;
            if(camRot.y > 2 * PI) camRot.y -= 2 * PI;
            else if(camRot.y < 0.0) camRot.y += 2 * PI;
            
            camRot.x = fmin(PI * 0.5, fmax(-PI * 0.5, camRot.x - (y - lastMouseY) * 0.01 * PI));
        }
        
    }
    else if(lastMouseButton == GLUT_MIDDLE_BUTTON) {
        vec3 moveX = rotateByCameraRotation(vec3_new(1.0, 0.0, 0.0));
        vec3 moveY = rotateByCameraRotation(vec3_new(0.0, 1.0, 0.0));
        
        if(settings.lockYMovement) {
            moveX.y = 0;
            moveY.y = 0;
        }
        
        camCenter = vec3_add(camCenter, vec3_mult(moveX, (lastMouseX - x)));
        camCenter = vec3_add(camCenter, vec3_mult(moveY, (y - lastMouseY)));
    }
    lastMouseX = x; lastMouseY = y;
    glutPostRedisplay();
}

void mouseWheelEvent(int wheel, int dir, int x, int y) {
    char slided = 0;
    for(ui_slider* s = sliders; s < sliders + sizeof sliders / sizeof(ui_slider); s += 1) {
        if(ui_slider_mouse_over(s, x, y)) {
            s->value += dir * 0.05;
            if(s->value < 0.0) s->value = 0.0;
            if(s->value > 1.0) s->value = 1.0;
            slided = 1;
        }
    }
    if(!slided) {
        camRadius *= 1.0 - (dir * 0.1);
    }
    glutPostRedisplay();
}

void keyboardEvent(unsigned char c, int x, int y) {
    if(c == '\e') {
        if(settings.showHelp) {
            settings.showHelp = 0;
            glutPostRedisplay();
        }
        else {
            glutLeaveMainLoop();
            exit(0);
        }
    }
    onCowKeyboardInput(c);
}

void onMenuItem(int item) {
    lastMouseButton = -1; // to prevent jumps in view position/rotation
    switch(item) {
        case 0: // Lock y
            settings.lockYMovement ^= 1;
            glutChangeToMenuEntry(3, settings.lockYMovement ? "Unlock Y position" : "Lock Y position", 0);
            break;
        case 1: // cow pov
            settings.cowsPOV ^= 1;
            glutPostRedisplay();
            break;
        case 2: // Show help
            settings.showHelp ^= 1;
            glutPostRedisplay();
            break;
        case 3: // reste view
            camRadius = 40.0;
            camCenter = vec3_new(0.0, 0.0, 0.0);
            camRot = vec3_new(-PI * 0.125, 0.0, 0.0);
            glutPostRedisplay();
            break;
        case 4: // quit
            glutLeaveMainLoop();
            exit(0);
    }
}

void onControlCowMenu(int item) {
    lastMouseButton = GLUT_RIGHT_BUTTON;
    updateCowControl(item);
}

void onFilterMenu(int item) {
    lastMouseButton = GLUT_RIGHT_BUTTON;
    GLenum filters[] = { GL_LINEAR, GL_NEAREST };
    settings.textureFilter = filters[item];
}

void menuSetup() {
    // control cow sub-menu
    menus.cowControl = glutCreateMenu(onControlCowMenu);
    glutAddMenuEntry("Movement", COW_CONTROL_MOVE);
    glutAddMenuEntry("Head", COW_CONTROL_HEAD);
    glutAddMenuEntry("Tail", COW_CONTROL_TAIL);
    // filters
    menus.filters = glutCreateMenu(onFilterMenu);
    glutAddMenuEntry("Linear", 0);
    glutAddMenuEntry("Nearest", 1);
    
    menus.main = glutCreateMenu(onMenuItem);
    glutAddSubMenu("Control Cow", menus.cowControl);
    glutAddSubMenu("Texture filters", menus.filters);
    glutAddMenuEntry("Lock Y position", 0);
    glutAddMenuEntry("Change to Cow POV", 1);
    glutAddMenuEntry("Reset view", 3);
    glutAddMenuEntry("Help", 2);
    glutAddMenuEntry("Quit", 4);
    glutAttachMenu(GLUT_RIGHT_BUTTON);
}

int main(int argc, char** argv) {
    /* initialize glut and window */
    glutInit(&argc, argv);
    glutInitDisplayMode(GLUT_DOUBLE | GLUT_RGB | GLUT_DEPTH);
    glutInitWindowSize(800, 500);
    glutInitWindowPosition(450, 450);
    /* create window and set callbacks */
    glutCreateWindow("World of Cow - Aviv Romem");
    glutDisplayFunc(display);
    glutMouseFunc(mouseEvent);
    glutKeyboardFunc(keyboardEvent);
    glutMouseWheelFunc(mouseWheelEvent);
    glutMotionFunc(mouseMotionEvent);
    menuSetup();
    // set up some rendering related stuff
    glEnable(GL_CULL_FACE);
    glCullFace(GL_BACK);

    generateGrassVertexList();
    glutMainLoop();

    return 0;
}