org.ggp.base.util.assignments.DependentAssignmentStrategy Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of alloy-ggp-base Show documentation
A modified version of the GGP-Base library for Alloy.
There is a newer version: 0.0.15
package org.ggp.base.util.assignments;

import java.util.Collection;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;

import javax.annotation.Nullable;
import javax.annotation.concurrent.NotThreadSafe;

import org.ggp.base.util.Immutables;
import org.ggp.base.util.gdl.grammar.GdlConstant;

import com.google.common.collect.ImmutableList;
import com.google.common.collect.Maps;

@NotThreadSafe
public class DependentAssignmentStrategy implements AssignmentStrategy {
    private final ImmutableList dependentIndices;
    private final ImmutableList definedIndices;

    //TODO: Something better and trie-based?
    private final Node contents;

    public DependentAssignmentStrategy(ImmutableList dependentIndices,
            ImmutableList definedIndices, Node contents) {
        this.dependentIndices = dependentIndices;
        this.definedIndices = definedIndices;
        this.contents = contents;
    }

    public static AssignmentStrategy create(List dependentIndices,
            List definedIndices,
            Map, ? extends Collection>> contents) {
        if (dependentIndices.isEmpty()) {
            return MultipleAssignmentStrategy.create(definedIndices,
                    contents.get(ImmutableList.of()));
        }
        Node contentsNode = toNode(contents);
        return new DependentAssignmentStrategy(
                ImmutableList.copyOf(dependentIndices),
                ImmutableList.copyOf(definedIndices),
                contentsNode);
    }

    private static Node toNode(
            Map, ? extends Collection>> contents) {
        Node rootNode = new BranchNode();
        for (Entry, ? extends Collection>> entry : contents.entrySet()) {
            List inputs = entry.getKey();
            Collection> partialAssignments = entry.getValue();
            Node curNode = rootNode;
            for (int i = 0; i < inputs.size(); i++) {
                GdlConstant input = inputs.get(i);
                if (i == inputs.size() - 1) {
                    curNode.createLeafChild(input, partialAssignments);
                } else {
                    curNode = curNode.getBranchChildConstructively(input);
                }
            }
        }
        return rootNode;
    }

    @Override
    public List getDependentIndices() {
        return dependentIndices;
    }

    @Override
    public List getDefinedIndices() {
        return definedIndices;
    }

    @Override
    public List> getPartialAssignments(
            List inputs) {
        Node curNode = contents;
        for (GdlConstant input : inputs) {
            curNode = curNode.getChildForInput(input);
            if (curNode == null) {
                return ImmutableList.of();
            }
        }
        return curNode.getPartialAssignments();
    }

    @Override
    public int getRejectedIndex(List inputs) {
        Node curNode = contents;
        int inputIndex = 0;
        for (GdlConstant input : inputs) {
            curNode = curNode.getChildForInput(input);
            if (curNode == null) {
                return dependentIndices.get(inputIndex);
            }
            inputIndex++;
        }
        return NO_INDEX_REJECTED;
    }

    private static interface Node {
        @Nullable Node getChildForInput(GdlConstant input);
        Node getBranchChildConstructively(GdlConstant input);
        void createLeafChild(GdlConstant input, Collection> partialAssignments);
        List> getPartialAssignments();
    }
    private static class BranchNode implements Node {
        private final Map children = Maps.newHashMap();
        @Override
        public Node getChildForInput(GdlConstant input) {
            return children.get(input);
        }
        @Override
        public List> getPartialAssignments() {
            throw new UnsupportedOperationException();
        }
        @Override
        public Node getBranchChildConstructively(GdlConstant input) {
            Node result = children.get(input);
            if (result != null) {
                return result;
            }
            result = new BranchNode();
            children.put(input, result);
            return result;
        }
        @Override
        public void createLeafChild(GdlConstant input,
                Collection> partialAssignments) {
            children.put(input, new LeafNode(partialAssignments.stream()
                    .map(ImmutableList::copyOf)
                    .collect(Immutables.collectList())));
        }
    }
    private static class LeafNode implements Node {
        private final ImmutableList> partialAssignments;
        public LeafNode(
                ImmutableList> partialAssignments) {
            this.partialAssignments = partialAssignments;
        }
        @Override
        public Node getChildForInput(GdlConstant input) {
            throw new UnsupportedOperationException();
        }
        @Override
        public List> getPartialAssignments() {
            return partialAssignments;
        }
        @Override
        public Node getBranchChildConstructively(GdlConstant input) {
            throw new UnsupportedOperationException();
        }
        @Override
        public void createLeafChild(GdlConstant input,
                Collection> partialAssignments) {
            throw new UnsupportedOperationException();
        }
    }

    @Override
    public String toString() {
        return "DependentAssignmentStrategy [From " + dependentIndices + " to " + definedIndices + "]";
    }
}