package com.mojang.blaze3d.resource;

import com.google.common.annotations.VisibleForTesting;
import java.util.ArrayDeque;
import java.util.Collection;
import java.util.Deque;
import java.util.Iterator;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;

@Environment(EnvType.CLIENT)
public class CrossFrameResourcePool implements GraphicsResourceAllocator, AutoCloseable {
	private final int framesToKeepResource;
	private final Deque<CrossFrameResourcePool.ResourceEntry<?>> pool = new ArrayDeque();

	public CrossFrameResourcePool(int i) {
		this.framesToKeepResource = i;
	}

	public void endFrame() {
		Iterator<? extends CrossFrameResourcePool.ResourceEntry<?>> iterator = this.pool.iterator();

		while (iterator.hasNext()) {
			CrossFrameResourcePool.ResourceEntry<?> resourceEntry = (CrossFrameResourcePool.ResourceEntry<?>)iterator.next();
			if (resourceEntry.framesToLive-- == 0) {
				resourceEntry.close();
				iterator.remove();
			}
		}
	}

	@Override
	public <T> T acquire(ResourceDescriptor<T> resourceDescriptor) {
		Iterator<? extends CrossFrameResourcePool.ResourceEntry<?>> iterator = this.pool.iterator();

		while (iterator.hasNext()) {
			CrossFrameResourcePool.ResourceEntry<?> resourceEntry = (CrossFrameResourcePool.ResourceEntry<?>)iterator.next();
			if (resourceEntry.descriptor.equals(resourceDescriptor)) {
				iterator.remove();
				return (T)resourceEntry.value;
			}
		}

		return resourceDescriptor.allocate();
	}

	@Override
	public <T> void release(ResourceDescriptor<T> resourceDescriptor, T object) {
		this.pool.addFirst(new CrossFrameResourcePool.ResourceEntry<>(resourceDescriptor, object, this.framesToKeepResource));
	}

	public void clear() {
		this.pool.forEach(CrossFrameResourcePool.ResourceEntry::close);
		this.pool.clear();
	}

	public void close() {
		this.clear();
	}

	@VisibleForTesting
	protected Collection<CrossFrameResourcePool.ResourceEntry<?>> entries() {
		return this.pool;
	}

	@Environment(EnvType.CLIENT)
	@VisibleForTesting
	protected static final class ResourceEntry<T> implements AutoCloseable {
		final ResourceDescriptor<T> descriptor;
		final T value;
		int framesToLive;

		ResourceEntry(ResourceDescriptor<T> resourceDescriptor, T object, int i) {
			this.descriptor = resourceDescriptor;
			this.value = object;
			this.framesToLive = i;
		}

		public void close() {
			this.descriptor.free(this.value);
		}
	}
}