package com.mojang.blaze3d.pipeline;

import com.google.common.collect.ImmutableList;
import com.mojang.blaze3d.GpuOutOfMemoryException;
import com.mojang.blaze3d.systems.RenderSystem;
import com.mojang.blaze3d.textures.AddressMode;
import com.mojang.blaze3d.textures.FilterMode;
import com.mojang.blaze3d.textures.GpuTexture;
import com.mojang.blaze3d.textures.TextureFormat;
import java.util.List;
import java.util.Objects;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import org.jetbrains.annotations.Nullable;

@Environment(EnvType.CLIENT)
public class MainTarget extends RenderTarget {
	public static final int DEFAULT_WIDTH = 854;
	public static final int DEFAULT_HEIGHT = 480;
	static final MainTarget.Dimension DEFAULT_DIMENSIONS = new MainTarget.Dimension(854, 480);

	public MainTarget(int width, int height) {
		super("Main", true);
		this.createFrameBuffer(width, height);
	}

	private void createFrameBuffer(int width, int height) {
		MainTarget.Dimension dimension = this.allocateAttachments(width, height);
		if (this.colorTexture != null && this.depthTexture != null) {
			this.colorTexture.setTextureFilter(FilterMode.NEAREST, false);
			this.colorTexture.setAddressMode(AddressMode.CLAMP_TO_EDGE);
			this.colorTexture.setTextureFilter(FilterMode.NEAREST, false);
			this.colorTexture.setAddressMode(AddressMode.CLAMP_TO_EDGE);
			this.viewWidth = dimension.width;
			this.viewHeight = dimension.height;
			this.width = dimension.width;
			this.height = dimension.height;
		} else {
			throw new IllegalStateException("Missing color and/or depth textures");
		}
	}

	private MainTarget.Dimension allocateAttachments(int width, int height) {
		RenderSystem.assertOnRenderThread();

		for (MainTarget.Dimension dimension : MainTarget.Dimension.listWithFallback(width, height)) {
			if (this.colorTexture != null) {
				this.colorTexture.close();
				this.colorTexture = null;
			}

			if (this.depthTexture != null) {
				this.depthTexture.close();
				this.depthTexture = null;
			}

			this.colorTexture = this.allocateColorAttachment(dimension);
			this.depthTexture = this.allocateDepthAttachment(dimension);
			if (this.colorTexture != null && this.depthTexture != null) {
				return dimension;
			}
		}

		throw new RuntimeException(
			"Unrecoverable GL_OUT_OF_MEMORY ("
				+ (this.colorTexture == null ? "missing color" : "have color")
				+ ", "
				+ (this.depthTexture == null ? "missing depth" : "have depth")
				+ ")"
		);
	}

	@Nullable
	private GpuTexture allocateColorAttachment(MainTarget.Dimension dimension) {
		try {
			return RenderSystem.getDevice().createTexture(() -> this.label + " / Color", TextureFormat.RGBA8, dimension.width, dimension.height, 1);
		} catch (GpuOutOfMemoryException var3) {
			return null;
		}
	}

	@Nullable
	private GpuTexture allocateDepthAttachment(MainTarget.Dimension dimension) {
		try {
			return RenderSystem.getDevice().createTexture(() -> this.label + " / Depth", TextureFormat.DEPTH32, dimension.width, dimension.height, 1);
		} catch (GpuOutOfMemoryException var3) {
			return null;
		}
	}

	@Environment(EnvType.CLIENT)
	static class Dimension {
		public final int width;
		public final int height;

		Dimension(int width, int height) {
			this.width = width;
			this.height = height;
		}

		static List<MainTarget.Dimension> listWithFallback(int width, int height) {
			RenderSystem.assertOnRenderThread();
			int i = RenderSystem.getDevice().getMaxTextureSize();
			return width > 0 && width <= i && height > 0 && height <= i
				? ImmutableList.of(new MainTarget.Dimension(width, height), MainTarget.DEFAULT_DIMENSIONS)
				: ImmutableList.of(MainTarget.DEFAULT_DIMENSIONS);
		}

		public boolean equals(Object object) {
			if (this == object) {
				return true;
			} else if (object != null && this.getClass() == object.getClass()) {
				MainTarget.Dimension dimension = (MainTarget.Dimension)object;
				return this.width == dimension.width && this.height == dimension.height;
			} else {
				return false;
			}
		}

		public int hashCode() {
			return Objects.hash(new Object[]{this.width, this.height});
		}

		public String toString() {
			return this.width + "x" + this.height;
		}
	}
}