geneticalgorithm.gui.view.VGraphic.java Source code

Introduction

Here is the source code for geneticalgorithm.gui.view.VGraphic.java

Source

/**
 * Copyright (c) 2014 Sal Pia <sauljabin@gmail.com>.
 * 
 * This file is part of GeneticAlgorithm.
 * 
 * GeneticAlgorithm is licensed under The MIT License.
 * For full copyright and license information please see the LICENSE file.
 */

package geneticalgorithm.gui.view;

import geneticalgorithm.util.HelperImage;

import java.awt.BorderLayout;
import java.awt.Font;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.IOException;

import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.SwingConstants;

import net.miginfocom.swing.MigLayout;

import org.jfree.chart.ChartFactory;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.plot.PlotOrientation;
import org.jfree.data.xy.XYSeries;
import org.jfree.data.xy.XYSeriesCollection;

/**
 * 
 * @author Saul Pina - sauljp07@gmail.com
 */
public class VGraphic extends JFrame {

    private static final long serialVersionUID = -7690162952744313595L;

    private XYSeriesCollection dataset;
    private JFreeChart chart;
    private XYSeries offline;
    private XYSeries average;
    private XYSeries online;

    private JLabel lblOffline;
    private JLabel lblOnline;
    private JLabel lblAverage;

    public VGraphic(String title, String xLabel, String yLabel) {
        setSize(800, 600);
        setLocationRelativeTo(this);
        setTitle(title);

        addWindowListener(new WindowAdapter() {
            @Override
            public void windowClosing(WindowEvent arg0) {
                dispose();
            }
        });
        setLayout(new BorderLayout());

        average = new XYSeries("Average");
        offline = new XYSeries("Offline");
        online = new XYSeries("Online");

        dataset = new XYSeriesCollection();
        dataset.addSeries(average);
        dataset.addSeries(offline);
        dataset.addSeries(online);

        chart = ChartFactory.createXYLineChart(title, xLabel, yLabel, dataset, PlotOrientation.VERTICAL, true, true,
                false);
        add(new ChartPanel(chart), BorderLayout.CENTER);

        JPanel south = new JPanel(new MigLayout());

        lblAverage = new JLabel("");
        lblAverage.setHorizontalTextPosition(SwingConstants.LEFT);
        lblAverage.setFont(new Font("ARIAL", Font.BOLD, 26));
        JLabel lblAverageTitle = new JLabel("Average: ");
        lblAverageTitle.setFont(new Font("ARIAL", Font.BOLD, 26));
        south.add(lblAverageTitle);
        south.add(lblAverage, "wrap");

        lblOffline = new JLabel("");
        lblOffline.setHorizontalTextPosition(SwingConstants.LEFT);
        lblOffline.setFont(new Font("ARIAL", Font.BOLD, 26));
        JLabel lblOfflineTitle = new JLabel("Offline: ");
        lblOfflineTitle.setFont(new Font("ARIAL", Font.BOLD, 26));
        south.add(lblOfflineTitle);
        south.add(lblOffline, "wrap");

        lblOnline = new JLabel("");
        lblOnline.setHorizontalTextPosition(SwingConstants.LEFT);
        lblOnline.setFont(new Font("ARIAL", Font.BOLD, 26));
        JLabel lblOnlineTitle = new JLabel("Online: ");
        lblOnlineTitle.setFont(new Font("ARIAL", Font.BOLD, 26));
        south.add(lblOnlineTitle);
        south.add(lblOnline, "wrap");

        add(south, BorderLayout.SOUTH);
    }

    public void addPoint(int t, double vAverage, double vOffLine, double vOnLine) {
        average.add(t, vAverage);
        offline.add(t, vOffLine);
        online.add(t, vOnLine);
        lblAverage.setText(String.format("%.15f", vAverage));
        lblOffline.setText(String.format("%.15f", vOffLine));
        lblOnline.setText(String.format("%.15f", vOnLine));
    }

    public void exportImage(String path) throws IOException {
        HelperImage.writeImage(chart.createBufferedImage(800, 600), path);
    }

}