net.minecraft.server.AttributeMapBase Maven / Gradle / Ivy

Go to download
package net.minecraft.server;

import com.google.common.collect.HashMultimap;
import com.google.common.collect.Maps;
import com.google.common.collect.Multimap;

import java.util.Collection;
import java.util.Map;
import java.util.Map.Entry;

public abstract class AttributeMapBase {

  protected final Map a = Maps.newHashMap();
  protected final Map b = new InsensitiveStringMap<>();
  protected final Multimap c = HashMultimap.create();

  public AttributeMapBase() {
  }

  public AttributeInstance a(IAttribute iattribute) {
    return this.a.get(iattribute);
  }

  public AttributeInstance a(String s) {
    return this.b.get(s);
  }

  public AttributeInstance b(IAttribute iattribute) {
    if (this.b.containsKey(iattribute.getName())) {
      throw new IllegalArgumentException("Attribute is already registered!");
    } else {
      AttributeInstance attributeinstance = this.c(iattribute);

      this.b.put(iattribute.getName(), attributeinstance);
      this.a.put(iattribute, attributeinstance);

      for (IAttribute iattribute1 = iattribute.d(); iattribute1 != null; iattribute1 = iattribute1.d()) {
        this.c.put(iattribute1, iattribute);
      }

      return attributeinstance;
    }
  }

  protected abstract AttributeInstance c(IAttribute iattribute);

  public Collection a() {
    return this.b.values();
  }

  public void a(AttributeInstance attributeinstance) {
  }

  public void a(Multimap multimap) {

    for (Entry stringAttributeModifierEntry : multimap.entries()) {
      AttributeInstance attributeinstance = this.a(stringAttributeModifierEntry.getKey());
      if (attributeinstance != null) {
        attributeinstance.c(stringAttributeModifierEntry.getValue());
      }
    }
  }

  public void b(Multimap multimap) {

    for (Entry stringAttributeModifierEntry : multimap.entries()) {
      AttributeInstance attributeinstance = this.a(stringAttributeModifierEntry.getKey());

      if (attributeinstance != null) {
        attributeinstance.c(stringAttributeModifierEntry.getValue());
        attributeinstance.b(stringAttributeModifierEntry.getValue());
      }
    }

  }
}