ArgListCompiler.java

// Copyright (c) Corporation for National Research Initiatives

package org.python.compiler;

import java.util.Vector;

import org.python.parser.ParseException;
import org.python.parser.PythonGrammarTreeConstants;
import org.python.parser.Visitor;
import org.python.parser.ast.Assign;
import org.python.parser.ast.Name;
import org.python.parser.ast.Suite;
import org.python.parser.ast.Tuple;
import org.python.parser.ast.argumentsType;
import org.python.parser.ast.exprType;
import org.python.parser.ast.stmtType;

public class ArgListCompiler extends Visitor implements PythonGrammarTreeConstants {
    public boolean arglist, keywordlist;
    public exprType[] defaults;
    public Vector names;
    public Vector fpnames;
    public Vector init_code;

    public ArgListCompiler() {
        arglist = keywordlist = false;
        defaults = null;
        names = new Vector();
        fpnames = new Vector();
        init_code = new Vector();
    }

    public void reset() {
        arglist = keywordlist = false;
        defaults = null;
        names.removeAllElements();
        init_code.removeAllElements();
    }

    public void appendInitCode(Suite node) {
        int n = node.body.length;
        stmtType[] newtree = new stmtType[init_code.size() + n];
        init_code.copyInto(newtree);
        System.arraycopy(node.body, 0, newtree, init_code.size(), n);
        node.body = newtree;
    }

    public exprType[] getDefaults() {
        return defaults;
    }

    public void visitArgs(argumentsType args) throws Exception {
        for (int i = 0; i < args.args.length; i++) {
            String name = (String) visit(args.args[i]);
            names.addElement(name);
            if (args.args[i] instanceof Tuple) {
                Assign ass = new Assign(new exprType[] { args.args[i] }, new Name(name, Name.Load, args.args[i]),
                        args.args[i]);
                init_code.addElement(ass);
            }
        }
        if (args.vararg != null) {
            arglist = true;
            names.addElement(args.vararg);
        }
        if (args.kwarg != null) {
            keywordlist = true;
            names.addElement(args.kwarg);
        }

        defaults = args.defaults;
        for (int i = 0; i < defaults.length; i++) {
            if (defaults[i] == null)
                throw new ParseException("non-default argument follows default argument", args.args[args.args.length
                        - defaults.length + i]);
        }
    }

    public Object visitName(Name node) throws Exception {
        if (node.ctx != Name.Store)
            return null;

        if (fpnames.contains(node.id)) {
            throw new ParseException("duplicate argument name found: " + node.id, node);
        }
        fpnames.addElement(node.id);
        return node.id;
    }

    public Object visitTuple(Tuple node) throws Exception {
        StringBuffer name = new StringBuffer("(");
        int n = node.elts.length;
        for (int i = 0; i < n - 1; i++) {
            name.append(visit(node.elts[i]));
            name.append(", ");
        }
        name.append(visit(node.elts[n - 1]));
        name.append(")");
        return name.toString();
    }
}