customcolumnspanel.java

It is all about project scheduling. GanttProject is a tool for creating a project schedule by means

package net.sourceforge.ganttproject.gui.taskproperties;

import java.awt.BorderLayout;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.GregorianCalendar;
import java.util.Iterator;
import java.util.Vector;

import javax.swing.JButton;
import javax.swing.JPanel;
import javax.swing.table.DefaultTableModel;

import net.sourceforge.ganttproject.Mediator;
import net.sourceforge.ganttproject.language.GanttLanguage;
import net.sourceforge.ganttproject.task.CustomColumn;
import net.sourceforge.ganttproject.task.CustomColumnsException;
import net.sourceforge.ganttproject.task.CustomColumnsStorage;

import org.jdesktop.jdnc.JNTable;
import org.jdesktop.swing.decorator.AlternateRowHighlighter;
import org.jdesktop.swing.decorator.Highlighter;
import org.jdesktop.swing.decorator.HighlighterPipeline;

/**
 * @author bbaranne Mar 4, 2005
 */
public class CustomColumnsPanel extends JPanel {
    private static GanttLanguage language = GanttLanguage.getInstance();

    private CustomColumnsStorage customColumnStorage = null;

    private Vector titles = null;

    private JNTable table = null;

    private JButton buttonAdd = null;

    private JButton buttonDel = null;

    public CustomColumnsPanel(CustomColumnsStorage customColHandler) {
        customColumnStorage = customColHandler;
        this.initComponents();
    }

    private void initComponents() {
        this.titles = new Vector(3);
        this.titles.add(language.getText("name"));
        this.titles.add(language.getText("typeClass"));
        this.titles.add(language.getText("default"));

        final CustomColumnTableModel model = new CustomColumnTableModel(
                customColumnStorage);
        table = new JNTable(model);
        table.setPreferredVisibleRowCount(10);
        this.add(table);
        table.getColumn((String) titles.get(0)).setPreferredWidth(150);
        table.getColumn((String) titles.get(1)).setPreferredWidth(150);
        table.getColumn((String) titles.get(2)).setPreferredWidth(200);

        table.setHighlighters(new HighlighterPipeline(new Highlighter[] {
                AlternateRowHighlighter.floralWhite,
                AlternateRowHighlighter.quickSilver }));

        table.getTable().setSortable(false);

        buttonAdd = new JButton(language.getText("add"));
        buttonAdd.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                Mediator.getUndoManager().undoableEdit("TaskPropertyNewColumn",
                        new Runnable() {
                            public void run() {
                                Mediator.getCustomColumnsManager()
                                        .addNewCustomColumn(null);
                                model.refreshData(Mediator
                                        .getCustomColumnsStorage());
                                repaint();
                            }
                        });
            }
        });
        this.add(buttonAdd, BorderLayout.EAST);

        buttonDel = new JButton(language.getText("delete"));
        buttonDel.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                int selectedRowsIndexes[] = table.getTable().getSelectedRows();
                for (int i = 0; i < selectedRowsIndexes.length; i++) {
                    String nameToDel = (String) model.getValueAt(
                            selectedRowsIndexes[i], 0);
                    Mediator.getCustomColumnsManager().deleteCustomColumn(
                            nameToDel);
                }
                model.refreshData(Mediator.getCustomColumnsStorage());
                repaint();
            }
        });
        this.add(buttonDel, BorderLayout.EAST);

    }

    /*
     * ----- INNER CLASSES -----
     */

    class CustomColumnTableModel extends DefaultTableModel {
        CustomColumnsStorage customColumnsHandler = null;

        public CustomColumnTableModel(CustomColumnsStorage ccHandler) {
            customColumnsHandler = ccHandler;
            this.columnIdentifiers = titles;

            this.dataVector = new Vector();

            Iterator it = customColumnsHandler.getCustomColumnsNames()
                    .iterator();
            while (it.hasNext()) {
                CustomColumn cc = null;
                try {
                    cc = customColumnsHandler.getCustomColumn((String) it
                            .next());
                } catch (CustomColumnsException e) {
                    // TODO Auto-generated catch block
                    e.printStackTrace();
                }
                Object[] o = { cc.getName(), cc.getType(), cc.getDefaultValue() };
                this.addRow(o);
            }
        }

        public void refreshData(CustomColumnsStorage cch) {
            customColumnsHandler = cch;
            this.dataVector = new Vector();

            Iterator it = customColumnsHandler.getCustomColumnsNames()
                    .iterator();
            while (it.hasNext()) {
                CustomColumn cc = null;
                try {
                    cc = customColumnsHandler.getCustomColumn((String) it
                            .next());
                } catch (CustomColumnsException e) {
                    e.printStackTrace();
                }
                Object[] o = { cc.getName(), cc.getType(), cc.getDefaultValue() };
                this.addRow(o);
            }
        }

        public String getColumnName(int column) {
            return (String) columnIdentifiers.get(column);
        }

        public Class getColumnClass(int column) {
            return String.class;
        }

        public boolean isCellEditable(int row, int col) {
            return col != 1;
        }

        public Object getValueAt(int row, int col) {
            String colName = (String) ((Vector) this.dataVector.get(row))
                    .get(0);
            CustomColumn res = null;
            try {
                res = customColumnsHandler.getCustomColumn(colName);
            } catch (CustomColumnsException e) {
                e.printStackTrace();
            }
            if (res != null) {
                switch (col) {
                case 0:
                    return res.getName();
                case 1: {
                    Class cl = res.getType();
                    if (cl.equals(String.class))
                        return language.getText("text");
                    if (GregorianCalendar.class.isAssignableFrom(cl))
                        return language.getText("date");
                    if (cl.equals(Boolean.class))
                        return language.getText("boolean");
                    if (cl.equals(Integer.class))
                        return language.getText("integer");
                    if (cl.equals(Double.class))
                        return language.getText("double");
                }
                case 2:
                    return res.getDefaultValue();
                }
            }
            return null;
        }

        public void setValueAt(Object o, int row, int col) {

            String oldName;
            Vector v = (Vector) dataVector.get(row);
            oldName = (String) v.get(0);

            if (col == 0) {
                if (!((String) o).equals(oldName)) {
                    if (!Mediator.getCustomColumnsStorage().exists((String) o)) {
                        v.setElementAt(o, col);
                        Mediator.getCustomColumnsManager()
                                .changeCustomColumnName(oldName, (String) o);
                    }

                }
            }
            if (col == 2) {

                try {
                    Mediator.getCustomColumnsManager()
                            .changeCustomColumnDefaultValue(oldName, o);
                    v.setElementAt(o, col);
                } catch (CustomColumnsException e) {
                    // TODO manage it
                    // e.printStackTrace();
                }
            }
        }
    }
}