/*
 * Created on 21-Jul-2005
 *
 * TODO To change the template for this generated file go to
 * Window - Preferences - Java - Code Style - Code Templates
 */

/**
 * @author Christian
 *
 * TODO To change the template for this generated type comment go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
Vector color_files;
Vector words;
PFont mainFont;
int nCol;
int nRow;
int totalConsidered;
Vector selectedWords;
float minSeg;
float maxSeg;
int minPhotos;
int maxPhotos;
float nPhotosFactW;
float nPhotosFactH;
float segmentationFactW;
float segmentationFactH;
String userInputStr;
Word lastRolled;
int bg_color;
color bg=color(221,221,204);
String[][] availableTags;

void setup(){
  size(700,480,P3D);
  words=new Vector();
  userInputStr="";
  String tag_lines[]=loadStrings("tags.txt");
  String[] tags_arr=tag_lines[0].split("_");
  int $av_tag_len=tags_arr.length;
  availableTags=new String[$av_tag_len][2];
  for(int i=0; i<$av_tag_len; i++){
    availableTags[i]=tags_arr[i].split(">");
  }
  String col_lines[] = loadStrings("colors.txt"); 
  parseColors(col_lines[0]);
  println("number words:"+words.size());

  // order colors inside
  Word $word;
  int len=words.size();
  for(int i=0; i<len; i++){
    $word=(Word) words.get(i);
    $word.orderColors();
  }
  System.out.println("Colors Ordered!");

  // assign related & find shared colors
  for(int i=0; i<len; i++){
    $word=(Word) words.get(i);
    $word.related_tags=cleanRelated($word.related_tags);
    $word.relatedVect=getRelated($word.related_tags);
    $word.findSharedColors();
  }



  //Selection of needed number by popularity
  selectedWords=new Vector();
  nCol=14;
  nRow=32;
  totalConsidered=nCol*nRow;
  System.out.println("total considered: "+totalConsidered);
  for(int i=0; i<len; i++){
    $word=(Word) words.get(i);
    int $len2=selectedWords.size();
    for(int j=0; j<$len2; j++){
      if($word.photos.size()>((Word) selectedWords.get(j)).photos.size()){
        selectedWords.add(j,$word);
        j=$len2;
      }
    }
    if(selectedWords.size()==$len2){
      selectedWords.add($word);
    }
  }
  while(selectedWords.size()>totalConsidered){
    selectedWords.remove(selectedWords.size()-1);
  }
  System.out.println("Number selected: "+selectedWords.size());
  minPhotos=((Word) selectedWords.get(totalConsidered-1)).photos.size();
  maxPhotos=((Word) selectedWords.get(0)).photos.size();
  System.out.println("minPhotos:"+minPhotos+" maxPhotos:"+maxPhotos);
  nPhotosFactW=(width-(width/nCol))/((maxPhotos-minPhotos)*1f);
  nPhotosFactH=(height-(height/nRow))/((maxPhotos-minPhotos)*1f);

  // find max and min Segementation
  minSeg=1000f;
  maxSeg=0f;
  for(int i=0; i<totalConsidered; i++){
    $word=(Word) selectedWords.get(i);
    if($word.segmentationRate<minSeg) minSeg=$word.segmentationRate;
    if($word.segmentationRate>maxSeg) maxSeg=$word.segmentationRate;
  }
  segmentationFactW=(width-(width/nCol))/(maxSeg-minSeg);
  segmentationFactH=(height-(height/nRow))/(maxSeg-minSeg);

  // order selected by hue
  Vector hueWords=new Vector();
  for(int i=0; i<totalConsidered; i++){
    $word=(Word) selectedWords.get(i);
    int $len2=hueWords.size();
    for(int j=0; j<$len2; j++){
      if($word.av_hue<((Word) hueWords.get(j)).av_hue){
        hueWords.add(j,$word);
        j=$len2;
      }
    }
    if(hueWords.size()==$len2){
      hueWords.add($word);
    }
  }
  // create columns based on hue
  Vector[] columnsVect=new Vector[nCol];
  for(int x=0; x<nCol; x++){
    columnsVect[x]=new Vector();
    for(int y=0; y<nRow; y++){
      columnsVect[x].add((Word) hueWords.get((x*nRow)+y));
    }
    //System.out.println("vect "+x+" size: "+columnsVect[x].size());
  }
  System.out.println("Columns created");
  // adjust columns based on brightness
  for(int x=0; x<nCol; x++){
    Vector tempVect=new Vector();
    for(int y=0; y<nRow; y++){
      $word=(Word) columnsVect[x].get(y);
      int $len2=tempVect.size();
      for(int i=0; i<$len2; i++){
        if($word.segmentationRate<((Word) tempVect.get(i)).segmentationRate){
          tempVect.add(i,$word);
          i=$len2;
        }
      }
      if(tempVect.size()==$len2){
        tempVect.add($word);
      }
    }
    columnsVect[x]=tempVect;
  }
  /*Word[][] wordsGrid=new Word[nCol][nRow];
   		for(int x=0; x<nCol; x++){
   			wordsGrid[x]=(Word[])columnsVect[x].toArray();
   		}*/
  mainFont= loadFont("HelveticaNeue-HeavyCond-40.vlw");
  textFont(mainFont,40);
  textAlign(CENTER);
  //textMode(SCREEN);
  //scale(0.5f);
  int dx=width/nCol;
  int dy=height/nRow;
  for(int x=0; x<nCol; x++){
    for(int y=0; y<nRow; y++){
      $word=(Word) columnsVect[x].get(y);
      //System.out.println($word.name.toUpperCase());
      $word.initMe(this,x*dx,y*dy,dx,dy);
      //fill($word.average_color);
      ///text($word.name.toUpperCase(), x*dx, y*dy);
    }
  }
  lastRolled=null;
  bg_color=0x000000;
  println("ended!");
  background(221,221,204);
  for(int i=0; i<totalConsidered; i++){
    ((Word) selectedWords.get(i)).arrangeBy((PApplet) this,3,nPhotosFactW,nPhotosFactH,minPhotos,maxPhotos,segmentationFactW,segmentationFactH,minSeg,maxSeg);
  }
}
String[] cleanRelated(String[] related_tags) {
  // TODO Auto-generated method stub
  Vector strings=new Vector();
  int $len=related_tags.length;
  int $wLen=words.size();
  for(int i=0; i<$len; i++){
    for(int j=0; j<$wLen; j++){
      if(((Word) words.get(j)).name.equals(related_tags[i])){
        strings.add(((Word) words.get(j)).name);
        j=$wLen;
      }
    }
  }
  $len=strings.size();
  String[] res=new String[$len];
  for(int i=0; i<$len; i++){
    res[i]=(String) strings.get(i);
  }
  return res;
}
//
//
void draw(){
  fill(221,221,204,100);
  noStroke();
  rect(0,0,width,height);
  Word $word;
  Word $rolledWord=null;
  noStroke();
  //
  //scale(0.5f);
  float $x,$y,$w,$h,$s;
  for(int i=0; i<totalConsidered; i++){
    $word=(Word) selectedWords.get(totalConsidered-i-1);
    $word.update();
    if($word._visible){
      $s=$word.wScale;
      $w=$word._width*$word.wScale;
      $h=$word._height*$word.wScale;
      $x=$word._x-($w*.5f);
      $y=$word._y-($h*.5f);
      // checkRoll
      if(mouseX>=$x && mouseX<$x+$w && mouseY>=$y && mouseY<$y+$h){
        $rolledWord=$word;
      }
      pushMatrix();
      translate($x,$y);
      scale($word.wScale);
      $word.drawMe(this);
      popMatrix();
    }
  }
  stroke(255);
  if($rolledWord!=null){
    //lastRolled=$rolledWord;
    fill(221,221,204,150);
    noStroke();
    rect(0,0,width,height);
    //strokeWeight(4);
    int $len=$rolledWord.relatedVect.size();
    for(int i=0; i<$len; i++){
      $word=(Word) $rolledWord.relatedVect.get(i);
      //System.out.println($word.name);
      if($word.isSelected){
        $s=$word.wScale;
        $w=$word._width*$word.wScale;
        $h=$word._height*$word.wScale;
        $x=$word._x-($w*.5f);
        $y=$word._y-($h*.5f);
        pushMatrix();
        translate($x,$y);
        scale($word.wScale);
        $word.drawMe(this);
        popMatrix();
      }
    }
    showLinks($rolledWord);
    $word=$rolledWord;
    $s=$word.wScale;
    $w=$word._width*$word.wScale;
    $h=$word._height*$word.wScale;
    $x=$word._x-($w*.5f);
    $y=$word._y-($h*.5f);
    pushMatrix();
    translate($x,$y);
    scale($word.wScale);
    $word.drawMe(this);
    popMatrix();
    //System.out.println("roll: "+$rolledWord.name);
    bg_color=$word.average_color;
  }
  else{
    bg_color=0x000000;
  }
}
//
//
void showLinks(Word $word){
  int $len=$word.relatedVect.size();
  if($len<1) return;
  Word $rWord;
  for(int i=0; i<$len; i++){
    $rWord=(Word) $word.relatedVect.get(i);
    if($rWord.isSelected && $rWord._visible){
      stroke($word.sharedColors[i]);
      float l=min(abs($word._x-$rWord._x),abs($word._y-$rWord._y));
      if($word._x==$rWord._x || $word._y==$rWord._y){
        line($word._x,$word._y,$rWord._x,$rWord._y);
      }else if($word._x>$rWord._x && $word._y>$rWord._y){
        //top left
        bezier($word._x, $word._y, $word._x, $word._y-l, $rWord._x+l, $rWord._y, $rWord._x, $rWord._y);
      }else if($word._x<$rWord._x && $word._y>$rWord._y){
        //top right
        bezier($word._x, $word._y, $word._x, $word._y-l, $rWord._x-l, $rWord._y, $rWord._x, $rWord._y);
      }else if($word._x>$rWord._x && $word._y<$rWord._y){
        //bott left
        bezier($word._x, $word._y, $word._x, $word._y+l, $rWord._x+l, $rWord._y, $rWord._x, $rWord._y);
      }else if($word._x<$rWord._x && $word._y<$rWord._y){
        //bott right
        bezier($word._x, $word._y, $word._x, $word._y+l, $rWord._x-l, $rWord._y, $rWord._x, $rWord._y);
      }
    }
  }
}
//
//
Vector getRelated(String[] $related_tags){
  Vector $relatedVect=new Vector();
  int $len=$related_tags.length;
  int $wLen=words.size();
  Word $word;
  for(int i=0; i<$len; i++){
    for(int j=0; j<$wLen; j++){
      $word=(Word) words.get(j);
      if($word.name.equals($related_tags[i])){
        $relatedVect.addElement($word);
        j=$wLen;
      }
    }
  }
  return $relatedVect;
}
//
//
void parseColors(String $superString){
  println("ss len: "+$superString.length());
  String[] colors_arr=$superString.split("_");
  int $len=colors_arr.length;
  println("nColors: "+$len);
  String[] $col_arr;
  String $colName;
  String $tagName;
  String $photoId;
  int $len2,$len3;
  for(int i=0; i<$len; i++){
    $col_arr=colors_arr[i].split(">");
    $colName=$col_arr[0];
    String[] $photos_arr=$col_arr[1].split("-");
    $len2=$photos_arr.length;
    for(int j=0; j<$len2; j++){
      String[] $photo_arr=$photos_arr[j].split(",");
      $len3=$photo_arr.length;
      if($len3>1){
        $photoId=$photo_arr[0];
        for(int k=1; k<$len3; k++){
          addTag($photo_arr[k], $colName,$photoId);
        }
      }
    }
  }
}
//
//
void addTag(String $name, String $colorname, String $photoId)
{
  //File $src = new File("data/related_tags/"+$name+".txt");
  //if(!$src.exists()) return;
  String[] $related_tags=new String[0];
  int $av_tag_len=availableTags.length;
  boolean isPresent=false;
  for(int i=0; i<$av_tag_len; i++){
    if($name.equals(availableTags[i][0])){
      isPresent=true;
      if(availableTags[i].length>1){
        $related_tags=availableTags[i][1].split(",");
      }
      i=$av_tag_len;
    }
  }
  if(!isPresent) return;
  int $wordsLen=words.size();
  Word $word;
  String $wordname;
  for(int i=0; i<$wordsLen; i++){
    $word = (Word) words.get(i);
    $wordname=$word.name;
    if($wordname.equals($name)){
      $word.addImage($colorname,$photoId);
      return;
    }
  }
  words.addElement(new Word($name,$colorname,$photoId,$related_tags));
}
//
//
void updateSelection(){
  for(int i=0; i<totalConsidered; i++){
    ((Word) selectedWords.get(i)).checkString(userInputStr);
  }
}
//
//
void keyPressed()
{
  if ( key == 's'){
    saveFrame("flickr_####");
  }
  if(keyCode>=112 && keyCode<112+12){
    int k=keyCode-112;
    for(int i=0; i<totalConsidered; i++){
      ((Word) selectedWords.get(i)).arrangeBy((PApplet) this,k,nPhotosFactW,nPhotosFactH,minPhotos,maxPhotos,segmentationFactW,segmentationFactH,minSeg,maxSeg);
    }
  }
  int $strLen=userInputStr.length();
  if(keyCode==BACKSPACE){
    if(userInputStr.length()>0){
      userInputStr=userInputStr.substring(0,userInputStr.length()-1);
    }
  }
  else if(keyCode >=48 && keyCode<=90){
    userInputStr=userInputStr+key;
  }
  //System.out.println("code: "+keyCode+" user: "+userInputStr);
  if(userInputStr.length()!=$strLen){
    updateSelection();
  }
}