Add MinecraftClientUtil

zbx1425 · Sep 29, 2023 · 8349b7c · 8349b7c
1 parent 64864b1
commit 8349b7c
Show file tree

Hide file tree

Showing 4 changed files with 29 additions and 26 deletions.
diff --git a/common/src/main/java/cn/zbx1425/mtrsteamloco/render/integration/NoopModelMapper.java b/common/src/main/java/cn/zbx1425/mtrsteamloco/render/integration/NoopModelMapper.java
diff --git a/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/ScriptHolder.java b/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/ScriptHolder.java
@@ -57,6 +57,8 @@ public void load(Map<ResourceLocation, String> scripts) {
 
             scope.put("MTRClientData", scope, new NativeJavaClass(scope, ClientData.class));
 
+            scope.put("MinecraftClient", scope, new NativeJavaClass(scope, MinecraftClientUtil.class));
+
             try {
                 String[] classesToLoad = {
                         "util.AddParticleHelper",

diff --git a/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/util/MinecraftClientUtil.java b/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/util/MinecraftClientUtil.java
@@ -0,0 +1,22 @@
+package cn.zbx1425.mtrsteamloco.render.scripting.util;
+
+import cn.zbx1425.sowcer.math.Vector3f;
+import net.minecraft.client.Minecraft;
+
+public class MinecraftClientUtil {
+
+    public static boolean worldIsRaining() {
+        return Minecraft.getInstance().level != null
+                && Minecraft.getInstance().level.isRaining();
+    }
+
+    public static boolean worldIsRainingAt(Vector3f pos) {
+        return Minecraft.getInstance().level != null
+                && Minecraft.getInstance().level.isRainingAt(pos.toBlockPos());
+    }
+
+    public static int worldDayTime() {
+        return Minecraft.getInstance().level != null
+                ? (int) Minecraft.getInstance().level.getDayTime() : 0;
+    }
+}
diff --git a/common/src/main/java/cn/zbx1425/sowcer/math/Vector3f.java b/common/src/main/java/cn/zbx1425/sowcer/math/Vector3f.java
@@ -1,5 +1,6 @@
 package cn.zbx1425.sowcer.math;
 
+import net.minecraft.core.BlockPos;
 import net.minecraft.util.Mth;
 
 public class Vector3f {
@@ -165,6 +166,10 @@ public float distanceSq(Vector3f other) {
         return (float)(dx * dx + dy * dy + dz * dz);
     }
 
+    public BlockPos toBlockPos() {
+        return new BlockPos(Mth.floor(x()), Mth.floor(y()), Mth.floor(z()));
+    }
+
     @Override
     public boolean equals(Object o) {
         if (this == o) return true;