summaryrefslogtreecommitdiff
path: root/src/main/java/tlIItools/AffixSet.java
blob: 91c0dcd9b98aa78ab5a30cfe6006011ac4bd19b2 (plain)
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
package tlIItools;

import java.util.*;

/** Container of a set of affixes.
 *
 * @author Ben Culkin */
public class AffixSet {
	private static class AffixComparator implements Comparator<Affix> {
		@Override
		public int compare(Affix a1, Affix a2) {
			if (a1.minLevel == a2.minLevel)  return a1.maxLevel - a2.maxLevel;
			else                             return a1.minLevel - a2.minLevel;
		}
	}

	/** All of the affix groups contained in this set.
	 *
	 * An affix group is a set of affixes that generally have the same or
	 * similar effects, but have different intensities or spawn levels. */
	public Map<AffixGroup, Set<Affix>> affixGroups;

	/** All of the ungrouped affixes contained in this set. */
	public Set<Affix> ungroupedAffixes;

	/** Create a new blank affix set. */
	public AffixSet() {
		affixGroups = new TreeMap<>();

		ungroupedAffixes = new TreeSet<>(new AffixComparator());
	}
	
	/** Add an affix to this set.
	 * 
	 * @param afx The affix to add. */
	public void addAffixByContents(Affix afx) {
		AffixGroup group = afx.toAffixGroup();
        String afxGroup = group.toString();
		
		if (afxGroup.equals("")) {
			ungroupedAffixes.add(afx);
		} else {
			affixGroups.compute(group, (key, val) -> {
				if (val == null) {
					Set<Affix> afxSet = new HashSet<>();
					afxSet.add(afx);
					return afxSet;
				} else {
					val.add(afx);
					return val;
				}
			});
		}
	}
}