org.teavm.platform.plugin.ResourceReadCIntrinsic Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of teavm-platform Show documentation
A low-level classes that help to implement Java class library
The newest version!
/*
 *  Copyright 2018 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.platform.plugin;

import org.teavm.ast.InvocationExpr;
import org.teavm.backend.c.intrinsic.Intrinsic;
import org.teavm.backend.c.intrinsic.IntrinsicContext;
import org.teavm.model.ClassReaderSource;
import org.teavm.model.MethodReference;
import org.teavm.platform.metadata.Resource;
import org.teavm.platform.metadata.ResourceArray;
import org.teavm.platform.metadata.ResourceMap;

public class ResourceReadCIntrinsic implements Intrinsic {
    private ClassReaderSource classSource;

    public ResourceReadCIntrinsic(ClassReaderSource classSource) {
        this.classSource = classSource;
    }

    @Override
    public boolean canHandle(MethodReference method) {
        return classSource.isSuperType(Resource.class.getTypeName(), method.getClassName()).orElse(false);
    }

    @Override
    public void apply(IntrinsicContext context, InvocationExpr invocation) {
        if (invocation.getMethod().getClassName().equals(ResourceMap.class.getName())) {
            applyForResourceMap(context, invocation);
            return;
        } else if (invocation.getMethod().getClassName().equals(ResourceArray.class.getName())) {
            applyForResourceArray(context, invocation);
            return;
        }

        String name = invocation.getMethod().getName();
        if (name.startsWith("get")) {
            name = name.substring(3);
        } else if (name.startsWith("is")) {
            name = name.substring(2);
        } else {
            throw new IllegalArgumentException();
        }

        name = Character.toLowerCase(name.charAt(0)) + name.substring(1);

        String resourceName = "resources/" + context.escapeFileName(invocation.getMethod().getClassName()) + ".h";
        context.includes().includePath(resourceName);
        boolean isString = invocation.getMethod().getReturnType().isObject("java.lang.String");
        if (isString) {
            context.writer().print("teavm_dereferenceNullable(");
        }
        context.writer().print("TEAVM_FIELD(");
        context.emit(invocation.getArguments().get(0));
        context.writer().print(", ").print(context.names().forClass(invocation.getMethod().getClassName()));
        context.writer().print(", ").print(name).print(")");
        if (isString) {
            context.writer().print(")");
        }
    }

    private void applyForResourceMap(IntrinsicContext context, InvocationExpr invocation) {
        context.includes().includePath("resource.h");
        switch (invocation.getMethod().getName()) {
            case "keys":
                context.writer().print("teavm_resourceMapKeys((TeaVM_ResourceMap*) ");
                context.emit(invocation.getArguments().get(0));
                context.writer().print(")");
                break;
            case "has":
                context.writer().print("(teavm_lookupResource((TeaVM_ResourceMap*) ");
                context.emit(invocation.getArguments().get(0));
                context.writer().print(", (TeaVM_String*) ");
                context.emit(invocation.getArguments().get(1));
                context.writer().print(") != NULL)");
                break;
            case "get":
                context.writer().print("teavm_lookupResourceValue((TeaVM_ResourceMap*) ");
                context.emit(invocation.getArguments().get(0));
                context.writer().print(", (TeaVM_String*) ");
                context.emit(invocation.getArguments().get(1));
                context.writer().print(")");
                break;
        }
    }

    private void applyForResourceArray(IntrinsicContext context, InvocationExpr invocation) {
        context.includes().includePath("resource.h");
        switch (invocation.getMethod().getName()) {
            case "size":
                context.writer().print("((TeaVM_ResourceArray*) ");
                context.emit(invocation.getArguments().get(0));
                context.writer().print(")->size");
                break;
            case "get":
                context.writer().print("((TeaVM_ResourceArray*) ");
                context.emit(invocation.getArguments().get(0));
                context.writer().print(")->data[");
                context.emit(invocation.getArguments().get(1));
                context.writer().print("]");
                break;
        }
    }
}