package net.minecraft.entity.ai.attributes; import com.google.common.collect.Lists; import com.google.common.collect.Maps; import com.google.common.collect.Sets; import java.util.ArrayList; import java.util.Collection; import java.util.HashSet; import java.util.Iterator; import java.util.Map; import java.util.Set; import java.util.UUID; public class ModifiableAttributeInstance implements IAttributeInstance { /** The BaseAttributeMap this attributeInstance can be found in */ private final BaseAttributeMap attributeMap; /** The Attribute this is an instance of */ private final IAttribute genericAttribute; private final Map mapByOperation = Maps.newHashMap(); private final Map mapByName = Maps.newHashMap(); private final Map mapByUUID = Maps.newHashMap(); private double baseValue; private boolean needsUpdate = true; private double cachedValue; private static final String __OBFID = "CL_00001567"; public ModifiableAttributeInstance(BaseAttributeMap p_i1608_1_, IAttribute p_i1608_2_) { this.attributeMap = p_i1608_1_; this.genericAttribute = p_i1608_2_; this.baseValue = p_i1608_2_.getDefaultValue(); for (int var3 = 0; var3 < 3; ++var3) { this.mapByOperation.put(Integer.valueOf(var3), Sets.newHashSet()); } } /** * Get the Attribute this is an instance of */ public IAttribute getAttribute() { return this.genericAttribute; } public double getBaseValue() { return this.baseValue; } public void setBaseValue(double p_111128_1_) { if (p_111128_1_ != this.getBaseValue()) { this.baseValue = p_111128_1_; this.flagForUpdate(); } } public Collection getModifiersByOperation(int p_111130_1_) { return (Collection)this.mapByOperation.get(Integer.valueOf(p_111130_1_)); } public Collection func_111122_c() { HashSet var1 = Sets.newHashSet(); for (int var2 = 0; var2 < 3; ++var2) { var1.addAll(this.getModifiersByOperation(var2)); } return var1; } /** * Returns attribute modifier, if any, by the given UUID */ public AttributeModifier getModifier(UUID p_111127_1_) { return (AttributeModifier)this.mapByUUID.get(p_111127_1_); } public boolean func_180374_a(AttributeModifier p_180374_1_) { return this.mapByUUID.get(p_180374_1_.getID()) != null; } public void applyModifier(AttributeModifier p_111121_1_) { if (this.getModifier(p_111121_1_.getID()) != null) { throw new IllegalArgumentException("Modifier is already applied on this attribute!"); } else { Object var2 = (Set)this.mapByName.get(p_111121_1_.getName()); if (var2 == null) { var2 = Sets.newHashSet(); this.mapByName.put(p_111121_1_.getName(), var2); } ((Set)this.mapByOperation.get(Integer.valueOf(p_111121_1_.getOperation()))).add(p_111121_1_); ((Set)var2).add(p_111121_1_); this.mapByUUID.put(p_111121_1_.getID(), p_111121_1_); this.flagForUpdate(); } } protected void flagForUpdate() { this.needsUpdate = true; this.attributeMap.func_180794_a(this); } public void removeModifier(AttributeModifier p_111124_1_) { for (int var2 = 0; var2 < 3; ++var2) { Set var3 = (Set)this.mapByOperation.get(Integer.valueOf(var2)); var3.remove(p_111124_1_); } Set var4 = (Set)this.mapByName.get(p_111124_1_.getName()); if (var4 != null) { var4.remove(p_111124_1_); if (var4.isEmpty()) { this.mapByName.remove(p_111124_1_.getName()); } } this.mapByUUID.remove(p_111124_1_.getID()); this.flagForUpdate(); } public void removeAllModifiers() { Collection var1 = this.func_111122_c(); if (var1 != null) { ArrayList var4 = Lists.newArrayList(var1); Iterator var2 = var4.iterator(); while (var2.hasNext()) { AttributeModifier var3 = (AttributeModifier)var2.next(); this.removeModifier(var3); } } } public double getAttributeValue() { if (this.needsUpdate) { this.cachedValue = this.computeValue(); this.needsUpdate = false; } return this.cachedValue; } private double computeValue() { double var1 = this.getBaseValue(); AttributeModifier var4; for (Iterator var3 = this.func_180375_b(0).iterator(); var3.hasNext(); var1 += var4.getAmount()) { var4 = (AttributeModifier)var3.next(); } double var7 = var1; Iterator var5; AttributeModifier var6; for (var5 = this.func_180375_b(1).iterator(); var5.hasNext(); var7 += var1 * var6.getAmount()) { var6 = (AttributeModifier)var5.next(); } for (var5 = this.func_180375_b(2).iterator(); var5.hasNext(); var7 *= 1.0D + var6.getAmount()) { var6 = (AttributeModifier)var5.next(); } return this.genericAttribute.clampValue(var7); } private Collection func_180375_b(int p_180375_1_) { HashSet var2 = Sets.newHashSet(this.getModifiersByOperation(p_180375_1_)); for (IAttribute var3 = this.genericAttribute.func_180372_d(); var3 != null; var3 = var3.func_180372_d()) { IAttributeInstance var4 = this.attributeMap.getAttributeInstance(var3); if (var4 != null) { var2.addAll(var4.getModifiersByOperation(p_180375_1_)); } } return var2; } }