org.teavm.model.MethodHolder Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of teavm-core Show documentation
TeaVM compiler and SPI
There is a newer version: 0.10.2
/*
 *  Copyright 2012 Alexey Andreev.
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *       http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.teavm.model;

import java.util.Objects;
import java.util.function.Function;

public class MethodHolder extends MemberHolder implements MethodReader {
    private MethodDescriptor descriptor;
    private GenericTypeParameter[] typeParameters;
    private GenericValueType genericReturnType;
    private GenericValueType[] genericParameterTypes;
    private ClassHolder owner;
    private Program program;
    private Function programSupplier;
    private AnnotationValue annotationDefault;
    private AnnotationContainer[] parameterAnnotations;
    private MethodReference reference;

    public MethodHolder(MethodDescriptor descriptor) {
        super(descriptor.getName());
        this.descriptor = descriptor;
        parameterAnnotations = new AnnotationContainer[descriptor.parameterCount()];
        for (int i = 0; i < parameterAnnotations.length; ++i) {
            parameterAnnotations[i] = new AnnotationContainer();
        }
    }

    public MethodHolder(String name, ValueType... signature) {
        this(new MethodDescriptor(name, signature));
    }

    @Override
    public ValueType getResultType() {
        return descriptor.getResultType();
    }

    @Override
    public GenericValueType getGenericResultType() {
        return genericReturnType;
    }

    @Override
    public int genericParameterCount() {
        return genericParameterTypes != null ? genericParameterTypes.length : 0;
    }

    @Override
    public GenericValueType genericParameterType(int index) {
        return genericParameterTypes != null ? genericParameterTypes[index] : null;
    }

    public void setGenericSignature(GenericValueType returnType, GenericValueType[] parameterTypes) {
        genericReturnType = Objects.requireNonNull(returnType);
        genericParameterTypes = parameterTypes.clone();
    }

    public void removeGenericSignature() {
        genericReturnType = null;
        genericParameterTypes = null;
    }

    @Override
    public GenericTypeParameter[] getTypeParameters() {
        return typeParameters != null ? typeParameters.clone() : new GenericTypeParameter[0];
    }

    public void setTypeParameters(GenericTypeParameter[] typeParameters) {
        this.typeParameters = typeParameters != null ? typeParameters.clone() : null;
    }

    @Override
    public int parameterCount() {
        return descriptor.parameterCount();
    }

    @Override
    public ValueType[] getSignature() {
        return descriptor.getSignature();
    }

    @Override
    public ValueType parameterType(int index) {
        return descriptor.parameterType(index);
    }

    @Override
    public ValueType[] getParameterTypes() {
        return descriptor.getParameterTypes();
    }

    @Override
    public AnnotationContainer parameterAnnotation(int index) {
        return parameterAnnotations[index];
    }

    @Override
    public AnnotationContainer[] getParameterAnnotations() {
        return parameterAnnotations.clone();
    }

    @Override
    public String getOwnerName() {
        return owner != null ? owner.getName() : null;
    }

    ClassHolder getOwner() {
        return owner;
    }

    void setOwner(ClassHolder owner) {
        reference = null;
        this.owner = owner;
    }

    @Override
    public MethodDescriptor getDescriptor() {
        return descriptor;
    }

    @Override
    public MethodReference getReference() {
        if (owner == null) {
            return null;
        }
        if (reference == null) {
            reference = new MethodReference(owner.getName(), descriptor);
        }
        return reference;
    }

    public void updateReference(ReferenceCache cache) {
        MethodReference reference = getReference();
        if (reference != null) {
            this.reference = cache.getCached(reference);
        }
    }

    @Override
    public Program getProgram() {
        if (program == null && programSupplier != null) {
            program = programSupplier.apply(this);
            programSupplier = null;
        }
        return program;
    }

    public void setProgram(Program program) {
        this.program = program;
        this.programSupplier = null;
    }

    public boolean hasProgram() {
        return program != null || programSupplier != null;
    }

    public void setProgramSupplier(Function programSupplier) {
        this.program = null;
        this.programSupplier = programSupplier;
    }

    @Override
    public AnnotationValue getAnnotationDefault() {
        return annotationDefault;
    }

    public void setAnnotationDefault(AnnotationValue annotationDefault) {
        this.annotationDefault = annotationDefault;
    }
}