From e33b4003a5c423894e7aef575faff359dd1d33b1 Mon Sep 17 00:00:00 2001
From: jeff
Date: Wed, 11 Mar 2015 11:03:16 -0400
Subject: generalized field matching added method matching

---
 src/cuchaz/enigma/convert/MemberMatches.java | 145 +++++++++++++++++++++++++++
 1 file changed, 145 insertions(+)
 create mode 100644 src/cuchaz/enigma/convert/MemberMatches.java

(limited to 'src/cuchaz/enigma/convert/MemberMatches.java')

diff --git a/src/cuchaz/enigma/convert/MemberMatches.java b/src/cuchaz/enigma/convert/MemberMatches.java
new file mode 100644
index 0000000..1078ab7
--- /dev/null
+++ b/src/cuchaz/enigma/convert/MemberMatches.java
@@ -0,0 +1,145 @@
+package cuchaz.enigma.convert;
+
+import java.util.Collection;
+import java.util.Set;
+
+import com.google.common.collect.BiMap;
+import com.google.common.collect.HashBiMap;
+import com.google.common.collect.HashMultimap;
+import com.google.common.collect.Multimap;
+import com.google.common.collect.Sets;
+
+import cuchaz.enigma.mapping.ClassEntry;
+import cuchaz.enigma.mapping.Entry;
+
+
+public class MemberMatches<T extends Entry> {
+
+	private BiMap<T,T> m_matches;
+	private Multimap<ClassEntry,T> m_matchedSourceEntries;
+	private Multimap<ClassEntry,T> m_unmatchedSourceEntries;
+	private Multimap<ClassEntry,T> m_unmatchedDestEntries;
+	private Multimap<ClassEntry,T> m_unmatchableSourceEntries;
+	
+	public MemberMatches() {
+		m_matches = HashBiMap.create();
+		m_matchedSourceEntries = HashMultimap.create();
+		m_unmatchedSourceEntries = HashMultimap.create();
+		m_unmatchedDestEntries = HashMultimap.create();
+		m_unmatchableSourceEntries = HashMultimap.create();
+	}
+	
+	public void addMatch(T srcEntry, T destEntry) {
+		boolean wasAdded = m_matches.put(srcEntry, destEntry) == null;
+		assert (wasAdded);
+		wasAdded = m_matchedSourceEntries.put(srcEntry.getClassEntry(), srcEntry);
+		assert (wasAdded);
+	}
+	
+	public void addUnmatchedSourceEntry(T sourceEntry) {
+		boolean wasAdded = m_unmatchedSourceEntries.put(sourceEntry.getClassEntry(), sourceEntry);
+		assert (wasAdded);
+	}
+	
+	public void addUnmatchedSourceEntries(Iterable<T> sourceEntries) {
+		for (T sourceEntry : sourceEntries) {
+			addUnmatchedSourceEntry(sourceEntry);
+		}
+	}
+	
+	public void addUnmatchedDestEntry(T destEntry) {
+		boolean wasAdded = m_unmatchedDestEntries.put(destEntry.getClassEntry(), destEntry);
+		assert (wasAdded);
+	}
+	
+	public void addUnmatchedDestEntries(Iterable<T> destEntriesntries) {
+		for (T entry : destEntriesntries) {
+			addUnmatchedDestEntry(entry);
+		}
+	}
+	
+	public void addUnmatchableSourceEntry(T sourceEntry) {
+		boolean wasAdded = m_unmatchableSourceEntries.put(sourceEntry.getClassEntry(), sourceEntry);
+		assert (wasAdded);
+	}
+	
+	public Set<ClassEntry> getSourceClassesWithUnmatchedEntries() {
+		return m_unmatchedSourceEntries.keySet();
+	}
+	
+	public Collection<ClassEntry> getSourceClassesWithoutUnmatchedEntries() {
+		Set<ClassEntry> out = Sets.newHashSet();
+		out.addAll(m_matchedSourceEntries.keySet());
+		out.removeAll(m_unmatchedSourceEntries.keySet());
+		return out;
+	}
+
+	public Collection<T> getUnmatchedSourceEntries() {
+		return m_unmatchedSourceEntries.values();
+	}
+
+	public Collection<T> getUnmatchedSourceEntries(ClassEntry sourceClass) {
+		return m_unmatchedSourceEntries.get(sourceClass);
+	}
+
+	public Collection<T> getUnmatchedDestEntries() {
+		return m_unmatchedDestEntries.values();
+	}
+
+	public Collection<T> getUnmatchedDestEntries(ClassEntry destClass) {
+		return m_unmatchedDestEntries.get(destClass);
+	}
+	
+	public Collection<T> getUnmatchableSourceEntries() {
+		return m_unmatchableSourceEntries.values();
+	}
+
+	public boolean hasSource(T sourceEntry) {
+		return m_matches.containsKey(sourceEntry) || m_unmatchedSourceEntries.containsValue(sourceEntry);
+	}
+	
+	public boolean hasDest(T destEntry) {
+		return m_matches.containsValue(destEntry) || m_unmatchedDestEntries.containsValue(destEntry);
+	}
+
+	public BiMap<T,T> matches() {
+		return m_matches;
+	}
+	
+	public boolean isMatchedSourceEntry(T sourceEntry) {
+		return m_matches.containsKey(sourceEntry);
+	}
+
+	public boolean isMatchedDestEntry(T destEntry) {
+		return m_matches.containsValue(destEntry);
+	}
+
+	public void makeMatch(T sourceEntry, T destEntry) {
+		boolean wasRemoved = m_unmatchedSourceEntries.remove(sourceEntry.getClassEntry(), sourceEntry);
+		assert (wasRemoved);
+		wasRemoved = m_unmatchedDestEntries.remove(destEntry.getClassEntry(), destEntry);
+		assert (wasRemoved);
+		addMatch(sourceEntry, destEntry);
+	}
+
+	public boolean isMatched(T sourceEntry, T destEntry) {
+		T match = m_matches.get(sourceEntry);
+		return match != null && match.equals(destEntry);
+	}
+
+	public void unmakeMatch(T sourceEntry, T destEntry) {
+		boolean wasRemoved = m_matches.remove(sourceEntry) != null;
+		assert (wasRemoved);
+		wasRemoved = m_matchedSourceEntries.remove(sourceEntry.getClassEntry(), sourceEntry);
+		assert (wasRemoved);
+		addUnmatchedSourceEntry(sourceEntry);
+		addUnmatchedDestEntry(destEntry);
+	}
+	
+	public void makeSourceUnmatchable(T sourceEntry) {
+		assert(!isMatchedSourceEntry(sourceEntry));
+		boolean wasRemoved = m_unmatchedSourceEntries.remove(sourceEntry.getClassEntry(), sourceEntry);
+		assert (wasRemoved);
+		addUnmatchableSourceEntry(sourceEntry);
+	}
+}
-- 
cgit v1.2.3