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.concurrent.ArrayBlockingQueue;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import org.jspecify.annotations.Nullable;
import org.slf4j.Logger;

@Environment(EnvType.CLIENT)
public class SectionBufferBuilderPool implements AutoCloseable {
	private static final Logger LOGGER = LogUtils.getLogger();
	private final ArrayBlockingQueue<SectionBufferBuilderPack> freeBuffers;

	private SectionBufferBuilderPool(final List<SectionBufferBuilderPack> buffers) {
		this.freeBuffers = Queues.newArrayBlockingQueue(buffers.size());
		this.freeBuffers.addAll(buffers);
	}

	public static SectionBufferBuilderPool allocate(final int maxWorkers) {
		int maxBuffers = Math.max(1, (int)(Runtime.getRuntime().maxMemory() * 0.3) / SectionBufferBuilderPack.TOTAL_BUFFERS_SIZE);
		int targetBufferCount = Math.max(1, Math.min(maxWorkers, maxBuffers));
		List<SectionBufferBuilderPack> buffers = new ArrayList(targetBufferCount);

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

			for (int i = 0; i < buffersToDrop; i++) {
				((SectionBufferBuilderPack)buffers.remove(buffers.size() - 1)).close();
			}
		}

		return new SectionBufferBuilderPool(buffers);
	}

	@Nullable
	public SectionBufferBuilderPack acquire() {
		return (SectionBufferBuilderPack)this.freeBuffers.poll();
	}

	public void release(final SectionBufferBuilderPack buffer) {
		this.freeBuffers.offer(buffer);
	}

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

	public int getFreeBufferCount() {
		return this.freeBuffers.size();
	}

	public void close() {
		this.freeBuffers.forEach(SectionBufferBuilderPack::close);
		this.freeBuffers.clear();
	}
}