StructureLib/src/main/java/com/gtnewhorizon/structurelib/structure/LazyStructureElement.java

package com.gtnewhorizon.structurelib.structure;

import net.minecraft.item.ItemStack;
import net.minecraft.world.World;

import java.util.function.Function;

class LazyStructureElement<T> implements IStructureElementDeferred<T> {
	private Function<T, IStructureElement<T>> to;
	private IStructureElement<T> elem;

	public LazyStructureElement(Function<T, IStructureElement<T>> to) {
		this.to = to;
	}

	private IStructureElement<T> get(T t) {
		if (to != null) {
			elem = to.apply(t);
			to = null;
		}
		return elem;
	}

	@Override
	public boolean check(T t, World world, int x, int y, int z) {
		return get(t).check(t, world, x, y, z);
	}

	@Override
	public boolean placeBlock(T t, World world, int x, int y, int z, ItemStack trigger) {
		return get(t).placeBlock(t, world, x, y, z, trigger);
	}

	@Override
	public boolean spawnHint(T t, World world, int x, int y, int z, ItemStack trigger) {
		return get(t).spawnHint(t, world, x, y, z, trigger);
	}
}