From 73313b0f1c1660986afc1449960889cac242eee0 Mon Sep 17 00:00:00 2001
From: jeff
Date: Thu, 5 Feb 2015 23:51:40 -0500
Subject: add new type/signature system

---
 src/cuchaz/enigma/mapping/BehaviorSignature.java | 96 ++++++++++++++++++++++++
 1 file changed, 96 insertions(+)
 create mode 100644 src/cuchaz/enigma/mapping/BehaviorSignature.java

(limited to 'src/cuchaz/enigma/mapping/BehaviorSignature.java')

diff --git a/src/cuchaz/enigma/mapping/BehaviorSignature.java b/src/cuchaz/enigma/mapping/BehaviorSignature.java
new file mode 100644
index 0000000..a6371f8
--- /dev/null
+++ b/src/cuchaz/enigma/mapping/BehaviorSignature.java
@@ -0,0 +1,96 @@
+package cuchaz.enigma.mapping;
+
+import java.util.List;
+
+import com.beust.jcommander.internal.Lists;
+
+public class BehaviorSignature {
+	
+	public static interface ClassReplacer {
+		ClassEntry replace(ClassEntry entry);
+	}
+	
+	private List<Type> m_argumentTypes;
+	private Type m_returnType;
+	
+	public BehaviorSignature(String signature) {
+		m_argumentTypes = Lists.newArrayList();
+		int i=0;
+		while (i<signature.length()) {
+			char c = signature.charAt(i);
+			if (c == '(') {
+				assert(m_argumentTypes.isEmpty());
+				assert(m_returnType == null);
+				i++;
+			} else if (c == ')') {
+				i++;
+				break;
+			} else {
+				String type = Type.parseFirst(signature.substring(i));
+				m_argumentTypes.add(new Type(type));
+				i += type.length();
+			}
+		}
+		m_returnType = new Type(Type.parseFirst(signature.substring(i)));
+	}
+	
+	public BehaviorSignature(BehaviorSignature other, ClassReplacer replacer) {
+		m_argumentTypes = Lists.newArrayList(other.m_argumentTypes);
+		for (int i=0; i<m_argumentTypes.size(); i++) {
+			Type type = m_argumentTypes.get(i);
+			if (type.isClass()) {
+				ClassEntry newClassEntry = replacer.replace(type.getClassEntry());
+				if (newClassEntry != null) {
+					m_argumentTypes.set(i, new Type(newClassEntry));
+				}
+			}
+		}
+		m_returnType = other.m_returnType;
+		if (other.m_returnType.isClass()) {
+			ClassEntry newClassEntry = replacer.replace(m_returnType.getClassEntry());
+			if (newClassEntry != null) {
+				m_returnType = new Type(newClassEntry);
+			}
+		}
+	}
+	
+	public List<Type> getArgumentTypes() {
+		return m_argumentTypes;
+	}
+	
+	public Type getReturnType() {
+		return m_returnType;
+	}
+	
+	@Override
+	public String toString() {
+		StringBuilder buf = new StringBuilder();
+		buf.append("(");
+		for (int i=0; i<m_argumentTypes.size(); i++) {
+			if (i > 0) {
+				buf.append(",");
+			}
+			buf.append(m_argumentTypes.get(i).toString());
+		}
+		buf.append(")");
+		buf.append(m_returnType.toString());
+		return buf.toString();
+	}
+	
+	public Iterable<Type> types() {
+		List<Type> types = Lists.newArrayList();
+		types.addAll(m_argumentTypes);
+		types.add(m_returnType);
+		return types;
+	}
+	
+	public Iterable<ClassEntry> classes() {
+		List<ClassEntry> out = Lists.newArrayList();
+		for (Type type : types()) {
+			if (type.isClass()) {
+				out.add(type.getClassEntry());
+			}
+		}
+		return out;
+	}
+}
-- 
cgit v1.2.3