root/trunk/ProjectFortress/src/com/sun/fortress/compiler/index/ParametricOperator.java @ 3915

/*******************************************************************************
    Copyright 2009 Sun Microsystems, Inc.,
    4150 Network Circle, Santa Clara, California 95054, U.S.A.
    All rights reserved.

    U.S. Government Rights - Commercial software.
    Government users are subject to the Sun Microsystems, Inc. standard
    license agreement and applicable provisions of the FAR and its supplements.

    Use is subject to license terms.

    This distribution may include materials developed by third parties.

    Sun, Sun Microsystems, the Sun logo and Java are trademarks or registered
    trademarks of Sun Microsystems, Inc. in the U.S. and other countries.
 ******************************************************************************/

package com.sun.fortress.compiler.index;

import java.util.Collections;
import java.util.List;

import com.sun.fortress.nodes.ArrowType;
import com.sun.fortress.nodes.BaseType;
import com.sun.fortress.nodes.Expr;
import com.sun.fortress.nodes.FnDecl;
import com.sun.fortress.nodes.Id;
import com.sun.fortress.nodes.Node;
import com.sun.fortress.nodes.NodeDepthFirstVisitor;
import com.sun.fortress.nodes.NodeUpdateVisitor;
import com.sun.fortress.nodes.Op;
import com.sun.fortress.nodes.Param;
import com.sun.fortress.nodes.StaticArg;
import com.sun.fortress.nodes.StaticParam;
import com.sun.fortress.nodes.Type;
import com.sun.fortress.nodes_util.NodeUtil;
import com.sun.fortress.nodes_util.Span;
import com.sun.fortress.useful.NI;

import edu.rice.cs.plt.lambda.SimpleBox;
import edu.rice.cs.plt.tuple.Option;

/**
 * This is a operator declared in a trait or object whose name is an opr parameter.
 * It needs to be promoted to top-level so it can match OpExprs appropriately.
 */
public class ParametricOperator extends FunctionalMethod {
    Op _name;

    public ParametricOperator(FnDecl ast, Id declaringTrait) {
        super(ast, declaringTrait);
        _name = (Op)NodeUtil.getName(ast);
        putThunk(SimpleBox.make(NodeUtil.getReturnType(_ast)));
    }

    public FnDecl ast() { return _ast; }

    public Op name() { return _name; }

    @Override
    public Span getSpan() { return NodeUtil.getSpan(_ast); }

        @Override
        public Option<Expr> body() {
                return _ast.accept(new NodeDepthFirstVisitor<Option<Expr>>(){
                        @Override
                        public Option<Expr> defaultCase(Node that) {
                                return Option.none();
                        }
                        @Override
                        public Option<Expr> forFnDecl(FnDecl that) {
                            return that.getBody();
                        }
                });
        }


        @Override
        public List<Param> parameters() {
                return NodeUtil.getParams(_ast);
        }

        @Override
        public List<StaticParam> staticParameters() {
                return NodeUtil.getStaticParams(_ast);
        }

        @Override
        public List<BaseType> thrownTypes() {
                if(  NodeUtil.getThrowsClause(_ast).isSome() )
                        return Collections.emptyList();
                else
                        return Collections.unmodifiableList(NodeUtil.getThrowsClause(_ast).unwrap());
        }

        @Override
        public Functional instantiate(List<StaticParam> params, List<StaticArg> args) {
                // TODO Auto-generated method stub
                return NI.nyi();
        }

        @Override
        public Functional acceptNodeUpdateVisitor(NodeUpdateVisitor visitor) {
                return new FunctionalMethod((FnDecl)this.ast().accept(visitor), this._declaringTrait);
        }
}