sourcecodefilter.filter.IfElseFilter.java Source code

Introduction

Here is the source code for sourcecodefilter.filter.IfElseFilter.java

Source

/*
 * ScratchToCatrobat: A tool for converting Scratch projects into Catrobat programs.
 * Copyright (C) 2013-2016 The Catrobat Team
 * (http://developer.catrobat.org/credits)
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * An additional term exception under section 7 of the GNU Affero
 * General Public License, version 3, is available at
 * http://developer.catrobat.org/license_additional_term
 *
 * This program 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 Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see http://www.gnu.org/licenses/.
 */

package sourcecodefilter.filter;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import org.eclipse.jdt.core.dom.ASTNode;
import org.eclipse.jdt.core.dom.ASTVisitor;
import org.eclipse.jdt.core.dom.AbstractTypeDeclaration;
import org.eclipse.jdt.core.dom.Block;
import org.eclipse.jdt.core.dom.BodyDeclaration;
import org.eclipse.jdt.core.dom.CompilationUnit;
import org.eclipse.jdt.core.dom.IfStatement;
import org.eclipse.jdt.core.dom.MethodDeclaration;

import sourcecodefilter.ConverterRelevantCatroidSource;

public class IfElseFilter extends ASTVisitor {

    final private ConverterRelevantCatroidSource catroidSource;
    final private Set<String> ifElseBlocksToBeRemoved;

    public IfElseFilter(ConverterRelevantCatroidSource catroidSource, Set<String> ifElseBlocksToBeRemoved) {
        this.catroidSource = catroidSource;
        this.ifElseBlocksToBeRemoved = ifElseBlocksToBeRemoved;
    }

    @SuppressWarnings("unchecked")
    public void removeUnallowedIfElseBlocks() {
        CompilationUnit cu = catroidSource.getSourceAst();
        final List<AbstractTypeDeclaration> types = cu.types();
        assert types.size() > 0;

        for (AbstractTypeDeclaration abstractTypeDecl : types) {
            for (BodyDeclaration bodyDecl : new ArrayList<BodyDeclaration>(abstractTypeDecl.bodyDeclarations())) {
                if (bodyDecl.getNodeType() != ASTNode.METHOD_DECLARATION) {
                    continue;
                }

                MethodDeclaration methodDeclaration = (MethodDeclaration) bodyDecl;
                Block body = methodDeclaration.getBody();
                if ((body == null) || (body.statements().size() == 0)) {
                    continue;
                }
                methodDeclaration.accept(this);
            }
        }
    }

    public boolean visit(final IfStatement node) {
        final String expression = node.getExpression().toString();
        for (String expectedExpression : ifElseBlocksToBeRemoved) {
            if (expression.equals(expectedExpression)) {
                node.delete();
            }
        }
        return false;
    }
}