vazkii.botania.common.item.equipment.tool.ItemThunderSword.java Source code

Introduction

Here is the source code for vazkii.botania.common.item.equipment.tool.ItemThunderSword.java

Source

/**
 * This class was created by <Vazkii>. It's distributed as
 * part of the Botania Mod. Get the Source Code in github:
 * https://github.com/Vazkii/Botania
 *
 * Botania is Open Source and distributed under the
 * Botania License: http://botaniamod.net/license.php
 *
 * File Created @ [Aug 27, 2015, 10:38:50 PM (GMT)]
 */
package vazkii.botania.common.item.equipment.tool;

import java.util.ArrayList;
import java.util.List;
import java.util.Random;
import java.util.function.Predicate;

import javax.annotation.Nonnull;

import com.google.common.collect.Multimap;

import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityLivingBase;
import net.minecraft.entity.SharedMonsterAttributes;
import net.minecraft.entity.ai.attributes.AttributeModifier;
import net.minecraft.entity.monster.IMob;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.inventory.EntityEquipmentSlot;
import net.minecraft.inventory.IInventory;
import net.minecraft.item.ItemStack;
import net.minecraft.stats.Achievement;
import net.minecraft.util.DamageSource;
import net.minecraft.util.math.AxisAlignedBB;
import vazkii.botania.api.BotaniaAPI;
import vazkii.botania.common.Botania;
import vazkii.botania.common.achievement.ICraftAchievement;
import vazkii.botania.common.achievement.ModAchievements;
import vazkii.botania.common.core.helper.ItemNBTHelper;
import vazkii.botania.common.core.helper.Vector3;
import vazkii.botania.common.item.equipment.tool.manasteel.ItemManasteelSword;
import vazkii.botania.common.lib.LibItemNames;

public class ItemThunderSword extends ItemManasteelSword implements ICraftAchievement {

    private static final String TAG_LIGHTNING_SEED = "lightningSeed";

    public ItemThunderSword() {
        super(BotaniaAPI.terrasteelToolMaterial, LibItemNames.THUNDER_SWORD);
    }

    @Override
    public boolean hitEntity(ItemStack stack, EntityLivingBase entity, @Nonnull EntityLivingBase attacker) {
        if (!(entity instanceof EntityPlayer) && entity != null) {
            double range = 8;
            List<EntityLivingBase> alreadyTargetedEntities = new ArrayList<>();
            int dmg = 5;
            long lightningSeed = ItemNBTHelper.getLong(stack, TAG_LIGHTNING_SEED, 0);

            Predicate<Entity> selector = e -> e instanceof EntityLivingBase && e instanceof IMob
                    && !(e instanceof EntityPlayer) && !alreadyTargetedEntities.contains(e);

            Random rand = new Random(lightningSeed);
            EntityLivingBase lightningSource = entity;
            int hops = entity.worldObj.isThundering() ? 10 : 4;
            for (int i = 0; i < hops; i++) {
                List<Entity> entities = entity.worldObj.getEntitiesInAABBexcluding(lightningSource,
                        new AxisAlignedBB(lightningSource.posX - range, lightningSource.posY - range,
                                lightningSource.posZ - range, lightningSource.posX + range,
                                lightningSource.posY + range, lightningSource.posZ + range),
                        selector::test);
                if (entities.isEmpty())
                    break;

                EntityLivingBase target = (EntityLivingBase) entities.get(rand.nextInt(entities.size()));
                if (attacker instanceof EntityPlayer)
                    target.attackEntityFrom(DamageSource.causePlayerDamage((EntityPlayer) attacker), dmg);
                else
                    target.attackEntityFrom(DamageSource.causeMobDamage(attacker), dmg);

                Botania.proxy.lightningFX(Vector3.fromEntityCenter(lightningSource),
                        Vector3.fromEntityCenter(target), 1, 0x0179C4, 0xAADFFF);

                alreadyTargetedEntities.add(target);
                lightningSource = target;
                dmg--;
            }

            if (!entity.worldObj.isRemote)
                ItemNBTHelper.setLong(stack, TAG_LIGHTNING_SEED, entity.worldObj.rand.nextLong());
        }

        return super.hitEntity(stack, entity, attacker);
    }

    @Override
    public Achievement getAchievementOnCraft(ItemStack stack, EntityPlayer player, IInventory matrix) {
        return ModAchievements.terrasteelWeaponCraft;
    }

    @Nonnull
    @Override
    public Multimap<String, AttributeModifier> getItemAttributeModifiers(EntityEquipmentSlot equipmentSlot) {
        Multimap<String, AttributeModifier> multimap = super.getItemAttributeModifiers(equipmentSlot);

        if (equipmentSlot == EntityEquipmentSlot.MAINHAND) {
            multimap.removeAll(SharedMonsterAttributes.ATTACK_SPEED.getAttributeUnlocalizedName());
            multimap.put(SharedMonsterAttributes.ATTACK_SPEED.getAttributeUnlocalizedName(),
                    new AttributeModifier(ATTACK_SPEED_MODIFIER, "Weapon modifier", -1.5, 0));
        }

        return multimap;
    }

}