tapestry/tapestry-contrib/src/java/org/apache/tapestry/contrib/tree/simple/SimpleTreeDataModel.java

// Copyright 2004, 2005 The Apache Software Foundation
//
// 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 org.apache.tapestry.contrib.tree.simple;

import java.io.Serializable;
import java.util.Iterator;

import javax.swing.tree.TreePath;

import org.apache.tapestry.contrib.tree.model.ITreeDataModel;
import org.apache.tapestry.contrib.tree.model.ITreeNode;

/**
 * @author ceco
 */
public class SimpleTreeDataModel implements ITreeDataModel, Serializable
{

    private static final long serialVersionUID = 9215832847660213349L;

    protected ITreeNode m_objRootNode;

    /**
     * Constructor for SimpleTreeDataModel.
     */
    public SimpleTreeDataModel(ITreeNode objRootNode)
    {
        super();
        m_objRootNode = objRootNode;
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeDataModel#getRoot()
     */
    public Object getRoot()
    {
        return m_objRootNode;
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeDataModel#getChildCount(Object)
     */
    public int getChildCount(Object objParent)
    {
        ITreeNode objParentNode = (ITreeNode) objParent;

        return objParentNode.getChildCount();
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeDataModel#getChildren(Object)
     */
    public Iterator getChildren(Object objParent)
    {
        ITreeNode objParentNode = (ITreeNode) objParent;
        return objParentNode.getChildren().iterator();
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeDataModel#getObject(Object)
     */
    public Object getObject(Object objUniqueKey)
    {
        if (objUniqueKey != null)
        {
            TreePath objPath = (TreePath) objUniqueKey;
            return objPath.getLastPathComponent();
        }
        return null;
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeDataModel#getUniqueKey(Object,
     *      Object)
     */
    public Object getUniqueKey(Object objTarget, Object objParentUniqueKey)
    {
        TreePath objPath = (TreePath) objParentUniqueKey;
        Object objTargetUID = null;
        if (objPath != null)
        {
            objTargetUID = objPath.pathByAddingChild(objTarget);
        }
        else
        {
            objTargetUID = new TreePath(objTarget);
        }
        return objTargetUID;
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeDataModel#isAncestorOf(Object,
     *      Object)
     */
    public boolean isAncestorOf(Object objTargetUniqueKey,
            Object objParentUniqueKey)
    {
        TreePath objParentPath = (TreePath) objParentUniqueKey;
        TreePath objTargetPath = (TreePath) objTargetUniqueKey;
        boolean bResult = objParentPath.isDescendant(objTargetPath);
        return bResult;
    }

    /**
     * @see org.apache.tapestry.contrib.tree.model.ITreeDataModel#getParentUniqueKey
     */
    public Object getParentUniqueKey(Object objChildUniqueKey)
    {
        TreePath objChildPath = (TreePath) objChildUniqueKey;
        TreePath objParentPath = objChildPath.getParentPath();
        if (objParentPath == null) return null;
        return objParentPath.getLastPathComponent();
    }

}