Rename

autopon.py

@@ -0,0 +1,88 @@
+#! /usr/bin/env python3
+import argparse
+import subprocess
+import time
+
+def ydotool(key):
+    subprocess.run([
+        "ydotool",
+        "type",
+        str(key)
+    ])
+
+def main():
+    # A list of all possible songs, expressed as their cardinal directions
+    # We'll map these to keystrokes later
+    songs = [
+            # March of Mobility
+            "L-L-L-R-",
+            # Aria of Attack
+            "R-R-L-R-",
+            # Lament of Defense
+            "U-U-L-R-",
+            # Hold-Tight Hoe-Down/Concerto of Charge
+            "R-R-U-U-",
+            # Melody with a Bounce/Jingle of Jump
+            "D-D-U-U-",
+            # Ballad of 1999/Pizzicato of Party
+            "L-R-D-U-",
+            # Song of Miracles (Djinn)
+            "D-DD-DD-",
+            # Leisurely Lullaby
+            # Not sure why you'd want to cast this on repeat, but you can I guess
+            "L-R-L-R-",
+            # Step Back Strut
+            "U-L-U-L-"
+    ]
+    # A mapping of characters in our encoded songs to keys to send to ydotool
+    # These should be the default bindings for PPSSPP
+    keymapping = {
+            "U": "s",
+            "D": "z",
+            "L": "a",
+            "R": "x"
+    }
+    drummapping = {
+            "U": '\033[32mCHAKA\033[0m',
+            "D": '\033[33mDON\033[0m',
+            "L": '\033[31mPATA\033[0m',
+            "R": '\033[34mPON\033[0m'
+    }
+    parser = argparse.ArgumentParser(
+        description="Play a sequence of Patapon commands on repeat forever"
+        )
+    parser.add_argument('song',default="R-R-L-R-",nargs="+",choices=songs,help="The song to play. Defaults to Aria of Attack. When expressing a song, use eighth notes, dashes, and cardinal directions to designate the drums. For example, party would be \"L-R-D-U-\", and djinn would be \"D-DD-DD-\"")
+    parser.add_argument('--bpm',type=int,default=120,help="The BPM of Patapon. Change if you're running the game at a higher speed")
+    parser.add_argument('--iterations',type=int,default=10000,help="Number of iterations of the sequence to run. Defaults to 10,000")
+    args = parser.parse_args()
+
+    # Schedule out our beat interval
+    beat_interval = 60 / (args.bpm * 2)
+    print(f"Beat interval: {beat_interval}")
+
+    # Set up the environment
+    sequence='--------' + '--------'.join(args.song)
+    print(f"Song sequence: {sequence}")
+    remaining_iterations = args.iterations
+    lastbeat = 0
+
+    # Wait for user confirmation
+    input("Press enter on-beat to sync up with Patapon...")
+    synctime = time.perf_counter()
+
+    # Play da notes
+    while remaining_iterations > 0:
+        for i, key in enumerate(sequence):
+            while time.perf_counter() < synctime + lastbeat:
+                pass
+            lastbeat += beat_interval
+            button = keymapping.get(key, '-')
+            if button != '-':
+                ydotool(button)
+                print(drummapping.get(key, '-'), end="", flush=True)
+            else:
+                print(" ", end="", flush=True)
+        remaining_iterations -= 1
+        print(f"~ ({remaining_iterations} remaining)")
+
+main()