/* GpGraph.java
 * J Scott Cameron
 * 
 * This class is a Swing component which can be dropped
 * into any swing frame or panel to graph the performance
 * of a Genetic Program
 */

import javax.swing.*;
import java.awt.*;
import java.awt.geom.*;
import java.awt.font.*;
import java.applet.*;
import GPNodeContainer;
import GPPopulation.*;
import java.util.*;
import java.awt.event.*;
import java.io.*;

public class GpGraph extends JPanel
{
    /* vector of best GP's */ 
    Vector data;
    
    int w, h; /* width/height */
    
    /* parameterized constructor */
    public GpGraph(Vector v)
    {
        data = v;
        
    }
    
    /* updates graph */ 
    public void updateGraph(Vector v)
    {
        data = v;
        repaint();
    }
    
    /* paints 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 quality to high and turn anti-aliasing on */
        g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING,
        RenderingHints.VALUE_ANTIALIAS_ON);
        
        g2.setRenderingHint(RenderingHints.KEY_RENDERING,
        RenderingHints.VALUE_RENDER_QUALITY);
        
        /* find the best and worst score to set the range */
        double bestScore = ((GPNodeContainer)data.get(
                             data.size()-1)).getRating();
        if (bestScore < 0) bestScore =0;
        int worstScore = 0;
        for(int i =0 ;i<data.size();i++)
        {
            if(((GPNodeContainer)(data.get(i))).getRating() < worstScore)
            {
                worstScore =  (int)((GPNodeContainer)(
                                      data.get(i))).getRating();
            }
        }
                        
        /* create a font object for writing text */
        FontRenderContext frc = g2.getFontRenderContext();
        Font f = new Font("Helvetica", 1, 8);
        
        /* find the interval for tick marks on the Y-Axis */
        double vInterval = ((bestScore-worstScore)/5);
        int vGraphInterval = (graphBottom-20)/5;
        
        /* find the graphunits/pixel */
        double vUnit = (((double)(graphBottom-20))/((double)(vInterval*5)));
        
        g2.setColor(Color.black);
        
        /* draw the Y axis*/
        g2.draw(new Line2D.Float(20,20,20,graphBottom));
        
        /* draw tick marks on the 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((int)(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));
            
        }
        
        /* find the interval for tick marks on the x-axis */
        int hInterval = (data.size()/10)+1;
        int hGraphInterval = (graphRight-20)/10;
        
        /* find the horizontal units/pixel */
        double hUnit = (((double)(graphRight-20))/((double)( hInterval*10)));
        
        /* draw the x-axis*/
        g2.draw(new Line2D.Float(20,graphBottom,graphRight,graphBottom));
        
        /* draw the tick marks on the 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 like at y=0 */
        g2.setColor(Color.lightGray);
        g2.draw(new Line2D.Float(16,graphBottom-(int)(0-worstScore*vUnit),
                                 graphRight,
                                 graphBottom-(int)(0-worstScore*vUnit)));
        
        
        /* plot the fitness of the best GP's */
        g2.setColor(Color.red);
        for(int i=0;i<data.size();i++)
        {
            Ellipse2D.Float point = new Ellipse2D.Float(20+(int)(i*hUnit),
            graphBottom-(int)((((GPNodeContainer)
                        (data.get(i))).getRating()-worstScore)*vUnit),3,3);
            g2.fill(point);
            
        }   
    }    
}