import javafx.application.Application;
import javafx.beans.property.ObjectProperty;
import javafx.beans.property.SimpleObjectProperty;
import javafx.beans.property.SimpleStringProperty;
import javafx.beans.property.StringProperty;
import javafx.beans.value.ChangeListener;
import javafx.beans.value.ObservableValue;
import javafx.collections.FXCollections;
import javafx.collections.ObservableList;
import javafx.event.ActionEvent;
import javafx.event.EventHandler;
import javafx.scene.Node;
import javafx.scene.Parent;
import javafx.scene.Scene;
import javafx.scene.chart.AreaChart;
import javafx.scene.chart.CategoryAxis;
import javafx.scene.chart.NumberAxis;
import javafx.scene.chart.XYChart;
import javafx.scene.control.ContextMenuBuilder;
import javafx.scene.control.MenuItemBuilder;
import javafx.scene.control.TextArea;
import javafx.scene.control.TextAreaBuilder;
import javafx.scene.control.TreeItem;
import javafx.scene.control.TreeTableCell;
import javafx.scene.control.TreeTableColumn;
import javafx.scene.control.TreeTableColumnBuilder;
import javafx.scene.control.TreeTableRow;
import javafx.scene.control.TreeTableView;
import javafx.scene.control.cell.TextFieldTreeTableCell;
import javafx.scene.input.ContextMenuEvent;
import javafx.scene.input.MouseEvent;
import javafx.scene.layout.HBoxBuilder;
import javafx.stage.Stage;
import javafx.util.Callback;
import javafx.util.StringConverter;


/*
 * Copyright (c) 2013, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 */

/**
 *
 * @author akouznet
 */
public class TreeTableViewBug extends Application {
    
    private AreaChart<Double, Double> chart;
    private NumberAxis xAxis;
    private NumberAxis yAxis;
    private StringProperty sp = new SimpleStringProperty();
    private TextArea textArea;

    public Parent createContent() {
        xAxis = new NumberAxis();
        xAxis.setLabel("X Values");
        yAxis = new NumberAxis();
        yAxis.setLabel("Y Values");
        ObservableList<AreaChart.Series> areaChartData = FXCollections.observableArrayList(
                new AreaChart.Series("Series 1",FXCollections.observableArrayList(
                    new AreaChart.Data(0,4),
                    new AreaChart.Data(2,5),
                    new AreaChart.Data(4,4),
                    new AreaChart.Data(6,2),
                    new AreaChart.Data(8,6),
                    new AreaChart.Data(10,8)
                )),
                new AreaChart.Series("Series 2", FXCollections.observableArrayList(
                    new AreaChart.Data(0,8),
                    new AreaChart.Data(2,2),
                    new AreaChart.Data(4,9),
                    new AreaChart.Data(6,7),
                    new AreaChart.Data(8,5),
                    new AreaChart.Data(10,7)
                )),
                new AreaChart.Series("Series 3", FXCollections.observableArrayList(
                    new AreaChart.Data(0,2),
                    new AreaChart.Data(2,5),
                    new AreaChart.Data(4,8),
                    new AreaChart.Data(6,6),
                    new AreaChart.Data(8,9),
                    new AreaChart.Data(10,7)
                ))
        );
        chart = new AreaChart(xAxis, yAxis, areaChartData);
        
        XYDataVisualizer xyVis = new XYDataVisualizer(chart);
        textArea = TextAreaBuilder.create()
                .text("Click here at anytime to get dump")
                .onMouseClicked(new EventHandler<MouseEvent>() {

                    @Override
                    public void handle(MouseEvent t) {
                        StringBuilder sb = new StringBuilder();
                        for (XYChart.Series<Double, Double> series : chart.getData()) {
                            sb.append(series).append('\n');
                            for (XYChart.Data<Double, Double> data : series.getData()) {
                                sb.append("   ").append(data).append('\n');
                            }
                        }
                        textArea.setText(sb.toString());
                    }
                })
                .build();
        
        return HBoxBuilder.create()
                .children(chart, xyVis, textArea)
                .build();
    }
    
    public static void main(String[] args) {
        launch(args);
    }

    @Override
    public void start(Stage primaryStage) throws Exception {
        primaryStage.setScene(new Scene(createContent()));
        primaryStage.show();
    }
    
    public static class XYDataVisualizer<X, Y> extends TreeTableView<XYDataVisualizer.XYChartItem<X, Y>> {

        XYChart<X, Y> chart;
        Class<? extends X> xType;
        Class<? extends Y> yType;
    //    Class<?> extraValueType;
        double minY, maxY;

        public XYDataVisualizer(final XYChart<X, Y> chart) {
            this.chart = chart;
            setShowRoot(false);
            XYDataVisualizer.XYChartItem<X, Y> root = new XYDataVisualizer.XYChartItem<>(chart.getData());
            setRoot(new XYDataVisualizer.MyTreeItem(root));
            setMinHeight(100);
            setMinWidth(100);

            parseData();

            if (!getRoot().getChildren().isEmpty()) {
                getRoot().getChildren().get(0).setExpanded(true);
            }

            chart.dataProperty().addListener(new ChangeListener<ObservableList<XYChart.Series<X, Y>>>() {

                @Override
                public void changed(ObservableValue<? extends ObservableList<XYChart.Series<X, Y>>> ov, ObservableList<XYChart.Series<X, Y>> t, ObservableList<XYChart.Series<X, Y>> t1) {
                    setRoot(new XYDataVisualizer.MyTreeItem(new XYDataVisualizer.XYChartItem<X, Y>(t1)));
                }
            });

            TreeTableColumn<XYDataVisualizer.XYChartItem<X, Y>, String> nameColumn = new TreeTableColumn<>("Name");
            nameColumn.setCellValueFactory(new Callback<TreeTableColumn.CellDataFeatures<XYDataVisualizer.XYChartItem<X, Y>, String>, ObservableValue<String>>() {

                @Override
                public ObservableValue<String> call(TreeTableColumn.CellDataFeatures<XYDataVisualizer.XYChartItem<X, Y>, String> p) {
                    return p.getValue().getValue().nameProperty();
                }
            });
            nameColumn.setEditable(true);
            nameColumn.setSortable(false);
            nameColumn.setMinWidth(70);

            TreeTableColumn<XYDataVisualizer.XYChartItem<X, Y>, X> xValueColumn = new TreeTableColumn<>("XValue");
            xValueColumn.setCellValueFactory(new Callback<TreeTableColumn.CellDataFeatures<XYDataVisualizer.XYChartItem<X, Y>, X>, ObservableValue<X>>() {

                @Override
                public ObservableValue<X> call(TreeTableColumn.CellDataFeatures<XYDataVisualizer.XYChartItem<X, Y>, X> p) {
                    return p.getValue().getValue().xValueProperty();
                }
            });
            xValueColumn.setCellFactory(new Callback<TreeTableColumn<XYDataVisualizer.XYChartItem<X, Y>, X>, TreeTableCell<XYDataVisualizer.XYChartItem<X, Y>, X>>() {

                @Override
                public TreeTableCell<XYDataVisualizer.XYChartItem<X, Y>, X> call(TreeTableColumn<XYDataVisualizer.XYChartItem<X, Y>, X> p) {
                    return new TextFieldTreeTableCell<>(new StringConverter<X>() {

                        @Override
                        public String toString(X t) {
                            return t == null ? null : t.toString();
                        }

                        @Override
                        public X fromString(String string) {
                            if (string == null) {
                                return null;
                            }
                            X x = (X) new Double(string);
                            return x;
                        }
                    });
                }
            });
            xValueColumn.setEditable(true);
            xValueColumn.setSortable(false);
            xValueColumn.setMinWidth(50);

            TreeTableColumn<XYDataVisualizer.XYChartItem<X, Y>, Y> yValueColumn = new TreeTableColumn<>("YValue");
            yValueColumn.setCellValueFactory(new Callback<TreeTableColumn.CellDataFeatures<XYDataVisualizer.XYChartItem<X, Y>, Y>, ObservableValue<Y>>() {

                @Override
                public ObservableValue<Y> call(TreeTableColumn.CellDataFeatures<XYDataVisualizer.XYChartItem<X, Y>, Y> p) {
                    return p.getValue().getValue().yValueProperty();
                }
            });
            yValueColumn.setCellFactory(new Callback<TreeTableColumn<XYDataVisualizer.XYChartItem<X, Y>, Y>, TreeTableCell<XYDataVisualizer.XYChartItem<X, Y>, Y>>() {

                @Override
                public TreeTableCell<XYDataVisualizer.XYChartItem<X, Y>, Y> call(TreeTableColumn<XYDataVisualizer.XYChartItem<X, Y>, Y> p) {
                    return new TextFieldTreeTableCell<>(new StringConverter<Y>() {

                        @Override
                        public String toString(Y t) {
                            return t == null ? null : t.toString();
                        }

                        @Override
                        public Y fromString(String string) {
                            if (string == null) {
                                return null;
                            }
                            Y y = (Y) new Double(string);
                            return y;
                        }
                    });
                }
            });
            yValueColumn.setEditable(true);
            yValueColumn.setSortable(false);
            yValueColumn.setMinWidth(50);

            Class<XYDataVisualizer.XYChartItem<X, Y>> clz = (Class<XYDataVisualizer.XYChartItem<X, Y>>) root.getClass();
            TreeTableColumn<XYDataVisualizer.XYChartItem<X, Y>, Object> extraValueColumn = TreeTableColumnBuilder.create(clz, Object.class)
                    .cellValueFactory(new Callback<TreeTableColumn.CellDataFeatures<XYDataVisualizer.XYChartItem<X, Y>, Object>, ObservableValue<Object>>() {

                        @Override
                        public ObservableValue<Object> call(TreeTableColumn.CellDataFeatures<XYDataVisualizer.XYChartItem<X, Y>, Object> p) {
                            return p.getValue().getValue().extraValueProperty();
                        }
                    })
                    .text("Extra Value")
                    .minWidth(100)
                    .sortable(false)
                    .build();

            getColumns().setAll(nameColumn, xValueColumn, yValueColumn, extraValueColumn);

           
            setContextMenu(ContextMenuBuilder.create()
                    .items(
                        MenuItemBuilder.create()
                            .text("Insert data item")
                            .disable(!isEditable())
                            .onAction(new EventHandler<ActionEvent>() {

                                @Override
                                public void handle(ActionEvent t) {
                                    TreeItem<XYDataVisualizer.XYChartItem<X, Y>> selectedItem = getSelectionModel().getSelectedItem();
                                    if (selectedItem == null) {
                                        return;
                                    }
                                    Object value = selectedItem.getValue().getValue();
                                    Object parentValue = selectedItem.getParent().getValue().getValue();
                                    if (value instanceof XYChart.Series) {
                                        XYChart.Series series = (XYChart.Series) value;
                                        insertItem(series.getData());
                                    } else if (parentValue instanceof XYChart.Series) {
                                        XYChart.Series series = (XYChart.Series) parentValue;
                                        insertItem(series.getData().indexOf(value), series.getData());
                                    }
                                }

                                private int categoryIndex = 1;

                                public XYChart.Data<Integer, Integer> insertItem(int index, ObservableList<XYChart.Data> list) {
                                    XYChart.Data prev = null, next = null;
                                    if (index >= 0 && index < list.size()) {
                                        next = list.get(index);
                                    }
                                    if (index > 0) {
                                        prev = list.get(index - 1);
                                    }
                                    if (chart.getXAxis() instanceof NumberAxis) {
                                        NumberAxis xAxis = (NumberAxis) chart.getXAxis();
                                        double lower = prev == null 
                                                ? xAxis.getLowerBound() - 2 * xAxis.getTickUnit() 
                                                : ((Number) prev.getXValue()).doubleValue();
                                        double upper = next == null 
                                                ? xAxis.getUpperBound() + 2 * xAxis.getTickUnit() 
                                                : ((Number) next.getXValue()).doubleValue();
                                        XYChart.Data item = new XYChart.Data<>(
                                                Math.random() * (upper - lower) + lower, 
                                                Math.random() * (maxY - minY) + minY);
                                        list.add(index, item);
                                        return item;
                                    } else if (chart.getXAxis() instanceof CategoryAxis) {
                                        CategoryAxis xAxis = (CategoryAxis) chart.getXAxis();
                                        int lower = prev == null 
                                                ? -1 
                                                : xAxis.getCategories().indexOf(prev.getXValue());
                                        int upper = next == null 
                                                ? xAxis.getCategories().size() 
                                                : xAxis.getCategories().indexOf(next.getXValue());
                                        String category;
                                        if (upper - lower <= 1) {
                                            category = "New category " + (categoryIndex++);
                                            System.out.println("before list = " + list);
                                            System.out.println("list index = " + index);
                                            System.out.println("before xAxis.getCategories() = " + xAxis.getCategories());
                                            System.out.println("category index = " + upper);
                                            xAxis.getCategories().add(upper, category);
                                            System.out.println("after  xAxis.getCategories() = " + xAxis.getCategories());
                                        } else {
                                            category = xAxis.getCategories().get(
                                                    (int) (Math.random() * (upper - lower - 1) + lower + 1));
                                        }
                                        XYChart.Data item = new XYChart.Data<>(category, 
                                                Math.random() * (maxY - minY) + minY);
                                        list.add(index, item);
                                        System.out.println("after  list = " + list);
                                        return item;
                                    }
                                    return new XYChart.Data<>(50, 50);
                                }

                                public XYChart.Data<Integer, Integer> insertItem(ObservableList<XYChart.Data> list) {
                                    return insertItem(list.size(), list);
                                }
                            })
                            .build(),
                        MenuItemBuilder.create()
                            .text("Insert Series")
                            .disable(!isEditable())
                            .onAction(new EventHandler<ActionEvent>() {

                                private int seriesIndex = 4;

                                @Override
                                public void handle(ActionEvent t) {
                                    TreeItem<XYDataVisualizer.XYChartItem<X, Y>> selectedItem = getSelectionModel().getSelectedItem();
                                    if (selectedItem == null) {
                                        return;
                                    }
                                    Object value = selectedItem.getValue().getValue();
                                    Object parentValue = selectedItem.getParent().getValue().getValue();
                                    if (parentValue instanceof ObservableList) {
                                        ObservableList parentList = (ObservableList) parentValue;
                                        parentList.add(parentList.indexOf(value), new XYChart.Series<>("Series " + (seriesIndex++),
                                                FXCollections.observableArrayList(new XYChart.Data<>(50, 50))));
                                    }
                                }
                            })
                            .build(),
                        MenuItemBuilder.create()
                            .text("Delete item")
                            .disable(!isEditable())
                            .onAction(new EventHandler<ActionEvent>() {

                                @Override
                                public void handle(ActionEvent t) {
                                    TreeItem<XYDataVisualizer.XYChartItem<X, Y>> selectedItem = getSelectionModel().getSelectedItem();
                                    if (selectedItem == null) {
                                        return;
                                    }
                                    Object value = selectedItem.getValue().getValue();
                                    Object parentValue = selectedItem.getParent().getValue().getValue();
                                    if (parentValue instanceof ObservableList) {
                                        ((ObservableList) parentValue).remove(value);
                                    } else if (parentValue instanceof XYChart.Series) {
                                        ((XYChart.Series) parentValue).getData().remove(value);
                                    }
                                }
                            })
                            .build(),
                        MenuItemBuilder.create()
                            .text("Remove all data")
                            .disable(!isEditable())
                            .onAction(new EventHandler<ActionEvent>() {

                                @Override
                                public void handle(ActionEvent t) {
    //                                chart.getData().clear();
                                    chart.setData(null);
                                }
                            })
                            .build(),
                        MenuItemBuilder.create()
                            .text("Set new data")
                            .disable(!isEditable())
                            .onAction(new EventHandler<ActionEvent>() {

                                @Override
                                public void handle(ActionEvent t) {
                                    chart.setData(generateData());
                                }
                            })
                            .build()
                    )
                    .build());
        }

        public ObservableList generateData() {
            return FXCollections.observableArrayList(
                    new AreaChart.Series("Series 1",FXCollections.observableArrayList(
                        new AreaChart.Data(0,4),
                        new AreaChart.Data(2,5),
                        new AreaChart.Data(4,4),
                        new AreaChart.Data(6,2),
                        new AreaChart.Data(8,6),
                        new AreaChart.Data(10,8)
                    )),
                    new AreaChart.Series("Series 2", FXCollections.observableArrayList(
                        new AreaChart.Data(0,8),
                        new AreaChart.Data(2,2),
                        new AreaChart.Data(4,9),
                        new AreaChart.Data(6,7),
                        new AreaChart.Data(8,5),
                        new AreaChart.Data(10,7)
                    )),
                    new AreaChart.Series("Series 3", FXCollections.observableArrayList(
                        new AreaChart.Data(0,2),
                        new AreaChart.Data(2,5),
                        new AreaChart.Data(4,8),
                        new AreaChart.Data(6,6),
                        new AreaChart.Data(8,9),
                        new AreaChart.Data(10,7)
                    ))
            );        
        }

        private void parseData() {
            boolean editable = true;
            for (XYChart.Series<X, Y> series : chart.getData()) {
                for (XYChart.Data<X, Y> data : series.getData()) {
                    if (data.getXValue() != null) {
                        if (xType == null) {
                            xType = (Class<? extends X>) data.getXValue().getClass();
                        }
                    }
                    Y y = data.getYValue();
                    if (y != null) {
                        if (yType == null) {
                            yType = (Class<? extends Y>) y.getClass();
                        }
                        if (yType != null && Number.class.isAssignableFrom(yType)) {
                            minY = Math.min(minY, ((Number) y).doubleValue());
                            maxY = Math.max(maxY, ((Number) y).doubleValue());
                        }
                    }
                    if (data.getExtraValue() != null) {
                        editable = false;
                    }
                }
            }
            if (chart.getYAxis() instanceof CategoryAxis) {
                editable = false;
            }
            setEditable(editable);
            System.out.println("editable = " + editable);
            System.out.println("minY = " + minY);
            System.out.println("maxY = " + maxY);
            System.out.println("xType = " + xType);
            System.out.println("yType = " + yType);
        }

        private static class MyTreeItem<X, Y> extends TreeItem<XYDataVisualizer.XYChartItem<X, Y>> {

            {
                expandedProperty().addListener(new ChangeListener<Boolean>() {

                    @Override
                    public void changed(ObservableValue<? extends Boolean> ov, Boolean t, Boolean expanded) {
                        if (expanded) {
                            ObservableList children = getValue().getChildren();
                            if (children != null && getChildren().isEmpty()) {
                                boolean expand = children.size() == 1;
                                for (Object child : children) {
                                    getChildren().add(new XYDataVisualizer.MyTreeItem(new XYDataVisualizer.XYChartItem(child), expand));
                                }
                            } 
                        }
                    }
                });
            }

            @Override
            public boolean isLeaf() {
                return getValue().isLeaf();
            }

            public MyTreeItem(XYDataVisualizer.XYChartItem t) {
                this(t, true);
            }

            public MyTreeItem(XYDataVisualizer.XYChartItem t, boolean expand) {
                super(t);
                setExpanded(expand);
            }
        }

        public static class XYChartItem<X, Y> {

            private boolean leaf = true;
            private ObservableList children;
            private Object value;

            public XYChartItem(Object value) {
                this.value = value;
                if (value == null) {
                    return;
                }
                name.set(value.toString());
                if (value instanceof ObservableList) {
                    children = (ObservableList) value;
                    leaf = false;
                } else if (value instanceof XYChart.Series) {
                    XYChart.Series<X, Y> series = (XYChart.Series<X, Y>) value;
                    name = series.nameProperty();
                    children = series.getData();
                    leaf = false;
                } else if (value instanceof XYChart.Data) {
                    XYChart.Data<X, Y> data = (XYChart.Data<X, Y>) value;
                    name.set("Data");
                    xValue = data.XValueProperty();
                    yValue = data.YValueProperty();
                    extraValue.bindBidirectional(data.extraValueProperty());
                }
            }

            public ObservableList getChildren() {
                return children;
            }

            public boolean isLeaf() {
                return leaf;
            }

            private ObjectProperty<X> xValue = new SimpleObjectProperty<>();

            public ObjectProperty<X> xValueProperty() {
                return xValue;
            }

            private ObjectProperty<Y> yValue = new SimpleObjectProperty<>();

            public ObjectProperty<Y> yValueProperty() {
                return yValue;
            }

            private ObjectProperty<Object> extraValue = new SimpleObjectProperty<>();

            public ObjectProperty<Object> extraValueProperty() {
                return extraValue;
            }

            private StringProperty name = new SimpleStringProperty();

            public StringProperty nameProperty() {
                return name;
            }

            public Object getValue() {
                return value;
            }
        }
    }    
}