Application: gvSIG desktop » gvSIG geoprocess

package org.gvsig.geoprocess.algorithm.groupby;

import java.awt.BorderLayout;

import java.awt.Component;

import java.awt.event.ActionEvent;

import java.awt.event.ActionListener;

import java.util.ArrayList;

import java.util.List;

import javax.swing.JButton;

import javax.swing.JComboBox;

import javax.swing.JFrame;

import javax.swing.JTable;

import javax.swing.event.TableModelEvent;

import javax.swing.event.TableModelListener;

import javax.swing.table.DefaultTableModel;

import javax.swing.table.TableCellEditor;

import javax.swing.table.TableCellRenderer;

import javax.swing.table.TableColumn;

import org.gvsig.gui.beans.swing.treeTable.AbstractCellEditor;

public class TestTableModel extends JFrame implements TableModelListener, ActionListener {

        private static final long serialVersionUID = -7971006561681605303L;

        private int            w                   = 400;

        private int            h                   = 200;

        private JTable         table               = null;

        private final String[] opList              = { "first", "sum", "average", "max", "min", "last"};

        private final String[] columnNames         = { "Field", "Function" };

        private JButton        but                 = null;

        public class GroupByTableModel extends DefaultTableModel {

                final private static long       serialVersionUID   = -3370601314380922368L;

                private List<JComboBox>         combos             = null;

                public GroupByTableModel(String[] columnNames, List<JComboBox> c) {

                        super(new Object[0][columnNames.length], columnNames);

                        this.combos = c;

                }

                public Object[] getNewLine() {

                        return new Object[] { "", new Integer(0)};

                }

                public boolean isCellEditable(int row, int col) {

                        return (col == 1) ? true : false;

                }

                public void addRow(Object[] list) {

                        if(list.length <= 2 && list[1] instanceof Integer && list[0] instanceof String) {

                                super.addRow(list);

                        }

                }

                public Object getValueAt(int row, int column) {

                        Object obj = super.getValueAt(row, column);

                        if(column == 1)

                                if(row < combos.size())

                                        return combos.get(row).getSelectedIndex();

                        return obj;

                }

        }

        class GroupByEditor extends AbstractCellEditor implements TableCellEditor {

                private List<JComboBox>         combos             = null;

                public GroupByEditor(List<JComboBox> c) {

                        combos = c;

                }

                private void loadCombo(JComboBox combo) {

                        for (int i = 0; i < opList.length; i++) {

                                combo.addItem(opList[i]);

                        }

                }

                public Component getTableCellEditorComponent(JTable table,

                                Object value, boolean isSelected, int row, int column) {

                        if(value instanceof Integer) {

                                if(row < combos.size()) {

                                        JComboBox c = combos.get(row);

                                        c.setSelectedIndex((Integer)value);

                                        return c;

                                } else {

                                        for (int i = combos.size(); i <= row; i++) {

                                                JComboBox c = new JComboBox();

                                                loadCombo(c);

                                                combos.add(c);

                                                c.setSelectedIndex((Integer)value);

                                        }

                                        JComboBox c = combos.get(row);

                                        c.setSelectedIndex((Integer)value);

                                        return c;

                                }

                        }

                        return null;

                }

        }

        class GroupByRenderer extends JComboBox implements TableCellRenderer {

                private static final long            serialVersionUID   = 1L;

                public GroupByRenderer() {

                        for (int i = 0; i < opList.length; i++) {

                                this.addItem(opList[i]);

                        }

                }

                public Component getTableCellRendererComponent(JTable table,

                                Object value, boolean isSelected, boolean hasFocus, int row,

                                int column) {

                        return this;

                }

        }

        public TestTableModel() {

                super("TestValidationTableModel");

                List<JComboBox> combos = new ArrayList<JComboBox>();

                GroupByTableModel model = new GroupByTableModel(columnNames, combos);

                table = new JTable(model);

                table.setRowHeight(24);

                TableColumn column = table.getColumnModel().getColumn(1);

                GroupByRenderer render = new GroupByRenderer();

                column.setCellRenderer(render);

                column.setCellEditor(new GroupByEditor(combos));

                loadData();

                table.getModel().addTableModelListener(this);

                getContentPane().setLayout(new BorderLayout());

                getContentPane().add(table, BorderLayout.CENTER);

                but = new JButton();

                but.addActionListener(this);

                getContentPane().add(but, BorderLayout.SOUTH);

                setSize(w, h);

                setVisible(true);

        }

        private void loadData() {

                Object row[] = new Object[2];

                row[0] = "Line 1";

                row[1] = 1;

                ((GroupByTableModel)table.getModel()).addRow(row);

                row[0] = "Line 2";

                row[1] = 2;

                ((GroupByTableModel)table.getModel()).addRow(row);

        }

        public static void main(String[] args) {

                new TestTableModel();

        }

        public void tableChanged(TableModelEvent e) {

                System.out.println("Ha cambiado");

        }

        public void actionPerformed(ActionEvent e) {

                for (int i = 0; i < ((GroupByTableModel)table.getModel()).getRowCount(); i++) {

                        Object v1 = ((GroupByTableModel)table.getModel()).getValueAt(i, 0);

                        Object v2 = ((GroupByTableModel)table.getModel()).getValueAt(i, 1);

                        System.out.println(v1 + "..." + v2);

                }

        }

}