/**
 * 
 * Doing some stuff with video.  
 */

// port forwarding if you need it:
// ssh -L 3306:localhost:3306 rtwomey@192.168.0.183

import hypermedia.video.*;
import java.awt.Rectangle;
import de.bezier.data.sql.*;
import processing.video.*;

//Movie myMovie;
SteppedMovie myMovie;

Gaggle agents;

// ---- face tracking stuff ----
PImage img;
PImage frameimg;
PFont font;

int filenum=257;
int facenum=0;
int frame=0;
int id=0;

int startframe=67423;

float moviepos=519.8;
float lasttime=0;
boolean newframe=false;

// ---- interactivity ---- //
// timing
int last=0;
boolean paused=false;

String DETECTION_CASCADE=OpenCV.CASCADE_FRONTALFACE_DEFAULT;
//String DETECTION_CASCADE=OpenCV.CASCADE_FRONTALFACE_ALT_TREE;
//String DETECTION_CASCADE=OpenCV.CASCADE_PROFILEFACE;

//String inpath="/Volumes/tera 4/Assets/Video";
String inpath="/Volumes/Reservoir/Assets/Video";
//String fname="abby gas station";

ArrayList videoIDs;
int j=0;
String file_name, fname;

String outpath="/Users/rtwomey/Desktop/";

// ---- database for face information ---- //
MySQL msql;

String user="bob";
String pass="password";
String database="looking";
String table="faces";

int video_id=-1;

void setup() {
  //size(1280, 720, P2D);
  size(screen.width, screen.height);

  background(0);

  strokeWeight(2);
  noFill();

  // Font and text drawing setup
  font = loadFont("HelveticaNeue-48.vlw");
  textAlign(LEFT);
  //  textAlign(CENTER,CENTER);
  rectMode(CORNER);
  fill(255);
  textFont(font, 24);

  // connect do db for face-tracking
  msql = new MySQL( this, "localhost", database, user, pass );

  if ( !msql.connect() )
  {   
    // connection failed !
    println("connection failed!");
    exit();
  }

  retrieveVideoIDs();
  nextMovie();

  agents = new Gaggle(this, myMovie.width, myMovie.height );

  last=-1000;

  println(DETECTION_CASCADE);
  exit();
}

void retrieveVideoIDs() {
  // query faces table for faces with width > 100
  videoIDs = new ArrayList();

  msql.query("SELECT DISTINCT(id) FROM media WHERE filetype=\"video\"");
  while(msql.next()) {
    video_id=msql.getInt("id");
    videoIDs.add(video_id);
    println(video_id);
  }

  println(videoIDs.size()+" video files");
  if(videoIDs.size()<=0) exit();
}

void loadMovie() {
  video_id=(Integer)videoIDs.get(j);
  msql.query("SELECT * FROM media WHERE id="+video_id);
  while(msql.next()) {
    file_name = msql.getString("fname");
    fname=file_name;
    String path=msql.getString("path");

    print("  opening video_id "+video_id+" "+path+"/"+file_name+". ");

    // open movie file
    myMovie = new SteppedMovie(this, savePath(path+"/"+file_name));
    myMovie.precalcFrameTimes();
    myMovie.read();
    println("file is "+myMovie.getFrameCount()+" frames long.");
  }
}

void nextMovie() {
  j++;
  if(j>=videoIDs.size())j=0;
  loadMovie();
}

void doFaceTracking() {
  frame++;
  myMovie.loadPixels();
  frameimg = new PImage(myMovie.width, myMovie.height, RGB);

  arraycopy(myMovie.pixels, frameimg.pixels);
  frameimg.updatePixels();

  int count = agents.checkframe(frameimg);

  if(count>0) {
    println("frame: "+str(myMovie.getCurrentFrameNumber())+", "+count+" faces");
  } 
  else {
    //println("  no faces in frame.");
  }
}


//void movieEvent(Movie myMovie) {
//  //  if(!paused) {
//  //    lasttime=myMovie.time();
//  //    myMovie.read();
//  //    newframe=true;
//  //  }
//}

void draw() {
  if((!paused)) {
    // erase image and put curr video frame
    myMovie.read();
    myMovie.stepForward();
    if(myMovie.done()) {
    } 
    else {
      background(0);
      image(myMovie, 0, 0);
      fill(64, 64, 64);
      text(myMovie.time()+" s\n"+myMovie.getCurrentFrameNumber()+"\n"+myMovie.getFrameCount()+" frames\n"+DETECTION_CASCADE, 50, myMovie.height+50);
      int linepos=round(map(myMovie.getCurrentFrameNumber(), 0, myMovie.getFrameCount(), 0, width));
      stroke(128);
      line(linepos, height-50, linepos, height);
      noFill();
      doFaceTracking();
      newframe=false;
    } 

    agents.relax();
  }
}

void saveSnapShot() {
  //myMovie.save(savePath("/Users/rtwomey/Desktop/images/"+vidName));
  save("snapshot_"+str(myMovie.time()));
}

void mousePressed() {
  int fnum=round(float(mouseX)/float(width)*myMovie.getFrameCount());
  println("jumping to frame "+fnum);
  myMovie.gotoFrameNumber(fnum);
}

void keyPressed() {
  if(key==' ')
    paused = (paused == true) ? false : true;
  if(key=='s')
    saveSnapShot();
}

public void stop() {
  agents.stop();
  super.stop();
}