package net.minecraft.client.renderer;

import com.google.common.collect.Lists;
import com.mojang.blaze3d.pipeline.RenderTarget;
import com.mojang.blaze3d.systems.RenderSystem;
import com.mojang.blaze3d.vertex.BufferBuilder;
import com.mojang.blaze3d.vertex.BufferUploader;
import com.mojang.blaze3d.vertex.DefaultVertexFormat;
import com.mojang.blaze3d.vertex.Tesselator;
import com.mojang.blaze3d.vertex.VertexFormat;
import java.io.IOException;
import java.util.List;
import java.util.function.IntSupplier;
import net.fabricmc.api.EnvType;
import net.fabricmc.api.Environment;
import net.minecraft.client.Minecraft;
import net.minecraft.server.packs.resources.ResourceProvider;
import org.joml.Matrix4f;

@Environment(EnvType.CLIENT)
public class PostPass implements AutoCloseable {
	private final EffectInstance effect;
	public final RenderTarget inTarget;
	public final RenderTarget outTarget;
	private final List<IntSupplier> auxAssets = Lists.<IntSupplier>newArrayList();
	private final List<String> auxNames = Lists.<String>newArrayList();
	private final List<Integer> auxWidths = Lists.<Integer>newArrayList();
	private final List<Integer> auxHeights = Lists.<Integer>newArrayList();
	private Matrix4f shaderOrthoMatrix;
	private final int filterMode;

	public PostPass(ResourceProvider resourceProvider, String name, RenderTarget inTarget, RenderTarget outTarget, boolean useLinearFilter) throws IOException {
		this.effect = new EffectInstance(resourceProvider, name);
		this.inTarget = inTarget;
		this.outTarget = outTarget;
		this.filterMode = useLinearFilter ? 9729 : 9728;
	}

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

	public final String getName() {
		return this.effect.getName();
	}

	public void addAuxAsset(String auxName, IntSupplier auxFramebuffer, int width, int height) {
		this.auxNames.add(this.auxNames.size(), auxName);
		this.auxAssets.add(this.auxAssets.size(), auxFramebuffer);
		this.auxWidths.add(this.auxWidths.size(), width);
		this.auxHeights.add(this.auxHeights.size(), height);
	}

	public void setOrthoMatrix(Matrix4f shaderOrthoMatrix) {
		this.shaderOrthoMatrix = shaderOrthoMatrix;
	}

	public void process(float partialTicks) {
		this.inTarget.unbindWrite();
		float f = this.outTarget.width;
		float g = this.outTarget.height;
		RenderSystem.viewport(0, 0, (int)f, (int)g);
		this.effect.setSampler("DiffuseSampler", this.inTarget::getColorTextureId);

		for (int i = 0; i < this.auxAssets.size(); i++) {
			this.effect.setSampler((String)this.auxNames.get(i), (IntSupplier)this.auxAssets.get(i));
			this.effect.safeGetUniform("AuxSize" + i).set((float)((Integer)this.auxWidths.get(i)).intValue(), (float)((Integer)this.auxHeights.get(i)).intValue());
		}

		this.effect.safeGetUniform("ProjMat").set(this.shaderOrthoMatrix);
		this.effect.safeGetUniform("InSize").set((float)this.inTarget.width, (float)this.inTarget.height);
		this.effect.safeGetUniform("OutSize").set(f, g);
		this.effect.safeGetUniform("Time").set(partialTicks);
		Minecraft minecraft = Minecraft.getInstance();
		this.effect.safeGetUniform("ScreenSize").set((float)minecraft.getWindow().getWidth(), (float)minecraft.getWindow().getHeight());
		this.effect.apply();
		this.outTarget.clear(Minecraft.ON_OSX);
		this.outTarget.bindWrite(false);
		RenderSystem.depthFunc(519);
		BufferBuilder bufferBuilder = Tesselator.getInstance().begin(VertexFormat.Mode.QUADS, DefaultVertexFormat.POSITION);
		bufferBuilder.addVertex(0.0F, 0.0F, 500.0F);
		bufferBuilder.addVertex(f, 0.0F, 500.0F);
		bufferBuilder.addVertex(f, g, 500.0F);
		bufferBuilder.addVertex(0.0F, g, 500.0F);
		BufferUploader.draw(bufferBuilder.buildOrThrow());
		RenderSystem.depthFunc(515);
		this.effect.clear();
		this.outTarget.unbindWrite();
		this.inTarget.unbindRead();

		for (Object object : this.auxAssets) {
			if (object instanceof RenderTarget) {
				((RenderTarget)object).unbindRead();
			}
		}
	}

	public EffectInstance getEffect() {
		return this.effect;
	}

	public int getFilterMode() {
		return this.filterMode;
	}
}