NCall.java

/**
 * Copyright 2009, Google Inc.  All rights reserved.
 * Licensed to PSF under a Contributor Agreement.
 */
package org.python.indexer.ast;

import org.python.indexer.Scope;
import org.python.indexer.types.NClassType;
import org.python.indexer.types.NFuncType;
import org.python.indexer.types.NInstanceType;
import org.python.indexer.types.NType;
import org.python.indexer.types.NUnionType;
import org.python.indexer.types.NUnknownType;

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

public class NCall extends NNode {

    static final long serialVersionUID = 5212954751978100639L;

    public NNode func;
    public List<NNode> args;
    public List<NKeyword> keywords;
    public NNode kwargs;
    public NNode starargs;

    public NCall(NNode func, List<NNode> args, List<NKeyword> keywords,
                 NNode kwargs, NNode starargs) {
        this(func, args, keywords, kwargs, starargs, 0, 1);
    }

    public NCall(NNode func, List<NNode> args, List<NKeyword> keywords,
                 NNode kwargs, NNode starargs, int start, int end) {
        super(start, end);
        this.func = func;
        this.args = args;
        this.keywords = keywords;
        this.kwargs = kwargs;
        this.starargs = starargs;
        addChildren(func, kwargs, starargs);
        addChildren(args);
        addChildren(keywords);
    }

    @Override
    public NType resolve(Scope s) throws Exception {
        NType ft = resolveExpr(func, s);
        List<NType> argTypes = new ArrayList<NType>();
        for (NNode a : args) {
            argTypes.add(resolveExpr(a, s));
        }
        resolveList(keywords, s);
        resolveExpr(starargs, s);
        resolveExpr(kwargs, s);

        if (ft.isClassType()) {
            return setType(ft);  // XXX:  was new NInstanceType(ft)
        }

        if (ft.isFuncType()) {
            return setType(ft.asFuncType().getReturnType().follow());
        }

        if (ft.isUnknownType()) {
            NUnknownType to = new NUnknownType();
            NFuncType at = new NFuncType(to);
            NUnionType.union(ft, at);
            return setType(to);
        }

        addWarning("calling non-function " + ft);
        return setType(new NUnknownType());
    }

    @Override
    public String toString() {
        return "<Call:" + func + ":" + args + ":" + start() + ">";
    }

    @Override
    public void visit(NNodeVisitor v) {
        if (v.visit(this)) {
            visitNode(func, v);
            visitNodeList(args, v);
            visitNodeList(keywords, v);
            visitNode(kwargs, v);
            visitNode(starargs, v);
        }
    }
}