1 files changed, 41 insertions, 0 deletions
diff --git a/src/main/java/org/chrisoft/trashyaddon/mixin/meteorclient/PostProcessShadersMixin.java b/src/main/java/org/chrisoft/trashyaddon/mixin/meteorclient/PostProcessShadersMixin.java
new file mode 100644
index 0000000..565e670
--- /dev/null
+++ b/src/main/java/org/chrisoft/trashyaddon/mixin/meteorclient/PostProcessShadersMixin.java
@@ -0,0 +1,41 @@
+package org.chrisoft.trashyaddon.mixin.meteorclient;
+
+import com.mojang.logging.LogUtils;
+import meteordevelopment.meteorclient.utils.render.postprocess.EntityShader;
+import net.minecraft.client.render.VertexConsumerProvider;
+import org.chrisoft.trashyaddon.misc.AddonPostProcessingShaders;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Inject;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
+import org.spongepowered.asm.mixin.injection.callback.CallbackInfoReturnable;
+
+@Mixin(value = meteordevelopment.meteorclient.utils.render.postprocess.PostProcessShaders.class, remap = false)
+public abstract class PostProcessShadersMixin {
+    @Inject(method = "beginRender()V", at = @At("TAIL"))
+    private static void beginRender(CallbackInfo __) {
+        for (EntityShader es : AddonPostProcessingShaders.entityShaders)
+            es.beginRender();
+    }
+
+    @Inject(method = "endRender()V", at = @At("TAIL"))
+    private static void endRender(CallbackInfo __) {
+        for (EntityShader es : AddonPostProcessingShaders.entityShaders)
+            es.endRender();
+    }
+
+    @Inject(method = "onResized(II)V", at = @At("TAIL"))
+    private static void onResized(int width, int height, CallbackInfo __) {
+        for (EntityShader es : AddonPostProcessingShaders.entityShaders)
+            es.onResized(width, height);
+    }
+
+    @Inject(method = "isCustom(Lnet/minecraft/client/render/VertexConsumerProvider;)Z", at = @At("HEAD"), cancellable = true, remap = true)
+    private static void isCustom(VertexConsumerProvider vcp, CallbackInfoReturnable<Boolean> info) {
+        for (EntityShader es : AddonPostProcessingShaders.entityShaders)
+            if (vcp == es.vertexConsumerProvider) {
+                info.setReturnValue(true);
+                info.cancel();
+            }
+    }
+}