tech.tablesaw.plotting.fx.FxPareto Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tablesaw-plot Show documentation
There is a newer version: 0.24.4
package tech.tablesaw.plotting.fx;

import javafx.collections.FXCollections;
import javafx.scene.chart.BarChart;
import javafx.scene.chart.CategoryAxis;
import javafx.scene.chart.NumberAxis;
import javafx.scene.chart.XYChart;
import javafx.scene.layout.Background;
import tech.tablesaw.api.CategoryColumn;
import tech.tablesaw.api.IntColumn;
import tech.tablesaw.api.NumericColumn;
import tech.tablesaw.api.ShortColumn;
import tech.tablesaw.api.Table;

import java.util.ArrayList;
import java.util.List;


/**
 *
 */
public class FxPareto extends FxBuilder {

    public static BarChart chart(
            String title,
            CategoryColumn x,
            NumericColumn y) {

        final CategoryAxis xAxis = new CategoryAxis();
        final NumberAxis yAxis = new NumberAxis();
        xAxis.setLabel(x.name());
        yAxis.setLabel(y.name());

        Table t = Table.create("", x, y);
        t = t.sortDescendingOn(y.name());

        final BarChart bar = new BarChart<>(xAxis, yAxis);

        bar.setTitle(title);

        List> d2 = new ArrayList<>(x.size());

        for (int i = 0; i < x.size(); i++) {
            d2.add(new XYChart.Data<>(t.categoryColumn(0).get(i), t.nCol(1).getFloat(i)));
        }
        XYChart.Series series1
                = new XYChart.Series<>(FXCollections.observableList(d2));

        series1.setName(y.name());

        bar.setLegendVisible(false);
        bar.setCategoryGap(0.0);
        bar.setBarGap(0.1);
        bar.setBackground(Background.EMPTY);
        bar.setVerticalGridLinesVisible(false);

        bar.getData().add(series1);
        return bar;
    }

    public static BarChart chart(String title, IntColumn categoryColumn, NumericColumn numericColumn) {

        Table t = Table.create("", categoryColumn, numericColumn);
        t = t.sortDescendingOn(numericColumn.name());

        final CategoryAxis categoryAxis = getCategoryAxis(t.categoryColumn(0));
        final NumberAxis numberAxis = getNumberAxis(t.numericColumn(1));

        final BarChart barChart = getBarChart(title, categoryAxis, numberAxis);
        List> data = new ArrayList<>(categoryColumn.size());

        for (int i = 0; i < categoryColumn.size(); i++) {
            data.add(new XYChart.Data<>(categoryColumn.getString(i), numericColumn.getFloat(i)));
        }

        barChart.getData().add(getSeries(numericColumn, data));
        return barChart;
    }

    public static BarChart chart(String title, ShortColumn categoryColumn, NumericColumn
            numericColumn) {

        Table t = Table.create("", categoryColumn, numericColumn);
        t = t.sortDescendingOn(numericColumn.name());

        final CategoryAxis categoryAxis = getCategoryAxis(t.categoryColumn(0));
        final NumberAxis numberAxis = getNumberAxis(t.numericColumn(1));

        final BarChart barChart = getBarChart(title, categoryAxis, numberAxis);
        List> data = new ArrayList<>(categoryColumn.size());

        for (int i = 0; i < categoryColumn.size(); i++) {
            data.add(new XYChart.Data<>(categoryColumn.getString(i), numericColumn.getFloat(i)));
        }

        barChart.getData().add(getSeries(numericColumn, data));
        return barChart;
    }

}