com.goncalomb.bukkit.nbteditor.nbt.MobNBT.java Source code

Introduction

Here is the source code for com.goncalomb.bukkit.nbteditor.nbt.MobNBT.java

Source

/*
 * Copyright (C) 2013-2015 Gonalo Baltazar <me@goncalomb.com>
 *
 * This file is part of NBTEditor.
 *
 * NBTEditor is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * NBTEditor is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with NBTEditor.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.goncalomb.bukkit.nbteditor.nbt;

import java.util.Arrays;

import org.apache.commons.lang.ArrayUtils;
import org.bukkit.inventory.ItemStack;

import com.goncalomb.bukkit.mylib.reflect.NBTTagList;
import com.goncalomb.bukkit.mylib.reflect.NBTUtils;
import com.goncalomb.bukkit.nbteditor.nbt.attributes.AttributeContainer;
import com.goncalomb.bukkit.nbteditor.nbt.variable.BooleanVariable;
import com.goncalomb.bukkit.nbteditor.nbt.variable.FloatVariable;
import com.goncalomb.bukkit.nbteditor.nbt.variable.NBTGenericVariableContainer;
import com.goncalomb.bukkit.nbteditor.nbt.variable.ShortVariable;
import com.goncalomb.bukkit.nbteditor.nbt.variable.StringVariable;

public class MobNBT extends EquippableNBT {

    static {
        NBTGenericVariableContainer variables = new NBTGenericVariableContainer("Mob");
        variables.add("Health", new FloatVariable("HealF", 0.0f));
        variables.add("AttackTime", new ShortVariable("AttackTime"));
        variables.add("HurtTime", new ShortVariable("HurtTime"));
        variables.add("DeathTime", new ShortVariable("DeathTime"));
        variables.add("PickLoot", new BooleanVariable("CanPickUpLoot"));
        variables.add("NoAI", new BooleanVariable("NoAI"));
        variables.add("Persistent", new BooleanVariable("PersistenceRequired"));
        variables.add("Name", new StringVariable("CustomName"));
        variables.add("NameVisible", new BooleanVariable("CustomNameVisible"));
        registerVariables(MobNBT.class, variables);
    }

    public void setDropChances(float hand, float feet, float legs, float chest, float head) {
        _data.setList("DropChances", hand, feet, legs, chest, head);
    }

    public float[] getDropChances() {
        if (_data.hasKey("DropChances")) {
            return ArrayUtils
                    .toPrimitive(Arrays.copyOfRange(_data.getListAsArray("DropChances"), 0, 5, Float[].class));
        }
        return null;
    }

    public void clearDropChances() {
        _data.remove("DropChances");
    }

    public void setEffectsFromPotion(ItemStack potion) {
        if (potion != null) {
            NBTTagList effects = NBTUtils.potionToNBTEffectsList(potion);
            if (effects != null) {
                _data.setList("ActiveEffects", effects);
                return;
            }
        } else {
            _data.remove("ActiveEffects");
        }
    }

    public ItemStack getEffectsAsPotion() {
        if (_data.hasKey("ActiveEffects")) {
            return NBTUtils.potionFromNBTEffectsList(_data.getList("ActiveEffects"));
        }
        return null;
    }

    public AttributeContainer getAttributes() {
        if (_data.hasKey("Attributes")) {
            return AttributeContainer.fromNBT(_data.getList("Attributes"));
        }
        return new AttributeContainer();
    }

    public void setAttributes(AttributeContainer container) {
        if (container == null || container.size() == 0) {
            _data.remove("Attributes");
        } else {
            _data.setList("Attributes", container.toNBT());
        }
    }

}