/*
 * Copyright (c) 2008, 2011 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.
 */
package com.javafx.experiments.swingjavafxcharts;

import java.awt.BorderLayout;
import java.awt.Color;
import java.awt.Component;
import java.awt.Dimension;
import java.text.DecimalFormat;
import java.util.Arrays;
import java.util.List;

import javax.swing.JFrame;
import javax.swing.SwingUtilities;
import javax.swing.JTable;
import javax.swing.JScrollPane;

import javax.swing.event.TableModelEvent;
import javax.swing.event.TableModelListener;

import javafx.beans.InvalidationListener;
import javafx.beans.Observable;

import javafx.collections.ObservableList;
import javafx.scene.Group;
import javafx.scene.Scene;
import javafx.embed.swing.JFXPanel;

import javafx.scene.chart.Chart;
import javafx.scene.chart.NumberAxis;
import javafx.scene.chart.BarChart;
import javafx.scene.chart.CategoryAxis;

import javafx.application.Platform;
import javafx.collections.FXCollections;
import javafx.scene.chart.XYChart;
import javax.swing.JApplet;
import javax.swing.JLabel;
import javax.swing.table.AbstractTableModel;
import javax.swing.table.DefaultTableCellRenderer;

/**
 * Sample showing a JavaFX chart linked to a Swing JTable
 *
 */
public class Main extends JApplet {

    private static JFXPanel javafxPanel;
    private static SampleTableModel tableModel;

    private Scene scene;
    private Chart chart;

    //init GUI
    public void init() {
        // create swing table
        tableModel = new SampleTableModel();

        JTable table = new JTable(tableModel);
        table.setAutoCreateRowSorter(true);
        table.setGridColor(Color.DARK_GRAY);
        DecimalFormatRenderer renderer = new DecimalFormatRenderer();
        renderer.setHorizontalAlignment(JLabel.RIGHT);
        for (int i=0; i<table.getColumnCount(); i++) {
            table.getColumnModel().getColumn(i).setCellRenderer(renderer);
        }
       
        JScrollPane panel = new JScrollPane(table);
        panel.setPreferredSize(new Dimension(550,100));
        setLayout(new BorderLayout());
        add(panel, BorderLayout.SOUTH);

        // create javafx panel
        javafxPanel = new JFXPanel();
        javafxPanel.setPreferredSize(new Dimension(550,400));
        add(javafxPanel, BorderLayout.CENTER);

        // create JavaFX scene
        Platform.runLater(new Runnable() {
            public void run() {
                createScene();
            }
        });
    }

    public void createScene() {
        Group root = new Group();
        scene = new Scene(root);

        root.getChildren().add(chart = createBarChart());
        // add scene to panel
        javafxPanel.setScene(scene);

        scene.widthProperty().addListener(new InvalidationListener() {
            @Override
            public void invalidated(Observable ov) {
                chart.setLayoutX((scene.getWidth() - chart.getWidth()) / 2);
            }
        });
        scene.heightProperty().addListener(new InvalidationListener() {
            @Override
            public void invalidated(Observable ov) {
                chart.setLayoutY((scene.getHeight() - chart.getHeight()) / 2);
            }
        });
    }

    private BarChart createBarChart() {
        CategoryAxis xAxis = new CategoryAxis();
        xAxis.setCategories(FXCollections.<String>observableArrayList(tableModel.getColumnNames()));
        xAxis.setLabel("Year");

        double tickUnit = tableModel.getTickUnit();

        NumberAxis yAxis = new NumberAxis("Units Sold", 0, 3000, tickUnit);

        final BarChart chart = new BarChart(xAxis, yAxis, tableModel.getBarChartData());
        tableModel.addTableModelListener(new TableModelListener() {
            public void tableChanged(TableModelEvent e) {
                if (e.getType() == TableModelEvent.UPDATE) {
                    final int row = e.getFirstRow();
                    final int column = e.getColumn();
                    final Object value = ((SampleTableModel)e.getSource()).getValueAt(row, column);
                    
                    Platform.runLater(new Runnable() {
                        public void run() {
                            XYChart.Series<String, Number> s = (XYChart.Series<String, Number>)chart.getData().get(row);
                            BarChart.Data data = s.getData().get(column);
                            data.setYValue(value);
                        }
                    });
                }
            }
        });

        return chart;
    }

    public static void main(String[] args) {
        SwingUtilities.invokeLater(new Runnable() {
            public void run() {
               JFrame frame = new JFrame("Swing JavaFX Chart");
               frame.setResizable(false);
               frame.setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);

               JApplet applet = new Main();
               applet.init();

               frame.setContentPane(applet.getContentPane());

               frame.pack();
               frame.setLocationRelativeTo(null);
               frame.setVisible(true);

               applet.start();
            }
        });
    }

    private static class DecimalFormatRenderer extends DefaultTableCellRenderer {
        private static final DecimalFormat formatter = new DecimalFormat("#.0");

        public Component getTableCellRendererComponent(JTable table, Object value, boolean isSelected, boolean hasFocus, int row, int column) {
            value = formatter.format((Number) value);
            return super.getTableCellRendererComponent(table, value, isSelected, hasFocus, row, column);
        }
    }

    /**
     * SampleTableModel
     */
    public static class SampleTableModel extends AbstractTableModel {
        private static ObservableList<BarChart.Series> bcData;

        private final String[] names = {"2007", "2008", "2009"};

        private Object[][] data = {
            {new Double(567), new Double(956), new Double(1154)},
                {new Double(1292), new Double(1665), new Double(1927)},
                {new Double(1292), new Double(2559), new Double(2774)}
            };

        public double getTickUnit() {
            return 1000;
        }
        public List<String> getColumnNames() {
            return Arrays.asList(names);
        }

        @Override
        public int getRowCount() {
            return data.length;
        }

        @Override
        public int getColumnCount() {
            return names.length;
        }

        @Override
        public Object getValueAt(int row, int column) {
            return data[row][column];
        }

        @Override
        public String getColumnName(int column) {
            return names[column];
        }

        @Override
        public Class getColumnClass(int column) {
            return getValueAt(0, column).getClass();
        }

        @Override
        public boolean isCellEditable(int row, int column) {
            return true;
        }

        @Override
        public void setValueAt(Object value, int row, int column) {
            if (value instanceof Double) {
                data[row][column] = (Double)value;
            }

            fireTableCellUpdated(row, column);
        }

        public ObservableList<BarChart.Series> getBarChartData() {
            if (bcData == null) {
                bcData = FXCollections.<BarChart.Series>observableArrayList();
                for (int row = 0; row < getRowCount(); row++) {
                    ObservableList<BarChart.Data> series = FXCollections.<BarChart.Data>observableArrayList();
                    for (int column = 0; column < getColumnCount(); column++) {
                        series.add(new BarChart.Data(getColumnName(column), getValueAt(row, column)));
                    }
                    bcData.add(new BarChart.Series(series));
                }
            }
            return bcData;
        }
    }
}