#include "stdafx.h"
#include "net.minecraft.world.level.newbiome.layer.h"
#include "System.h"

ZoomLayer::ZoomLayer(__int64 seedMixup, shared_ptr<Layer>parent) : Layer(seedMixup)
{
	this->parent = parent;
}

intArray ZoomLayer::getArea(int xo, int yo, int w, int h)
{
	int px = xo >> 1;
	int py = yo >> 1;
	int pw = (w >> 1) + 3;
	int ph = (h >> 1) + 3;
	intArray p = parent->getArea(px, py, pw, ph);

	intArray tmp = IntCache::allocate((pw * 2) * (ph * 2));
	int ww = (pw << 1);
	for (int y = 0; y < ph - 1; y++)
	{
		int ry = y << 1;
		int pp = ry * ww;
		int ul = p[(0 + 0) + (y + 0) * pw];
		int dl = p[(0 + 0) + (y + 1) * pw];
		for (int x = 0; x < pw - 1; x++)
		{
			initRandom((x + px) << 1, (y + py) << 1);
			int ur = p[(x + 1) + (y + 0) * pw];
			int dr = p[(x + 1) + (y + 1) * pw];

			tmp[pp] = ul;
			tmp[pp++ + ww] = random(ul, dl);
			tmp[pp] = random(ul, ur);
			tmp[pp++ + ww] = random(ul, ur, dl, dr);

			ul = ur;
			dl = dr;
		}
	}
	intArray result = IntCache::allocate(w * h);
	for (int y = 0; y < h; y++)
	{
		System::arraycopy(tmp, (y + (yo & 1)) * (pw << 1) + (xo & 1), &result, y * w, w);
	}
	return result;
}

int ZoomLayer::random(int a, int b)
{
	return nextRandom(2) == 0 ? a : b;
}

int ZoomLayer::random(int a, int b, int c, int d)
{
	if (b == c && c == d) return b;
	if (a == b && a == c) return a;
	if (a == b && a == d) return a;
	if (a == c && a == d) return a;

	if (a == b && c != d) return a;
	if (a == c && b != d) return a;
	if (a == d && b != c) return a;

	if (b == a && c != d) return b;
	if (b == c && a != d) return b;
	if (b == d && a != c) return b;

	if (c == a && b != d) return c;
	if (c == b && a != d) return c;
	if (c == d && a != b) return c;

	if (d == a && b != c) return c;
	if (d == b && a != c) return c;
	if (d == c && a != b) return c;

	int s = nextRandom(4);
	if (s == 0) return a;
	if (s == 1) return b;
	if (s == 2) return c;
	return d;
}

shared_ptr<Layer>ZoomLayer::zoom(__int64 seed, shared_ptr<Layer> sup, int count)
{
	shared_ptr<Layer>result = sup;
	for (int i = 0; i < count; i++)
	{
		result = shared_ptr<Layer>(new ZoomLayer(seed + i, result));
	}
	return result;
}