org.jruby.truffle.aot.JRubySupport Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jruby-truffle Show documentation
The newest version!
/*
 * Copyright (c) 2015, 2016 Oracle and/or its affiliates. All rights reserved. This
 * code is released under a tri EPL/GPL/LGPL license. You can use it,
 * redistribute it and/or modify it under the terms of the:
 *
 * Eclipse Public License version 1.0
 * GNU General Public License version 2
 * GNU Lesser General Public License version 2.1
 *
 *
 * Some of the code in this class is modified from org.jruby.util.StringSupport,
 * licensed under the same EPL1.0/GPL 2.0/LGPL 2.1 used throughout.
 */
package org.jruby.truffle.aot;

import org.jcodings.Encoding;
import org.jcodings.EncodingDB;
import org.jcodings.util.ArrayReader;
import org.jcodings.util.CaseInsensitiveBytesHash;

import java.io.DataInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.FileVisitResult;
import java.nio.file.Path;
import java.nio.file.attribute.BasicFileAttributes;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public final class JRubySupport {
    public static final Map allEncodings = getEncodings();
    public static final Map allJCodingsTables = getJcodingsTables();

    private static Map getEncodings() {
        final Map encodings = new HashMap<>();
        final CaseInsensitiveBytesHash encodingdb = EncodingDB.getEncodings();
        for (EncodingDB.Entry entry : encodingdb) {
            final String encodingClassName = entry.getEncodingClass();
            final Encoding encoding = Encoding.load(encodingClassName);
            encodings.put(encodingClassName, new EncodingInstance(encoding, encoding));
        }

        return encodings;
    }

    private static Map getJcodingsTables() {
        Map jcodingsTables = new HashMap<>();
        Set jcodingsTableNames = new HashSet<>();

        RootedFileVisitor visitor = new SimpleRootedFileVisitor() {
            // match files that start with "tables/" and end with ".bin"
            Pattern filePattern = Pattern.compile("^tables/(.*)\\.bin$");

            @Override
            public FileVisitResult visitFile(Path file, BasicFileAttributes attrs) throws IOException {
                String fileName = getRoot().relativize(file).toString();
                Matcher m = filePattern.matcher(fileName);
                if (m.matches()) {
                    jcodingsTableNames.add(m.group(1));
                }
                return FileVisitResult.CONTINUE;
            }
        };

        RootedFileVisitor.visitEachFileOnClassPath(visitor);

        for (String name : jcodingsTableNames) {
            String entry = "/tables/" + name + ".bin";
            try (InputStream is = ArrayReader.class.getResourceAsStream(entry)) {
                if (is != null) {
                    byte[] buf = new byte[is.available()];
                    new DataInputStream(is).readFully(buf);
                    jcodingsTables.put(name, buf);
                } else {
                    throw new Error("Unable to load Jcodings table " + name);
                }
            } catch (IOException e) {
            }
        }
        return jcodingsTables;
    }

    static class EncodingInstance {
        Encoding instance;
        Encoding dummy;

        EncodingInstance(Encoding instance, Encoding dummy) {
            this.instance = instance;
            this.dummy = dummy;
        }

        Encoding get(boolean useDummy) {
            if (useDummy && this.dummy != null) {
                return this.dummy;
            } else {
                return this.instance;
            }
        }
    }

}