Mekanism-tilera-Edition/common/mekanism/induction/client/render/RenderPartWire.java

package mekanism.induction.client.render;

import java.nio.FloatBuffer;
import java.util.Map;

import mekanism.common.util.MekanismUtils;
import mekanism.common.util.MekanismUtils.ResourceType;
import mekanism.induction.common.MekanismInduction;
import mekanism.induction.common.wire.PartConductor;
import mekanism.induction.common.wire.PartWire;
import net.minecraft.client.renderer.texture.IconRegister;
import net.minecraft.util.Icon;
import net.minecraft.util.ResourceLocation;
import net.minecraftforge.common.ForgeDirection;

import org.lwjgl.BufferUtils;
import org.lwjgl.opengl.GL11;

import universalelectricity.core.vector.Vector3;
import codechicken.lib.colour.Colour;
import codechicken.lib.colour.ColourRGBA;
import codechicken.lib.lighting.LightModel;
import codechicken.lib.render.CCModel;
import codechicken.lib.render.CCRenderState;
import codechicken.lib.render.ColourMultiplier;
import codechicken.lib.render.IconTransformation;
import codechicken.lib.render.TextureUtils;
import codechicken.lib.vec.Matrix4;
import codechicken.lib.vec.Rotation;
import codechicken.lib.vec.Transformation;
import codechicken.lib.vec.Translation;
import codechicken.lib.vec.VariableTransformation;
import cpw.mods.fml.relauncher.Side;
import cpw.mods.fml.relauncher.SideOnly;

/**
 * 
 * @author unpairedbracket
 * 
 */
@SideOnly(Side.CLIENT)
public class RenderPartWire
{
	private static final ResourceLocation WIRE_SHINE = MekanismUtils.getResource(ResourceType.RENDER, "white.png");
	public static final Map<String, CCModel> models;
	public static final Map<String, CCModel> shinyModels;
	public static Icon wireIcon;
	public static Icon insulationIcon;
	public static Icon breakIcon;
	public static FloatBuffer location = BufferUtils.createFloatBuffer(4);
	public static FloatBuffer specular = BufferUtils.createFloatBuffer(4);
	public static FloatBuffer zero = BufferUtils.createFloatBuffer(4);
	public static FloatBuffer defaultAmbient = BufferUtils.createFloatBuffer(4);
	public static final RenderPartWire INSTANCE = new RenderPartWire();

	static
	{
		models = CCModel.parseObjModels(MekanismUtils.getResource(ResourceType.MODEL, "wire.obj"), 7, new InvertX());
		
		for(CCModel c : models.values())
		{
			c.apply(new Translation(.5, 0, .5));
			c.computeLighting(LightModel.standardLightModel);
			c.shrinkUVs(0.0005);
		}

		shinyModels = CCModel.parseObjModels(MekanismUtils.getResource(ResourceType.MODEL, "wireShine.obj"), 7, new InvertX());
		
		for(CCModel c : shinyModels.values())
		{
			c.apply(new Translation(.5, 0, .5));
			c.computeLighting(LightModel.standardLightModel);
			c.shrinkUVs(0.0005);
		}

		loadBuffer(location, 0, 0, 0, 1);
		loadBuffer(specular, 1, 1, 1, 1);
		loadBuffer(zero, 0, 0, 0, 0);
		loadBuffer(defaultAmbient, 0.4F, 0.4F, 0.4F, 1);

		GL11.glLightModel(GL11.GL_LIGHT_MODEL_AMBIENT, zero);

		GL11.glLight(GL11.GL_LIGHT3, GL11.GL_SPECULAR, specular);

		GL11.glMaterial(GL11.GL_FRONT, GL11.GL_SPECULAR, specular);
		GL11.glMaterial(GL11.GL_FRONT, GL11.GL_AMBIENT, zero);
		GL11.glMaterial(GL11.GL_FRONT, GL11.GL_DIFFUSE, zero);
		GL11.glMaterialf(GL11.GL_FRONT, GL11.GL_SHININESS, 128F);
	}

	public static void loadBuffer(FloatBuffer buffer, float... src)
	{
		buffer.clear();
		buffer.put(src);
		buffer.flip();
	}

	public void renderShine(PartWire wire, double x, double y, double z, float f)
	{
		if(wire != null)
		{
			GL11.glPushMatrix();
			GL11.glEnable(GL11.GL_BLEND);
			GL11.glBlendFunc(GL11.GL_SRC_COLOR, GL11.GL_ONE_MINUS_SRC_COLOR);
			GL11.glEnable(GL11.GL_LIGHTING);
			GL11.glDisable(GL11.GL_LIGHT0);
			GL11.glDisable(GL11.GL_LIGHT1);
			GL11.glEnable(GL11.GL_LIGHT3);
			GL11.glLight(GL11.GL_LIGHT3, GL11.GL_POSITION, location);

			GL11.glTranslatef((float)x, (float)y, (float)z);
			GL11.glLightModel(GL11.GL_LIGHT_MODEL_AMBIENT, zero);

			CCRenderState.reset();
			CCRenderState.useNormals(true);
			CCRenderState.changeTexture(WIRE_SHINE);
			CCRenderState.startDrawing(7);
			renderSideShine(ForgeDirection.UNKNOWN, wire);
			
			byte renderSides = wire.getAllCurrentConnections();
			
			for(ForgeDirection side : ForgeDirection.VALID_DIRECTIONS)
			{
				if(PartConductor.connectionMapContainsSide(renderSides, side))
				{
					renderSideShine(side, wire);
				}
			}
			
			CCRenderState.draw();

			GL11.glDisable(GL11.GL_LIGHTING);
			GL11.glEnable(GL11.GL_LIGHT0);
			GL11.glEnable(GL11.GL_LIGHT1);
			GL11.glDisable(GL11.GL_LIGHT3);

			GL11.glLightModel(GL11.GL_LIGHT_MODEL_AMBIENT, defaultAmbient);

			GL11.glBlendFunc(GL11.GL_SRC_ALPHA, GL11.GL_ONE_MINUS_SRC_ALPHA);

			GL11.glPopMatrix();
		}
	}

	public static void registerIcons(IconRegister iconReg)
	{
		wireIcon = iconReg.registerIcon("mekanism:render/Wire");
		insulationIcon = iconReg.registerIcon("mekanism:render/" + "Insulation" + (MekanismInduction.LO_FI_INSULATION ? "Tiny" : ""));
		breakIcon = iconReg.registerIcon("mekanism:Wire");
	}

	public void renderStatic(PartWire wire)
	{
		TextureUtils.bindAtlas(0);
		CCRenderState.reset();
		CCRenderState.useModelColours(true);
		CCRenderState.setBrightness(wire.world(), wire.x(), wire.y(), wire.z());
		renderSide(ForgeDirection.UNKNOWN, wire);
		byte renderSides = wire.getAllCurrentConnections();
		
		for(ForgeDirection side : ForgeDirection.VALID_DIRECTIONS)
		{
			if(PartConductor.connectionMapContainsSide(renderSides, side))
			{
				renderSide(side, wire);
			}
		}
	}

	public void renderSide(ForgeDirection side, PartWire wire)
	{
		String name = side.name().toLowerCase();
		name = name.equals("unknown") ? "center" : name;
		Vector3 materialColour = wire.getMaterial().color;
		Colour colour = new ColourRGBA(materialColour.x, materialColour.y, materialColour.z, 1);
		renderPart(wireIcon, models.get(name), wire.x(), wire.y(), wire.z(), colour);
		
		if(wire.isInsulated())
		{
			Vector3 vecColour = MekanismInduction.DYE_COLORS[wire.dyeID];
			Colour insulationColour = new ColourRGBA(vecColour.x, vecColour.y, vecColour.z, 1);
			renderPart(insulationIcon, models.get(name + "Insulation"), wire.x(), wire.y(), wire.z(), insulationColour);
		}
	}

	public void renderSideShine(ForgeDirection side, PartWire wire)
	{
		String name = side.name().toLowerCase();
		name = name.equals("unknown") ? "center" : name;
		Vector3 materialColour = wire.getMaterial().color;
		renderPartShine(shinyModels.get(name));
	}

	public void renderPart(Icon icon, CCModel cc, double x, double y, double z, Colour colour)
	{
		cc.render(0, cc.verts.length, Rotation.sideOrientation(0, Rotation.rotationTo(0, 2)).at(codechicken.lib.vec.Vector3.center).with(new Translation(x, y, z)), new IconTransformation(icon), new ColourMultiplier(colour));
	}

	public void renderPartShine(CCModel cc)
	{
		cc.render(null, 0, 0);
	}

	public static class InvertX extends VariableTransformation
	{
		public InvertX()
		{
			super(new Matrix4(1, 0, 0, 0, 0, 0, 1, 0, 0, 1, 0, 0, 0, 0, 0, 1));
		}

		@Override
		public Transformation inverse()
		{
			return this;
		}

		@Override
		public void apply(codechicken.lib.vec.Vector3 vec)
		{
			vec.x = -vec.x;
		}
	}
}