summaryrefslogtreecommitdiff
path: root/src/main/java/cuchaz/enigma/mapping/Signature.java
blob: 78130d6bb9d0c79f793a004ac8d454482a15f2dc (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
/*******************************************************************************
 * Copyright (c) 2015 Jeff Martin.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the GNU Lesser General Public
 * License v3.0 which accompanies this distribution, and is available at
 * http://www.gnu.org/licenses/lgpl.html
 * <p>
 * Contributors:
 * Jeff Martin - initial API and implementation
 ******************************************************************************/

package cuchaz.enigma.mapping;

import com.google.common.collect.Lists;
import cuchaz.enigma.utils.Utils;

import java.util.List;

public class Signature {

	private List<Type> argumentTypes;
	private Type returnType;

	public Signature(String signature) {
		try {
			this.argumentTypes = Lists.newArrayList();
			int i = 0;
			while (i < signature.length()) {
				char c = signature.charAt(i);
				if (c == '(') {
					assert (this.argumentTypes.isEmpty());
					assert (this.returnType == null);
					i++;
				} else if (c == ')') {
					i++;
					break;
				} else {
					String type = Type.parseFirst(signature.substring(i));
					this.argumentTypes.add(new Type(type));
					i += type.length();
				}
			}
			this.returnType = new Type(Type.parseFirst(signature.substring(i)));
		} catch (Exception ex) {
			throw new IllegalArgumentException("Unable to parse signature: " + signature, ex);
		}
	}

	public Signature(Signature other, ClassNameReplacer replacer) {
		this.argumentTypes = Lists.newArrayList(other.argumentTypes);
		for (int i = 0; i < this.argumentTypes.size(); i++) {
			this.argumentTypes.set(i, new Type(this.argumentTypes.get(i), replacer));
		}
		this.returnType = new Type(other.returnType, replacer);
	}

	public List<Type> getArgumentTypes() {
		return this.argumentTypes;
	}

	public Type getReturnType() {
		return this.returnType;
	}

	@Override
	public String toString() {
		StringBuilder buf = new StringBuilder();
		buf.append("(");
		for (Type type : this.argumentTypes) {
			buf.append(type);
		}
		buf.append(")");
		buf.append(this.returnType);
		return buf.toString();
	}

	public Iterable<Type> types() {
		List<Type> types = Lists.newArrayList();
		types.addAll(this.argumentTypes);
		types.add(this.returnType);
		return types;
	}

	@Override
	public boolean equals(Object other) {
		return other instanceof Signature && equals((Signature) other);
	}

	public boolean equals(Signature other) {
		return this.argumentTypes.equals(other.argumentTypes) && this.returnType.equals(other.returnType);
	}

	@Override
	public int hashCode() {
		return Utils.combineHashesOrdered(this.argumentTypes.hashCode(), this.returnType.hashCode());
	}

	public boolean hasClass(ClassEntry classEntry) {
		for (Type type : types()) {
			if (type.hasClass() && type.getClassEntry().equals(classEntry)) {
				return true;
			}
		}
		return false;
	}
}