/*****************************************************************
  flock0.cpp
  by Dave Pape
  18 March 2003

*****************************************************************/
#include <stdlib.h>
#include <math.h>
#include <unistd.h>
#include <stdio.h>
#include <GL/glut.h>
#include <GL/gl.h>
#include <dms/dms.h>
#include "Square.h"
#include "boid0.h"

using namespace dms;

void createScene(Object& root,int numFish);
void attachFishGeometry(boid *b, Material& material);
void drawEverything(void);

void key(unsigned char k, int x, int y);
void specialkey(int k, int x, int y);
void idle(void);


PerspCamera camera;
Object root;
Light light;

int main(int argc, char *argv[])
    {
    int numFish = 2;
    
    glutInit(&argc, argv);
    glutInitDisplayMode(GLUT_RGB | GLUT_DOUBLE | GLUT_DEPTH);
    glutInitWindowSize(512,512);
    glutCreateWindow(argv[0]);
    
    glutDisplayFunc(drawEverything);
    glutKeyboardFunc(key);
    glutSpecialFunc(specialkey);
    glutIdleFunc(idle);
    
    camera.setPosition(0, 30, 0);
    camera.setXRotation(-90);
    light.setInfinitePosition(0, 1, 0);
    if (argc > 1)
        numFish = atoi(argv[1]);
    createScene(root, numFish);
    
    glutMainLoop();
    return 0;
    }


void createScene(Object& root,int numFish)
    {
    Square * ground = new Square;
    ground->setTexture(*(new Texture2D("bigpble.jpg", GL_REPEAT, GL_LINEAR_MIPMAP_LINEAR)));
    ground->setTexCoords(0,0, 10,10);
    SimpleTransform * groundXform = new SimpleTransform;
    groundXform->setRotation(-90, 1, 0, 0);
    groundXform->setScaling(10);
    ground->setTransform(*groundXform);
    root.attach(*ground);
    
    Material * material = new Material(Color::Yellow);
    for (int i=0; i < (numFish+1)/2; i++)
        {
        boid *b = new boid;
        attachFishGeometry(b, *material);
        root.attach(*b);
        }
    
    material = new Material(Color::Blue);
    for (int i=0; i < numFish/2; i++)
        {
        boid *b = new boid;
        attachFishGeometry(b, *material);
        root.attach(*b);
        }
    }


void attachFishGeometry(boid *b, Material& material)
    {
    QuadricObject *fish = new QuadricObject;
    fish->makeCone(0.25, 1.0, 8, 1, DMS_X);
    fish->setMaterial(material);
    b->attach(*fish);
    }


void drawEverything(void)
    {
    GLfloat bgcolor[] = { 0.1, 0.5, 0.4, 0.0 };
    glClearColor(bgcolor[0], bgcolor[1], bgcolor[2], bgcolor[3]);
    glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
    glEnable(GL_DEPTH_TEST);
    
    camera.apply();
    
    glDisable(GL_FOG);
    glFogi(GL_FOG_MODE, GL_LINEAR);
    glFogfv(GL_FOG_COLOR, bgcolor);
    glFogf(GL_FOG_START, 4.0);
    glFogf(GL_FOG_END, 30.0);
    
    light.apply();
    
    root.drawAll();

    glutSwapBuffers();

    checkGLError("end-of-frame");
    }


void key(unsigned char k, int x, int y)
    {
    if (k == 27)
        exit(0);
    else if (k == '0')
        {
        camera.setPosition(0, 30, 0);
        camera.setXRotation(-90);
        camera.setYRotation(0);
        }
    else if (k == '1')
        {
        camera.setPosition(0, 5, 30);
        camera.setXRotation(0);
        camera.setYRotation(0);
        }
    }


void specialkey(int k, int x, int y)
    {
    if (k == GLUT_KEY_LEFT)
        camera.turn(3);
    else if (k == GLUT_KEY_RIGHT)
        camera.turn(-3);
    else if (k == GLUT_KEY_UP)
        camera.pitch(2);
    else if (k == GLUT_KEY_DOWN)
        camera.pitch(-2);
    else if (k == GLUT_KEY_HOME)
        camera.moveForward(0.25);
    else if (k == GLUT_KEY_END)
        camera.moveForward(-0.25);
    else if (k == GLUT_KEY_PAGE_UP)
        camera.zoom(-1);
    else if (k == GLUT_KEY_PAGE_DOWN)
        camera.zoom(1);
    }


void idle(void)
    {
    beginFrame();
    root.updateAll();
    glutPostRedisplay();
    }