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

package cuchaz.enigma.translation.mapping;

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

import java.util.stream.Stream;

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

	private final EntryTree<T> baseMappings;

	private final EntryTree<Object> changes;

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

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

	public static <T> MappingDelta<T> added(EntryTree<T> mappings) {
		EntryTree<Object> changes = new HashEntryTree<>();
		mappings.getAllEntries().forEach(entry -> changes.insert(entry, PLACEHOLDER));

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

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

	public EntryTree<?> getChanges() {
		return changes;
	}

	public Stream<Entry<?>> getChangedRoots() {
		return changes.getRootNodes().map(EntryTreeNode::getEntry);
	}

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