bigdataproject.ScatterPlot.java Source code

Introduction

Here is the source code for bigdataproject.ScatterPlot.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package bigdataproject;

/**
 *
 * @author raffaele
 */
import java.awt.RenderingHints;
import java.util.HashMap;
import java.util.List;
import javax.swing.JFrame;
import org.apache.commons.math3.ml.clustering.Cluster;
import org.apache.commons.math3.ml.clustering.DoublePoint;
import org.jfree.chart.ChartPanel;
import org.jfree.chart.JFreeChart;
import org.jfree.chart.axis.NumberAxis;

/**
 * A demo of the fast scatter plot.
 *
 */
public class ScatterPlot extends JFrame {

    /**
     * The data.
     */
    private float[][] data;
    List<Cluster<DoublePoint>> list;
    HashMap<Integer, double[][]> clusters;

    /**
     * Creates a new fast scatter plot.
     *
     * @param title the frame title.
     * @param samples the 2D matrix to plot.
     */
    public ScatterPlot(final String title, double[][] samples, List<Cluster<DoublePoint>> list) {

        super(title);
        this.list = list;
        convertToFloat(samples);
        listToHashMap();
        HashMap<Integer, float[][]> clusterData = HashMapDoubleToFloat();
        final NumberAxis domainAxis = new NumberAxis("X1");
        domainAxis.setAutoRangeIncludesZero(false);
        final NumberAxis rangeAxis = new NumberAxis("X2");
        rangeAxis.setAutoRangeIncludesZero(false);
        final CustomFastScatterPlot plot = new CustomFastScatterPlot(this.data, domainAxis, rangeAxis, clusterData);
        final JFreeChart chart = new JFreeChart(title, plot);
        //chart.setLegend(null);

        // force aliasing of the rendered content..
        chart.getRenderingHints().put(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);

        final ChartPanel panel = new ChartPanel(chart, true);
        panel.setPreferredSize(new java.awt.Dimension(1000, 540));
        panel.setMinimumDrawHeight(10);
        panel.setMaximumDrawHeight(2000);
        panel.setMinimumDrawWidth(20);
        panel.setMaximumDrawWidth(2000);
        setContentPane(panel);

    }

    private HashMap<Integer, float[][]> HashMapDoubleToFloat() {
        HashMap<Integer, float[][]> hashMapFloat = new HashMap<>();
        if (clusters != null) {
            for (Integer i : clusters.keySet()) {
                double[][] matrix = clusters.get(i);
                float[][] matrixFloat = convertToFloat2(matrix);
                hashMapFloat.put(i, matrixFloat);
            }
        }
        return hashMapFloat;
    }

    private void convertToFloat(double[][] matrix) {
        data = new float[matrix.length][matrix[0].length];
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[i].length; j++) {
                data[i][j] = (float) matrix[i][j];
            }
        }
    }

    float[][] convertToFloat2(double[][] matrix) {
        float[][] floatMatrix = new float[matrix.length][matrix[0].length];
        for (int i = 0; i < matrix.length; i++) {
            for (int j = 0; j < matrix[i].length; j++) {
                floatMatrix[i][j] = (float) matrix[i][j];
            }
        }
        return floatMatrix;
    }

    private void listToHashMap() {
        if (list != null) {
            int index = 0;
            clusters = new HashMap<>();
            for (Cluster<DoublePoint> c : list) {
                List cluster = c.getPoints();
                int size = cluster.size();
                double[][] clusterMatrix = new double[size][2];
                int i = 0;
                for (Object p : cluster) {
                    DoublePoint point = (DoublePoint) p;
                    clusterMatrix[i++] = point.getPoint();
                }
                clusters.put(index++, clusterMatrix);
            }
        }
    }

}