1 files changed, 88 insertions, 0 deletions
diff --git a/src/main/java/tlIItools/AffixGroup.java b/src/main/java/tlIItools/AffixGroup.java
index 7ae57453..4069c632 100644
--- a/src/main/java/tlIItools/AffixGroup.java
+++ b/src/main/java/tlIItools/AffixGroup.java
@@ -1,5 +1,9 @@
 package tlIItools;
 
+import java.util.*;
+
+import tlIItools.Affix.*;
+
 /**
  * Represents a 'group' of related affixes.
  *
@@ -8,4 +12,88 @@ package tlIItools;
  *
  * @author Ben Culkin */
 public class AffixGroup {
+    public List<EffectGroup> effects;
+    /** The types of equipment this can spawn on. */
+    public List<String> enchantSources;
+    /** The types of equipment this can spawn on. */
+    public List<String> equipTypes;
+    /** The types of equipment this can spawn on. */
+    public List<String> nonequipTypes;
+    /** The types of equipment this can spawn on. */
+    public List<String> socketableTypes;
+    
+    /** The type of thing this affix applies to. */
+    public AffixType type = AffixType.ITEM;
+    
+    /** Create a new affix group. */
+    public AffixGroup() {
+        effects = new ArrayList<>();
+        enchantSources = new ArrayList<>();
+        equipTypes = new ArrayList<>();
+        nonequipTypes = new ArrayList<>();
+        socketableTypes = new ArrayList<>();
+    }
+
+    /** Determine whether this affix group contains a particular affix.
+     * 
+     * By 'affix group', what we mean is that it is essentially the same affix, with
+     * generally just differing levels/values.
+     * 
+     * For instance, an affix that granted +2 strength, and one that granted +4
+     * strength would be considered to be in the same affix group (assuming that
+     * both of those strength bonuses had the same effect name).
+     * 
+     * @param afx The affix to check if it is in the affix group with.
+     * 
+     * @return Whether or the specified affix is in this affix group. */
+    public boolean contains(Affix afx) {
+        return afx.toAffixGroup().equals(this);
+    }
+    
+    /** Get the 'header' for displaying this group
+     * 
+     * @return The header for this group. */
+    public String groupHeader() {
+       StringBuilder sb = new StringBuilder();
+       
+       sb.append("ID: " + hashCode());
+       
+       return sb.toString();
+    }
+    
+    @Override
+    public String toString() {
+        StringBuilder sb = new StringBuilder();
+        
+        for (EffectGroup group : effects) sb.append(group);
+        for (String enchantSource : enchantSources) sb.append(enchantSource);
+        for (String equipType : equipTypes) sb.append(equipType);
+        for (String nonEquipType : nonequipTypes) sb.append(nonEquipType);
+        for (String socketableType : socketableTypes) sb.append(socketableType);
+        
+        return sb.toString();
+    }
+    
+    @Override
+    public int hashCode() {
+        return Objects.hash(effects, enchantSources, equipTypes, nonequipTypes,
+                socketableTypes, type);
+    }
+
+    @Override
+    public boolean equals(Object obj) {
+        if (this == obj)
+            return true;
+        if (obj == null)
+            return false;
+        if (getClass() != obj.getClass())
+            return false;
+        AffixGroup other = (AffixGroup) obj;
+        return Objects.equals(effects, other.effects)
+                && Objects.equals(enchantSources, other.enchantSources)
+                && Objects.equals(equipTypes, other.equipTypes)
+                && Objects.equals(nonequipTypes, other.nonequipTypes)
+                && Objects.equals(socketableTypes, other.socketableTypes)
+                && type == other.type;
+    }
 }