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

package noppes.npcs;

import net.minecraft.entity.EnumCreatureAttribute;
import net.minecraft.entity.SharedMonsterAttributes;
import net.minecraft.nbt.NBTTagCompound;
import noppes.npcs.constants.EnumParticleType;
import noppes.npcs.constants.EnumPotionType;
import noppes.npcs.controllers.data.MagicData;
import noppes.npcs.entity.EntityNPCInterface;
import noppes.npcs.scripted.ScriptParticle;
import noppes.npcs.util.ValueUtil;

public class DataStats {

    private float attackStrength = 5;
    public int attackSpeed = 20, swingWarmUp = 0, attackRange = 2, knockback = 0;
    public int minDelay = 20, maxDelay = 40, rangedRange = 15, fireRate = 5, burstCount = 1, shotCount = 1, accuracy = 60;
    public int aggroRange = 16;

    public EnumPotionType potionType = EnumPotionType.None;
    public int potionDuration = 5; //20 = 1 second
    public int potionAmp = 0;

    public double maxHealth = 20;
    public int respawnTime = 20;
    public int spawnCycle = 0;
    public boolean hideKilledBody = false;
    public boolean canDespawn = false;
    public boolean playerSetCanDespawn = false;

    public Resistances resistances = new Resistances();
    public MagicData magicData = new MagicData();

    public boolean ignoreCobweb = false;
    public boolean immuneToFire = false;
    public boolean potionImmune = false;
    public int drowningType = 1; // 0 - Never, 1 - Water,  2 - Air
    public boolean burnInSun = false;
    public boolean noFallDamage = false;
    public float healthRegen = 1;
    public float combatRegen = 0;

    public float pDamage = 4;
    public int pImpact = 0, pSize = 5, pSpeed = 10, pArea = 0, pDur = 5;
    public boolean pPhysics = true, pXlr8 = false, pGlows = false, pExplode = false;
    public boolean pRender3D = false, pSpin = false, pStick = false, pBurnItem = false;
    public EnumPotionType pEffect = EnumPotionType.None;
    public EnumParticleType pTrail = EnumParticleType.None;
    public ScriptParticle pCustom = new ScriptParticle("");
    public int pEffAmp = 0;
    public String fireSound = "random.bow";
    public boolean onSoundBegin = false;
    public boolean playBurstSound = true; // Internal Only
    public byte aimType = 0;
    public boolean projectilesKeepTerrain = false;
    public boolean projectileInvincibility = true;


    public EnumCreatureAttribute creatureType = EnumCreatureAttribute.UNDEFINED;

    private EntityNPCInterface npc;
    public boolean attackInvisible = false;

    // 0 - Yes, 1 - No, 2 - Only NPCs, 3 - Only Players, 4 - NPCs and Players
    public int collidesWith = 1;

    public DataStats(EntityNPCInterface npc) {
        this.npc = npc;
    }

    public NBTTagCompound writeToNBT(NBTTagCompound compound) {
        compound.setTag("Resistances", resistances.writeToNBT());
        compound.setDouble("MaxHealth", maxHealth);
        compound.setInteger("AggroRange", aggroRange);
        compound.setBoolean("HideBodyWhenKilled", hideKilledBody);
        compound.setInteger("RespawnTime", respawnTime);
        compound.setInteger("SpawnCycle", spawnCycle);
        compound.setInteger("CreatureType", creatureType.ordinal());
        compound.setBoolean("IgnoreCobweb", ignoreCobweb);
        compound.setFloat("HealthRegen", healthRegen);
        compound.setFloat("CombatRegen", combatRegen);

        compound.setFloat("AttackStrenght", attackStrength);
        compound.setInteger("AttackRange", attackRange);
        compound.setInteger("AttackSpeed", attackSpeed);
        compound.setInteger("SwingWarmup", swingWarmUp);
        compound.setInteger("KnockBack", knockback);
        compound.setInteger("PotionEffect", potionType.ordinal());
        compound.setInteger("PotionDuration", potionDuration);
        compound.setInteger("PotionAmp", potionAmp);

        compound.setInteger("MaxFiringRange", rangedRange);
        compound.setInteger("FireRate", fireRate);
        compound.setInteger("minDelay", minDelay);
        compound.setInteger("maxDelay", maxDelay);
        compound.setInteger("BurstCount", burstCount);
        compound.setInteger("ShotCount", shotCount);
        compound.setInteger("Accuracy", accuracy);

        compound.setFloat("pDamage", pDamage);
        compound.setInteger("pImpact", pImpact);
        compound.setInteger("pSize", pSize);
        compound.setInteger("pSpeed", pSpeed);
        compound.setInteger("pArea", pArea);
        compound.setInteger("pDur", pDur);
        compound.setBoolean("pPhysics", pPhysics);
        compound.setBoolean("pXlr8", pXlr8);
        compound.setBoolean("pGlows", pGlows);
        compound.setBoolean("pExplode", pExplode);
        compound.setBoolean("pRender3D", pRender3D);
        compound.setBoolean("pSpin", pSpin);
        compound.setBoolean("pStick", pStick);
        compound.setInteger("pEffect", pEffect.ordinal());
        compound.setInteger("pTrail", pTrail.ordinal());
        compound.setInteger("pEffAmp", pEffAmp);
        compound.setString("FiringSound", fireSound);
        compound.setBoolean("FiringSoundBegin", onSoundBegin);
        compound.setByte("AimType", aimType);
        compound.setBoolean("ProjectilesKeepTerrain", projectilesKeepTerrain);
        compound.setBoolean("ProjectileInvincibility", projectileInvincibility);
        if (pTrail == EnumParticleType.Custom)
            compound.setTag("pCustom", pCustom.writeToNBT());
        if (pEffect == EnumPotionType.Fire)
            compound.setBoolean("pBurnItem", pBurnItem);

        compound.setBoolean("ImmuneToFire", immuneToFire);
        compound.setBoolean("PotionImmune", potionImmune);
        compound.setInteger("DrowningType", drowningType);
        compound.setBoolean("BurnInSun", burnInSun);
        compound.setBoolean("NoFallDamage", noFallDamage);
        compound.setBoolean("AttackInvisible", attackInvisible);
        compound.setBoolean("CanDespawn", canDespawn);
        compound.setBoolean("PlayerSetCanDespawn", playerSetCanDespawn);
        compound.setInteger("CollidesWith", collidesWith);

        magicData.writeToNBT(compound);

        return compound;
    }

    public void readToNBT(NBTTagCompound compound) {
        resistances.readToNBT(compound.getCompoundTag("Resistances"));
        setMaxHealth(compound.getDouble("MaxHealth"));
        hideKilledBody = compound.getBoolean("HideBodyWhenKilled");
        aggroRange = compound.getInteger("AggroRange");
        respawnTime = compound.getInteger("RespawnTime");
        spawnCycle = compound.getInteger("SpawnCycle");
        creatureType = EnumCreatureAttribute.values()[compound.getInteger("CreatureType") % EnumPotionType.values().length];
        ignoreCobweb = compound.getBoolean("IgnoreCobweb");
        healthRegen = compound.getFloat("HealthRegen");
        combatRegen = compound.getFloat("CombatRegen");

        setAttackStrength(compound.getFloat("AttackStrenght"));
        attackSpeed = compound.getInteger("AttackSpeed");
        swingWarmUp = ValueUtil.clamp(compound.getInteger("SwingWarmup"), 0, 1000);
        attackRange = compound.getInteger("AttackRange");
        knockback = compound.getInteger("KnockBack");
        potionType = EnumPotionType.values()[compound.getInteger("PotionEffect") % EnumPotionType.values().length];
        potionDuration = compound.getInteger("PotionDuration");
        potionAmp = compound.getInteger("PotionAmp");

        rangedRange = compound.getInteger("MaxFiringRange");
        fireRate = compound.getInteger("FireRate");
        minDelay = ValueUtil.clamp(compound.getInteger("minDelay"), 1, 9999);
        maxDelay = ValueUtil.clamp(compound.getInteger("maxDelay"), 1, 9999);
        burstCount = compound.getInteger("BurstCount");
        shotCount = ValueUtil.clamp(compound.getInteger("ShotCount"), 1, 10);
        accuracy = compound.getInteger("Accuracy");

        pDamage = compound.getFloat("pDamage");
        pImpact = compound.getInteger("pImpact");
        pSize = compound.getInteger("pSize");
        pSpeed = compound.getInteger("pSpeed");
        pArea = compound.getInteger("pArea");
        pDur = compound.getInteger("pDur");
        pPhysics = compound.getBoolean("pPhysics");
        pXlr8 = compound.getBoolean("pXlr8");
        pGlows = compound.getBoolean("pGlows");
        pExplode = compound.getBoolean("pExplode");
        pRender3D = compound.getBoolean("pRender3D");
        pSpin = compound.getBoolean("pSpin");
        pStick = compound.getBoolean("pStick");
        pEffect = EnumPotionType.values()[compound.getInteger("pEffect") % EnumPotionType.values().length];
        pTrail = EnumParticleType.values()[compound.getInteger("pTrail") % EnumParticleType.values().length];
        pEffAmp = compound.getInteger("pEffAmp");
        fireSound = compound.getString("FiringSound");
        onSoundBegin = compound.getBoolean("FiringSoundBegin");
        aimType = compound.getByte("AimType");
        projectilesKeepTerrain = compound.getBoolean("ProjectilesKeepTerrain");
        if (compound.hasKey("ProjectileInvincibility"))
            projectileInvincibility = compound.getBoolean("ProjectileInvincibility");
        else
            projectileInvincibility = true;
        if (pTrail == EnumParticleType.Custom)
            pCustom = ScriptParticle.fromNBT(compound.getCompoundTag("pCustom"));
        if (pEffect == EnumPotionType.Fire)
            pBurnItem = compound.getBoolean("pBurnItem");

        immuneToFire = compound.getBoolean("ImmuneToFire");
        potionImmune = compound.getBoolean("PotionImmune");
        drowningType = compound.getInteger("DrowningType");

        burnInSun = compound.getBoolean("BurnInSun");
        noFallDamage = compound.getBoolean("NoFallDamage");
        attackInvisible = compound.getBoolean("AttackInvisible");
        canDespawn = compound.getBoolean("CanDespawn");
        playerSetCanDespawn = compound.getBoolean("PlayerSetCanDespawn");
        magicData.readToNBT(compound);

        npc.setImmuneToFire(immuneToFire);

        if (compound.hasKey("CollidesWith"))
            collidesWith = compound.getInteger("CollidesWith");
        else
            collidesWith = 1;
    }


    public float getAttackStrength() {
        return attackStrength;
    }

    public void setAttackStrength(double strength) {
        attackStrength = (float) (Math.floor(strength));
        npc.getEntityAttribute(SharedMonsterAttributes.attackDamage).setBaseValue(attackStrength);
    }

    public void setMaxHealth(double maxHealth) {
        this.maxHealth = Math.floor(maxHealth);
        npc.getEntityAttribute(SharedMonsterAttributes.maxHealth).setBaseValue(maxHealth);
    }
}