CustomNPC-Plus/src/main/java/noppes/npcs/Resistances.java

package noppes.npcs;

import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.util.DamageSource;

public class Resistances {

	public float knockback = 1f;
	public float arrow = 1f;
	public float playermelee = 1f;
	public float explosion = 1f;

	public boolean disableDamage = false;
	
	public NBTTagCompound writeToNBT() {
		NBTTagCompound compound = new NBTTagCompound();
		compound.setFloat("Knockback", knockback);
		compound.setFloat("Arrow", arrow);
		compound.setFloat("Melee", playermelee);
		compound.setFloat("Explosion", explosion);
		compound.setBoolean("disableDamage", disableDamage);
		return compound;
	}

	public void readToNBT(NBTTagCompound compound) {
		knockback = compound.getFloat("Knockback");
		arrow = compound.getFloat("Arrow");
		playermelee = compound.getFloat("Melee");
		explosion = compound.getFloat("Explosion");
		disableDamage = compound.getBoolean("disableDamage");
	}

	public float applyResistance(DamageSource source, float damage) {
		if(disableDamage){
			return 0;
		}

		if(source.damageType.equals("arrow") || source.damageType.equals("thrown")){
			damage *= (2 - arrow);
		}
		else if(source.damageType.equals("player") || source.damageType.equals("mob")){
			damage *= (2 - playermelee);
		}
		else if(source.damageType.equals("explosion") || source.damageType.equals("explosion.player")){
			damage *= (2 - explosion);
		}
		
		return damage;
	}

}