BetterEnd/src/main/java/ru/betterend/world/features/EndLakeFeature.java

package ru.betterend.world.features;

import java.util.Random;

import net.minecraft.block.BlockState;
import net.minecraft.block.Blocks;
import net.minecraft.util.math.BlockPos;
import net.minecraft.util.math.BlockPos.Mutable;
import net.minecraft.world.StructureWorldAccess;
import net.minecraft.world.gen.chunk.ChunkGenerator;
import net.minecraft.world.gen.feature.DefaultFeatureConfig;
import ru.betterend.noise.OpenSimplexNoise;
import ru.betterend.registry.BlockRegistry;
import ru.betterend.util.BlocksHelper;
import ru.betterend.util.MHelper;

public class EndLakeFeature extends DefaultFeature {
	private static final OpenSimplexNoise NOISE = new OpenSimplexNoise(15152);
	private static final Mutable POS = new Mutable();
	
	@Override
	public boolean generate(StructureWorldAccess world, ChunkGenerator chunkGenerator, Random random, BlockPos blockPos, DefaultFeatureConfig featureConfig) {
		double radius = MHelper.randRange(10.0, 20.0, random);
		double depth = radius * 0.5 * MHelper.randRange(0.8, 1.2, random);
		int dist = MHelper.floor(radius);
		int dist2 = MHelper.floor(radius * 1.5);
		int bott = MHelper.floor(depth);
		blockPos = getTopPos(world, blockPos);
		if (blockPos.getY() < 10) return false;
		
		int waterLevel = blockPos.getY();
		
		BlockPos pos = getTopPos(world, blockPos.north(dist));
		if (pos.getY() < 10) return false;
		waterLevel = MHelper.min(pos.getY(), waterLevel);
		
		pos = getTopPos(world, blockPos.south(dist));
		if (pos.getY() < 10) return false;
		waterLevel = MHelper.min(pos.getY(), waterLevel);
		
		pos = getTopPos(world, blockPos.east(dist));
		if (pos.getY() < 10) return false;
		waterLevel = MHelper.min(pos.getY(), waterLevel);
		
		pos = getTopPos(world, blockPos.west(dist));
		if (pos.getY() < 10) return false;
		waterLevel = MHelper.min(pos.getY(), waterLevel);
		
		int minX = MHelper.floor((blockPos.getX() - dist - 16) >> 4) << 4;
		int minZ = MHelper.floor((blockPos.getZ() - dist - 16) >> 4) << 4;
		int maxX = MHelper.floor((blockPos.getX() + dist + 16) >> 4) << 4;
		int maxZ = MHelper.floor((blockPos.getZ() + dist + 16) >> 4) << 4;
		
		for (int y = blockPos.getY(); y <= blockPos.getY() + 10; y++) {
			POS.setY(y);
			for (int x = minX; x <= maxX; x++) {
				POS.setX(x);
				for (int z = minZ; z <= maxZ; z++) {
					POS.setZ(z);
					if (!world.getFluidState(POS).isEmpty())
						BlocksHelper.setWithoutUpdate(world, POS, AIR);
				}
			}
		}
		
		for (int y = blockPos.getY(); y <= blockPos.getY() + 10; y++) {
			POS.setY(y);
			int add = y - blockPos.getY();
			for (int x = blockPos.getX() - dist2; x <= blockPos.getX() + dist2; x++) {
				POS.setX(x);
				int x2 = x - blockPos.getX();
				x2 *= x2;
				for (int z = blockPos.getZ() - dist2; z <= blockPos.getZ() + dist2; z++) {
					POS.setZ(z);
					int z2 = z - blockPos.getZ();
					z2 *= z2;
					double r = add * 1.5 + radius  * (NOISE.eval(x * 0.2, y * 0.2, z * 0.2) * 0.25 + 0.75);
					r *= r;
					if (x2 + z2 <= r) {
						BlocksHelper.setWithoutUpdate(world, POS, AIR);
						pos = POS.down();
						if (world.getBlockState(pos).getBlock() == Blocks.END_STONE)
						{
							BlockState state = world.getBiome(POS).getGenerationSettings().getSurfaceConfig().getTopMaterial();
							if (y > waterLevel + 1)
								BlocksHelper.setWithoutUpdate(world, POS.down(), state);
							else if (y > waterLevel)
								BlocksHelper.setWithoutUpdate(world, POS.down(), random.nextBoolean() ? state : BlockRegistry.ENDSTONE_DUST.getDefaultState());
							else
								BlocksHelper.setWithoutUpdate(world, POS.down(), BlockRegistry.ENDSTONE_DUST.getDefaultState());
						}
						pos = POS.up();
						if (!world.getBlockState(pos).isAir()) {
							while (!world.getBlockState(pos).isAir()) {
								BlocksHelper.setWithoutUpdate(world, pos, AIR);
								pos = pos.up();
							}
						}
					}
				}
			}
		}
		
		double aspect = ((double) radius / (double) depth);
		for (int y = blockPos.getY() - bott; y < blockPos.getY(); y++) {
			POS.setY(y);
			double y2 = (double) (y - blockPos.getY()) * aspect;
			y2 *= y2;
			for (int x = blockPos.getX() - dist; x <= blockPos.getX() + dist; x++) {
				POS.setX(x);
				int x2 = x - blockPos.getX();
				x2 *= x2;
				for (int z = blockPos.getZ() - dist; z <= blockPos.getZ() + dist; z++) {
					POS.setZ(z);
					int z2 = z - blockPos.getZ();
					z2 *= z2;
					double r = radius * (NOISE.eval(x * 0.2, y * 0.2, z * 0.2) * 0.25 + 0.75);
					r *= r;
					if (y2 + x2 + z2 <= r) {
						BlocksHelper.setWithoutUpdate(world, POS, y < waterLevel ? WATER : AIR);
						pos = POS.down();
						if (world.getBlockState(pos).getBlock() == Blocks.END_STONE)
							BlocksHelper.setWithoutUpdate(world, POS.down(), BlockRegistry.ENDSTONE_DUST.getDefaultState());
						pos = POS.up();
						if (!world.getBlockState(pos).isAir()) {
							while (!world.getBlockState(pos).isAir()) {
								BlocksHelper.setWithoutUpdate(world, pos, y < waterLevel ? WATER : AIR);
								pos = pos.up();
							}
						}
					}
				}
			}
		}
		
		return true;
	}
}