org.walkmod.javalang.ast.stmt.AssertStmt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of javalang Show documentation
Java 8 parser library
There is a newer version: 4.8.8
/*
 * Copyright (C) 2013 Raquel Pau and Albert Coroleu.
 * 
 * Walkmod is free software: you can redistribute it and/or modify it under the terms of the GNU
 * Lesser General Public License as published by the Free Software Foundation, either version 3 of
 * the License, or (at your option) any later version.
 * 
 * Walkmod is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even
 * the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser
 * General Public License for more details.
 * 
 * You should have received a copy of the GNU Lesser General Public License along with Walkmod. If
 * not, see .
 */
package org.walkmod.javalang.ast.stmt;

import java.util.LinkedList;
import java.util.List;

import org.walkmod.javalang.ast.Node;
import org.walkmod.javalang.ast.expr.Expression;
import org.walkmod.javalang.visitors.GenericVisitor;
import org.walkmod.javalang.visitors.VoidVisitor;

/**
 * @author Julio Vilmar Gesser
 */
public final class AssertStmt extends Statement {

    private Expression check;

    private Expression msg;

    public AssertStmt() {}

    public AssertStmt(Expression check) {
        setCheck(check);
    }

    public AssertStmt(Expression check, Expression msg) {
        setCheck(check);
        setMessage(msg);
    }

    public AssertStmt(int beginLine, int beginColumn, int endLine, int endColumn, Expression check, Expression msg) {
        super(beginLine, beginColumn, endLine, endColumn);
        setCheck(check);
        setMessage(msg);
    }

    @Override
    public boolean removeChild(Node child) {
        boolean result = false;
        if (child != null) {
            if (check == child) {
                check = null;
                result = true;
            }

            if (!result) {
                if (msg == child) {
                    msg = null;
                    result = true;
                }
            }
        }
        if (result) {
            updateReferences(child);
        }
        return result;
    }

    @Override
    public List getChildren() {
        List children = new LinkedList();
        if (check != null) {
            children.add(check);
        }
        if (msg != null) {
            children.add(msg);
        }
        return children;
    }

    @Override
    public  R accept(GenericVisitor v, A arg) {
        if (!check()) {
            return null;
        }
        return v.visit(this, arg);
    }

    @Override
    public  void accept(VoidVisitor v, A arg) {
        if (check()) {
            v.visit(this, arg);
        }
    }

    public Expression getCheck() {
        return check;
    }

    public Expression getMessage() {
        return msg;
    }

    public void setCheck(Expression check) {
        if (this.check != null) {
            updateReferences(this.check);
        }
        this.check = check;
        setAsParentNodeOf(check);
    }

    public void setMessage(Expression msg) {
        if (this.msg != null) {
            updateReferences(this.msg);
        }
        this.msg = msg;
        setAsParentNodeOf(msg);
    }

    @Override
    public boolean replaceChildNode(Node oldChild, Node newChild) {
        boolean updated = false;
        if (oldChild == check) {
            setCheck((Expression) newChild);
            updated = true;
        }
        if (!updated) {
            if (oldChild == msg) {
                setMessage((Expression) newChild);
                updated = true;
            }
        }
        return updated;
    }

    @Override
    public AssertStmt clone() throws CloneNotSupportedException {
        return new AssertStmt(clone(check), clone(msg));
    }
}