/**
 * GaGraph.java
 * J Scott Cameron
 *This is a swing component that you can drop into any
 *Swing panel or frame that will graph data for
 *a GeneticAlgorithm
 **/

import javax.swing.*;
import java.awt.*;
import java.awt.geom.*;
import java.awt.font.*;
import java.applet.*;
import Knapsack.*;
import GeneticAlgorithm.*;
import java.util.*;
import java.awt.event.*;
import java.io.*;

public class GaGraph extends JPanel
{
    
    Vector data;/*Vector that contains Genotypes of the
                 *best rated individuals from each
                 *generation*/
    Vector avgData;/*Vector that contains average ratings
                    * from each generation*/
    
    
    int w, h; /*width/height of panel*/
    
    /* initialize the graph with appropriate vectors */
    public GaGraph(Vector v,Vector v2)
    {
        data = v;
        avgData = v2;
    }
    
    /* update the data for the graph and repaint*/
    public void updateGraph(Vector v,Vector v2)
    {
        data = v;
        avgData = v2;
        repaint();
    }
    
    /* draw the graph */
    public void paintComponent(Graphics g) {
        super.paintComponent(g);
        setBackground(Color.white);
        w = getSize().width;
        h = getSize().height;
        
        /* give the graph a margin of 20 pixels */
        int graphBottom = h-20;
        int graphRight = w-20;
        
        Graphics2D g2;
        g2 = (Graphics2D) g;
        
        /* set Rendering state so that the graph is pretty
         * and anti-aliased */
        g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING,
        RenderingHints.VALUE_ANTIALIAS_ON);
        
        g2.setRenderingHint(RenderingHints.KEY_RENDERING,
        RenderingHints.VALUE_RENDER_QUALITY);
        
        /* find the best score and worst score as the range of the graph*/
        int bestScore = ((Genotype)data.get(data.size()-1)).getRating();
        if (bestScore < 0) bestScore =0;
        int worstScore = 0;
        for(int i =0 ;i<avgData.size();i++)
        {
            if(((Double)avgData.get(i)).doubleValue() < worstScore)
            {
                worstScore =  ((Double)avgData.get(i)).intValue();
            }
        }
        
        /* create font object for text*/
        FontRenderContext frc = g2.getFontRenderContext();
        Font f = new Font("Helvetica", 1, 8);
        
        /* find intervals for axis tick marks */
        int vInterval = ((bestScore-worstScore)/5);
        int vGraphInterval = (graphBottom-20)/5;
        int hInterval = (data.size()/10)+1;
        int hGraphInterval = (graphRight-20)/10;
        
        /* find the number of graph units each pixel is worth*/
        double vUnit = (((double)(graphBottom-20))/((double)(vInterval*5)));
        double hUnit = (((double)(graphRight-20))/((double)( hInterval*10)));
        
        g2.setColor(Color.black);
        
        /* draw Y-Axis */
        g2.draw(new Line2D.Float(20,20,20,graphBottom));
        /* draw ticks of Y-Axis */
        for(int i=0;i<6;i++)
        {
            g2.draw(new Line2D.Float(16,20+i*vGraphInterval,
                                     20,20+i*vGraphInterval));
            String s = new String(
                       (new Integer(worstScore +vInterval*(5-i))).toString());
            TextLayout textTl = new TextLayout(s, f, frc);
            Rectangle r= (textTl.getBounds()).getBounds();
            textTl.draw(g2,0,20+i*vGraphInterval+(r.height/2));
            
        }
        
        /*draw X-Axis */
        g2.draw(new Line2D.Float(20,graphBottom,graphRight,graphBottom));
        /*draw ticks of X-Axis */
        for(int i=0;i<11;i++)
        {
            g2.draw(new Line2D.Float(20+i*hGraphInterval,graphBottom,
                                     20+i*hGraphInterval,graphBottom+4));
            String s = new String((new Integer(hInterval*i)).toString());
            TextLayout textTl = new TextLayout(s, f, frc);
            Rectangle r= (textTl.getBounds()).getBounds();
            textTl.draw(g2,20+i*hGraphInterval-(r.width/2),graphBottom+10);
        }
        
        /* draw a horizontal line at y=0 */
        g2.setColor(Color.lightGray);
        g2.draw(new Line2D.Float(16,
                                 graphBottom-(int)(0-worstScore*vUnit),
                                 graphRight,
                                 graphBottom-(int)(0-worstScore*vUnit)));
        
        String s = "0";
        TextLayout textTl = new TextLayout(s, f, frc);
        Rectangle r= (textTl.getBounds()).getBounds();
        textTl.draw(g2,0,graphBottom-(int)(0-worstScore*vUnit));
        
        /* plot all the best values in red*/
        g2.setColor(Color.red);
        for(int i=0;i<data.size();i++)
        {
            Ellipse2D.Float point = new Ellipse2D.Float(20+(int)(i*hUnit),
            graphBottom-(int)((((Genotype)(data.get(i))).getRating()
                       -worstScore)*vUnit),3,3);
            g2.fill(point);
            
        }
        
        /* plot all the average values in blue */
        g2.setColor(Color.blue);
        for(int i=0;i<data.size();i++)
        {
            Ellipse2D.Float point = new Ellipse2D.Float(20+(int)(i*hUnit),
            graphBottom-(int)((((Double)avgData.get(i)).doubleValue()
                         -worstScore) *vUnit),3,3);
            g2.fill(point);
            
        }
    }
}