allow different ports and addrs

misko · Oct 8, 2024 · 82a56a8 · 82a56a8
1 parent 804bd2f
commit 82a56a8
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/hudes/hudes_client.py b/hudes/hudes_client.py
@@ -19,6 +19,8 @@
 class HudesClient:
     def __init__(
         self,
+        addr,
+        port,
         step_size_resolution=-0.05,
         inital_step_size_idx=10,
         seed=0,
@@ -30,7 +32,7 @@ def __init__(
         self.quit_count = 0
         self.seed = seed
 
-        self.hudes_websocket_client = HudesWebsocketClient("ws://localhost:8765")
+        self.hudes_websocket_client = HudesWebsocketClient(f"ws://{addr}:{port}")
 
         self.request_idx = 0
 

diff --git a/hudes/hudes_play.py b/hudes/hudes_play.py
@@ -15,14 +15,18 @@ def main():
     parser.add_argument("--seed", type=int, default=0)
     parser.add_argument("--grid-size", type=int, default=41)
     parser.add_argument("--grids", type=int, default=2)
+    parser.add_argument("--addr", type=str, default="localhost")
+    parser.add_argument("--port", type=int, default=8765)
 
     args = parser.parse_args()
 
     if args.input == "keyboard":
-        controller = KeyboardClient(seed=args.seed)
+        controller = KeyboardClient(addr=args.addr, port=args.port, seed=args.seed)
         view = View()
     elif args.input == "keyboardGL":
         controller = KeyboardClientGL(
+            addr=args.addr,
+            port=args.port,
             seed=args.seed,
             step_size_resolution=-0.005,
             inital_step_size_idx=100,
@@ -34,7 +38,7 @@ def main():
     #     controller = AkaiClient(seed=args.seed)
     #     view = View()
     elif args.input == "xtouch":
-        controller = XTouchClient(seed=args.seed)
+        controller = XTouchClient(addr=args.addr, port=args.port, seed=args.seed)
         view = View(use_midi=True)
     else:
         raise ValueError