com.llamalad7.mixinextras.expression.impl.pool.IdentifierPool Maven / Gradle / Ivy

Go to download
package com.llamalad7.mixinextras.expression.impl.pool;

import com.llamalad7.mixinextras.expression.impl.flow.FlowValue;
import org.objectweb.asm.Type;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class IdentifierPool {
    private final Map> members = new HashMap<>();
    private final Map> types = new HashMap<>();

    public IdentifierPool() {
        addType("byte", new ExactTypeDef(Type.BYTE_TYPE));
        addType("char", new ExactTypeDef(Type.CHAR_TYPE));
        addType("double", new ExactTypeDef(Type.DOUBLE_TYPE));
        addType("float", new ExactTypeDef(Type.FLOAT_TYPE));
        addType("int", new ExactTypeDef(Type.INT_TYPE));
        addType("long", new ExactTypeDef(Type.LONG_TYPE));
        addType("short", new ExactTypeDef(Type.SHORT_TYPE));
        addMember("length", new ArrayLengthDef());
    }

    public boolean matchesMember(String id, FlowValue node) {
        List matching = members.get(id);
        if (matching == null) {
            throw new IllegalStateException("Use of undeclared identifier '" + id + '\'');
        }
        return matching.stream().anyMatch(it -> it.matches(node));
    }

    public boolean matchesType(String id, Type type) {
        List matching = types.get(id);
        if (matching == null) {
            throw new IllegalStateException("Use of undeclared identifier '" + id + '\'');
        }
        return matching.stream().anyMatch(it -> it.matches(type));
    }

    public void addMember(String id, MemberDefinition entry) {
        members.computeIfAbsent(id, k -> new ArrayList<>()).add(entry);
    }

    public void addType(String id, TypeDefinition entry) {
        types.computeIfAbsent(id, k -> new ArrayList<>()).add(entry);
    }

    public boolean memberExists(String id) {
        return members.containsKey(id);
    }

    public boolean typeExists(String id) {
        return types.containsKey(id);
    }
}