ClassiCube/ClassicalSharp/Map/ChunkMeshBuilder.FloodFill.cs

using System;
using ClassicalSharp.GraphicsAPI;

namespace ClassicalSharp {
	
	public partial class ChunkMeshBuilder {

		unsafe int ComputeOcclusion() {
			int* didFlags = stackalloc int[chunkSize2];
			OpenTK.MemUtils.memset( (IntPtr)didFlags, 0x00, 0, chunkSize2 * sizeof( int ) );
			int* stack = stackalloc int[chunkSize3 * 4];
			
			int i = 0;
			bool solidX = true, solidY = true, solidZ = true;
			for( int y = 0; y < 16; y++ ) {
				for( int z = 0; z < 16; z++ ) {
					int flagIndex = (y << 4) | z;
					int chunkIndex = (y + 1) * extChunkSize2 + (z + 1) * extChunkSize + (0 + 1);
					for( int x = 0; x < 16; x++ ) {
						byte block = chunk[chunkIndex];
						if( info.IsOpaque[block] ) {
							didFlags[flagIndex] |= (1 << x);
						} else {
							FloodFill( didFlags, stack, i, ref solidX, ref solidY, ref solidZ );
						}
						
						i++;
						chunkIndex++;
					}
				}
			}
			return (solidX ? 0x1 : 0) | (solidZ ? 0x2 : 0) | (solidY ? 0x4 : 0);
		}
		
		unsafe void FloodFill( int* didFlags, int* stack, int startIndex,
		                      ref bool solidX, ref bool solidY, ref bool solidZ ) {
			int index = 0;
			stack[index++] = startIndex;
			bool tX0 = false, tX1 = false, tY0 = false, 
			tY1 = false, tZ0 = false, tZ1 = false;
			
			while( index > 0 ) {
				int bIndex = stack[--index];
				int x = (bIndex & 0x0F);
				int z = ((bIndex >> 4) & 0x0F );
				int y = ((bIndex >> 8) & 0x0F);
				int flagIndex = (y << 4) | z;
				didFlags[flagIndex] |= (1 << x);
				
				int chunkIndex = (y + 1) * extChunkSize2 + (z + 1) * extChunkSize + (x + 1);
				byte block = chunk[chunkIndex];
				if( !info.IsOpaque[block] ) {
					if( x == 0 )
						tX0 = true;
					else if( (didFlags[flagIndex] & (1 << (x - 1))) == 0 )
						stack[index++] = (y << 8) | (z << 4) | (x - 1);
					
					if( x == 15 )
						tX1 = true;
					else if( (didFlags[flagIndex] & (1 << (x + 1))) == 0 )
						stack[index++] = (y << 8) | (z << 4) | (x + 1);
					
					if( z == 0 )
						tZ0 = true;
					else if( (didFlags[flagIndex - 1] & (1 << x)) == 0 )
						stack[index++] = (y << 8) | ((z - 1) << 4) | x;
					
					if( z == 15 )
						tZ1 = true;
					else if( (didFlags[flagIndex + 1] & (1 << x)) == 0 )
						stack[index++] = (y << 8) | ((z + 1) << 4) | x;
					
					if( y == 0 ) 
						tY0 = true;
					else if( (didFlags[flagIndex - 16] & (1 << x)) == 0 )
						stack[index++] = ((y - 1) << 8) | (z << 4) | x;
					
					if( y == 15 ) 
						tY1 = true;
					else if( (didFlags[flagIndex + 16] & (1 << x)) == 0 )
						stack[index++] = ((y + 1) << 8) | (z << 4) | x;
				}
			}
			if( tX0 && tX1 ) solidX = false;
			if( tY0 && tY1 ) solidY = false;
			if( tZ0 && tZ1 ) solidZ = false;
		}
	}
}