#include <stdio.h>
#include <gtk/gtk.h>
#include <gdk/gdkgl.h>
#include <gtk/gtkgl.h>
#include <GL/glut.h>
#include <glade/glade.h>

#include "config.h"
#include "display.h"

#include "../firmware/globals.h"

extern void lc_setMode(int);
extern void lc_setFrame(unsigned long);
extern int lc_saveFrame(unsigned long frame, int delay, int index);

void drawLEDs(gint mode) {
  gint x, y, z;
  gint ledIndex = 0;
  gint iLED = 0;

  if (ledOrientation == TOP_ORIENTATION) {
    glRotatef(90, 2, 0, 0);
  }

  for (z=10; z>=-10; z-=10) // Ebene
    for (y=10; y>=-10; y-=10) // Zeile
      for (x=10; x>=-10; x-=10) { // Spalte
        ledIndex++;
        if (mode == PICKING_MODE) {
          glColor3ub(0, 0, ledIndex*8);
        } else {
          iLED = animation[currentFrame][ledIndex-1];
          glMaterialfv(GL_FRONT, GL_AMBIENT, (iLED == 1 ? ledOnMaterial : ledOffMaterial));
          glMaterialfv(GL_FRONT, GL_DIFFUSE, (iLED == 1 ? ledOnMaterial : ledOffMaterial));
        }

        glPushMatrix();
        glTranslatef(x, y, z-0.8);
        glutSolidSphere(1, 16, 16);
        gluCylinder(quadric, 1, 1, 1.9, 16, 2);

        glTranslatef(0, 0, 1.9);
        gluCylinder(quadric, 1.25, 1.25, 0.35, 16, 2);

        gluDisk(quadric, 0, 1.25, 16, 16);
        glTranslatef(0, 0, 0.35);
        gluDisk(quadric, 0, 1.25, 16, 16);

        glPopMatrix();
      }
}

void drawWires() {
  gint x, y;
  for (y=10; y>=-10; y-=10)
    for (x=10; x>=-10; x-=10) {
      glMaterialfv(GL_FRONT, GL_AMBIENT, ((x == 0 || y == 0) ? innerWireMaterial : wireMaterial));

      // Front
      glBegin(GL_LINES);
        glVertex3f(x, y, -10);
        glVertex3f(x, y, 10);
      glEnd();

      // Side
      glBegin(GL_LINES);
        glVertex3f(-10, x, y);
        glVertex3f(10, x, y);
      glEnd();

      // Top
      glBegin(GL_LINES);
        glVertex3f(x, -10, y);
        glVertex3f(x, 10, y);
      glEnd();
    }
}

void setScene() {
  glClear(GL_COLOR_BUFFER_BIT | GL_DEPTH_BUFFER_BIT);
  glMatrixMode(GL_PROJECTION);
  glLoadIdentity();
  gluPerspective(ZOOM_LEVEL, WINDOW_WIDTH/WINDOW_HEIGHT, 1.0, 350.0);

  glMatrixMode(GL_MODELVIEW);
  glLoadIdentity();
  gluLookAt(lookX, eyeAngle, lookZ, 0, 0, 0, 0, 1, 0);
}

// OpenGL Display function
void display(gboolean onlyForPicking) {
  glClearColor(0.0, 0.0, 0.0, 1.0);

  glEnable(GL_LIGHTING);
  glEnable(GL_DEPTH_TEST);
  glEnable(GL_LIGHT0);
  glLightfv(GL_LIGHT0, GL_AMBIENT, backgroundColor);

  setScene();

  if (onlyForPicking == TRUE) {
    glDisable(GL_DITHER);
    glDisable(GL_LIGHTING);

    drawLEDs(PICKING_MODE);

    glEnable(GL_LIGHTING);
    glEnable(GL_DITHER);
  } else {
    drawLEDs(RENDER_MODE);
    drawWires();
  }
}


void displayCurrentFrame() {
  int i;
  unsigned long frame = 0;
  for (i=0; i<27; ++i) {
    if (animation[currentFrame][i] == 1) frame |= (1 << i);
  }
  lc_setMode(MODE_ANIMATION_STOP);
  lc_setFrame(frame);
}

void uploadAnimation() {
  int i, j;
  unsigned long frame = 0;
  int delay = 0; // TODO: Implement the delay
  int skip = 1; // Skip

  // Maximum delay
  for (i=0; i<5; ++i) delay |= (1 << i);

  lc_setMode(MODE_ANIMATION_STOP);
  for (i=0; i<32; ++i) {
    frame = 0;
    for (j=0; j<27; ++j) {
      if (animation[i][j] == 1) frame |= (1 << j);
    }

    lc_saveFrame(frame, (i<animationLength ? delay : skip), i);
  }
  lc_setMode(MODE_ANIMATION_LOOP);
}