VAqua.src.org.violetlib.treetable.DefaultTreeColumnModel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vaqua Show documentation
An improved native Swing look and feel for macOS
The newest version!
/*
 *    This program is free software: you can redistribute it and/or modify
 *    it under the terms of the GNU Lesser General Public License as published
 *    by the Free Software Foundation, either version 3 of the License, or
 *    (at your option) any later version.
 *
 *    This program is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *    GNU Lesser General Public License for more details.
 *
 *    You should have received a copy of the GNU Lesser General Public License
 *    along with this program.  If not, see .
 */
package org.violetlib.treetable;

import java.util.Arrays;
import java.util.Collections;
import java.util.List;

import javax.swing.tree.TreeNode;

public class DefaultTreeColumnModel extends AbstractTreeColumnModel {

    public DefaultTreeColumnModel(int columns) {
        this(null, columns);
    }

    public DefaultTreeColumnModel(TreeTableNode root) {
        this(root, Collections.nCopies(root.getColumnCount(), null));
    }

    public DefaultTreeColumnModel(TreeTableNode root, Object ... columnNames) {
        this(root, Arrays.asList(columnNames));
    }

    public DefaultTreeColumnModel(TreeTableNode root, List columnNames) {
        this.root = root;
        this.columnNames = columnNames;
    }

    private List columnNames;

    private int editableColumns = -1;

    private int hierarchialColumn;

    private TreeTableNode root;

    @Override
    public String getColumnName(int column) {
        Object name = columnNames.get(column);
        String str = name == null ? null : name.toString();
        return str != null ? str : super.getColumnName(column);
    }

    @Override
    public Class getColumnClass(int column) {
        if (root == null)
            return Object.class;
        Object value = root.getValueAt(column);
        if (value instanceof Class)
            return (Class)value;
        return value == null ? Object.class : value.getClass();
    }

    @Override
    public int getColumnCount() {
        return columnNames.size();
    }

    @Override
    public Object getValueAt(Object node, int column) {
        if (column == hierarchialColumn) {
            return "";
        }
        return ((TreeTableNode)node).getValueAt(column);
    }

    @Override
    public void setValueAt(Object value, Object node, int column) {
        ((MutableTreeTableNode)node).setValueAt(
                convertValue(value, node, column), column);
        fireTreeColumnChanged(pathToRoot(root, (TreeNode)node), column);
    }

    @Override
    public boolean isCellEditable(Object node, int column) {
        return (editableColumns & (1 << column)) != 0;
    }

    public void setAllColumnsEditable(boolean editable) {
        editableColumns = editable ? -1 : 0;
    }

    public void setColumnEditable(int column, boolean editable) {
        if (column > 31)
            throw new IllegalArgumentException();
        if (editable) {
            editableColumns |= 1 << column;
        } else {
            editableColumns &= ~(1 << column);
        }
    }

    @Override
    public int getHierarchicalColumn() {
        return hierarchialColumn;
    }

    public void setHierarchialColumn(int column) {
        hierarchialColumn = column;
    }
}