添加极其逆天的molang渲染法

src/main/java/com/atsuishio/superbwarfare/client/model/item/CustomGunModel.java

@@ -3,21 +3,68 @@ package com.atsuishio.superbwarfare.client.model.item;
 import com.atsuishio.superbwarfare.client.molang.MolangVariable;
 import com.atsuishio.superbwarfare.data.gun.GunData;
 import com.atsuishio.superbwarfare.item.gun.GunItem;
+import com.atsuishio.superbwarfare.mixins.AnimationProcessorAccessor;
+import com.atsuishio.superbwarfare.mixins.GeoModelAccessor;
 import net.minecraft.client.Minecraft;
 import net.minecraft.world.item.ItemDisplayContext;
 import net.minecraft.world.item.ItemStack;
 import software.bernie.geckolib.animatable.GeoAnimatable;
 import software.bernie.geckolib.animatable.GeoItem;
+import software.bernie.geckolib.animation.AnimatableManager;
+import software.bernie.geckolib.animation.AnimationProcessor;
 import software.bernie.geckolib.animation.AnimationState;
 import software.bernie.geckolib.constant.DataTickets;
 import software.bernie.geckolib.loading.math.MathParser;
 import software.bernie.geckolib.loading.math.MolangQueries;
 import software.bernie.geckolib.model.GeoModel;
+import software.bernie.geckolib.util.RenderUtil;
 
 import java.util.function.DoubleSupplier;
 
 public abstract class CustomGunModel<T extends GunItem & GeoAnimatable> extends GeoModel<T> {
 
+    // TODO 优化这一坨
+    @Override
+    public void setCustomAnimations(T animatable, long instanceId, AnimationState<T> animationState) {
+        Minecraft mc = Minecraft.getInstance();
+        AnimatableManager<T> animatableManager = animatable.getAnimatableInstanceCache().getManagerForId(instanceId);
+        Double currentTick = animationState.getData(DataTickets.TICK);
+
+        if (currentTick == null)
+            currentTick = RenderUtil.getCurrentTick();
+
+//        if (animatableManager.getFirstTickTime() == -1)
+//            animatableManager.startedAt(currentTick + mc.getFrameTime());
+
+        double currentFrameTime = currentTick - animatableManager.getFirstTickTime();
+        boolean isReRender = !animatableManager.isFirstTick() && currentFrameTime == animatableManager.getLastUpdateTime();
+
+        if (isReRender && instanceId == ((GeoModelAccessor) this).getLastRenderedInstance())
+            return;
+
+        if (!mc.isPaused() || animatable.shouldPlayAnimsWhileGamePaused()) {
+            animatableManager.updatedAt(currentFrameTime);
+
+            double lastUpdateTime = animatableManager.getLastUpdateTime();
+            ((GeoModelAccessor) this).setAnimTime(((GeoModelAccessor) this).getAnimTime() + lastUpdateTime - ((GeoModelAccessor) this).getLastGameTickTime());
+            ((GeoModelAccessor) this).setLastGameTickTime(lastUpdateTime);
+        }
+
+        animationState.animationTick = ((GeoModelAccessor) this).getAnimTime();
+        ((GeoModelAccessor) this).setLastRenderedInstance(instanceId);
+        AnimationProcessor<T> processor = getAnimationProcessor();
+
+        var model = ((AnimationProcessorAccessor<T>) processor).getModel();
+        if (model instanceof CustomGunModel<T> customGunModel) {
+            customGunModel.applyCustomMolangQueries(animationState, ((GeoModelAccessor) this).getAnimTime());
+        }
+
+        if (!processor.getRegisteredBones().isEmpty())
+            processor.tickAnimation(animatable, this, animatableManager, ((GeoModelAccessor) this).getAnimTime(), animationState, crashIfBoneMissing());
+
+        setCustomAnimations(animatable, instanceId, animationState);
+    }
+
     @Override
     public void applyMolangQueries(AnimationState<T> animationState, double animTime) {
         Minecraft mc = Minecraft.getInstance();
@@ -31,14 +78,37 @@ public abstract class CustomGunModel<T extends GunItem & GeoAnimatable> extends
         }
 
         set(MolangVariable.SBW_SYSTEM_TIME, System::currentTimeMillis);
+    }
+
+    public void applyCustomMolangQueries(AnimationState<T> animationState, double animTime) {
+        this.applyMolangQueries(animationState, animTime);
+
+        Minecraft mc = Minecraft.getInstance();
 
         // GunData
 
         var player = mc.player;
-        if (player == null) return;
+        if (player == null) {
+            resetQueryValue();
+            return;
+        }
 
         var stack = player.getMainHandItem();
-        if (!(stack.getItem() instanceof GunItem)) return;
+        if (!(stack.getItem() instanceof GunItem)) {
+            resetQueryValue();
+            return;
+        }
+
+        var item = animationState.getData(DataTickets.ITEMSTACK);
+        if (item == null || GeoItem.getId(item) != GeoItem.getId(stack)) {
+            resetQueryValue();
+            return;
+        }
+
+        if (animationState.getData(DataTickets.ITEM_RENDER_PERSPECTIVE) != ItemDisplayContext.FIRST_PERSON_RIGHT_HAND) {
+            resetQueryValue();
+            return;
+        }
 
         var data = GunData.from(stack);
 
@@ -50,6 +120,10 @@ public abstract class CustomGunModel<T extends GunItem & GeoAnimatable> extends
         MathParser.setVariable(key, value);
     }
 
+    private void resetQueryValue() {
+        MathParser.setVariable(MolangVariable.SBW_IS_EMPTY, () -> 0);
+    }
+
     public boolean shouldCancelRender(ItemStack stack, AnimationState<T> animationState) {
         if (!(stack.getItem() instanceof GunItem)) return true;
         var item = animationState.getData(DataTickets.ITEMSTACK);

src/main/java/com/atsuishio/superbwarfare/mixins/AnimationProcessorAccessor.java

@@ -0,0 +1,14 @@
+package com.atsuishio.superbwarfare.mixins;
+
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.gen.Accessor;
+import software.bernie.geckolib.animatable.GeoAnimatable;
+import software.bernie.geckolib.animation.AnimationProcessor;
+import software.bernie.geckolib.model.GeoModel;
+
+@Mixin(AnimationProcessor.class)
+public interface AnimationProcessorAccessor<T extends GeoAnimatable> {
+
+    @Accessor(value = "model", remap = false)
+    GeoModel<T> getModel();
+}

src/main/java/com/atsuishio/superbwarfare/mixins/GeoModelAccessor.java

@@ -0,0 +1,27 @@
+package com.atsuishio.superbwarfare.mixins;
+
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.gen.Accessor;
+import software.bernie.geckolib.model.GeoModel;
+
+@Mixin(GeoModel.class)
+public interface GeoModelAccessor {
+
+    @Accessor(value = "animTime", remap = false)
+    double getAnimTime();
+
+    @Accessor(value = "animTime", remap = false)
+    void setAnimTime(double animTime);
+
+    @Accessor(value = "lastGameTickTime", remap = false)
+    double getLastGameTickTime();
+
+    @Accessor(value = "lastGameTickTime", remap = false)
+    void setLastGameTickTime(double lastGameTickTime);
+
+    @Accessor(value = "lastRenderedInstance", remap = false)
+    long getLastRenderedInstance();
+
+    @Accessor(value = "lastRenderedInstance", remap = false)
+    void setLastRenderedInstance(long lastRenderedInstance);
+}

src/main/resources/mixins.superbwarfare.json

@@ -4,10 +4,12 @@
   "compatibilityLevel": "JAVA_21",
   "refmap": "mixins.superbwarfare.refmap.json",
   "mixins": [
+    "AnimationProcessorAccessor",
     "BeastMixin",
     "ClientboundSetPassengersPacketMixin",
     "EntityMixin",
     "FishingHookMixin",
+    "GeoModelAccessor",
     "LivingEntityMixin",
     "PlayerMixin",
     "VillagerMixin"