package org.ggp.base.util.prover.aima.cache;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

import org.ggp.base.util.gdl.grammar.GdlSentence;
import org.ggp.base.util.prover.aima.substituter.Substituter;
import org.ggp.base.util.prover.aima.substitution.Substitution;
import org.ggp.base.util.prover.aima.unifier.Unifier;


public final class ProverCache
{

    private final Map<GdlSentence, Set<GdlSentence>> contents;

    private ProverCache(Map<GdlSentence, Set<GdlSentence>> mapForContents) {
        this.contents = mapForContents;
    }

    public static ProverCache createSingleThreadedCache() {
        return new ProverCache(new HashMap<GdlSentence, Set<GdlSentence>>());
    }

    public static ProverCache createMultiThreadedCache() {
        return new ProverCache(new ConcurrentHashMap<GdlSentence, Set<GdlSentence>>());
    }

    /**
     * NOTE: The given sentence must have been renamed with a VariableRenamer.
     */
    public boolean contains(GdlSentence renamedSentence)
    {
        return contents.containsKey(renamedSentence);
    }

    public List<Substitution> get(GdlSentence sentence, GdlSentence varRenamedSentence)
    {
        Set<GdlSentence> cacheContents = contents.get(varRenamedSentence);
        if (cacheContents == null) {
            return null;
        }
        Set<Substitution> results = new HashSet<Substitution>();
        for (GdlSentence answer : cacheContents)
        {
            results.add(Unifier.unify(sentence, answer));
        }

        return new ArrayList<Substitution>(results);
    }

    public void put(GdlSentence sentence, GdlSentence renamedSentence,
            Set<Substitution> answers)
    {
        Set<GdlSentence> results = new HashSet<GdlSentence>();
        for (Substitution answer : answers)
        {
            results.add(Substituter.substitute(sentence, answer));
        }

        contents.put(renamedSentence, results);
    }

}
