/*
 * Main.java
 *
 * Created on March 27, 2007, 5:53 PM
 */

package spldata;

import java.awt.AlphaComposite;
import java.awt.Color;
import java.awt.Dimension;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.RenderingHints;
import java.awt.event.ComponentEvent;
import java.awt.event.ComponentListener;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.awt.geom.Ellipse2D;
import java.io.File;
import java.io.RandomAccessFile;
import java.nio.ByteBuffer;
import java.nio.ByteOrder;
import java.nio.CharBuffer;
import java.nio.channels.FileChannel;
import java.util.Arrays;
import java.util.Comparator;
import javax.swing.JFrame;
import javax.swing.JPanel;

public class Main extends JPanel
{
  int numTop = 10;
  public Category[] cs = new Category[numTop]; //tracking top 20 categories
  int biggest = 0;
  
  
  public Category[] fillCategories()
  {
    Category[] cats = new Category[numTop];
    
    int i = 0;
    
    //cats[i++] = new Category(800, "782", "Vocal Music");
    //cats[i++] = new Category(759, "741", "Drawing & Drawings");
    //cats[i++] = new Category(807, "789", "Instrumental Music");
    //cats[i++] = new Category(799, "781", "Musical Forms");
    //cats[i++] = new Category(659, "641", "Food & Drink");
    //cats[i++] = new Category(802, "784", "Instrumental Ensembles");
    cats[i++] = new Category(631, "613", "Promotion of Health");
    cats[i++] = new Category(416, "398", "Folklore");
    cats[i++] = new Category(935, "917", "North America");
    cats[i++] = new Category(814, "796", "Athletics");
    cats[i++] = new Category(809, "791", "Public Performances");
    cats[i++] = new Category(991, "973", "US History");
    cats[i++] = new Category(764, "746", "Textile Arts");
    cats[i++] = new Category(932, "914", "Europe");
    cats[i++] = new Category(634, "616", "Diseases");
    cats[i++] = new Category(803, "785", "Chamber Music");
    
    return cats;
  }
  
  
  public void setCategoryDeweyData(Category c)
  {
    switch(c.index)
    {
      case 807:
        c.deweyNum = "789";
        c.deweyTitle = "Instrumental Music";
        break;
      case 799:
        c.deweyNum = "781";
        c.deweyTitle = "Musical Forms";
        break;
      case 659:
        c.deweyNum = "641";
        c.deweyTitle = "Food & Drink";
        break;
      case 802:
        c.deweyNum = "784";
        c.deweyTitle = "Instrumental Ensembles";
        break;
      case 631:
        c.deweyNum = "613";
        c.deweyTitle = "Promotion of Health";
        break;
      case 416:
        c.deweyNum = "398";
        c.deweyTitle = "Folklore";
        break;
      case 935:
        c.deweyNum = "917";
        c.deweyTitle = "North America";
        break;
      case 814:
        c.deweyNum = "796";
        c.deweyTitle = "Athletics";
        break;
      case 991:
        c.deweyNum = "973";
        c.deweyTitle = "US History";
        break;
      case 932:
        c.deweyNum = "914";
        c.deweyTitle = "Europe";
        break;
      case 809:
        c.deweyNum = "791";
        c.deweyTitle = "Public Performances";
        break;
      default:
        break;
    }
    
    return;
  }
  
  public boolean useCateogryIndex(int idx)
  {
    if (idx == 800 || idx == 759 || idx == 807 || idx == 799 || idx == 659 || idx == 802
            || idx == 631 || idx == 416 || idx == 935 || idx == 814 || idx == 809
            || idx == 991 || idx == 764 || idx == 932 || idx == 634 || idx == 803
            )
    {
      return true;
    }
    return false;
  }
  
  public Category[] gatherWeekData(CharBuffer ibuf, Category[] cats)
  {
    for (int hour = 0; hour < 168; hour++)
    {
      System.out.println("\thour " + hour);
      for (int cell = 0; cell < 1017; cell++)
      {
        char b = ibuf.get((hour * 3000) + cell + 4);
        {
          //System.out.println("b("+cell+") = "  + (int)b);
          if (cats[cell] == null)
          {
            cats[cell] = new Category(cell);
          }
          if (useCateogryIndex(cell))
          {
            cats[cell].count += b;
          }
        }
      }
      
      /*
      for (int cell = 0; cell < 1017; cell++)
      {
        System.out.println("cell["+cell+"].index("+cats[cell].index+" = " + cats[cell].count);
      }
       */
    }
    return cats;
  }
  
  public Category[] gatherYearData(CharBuffer ibuf, Category[] cats)
  {
    for (int day = 0; day < 365; day++)
    {
      System.out.println("\tday " + day);
      for (int cell = 0; cell < 1017; cell++)
      {
        char b = ibuf.get((day * 3000) + cell + 4);
        {
          //System.out.println("b("+cell+") = "  + (int)b);
          if (cats[cell] == null)
          {
            cats[cell] = new Category(cell);
          }
          if (cell > 17)
          {
            cats[cell].count += b;
          }
        }
      }
      
      /*
      for (int cell = 0; cell < 1017; cell++)
      {
        System.out.println("cell["+cell+"].index("+cats[cell].index+" = " + cats[cell].count);
      }
       */
    }
    return cats;
  }
  
  public Category[] averageData(Category[] cats, int divisor)
  {
    for (int cell = 0; cell < cats.length; cell++)
    {
      System.out.println("\tcell["+cell+"].index("+cats[cell].deweyTitle+") = " + cats[cell].count);
      
      for (int hour = 0; hour < 24; hour++)
      {
        cats[cell].hours[hour] /= (17 * 7);
        if (cats[cell].hours[hour] > biggest)
        {
          biggest = cats[cell].hours[hour];
        }
        
        System.out.println("hour " + hour + ": " + cats[cell].hours[hour]);
      }
    }
    
    return cats;
  }
  
  public void sortCategoriesByRankForHour(Category[] cats, final int hour)
  {
    Arrays.sort(cats, new Comparator()
    {
      public int compare(Object a, Object b)
      {
        Category c1 = (Category)a;
        Category c2 = (Category)b;
        
        //     System.out.println("c1 (index = " + c1.index + " count = "
        //            + c1.count + ")> vs <c2 (index = " + c2.index + " count = " + c2.count);
        
        if (c1.hours[hour] < c2.hours[hour])
        {
          return 1;
        }
        else if (c1.hours[hour] > c2.hours[hour])
        {
          return -1;
        }
        else
        {
          return 0;
        }
      }
    });
  }
  
  public Category[] rankData(Category[] cats)
  {
    //sortCategories(cats);
    
    for (int i = 0; i < 24; i++)
    {
      sortCategoriesByRankForHour(cats, i);
      
      for (int ii = 0; ii < cats.length; ii++)
      {
        cats[ii].ranksByHour[i] = ii;
      }
    }
    
    //end up sorted by FIRST hour-- ie 8am
    sortCategoriesByRankForHour(cats, 8);
    
    return cats;
  }
  
  public void setBiggest(Category[] cats)
  {
    for (int cell = 0; cell < numTop; cell++)
    {
      //System.out.println("cell["+cell+"].index("+cs[cell].index+" = " + cs[cell].count);
      for (int h = 0; h < 24; h++)
      {
        int hc = cats[cell].hours[h];
        if (hc > biggest)
        {
          biggest = hc;
        }
        //System.out.println("month " + m + ": " + mc);
      }
    }
  }  
  
  public Main(String filename, String type)
  {
    Category[] cats = null; // = new Category[1017];
    
    if (type.equals("week"))
    {
      cats = fillCategories();
    }
    
    ByteBuffer buf = null;
    
    for (int num = 0; num < 17; num++) //loop through 20 weeks
    {
      int weeknum = 86 + num;
      String fname;
      if (weeknum < 100)
      {
        fname = "data/week-00"+weeknum+".dat";
      }
      else
      {
        fname = "data/week-0"+weeknum+".dat";
      }
      
      try
      {
        File file = new File(fname);
        FileChannel roChannel = new RandomAccessFile(file, "r").getChannel();
        buf = roChannel.map(FileChannel.MapMode.READ_ONLY, 0, (int)roChannel.size());
      }
      catch(Exception e)
      {
        e.printStackTrace();
        System.exit(0);
      }
      
      buf.order(ByteOrder.LITTLE_ENDIAN); //windows
      CharBuffer ibuf = buf.asCharBuffer();
      
      for (int i = 0; i < cats.length; i++) //loop through each category
      {
        int idx = cats[i].index;
        
        for (int hour = 0; hour < 168; hour++)
        {
          System.out.println("weeknum = " + weeknum + " hour = " + hour);
          char b = ibuf.get((hour * 3000) + idx + 4);
          
          cats[i].hours[hour % 24] += (int) b;
        }
      }
    }
    
    //cats = averageData(cats, (17 * 7)); ////divide by weeks * days to get average
    //sort by most popular categories
    //cats = rankData(cats);
    
    setBiggest(cats);
    System.arraycopy(cats, 0, cs, 0, numTop);
    
    //System.exit(0);
    /*
    else if (type.equals("year"))
    {
      cats = new Category[1017];
     
      cats = gatherYearData(ibuf, cats);
    }
     
    //sort by most popular categories
    sortCategories(cats);
    System.arraycopy(cats, 0, cs, 0, 20);
     
    if (type.equals("week"))
    {
      for (int cell = 0; cell < 20; cell++)
      {
        setCategoryDeweyData(cs[cell]);
        System.out.println("cell["+cell+"].index("+cs[cell].index+" = " + cs[cell].count);
      }
     
      //System.exit(0);
    }
     
    for (int cell = 2; cell < 20; cell++)
    {
      setCategoryDeweyData(cs[cell]);
      System.out.println("cell["+cell+"].index("+cs[cell].index+" = " + cs[cell].count);
     
      int idx = cs[cell].index;
     
      for (int day = 0; day < 365; day++)
      {
        char b = ibuf.get((day * 3000) + idx + 4);
     
        if (day <= 30)
        {
          cs[cell].months[0] += (int) b;
        }
        else if (day <= 61)
        {
          cs[cell].months[1] += (int) b;
        }
        else if (day <= 91)
        {
          cs[cell].months[2] += (int) b;
        }
        else if (day <= 122)
        {
          cs[cell].months[3] += (int) b;
        }
        else if (day <= 153)
        {
          cs[cell].months[4] += (int) b;
        }
        else if (day <= 183) //Sept
        {
          cs[cell].months[5] += (int) b;
        }
        else if (day <= 214)
        {
          cs[cell].months[6] += (int) b;
        }
        else if (day <= 244)
        {
          cs[cell].months[7] += (int) b;
        }
        else if (day <= 275) //Dec
        {
          cs[cell].months[8] += (int) b;
        }
        else if (day <= 306)
        {
          cs[cell].months[9] += (int) b;
        }
        else if (day <= 334) //Feb
        {
          cs[cell].months[10] += (int) b;
        }
        else if (day <= 365)
        {
          cs[cell].months[11] += (int) b;
        }
      }
    }
     
     
     
    for (int cell = 0; cell < 20; cell++)
    {
      //System.out.println("cell["+cell+"].index("+cs[cell].index+" = " + cs[cell].count);
      for (int m = 0; m < 12; m++)
      {
        int mc = cs[cell].months[m];
        if (mc > biggest)
        {
          biggest = mc;
        }
        //System.out.println("month " + m + ": " + mc);
      }
    }
     
    System.out.println("biggest = " + biggest);
     */
    
//System.out.println("ibuf capacity = " + ibuf.capacity());
    
  }
  
  public void sortCategories(Category[] cats)
  {
    Arrays.sort(cats, new Comparator()
    {
      public int compare(Object a, Object b)
      {
        Category c1 = (Category)a;
        Category c2 = (Category)b;
        
        //     System.out.println("c1 (index = " + c1.index + " count = "
        //            + c1.count + ")> vs <c2 (index = " + c2.index + " count = " + c2.count);
        
        if (c1.count < c2.count)
        {
          return 1;
        }
        else if (c1.count > c2.count)
        {
          return -1;
        }
        else
        {
          return 0;
        }
      }
    });
  }
  
  public Color getColorFromNum(int num)
  {
    switch(num)
    {
      case 0:
        return new Color(49, 65, 24);
      case 1:
        return new Color(24, 65, 33);
      case 2:
        return new Color(255, 36, 8);
      case 3:
        return new Color(57, 85, 24);
      case 4:
        return new Color(24, 85, 49);
      case 5:
        return new Color(115, 36, 0);
      case 6:
        return new Color(99, 130, 41);
      case 7:
        return new Color(49, 130, 74);
      case 8:
        return new Color(173, 60, 0);
      case 9:
        return new Color(140, 195, 57);
      case 10:
        return new Color(66, 190, 107);
      case 11:
        return new Color(255, 93, 8);
      case 12:
        return new Color(165, 207, 90);
      case 13:
        return new Color(107, 211, 148);
      case 14:
        return new Color(255, 121, 49);
      case 15:
        return new Color(255, 223, 8);
      default:
        return Color.GRAY;
    }
  }
  
  public String getMonthStringFromNum(int num)
  {
    switch(num)
    {
      case 0:
        return "april";
      case 1:
        return "may";
      case 2:
        return "june";
      case 3:
        return "july";
      case 4:
        return "august";
      case 5:
        return "september";
      case 6:
        return "october";
      case 7:
        return "november";
      case 8:
        return "december";
      case 9:
        return "january";
      case 10:
        return "february";
      case 11:
        return "march";
      default:
        return "bad month!";
    }
  }
  
  
  public String getHourStringFromNum(int num)
  {
    switch(num)
    {
      case 0:
        return "8am";
      case 1:
        return "9am";
      case 2:
        return "10am";
      case 3:
        return "11am";
      case 4:
        return "12am";
      case 5:
        return "1pm";
      case 6:
        return "2pm";
      case 7:
        return "3pm";
      case 8:
        return "4pm";
      case 9:
        return "5pm";
      case 10:
        return "6pm";
      case 11:
        return "7pm";
      case 12:
        return "8pm";
      default:
        return "bad hour";
    }
  }
  public void paintComponent(Graphics g)
  {
    boolean useRank = false;
    int circlesize = 8;
    
    Graphics2D g2 = (Graphics2D) g;
    
    g2.setColor(Color.WHITE);
    g2.fillRect(0,0, getWidth(), getHeight());
    
    g2.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER, 0.7f));
    g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
    g2.setFont(new Font("Arial", Font.PLAIN, 12));
    
    int leftInset = 100;
    int dataSliceWidth = (getWidth() - leftInset) / 13;
    
    int topInset = 80;
    int bottomInset = 80;
    int dataHeight = (getHeight() - (topInset + bottomInset));
    int dataSliceHeight = dataHeight / numTop;
    
    /* debug
     g2.setColor(Color.PINK);
    g2.fillRect(0,0,50,topInset);
    g2.fillRect(0,topInset + dataHeight,50,bottomInset);
    g2.setColor(Color.BLUE);
    g2.fillRect(20,topInset,50,dataHeight);
     */
    
    for (int ii = 0; ii < cs.length; ii++)
    {
      System.out.println("paint 1 : " + cs[ii].deweyTitle + " rank for 8am : " + cs[ii]);
      int prevx = 0;
      int prevy = 0;
      
      for (int i = 9; i <= 20; i++) //8am to 8pm
      {
        int x = leftInset + ((i - 8) * dataSliceWidth);
        int y;
        
        if (useRank)
        {
          int rank = cs[ii].ranksByHour[i];
          int pix = (rank) * dataSliceHeight;
          
          //double percent = (double)cs[ii].hours[i] / (double)biggest;
          //int pix = (int) (percent * dataHeight);
          y = topInset + pix;
        }
        else
        {//double percent = (double)cs[ii].months[i] / (double)biggest;
          double percent = (double)cs[ii].hours[i] / (double)biggest;
          int pix = (int) (percent * dataHeight);
          //y = getHeight() - bottomInset - pix;
          //y = topInset + dataHeight  - pix;
          y = topInset + dataHeight - pix;
        }
        
        if (!useRank && i == 9 || i == 20)
        {
          if (ii == 0)
          {
            g2.setColor(Color.DARK_GRAY);
            Ellipse2D.Double e = new Ellipse2D.Double(x - circlesize,y - circlesize,circlesize*2,circlesize*2);
            g2.fill(e);
          }
        }
        else
        {
          g2.setColor(getColorFromNum(ii));
          
          Ellipse2D.Double e = new Ellipse2D.Double(x - circlesize,y - circlesize,circlesize*2,circlesize*2);
          g2.fill(e);
        }
        
        g2.setColor(getColorFromNum(ii));
        
        if (i > 9) //9am
        {
          g2.drawLine(prevx, prevy, x, y);
        }
        prevx = x;
        prevy = y;
        
      }
    }
    
    /* useRank! */
    /*
    for (int ii = 0; ii < cs.length; ii++)
    {
      System.out.println("ii = " + ii);
      int y;
        if (useRank)
        {
          int rank = cs[ii].ranksByHour[8];
          int pix = (rank) * dataSliceHeight;
     
          //double percent = (double)cs[ii].hours[i] / (double)biggest;
          //int pix = (int) (percent * dataHeight);
          y = topInset + pix;
        }
      else
      {
        y = topInset + (dataSliceHeight * ii);
      }
      System.out.println(" BBB ii = " + ii);
      System.out.println("paint 1 : " + cs[ii].deweyTitle + " rank for 8am : " + cs[ii]);
     
      g2.setColor(getColorFromNum(ii));
      //g2.fillRect(0, y - 6, leftInset - 30, dataSliceHeight);
      g2.fillRect(0, y - circlesize, leftInset - 30, circlesize * 2 );
      //g2.drawRect(0, y - 6, leftInset - 30, dataSliceHeight);
      //g2.fillRect(getWidth() - 25, y - circlesize, getWidth() - 5, circlesize * 2 );
     
      g2.setColor(Color.BLACK);
      System.out.println(" CCC ii = " + ii);
     
      g2.drawString(cs[ii].deweyNum + ": " + cs[ii].deweyTitle, 5, y + circlesize - 4);
     
      if (useRank)
      {
      int spix = (ii) * dataSliceHeight;
      y = topInset + spix;
     
      g2.drawString("" + (ii + 1), getWidth() - 25, y + circlesize - 4);
      }
    }
     */
    
    if (!useRank)
    {
      for (int ii = 0; ii < cs.length; ii++)
      {
        g2.setColor(getColorFromNum(ii));
        
        int x = 10;
        int y = topInset + 20 + (20 * ii);
        Ellipse2D.Double e = new Ellipse2D.Double(x,y,circlesize*2,circlesize*2);
        g2.fill(e);
        
        g2.setColor(Color.BLACK);
        g2.drawString(cs[ii].deweyNum + ": " + cs[ii].deweyTitle, 30, y + circlesize + 5);
        
      }
      
      g2.setComposite(AlphaComposite.getInstance(AlphaComposite.SRC_OVER, 1.0f));
    
      g2.setColor(Color.BLACK);
      
      int inc = 100;
      for (int i = 0; i < biggest + 100; i+=100)
      {
        int pix = (int) (((double)dataHeight / (double)biggest ) * (double)i);
        
        g2.drawString("" + i, getWidth() - 45, topInset + dataHeight - pix + 5);
      }
    }
    
    
    for (int i = 1; i <= 12; i++)
    {
      int x = leftInset + (i * dataSliceWidth) - 8;
      
      g2.setRenderingHint(RenderingHints.KEY_TEXT_ANTIALIASING, RenderingHints.VALUE_TEXT_ANTIALIAS_ON);
      
      //String month = getMonthStringFromNum(i);
      String hour = getHourStringFromNum(i);
      g2.setColor(Color.BLACK);
      g2.drawString(hour, x - 3, getHeight() - bottomInset + 25);
    }
    
    g2.setFont(new Font("Arial", Font.BOLD, 24));
    if (useRank)
    {
      g2.drawString("Average Rank of Category by Hour: January 2007 through April 2007", 30,45);
    }
    else
    {
      //g2.drawString("Average Num Items Checked-out per Hour: January 2007 through April 2007", 30,45);
      g2.drawString("Total Num Items Checked-out per Hour: January 2007 through April 2007", 30,45);
    }
    System.out.println("biggest = " + biggest);
  }
  
  public static void main(String[] args)
  {
    
    JFrame jf = new JFrame();
    jf.setSize(new Dimension(400,300));
    jf.setLocation(0, 0);
   // jf.setVisible(true);
    
    jf.addWindowListener(new WindowAdapter()
    {
      public void windowClosing(WindowEvent e)
      {
        System.out.println("exit...");
        System.exit(0);
      }
    });
    
    jf.addComponentListener(new ComponentListener()
    {
      public void componentHidden(ComponentEvent e)
      {}
      
      public void componentMoved(ComponentEvent e)
      {
        //Main.makeBox = true;
        //RectangleFloat.makeRectangles = true;
      }
      public void componentResized(ComponentEvent e)
      {
        //Main.makeBox = true;
        //RectangleFloat.makeRectangles = true;
      }
      
      public void componentShown(ComponentEvent e)
      {
      }
    });
    
    JPanel mainPanel = new Main("data/week-0102.dat", "week");
    jf.add(mainPanel);
     jf.setVisible(true);
   
  }
}

class Category
{
  public int rank = 0;
  public int count = 0;
  public int index = 0;
  public int[] ranksByHour = new int[24];
  public int[] hours = new int[24];
  public int[] months = new int[12];
  public String deweyNum = "na";
  public String deweyTitle = "na";
  
  public Category(int index)
  {
    this.index = index;
  }
  public Category(int index, String deweyNum, String deweyTitle)
  {
    this.index = index;
    this.deweyNum = deweyNum;
    this.deweyTitle = deweyTitle;
  }
}