TreeParent.java

// ============================================================================
//
// Copyright (C) 2006-2016 Talend Inc. - www.talend.com
//
// This source code is available under agreement available at
// %InstallDIR%\features\org.talend.rcp.branding.%PRODUCTNAME%\%PRODUCTNAME%license.txt
//
// You should have received a copy of the agreement
// along with this program; if not, write to Talend SA
// 9 rue Pages 92150 Suresnes, France
//
// ============================================================================
package com.amalto.workbench.models;

import java.util.ArrayList;

import com.amalto.workbench.utils.LocalTreeObjectRepository;

public class TreeParent extends TreeObject {

    private ArrayList children;

    public TreeParent(String name, TreeParent serverRoot, int type, Object wsKey, Object wsObject) {

        super(name, serverRoot, type, wsKey, wsObject);
        children = new ArrayList();
        this.setXObject(false);
    }

    public void addChild(TreeObject child) {
        if (child != null) {
            children.remove(child);
            children.add(child);
            child.setParent(this);
            child.fireEvent(IXObjectModelListener.ADD, this, child);
        }
        LocalTreeObjectRepository.getInstance().correctDisplayNameForCategory(child);
    }

    public void addChild(TreeObject child, boolean isMergeCategory) {
        if (child != null) {
            children.remove(child);
            children.add(child);
            child.setParent(this);
            if (isMergeCategory)
                child.fireEvent(IXObjectModelListener.ADD, this, child);
        }
        if (isMergeCategory)
            LocalTreeObjectRepository.getInstance().correctDisplayNameForCategory(child);
    }

    public void removeChild(TreeObject child) {
        if (child instanceof TreeParent) {
            TreeObject[] subchildren = ((TreeParent) child).getChildren();
            for (int i = 0; i < subchildren.length; i++) {
                ((TreeParent) child).removeChild(subchildren[i]);
            }
        }
        children.remove(child);
        child.setParent(null);
        this.fireEvent(IXObjectModelListener.DELETE, this, child);

    }

    public void removeChildFromUI(TreeObject child) {
        if (child instanceof TreeParent) {
            TreeObject[] subchildren = ((TreeParent) child).getChildren();
            for (int i = 0; i < subchildren.length; i++) {
                ((TreeParent) child).removeChildFromUI(subchildren[i]);
            }
        }
        children.remove(child);
        child.setParent(null);
        // this.fireEvent(IXObjectModelListener.DELETE, this, child);

    }

    public void removeChildren() {
        TreeObject[] allchildren = this.getChildren();
        for (int i = 0; i < allchildren.length; i++) {
            this.removeChild(allchildren[i]);
        }
    }

    public boolean containsChild(TreeObject child) {
        TreeObject[] allchildren = this.getChildren();
        for (int i = 0; i < allchildren.length; i++) {
            if (allchildren[i].getDisplayName().equals(child.getDisplayName()) && allchildren[i].getType() == child.getType())
                return true;
        }
        return false;
    }

    public TreeObject[] getChildren() {
        return (TreeObject[]) children.toArray(new TreeObject[children.size()]);
    }

    public void setChildren(ArrayList children) {
        this.children = children;
    }

    public ArrayList getChildrenList() {
        return children;
    }

    public boolean hasChildren() {
        return children.size() > 0;
    }

    public void synchronizeWith(TreeParent target) {
        TreeObject[] thisChildren = this.getChildren();
        TreeObject[] targetChildren = target.getChildren();
        // check what to remove
        for (int i = 0; i < thisChildren.length; i++) {
            TreeObject targetChild = null;
            for (int j = 0; j < targetChildren.length; j++) {
                if (targetChildren[j].getDisplayName().equals(thisChildren[i].getDisplayName())
                        && targetChildren[j].getClass().equals(thisChildren[i].getClass())
                        && targetChildren[j].getType() == thisChildren[i].getType()) {
                    targetChild = targetChildren[j];
                    break;
                }
            }// for target
            if (targetChild != null) {
                if (thisChildren[i] instanceof TreeParent)
                    ((TreeParent) thisChildren[i]).synchronizeWith((TreeParent) targetChild);
            } else {
                this.removeChild(thisChildren[i]);
            }
        }

        // check what to add
        for (int j = 0; j < targetChildren.length; j++) {
            boolean found = false;
            for (int i = 0; i < thisChildren.length; i++) {
                if (targetChildren[j].getDisplayName().equals(thisChildren[i].getDisplayName())
                        && targetChildren[j].getClass().equals(thisChildren[i].getClass())
                        && targetChildren[j].getType() == thisChildren[i].getType()) {
                    found = true;
                    break;
                }
            }// for thist
            if (!found) {
                this.addChild(targetChildren[j]);
                // fliu set the root node value of the new created child
                targetChildren[j].setServerRoot(this.getServerRoot());
            }
        }// for add

    }// synchronizewith

    public TreeObject findObject(int type, String name) {
        TreeObject[] allchildren = this.getChildren();
        TreeObject obj = null;
        for (int i = 0; i < allchildren.length; i++) {
            if (allchildren[i] instanceof TreeParent) {
                obj = ((TreeParent) allchildren[i]).findObject(type, name);
                if ((allchildren[i].getType() == type) && (allchildren[i].getDisplayName().equals(name)))
                    return allchildren[i];
            } else if ((allchildren[i].getType() == type) && (allchildren[i].getDisplayName().equals(name)))
                obj = allchildren[i];
        }
        return obj;
    }

}