gvSIG tools: basic charts

package org.gvsig.chart.swing.impl.editors.panels;

import java.awt.Color;

import java.awt.Dimension;

import java.awt.Font;

import java.awt.GridBagConstraints;

import java.awt.GridBagLayout;

import java.awt.Insets;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.ArrayList;

import java.util.Iterator;

import java.util.List;

import javax.swing.ButtonGroup;

import javax.swing.DefaultComboBoxModel;

import javax.swing.JButton;

import javax.swing.JCheckBox;

import javax.swing.JComboBox;

import javax.swing.JLabel;

import javax.swing.JPanel;

import javax.swing.JRadioButton;

import javax.swing.JScrollPane;

import javax.swing.JTable;

import javax.swing.ListSelectionModel;

import javax.swing.table.DefaultTableModel;

import org.gvsig.chart.ChartProperties;

import org.gvsig.chart.ChartService;

import org.gvsig.chart.model.variables.RequiredField;

import org.gvsig.chart.renderer.ChartSerieEntity;

import org.gvsig.chart.swing.ChartSwingLocator;

import org.gvsig.chart.swing.ChartSwingManager;

import org.gvsig.chart.swing.editors.panels.ChartServiceDataPanel;

import org.gvsig.chart.swing.impl.editors.DefaultEditorChartServicePanel;

import org.gvsig.chart.swing.impl.utils.ColorEditor;

import org.gvsig.chart.swing.impl.utils.ColorRenderer;

import org.gvsig.tools.observer.Observable;

import org.gvsig.tools.observer.Observer;

public class DefaultChartServiceDataPanel extends JPanel implements ChartServiceDataPanel{

        /**

         * 

         */

        private static final long serialVersionUID = 3973104135478200344L;

        private List<JComboBox> cbos = new ArrayList<JComboBox>();

        private JComboBox cboTable;

        protected List<Observer> observers = new ArrayList<Observer>();

        public static final int MAIN_PANEL_WIDTH = 680;

        public static final int MAIN_PANEL_HEIGHT = 410;

        public static final int TABLE_WIDTH = 270;

        public static final int TABLE_HEIGHT = 150;

        private DefaultEditorChartServicePanel parentPanel;

        private JCheckBox chckbxUseSelectedRows;

        private JButton editSeriesButton;

        private JTable editSeriesTable;

        private DefaultTableModel editSeriesTableModel;

        private JLabel lblDataOrigin;

        private ChartSwingManager swingManager = ChartSwingLocator.getSwingManager().getChartSwingManager();

        private JRadioButton nullGroupIgnoredRb;

        private JRadioButton addInSpecialCategory;

        private JCheckBox cbSortedField;

        private boolean hasSortedField = false;

        private boolean allowOnlySelectedData;

        public DefaultChartServiceDataPanel(DefaultEditorChartServicePanel parent) {

                this.parentPanel = parent;

                ChartProperties cp = getChartProperties();

                GridBagLayout gridBagLayout = new GridBagLayout();

                /*gridBagLayout.columnWidths = new int[]{0, 0, 0, TABLE_WIDTH, 0, 0, 0, 0, 0,0};

                gridBagLayout.rowWeights = new double[]{0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0};

                gridBagLayout.rowHeights = new int[8 + cp.getRequiredFields().size()+1];

                gridBagLayout.columnWeights = new double[]{0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0};

                */setLayout(gridBagLayout);

                /*for(int i=0; i< 8 + cp.getRequiredFields().size()+1; i++){

                        gridBagLayout.rowHeights[i]=0;

                }

                gridBagLayout.rowHeights[8 + cp.getRequiredFields().size()]=TABLE_HEIGHT;

                */

                lblDataOrigin = new JLabel(this.swingManager.getTranslation("Data_origin"));

                GridBagConstraints gbc_lblDataOrigin = new GridBagConstraints();

                gbc_lblDataOrigin.anchor = GridBagConstraints.WEST;

                gbc_lblDataOrigin.gridwidth = 3;

                gbc_lblDataOrigin.insets = new Insets(0, 0, 5, 5);

                gbc_lblDataOrigin.gridx = 1;

                gbc_lblDataOrigin.gridy = 0;

                add(lblDataOrigin, gbc_lblDataOrigin);

                JLabel lblTable = new JLabel(this.swingManager.getTranslation("Table"));

                lblTable.setFont(new Font("Dialog", Font.PLAIN, 12));

                GridBagConstraints gbc_lblTable = new GridBagConstraints();

                gbc_lblTable.anchor = GridBagConstraints.WEST;

                gbc_lblTable.insets = new Insets(0, 0, 5, 5);

                gbc_lblTable.gridx = 1;

                gbc_lblTable.gridy = 1;

                add(lblTable, gbc_lblTable);

                cboTable = new JComboBox();

                cboTable.addActionListener(new ActionListener() {

                        public void actionPerformed(ActionEvent arg0) {

                                String tableName = (String) cboTable.getSelectedItem();

                                notifyObservers(DefaultChartServiceDataPanel.this, tableName);

                        }

                });

                GridBagConstraints gbc_cboTable = new GridBagConstraints();

                gbc_cboTable.gridwidth = 4;

                gbc_cboTable.insets = new Insets(0, 0, 5, 5);

                gbc_cboTable.fill = GridBagConstraints.HORIZONTAL;

                gbc_cboTable.gridx = 3;

                gbc_cboTable.gridy = 1;

                add(cboTable, gbc_cboTable);

                JLabel lblChartDataDefinition = new JLabel(this.swingManager.getTranslation("Chart_Data_Definition"));

                GridBagConstraints gbc_lblChartDataDefinition = new GridBagConstraints();

                gbc_lblChartDataDefinition.insets = new Insets(0, 0, 5, 0);

                gbc_lblChartDataDefinition.gridwidth = 6;

                gbc_lblChartDataDefinition.anchor = GridBagConstraints.WEST;

                gbc_lblChartDataDefinition.gridx = 1;

                gbc_lblChartDataDefinition.gridy = 3;

                add(lblChartDataDefinition, gbc_lblChartDataDefinition);

                this.initialize();

                chckbxUseSelectedRows = new JCheckBox(this.swingManager.getTranslation("use_selected_rows"));

                chckbxUseSelectedRows.addActionListener(new ActionListener() {

                        public void actionPerformed(ActionEvent e) {

                                List<Color> colors = new ArrayList<Color>();

                                for(int index=0; index< editSeriesTable.getRowCount(); index++){

                                        colors.add((Color)editSeriesTable.getModel().getValueAt(index, 0));

                                }

                                refreshData();

                                if(colors.size()==editSeriesTable.getRowCount()){

                                        for(int index=0; index< editSeriesTable.getRowCount(); index++){

                                                editSeriesTable.getModel().setValueAt(colors.get(index), index, 0);

                                        }

                                }

                        }

                });

                chckbxUseSelectedRows.setFont(new Font("Dialog", Font.PLAIN, 12));

                if (cp.getUseSelectedFeatures())

                        chckbxUseSelectedRows.setSelected(true);

                GridBagConstraints gbc_chckbxUseSelectedRows = new GridBagConstraints();

                gbc_chckbxUseSelectedRows.anchor = GridBagConstraints.WEST;

                gbc_chckbxUseSelectedRows.gridwidth = 3;

                gbc_chckbxUseSelectedRows.insets = new Insets(0, 0, 5, 5);

                gbc_chckbxUseSelectedRows.gridx = 1;

                gbc_chckbxUseSelectedRows.gridy = 5 + cp.getRequiredFields().size();

                add(chckbxUseSelectedRows, gbc_chckbxUseSelectedRows);

                JLabel groupedOptionslabel = new JLabel(this.swingManager.getTranslation("grouped_options"));

                GridBagConstraints gbc_chckbxUseSelectedRows15 = new GridBagConstraints();

                gbc_chckbxUseSelectedRows15.anchor = GridBagConstraints.WEST;

                gbc_chckbxUseSelectedRows15.gridwidth = 3;

                gbc_chckbxUseSelectedRows15.insets = new Insets(0, 0, 5, 5);

                gbc_chckbxUseSelectedRows15.gridx = 1;

                gbc_chckbxUseSelectedRows15.gridy = 6 + cp.getRequiredFields().size();

                add(groupedOptionslabel, gbc_chckbxUseSelectedRows15);

                nullGroupIgnoredRb = new JRadioButton(this.swingManager.getTranslation("ignore_null_groups"));

                nullGroupIgnoredRb.addActionListener(new ActionListener() {

                        public void actionPerformed(ActionEvent e) {

                                getChartProperties().setNullGroupIgnored(true);

                                refreshData();

                        }

                });

                nullGroupIgnoredRb.setFont(new Font("Dialog", Font.PLAIN, 12));

                GridBagConstraints gbc_chckbxUseSelectedRows2 = new GridBagConstraints();

                gbc_chckbxUseSelectedRows2.anchor = GridBagConstraints.WEST;

                gbc_chckbxUseSelectedRows2.gridwidth = 3;

                gbc_chckbxUseSelectedRows2.insets = new Insets(0, 0, 5, 5);

                gbc_chckbxUseSelectedRows2.gridx = 2;

                gbc_chckbxUseSelectedRows2.gridy = 6 + cp.getRequiredFields().size();

                add(nullGroupIgnoredRb, gbc_chckbxUseSelectedRows2);

                addInSpecialCategory = new JRadioButton(this.swingManager.getTranslation("add_null_in_special_category"));

                addInSpecialCategory.addActionListener(new ActionListener() {

                        public void actionPerformed(ActionEvent e) {

                                getChartProperties().setNullGroupIgnored(false);

                                refreshData();

                        }

                });

                addInSpecialCategory.setFont(new Font("Dialog", Font.PLAIN, 12));

                GridBagConstraints gbc_chckbxUseSelectedRows3 = new GridBagConstraints();

                gbc_chckbxUseSelectedRows3.anchor = GridBagConstraints.WEST;

                gbc_chckbxUseSelectedRows3.gridwidth = 3;

                gbc_chckbxUseSelectedRows3.insets = new Insets(0, 0, 5, 5);

                gbc_chckbxUseSelectedRows3.gridx = 2;

                gbc_chckbxUseSelectedRows3.gridy = 7 + cp.getRequiredFields().size();

                add(addInSpecialCategory, gbc_chckbxUseSelectedRows3);

                ButtonGroup group = new ButtonGroup();

                group.add(nullGroupIgnoredRb);

                group.add(addInSpecialCategory);

                if (cp.getNullGroupIgnored()){

                        nullGroupIgnoredRb.setSelected(true);

                }else{

                        addInSpecialCategory.setSelected(true);

                }

//                editSeriesButton = new JButton(this.swingManager.getTranslation("Edit_Series_Values"));

//                editSeriesButton.addActionListener(new ActionListener() {

//                        public void actionPerformed(ActionEvent arg0) {

//                                listSeries();

//                        }

//                });

//                GridBagConstraints gbc_editSeriesButton = new GridBagConstraints();

//                gbc_editSeriesButton.anchor = GridBagConstraints.EAST;

//                gbc_editSeriesButton.gridwidth = 3;

//                gbc_editSeriesButton.insets = new Insets(0, 0, 5, 5);

//                gbc_editSeriesButton.gridx = 3;

//                gbc_editSeriesButton.gridy = 5 + getChartProperties().getRequiredFields().size();

//                add(editSeriesButton, gbc_editSeriesButton);

                editSeriesTable = new JTable(){

                                /**

                         * 

                         */

                        private static final long serialVersionUID = -7016858079737088567L;

                                public boolean isCellEditable(int rowIndex, int colIndex) {

                                        if (colIndex == 0)

                                                return true;

                                        return false; // Disallow the editing of any cell

                                }

                        };

                editSeriesTable.setSelectionMode(ListSelectionModel.SINGLE_SELECTION);

                editSeriesTable.setColumnSelectionAllowed(false);

                editSeriesTable.setCellSelectionEnabled(false);

                JScrollPane editSeriesTableScroll = new JScrollPane(editSeriesTable);

                GridBagConstraints gbc_editSeriesTable = new GridBagConstraints();

                gbc_editSeriesTable.insets = new Insets(0, 0, 5, 0);

                gbc_editSeriesTable.gridheight = 3;

                gbc_editSeriesTable.gridwidth = 5;

                gbc_editSeriesTable.fill = GridBagConstraints.BOTH;

                gbc_editSeriesTable.weightx = 1.0;

                gbc_editSeriesTable.weighty = 1.0;

                gbc_editSeriesTable.gridx = 1;

                gbc_editSeriesTable.gridy = 8 + cp.getRequiredFields().size();

                add(editSeriesTableScroll, gbc_editSeriesTable);

                this.initializeTable();

                setPreferredSize(new Dimension(680, 478));

        }

        protected void refreshData() {

                getChartProperties().setUseSelectedFeatures(chckbxUseSelectedRows.isSelected());

                listSeries();

        }

        public void listSeries(){

                if(!parentPanel.getNewChartService().getChartLegendItems().isEmpty()){

                        fillTableData();

                }

                List<RequiredField> rfields = parentPanel.getNewChartService().getChartProperties().getRequiredFields();

                int index = -1;

                String previousSeriesField = null;

                Iterator<RequiredField> it = rfields.iterator();

                while(it.hasNext()){

                        RequiredField req = (RequiredField)it.next();

                        if(req.getName()== parentPanel.getNewChartService().getChartDataSet().getKeyField()){

                                index = rfields.indexOf((RequiredField)req);

                                previousSeriesField = req.getValue();

                        }

                }

                String currentSeriesField = null;

                if(index != -1){

                        if(cbos.get(index).getSelectedItem()!= null)

                                currentSeriesField = cbos.get(index).getSelectedItem().toString();

                }

                /*

                 * Si se ha cambiado el campo, o las series est?n por cargar

                 */

//                if((currentSeriesField != null && 

//                                !(currentSeriesField.equals(previousSeriesField))) ||

//                                parentPanel.getNewChartService().getChartLegendItems().isEmpty()){

                        registerParameters();

                        notifyObservers(DefaultChartServiceDataPanel.this, parentPanel.getNewChartService());

                        parentPanel.getNewChartService().update();

                        parentPanel.getNewChartService().createChartLegendItems();

                        fillTableData();

//                }

        }

        private void initialize() {

                ChartProperties chartProperties = getChartProperties();

                int i = 0;

                int j = 0;

                for (i = 0; i < chartProperties.getRequiredFields().size(); i++) {

                        RequiredField f = chartProperties.getRequiredFields().get(i);

                        String name = f.getDescription();

                        if(f.isOptional()){

                                name += " ("+this.swingManager.getTranslation("optional")+")";

                        }

                        JLabel lblNewLabel = new JLabel(name);

                        GridBagConstraints gbc_lblNewLabel = new GridBagConstraints();

                        gbc_lblNewLabel.fill = GridBagConstraints.BOTH;

                        gbc_lblNewLabel.insets = new Insets(0, 0, 5, 5);

                        gbc_lblNewLabel.gridx = 1;

                        gbc_lblNewLabel.gridy = 4+i+j;

                        add(lblNewLabel, gbc_lblNewLabel);

                        JComboBox cboTableField = new JComboBox();

                        cboTableField.addActionListener(new ActionListener() {

                                public void actionPerformed(ActionEvent arg0) {

                                        listSeries();                        

                                        parentPanel.getChartServiceFormatPanel().refreshContent();

                                }

                        });

                        GridBagConstraints gbc_cboTable = new GridBagConstraints();

                        gbc_cboTable.gridwidth = 4;

                        gbc_cboTable.insets = new Insets(0, 0, 5, 5);

                        gbc_cboTable.fill = GridBagConstraints.HORIZONTAL;

                        gbc_cboTable.gridx = 3;

                        gbc_cboTable.gridy = 4+i+j;

                        add(cboTableField, gbc_cboTable);

                        if(f.canBeSorted()){

                                cbSortedField = new JCheckBox(this.swingManager.getTranslation("ordered_field"));

                                GridBagConstraints gbc_cboSorted = new GridBagConstraints();

                                gbc_cboSorted.insets = new Insets(0, 0, 5, 5);

                                gbc_cboSorted.fill = GridBagConstraints.HORIZONTAL;

                                gbc_cboSorted.gridx = 3;

                                gbc_cboSorted.gridy = 5+i+j;

                                add(cbSortedField, gbc_cboSorted);

                                cbSortedField.setSelected(

                                                getParentPanel().getNewChartService().getChartDataSet().getIsSorted());

                                this.hasSortedField = true;

                                j++;

                        }

                        cbos.add(cboTableField);

                }

        }

        private void initializeTable() {

                String[] colName = { this.swingManager.getTranslation("Symbol"), 

                                this.swingManager.getTranslation("Series_name")};

                editSeriesTableModel = (DefaultTableModel) editSeriesTable.getModel();

                editSeriesTableModel.setColumnIdentifiers(colName);

                editSeriesTable.getColumnModel().getColumn(0).setPreferredWidth(50);

                editSeriesTable.setDefaultRenderer(Color.class, new ColorRenderer(true));

                editSeriesTable.setDefaultEditor(Color.class, new ColorEditor());

                editSeriesTable.getColumnModel().getColumn(0)

                                .setCellEditor(editSeriesTable.getDefaultEditor(Color.class));

                editSeriesTable.getColumnModel().getColumn(0)

                                .setCellRenderer(editSeriesTable.getDefaultRenderer(Color.class));

        }

        public void fillTableData() {

                ChartService service = parentPanel.getNewChartService();

                List<ChartSerieEntity> legendItems = service.getChartLegendItems();

                if(legendItems!=null){

                        removeTable();

                        for(int k=0; k<legendItems.size(); k++){

                                if(legendItems.get(k)!=null){

                                        Object[] data = {

                                                        legendItems.get(k).getSeriePaint(),

                                                        legendItems.get(k).getSeriesName()

                                        };

                                        //renderer.setSeriesPaint(series, paint)

                                        ((DefaultTableModel) editSeriesTableModel).addRow(data);

                                }

                        }

                }

        //        List<ChartService> chartServices = getGroupedChart().getChartServices();

                editSeriesTable.revalidate();

                editSeriesTable.repaint();

        }

        public void removeTable() {

                int numRows = editSeriesTableModel.getRowCount();

                for (int i = numRows - 1; i >= 0; i--) {

                        editSeriesTableModel.removeRow(i);

                }

        }

        /**

         * TODO Check if the values have the correct format

         * @return boolean

         */

        public boolean validateValues() {

                // TODO Auto-generated method stub

                return true;

        }

        public void acceptForm(){

                ChartService service = parentPanel.getNewChartService();

                if(service.getChartDataSet().getSeries().isEmpty()){

                        listSeries();

                }

                if(this.hasSortedField ){

                        service.getChartDataSet().setIsSorted(cbSortedField.isSelected());

                }

                service.getChartProperties().setNullGroupIgnored(nullGroupIgnoredRb.isSelected());

                registerParameters();

        }

        /**

         * Stores the information from swing controls to ChartProperties

         * in order to be usable by chart services.

         */

        public void registerParameters(){

                ChartProperties chartProperties = getChartProperties();

                //Rellenamos los datos del panel actual

                chartProperties.setChartTableName((String)cboTable.getSelectedItem());

                for (int i = 0; i < chartProperties.getRequiredFields().size(); i++) {

                        RequiredField f = chartProperties.getRequiredFields().get(i);

                        JComboBox cbo = cbos.get(i);

                        f.setValue((String) cbo.getSelectedItem());

                }

                chartProperties.setUseSelectedFeatures(chckbxUseSelectedRows.isSelected());

                ChartService service = parentPanel.getNewChartService();

                if(service.getChartLegendItems().size() == editSeriesTableModel.getRowCount()){

                        for(int j=0; j<service.getChartLegendItems().size();j++){

                                Color color = (Color) editSeriesTableModel.getValueAt(j, 0);

                                service.getChartLegendItems().get(j).setSeriePaint(color);

                        }

                }

        }

        public synchronized void deleteObserver(Observer o) {

                this.observers.remove(o);

        }

        public synchronized void deleteObservers() {

                this.observers = new ArrayList<Observer>();

        }

        public synchronized void notifyObservers(Observable observable, Object tableName) {

                for(int i =0; i<this.observers.size(); i++ ) {

                        Observer o = (Observer) this.observers.get(i);

                        o.update(this, tableName);

                }

        }

        public void addObserver(Observer o) {

                this.observers.add(o);

        }

        public void setTableNameModel(List<String> model) {

                DefaultComboBoxModel cboModel = new DefaultComboBoxModel(model.toArray());

                cboTable.setModel(cboModel);

        }

        public void setSelectedTableName(String tableName) {

                cboTable.setSelectedItem(tableName);

        }

        public void setRequiredFieldModel(List<RequiredField> rFields, RequiredField rf, List<String> model, boolean active) {

                int index = rFields.indexOf(rf);

                DefaultComboBoxModel cboModel = new DefaultComboBoxModel(model.toArray());

                cbos.get(index).setEnabled(active);

                nullGroupIgnoredRb.setEnabled(active);

                chckbxUseSelectedRows.setEnabled(active);

                addInSpecialCategory.setEnabled(active);

                if(allowOnlySelectedData){

                        setOnlySelectedRows(allowOnlySelectedData);

                }

                if(cbSortedField != null){

                        cbSortedField.setEnabled(active);

                }

                editSeriesTable.setEnabled(active);

                if(cbos.size()>=index){

                        cbos.get(index).setModel(cboModel);

                }

        }

        public void setSelectedRequiredField(List<RequiredField> rFields, RequiredField rf, String name, boolean active) {

                int index = rFields.indexOf(rf);

                if(cbos.size()>=index){

                        ActionListener[] listeners = cbos.get(index).getActionListeners();

                        for(int l=0; l<listeners.length;l++){

                                cbos.get(index).removeActionListener(listeners[l]);

                        }

                        cbos.get(index).setSelectedItem(name);

                        cbos.get(index).setEnabled(active);

                        nullGroupIgnoredRb.setEnabled(active);

                        chckbxUseSelectedRows.setEnabled(active);

                        addInSpecialCategory.setEnabled(active);

                        if(allowOnlySelectedData){

                                setOnlySelectedRows(allowOnlySelectedData);

                        }

                        if(cbSortedField != null){

                                cbSortedField.setEnabled(active); 

                        }

                        editSeriesTable.setEnabled(active);

                        if(active){

                                cbos.get(index).addActionListener(new ActionListener() {

                                        public void actionPerformed(ActionEvent arg0) {

                                                listSeries();                                        

                                        }

                                });

                        }

                }

        }

        public DefaultEditorChartServicePanel getParentPanel() {

                return parentPanel;

        }

        public ChartProperties getChartProperties() {

                return getParentPanel().getNewChartService().getChartProperties();

        }

        public ChartProperties getPreviousChartProperties(){

                if(getParentPanel().getPreviousChartService() == null){

                        return this.getChartProperties();

                }

                return getParentPanel().getPreviousChartService().getChartProperties();

        }

        public boolean getOnlySelectedRows() {

                return chckbxUseSelectedRows.isSelected();

        }

        public void setOnlySelectedRows(boolean selected) {

                if(selected){

                        chckbxUseSelectedRows.setSelected(selected);

                }

                this.allowOnlySelectedData = selected;

                chckbxUseSelectedRows.setEnabled(!selected);

        }

}