#include <GL/glew.h>
#include <GLFW/glfw3.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <pthread.h>
#include <sys/types.h>
#include <sys/stat.h>
#include <unistd.h>
#include <alsa/asoundlib.h>

#define STB_IMAGE_IMPLEMENTATION
#include "stb_image.h"
#define STB_IMAGE_WRITE_IMPLEMENTATION
#include "stb_image_write.h"

#define MIDICHK(stmt, msg) if((stmt) < 0) {puts("ERROR: "#msg); exit(1);}

int width = 1920;
int height = 1080;
GLFWwindow* window;
GLuint vertex;
GLuint fragment;
GLuint shaderProgram;
GLuint backbuffer;
int bars;

typedef struct Shad {
  GLuint id;
  char fragment[40];
  time_t mtime;
  int new;
} Shader;

Shader shader;

typedef struct Img {
  GLuint id;
  char path[40];
  float ratio;
  int new;
} Image;

Image images[8];

typedef struct Param {
  float value;
  int new;
} Parameter;

Parameter params[4];

static snd_seq_t *seq_handle;
static int in_port;

void midi_open(void) {
  MIDICHK(snd_seq_open(&seq_handle, "default", SND_SEQ_OPEN_INPUT, 0), "Could not open sequencer");
  MIDICHK(snd_seq_set_client_name(seq_handle, "Midi Listener"), "Could not set client name");
  MIDICHK(in_port = snd_seq_create_simple_port(seq_handle, "listen:in", SND_SEQ_PORT_CAP_WRITE|SND_SEQ_PORT_CAP_SUBS_WRITE, SND_SEQ_PORT_TYPE_APPLICATION), "Could not open port");
}

void screenshot() {
  unsigned char pixels[width*height*4];
  time_t current_time = time(NULL);
  char output_file[25];
  strftime(output_file, 25, "%Y-%m-%d_%H%M%S.png", localtime(&current_time));
  glReadPixels(0, 0, width, height, GL_RGBA, GL_UNSIGNED_BYTE, pixels);

  // Flip the image on Y
  int depth = 4;
  int row,col,z;
  stbi_uc temp;
  for (row = 0; row < (height>>1); row++) {
   for (col = 0; col < width; col++) {
      for (z = 0; z < depth; z++) {
         temp = pixels[(row * width + col) * depth + z];
         pixels[(row * width + col) * depth + z] = pixels[((height - row - 1) * width + col) * depth + z];
         pixels[((height - row - 1) * width + col) * depth + z] = temp;
      }
   }
  }
  if (0 == stbi_write_png(output_file, width, height, 4, pixels, width * 4)) { printf("can't create file %s",output_file); }
}

static void error_callback(int error, const char* description) { fputs(description, stderr); }

static void key_callback(GLFWwindow* window, int key, int scancode, int action, int mods) {
    if (key == GLFW_KEY_Q && action == GLFW_PRESS) glfwSetWindowShouldClose(window, GL_TRUE);
    else if (key == GLFW_KEY_SPACE && action == GLFW_PRESS) screenshot();
}

const char * readShader(char * path) {
  FILE *file = fopen(path, "r");
  if (!file) { fprintf(stderr,"Cannot read %s\n",path); }
  char source[100000];
  int res = fread(source,1,100000-1,file);
  source[res] = 0;
  fclose(file);
  const char *c_str = source;
  return c_str;
}

GLuint compileShader(const char * source, GLenum type) {
  GLuint sh = glCreateShader(type);
  glShaderSource(sh, 1, &source, NULL);
  glCompileShader(sh);
  int success;
  glGetShaderiv(sh, GL_COMPILE_STATUS, &success);
  if (!success) {
    char infoLog[512];
    glGetShaderInfoLog(sh, 512, NULL, infoLog);
    fprintf(stderr,"Shader compilation failed: %s\n",infoLog);
  }
  else { return sh; }
}

GLuint linkShader(GLuint vertex, GLuint fragment) {
  glUseProgram(0);
  glDeleteProgram(shader.id);
  shader.id = glCreateProgram();
  glAttachShader(shader.id, vertex);
  glAttachShader(shader.id, fragment);
  glLinkProgram(shader.id);
  int success;
  glGetProgramiv(shader.id, GL_LINK_STATUS, &success);
  if (!success) {
    char infoLog[512];
    glGetProgramInfoLog(shader.id, 512, NULL, infoLog);
    fprintf(stderr,"Shader linking failed: %s\n",infoLog);
  }
  glDetachShader(shader.id, vertex); 
  glDetachShader(shader.id, fragment); 
  glDeleteShader(vertex);
  glDeleteShader(fragment);
}

void createShader() {
  static const char vertex_src[] = {
    "#version 450 core\n"
    "const vec2 quadVertices[4] = { vec2(-1.0, -1.0), vec2(1.0, -1.0), vec2(-1.0, 1.0), vec2(1.0, 1.0) };\n"
    "void main() { gl_Position = vec4(quadVertices[gl_VertexID], 0.0, 1.0); }\n"
  };
  vertex = compileShader(vertex_src, GL_VERTEX_SHADER);
  fragment = compileShader(readShader(shader.fragment),GL_FRAGMENT_SHADER);
  linkShader(vertex,fragment);
  struct stat file_stat;
  stat(shader.fragment, &file_stat);
  shader.mtime = file_stat.st_mtime;
  glUseProgram(shader.id);
};

void createWindow() {
  glfwInit();
  glfwWindowHint(GLFW_CONTEXT_VERSION_MAJOR, 4);
  window = glfwCreateWindow(width,height, "", glfwGetPrimaryMonitor(), NULL);
  glfwMakeContextCurrent(window);
  glfwSetKeyCallback(window, key_callback);
  glfwSetErrorCallback(error_callback);
  glewInit();
}

void imageUniforms(int i) {
  glUniform1i(glGetUniformLocation(shader.id, "images")+i,i);
  glUniform1f(glGetUniformLocation(shader.id, "ratios")+i,images[i].ratio);
}

void readImage(int i) {

  images[i].new = 1;

  int w,h,comp;
  stbi_set_flip_vertically_on_load(1);
  unsigned char* pixels = stbi_load(images[i].path, &w, &h, &comp, STBI_rgb_alpha);

  images[i].ratio = (float) w/h;
  imageUniforms(i);

  glActiveTexture(GL_TEXTURE0+i);
  glGenTextures(1, &images[i].id);
  glBindTexture(GL_TEXTURE_2D,images[i].id);
  glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_LINEAR);
  glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);
  glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_BORDER);
  glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_BORDER);
  //glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_MIRRORED_REPEAT);
  //glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_MIRRORED_REPEAT);
  glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA, w, h, 0, GL_RGBA, GL_UNSIGNED_BYTE, pixels);
  stbi_image_free(pixels);
}

void updateImages(int force) {
  for (int i = 0; i<8; i++) {
    if (images[i].new || force) {
      readImage(i);
      printf("%s\n",images[i].path);
      images[i].new = 0;
    }
  }
}

void updateParams(int force) {
  glUniform1i(glGetUniformLocation(shader.id, "bars"), bars);
  for (int i = 0; i<4; i++) {
    if (params[i].new || force) {
      glUniform1f(glGetUniformLocation(shader.id, "params")+i, params[i].value);
      params[i].new = 0;
    }
  }
}

int ticks =0;
int running = 0;
int idx[2] = {0,1};
float vel[2] = {0,0};

void *readMidi() {
  midi_open();
  while(1) {
    snd_seq_event_t *ev = NULL;
    snd_seq_event_input(seq_handle, &ev);
    if (ev->type == SND_SEQ_EVENT_NOTEON) {
      int note = ev->data.note.note;
      int v = ev->data.note.velocity;
      int i,j;
      if (note < 48) {
        i = 0;
        j = note-36;
      }
      else {
        i = 1;
        j = note-48;
      }
      // only white keys
      if (j == 0) {
        idx[i] = 0; vel[i] = (float)v/127.0;
      }
      else if (j == 2) {
        idx[i] = 1; vel[i] = (float)v/127.0;
      }
      else if (j == 4) {
        idx[i] = 2; vel[i] = (float)v/127.0;
      }
      else if (j == 5) {
        idx[i] = 3; vel[i] = (float)v/127.0;
      }
      else if (j == 7) {
        idx[i] = 4; vel[i] = (float)v/127.0;
      }
      else if (j == 9) {
        idx[i] = 5; vel[i] = (float)v/127.0;
      }
      else if (j == 11) {
        idx[i] = 6; vel[i] = (float)v/127.0;
      }
      else if (j == 12) {
        idx[i] = 7; vel[i] = (float)v/127.0;
      }
      
    }
    else if (ev->type == SND_SEQ_EVENT_NOTEOFF) {
      int note = ev->data.note.note;
      int i,j;
      if (note < 48) {
        i = 0;
        j = note-36;
      }
      else {
        i = 1;
        j = note-48;
      }
      // only white keys
      if (j == 0) {
        idx[i] = 0; vel[i] = 0;
      }
      else if (j == 2) {
        idx[i] = 1; vel[i] = 0;
      }
      else if (j == 4) {
        idx[i] = 2; vel[i] = 0;
      }
      else if (j == 5) {
        idx[i] = 3; vel[i] = 0;
      }
      else if (j == 7) {
        idx[i] = 4; vel[i] = 0;
      }
      else if (j == 9) {
        idx[i] = 5; vel[i] = 0;
      }
      else if (j == 11) {
        idx[i] = 6; vel[i] = 0;
      }
      else if (j == 12) {
        idx[i] = 7; vel[i] = 0;
      }
    }
    else if(ev->type == SND_SEQ_EVENT_CONTROLLER) {
      int p = ev->data.control.param;
      int v = ev->data.control.value;
      if (p < 4) {
        params[p].value = (float)v/127;
        params[p].new = 1;
      }
      /*
        printf("[%d] Control:  %2x val(%2x)\n", ev->time.tick,
                                                ev->data.control.param,
                                                ev->data.control.value);
                                                */
    }
    else if(ev->type == SND_SEQ_EVENT_SONGPOS) {
      ticks = ev->data.control.value*24;
        printf("[%d] SPP:  %i val(%i)\n", ev->time.tick,
                                                ev->data.control.param,
                                                ev->data.control.value);
    }
    else if(ev->type == SND_SEQ_EVENT_START)
      running = 1;
    else if(ev->type == SND_SEQ_EVENT_STOP)
      running = 0;
    else if(ev->type == SND_SEQ_EVENT_CONTINUE)
      running = 1;
    else if(ev->type == SND_SEQ_EVENT_CLOCK) {
      if (running == 1) ticks++;
    }
    else
        printf("[%d] Unknown:  Unhandled Event Received\n", ev->time.tick);
  }
}


void *readStdin() {
  while (1) {
    char * str;
    size_t l = 80;
    getline(&str,&l,stdin);
    char *n = strtok(str," ");
    if (n[0] == 'i') {
      int i = atoi(strtok(NULL," "));
      strncpy(images[i].path, strtok(NULL,"\n"),40);
      images[i].new = 1;
    }
    else if (n[0] == 'f') {
      strncpy(shader.fragment, strtok(NULL,"\n"),40);
      shader.new = 1;
    }
    else if (n[0] == 'b') {
      bars = atof(strtok (NULL,"\n"));
      glfwSetTime(0);	
      printf("%i\n",bars);
    }
    else if (n[0] == 'p') {
      int i = atoi(strtok(NULL," "));
      params[i].value = atof(strtok (NULL,"\n"));
      params[i].new = 1;
    }
    else if (n[0] == 'q') {
      glfwSetWindowShouldClose(window, GL_TRUE);
    }
  }
}

void *watchShader() {
  while (1) {
    sleep(0.2);
    struct stat file_stat;
    stat(shader.fragment, &file_stat);
    if (file_stat.st_mtime > shader.mtime) shader.new = 1; 
  }
}

int main(int argc, char **argv) {

  createWindow();
  strncpy(shader.fragment,"shader.frag",40);
  shader.new = 1;
  unsigned int VAO;
  glGenVertexArrays(1, &VAO);
  glBindVertexArray(VAO);
  for (int i = 1; i<argc; i++) {
    strncpy(images[i-1].path, argv[i],40);
    images[i-1].new = 1;
  }
  glCreateTextures(GL_TEXTURE_2D,1,&backbuffer);
  glTextureStorage2D(backbuffer,1,GL_RGBA32F,width,height);
  glBindImageTexture(0,backbuffer, 0, GL_FALSE, 0, GL_READ_WRITE, GL_RGBA32F);
  
  //pthread_t stdin_t;
  //pthread_create(&stdin_t, NULL, readStdin, NULL);
  pthread_t midiin_t;
  pthread_create(&midiin_t, NULL, readMidi, NULL);
  pthread_t frag_t;
  pthread_create(&frag_t, NULL, watchShader, NULL);

  while (!glfwWindowShouldClose(window)) {
    //glUniform1f(glGetUniformLocation(shader.id, "time"),glfwGetTime());
    //glUniform1f(glGetUniformLocation(shader.id, "time"),glfwGetTime());
    glUniform1f(glGetUniformLocation(shader.id, "ticks"),(float)ticks);
    //printf("%i\n",note);
    //glUniform1i(glGetUniformLocation(shader.id, "program"),note-36);
    for (int i = 0; i<2; i++) { 
      glUniform1i(glGetUniformLocation(shader.id, "idx")+i,idx[i]);
      glUniform1f(glGetUniformLocation(shader.id, "vel")+i,vel[i]);
      }
    if (shader.new) {
      createShader();
      glUniform2f(glGetUniformLocation(shader.id, "resolution"),width,height); // important!!
      for (int i = 0; i<4; i++) { imageUniforms(i); }
      updateParams(1);
      shader.new = 0;
    }
    updateImages(0);
    updateParams(0);
    glDrawArrays(GL_TRIANGLE_STRIP, 0, 4);
    glfwSwapBuffers(window);
    glfwPollEvents();
  }

  //pthread_cancel(stdin_t);
  pthread_cancel(midiin_t);
  pthread_cancel(frag_t);
  glfwDestroyWindow(window);
  glfwTerminate();
 
  exit(EXIT_SUCCESS);
}