Initial commit

pyground/sugar/SugarRushSimulator.py

@@ -0,0 +1,173 @@
+import json
+import random
+from typing import List, Dict, Tuple, Optional
+import time
+
+from SugarRush1000 import SugarRush1000
+
+"""
+此程序主要模拟用户使用过程中的体验指标
+
+返奖率 RTP = 总赢 / 总投注
+中奖率/级联率
+    平均多久一次获取一次连线 = 带连线的旋转次数 / 总旋转次数
+免费旋转率
+    平均多久一次获取一次免费旋转 = 命中免费旋转次数 / 总旋转次数(不算免费旋转)
+    平均免费旋转次数 = 总免费旋转次数 / 总旋转次数
+"""
+
+
+class SugarRushSimulator:
+
+    def __init__(self, buy_type: Optional[str] = None):
+        self.buy_type = buy_type
+        self.f = open(
+            f"simulator_log_{'normal' if buy_type is None else buy_type}_{time.time()}.log",
+            "w",
+        )
+
+    def simulate_batch(self, weights, scotter_counts_weights, spins: int) -> Dict:
+        """
+        模拟指定次数的旋转
+        :param spins: 模拟次数
+        :param buy_type: None=普通混合模式, 'standard'=只测买普通, 'super'=只测买超级
+        :return: 统计结果
+        """
+
+        begin_balance = 1_0000_0000
+        game = SugarRush1000(
+            balance=begin_balance,
+            weights=weights,
+            scotter_counts_weights=scotter_counts_weights,
+        )
+
+        total_bet = 0.0
+        total_win = 0.0
+        game_spins = []
+        for index in range(spins):
+            spin = {
+                "gid": index + 1,
+                "score": 0,
+                "count": 0,
+                "feature": "normal" if self.buy_type is None else self.buy_type,
+                "extra_free": 0,
+                "steps": [],
+            }
+            game_spins.append(spin)
+
+            if self.buy_type and self.buy_type != "normal":
+                r = game.buy_free_spins(self.buy_type)
+                total_bet += r["cost"]
+
+            can_spins = 1
+            free_spind_id = 0
+            aes = 0
+            while can_spins > 0:
+                can_spins -= 1
+
+                res = game.doSpin()
+                game.mock_grid = None
+                if res["error"]:
+                    raise Exception("模拟器异常", res["error"])
+                if res["free_spins_remaining"] >= 0:
+                    can_spins = res["free_spins_remaining"]
+
+                actual_cost = res["actual_bet"]
+                if actual_cost > 0:
+                    total_bet += actual_cost
+
+                free_spind_id += 1
+
+                for step in res["steps"]:
+                    step["gid"] = spin["gid"]
+                    step["free_spin_id"] = free_spind_id
+                    step["aes"] = aes
+                    aes += 1
+                    spin["steps"].append(step)
+                if res["extra_free"]:
+                    spin["extra_free"] += 1
+
+            spin["score"] = res["spin_total_win"]
+            spin["count"] = len(spin["steps"])
+
+            total_win += spin["score"]
+
+            if len(game_spins) >= 100:
+                self.write_log(game_spins)
+                game_spins = []
+
+        if len(game_spins) > 0:
+            self.write_log(game_spins)
+            game_spins = []
+        print(f"旋转{spins} 次，RTP {(total_win / total_bet) * 100}")
+        return {}
+
+    def write_log(self, games):
+        buf = ""
+        for game in games:
+            buf += json.dumps(game, ensure_ascii=False) + "\n"
+
+        self.f.write(buf)
+        self.f.flush()
+
+
+if __name__ == "__main__":
+    scotter_count_weights = {
+        "3": 7.26421894353717,
+        "4": 4.1724734692682395,
+        "5": 0.8119106579617028,
+        "6": 0.20313929837878217,
+        "7": 0.04857599989214818,
+    }
+    batchs = [
+        # {
+        #     "buy_type": "normal",
+        #     "spins": 20000,
+        #     "weights": {
+        #         "A": 16.559326336631095,
+        #         "B": 17.242437197138468,
+        #         "C": 23.794844051571708,
+        #         "D": 24.480364769888073,
+        #         "E": 31.29327382410323,
+        #         "F": 38.17160528628571,
+        #         "G": 17.242437197138468,
+        #         "S": 2.2546844695944754,
+        #     },
+        #     "scotter_count_weights": scotter_count_weights,
+        # },
+        {
+            "buy_type": "standard",
+            "spins": 5000,
+            "weights": {
+                "A": 18.92167765262542,
+                "B": 21.106974838197058,
+                "C": 31.250953223681833,
+                "D": 34.54623003651251,
+                "E": 47.52175529241897,
+                "F": 62.474728710698884,
+                "G": 21.106974838197058,
+                "S": 2.666109542063759,
+            },
+            "scotter_count_weights": scotter_count_weights,
+        },
+        # {
+        #     "buy_type": "super",
+        #     "spins": 5000,
+        #     "weights": {
+        #         "A": 19.015957779792195,
+        #         "B": 21.291015318701493,
+        #         "C": 31.66660200727613,
+        #         "D": 35.193596023259865,
+        #         "E": 48.7122724047052,
+        #         "F": 64.49005324700025,
+        #         "G": 21.291015318701493,
+        #         "S": 2.6840958157151236,
+        #     },
+        #     "scotter_count_weights": scotter_count_weights,
+        # },
+    ]
+    for batch in batchs:
+        sim = SugarRushSimulator(buy_type=batch["buy_type"])
+        sim.simulate_batch(
+            batch["weights"], batch["scotter_count_weights"], batch["spins"]
+        )