package com.genuineflix.caption.caption;

import java.util.Comparator;

import net.minecraft.client.Minecraft;
import net.minecraft.client.audio.ISound;
import net.minecraft.entity.Entity;

public class CaptionWorld extends Caption {

	public Entity entity;
	public ISound sound;
	public double posX, prevPosX;
	public double posY, prevPosY;
	public double posZ, prevPosZ;
	public double size = 1;
	public static Comparator<CaptionWorld> DISTANCE = new Comparator<CaptionWorld>() {

		@Override
		public int compare(final CaptionWorld o1, final CaptionWorld o2) {
			if (Minecraft.getMinecraft().thePlayer == null)
				return 0;
			final double d1 = o1.getDistanceTo(Minecraft.getMinecraft().thePlayer);
			final double d2 = o2.getDistanceTo(Minecraft.getMinecraft().thePlayer);
			if (d1 < d2)
				return 1;
			if (d1 > d2)
				return -1;
			return 0;
		}
	};

	public CaptionWorld(final String message, final Entity entity, final float volume, final float pitch) {
		super(message, volume, pitch);
		this.attach(entity);
	}

	public CaptionWorld(final String message, final ISound sound, final float volume, final float pitch) {
		super(message, volume, pitch);
		this.attach(sound);
	}

	public void attach(final Entity entity) {
		if (this.entity != null || entity == null)
			return;
		this.entity = entity;
		posX = entity.posX;
		posY = entity.posY;
		posZ = entity.posZ;
		prevPosX = entity.prevPosX;
		prevPosY = entity.prevPosY;
		prevPosZ = entity.prevPosZ;
		size = entity.height + 0.5;
	}

	public void attach(final ISound sound) {
		if (sound == null)
			return;
		this.sound = sound;
		posX = sound.getXPosF();
		posY = sound.getYPosF();
		posZ = sound.getZPosF();
	}

	@Override
	public int compareTo(final Caption o) {
		if (o instanceof CaptionWorld)
			return (int) (getDistanceTo((CaptionWorld) o) * 10000f);
		return key.compareTo(o.key);
	}

	public boolean equalTo(final CaptionWorld caption) {
		if (!nameEquals(caption.key))
			return false;
		return getDistanceTo(caption) <= 0.1;
	}

	public double getDistanceTo(final CaptionWorld caption) {
		return getDistanceTo(caption.posX, caption.posY, caption.posZ);
	}

	public double getDistanceTo(final double posX, final double posY, final double posZ) {
		final double distanceX = this.posX - posX;
		final double distanceY = this.posY - posY;
		final double distanceZ = this.posZ - posZ;
		return Math.sqrt(distanceX * distanceX + distanceY * distanceY + distanceZ * distanceZ);
	}

	public double getDistanceTo(final Entity entity) {
		return getDistanceTo(entity.posX, entity.posY, entity.posZ);
	}

	public double getDistanceToPlayer() {
		return getDistanceTo(Minecraft.getMinecraft().thePlayer);
	}

	public float getScale() {
		if (isEntity()) {
			float out = (float) (size / 2);
			if (out < 1)
				out = 1;
			return out;
		}
		return 1;
	}

	public boolean isEntity() {
		return entity != null;
	}

	public boolean isSound() {
		return sound != null;
	}

	public boolean isWithin(final CaptionWorld caption, final double distance) {
		return entity != null && getDistanceTo(caption) <= distance;
	}

	public boolean isWithin(final Entity entity, final double distance) {
		return entity != null && getDistanceTo(entity) <= distance;
	}

	public boolean isWithinPlayer(final double distance) {
		return Minecraft.getMinecraft().thePlayer != null || getDistanceToPlayer() <= distance;
	}

	@Override
	public boolean tick() {
		if (isEntity() && entity.isDead)
			entity = null;
		updatePos();
		return super.tick();
	}

	private void updatePos() {
		if (isEntity()) {
			prevPosX = entity.prevPosX;
			prevPosY = entity.prevPosY;
			prevPosZ = entity.prevPosZ;
			posX = entity.posX;
			posY = entity.posY;
			posZ = entity.posZ;
		} else if (isSound()) {
			prevPosX = posX;
			prevPosY = posY;
			prevPosZ = posZ;
			posX = sound.getXPosF();
			posY = sound.getYPosF();
			posZ = sound.getZPosF();
		} else {
			prevPosX = posX;
			prevPosY = posY;
			prevPosZ = posZ;
		}
	}
}