CustomNPC-Plus/src/main/java/noppes/npcs/scripted/ScriptEntity.java

package noppes.npcs.scripted;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import net.minecraft.block.material.Material;
import net.minecraft.client.Minecraft;
import net.minecraft.entity.Entity;
import net.minecraft.entity.EntityList;
import net.minecraft.entity.EntityLivingBase;
import net.minecraft.entity.monster.EntityMob;
import net.minecraft.entity.passive.EntityAnimal;
import net.minecraft.entity.player.EntityPlayer;
import net.minecraft.nbt.NBTBase;
import net.minecraft.nbt.NBTBase.NBTPrimitive;
import net.minecraft.nbt.NBTTagCompound;
import net.minecraft.nbt.NBTTagString;
import net.minecraft.util.DamageSource;
import net.minecraft.util.MathHelper;
import net.minecraft.world.World;
import noppes.npcs.NoppesUtilServer;
import noppes.npcs.Server;
import noppes.npcs.constants.EnumPacketClient;
import noppes.npcs.controllers.ScriptController;
import noppes.npcs.entity.EntityNPCInterface;
import noppes.npcs.scripted.constants.EntityType;

public class ScriptEntity {
	protected Entity entity;
	private Map<String,Object> tempData = new HashMap<String,Object>();

	public ScriptEntity(Entity entity){
		this.entity = entity;
	}

	/**
	 * @param directory The particle's directory. Use only forward slashes when writing a directory. Example: "customnpcs:textures/particle/tail.png"
	 * @param HEXcolor The particle's color as a HEX integer
	 * @param amount The amount of particles to spawn
	 * @param maxAge The particle's maximum age in MC ticks

	 * @param x The particle's x position
	 * @param y The particle's y position
	 * @param z The particle's z position

	 * @param motionX The particle's speed in the x axis
	 * @param motionY The particle's speed in the y axis
	 * @param motionZ The particle's speed in the z axis
	 * @param gravity The particle's gravity

	 * @param scale1 The particle's starting scale
	 * @param scale2 The particle's ending scale
	 * @param scaleRate Multiplier for the particle's scale growth rate
	 * @param scaleRateStart The time at which the particle begins growing/shrinking

	 * @param alpha1 The particle's starting transparency
	 * @param alpha2 The particle's ending transparency
	 * @param scaleRate Multiplier for the particle's transparency growth rate
	 * @param alphaRateStart The time at which the particle begins appearing/fading
	 */
	public void spawnParticle(String directory, int HEXcolor, int amount, int maxAge,
							  double x, double y, double z,
							  double motionX, double motionY, double motionZ, float gravity,
							  float scale1, float scale2, float scaleRate, int scaleRateStart,
							  float alpha1, float alpha2, float alphaRate, int alphaRateStart
	) {
		int entityID = entity.getEntityId();
		NoppesUtilServer.spawnScriptedParticle(entity, directory, HEXcolor, amount, maxAge,
				x, y, z,
				motionX, motionY, motionZ, gravity,
				scale1, scale2, scaleRate, scaleRateStart,
				alpha1, alpha2, alphaRate, alphaRateStart,
				entityID
		);
	}
	public void spawnParticle(ScriptEntityParticle entityParticle) { entityParticle.spawnOnEntity(this); }

	public double getYOffset() { return entity.getYOffset(); }

	/**
	 * @return The entities width
	 */
	public double getWidth(){ return entity.width; }

	/**
	 * @return The entities height
	 */
	public double getHeight(){ return entity.height; }

	/**
	 * @return The entities x position
	 */
	public double getX(){
		return entity.posX;
	}
	
	/**
	 * @param x The entities x position
	 */
	public void setX(double x){
		entity.posX = x;
	}
	
	/**
	 * @return The entities y position
	 */
	public double getY(){
		return entity.posY;
	}
	
	/**
	 * @param y The entities y position
	 */
	public void setY(double y){
		entity.posY = y;
	}
	
	/**
	 * @return The entities x position
	 */
	public double getZ(){
		return entity.posZ;
	}
	
	/**
	 * @param z The entities x position
	 */
	public void setZ(double z){
		entity.posZ = z;
	}

	/**
	 * @return The block x position
	 */
	public int getBlockX(){
		return MathHelper.floor_double(entity.posX);
	}

	/**
	 * @return The block y position
	 */
	public int getBlockY(){
		return MathHelper.floor_double(entity.posY);
	}

	/**
	 * @return The block z position
	 */
	public int getBlockZ(){
		return MathHelper.floor_double(entity.posZ);
	}
		
	/**
	 * @param x The x position
	 * @param y The y position
	 * @param z The z position
	 */
	public void setPosition(double x, double y, double z){
		entity.setPosition(x, y, z);
	}

	
	/**
	 * @param range The search range for entities around this entity
	 * @return Array of entities within range
	 */
	public ScriptEntity[] getSurroundingEntities(int range){
		List<Entity> entities = entity.worldObj.getEntitiesWithinAABB(Entity.class, entity.boundingBox.expand(range, range, range));
		List<ScriptEntity> list = new ArrayList<ScriptEntity>();
		for(Entity living : entities){
			if(living != entity)
				list.add(ScriptController.Instance.getScriptForEntity(living));
		}
		return list.toArray(new ScriptEntity[list.size()]);
	}
		
	/**
	 * @param range The search range for entities around this entity
	 * @param type The EntityType you want to find
	 * @return Array of entities within range
	 */
	public ScriptEntity[] getSurroundingEntities(int range, int type){
		Class cls = Entity.class;
		if(type == EntityType.LIVING)
			cls = EntityLivingBase.class;
		else if(type == EntityType.PLAYER)
			cls = EntityPlayer.class;
		else if(type == EntityType.ANIMAL)
			cls = EntityAnimal.class;
		else if(type == EntityType.MONSTER)
			cls = EntityMob.class;
		else if(type == EntityType.NPC)
			cls = EntityNPCInterface.class;
		
		List<Entity> entities = entity.worldObj.getEntitiesWithinAABB(cls, entity.boundingBox.expand(range, range, range));
		List<ScriptEntity> list = new ArrayList<ScriptEntity>();
		for(Entity living : entities){
			if(living != entity)
				list.add(ScriptController.Instance.getScriptForEntity(living));
		}
		return list.toArray(new ScriptEntity[list.size()]);
	}	

	
	/**
	 * @return Whether the entity is alive or not
	 */
	public boolean isAlive(){
		return entity.isEntityAlive();
	}
	
	/**
	 * @param key Get temp data for this key
	 * @return Returns the stored temp data
	 */
	public Object getTempData(String key){
		return tempData.get(key);
	}
	
	/**
	 * Tempdata gets cleared when the entity gets unloaded or the world restarts
	 * @param key The key for the data stored
	 * @param value The data stored
	 */
	public void setTempData(String key, Object value){
		tempData.put(key, value);
	}
	
	/**
	 * @param key The key thats going to be tested against the temp data
	 * @return Whether or not temp data containes the key
	 */
	public boolean hasTempData(String key){
		return tempData.containsKey(key);
	}
	
	/**
	 * @param key The key for the temp data to be removed
	 */
	public void removeTempData(String key){
		tempData.remove(key);
	}
	
	/**
	 * Remove all tempdata
	 */
	public void clearTempData(){
		tempData.clear();
	}
	
	/**
	 * @param key The key of the data to be returned
	 * @return Returns the stored data
	 */
	public Object getStoredData(String key){
		NBTTagCompound compound = getStoredCompound();
		if(!compound.hasKey(key))
			return null;
		NBTBase base = compound.getTag(key);
		if(base instanceof NBTPrimitive)
			return ((NBTPrimitive)base).func_150286_g();
		return ((NBTTagString)base).func_150285_a_();
	}
	
	/**
	 * Stored data persists through world restart. Unlike tempdata only Strings and Numbers can be saved
	 * @param key The key for the data stored
	 * @param value The data stored. This data can be either a Number or a String. Other data is not stored
	 */
	public void setStoredData(String key, Object value){
		NBTTagCompound compound = getStoredCompound();
		if(value instanceof Number){
			compound.setDouble(key, ((Number) value).doubleValue());
		}
		else if(value instanceof String)
			compound.setString(key, (String)value);
		saveStoredCompound(compound);
	}
	
	/**
	 * @param key The key of the data to be checked
	 * @return Returns whether or not the stored data contains the key
	 */
	public boolean hasStoredData(String key){
		return getStoredCompound().hasKey(key);
	}
	
	/**
	 * @param key The key of the data to be removed
	 */
	public void removeStoredData(String key){
		NBTTagCompound compound = getStoredCompound();
		compound.removeTag(key);
		saveStoredCompound(compound);
	}
	
	/**
	 * Remove all stored data
	 */
	public void clearStoredData(){
		entity.getEntityData().removeTag("CNPCStoredData");
	}
	
	private NBTTagCompound getStoredCompound(){
		NBTTagCompound compound = entity.getEntityData().getCompoundTag("CNPCStoredData");
		if(compound == null)
			entity.getEntityData().setTag("CNPCStoredData", compound = new NBTTagCompound());
		return compound;
	}
	
	private void saveStoredCompound(NBTTagCompound compound){
		entity.getEntityData().setTag("CNPCStoredData", compound);
	}
	
	/**
	 * @return The age of this entity in ticks
	 */
	public long getAge(){
		return entity.ticksExisted;
	}
	
	/**
	 * Despawns this entity. Removes it permanently
	 */
	public void despawn(){
		entity.isDead = true;
	}
	
	/**
	 * @return Return whether or not this entity is standing in water
	 */
	public boolean inWater(){
		return entity.isInsideOfMaterial(Material.water);
	}
	
	/**
	 * @return Return whether or not this entity is standing in lava
	 */
	public boolean inLava(){
		return entity.isInsideOfMaterial(Material.lava);
	}
	
	/**
	 * @return Return whether or not this entity is standing in fire
	 */
	public boolean inFire(){
		return entity.isInsideOfMaterial(Material.fire);
	}
	
	/**
	 * @return Return whether or not this entity is on fire
	 */
	public boolean isBurning(){
		return entity.isBurning();
	}
	
	/**
	 * @param ticks Amount of world ticks this entity will burn. 20 ticks equals 1 second
	 */
	public void setBurning(int ticks){
		entity.setFire(ticks);
	}
	
	/**
	 * Removes fire from this entity
	 */
	public void extinguish(){
		entity.extinguish();
	}

	/**
	 * @return Name as which it's registered in minecraft
	 */
	public String getTypeName(){
		return EntityList.getEntityString(entity);
	}
	
	/**
	 * @param item Item to be dropped
	 */
	public void dropItem(ScriptItemStack item){
		entity.entityDropItem(item.item, 0);
	}

	/**
	 * @return Return the rider
	 */
	public ScriptEntity getRider(){
		return ScriptController.Instance.getScriptForEntity(entity.riddenByEntity);
	}
	
	/**
	 * @param entity The entity to ride this entity
	 */
	public void setRider(ScriptEntity entity){
		if(entity != null){
			entity.entity.mountEntity(this.entity);;
		}
		else if(this.entity.riddenByEntity != null)
			this.entity.riddenByEntity.mountEntity(null);
	}

	/**
	 * @return Return the entity, this entity is riding
	 */
	public ScriptEntity getMount(){
		return ScriptController.Instance.getScriptForEntity(entity.ridingEntity);
	}
	
	/**
	 * @param entity The entity this entity will mount
	 */
	public void setMount(ScriptEntity entity){
		if(entity == null)
			this.entity.mountEntity(null);
		else
			this.entity.mountEntity(entity.entity);
	}

	/**
	 * @see noppes.npcs.scripted.constants.EntityType
	 * @return Returns the EntityType of this entity
	 */
	public int getType(){
		return EntityType.UNKNOWN;
	}
	
	/**
	 * @since 1.7.10c
	 * @param type @EntityType to check
	 * @return Returns whether the entity is type of the given @EntityType
	 */
	public boolean typeOf(int type){
		return type == EntityType.UNKNOWN;
	}
	
	/**
	 * @param rotation The rotation to be set (0-360)
	 */
	public void setRotation(float rotation){
		entity.rotationYaw = rotation;
	}
	
	/**
	 * @return Current rotation of the npc
	 */
	public float getRotation(){
		return entity.rotationYaw;
	}
	
	/**
	 * @param power How strong the knockback is
	 * @param direction The direction in which he flies back (0-360). Usually based on getRotation()
	 */
	public void knockback(int power, float direction){
		float v = direction * (float)Math.PI / 180.0F;
        entity.addVelocity(-MathHelper.sin(v) * (float)power, 0.1D + power * 0.04f, MathHelper.cos(v) * (float)power);
        entity.motionX *= 0.6D;
        entity.motionZ *= 0.6D;
        entity.attackEntityFrom(DamageSource.outOfWorld, 0.0001F);
	}
	
	/**
	 * @since 1.7.10c
	 * @return Returns whether or not this entity is sneaking
	 */
	public boolean isSneaking(){
		return entity.isSneaking();
	}
	
	/**
	 * @since 1.7.10c
	 * @return Returns whether or not this entity is sprinting
	 */
	public boolean isSprinting(){
		return entity.isSprinting();
	}
	
	/**
	 * @since 1.7.10c
	 * Expert users only
	 * @return Returns minecrafts entity
	 */
	public Entity getMCEntity(){
		return entity;
	}
}