/*
 * Copyright (c) 2008, 2013 Oracle and/or its affiliates.
 * All rights reserved. Use is subject to license terms.
 *
 * This file is available and licensed under the following license:
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *  - Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 *  - Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in
 *    the documentation and/or other materials provided with the distribution.
 *  - Neither the name of Oracle Corporation nor the names of its
 *    contributors may be used to endorse or promote products derived
 *    from this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */



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.BarChart;
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;


public class BarChartBug extends Application {
    
    private BarChart<String, Double> chart;
    private CategoryAxis xAxis;
    private NumberAxis yAxis;
    private TextArea textArea;
    
    private BarChartBug.XYDataVisualizer xyVis;

    public Parent createContent() {
        String[] years = {"2007", "2008", "2009"};
        xAxis = new CategoryAxis();
        xAxis.setCategories(FXCollections.<String>observableArrayList(years));
        xAxis.setAutoRanging(false);
        yAxis = new NumberAxis("Units Sold", 0.0d, 3000.0d, 1000.0d);
        ObservableList<BarChart.Series> barChartData = FXCollections.observableArrayList(
            new BarChart.Series("Apples", FXCollections.observableArrayList(
//               new BarChart.Data(years[0], 567d),
               new BarChart.Data(years[1], 1292d),
               new BarChart.Data(years[2], 1292d)
            )),
            new BarChart.Series("Lemons", FXCollections.observableArrayList(
               new BarChart.Data(years[0], 956),
               new BarChart.Data(years[1], 1665),
               new BarChart.Data(years[2], 2559)
            )),
            new BarChart.Series("Oranges", FXCollections.observableArrayList(
               new BarChart.Data(years[0], 1154),
               new BarChart.Data(years[1], 1927),
               new BarChart.Data(years[2], 2774)
            ))
        );
        chart = new BarChart(xAxis, yAxis, barChartData, 25.0d);
//        FXCollections.sort(xAxis.getCategories());
        
        xyVis = new BarChartBug.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();
                        sb.append("Categories: ").append(((CategoryAxis) chart.getXAxis()).getCategories()).append('\n');
                        for (XYChart.Series<String, Double> series : chart.getData()) {
                            sb.append(series).append('\n');
                            for (XYChart.Data<String, Double> data : series.getData()) {
                                sb.append("   ").append(data).append('\n');
                            }
                        }
                        
                        sb.append("\nTreeTableView data:\n");
                        printTree(xyVis.getRoot(), sb, "");
                        
                        textArea.setText(sb.toString());
                    }
                    
                    private void printTree(TreeItem<?> item, StringBuilder sb, String indent) {
                        if (item == null) return;
                        sb.append(indent);
                        sb.append(item.getValue());
                        sb.append("\n");
                        for (TreeItem child : item.getChildren()) {
                            printTree(child, sb, indent + "\t");
                        }
                    }
                })
                .build();
        
        return HBoxBuilder.create()
                .children(chart, xyVis, textArea)
                .build();
    }

    @Override public void start(Stage primaryStage) throws Exception {
        primaryStage.setScene(new Scene(createContent()));
        primaryStage.show();
    }

    /**
     * Java main for when running without JavaFX launcher
     */
    public static void main(String[] args) {
        launch(args);
    }
    
        public static class XYDataVisualizer<X, Y> extends TreeTableView<BarChartBug.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);
            BarChartBug.XYDataVisualizer.XYChartItem<X, Y> root = new BarChartBug.XYDataVisualizer.XYChartItem<>(chart.getData());
            setRoot(new BarChartBug.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 BarChartBug.XYDataVisualizer.MyTreeItem(new BarChartBug.XYDataVisualizer.XYChartItem<X, Y>(t1)));
                }
            });

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

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

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

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

                @Override
                public TreeTableCell<BarChartBug.XYDataVisualizer.XYChartItem<X, Y>, X> call(TreeTableColumn<BarChartBug.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<BarChartBug.XYDataVisualizer.XYChartItem<X, Y>, Y> yValueColumn = new TreeTableColumn<>("YValue");
            yValueColumn.setCellValueFactory(new Callback<TreeTableColumn.CellDataFeatures<BarChartBug.XYDataVisualizer.XYChartItem<X, Y>, Y>, ObservableValue<Y>>() {

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

                @Override
                public TreeTableCell<BarChartBug.XYDataVisualizer.XYChartItem<X, Y>, Y> call(TreeTableColumn<BarChartBug.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<BarChartBug.XYDataVisualizer.XYChartItem<X, Y>> clz = (Class<BarChartBug.XYDataVisualizer.XYChartItem<X, Y>>) root.getClass();
            TreeTableColumn<BarChartBug.XYDataVisualizer.XYChartItem<X, Y>, Object> extraValueColumn = TreeTableColumnBuilder.create(clz, Object.class)
                    .cellValueFactory(new Callback<TreeTableColumn.CellDataFeatures<BarChartBug.XYDataVisualizer.XYChartItem<X, Y>, Object>, ObservableValue<Object>>() {

                        @Override
                        public ObservableValue<Object> call(TreeTableColumn.CellDataFeatures<BarChartBug.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<BarChartBug.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<BarChartBug.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<BarChartBug.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<BarChartBug.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 BarChartBug.XYDataVisualizer.MyTreeItem(new BarChartBug.XYDataVisualizer.XYChartItem(child), expand));
                                }
                            } 
                        }
                    }
                });
            }

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

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

            public MyTreeItem(BarChartBug.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;
            }
            
            @Override
            public String toString() {
                return "XYChartItem [ name: " + name.get() + 
                        ", x: " + xValueProperty().get() + 
                        ", y: " + yValueProperty().get() + " ]";
            }
        }
    }  
}