view graal/com.oracle.truffle.codegen.processor/src/com/oracle/truffle/codegen/processor/node/MethodParser.java @ 8592:a80bf36c6a1e

Refactor to shared template method signature comparison.
author Christian Humer <christian.humer@gmail.com>
date Mon, 01 Apr 2013 11:52:38 +0200
parents 89006c76f737
children 54e227b2e199
line wrap: on
line source

/*
 * Copyright (c) 2012, 2012, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.
 *
 * This code 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 General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */
package com.oracle.truffle.codegen.processor.node;

import java.util.*;

import javax.lang.model.element.*;
import javax.lang.model.type.*;

import com.oracle.truffle.codegen.processor.*;
import com.oracle.truffle.codegen.processor.node.NodeFieldData.*;
import com.oracle.truffle.codegen.processor.template.*;
import com.oracle.truffle.codegen.processor.template.ParameterSpec.Cardinality;

public abstract class MethodParser<E extends TemplateMethod> extends TemplateMethodParser<NodeData, E> {

    public MethodParser(ProcessorContext context, NodeData node) {
        super(context, node);
    }

    public NodeData getNode() {
        return template;
    }

    protected ParameterSpec createValueParameterSpec(String valueName, NodeData nodeData, boolean optional) {
        return new ParameterSpec(valueName, nodeData, optional, Cardinality.ONE, true);
    }

    protected ParameterSpec createReturnParameterSpec() {
        return createValueParameterSpec("operation", getNode(), false);
    }

    @Override
    public boolean isParsable(ExecutableElement method) {
        return Utils.findAnnotationMirror(getContext().getEnvironment(), method, getAnnotationType()) != null;
    }

    @SuppressWarnings("unused")
    protected final MethodSpec createDefaultMethodSpec(ExecutableElement method, AnnotationMirror mirror, String shortCircuitName) {
        List<ParameterSpec> defaultParameters = new ArrayList<>();

        if (getNode().supportsFrame()) {
            defaultParameters.add(new ParameterSpec("frame", getContext().getTruffleTypes().getFrame(), true, false));
        }

        TypeMirror declaredType = Utils.findNearestEnclosingType(method).asType();

        List<TypeMirror> prefixTypes = new ArrayList<>();

        if (!method.getModifiers().contains(Modifier.STATIC) && !Utils.isAssignable(declaredType, template.getNodeType())) {
            prefixTypes.add(getNode().getTemplateType().asType());
        }

        for (NodeFieldData field : getNode().getFields()) {
            if (field.getKind() == FieldKind.FIELD) {
                ParameterSpec spec = new ParameterSpec(field.getName(), field.getType(), true, false);
                spec.setLocal(true);
                defaultParameters.add(spec);
            }
        }

        for (NodeFieldData field : getNode().getFields()) {
            if (field.getExecutionKind() == ExecutionKind.IGNORE) {
                continue;
            }

            if (field.getExecutionKind() == ExecutionKind.DEFAULT) {
                ParameterSpec spec = createValueParameterSpec(field.getName(), field.getNodeData(), false);
                if (field.getKind() == FieldKind.CHILDREN) {
                    spec.setCardinality(Cardinality.MULTIPLE);
                    spec.setIndexed(true);
                }
                defaultParameters.add(spec);
            } else if (field.getExecutionKind() == ExecutionKind.SHORT_CIRCUIT) {
                String valueName = field.getName();
                if (shortCircuitName != null && valueName.equals(shortCircuitName)) {
                    break;
                }

                defaultParameters.add(new ParameterSpec(shortCircuitValueName(valueName), getContext().getType(boolean.class), false, false));
                defaultParameters.add(createValueParameterSpec(valueName, field.getNodeData(), false));
            } else {
                assert false;
            }
        }

        return new MethodSpec(prefixTypes, createReturnParameterSpec(), defaultParameters);
    }

    private static String shortCircuitValueName(String valueName) {
        return "has" + Utils.firstLetterUpperCase(valueName);
    }

}