
package net.minecraftforge.fluids;

import java.util.HashMap;
import java.util.Map;
import java.util.Random;

import net.minecraft.block.Block;
import net.minecraft.block.material.Material;
import net.minecraft.entity.Entity;
import net.minecraft.util.AxisAlignedBB;
import net.minecraft.util.Vec3;
import net.minecraft.world.IBlockAccess;
import net.minecraft.world.World;

/**
 * This is a base implementation for Fluid blocks.
 * 
 * It is highly recommended that you extend this class or one of the Forge-provided child classes.
 * 
 * @author King Lemming, OvermindDL1
 * 
 */
public abstract class BlockFluidBase extends Block implements IFluidBlock {

    protected final static Map<Integer, Boolean> defaultDisplacementIds = new HashMap<Integer, Boolean>();

    static {
        defaultDisplacementIds.put(Block.doorWood.blockID, false);
        defaultDisplacementIds.put(Block.doorIron.blockID, false);
        defaultDisplacementIds.put(Block.signPost.blockID, false);
        defaultDisplacementIds.put(Block.signWall.blockID, false);
        defaultDisplacementIds.put(Block.reed.blockID, false);
    }
    protected Map<Integer, Boolean> displacementIds = new HashMap<Integer, Boolean>();

    protected int quantaPerBlock = 8;
    protected float quantaPerBlockFloat = 8F;
    protected int density = 1;
    protected int densityDir = -1;

    protected int tickRate = 20;
    protected int renderPass = 1;
    protected int maxScaledLight = 0;

    protected final String fluidName;

    public BlockFluidBase(int id, Fluid fluid, Material material) {

        super(id, material);
        this.setBlockBounds(0.0F, 0.0F, 0.0F, 1.0F, 1.0F, 1.0F);
        this.setTickRandomly(true);
        this.disableStats();

        this.fluidName = fluid.getName();
        this.density = fluid.density;
        this.maxScaledLight = fluid.luminosity;
        this.tickRate = fluid.viscosity / 200;
        fluid.setBlockID(id);

        displacementIds.putAll(defaultDisplacementIds);
    }

    public BlockFluidBase setQuantaPerBlock(int quantaPerBlock) {

        if (quantaPerBlock > 16 || quantaPerBlock < 1) {
            quantaPerBlock = 8;
        }
        this.quantaPerBlock = quantaPerBlock;
        this.quantaPerBlockFloat = quantaPerBlock;
        return this;
    }

    public BlockFluidBase setDensity(int density) {

        if (density == 0) {
            density = 1;
        }
        this.density = density;
        this.densityDir = density > 0 ? -1 : 1;
        return this;
    }

    public BlockFluidBase setTickRate(int tickRate) {

        if (tickRate <= 0) {
            tickRate = 20;
        }
        this.tickRate = tickRate;
        return this;
    }

    public BlockFluidBase setRenderPass(int renderPass) {

        this.renderPass = renderPass;
        return this;
    }

    public BlockFluidBase setMaxScaledLight(int maxScaledLight) {

        this.maxScaledLight = maxScaledLight;
        return this;
    }

    /**
     * Returns true if the block at (x, y, z) is displaceable. Does not displace the block.
     */
    public boolean canDisplace(IBlockAccess world, int x, int y, int z) {

        int bId = world.getBlockId(x, y, z);

        if (bId == 0) {
            return true;
        }
        if (bId == blockID) {
            return false;
        }
        if (displacementIds.containsKey(bId)) {
            return displacementIds.get(bId);
        }
        Material material = Block.blocksList[bId].blockMaterial;

        if (material.blocksMovement() || material == Material.portal) {
            return false;
        }
        return true;
    }

    /**
     * Attempt to displace the block at (x, y, z), return true if it was displaced.
     */
    public boolean displaceIfPossible(World world, int x, int y, int z) {

        int bId = world.getBlockId(x, y, z);

        if (bId == 0) {
            return true;
        }
        if (bId == blockID) {
            return false;
        }
        if (displacementIds.containsKey(bId)) {
            if (displacementIds.get(bId)) {
                Block.blocksList[bId].dropBlockAsItem(world, x, y, z, world.getBlockMetadata(x, y, z), 0);
                return true;
            }
            return false;
        }
        Material material = Block.blocksList[bId].blockMaterial;

        if (material.blocksMovement() || material == Material.portal) {
            return false;
        }
        Block.blocksList[bId].dropBlockAsItem(world, x, y, z, world.getBlockMetadata(x, y, z), 0);
        return true;
    }

    public abstract int getQuantaValue(IBlockAccess world, int x, int y, int z);

    @Override
    public abstract boolean canCollideCheck(int meta, boolean fullHit);

    public abstract int getMaxRenderHeightMeta();

    /* BLOCK FUNCTIONS */
    @Override
    public void onBlockAdded(World world, int x, int y, int z) {

        world.scheduleBlockUpdate(x, y, z, blockID, tickRate);
    }

    @Override
    public void onNeighborBlockChange(World world, int x, int y, int z, int blockId) {

        world.scheduleBlockUpdate(x, y, z, blockID, tickRate);
    }

    // Used to prevent updates on chunk generation
    @Override
    public boolean func_82506_l() {

        return false;
    }

    @Override
    public boolean getBlocksMovement(IBlockAccess world, int x, int y, int z) {

        return true;
    }

    @Override
    public AxisAlignedBB getCollisionBoundingBoxFromPool(World world, int x, int y, int z) {

        return null;
    }

    @Override
    public int idDropped(int par1, Random par2Random, int par3) {

        return 0;
    }

    @Override
    public int quantityDropped(Random par1Random) {

        return 0;
    }

    @Override
    public int tickRate(World world) {

        return tickRate;
    }

    @Override
    public void velocityToAddToEntity(World world, int x, int y, int z, Entity entity, Vec3 vec) {

        if (densityDir > 0) {
            return;
        }
        Vec3 vec_flow = this.getFlowVector(world, x, y, z);
        vec.xCoord += vec_flow.xCoord * (quantaPerBlock * 4);
        vec.yCoord += vec_flow.yCoord * (quantaPerBlock * 4);
        vec.zCoord += vec_flow.zCoord * (quantaPerBlock * 4);
    }

    @Override
    public int getLightValue(IBlockAccess world, int x, int y, int z) {

        if (maxScaledLight == 0) {
            return super.getLightValue(world, x, y, z);
        }
        int data = world.getBlockMetadata(x, y, z);
        return (int) (data / quantaPerBlockFloat * maxScaledLight);
    }

    @Override
    public int getRenderType() {

        return FluidRegistry.renderIdFluid;
    }

    @Override
    public boolean isOpaqueCube() {

        return false;
    }

    @Override
    public boolean renderAsNormalBlock() {

        return false;
    }

    @Override
    public float getBlockBrightness(IBlockAccess world, int x, int y, int z) {

        float lightThis = world.getLightBrightness(x, y, z);
        float lightUp = world.getLightBrightness(x, y + 1, z);

        return lightThis > lightUp ? lightThis : lightUp;
    }

    @Override
    public int getMixedBrightnessForBlock(IBlockAccess world, int x, int y, int z) {

        int lightThis = world.getLightBrightnessForSkyBlocks(x, y, z, 0);
        int lightUp = world.getLightBrightnessForSkyBlocks(x, y + 1, z, 0);
        int lightThisBase = lightThis & 255;
        int lightUpBase = lightUp & 255;
        int lightThisExt = lightThis >> 16 & 255;
        int lightUpExt = lightUp >> 16 & 255;

        return (lightThisBase > lightUpBase ? lightThisBase : lightUpBase) | (lightThisExt > lightUpExt ? lightThisExt : lightUpExt) << 16;
    }

    @Override
    public int getRenderBlockPass() {

        return renderPass;
    }

    @Override
    public boolean shouldSideBeRendered(IBlockAccess world, int x, int y, int z, int side) {

        if (world.getBlockId(x, y, z) != blockID) {
            return !world.isBlockOpaqueCube(x, y, z);
        }
        Material mat = world.getBlockMaterial(x, y, z);
        return mat == this.blockMaterial ? false : super.shouldSideBeRendered(world, x, y, z, side);
    }

    /* FLUID FUNCTIONS */
    public static final int getDensity(IBlockAccess world, int x, int y, int z) {

        Block block = Block.blocksList[world.getBlockId(x, y, z)];

        if (!(block instanceof BlockFluidBase)) {
            return Integer.MAX_VALUE;
        }
        return ((BlockFluidBase) block).density;
    }

    public static double getFlowDirection(IBlockAccess world, int x, int y, int z) {

        Block block = Block.blocksList[world.getBlockId(x, y, z)];

        if (!(Block.blocksList[world.getBlockId(x, y, z)] instanceof BlockFluidBase)) {
            return -1000.0;
        }
        Vec3 vec = ((BlockFluidBase) block).getFlowVector(world, x, y, z);
        return vec.xCoord == 0.0D && vec.zCoord == 0.0D ? -1000.0D : Math.atan2(vec.zCoord, vec.xCoord) - Math.PI / 2D;
    }

    public final int getQuantaValueBelow(IBlockAccess world, int x, int y, int z, int belowThis) {

        int quantaRemaining = getQuantaValue(world, x, y, z);

        if (quantaRemaining >= belowThis) {
            return -1;
        }
        return quantaRemaining;
    }

    public final int getQuantaValueAbove(IBlockAccess world, int x, int y, int z, int aboveThis) {

        int quantaRemaining = getQuantaValue(world, x, y, z);

        if (quantaRemaining <= aboveThis) {
            return -1;
        }
        return quantaRemaining;
    }

    public final float getQuantaPercentage(IBlockAccess world, int x, int y, int z) {

        int quantaRemaining = getQuantaValue(world, x, y, z);
        return quantaRemaining / quantaPerBlockFloat;
    }

    public Vec3 getFlowVector(IBlockAccess world, int x, int y, int z) {

        Vec3 vec = world.getWorldVec3Pool().getVecFromPool(0.0D, 0.0D, 0.0D);
        int decay = quantaPerBlock - getQuantaValue(world, x, y, z);

        for (int side = 0; side < 4; ++side) {
            int x2 = x;
            int z2 = z;

            switch (side) {
            case 0:
                --x2;
                break;
            case 1:
                --z2;
                break;
            case 2:
                ++x2;
                break;
            case 3:
                ++z2;
                break;
            }

            int otherDecay = quantaPerBlock - getQuantaValue(world, x2, y, z2);

            if (otherDecay >= quantaPerBlock) {
                if (!world.getBlockMaterial(x2, y, z2).blocksMovement()) {
                    otherDecay = quantaPerBlock - getQuantaValue(world, x2, y - 1, z2);

                    if (otherDecay >= 0) {
                        int power = otherDecay - (decay - quantaPerBlock);
                        vec = vec.addVector((x2 - x) * power, (y - y) * power, (z2 - z) * power);
                    }
                }
            } else if (otherDecay >= 0) {
                int power = otherDecay - decay;
                vec = vec.addVector((x2 - x) * power, (y - y) * power, (z2 - z) * power);
            }
        }
        if (world.getBlockId(x, y + 1, z) == blockID) {
            boolean flag = false;

            if (this.isBlockSolid(world, x, y, z - 1, 2)) {
                flag = true;
            } else if (this.isBlockSolid(world, x, y, z + 1, 3)) {
                flag = true;
            } else if (this.isBlockSolid(world, x - 1, y, z, 4)) {
                flag = true;
            } else if (this.isBlockSolid(world, x + 1, y, z, 5)) {
                flag = true;
            } else if (this.isBlockSolid(world, x, y + 1, z - 1, 2)) {
                flag = true;
            } else if (this.isBlockSolid(world, x, y + 1, z + 1, 3)) {
                flag = true;
            } else if (this.isBlockSolid(world, x - 1, y + 1, z, 4)) {
                flag = true;
            } else if (this.isBlockSolid(world, x + 1, y + 1, z, 5)) {
                flag = true;
            }
            if (flag) {
                vec = vec.normalize().addVector(0.0D, -6.0D, 0.0D);
            }
        }
        vec = vec.normalize();
        return vec;
    }

    /* IFluidBlock */
    @Override
    public Fluid getFluid() {

        return FluidRegistry.getFluid(fluidName);
    }

}
