treesorter.java

Tree taglib,生成树的标签库

/*
    Copyright 2004 Jenkov Development

    Licensed under the Apache License, Version 2.0 (the "License");
    you may not use this file except in compliance with the License.
    You may obtain a copy of the License at

        http://www.apache.org/licenses/LICENSE-2.0

    Unless required by applicable law or agreed to in writing, software
    distributed under the License is distributed on an "AS IS" BASIS,
    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    See the License for the specific language governing permissions and
    limitations under the License.
*/



package com.jenkov.prizetags.tree.impl;

import com.jenkov.prizetags.tree.itf.ITreeNode;

import java.util.Comparator;
import java.util.Collections;
import java.util.Iterator;

/**
 * @author Jakob Jenkov - Copyright 2005 Jenkov Development
 */
public class TreeSorter {

    public static void sort(ITreeNode node, Comparator comparator){
        Collections.sort(node.getChildren(), comparator);
    }

    public static void sortRecursive(ITreeNode node, Comparator comparator){
        sort(node, comparator);
        Iterator iterator = node.getChildren().iterator();
        while(iterator.hasNext()){
            ITreeNode child = (ITreeNode) iterator.next();
            sortRecursive(child, comparator);
        }
    }

    public static void sortById(ITreeNode node){
        sort(node, createIdComparator());
    }

    public static void sortRecursiveById(ITreeNode node){
        sortRecursive(node, createIdComparator());
    }

    public static void sortByName(ITreeNode node){
        sort(node, createNameComparator());
    }

    public static void sortRecursiveByName(ITreeNode node){
        sortRecursive(node, createNameComparator());
    }

    public static void sortByType(ITreeNode node){
        sort(node, createTypeComparator());
    }

    public static void sortRecursiveByType(ITreeNode node){
        sortRecursive(node, createTypeComparator());
    }


    private static Comparator createIdComparator() {
        return new Comparator(){
            public int compare(Object o1, Object o2) {
                return ((ITreeNode)o1).getId().compareTo(((ITreeNode)o2).getId());
            }
        };
    }

    private static Comparator createNameComparator() {
        return new Comparator(){
            public int compare(Object o1, Object o2) {
                return ((ITreeNode)o1).getName().compareTo(((ITreeNode)o2).getName());
            }
        };
    }

    private static Comparator createTypeComparator() {
        return new Comparator(){
            public int compare(Object o1, Object o2) {
                return ((ITreeNode)o1).getType().compareTo(((ITreeNode)o2).getType());
            }
        };
    }

}