#!/usr/bin/env python3 import numpy as np from typing import List, Dict, Tuple import itertools from poker.card import Card, ShortDeckRank, Suit from shortdeck.hand_evaluator import HandEvaluator class ShortDeckHistGenerator: def __init__(self): # （36张牌：6-A） self.full_deck = [] for rank in ShortDeckRank: for suit in Suit: self.full_deck.append(Card(rank, suit)) self.hand_evaluator = HandEvaluator() print(f"初始化短牌型EHS直方图生成器，牌组大小: {len(self.full_deck)}") print(f"牌型范围: {ShortDeckRank.SIX.name}-{ShortDeckRank.ACE.name}") def generate_river_ehs(self, player_cards, board_cards) -> float: """ River阶段胜率计算（确定性结果） 5张公共牌已知，直接计算对所有可能对手的胜率 player_cards: 玩家的2张底牌 board_cards: 5张公共牌 Returns: 胜率值 (0-1之间) """ if len(player_cards) != 2: raise ValueError("玩家必须有2张底牌") if len(board_cards) != 5: raise ValueError("River阶段必须有5张公共牌") # 计算玩家的7张牌牌力 player_7_cards = player_cards + board_cards player_strength = self.hand_evaluator.evaluate_hand(player_7_cards) # 获取剩余的牌 used_cards = player_cards + board_cards remaining_cards = [card for card in self.full_deck if card not in used_cards] # 计算对所有可能对手的胜率 wins = 0 total_player2s = 0 # 遍历所有可能的对手底牌组合 for player2_cards in itertools.combinations(remaining_cards, 2): player2_7_cards = list(player2_cards) + board_cards player2_strength = self.hand_evaluator.evaluate_hand(player2_7_cards) if player_strength > player2_strength: wins += 2 elif player_strength == player2_strength: wins += 1 total_player2s += 2 ehs = wins / total_player2s return ehs def generate_turn_histogram(self, player_cards, board_cards, num_bins: int = 30) -> np.ndarray: """ player_cards: 玩家的2张底牌 board_cards: 4张公共牌 num_bins: 30个bin（短牌型应该是30） """ if len(player_cards) != 2: raise ValueError("玩家必须有2张底牌") if len(board_cards) != 4: raise ValueError("Turn阶段必须有4张公共牌") # 获取剩余的牌（可作为River牌） used_cards = player_cards + board_cards remaining_cards = [card for card in self.full_deck if card not in used_cards] ehs_values = [] histogram = [] # 对每张可能的River牌计算EHS for river_card in remaining_cards: full_board = board_cards + [river_card] ehs = self.generate_river_ehs(player_cards, full_board) # ehs_values.append(ehs) histogram.append(ehs) if (len(histogram) != num_bins): print(f" Turn计算: generate nums{len(histogram)} ,num_bins={num_bins}") raise # 生成直方图 # histogram, _ = np.histogram(ehs_values, bins=num_bins, range=(0, 1)) # 归一化 # histogram = histogram.astype(float) # if histogram.sum() > 0: # histogram /= histogram.sum() return histogram def generate_flop_histogram(self, player_cards, board_cards, num_bins: int = 465) -> np.ndarray: """ Flop阶段EHS直方图 3张公共牌已知，枚举所有可能的Turn+River牌组合计算EHS分布 player_cards: 玩家的2张牌 board_cards: 3张公共牌 num_bins: 直方图bin数量（短牌型应该是C(36-5,2) = 465） """ if len(player_cards) != 2: raise ValueError("玩家必须有2张牌") if len(board_cards) != 3: raise ValueError("Flop阶段必须有3张公共牌") # 获取剩余的牌 used_cards = player_cards + board_cards remaining_cards = [card for card in self.full_deck if card not in used_cards] ehs_values = [] histogram = [] # 枚举所有可能的Turn+River组合：C(31,2) = 465 for turn_river_combo in itertools.combinations(remaining_cards, 2): turn_card, river_card = turn_river_combo full_board = board_cards + [turn_card, river_card] ehs = self.generate_river_ehs(player_cards, full_board) # ehs_values.append(ehs) histogram.append(ehs) # # 验证组合数 # expected_combinations = len(list(itertools.combinations(remaining_cards, 2))) # print(f" Flop计算: C({len(remaining_cards)},2) = {expected_combinations} 种组合") # 生成直方图 # histogram, _ = np.histogram(ehs_values, bins=num_bins, range=(0, 1)) # 归一化 # histogram = histogram.astype(float) # if histogram.sum() > 0: # histogram /= histogram.sum() if (len(histogram) != num_bins): print(f" Turn计算: generate nums{len(histogram)} ,num_bins={num_bins}") raise return histogram # # 从一副牌中抽样验证 或者 从解析的数据中抽样验证 # def generate_sample_data(self, num_samples: int = 3) -> Dict: # print(f"\n 生成样本数据 (每阶段 {num_samples} 个样本)") # results = { # 'river': [], # 'turn': [], # 'flop': [] # } # for i in range(num_samples): # # River样本（7张牌） # river_cards = np.random.choice(self.full_deck, size=7, replace=False).tolist() # player_cards = river_cards[:2] # board_cards = river_cards[2:7] # river_ehs = self.generate_river_ehs(player_cards, board_cards) # results['river'].append({ # 'player_cards': player_cards, # 'board_cards': board_cards, # 'ehs': river_ehs # }) # # Turn样本（6张牌） # turn_cards = np.random.choice(self.full_deck, size=6, replace=False).tolist() # player_cards = turn_cards[:2] # board_cards = turn_cards[2:6] # turn_hist = self.generate_turn_histogram(player_cards, board_cards, num_bins=30) # results['turn'].append({ # 'player_cards': player_cards, # 'board_cards': board_cards, # 'histogram': turn_hist, # 'mean': float(np.mean(turn_hist)), # 'std': float(np.std(turn_hist)) # }) # # Flop样本（5张牌） # flop_cards = np.random.choice(self.full_deck, size=5, replace=False).tolist() # player_cards = flop_cards[:2] # board_cards = flop_cards[2:5] # flop_hist = self.generate_flop_histogram(player_cards, board_cards, num_bins=465) # results['flop'].append({ # 'player_cards': [str(c) for c in player_cards], # 'board_cards': [str(c) for c in board_cards], # 'histogram_stats': flop_hist, # 'mean': float(np.mean(flop_hist)), # 'std': float(np.std(flop_hist)) # }) # return results # def main(): # """测试短牌型EHS直方图生成器""" # print("短牌型EHS直方图生成器测试") # print("="*60) # generator = ShortDeckHistGenerator() # # 测试River EHS # print(f"\n测试River EHS计算...") # test_cards = np.random.choice(generator.full_deck, size=7, replace=False).tolist() # player_cards = test_cards[:2] # board_cards = test_cards[2:7] # player_str = " ".join(str(c) for c in player_cards) # board_str = " ".join(str(c) for c in board_cards) # river_ehs = generator.generate_river_ehs(player_cards, board_cards) # print(f" 玩家底牌: [{player_str}]") # print(f" 公共牌: [{board_str}]") # print(f" River EHS: {river_ehs:.4f}") # # 测试Turn直方图 # print(f"\n测试Turn直方图生成...") # turn_cards = np.random.choice(generator.full_deck, size=6, replace=False).tolist() # player_cards = turn_cards[:2] # board_cards = turn_cards[2:6] # turn_hist = generator.generate_turn_histogram(player_cards, board_cards, num_bins=30) # # 测试Flop直方图 # print(f"\n测试Flop直方图生成...") # flop_cards = np.random.choice(generator.full_deck, size=5, replace=False).tolist() # player_cards = flop_cards[:2] # board_cards = flop_cards[2:5] # flop_hist = generator.generate_flop_histogram(player_cards, board_cards, num_bins=465) # print(f" Flop直方图: mean={np.mean(flop_hist):.3f}, 非零bins={np.count_nonzero(flop_hist)}") # # 生成样本数据 # print(f"\n生成样本数据...") # sample_data = generator.generate_sample_data(num_samples=3)