root/org.axdt.as3/src/org/axdt/as3/imp/treeModelBuilder/AS3TreeModelBuilder.java @ 0f874443124e53e6b4e3d1e9dd1c4c5a30e9b80c

package org.axdt.as3.imp.treeModelBuilder;

import org.eclipse.imp.services.base.TreeModelBuilderBase;

import org.axdt.as3.imp.parser.Ast.*;

public class AS3TreeModelBuilder extends TreeModelBuilderBase {

        public AS3TreeModelBuilder() {
        }

        @Override
        public void visitTree(Object root) {
                if (root == null) return;
                ASTNode rootNode = (ASTNode) root;
                AS3ModelVisitor visitor = new AS3ModelVisitor();

                rootNode.accept(visitor);
        }

        private class AS3ModelVisitor extends AbstractVisitor {
                // private StringBuffer fRHSLabel;

                @Override
                public void unimplementedVisitor(String s) {
                }

                public boolean visit(VariableBinding n) {
                        if (n.getParent().getParent() instanceof VariableDefinition) {
                                createSubItem(n);
                                return true;
                        }
                        return false;
                }

                public boolean visit(FunctionDefinition n) {
                        createSubItem(n);
                        return false;
                }

                @Override
                public boolean visit(ImportDirectiveList n) {
                        pushSubItem(n);
                        return true;
                }

                public boolean visit(ImportDirective n) {
                        createSubItem(n);
                        return true;
                }

                @Override
                public void endVisit(ImportDirectiveList n) {
                        popSubItem();
                }

                @Override
                public boolean visit(ClassDefinition n) {
                        pushSubItem(n);
                        return true;
                }

                @Override
                public void endVisit(ClassDefinition n) {
                        popSubItem();
                }

                @Override
                public boolean visit(InterfaceDefinition n) {
                        pushSubItem(n);
                        return true;
                }

                @Override
                public void endVisit(InterfaceDefinition n) {
                        popSubItem();
                }

                @Override
                public boolean visit(PackageDefinition n) {
                        pushSubItem(n);
                        return true;
                }

                @Override
                public void endVisit(PackageDefinition n) {
                        popSubItem();
                }
        }
}