RhinoTreeVisitor.java

/*
 * This file is part of GumTree.
 *
 * GumTree 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.
 *
 * GumTree 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 GumTree.  If not, see <http://www.gnu.org/licenses/>.
 *
 * Copyright 2011-2015 Jean-Rémy Falleri <jr.falleri@gmail.com>
 * Copyright 2011-2015 Floréal Morandat <florealm@gmail.com>
 */

package com.github.gumtreediff.gen.js;

import java.util.HashMap;
import java.util.Map;

import org.mozilla.javascript.Token;
import org.mozilla.javascript.ast.AstNode;
import org.mozilla.javascript.ast.AstRoot;
import org.mozilla.javascript.ast.Name;
import org.mozilla.javascript.ast.NodeVisitor;
import org.mozilla.javascript.ast.NumberLiteral;
import org.mozilla.javascript.ast.StringLiteral;

import com.github.gumtreediff.tree.ITree;
import com.github.gumtreediff.tree.TreeContext;

public class RhinoTreeVisitor implements NodeVisitor {

    private Map<AstNode, ITree> trees;
    private TreeContext context;

    public RhinoTreeVisitor(AstRoot root) {
        trees = new HashMap<>();
        context = new TreeContext();
        ITree tree = buildTree(root);
        context.setRoot(tree);
    }

    public TreeContext getTree(AstNode root) {
        return context;
    }

    @Override
    public boolean visit(AstNode node) {
        if (node instanceof AstRoot)
            return true;
        else {
            ITree t = buildTree(node);
            ITree p = trees.get(node.getParent());
            p.addChild(t);

            if (node instanceof Name) {
                Name name = (Name) node;
                t.setLabel(name.getIdentifier());
            } else if ( node instanceof StringLiteral) {
                StringLiteral literal = (StringLiteral) node;
                t.setLabel(literal.getValue());
            } else if ( node instanceof NumberLiteral) {
                NumberLiteral l = (NumberLiteral) node;
                t.setLabel(l.getValue());
            }

            return true;
        }
    }

    private ITree buildTree(AstNode node)  {
        ITree t = context.createTree(node.getType(), ITree.NO_LABEL, Token.typeToName(node.getType()));
        t.setPos(node.getAbsolutePosition());
        t.setLength(node.getLength());
        trees.put(node, t);
        return t;
    }

}