package chartb;

import java.util.Arrays;
import java.util.List;
import javafx.application.Application;
import javafx.beans.property.ReadOnlyObjectWrapper;
import javafx.beans.value.ObservableValue;
import javafx.collections.FXCollections;
import javafx.scene.Scene;
import javafx.scene.chart.AreaChart;
import javafx.scene.chart.NumberAxis;
import javafx.scene.chart.XYChart;
import javafx.scene.control.ContentDisplay;
import javafx.scene.control.Label;
import javafx.scene.control.TableCell;
import javafx.scene.control.TableColumn;
import javafx.scene.control.TableColumn.CellDataFeatures;
import javafx.scene.control.TableView;
import javafx.stage.Stage;
import javafx.util.Callback;
import javafx.util.Pair;

public class ChartB extends Application {

    public static void main(String[] args) {
        launch(args);
    }

    @Override public void start(Stage stage) throws Exception {
       TableView<Pair<String,List<Double>>> table = new TableView<Pair<String,List<Double>>>();
       TableColumn<Pair<String,List<Double>>, String> c1 = new TableColumn<Pair<String,List<Double>>, String>("A");
        c1.setCellValueFactory(new Callback<CellDataFeatures<Pair<String, List<Double>>, String>, ObservableValue<String>>() {
            @Override public ObservableValue<String> call(CellDataFeatures<Pair<String, List<Double>>, String> p) {
                return new ReadOnlyObjectWrapper<String>(p.getValue().getKey());
            }
        });
       TableColumn<Pair<String,List<Double>>, List<Double>> c2 = new TableColumn<Pair<String,List<Double>>, List<Double>>("B");
       c2.setMinWidth(400);
        c2.setCellFactory(new Callback<TableColumn<Pair<String, List<Double>>, List<Double>>, TableCell<Pair<String, List<Double>>, List<Double>>>() {
            @Override public TableCell<Pair<String, List<Double>>, List<Double>> call(TableColumn<Pair<String, List<Double>>, List<Double>> p) {
                return new SeriesChartTableCell();
            }
        });
        c2.setCellValueFactory(new Callback<TableColumn.CellDataFeatures<Pair<String, List<Double>>, List<Double>>, ObservableValue<List<Double>>>() {
            @Override public ObservableValue<List<Double>> call(CellDataFeatures<Pair<String, List<Double>>, List<Double>> p) {
                return new ReadOnlyObjectWrapper<List<Double>>(p.getValue().getValue());
            }
        });
       table.getColumns().addAll(c1,c2);
       table.setItems(FXCollections.observableArrayList(
               new Pair<String,List<Double>>("aaa", Arrays.asList(new Double[]{1d,6d,2d,7d,3d,8d,1d})),
               new Pair<String,List<Double>>("bbb", Arrays.asList(new Double[]{1d,6d,2d,7d,3d,8d,1d})),
               new Pair<String,List<Double>>("ccc", Arrays.asList(new Double[]{1d,6d,2d,7d,3d,8d,1d})),
               new Pair<String,List<Double>>("ddd", Arrays.asList(new Double[]{1d,6d,2d,7d,3d,8d,1d}))
               ));
        
        Scene scene = new Scene(table, 500,400);
        stage.setScene(scene);
        stage.show();
    }
    
    public static class SeriesChartTableCell extends TableCell<Pair<String,List<Double>>, List<Double>> {

        NumberAxis xAxis;
        NumberAxis yAxis;
        AreaChart<Number, Number> chart;

        public SeriesChartTableCell() {
            super();
            getStyleClass().add("table-chart-cell");
            xAxis = new NumberAxis();
            yAxis= new NumberAxis();
            chart = new AreaChart<Number, Number>(xAxis, yAxis);
            chart.getStyleClass().add("table-chart");
            chart.setLegendVisible(false);
            chart.setTitle(null);
            
            xAxis.setLabel(null);
            xAxis.setTickMarkVisible(false);
            xAxis.setMinorTickVisible(false);
            // UNCOMMENT ME TO CAUSE HANG
//            xAxis.setTickLabelsVisible(false);

            chart.setMinSize(400,80);
            chart.setPrefSize(400,80);
            setGraphic(chart);
            setContentDisplay(ContentDisplay.GRAPHIC_ONLY);
         }

        @Override public void updateItem(List<Double> item, boolean empty) {
            super.updateItem(item, empty);
            if (empty) {
                setGraphic(null);
            } else {
                XYChart.Series<Number, Number> salesSeries = new XYChart.Series<Number, Number>();
                for (int i = 0; i < item.size(); i++) {
                    salesSeries.getData().add(new XYChart.Data<Number, Number>(i, item.get(i)));
                }
                chart.getData().setAll(salesSeries);
                setContentDisplay(ContentDisplay.GRAPHIC_ONLY);
            }
        }

    }
    
}