Mekanism-tilera-Edition/src/main/java/mekanism/common/LaserManager.java

package mekanism.common;

import java.util.List;

import com.mojang.realmsclient.util.Pair;

import mekanism.api.Coord4D;
import mekanism.api.MekanismConfig.general;
import mekanism.api.Pos3D;
import mekanism.api.lasers.ILaserReceptor;
import net.minecraft.block.Block;
import net.minecraft.entity.Entity;
import net.minecraft.item.ItemStack;
import net.minecraft.tileentity.TileEntity;
import net.minecraft.util.DamageSource;
import net.minecraft.util.MovingObjectPosition;
import net.minecraft.util.Vec3;
import net.minecraft.world.World;
import net.minecraftforge.common.util.ForgeDirection;

public class LaserManager
{
	public static LaserInfo fireLaser(TileEntity from, ForgeDirection direction, double energy, World world)
	{
		return fireLaser(new Pos3D(from).centre().translate(direction, 0.501), direction, energy, world);
	}

	public static LaserInfo fireLaser(Pos3D from, ForgeDirection direction, double energy, World world)
	{
		Pos3D to = from.clone().translate(direction, general.laserRange - 0.002);

		MovingObjectPosition mop = world.rayTraceBlocks(Vec3.createVectorHelper(from.xPos, from.yPos, from.zPos), Vec3.createVectorHelper(to.xPos, to.yPos, to.zPos));

		if(mop != null)
		{
			to = new Pos3D(mop.hitVec);
			Coord4D toCoord = new Coord4D(mop.blockX, mop.blockY, mop.blockZ);
			TileEntity tile = toCoord.getTileEntity(world);

			if(tile instanceof ILaserReceptor)
			{
				if(!(((ILaserReceptor)tile).canLasersDig()))
				{
					((ILaserReceptor)tile).receiveLaserEnergy(energy, ForgeDirection.getOrientation(mop.sideHit));
				}
			}
		}

		from.translateExcludingSide(direction, -0.1);
		to.translateExcludingSide(direction, 0.1);
		
		boolean foundEntity = false;

		for(Entity e : (List<Entity>)world.getEntitiesWithinAABB(Entity.class, Pos3D.getAABB(from, to)))
		{
			foundEntity = true;
			
			if(!e.isImmuneToFire()) 
			{
				e.setFire((int)(energy / 1000));
			}
			
			if(energy > 256)
			{
				e.attackEntityFrom(DamageSource.generic, (float)energy/1000F);
			}
		}
		
		return new LaserInfo(mop, foundEntity);
	}

	public static List<ItemStack> breakBlock(Coord4D blockCoord, boolean dropAtBlock, World world)
	{
		List<ItemStack> ret = null;
		Block blockHit = blockCoord.getBlock(world);
		
		if(dropAtBlock)
		{
			blockHit.dropBlockAsItem(world, blockCoord.xCoord, blockCoord.yCoord, blockCoord.zCoord, blockCoord.getMetadata(world), 0);
		}
		else {
			ret = blockHit.getDrops(world, blockCoord.xCoord, blockCoord.yCoord, blockCoord.zCoord, blockCoord.getMetadata(world), 0);
		}
		
		blockHit.breakBlock(world, blockCoord.xCoord, blockCoord.yCoord, blockCoord.zCoord, blockHit, blockCoord.getMetadata(world));
		world.setBlockToAir(blockCoord.xCoord, blockCoord.yCoord, blockCoord.zCoord);
		world.playAuxSFX(2001, blockCoord.xCoord, blockCoord.yCoord, blockCoord.zCoord, Block.getIdFromBlock(blockHit));
		
		return ret;
	}

	public static MovingObjectPosition fireLaserClient(TileEntity from, ForgeDirection direction, double energy, World world)
	{
		return fireLaserClient(new Pos3D(from).centre().translate(direction, 0.501), direction, energy, world);
	}

	public static MovingObjectPosition fireLaserClient(Pos3D from, ForgeDirection direction, double energy, World world)
	{
		Pos3D to = from.clone().translate(direction, general.laserRange - 0.002);
		MovingObjectPosition mop = world.rayTraceBlocks(Vec3.createVectorHelper(from.xPos, from.yPos, from.zPos), Vec3.createVectorHelper(to.xPos, to.yPos, to.zPos));

		if(mop != null)
		{
			to = new Pos3D(mop.hitVec);
		}
		
		from.translate(direction, -0.501);
		Mekanism.proxy.renderLaser(world, from, to, direction, energy);
		
		return mop;
	}
	
	public static class LaserInfo
	{
		public MovingObjectPosition movingPos;
		
		public boolean foundEntity;
		
		public LaserInfo(MovingObjectPosition mop, boolean b)
		{
			movingPos = mop;
			foundEntity = b;
		}
	}
}