1.4

dist/ba_root/mods/tools/wavedash.py

@@ -0,0 +1,124 @@
+"""Wavedash by TheMikirog
+    
+    This is an early version of the plugin. Feedback appreciated!
+
+"""
+
+# ba_meta require api 6
+
+from __future__ import annotations
+
+from typing import TYPE_CHECKING
+
+import ba
+import math
+import bastd
+from bastd.actor.spaz import Spaz
+
+if TYPE_CHECKING:
+    pass
+
+# ba_meta export plugin
+class MikiWavedashTest(ba.Plugin):
+    
+
+    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 = ba.time(timeformat=ba.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
+            ba.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
+    bastd.actor.spaz.Spaz.__init__ = new_spaz_init(bastd.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', bastd.gameutils.SharedObjects.get().footing_material),
+            actions=(('message', 'our_node', 'at_connect', MikiWavedashTest.FootConnectMessage),
+                     ('message', 'our_node', 'at_disconnect', MikiWavedashTest.FootDisconnectMessage)))
+        return wrapper
+    bastd.actor.spazfactory.SpazFactory.__init__ = new_factory(bastd.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
+    bastd.actor.spaz.Spaz.handlemessage = new_handlemessage(bastd.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
+    bastd.actor.spaz.Spaz.on_run = new_on_run(bastd.actor.spaz.Spaz.on_run)