com.github.antlrjavaparser.adapter.NormalForStatementContextAdapter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of antlr-java-parser Show documentation

Antlr Java Parser aims to create a Java parser using Antlr 4 grammar rules.

There is a newer version: 1.0.17

/*
 * This file is part of antlr-java-parser.
 *
 *     antlr-java-parser is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     antlr-java-parser 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 General Public License for more details.
 *
 *     You should have received a copy of the GNU General Public License
 *     along with antlr-java-parser.  If not, see .
 */

package com.github.antlrjavaparser.adapter;

import com.github.antlrjavaparser.Java7Parser;
import com.github.antlrjavaparser.api.expr.Expression;
import com.github.antlrjavaparser.api.stmt.ForStmt;

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

/**
 * Created with IntelliJ IDEA.
 * User: Mike De Haan
 * Date: 12/20/12
 * Time: 1:07 PM
 * To change this template use File | Settings | File Templates.
 */
public class NormalForStatementContextAdapter implements Adapter {
    @Override
    public ForStmt adapt(Java7Parser.NormalForStatementContext context, AdapterParameters adapterParameters) {

        /*
            normalForStatement
                :    FOR LPAREN (forInit)? SEMI (expression)? SEMI (expressionList)? RPAREN statement
                ;

            forInit
                :   localVariableDeclaration
                |   expressionList
                ;

         */

        ForStmt forStmt = new ForStmt();
        AdapterUtil.setComments(forStmt, context, adapterParameters);
        AdapterUtil.setPosition(forStmt, context);

        forStmt.setBody(Adapters.getStatementContextAdapter().adapt(context.statement(), adapterParameters));

        if (context.forInit() != null) {
            if (context.forInit().localVariableDeclaration() != null) {
                List expressionList = new LinkedList();
                expressionList.add(Adapters.getLocalVariableDeclarationContextAdapter().adapt(context.forInit().localVariableDeclaration(), adapterParameters));
                forStmt.setInit(expressionList);
            } else if (context.forInit().expressionList() != null) {
                forStmt.setInit(Adapters.getExpressionListContextAdapter().adapt(context.forInit().expressionList(), adapterParameters));
            }
        }

        if (context.expression() != null) {
            forStmt.setCompare(Adapters.getExpressionContextAdapter().adapt(context.expression(), adapterParameters));
        }

        if (context.expressionList() != null) {
            forStmt.setUpdate(Adapters.getExpressionListContextAdapter().adapt(context.expressionList(), adapterParameters));
        }

        return forStmt;
    }
}