package biomesoplenty.worldgen.tree;

import java.util.Random;

import net.minecraft.block.Block;
import net.minecraft.util.MathHelper;
import net.minecraft.world.World;
import net.minecraft.world.gen.feature.WorldGenerator;

public class WorldGenLargeTree extends WorldGenerator
{
	/**
	 * Contains three sets of two values that provide complimentary indices for a given 'major' index - 1 and 2 for 0, 0
	 * and 2 for 1, and 0 and 1 for 2.
	 */
	static final byte[] otherCoordPairs = new byte[] {(byte)2, (byte)0, (byte)0, (byte)1, (byte)2, (byte)1};

	/** random seed for GenBigTree */
	Random rand = new Random();

	/** Reference to the World object. */
	World worldObj;
	int[] basePos = new int[] {0, 0, 0};
	int heightLimit = 0;
	int height;
	double heightAttenuation = 0.45D;
	double branchDensity = 1.0D;
	double branchSlope = 0.0D;
	double scaleWidth = 2.0D;
	double leafDensity = 1.0D;

	/**
	 * Currently always 1, can be set to 2 in the class constructor to generate a double-sized tree trunk for big trees.
	 */
	int trunkSize = 2;

	/**
	 * Sets the limit of the random value used to initialize the height limit.
	 */
	int heightLimitLimit = 12;

	/**
	 * Sets the distance limit for how far away the generator will populate leaves from the base leaf node.
	 */
	int leafDistanceLimit = 4;

	/** Contains a list of a points at which to generate groups of leaves. */
	int[][] leafNodes;

	public WorldGenLargeTree(boolean par1)
	{
		super(par1);
	}

	/**
	 * Generates a list of leaf nodes for the tree, to be populated by generateLeaves.
	 */
	void generateLeafNodeList()
	{
		height = (int)(heightLimit * heightAttenuation);

		if (height >= heightLimit)
		{
			height = heightLimit - 1;
		}

		int var1 = (int)(1.382D + Math.pow(leafDensity * heightLimit / 13.0D, 2.0D));

		if (var1 < 1)
		{
			var1 = 1;
		}

		int[][] var2 = new int[var1 * heightLimit][4];
		int var3 = basePos[1] + heightLimit - leafDistanceLimit;
		int var4 = 1;
		int var5 = basePos[1] + height;
		int var6 = var3 - basePos[1];
		var2[0][0] = basePos[0];
		var2[0][1] = var3;
		var2[0][2] = basePos[2];
		var2[0][3] = var5;
		--var3;

		while (var6 >= 0)
		{
			int var7 = 0;
			float var8 = this.layerSize(var6);

			if (var8 < 0.0F)
			{
				--var3;
				--var6;
			}
			else
			{
				for (double var9 = 0.5D; var7 < var1; ++var7)
				{
					double var11 = scaleWidth * var8 * (rand.nextFloat() + 0.328D);
					double var13 = rand.nextFloat() * 2.0D * Math.PI;
					int var15 = MathHelper.floor_double(var11 * Math.sin(var13) + basePos[0] + var9);
					int var16 = MathHelper.floor_double(var11 * Math.cos(var13) + basePos[2] + var9);
					int[] var17 = new int[] {var15, var3, var16};
					int[] var18 = new int[] {var15, var3 + leafDistanceLimit, var16};

					if (this.checkBlockLine(var17, var18) == -1)
					{
						int[] var19 = new int[] {basePos[0], basePos[1], basePos[2]};
						double var20 = Math.sqrt(Math.pow(Math.abs(basePos[0] - var17[0]), 2.0D) + Math.pow(Math.abs(basePos[2] - var17[2]), 2.0D));
						double var22 = var20 * branchSlope;

						if (var17[1] - var22 > var5)
						{
							var19[1] = var5;
						}
						else
						{
							var19[1] = (int)(var17[1] - var22);
						}

						if (this.checkBlockLine(var19, var17) == -1)
						{
							var2[var4][0] = var15;
							var2[var4][1] = var3;
							var2[var4][2] = var16;
							var2[var4][3] = var19[1];
							++var4;
						}
					}
				}

				--var3;
				--var6;
			}
		}

		leafNodes = new int[var4][4];
		System.arraycopy(var2, 0, leafNodes, 0, var4);
	}

	void genTreeLayer(int par1, int par2, int par3, float par4, byte par5, int par6)
	{
		int var7 = (int)(par4 + 0.618D);
		byte var8 = otherCoordPairs[par5];
		byte var9 = otherCoordPairs[par5 + 3];
		int[] var10 = new int[] {par1, par2, par3};
		int[] var11 = new int[] {0, 0, 0};
		int var12 = -var7;
		int var13 = -var7;

		for (var11[par5] = var10[par5]; var12 <= var7; ++var12)
		{
			var11[var8] = var10[var8] + var12;
			var13 = -var7;

			while (var13 <= var7)
			{
				double var15 = Math.pow(Math.abs(var12) + 0.5D, 2.0D) + Math.pow(Math.abs(var13) + 0.5D, 2.0D);

				if (var15 > par4 * par4)
				{
					++var13;
				}
				else
				{
					var11[var9] = var10[var9] + var13;
					int var14 = worldObj.getBlockId(var11[0], var11[1], var11[2]);

					if (var14 != 0 && var14 != Block.leaves.blockID)
					{
						++var13;
					}
					else
					{
						this.setBlockAndMetadata(worldObj, var11[0], var11[1], var11[2], par6, 0);
						++var13;
					}
				}
			}
		}
	}

	/**
	 * Gets the rough size of a layer of the tree.
	 */
	 float layerSize(int par1)
	{
		if (par1 < (heightLimit) * 0.3D)
			return -1.618F;
		else
		{
			float var2 = heightLimit / 2.0F;
			float var3 = heightLimit / 2.0F - par1;
			float var4;

			if (var3 == 0.0F)
			{
				var4 = var2;
			}
			else if (Math.abs(var3) >= var2)
			{
				var4 = 0.0F;
			}
			else
			{
				var4 = (float)Math.sqrt(Math.pow(Math.abs(var2), 2.0D) - Math.pow(Math.abs(var3), 2.0D));
			}

			var4 *= 0.5F;
			return var4;
		}
	}

	 float leafSize(int par1)
	 {
		 return par1 >= 0 && par1 < leafDistanceLimit ? (par1 != 0 && par1 != leafDistanceLimit - 1 ? 3.0F : 2.0F) : -1.0F;
	 }

	 /**
	  * Generates the leaves surrounding an individual entry in the leafNodes list.
	  */
	 void generateLeafNode(int par1, int par2, int par3)
	 {
		 int var4 = par2;

		 for (int var5 = par2 + leafDistanceLimit; var4 < var5; ++var4)
		 {
			 float var6 = this.leafSize(var4 - par2);
			 this.genTreeLayer(par1, var4, par3, var6, (byte)1, Block.leaves.blockID);
		 }
	 }

	 /**
	  * Places a line of the specified block ID into the world from the first coordinate triplet to the second.
	  */
	 void placeBlockLine(int[] par1ArrayOfInteger, int[] par2ArrayOfInteger, int par3)
	 {
		 int[] var4 = new int[] {0, 0, 0};
		 byte var5 = 0;
		 byte var6;

		 for (var6 = 0; var5 < 3; ++var5)
		 {
			 var4[var5] = par2ArrayOfInteger[var5] - par1ArrayOfInteger[var5];

			 if (Math.abs(var4[var5]) > Math.abs(var4[var6]))
			 {
				 var6 = var5;
			 }
		 }

		 if (var4[var6] != 0)
		 {
			 byte var7 = otherCoordPairs[var6];
			 byte var8 = otherCoordPairs[var6 + 3];
			 byte var9;

			 if (var4[var6] > 0)
			 {
				 var9 = 1;
			 }
			 else
			 {
				 var9 = -1;
			 }

			 double var10 = (double)var4[var7] / (double)var4[var6];
			 double var12 = (double)var4[var8] / (double)var4[var6];
			 int[] var14 = new int[] {0, 0, 0};
			 int var15 = 0;

			 for (int var16 = var4[var6] + var9; var15 != var16; var15 += var9)
			 {
				 var14[var6] = MathHelper.floor_double(par1ArrayOfInteger[var6] + var15 + 0.5D);
				 var14[var7] = MathHelper.floor_double(par1ArrayOfInteger[var7] + var15 * var10 + 0.5D);
				 var14[var8] = MathHelper.floor_double(par1ArrayOfInteger[var8] + var15 * var12 + 0.5D);
				 byte var17 = 0;
				 int var18 = Math.abs(var14[0] - par1ArrayOfInteger[0]);
				 int var19 = Math.abs(var14[2] - par1ArrayOfInteger[2]);
				 int var20 = Math.max(var18, var19);

				 if (var20 > 0)
				 {
					 if (var18 == var20)
					 {
						 var17 = 4;
					 }
					 else if (var19 == var20)
					 {
						 var17 = 8;
					 }
				 }

				 this.setBlockAndMetadata(worldObj, var14[0], var14[1], var14[2], par3, var17);
			 }
		 }
	 }

	 /**
	  * Generates the leaf portion of the tree as specified by the leafNodes list.
	  */
	 void generateLeaves()
	 {
		 int var1 = 0;

		 for (int var2 = leafNodes.length; var1 < var2; ++var1)
		 {
			 int var3 = leafNodes[var1][0];
			 int var4 = leafNodes[var1][1];
			 int var5 = leafNodes[var1][2];
			 this.generateLeafNode(var3, var4, var5);
		 }
	 }

	 /**
	  * Indicates whether or not a leaf node requires additional wood to be added to preserve integrity.
	  */
	 boolean leafNodeNeedsBase(int par1)
	 {
		 return par1 >= heightLimit * 0.2D;
	 }

	 /**
	  * Places the trunk for the big tree that is being generated. Able to generate double-sized trunks by changing a
	  * field that is always 1 to 2.
	  */
	 void generateTrunk()
	 {
		 int var1 = basePos[0];
		 int var2 = basePos[1];
		 int var3 = basePos[1] + height;
		 int var4 = basePos[2];
		 int[] var5 = new int[] {var1, var2, var4};
		 int[] var6 = new int[] {var1, var3, var4};
		 this.placeBlockLine(var5, var6, Block.wood.blockID);

		 if (trunkSize == 2)
		 {
			 ++var5[0];
			 ++var6[0];
			 this.placeBlockLine(var5, var6, Block.wood.blockID);
			 ++var5[2];
			 ++var6[2];
			 this.placeBlockLine(var5, var6, Block.wood.blockID);
			 var5[0] += -1;
			 var6[0] += -1;
			 this.placeBlockLine(var5, var6, Block.wood.blockID);
		 }
	 }

	 /**
	  * Generates additional wood blocks to fill out the bases of different leaf nodes that would otherwise degrade.
	  */
	 void generateLeafNodeBases()
	 {
		 int var1 = 0;
		 int var2 = leafNodes.length;

		 for (int[] var3 = new int[] {basePos[0], basePos[1], basePos[2]}; var1 < var2; ++var1)
		 {
			 int[] var4 = leafNodes[var1];
			 int[] var5 = new int[] {var4[0], var4[1], var4[2]};
			 var3[1] = var4[3];
			 int var6 = var3[1] - basePos[1];

			 if (this.leafNodeNeedsBase(var6))
			 {
				 this.placeBlockLine(var3, var5, Block.wood.blockID);
			 }
		 }
	 }

	 /**
	  * Checks a line of blocks in the world from the first coordinate to triplet to the second, returning the distance
	  * (in blocks) before a non-air, non-leaf block is encountered and/or the end is encountered.
	  */
	 int checkBlockLine(int[] par1ArrayOfInteger, int[] par2ArrayOfInteger)
	 {
		 int[] var3 = new int[] {0, 0, 0};
		 byte var4 = 0;
		 byte var5;

		 for (var5 = 0; var4 < 3; ++var4)
		 {
			 var3[var4] = par2ArrayOfInteger[var4] - par1ArrayOfInteger[var4];

			 if (Math.abs(var3[var4]) > Math.abs(var3[var5]))
			 {
				 var5 = var4;
			 }
		 }

		 if (var3[var5] == 0)
			 return -1;
		 else
		 {
			 byte var6 = otherCoordPairs[var5];
			 byte var7 = otherCoordPairs[var5 + 3];
			 byte var8;

			 if (var3[var5] > 0)
			 {
				 var8 = 1;
			 }
			 else
			 {
				 var8 = -1;
			 }

			 double var9 = (double)var3[var6] / (double)var3[var5];
			 double var11 = (double)var3[var7] / (double)var3[var5];
			 int[] var13 = new int[] {0, 0, 0};
			 int var14 = 0;
			 int var15;

			 for (var15 = var3[var5] + var8; var14 != var15; var14 += var8)
			 {
				 var13[var5] = par1ArrayOfInteger[var5] + var14;
				 var13[var6] = MathHelper.floor_double(par1ArrayOfInteger[var6] + var14 * var9);
				 var13[var7] = MathHelper.floor_double(par1ArrayOfInteger[var7] + var14 * var11);
				 int var16 = worldObj.getBlockId(var13[0], var13[1], var13[2]);

				 if (var16 != 0 && var16 != Block.leaves.blockID)
				 {
					 break;
				 }
			 }

			 return var14 == var15 ? -1 : Math.abs(var14);
		 }
	 }

	 /**
	  * Returns a boolean indicating whether or not the current location for the tree, spanning basePos to to the height
	  * limit, is valid.
	  */
	 boolean validTreeLocation()
	 {
		 int[] var1 = new int[] {basePos[0], basePos[1], basePos[2]};
		 int[] var2 = new int[] {basePos[0], basePos[1] + heightLimit - 1, basePos[2]};
		 int var3 = worldObj.getBlockId(basePos[0], basePos[1] - 1, basePos[2]);

		 if (var3 != 2 && var3 != 3)
			 return false;
		 else
		 {
			 int var4 = this.checkBlockLine(var1, var2);

			 if (var4 == -1)
				 return true;
			 else if (var4 < 6)
				 return false;
			 else
			 {
				 heightLimit = var4;
				 return true;
			 }
		 }
	 }

	 /**
	  * Rescales the generator settings, only used in WorldGenBigTree
	  */
	 @Override
	 public void setScale(double par1, double par3, double par5)
	 {
		 heightLimitLimit = (int)(par1 * 12.0D);

		 if (par1 > 0.5D)
		 {
			 leafDistanceLimit = 5;
		 }

		 scaleWidth = par3;
		 leafDensity = par5;
	 }

	 @Override
	 public boolean generate(World par1World, Random par2Random, int par3, int par4, int par5)
	 {
		 worldObj = par1World;
		 long var6 = par2Random.nextLong();
		 rand.setSeed(var6);
		 basePos[0] = par3;
		 basePos[1] = par4;
		 basePos[2] = par5;

		 if (heightLimit == 0)
		 {
			 heightLimit = 12 + par2Random.nextInt(10);
		 }

		 if (!this.validTreeLocation())
			 return false;
		 else
		 {
			 this.generateLeafNodeList();
			 this.generateLeaves();
			 this.generateTrunk();
			 this.generateLeafNodeBases();
			 return true;
		 }
	 }
}