org.eclipse.nebula.widgets.nattable.core/src/org/eclipse/nebula/widgets/nattable/columnCategories/Node.java

/*******************************************************************************
 * Copyright (c) 2012, 2020 Original authors and others.
 *
 * This program and the accompanying materials are made
 * available under the terms of the Eclipse Public License 2.0
 * which is available at https://www.eclipse.org/legal/epl-2.0/
 *
 * SPDX-License-Identifier: EPL-2.0
 *
 * Contributors:
 *     Original authors and others - initial API and implementation
 ******************************************************************************/
package org.eclipse.nebula.widgets.nattable.columnCategories;

import java.io.Serializable;
import java.util.ArrayList;
import java.util.List;

/**
 * Represents a node of the Tree class.
 */
public class Node implements Serializable {

    private static final long serialVersionUID = 7855L;

    public enum Type {
        ROOT, COLUMN, CATEGORY, UNKNOWN
    }

    private Type type;
    private String data;
    private List<Node> children;
    private Node parent;

    public Node(String data) {
        this(data, Type.UNKNOWN);
    }

    public Node(String newCategoryName, Type type) {
        setData(newCategoryName);
        setType(type);
    }

    public Node getParent() {
        return this.parent;
    }

    public void setParent(Node parent) {
        this.parent = parent;
    }

    public void setType(Type type) {
        this.type = type;
    }

    public Type getType() {
        return this.type;
    }

    /**
     * Return the children of Node. The Tree is represented by a single root
     * Node whose children are represented by a List&lt;Node&gt;. Each of these
     * Node elements in the List can have children. The getChildren() method
     * will return the children of a Node.
     *
     * @return the children of Node
     */
    public List<Node> getChildren() {
        if (this.children == null) {
            return new ArrayList<>();
        }
        return this.children;
    }

    /**
     * Returns the number of immediate children of this Node.
     *
     * @return the number of immediate children.
     */
    public int getNumberOfChildren() {
        if (this.children == null) {
            return 0;
        }
        return this.children.size();
    }

    /**
     * Adds a child to the list of children for this Node. The addition of the
     * first child will create a new List&lt;Node&gt;.
     *
     * @param child
     *            a Node object to set.
     * @return Child node just added
     */
    public Node addChild(Node child) {
        if (this.children == null) {
            this.children = new ArrayList<>();
        }
        this.children.add(child);
        child.setParent(this);
        return child;
    }

    public Node addChildCategory(String categoryName) {
        return addChild(new Node(categoryName, Type.CATEGORY));
    }

    public void addChildColumnIndexes(int... columnIndexes) {
        for (int columnIndex : columnIndexes) {
            addChild(new Node(String.valueOf(columnIndex), Type.COLUMN));
        }
    }

    /**
     * Inserts a Node at the specified position in the child list. Will throw an
     * ArrayIndexOutOfBoundsException if the index does not exist.
     *
     * @param index
     *            the position to insert at.
     * @param child
     *            the Node object to insert.
     */
    public void insertChildAt(int index, Node child) {
        if (index == getNumberOfChildren()) {
            // this is really an append
            addChild(child);
        } else {
            // just to throw the exception, and stop here
            this.children.get(index);
            this.children.add(index, child);
        }
    }

    /**
     * Remove the Node element at index index of the List&lt;Node&gt;.
     *
     * @param index
     *            the index of the element to delete.
     */
    public void removeChildAt(int index) {
        this.children.remove(index);
    }

    public String getData() {
        return this.data;
    }

    public void setData(String data) {
        this.data = data;
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append("{").append(this.type).append(",").append(getData()).append(",["); //$NON-NLS-1$ //$NON-NLS-2$ //$NON-NLS-3$
        int i = 0;
        for (Node e : getChildren()) {
            if (i > 0) {
                sb.append(","); //$NON-NLS-1$
            }
            sb.append(e.getData());
            i++;
        }
        sb.append("]").append("}"); //$NON-NLS-1$ //$NON-NLS-2$
        return sb.toString();
    }
}