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 implements Translatable {
	public static final Object PLACEHOLDER = new Object();

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

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

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

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

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

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

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

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

	private EntryTree<Object> translate(Translator translator, EntryTree<Object> tree) {
		EntryTree<Object> translatedTree = new HashEntryTree<>();
		for (Entry<?> entry : tree.getAllEntries()) {
			translatedTree.insert(translator.translate(entry), PLACEHOLDER);
		}
		return translatedTree;
	}
}