updating mods to api8

dist/ba_root/mods/plugins/wavedash.py

@@ -0,0 +1,134 @@
+"""Wavedash by TheMikirog
+
+    This is an early version of the plugin. Feedback appreciated!
+
+"""
+
+# ba_meta require api 8
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import babase
+import bauiv1 as bui
+import bascenev1 as bs
+import math
+import bascenev1lib
+from bascenev1lib.actor.spaz import Spaz
+
+if TYPE_CHECKING:
+    pass
+
+
+class MikiWavedashTest:
+
+    class FootConnectMessage:
+        """Spaz started touching the ground"""
+
+    class FootDisconnectMessage:
+        """Spaz stopped touching the ground"""
+
+    def wavedash(self) -> None:
+        if not self.node:
+            return
+
+        isMoving = abs(self.node.move_up_down) >= 0.5 or abs(
+            self.node.move_left_right) >= 0.5
+
+        if self._dead or not self.grounded or not isMoving:
+            return
+
+        if self.node.knockout > 0.0 or self.frozen or self.node.hold_node:
+            return
+
+        t_ms = bs.time(timeformat=babase.TimeFormat.MILLISECONDS)
+        assert isinstance(t_ms, int)
+
+        if t_ms - self.last_wavedash_time_ms >= self._wavedash_cooldown:
+
+            move = [self.node.move_left_right, -self.node.move_up_down]
+            vel = [self.node.velocity[0], self.node.velocity[2]]
+
+            move_length = math.hypot(move[0], move[1])
+            vel_length = math.hypot(vel[0], vel[1])
+            if vel_length < 1.25:
+                return
+            move_norm = [m/move_length for m in move]
+            vel_norm = [v/vel_length for v in vel]
+            dot = sum(x*y for x, y in zip(move_norm, vel_norm))
+            turn_power = min(round(math.acos(dot) / math.pi, 2)*1.3, 1)
+            if turn_power < 0.2:
+                return
+
+            boost_power = math.sqrt(
+                math.pow(vel[0], 2) + math.pow(vel[1], 2)) * 1.2
+            boost_power = min(pow(boost_power, 4), 160)
+            # print(boost_power * turn_power)
+
+            self.last_wavedash_time_ms = t_ms
+
+            # FX
+            bs.emitfx(position=self.node.position,
+                      velocity=(vel[0]*0.5, -1, vel[1]*0.5),
+                      chunk_type='sweat',
+                      count=8,
+                      scale=boost_power / 160 * turn_power,
+                      spread=0.25)
+
+            # Boost itself
+            pos = self.node.position
+            for i in range(6):
+                self.node.handlemessage('impulse', pos[0], -0.1+pos[1]+i*0.1, pos[2],
+                                        0, 0, 0,
+                                        boost_power * turn_power,
+                                        boost_power * turn_power, 0, 0,
+                                        move[0], 0, move[1])
+
+    def new_spaz_init(func):
+        def wrapper(*args, **kwargs):
+
+            func(*args, **kwargs)
+
+            # args[0] = self
+            args[0]._wavedash_cooldown = 30
+            args[0].last_wavedash_time_ms = -9999
+            args[0].grounded = 0
+
+        return wrapper
+    bascenev1lib.actor.spaz.Spaz.__init__ = new_spaz_init(
+        bascenev1lib.actor.spaz.Spaz.__init__)
+
+    def new_factory(func):
+        def wrapper(*args, **kwargs):
+            func(*args, **kwargs)
+
+            args[0].roller_material.add_actions(
+                conditions=('they_have_material',
+                            bascenev1lib.gameutils.SharedObjects.get().footing_material),
+                actions=(('message', 'our_node', 'at_connect', MikiWavedashTest.FootConnectMessage),
+                         ('message', 'our_node', 'at_disconnect', MikiWavedashTest.FootDisconnectMessage)))
+        return wrapper
+    bascenev1lib.actor.spazfactory.SpazFactory.__init__ = new_factory(
+        bascenev1lib.actor.spazfactory.SpazFactory.__init__)
+
+    def new_handlemessage(func):
+        def wrapper(*args, **kwargs):
+            if args[1] == MikiWavedashTest.FootConnectMessage:
+                args[0].grounded += 1
+            elif args[1] == MikiWavedashTest.FootDisconnectMessage:
+                if args[0].grounded > 0:
+                    args[0].grounded -= 1
+
+            func(*args, **kwargs)
+        return wrapper
+    bascenev1lib.actor.spaz.Spaz.handlemessage = new_handlemessage(
+        bascenev1lib.actor.spaz.Spaz.handlemessage)
+
+    def new_on_run(func):
+        def wrapper(*args, **kwargs):
+            if args[0]._last_run_value < args[1] and args[1] > 0.8:
+                MikiWavedashTest.wavedash(args[0])
+            func(*args, **kwargs)
+        return wrapper
+    bascenev1lib.actor.spaz.Spaz.on_run = new_on_run(bascenev1lib.actor.spaz.Spaz.on_run)