Add CycleTracker

zbx1425 · Aug 1, 2023 · 8bfd34f · 8bfd34f
1 parent 55f9549
commit 8bfd34f
Show file tree

Hide file tree

Showing 6 changed files with 80 additions and 7 deletions.
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
@@ -4,8 +4,9 @@
 import cn.zbx1425.mtrsteamloco.MainClient;
 import cn.zbx1425.mtrsteamloco.render.scripting.eyecandy.EyeCandyScriptContext;
 import cn.zbx1425.mtrsteamloco.render.scripting.train.TrainScriptContext;
+import cn.zbx1425.mtrsteamloco.render.scripting.util.CycleTracker;
 import cn.zbx1425.mtrsteamloco.render.scripting.util.GraphicsTexture;
-import cn.zbx1425.mtrsteamloco.render.scripting.util.ScriptTimingUtil;
+import cn.zbx1425.mtrsteamloco.render.scripting.util.TimingUtil;
 import cn.zbx1425.mtrsteamloco.render.scripting.util.StateTracker;
 import cn.zbx1425.sowcer.math.Matrices;
 import cn.zbx1425.sowcer.math.Matrix4f;
@@ -43,8 +44,9 @@ public void load(Map<ResourceLocation, String> scripts) {
             scope.put("Resources", scope, new NativeJavaClass(scope, ScriptResourceUtil.class));
             scope.put("GraphicsTexture", scope, new NativeJavaClass(scope, GraphicsTexture.class));
 
-            scope.put("Timing", scope, new NativeJavaClass(scope, ScriptTimingUtil.class));
+            scope.put("Timing", scope, new NativeJavaClass(scope, TimingUtil.class));
             scope.put("StateTracker", scope, new NativeJavaClass(scope, StateTracker.class));
+            scope.put("CycleTracker", scope, new NativeJavaClass(scope, CycleTracker.class));
 
             scope.put("Color", scope, new NativeJavaClass(scope, Color.class));
             scope.put("RawModel", scope, new NativeJavaClass(scope, RawModel.class));
@@ -68,10 +70,10 @@ public void load(Map<ResourceLocation, String> scripts) {
                             new NativeJavaClass(scope, classToLoadClass));
                 }
                 scope.put("CompoundTag", scope, new NativeJavaClass(scope, CompoundTag.class));
-                scope.put("installedMadParticle", scope, true);
+                scope.put("foundMadParticle", scope, true);
             } catch (ClassNotFoundException ignored) {
                 Main.LOGGER.warn("MadParticle", ignored);
-                scope.put("installedMadParticle", scope, false);
+                scope.put("foundMadParticle", scope, false);
             }
 
             ScriptResourceUtil.scriptsToExecute = new ArrayList<>(scripts.entrySet());

diff --git a/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/ScriptResourceUtil.java b/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/ScriptResourceUtil.java
@@ -15,6 +15,7 @@
 import java.util.Map;
 import java.util.Optional;
 
+@SuppressWarnings("unused")
 public class ScriptResourceUtil {
 
     protected static List<Map.Entry<ResourceLocation, String>> scriptsToExecute;

diff --git a/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/util/CycleTracker.java b/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/util/CycleTracker.java
@@ -0,0 +1,67 @@
+package cn.zbx1425.mtrsteamloco.render.scripting.util;
+
+@SuppressWarnings("unused")
+public class CycleTracker {
+
+    private final String[] states;
+    private final float[] offsets;
+    private final float cycleDuration;
+
+    public CycleTracker(Object[] params) {
+        if (params.length % 2 != 0) throw new IllegalArgumentException();
+        float offset = 0;
+        states = new String[params.length / 2];
+        offsets = new float[params.length / 2];
+        for (int i = 0; i < params.length; i += 2) {
+            if (!(params[i] instanceof String)) throw new IllegalArgumentException();
+            if (!(params[i + 1] instanceof Float)) throw new IllegalArgumentException();
+            states[i / 2] = params[i].toString();
+            float elemDuration = Float.parseFloat(params[i + 1].toString());
+            offsets[i / 2] = offset;
+            offset += elemDuration;
+        }
+        cycleDuration = offset;
+    }
+
+    private String lastState;
+    private String currentState;
+    private float currentStateTime;
+    private int lastStateNum;
+    private boolean firstTimeCurrentState;
+
+    public void tick() {
+        float time = TimingUtil.elapsed() % cycleDuration;
+        int cycleNum = (int) (TimingUtil.elapsed() / cycleDuration);
+        for (int i = 0; i < offsets.length; i++) {
+            if (time >= offsets[i]) {
+                int stateNum = cycleNum * offsets.length + i;
+                currentState = states[i];
+                currentStateTime = offsets[i];
+                lastState = states[i == 0 ? offsets.length - 1 : i - 1];
+                if (lastStateNum != stateNum) {
+                    firstTimeCurrentState = true;
+                    lastStateNum = stateNum;
+                } else {
+                    firstTimeCurrentState = false;
+                }
+                break;
+            }
+        }
+    }
+
+    public String stateNow() {
+        return currentState;
+    }
+
+    public String stateLast() {
+        return lastState;
+    }
+
+    public float stateNowDuration() {
+        return TimingUtil.elapsed() - currentStateTime;
+    }
+
+    public boolean stateNowFirst() {
+        return firstTimeCurrentState;
+    }
+}
diff --git a/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/util/GraphicsTexture.java b/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/util/GraphicsTexture.java
@@ -14,6 +14,7 @@
 import java.awt.image.*;
 import java.util.UUID;
 
+@SuppressWarnings("unused")
 public class GraphicsTexture {
 
     private final DynamicTexture dynamicTexture;

diff --git a/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/util/StateTracker.java b/common/src/main/java/cn/zbx1425/mtrsteamloco/render/scripting/util/StateTracker.java
@@ -1,5 +1,6 @@
 package cn.zbx1425.mtrsteamloco.render.scripting.util;
 
+@SuppressWarnings("unused")
 public class StateTracker {
 
     private String lastState;
@@ -11,7 +12,7 @@ public void setState(String value) {
         if (value != null && !value.equals(currentState)) {
             lastState = currentState;
             currentState = value;
-            currentStateTime = ScriptTimingUtil.elapsed();
+            currentStateTime = TimingUtil.elapsed();
             firstTimeCurrentState = true;
         } else if (value != null) {
             firstTimeCurrentState = false;
@@ -27,7 +28,7 @@ public String stateLast() {
     }
 
     public float stateNowDuration() {
-        return ScriptTimingUtil.elapsed() - currentStateTime;
+        return TimingUtil.elapsed() - currentStateTime;
     }
 
     public boolean stateNowFirst() {

diff --git a/...nder/scripting/util/ScriptTimingUtil.java → ...oco/render/scripting/util/TimingUtil.java b/...nder/scripting/util/ScriptTimingUtil.java → ...oco/render/scripting/util/TimingUtil.java
@@ -2,7 +2,8 @@
 
 import mtr.MTRClient;
 
-public class ScriptTimingUtil {
+@SuppressWarnings("unused")
+public class TimingUtil {
 
     public static float elapsed() {
         return MTRClient.getGameTick() / 20;