Improved #59

versions/1.21.4/src/main/java/org/saturnclient/impl/mixins/MatrixStackMixin.java

@@ -1,13 +1,21 @@
 package org.saturnclient.impl.mixins;
 
+import org.joml.Quaternionf;
 import org.saturnclient.common.ref.render.MatrixStackRef;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.Shadow;
 
 import net.minecraft.client.util.math.MatrixStack;
+import net.minecraft.util.math.RotationAxis;
 
 @Mixin(MatrixStack.class)
 public abstract class MatrixStackMixin implements MatrixStackRef {
+    @Shadow
+    public abstract void push();
+
+    @Shadow
+    public abstract void pop();
+
     @Shadow
     public abstract void translate(float x, float y, float z);
 
@@ -15,10 +23,10 @@ public abstract class MatrixStackMixin implements MatrixStackRef {
     public abstract void scale(float x, float y, float z);
 
     @Shadow
-    public abstract void push();
+    public abstract void multiply(Quaternionf quaternion);
 
     @Shadow
-    public abstract void pop();
+    public abstract void multiply(Quaternionf quaternion, float originX, float originY, float originZ);
 
     @Override
     public void translate(float x, float y) {
@@ -29,4 +37,14 @@ public abstract class MatrixStackMixin implements MatrixStackRef {
     public void scale(float x, float y) {
         scale(x, y, 1.0f);
     }
-}
+
+    @Override
+    public void rotate(float angle, float originX, float originY, float originZ) {
+        multiply(RotationAxis.NEGATIVE_Z.rotationDegrees(angle), originX, originY, originZ);
+    }
+
+    @Override
+    public void rotate(float angle) {
+        multiply(RotationAxis.NEGATIVE_Z.rotationDegrees(angle));
+    }
+}