src/main/java/cuchaz/enigma/translation/mapping/MappingDelta.java


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

package cuchaz.enigma.translation.mapping;

import cuchaz.enigma.translation.Translatable;
import cuchaz.enigma.translation.Translator;
import cuchaz.enigma.translation.mapping.tree.HashEntryTree;
import cuchaz.enigma.translation.mapping.tree.EntryTree;
import cuchaz.enigma.translation.representation.entry.Entry;

public class MappingDelta<T> implements Translatable {
	public static final Object PLACEHOLDER = new Object();

	private final EntryTree<T> baseMappings;

	private final EntryTree<Object> additions;
	private final EntryTree<Object> deletions;

	public MappingDelta(EntryTree<T> baseMappings, EntryTree<Object> additions, EntryTree<Object> deletions) {
		this.baseMappings = baseMappings;
		this.additions = additions;
		this.deletions = deletions;
	}

	public MappingDelta(EntryTree<T> baseMappings) {
		this(baseMappings, new HashEntryTree<>(), new HashEntryTree<>());
	}

	public static <T> MappingDelta<T> added(EntryTree<T> mappings) {
		EntryTree<Object> additions = new HashEntryTree<>();
		for (Entry<?> entry : mappings.getAllEntries()) {
			additions.insert(entry, PLACEHOLDER);
		}

		return new MappingDelta<>(new HashEntryTree<>(), additions, new HashEntryTree<>());
	}

	public EntryTree<T> getBaseMappings() {
		return baseMappings;
	}

	public EntryTree<?> getAdditions() {
		return additions;
	}

	public EntryTree<?> getDeletions() {
		return deletions;
	}

	@Override
	public MappingDelta<T> translate(Translator translator, EntryResolver resolver, EntryMap<EntryMapping> mappings) {
		return new MappingDelta<>(
				translator.translate(baseMappings),
				translator.translate(additions),
				translator.translate(deletions)
		);
	}
}