package net.minecraft.client.renderer;

import com.google.common.collect.Queues;
import com.mojang.logging.LogUtils;
import java.util.ArrayList;
import java.util.List;
import java.util.Queue;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import org.jetbrains.annotations.Nullable;
import org.slf4j.Logger;

@Environment(EnvType.CLIENT)
public class SectionBufferBuilderPool {
	private static final Logger LOGGER = LogUtils.getLogger();
	private final Queue<SectionBufferBuilderPack> freeBuffers;
	private volatile int freeBufferCount;

	private SectionBufferBuilderPool(List<SectionBufferBuilderPack> freeBuffers) {
		this.freeBuffers = Queues.<SectionBufferBuilderPack>newArrayDeque(freeBuffers);
		this.freeBufferCount = this.freeBuffers.size();
	}

	public static SectionBufferBuilderPool allocate(int bufferCount) {
		int i = Math.max(1, (int)(Runtime.getRuntime().maxMemory() * 0.3) / SectionBufferBuilderPack.TOTAL_BUFFERS_SIZE);
		int j = Math.max(1, Math.min(bufferCount, i));
		List<SectionBufferBuilderPack> list = new ArrayList(j);

		try {
			for (int k = 0; k < j; k++) {
				list.add(new SectionBufferBuilderPack());
			}
		} catch (OutOfMemoryError var7) {
			LOGGER.warn("Allocated only {}/{} buffers", list.size(), j);
			int l = Math.min(list.size() * 2 / 3, list.size() - 1);

			for (int m = 0; m < l; m++) {
				((SectionBufferBuilderPack)list.remove(list.size() - 1)).close();
			}
		}

		return new SectionBufferBuilderPool(list);
	}

	@Nullable
	public SectionBufferBuilderPack acquire() {
		SectionBufferBuilderPack sectionBufferBuilderPack = (SectionBufferBuilderPack)this.freeBuffers.poll();
		if (sectionBufferBuilderPack != null) {
			this.freeBufferCount = this.freeBuffers.size();
			return sectionBufferBuilderPack;
		} else {
			return null;
		}
	}

	public void release(SectionBufferBuilderPack buffer) {
		this.freeBuffers.add(buffer);
		this.freeBufferCount = this.freeBuffers.size();
	}

	public boolean isEmpty() {
		return this.freeBuffers.isEmpty();
	}

	public int getFreeBufferCount() {
		return this.freeBufferCount;
	}
}