ClassiCube/ClassicalSharp/TexturePack/WaterAnimation.cs

// Copyright 2014-2017 ClassicalSharp | Licensed under BSD-3
// Based off the incredible work from https://dl.dropboxusercontent.com/u/12694594/lava.txt
using System;
using ClassicalSharp.Generator;

namespace ClassicalSharp {
	public unsafe class WaterAnimation	{
		float[] flameHeat, potHeat, soupHeat;
		JavaRandom rnd = null;
		
		public unsafe void Tick(int* ptr, int size) {
			if (rnd == null)
				rnd = new JavaRandom(new Random().Next());
			int mask = size - 1, shift = CheckSize(size);
			
			int i = 0;
			for (int y = 0; y < size; y++)
				for (int x = 0; x < size; x++)
			{
				// Calculate the colour at this coordinate in the heatmap
				float lSoupHeat = 0;
				for (int j = 0; j < 3; j++) {
					int xx = x + (j - 1);
					lSoupHeat += soupHeat[y << shift | (xx & mask)];
				}								
				
				soupHeat[i] = lSoupHeat / 3.3f + potHeat[i] * 0.8f;
				potHeat[i] += flameHeat[i] * 0.05f;
				if (potHeat[i] < 0) potHeat[i] = 0;
				flameHeat[i] -= 0.1f;
				
				if (rnd.NextFloat() <= 0.05f)
					flameHeat[i] = 0.5f;
				
				// Output the pixel
				float col = soupHeat[i];
				col = col < 0 ? 0 : col;
				col = col > 1 ? 1 : col;
				col = col * col;
				
				float r = 32 + col * 32;
				float g = 50 + col * 64;
				float a = 146 + col * 50;
				
				*ptr = (byte)a << 24 | (byte)r << 16 | (byte)g << 8 | 255;
				
				ptr++; i++;
			}
		}
		
		int CheckSize(int size) {
			if (potHeat == null || potHeat.Length < size * size) {
				flameHeat = new float[size * size];
				potHeat = new float[size * size];
				soupHeat = new float[size * size];
			}
			
			int shift = 0;
			while (size > 1) { shift++; size >>= 1; }
			return shift;
		}
	}
}