package nc.render.tile;

import javax.vecmath.Vector3f;

import nc.multiblock.turbine.*;
import nc.multiblock.turbine.Turbine.PlaneDir;
import nc.multiblock.turbine.TurbineRotorBladeUtil.TurbinePartDir;
import nc.multiblock.turbine.tile.TileTurbineController;
import nc.util.NCMath;
import net.minecraft.block.state.IBlockState;
import net.minecraft.client.Minecraft;
import net.minecraft.client.renderer.*;
import net.minecraft.client.renderer.texture.TextureMap;
import net.minecraft.client.renderer.tileentity.TileEntitySpecialRenderer;
import net.minecraft.util.EnumFacing;
import net.minecraft.util.EnumFacing.*;
import net.minecraft.util.math.BlockPos;
import net.minecraftforge.fml.relauncher.*;

@SideOnly(Side.CLIENT)
public class RenderTurbineRotor extends TileEntitySpecialRenderer<TileTurbineController> {

    private static final Minecraft MC = Minecraft.getMinecraft();

    private final float[] brightness = new float[] {1F, 1F, 1F, 1F, 1F, 1F, 1F, 1F};
    private byte count = 0;

    @Override
    public boolean isGlobalRenderer(TileTurbineController tile) {
        // return tile.isRenderer && tile.isMultiblockAssembled();
        return false;
    }

    @Override
    public void render(TileTurbineController controller, double posX, double posY, double posZ, float partialTicks, int destroyStage, float alpha) {
        if (!controller.isRenderer || !controller.isMultiblockAssembled()) {
            return;
        }

        Turbine turbine = controller.getMultiblock();
        if (turbine == null) return;

        EnumFacing dir = turbine.flowDir;
        if (dir == null) return;

        BlockRendererDispatcher renderer = MC.getBlockRendererDispatcher();

        // brightness = tile.getWorld().getLightBrightness(turbine.getMinimumInteriorPlaneCoord(dir, turbine.getFlowLength()/2, bladeLength - 1,bladeLength - 1));
        brightness[count] = controller.getWorld().getLightBrightness(turbine.getExtremeInteriorCoord(NCMath.getBit(count, 0) == 1, NCMath.getBit(count, 1) == 1, NCMath.getBit(count, 2) == 1));
        count++;
        count %= 8;
        float bright = (brightness[0] + brightness[1] + brightness[2] + brightness[3] + brightness[4] + brightness[5] + brightness[6] + brightness[7]) / 8F;

        MC.renderEngine.bindTexture(TextureMap.LOCATION_BLOCKS_TEXTURE);
        GlStateManager.color(1F, 1F, 1F, 1F);
        OpenGlHelper.setLightmapTextureCoords(OpenGlHelper.lightmapTexUnit, 0, 15 * 16);

        BlockPos pos = controller.getPos();
        double r = turbine.getRotorRadius();
        double rX = -turbine.getMaxX() + pos.getX() + (dir.getAxis() == Axis.X ? 0D : r);
        double rY = -turbine.getMaxY() + pos.getY() + (dir.getAxis() == Axis.Y ? 0D : r);
        double rZ = -turbine.getMaxZ() + pos.getZ() + (dir.getAxis() == Axis.Z ? 0D : r);

        int flowLength = turbine.getFlowLength(), bladeLength = turbine.bladeLength, shaftWidth = turbine.shaftWidth;
        double scale = r / Math.sqrt(r * r + NCMath.sq(shaftWidth) / 4D);

        // Enter rotated frame
        GlStateManager.pushMatrix();
        GlStateManager.translate(posX - rX, posY - rY, posZ - rZ);
        GlStateManager.scale(dir.getAxis() == Axis.X ? 1D : scale, dir.getAxis() == Axis.Y ? 1D : scale, dir.getAxis() == Axis.Z ? 1D : scale);
        {
            long systemTime = Minecraft.getSystemTime();
            if (!MC.isGamePaused()) turbine.rotorAngle = (turbine.rotorAngle + (systemTime - turbine.prevRenderTime) * turbine.angVel) % 360F;
            turbine.prevRenderTime = systemTime;
            GlStateManager.rotate(turbine.rotorAngle, dir.getAxis() == Axis.X ? 1F : 0F, dir.getAxis() == Axis.Y ? 1F : 0F, dir.getAxis() == Axis.Z ? 1F : 0F);
        }
        GlStateManager.translate(-pos.getX() + rX, -pos.getY() + rY, -pos.getZ() + rZ);

        if (turbine.rotorStateArray == null) return;
        IBlockState shaftState = turbine.rotorStateArray[turbine.bladePosArray.length];

        double bladeWidth = NCMath.SQRT2;

        for (int depth : turbine.bladeDepths) {
            renderRotor(turbine, renderer, bright, shaftState, dir, flowLength, bladeLength, shaftWidth, bladeWidth, depth);
        }

        // Leave rotated frame
        GlStateManager.popMatrix();

        // Enter stationary frame
        GlStateManager.pushMatrix();
        pos = controller.getPos();
        GlStateManager.translate(posX - rX, posY - rY, posZ - rZ);
        GlStateManager.scale(dir.getAxis() == Axis.X ? 1D : scale, dir.getAxis() == Axis.Y ? 1D : scale, dir.getAxis() == Axis.Z ? 1D : scale);
        GlStateManager.translate(-pos.getX() + rX, -pos.getY() + rY, -pos.getZ() + rZ);

        for (int depth : turbine.statorDepths) {
            renderRotor(turbine, renderer, bright, shaftState, dir, flowLength, bladeLength, shaftWidth, bladeWidth, depth);
        }

        GlStateManager.popMatrix();
    }

    public void renderRotor(Turbine turbine, BlockRendererDispatcher renderer, float bright, IBlockState shaftState, EnumFacing flowDir, int flowLength, int bladeLength, int shaftWidth, double bladeWidth, int depth) {
        double depthScale = (0.0D + Math.pow(4D, (double) (1 + depth - flowLength) / (double) flowLength)) / 1.0D;

        GlStateManager.pushMatrix();

        Vector3f renderPos = turbine.renderPosArray[4 * flowLength * shaftWidth + depth];
        GlStateManager.translate(renderPos.x + 0.5D, renderPos.y + 0.5D, renderPos.z + 0.5D);
        GlStateManager.scale(flowDir.getAxis() == Axis.X ? 1D : depthScale, flowDir.getAxis() == Axis.Y ? 1D : depthScale, flowDir.getAxis() == Axis.Z ? 1D : depthScale);
        GlStateManager.translate(-renderPos.x - 0.5D, -renderPos.y - 0.5D, -renderPos.z - 0.5D);

        renderShaft(turbine, renderer, bright, shaftState, flowDir, flowLength, shaftWidth, depth);
        for (int j : new int[] { 0, flowLength, 2 * flowLength, 3 * flowLength }) {
            renderBlades(turbine, renderer, bright, flowDir, flowLength, bladeLength, shaftWidth, bladeWidth, j, depth);
        }

        GlStateManager.popMatrix();
    }

    public void renderShaft(Turbine turbine, BlockRendererDispatcher renderer, float bright, IBlockState shaftState, EnumFacing flowDir, int flowLength, int shaftWidth, int depth) {
        GlStateManager.pushMatrix();

        Vector3f renderPos = turbine.renderPosArray[4 * flowLength * shaftWidth + depth];
        GlStateManager.translate(renderPos.x + 0.5D, renderPos.y + 0.5D, renderPos.z + 0.5D);
        GlStateManager.scale(flowDir.getAxis() == Axis.X ? 1D : shaftWidth, flowDir.getAxis() == Axis.Y ? 1D : shaftWidth, flowDir.getAxis() == Axis.Z ? 1D : shaftWidth);

        GlStateManager.translate(-0.5D, -0.5D, -0.5D);
        GlStateManager.rotate(-90F, 0F, 1F, 0F);

        renderer.renderBlockBrightness(shaftState, bright);

        GlStateManager.popMatrix();
    }

    public void renderBlades(Turbine turbine, BlockRendererDispatcher renderer, float bright, EnumFacing flowDir, int flowLength, int bladeLength, int shaftWidth, double bladeWidth, int jMult, int depth) {
        Vector3f renderPos;
        IBlockState rotorState;
        TurbinePartDir bladeDir;
        PlaneDir planeDir;

        int i = jMult + depth;

        for (int w = 0; w < shaftWidth; w++) {
            GlStateManager.pushMatrix();

            renderPos = turbine.renderPosArray[w + i * shaftWidth];
            rotorState = turbine.rotorStateArray[i];
            bladeDir = rotorState.getValue(TurbineRotorBladeUtil.DIR);
            planeDir = (i < flowLength || i >= 3 * flowLength) ? PlaneDir.V : PlaneDir.U;

            GlStateManager.translate(renderPos.x + 0.5D, renderPos.y + 0.5D, renderPos.z + 0.5D);
            GlStateManager.scale(flowDir.getAxis() == Axis.X ? 1D : (turbine.getBladeDir(planeDir) == TurbinePartDir.X ? bladeLength : bladeWidth), flowDir.getAxis() == Axis.Y ? 1D : (turbine.getBladeDir(planeDir) == TurbinePartDir.Y ? bladeLength : bladeWidth), flowDir.getAxis() == Axis.Z ? 1D : (turbine.getBladeDir(planeDir) == TurbinePartDir.Z ? bladeLength : bladeWidth));
            GlStateManager.rotate(turbine.bladeAngleArray[i] * (flowDir.getAxisDirection() == AxisDirection.POSITIVE ^ flowDir.getAxis() == Axis.X ? 1F : -1F), bladeDir == TurbinePartDir.X ? 1F : 0F, bladeDir == TurbinePartDir.Y ? 1F : 0F, bladeDir == TurbinePartDir.Z ? 1F : 0F);

            GlStateManager.translate(-0.5D, -0.5D, -0.5D);
            GlStateManager.rotate(-90F, 0F, 1F, 0F);

            renderer.renderBlockBrightness(rotorState, bright);

            GlStateManager.popMatrix();
        }
    }
}