Selection.java

package net.squiz.matrix.matrixtree;

import java.util.*;
import javax.swing.tree.*;

public class Selection {

    private static List nodes = new ArrayList();

    // cannot instantiate
    private Selection() {}

    public static void setNodes(MatrixTreeNode[] newNodes) {
        // dont remove the current selection if the specified nodes
        // is null; keep the current selection
        if (newNodes == null)
            return;
        removeAllNodes();
        for (int i = 0; i < newNodes.length; i++) {
            nodes.add(newNodes[i]);
        }
    }

    public static MatrixTreeNode[] getNodes() {
        if (nodes != null)
            return (MatrixTreeNode[]) nodes.toArray(new MatrixTreeNode[nodes.size()]);
        return null;
    }

    public static void removeNode(MatrixTreeNode node) {
        nodes.remove(node);
    }

    public static void removeNodes(TreePath[] paths) {
        for (int i = 0; i < paths.length; i++) {
            nodes.remove(paths[i].getLastPathComponent());
        }
    }

    public static void removeNodes(MatrixTreeNode[] removeNodes) {
        for (int i = 0; i < removeNodes.length; i++) {
            nodes.remove(removeNodes[i]);
        }
    }

    public static void removeAllNodes() {
        nodes.clear();
    }

    public static void addNode(MatrixTreeNode node) {
        nodes.add(node);
    }
}