Class TypeSystemImpl

src/main/java/pascal/taie/language/type/TypeSystemImpl.java:36–213

// TODO: optimize maps (classTypes and arrayTypes)
public class TypeSystemImpl implements TypeSystem {

    private final ClassHierarchy hierarchy;

    private final Map<JClassLoader, Map<String, ClassType>> classTypes = newSmallMap();

    /**
     * This map may be concurrently written during IR construction,
     * thus we use concurrent map to ensure its thread-safety.
     */
    private final ConcurrentMap<Integer, ConcurrentMap<Type, ArrayType>> arrayTypes
            = newConcurrentMap(8);

    private final ClassType OBJECT;
    private final ClassType SERIALIZABLE;
    private final ClassType CLONEABLE;

    // Boxed types
    private final ClassType BOOLEAN;
    private final ClassType BYTE;
    private final ClassType SHORT;
    private final ClassType CHARACTER;
    private final ClassType INTEGER;
    private final ClassType LONG;
    private final ClassType FLOAT;
    private final ClassType DOUBLE;

    public TypeSystemImpl(ClassHierarchy hierarchy) {
        this.hierarchy = hierarchy;
        // Initialize special types
        JClassLoader loader = hierarchy.getBootstrapClassLoader();
        OBJECT = getClassType(loader, ClassNames.OBJECT);
        SERIALIZABLE = getClassType(loader, ClassNames.SERIALIZABLE);
        CLONEABLE = getClassType(loader, ClassNames.CLONEABLE);
        BOOLEAN = getClassType(loader, ClassNames.BOOLEAN);
        BYTE = getClassType(loader, ClassNames.BYTE);
        SHORT = getClassType(loader, ClassNames.SHORT);
        CHARACTER = getClassType(loader, ClassNames.CHARACTER);
        INTEGER = getClassType(loader, ClassNames.INTEGER);
        LONG = getClassType(loader, ClassNames.LONG);
        FLOAT = getClassType(loader, ClassNames.FLOAT);
        DOUBLE = getClassType(loader, ClassNames.DOUBLE);
    }

    @Override
    public Type getType(JClassLoader loader, String typeName) {
        try {
            if (typeName.endsWith("[]")) {
                int dim = 0;
                int i = typeName.length() - 1;
                while (i > 0) {
                    if (typeName.charAt(i - 1) == '[' && typeName.charAt(i) == ']') {
                        ++dim;
                        i -= 2;
                    } else {
                        break;
                    }
                }