From 974c1764d4ebac4781b1e05a04c9b3f97c3cd465 Mon Sep 17 00:00:00 2001
From: Eason <30045503+Eason0729@users.noreply.github.com>
Date: Thu, 25 Apr 2024 17:45:26 +0800
Subject: [PATCH] add pyr binding

---
 ml/ml_play_pyr_test.py | 122 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 122 insertions(+)
 create mode 100644 ml/ml_play_pyr_test.py

diff --git a/ml/ml_play_pyr_test.py b/ml/ml_play_pyr_test.py
new file mode 100644
index 0000000..14232ed
--- /dev/null
+++ b/ml/ml_play_pyr_test.py
@@ -0,0 +1,122 @@
+import random
+from pprint import pprint
+
+import orjson
+import pygame
+
+from ctypes import *
+
+libpyr = cdll.LoadLibrary("./pyr/target/release/libpyr.so")
+app_ptr = c_ulonglong(0)
+
+class Overall(Structure):
+    _fields_ = [
+        ("frame", c_uint64),
+        # ("collision_mode", c_bool),
+        ("score", c_int64),
+        ("score_to_pass", c_int64),
+        ("self_x", c_int64),
+        ("self_y", c_int64),
+        ("self_w", c_int64),
+        ("self_h", c_int64),
+        ("self_vel", c_int64),
+        ("self_lv", c_int64),
+        ("opponent_x", c_int64),
+        ("opponent_y", c_int64),
+        ("opponent_lv", c_int64),
+    ]
+
+class Food(Structure):
+    _fields_= [
+        ("h", c_int64),
+        ("w", c_int64),
+        ("x", c_int64),
+        ("y", c_int64),
+        ("score", c_int64),
+        ("kind", c_int32),
+    ]
+
+def new_app():
+    global app_ptr
+    app_ptr=c_ulonglong(libpyr.new_app())
+
+new_app()
+
+def drop_app():
+    global app_ptr
+    libpyr.drop_app(app_ptr)
+
+def check_point():
+    global app_ptr
+    libpyr.check_point(app_ptr)
+
+def tick(scene_info: dict):
+    overall=Overall()
+    overall.frame=c_uint64(scene_info["frame"])
+    overall.score=c_int64(scene_info["score"])
+    overall.score_to_pass=c_int64(scene_info["score_to_pass"])
+    overall.self_x=c_int64(scene_info["self_x"])
+    overall.self_y=c_int64(scene_info["self_y"])
+    overall.self_w=c_int64(scene_info["self_w"])
+    overall.self_h=c_int64(scene_info["self_h"])
+    overall.self_vel=c_int64(scene_info["self_vel"])
+    overall.self_lv=c_int64(scene_info["self_lv"])
+    overall.opponent_x=c_int64(scene_info["opponent_x"])
+    overall.opponent_y=c_int64(scene_info["opponent_y"])
+    overall.opponent_lv=c_int64(scene_info["opponent_lv"])
+
+    n=len(scene_info["foods"])
+
+    foods= (Food * n)()
+    for i in range(n):
+        foods[i]=Food()
+        foods[i].h=c_int64(scene_info["foods"][i]["h"])
+        foods[i].w=c_int64(scene_info["foods"][i]["w"])
+        foods[i].x=c_int64(scene_info["foods"][i]["x"])
+        foods[i].y=c_int64(scene_info["foods"][i]["y"])
+        foods[i].score=c_int64(scene_info["foods"][i]["score"])
+        kind=scene_info["foods"][i]["type"]
+        if kind=="FOOD_1":
+            foods[i].kind=c_int(1)
+        elif kind=="FOOD_2":
+            foods[i].kind=c_int(2)
+        elif kind=="FOOD_3":
+            foods[i].kind=c_int(3)
+        elif kind=="GARBAGE_1":
+            foods[i].kind=c_int(4)
+        elif kind=="GARBAGE_2":
+            foods[i].kind=c_int(5)
+        elif kind=="GARBAGE_3":
+            foods[i].kind=c_int(6)
+    
+    result=libpyr.tick(app_ptr,pointer(overall),pointer(foods),c_uint64(n))
+    if result==1:
+        return ["UP"]
+    if result==2:
+        return ["DOWN"]
+    if result==3:
+        return ["LEFT"]
+    if result==4:
+        return ["RIGHT"]
+    return ["NONE"]
+
+class MLPlay:
+    def __init__(self,ai_name,*args,**kwargs):
+        print("Initial ml script")
+
+    def update(self, scene_info: dict, keyboard:list=[], *args, **kwargs):
+        """
+        Generate the command according to the received scene information
+        """
+
+        if scene_info["status"] == "GAME_ALIVE":
+            return tick(scene_info)
+        else:
+            check_point()
+        
+    def reset(self):
+        """
+        Reset the status
+        """
+        print("reset ml script")
+        pass