/*
 *                 Sun Public License Notice
 *
 * The contents of this file are subject to the Sun Public License
 * Version 1.0 (the "License"). You may not use this file except in
 * compliance with the License. A copy of the License is available at
 * http://www.sun.com/
 *
 * The Original Code is NetBeans. The Initial Developer of the Original
 * Code is Sun Microsystems, Inc. Portions Copyright 1997-2003 Sun
 * Microsystems, Inc. All Rights Reserved.
 */
package org.netbeans.modules.javacore.jmiimpl.javamodel;

import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
import java.util.Set;
import org.netbeans.jmi.javamodel.Element;
import org.netbeans.jmi.javamodel.Expression;
import org.netbeans.jmi.javamodel.Statement;
import org.netbeans.jmi.javamodel.WhileStatement;
import org.netbeans.lib.java.parser.ASTree;
import org.netbeans.mdr.storagemodel.StorableObject;
import org.netbeans.modules.javacore.parser.ASTProvider;

/**
 *
 * @author  mm109185
 */
public abstract class WhileStatementImpl extends ConditionImpl implements WhileStatement {
    protected Statement body = null;
    
    /** Creates a new instance of WhileStatementImpl */
    public WhileStatementImpl(StorableObject o) {
        super(o);
    }
    
    public void setBody(Statement body) {
        objectChanged(CHANGED_BODY);
        changeChild(getBody(), body);
        this.body = body;
    }

    public List getChildren() {
        List list = new ArrayList(4);
        addIfNotNull(list, getBody());
        list.addAll(super.getChildren());
        return list;
    }
    
    protected void initChildren() {
        childrenInited = false;
        ASTree tree = getASTree();
        if (tree != null) {
            ASTree[] parts = tree.getSubTrees();
            expression = (Expression) initOrCreate(expression, parts[0]);
            body = (Statement) initOrCreate(body, parts[1]);
        }
        childrenInited = true;
    }
    
    public Statement getBody() {
        if (!childrenInited) {
            initChildren();
        }
        return body;
    }
    
    public String getSourceText() {
        String origElem;
        if ((origElem = checkChange()) != null)
            return origElem;
        StringBuffer buf = new StringBuffer();
        StatementImpl cond = (StatementImpl) getExpression();
        StatementImpl body = (StatementImpl) getBody();
        
        buf.append('\n');
        buf.append(getIndentation());
        formatElementPart(WHILE_KEYWORD, buf);
        formatElementPart(STMT_OPEN_BRACKET);
        buf.append(cond.getSourceText());
        formatElementPart(STMT_CLOSE_BRACKET);
        buf.append(body.getSourceText());
        return buf.toString();
    }
    
    public void getDiff(List diff){
        ASTProvider parser = getParser();
        
        ASTree[] children = getASTree().getSubTrees();
        
        getChildDiff(diff, parser, children[0], (MetadataElement) getExpression(), CHANGED_EXPRESSION);
        getChildDiff(diff, parser, children[1], (MetadataElement) getBody(), CHANGED_BODY);
    }
    
    void setData(Expression expression, Statement body) {
        changeChild(null, expression);
        changeChild(null, body);
        this.expression = expression;
        this.body = body;
    }

    protected void _delete() {
        // --- delete components -------------------------------------------
        if (childrenInited) {
            deleteChild(body);
        }
        // --- delete links -----------------------------------------------
        // no links to delete
        // --- call super ---------------------------------------
        super._delete();
    }
    
    public void replaceChild(Element oldElement,Element newElement) {
        if (childrenInited) {
            if (oldElement.equals(body)) {
                setBody((Statement)newElement);
            } else
                super.replaceChild(oldElement,newElement);
        }
    }
}