streak-hunting-spreadsheet/endgames.py

import json
import re
import subprocess
from typing import Dict

from games import GAMES_PATH

from pathlib import Path
DATA_FILE = Path('endgame-data.json')

if not DATA_FILE.exists():
    DATA_FILE.write_text('{}')


with open(DATA_FILE, 'r') as f:
    DATA: Dict = json.loads(f.read())


def analyze_game(filename: str):
    max_draw_pile_size = 15
    try:
        result = subprocess.run(['./endgame-analyzer', '-f', filename, '-d', str(max_draw_pile_size), '--interactive', '0', '--quiet', '-r'], stdout=subprocess.PIPE, timeout=60*15)
        raw_output = result.stdout
    except subprocess.TimeoutExpired as time_err:
        raw_output = time_err.stdout
    output = raw_output.decode('utf-8')

    # Now, parse all results that we obtained (unclear how many depending on whether we ran into the timeout)
    probabilities = {}

    m = re.search(r"Specified draw pile size of ([0-9]+) cannot be reached with specified replay\.\nReplay ends at turn [0-9]+ with score of ([0-9]+)\.", output, re.M)
    if m:
        won = '100' if m.group(2) == '25' else 0
        for draw_pile_size in range(1, int(m.group(1)) + 1):
            probabilities[str(draw_pile_size)] = won

    for m in re.finditer('Probability with ([0-9]+) cards left in deck: .*/.* ~ ([0-9.]+)', output):
        probabilities[str(m.group(1))] = m.group(2)

    return probabilities


def full_analyze_game(filename: str):
    max_draw_pile_size = 10
    try:
        result = subprocess.run(['./endgame-analyzer', '-f', filename, '-d', str(max_draw_pile_size), '-i', '0', '--all-clues', '-r', '--quiet'], stdout=subprocess.PIPE, timeout=180)
        raw_output = result.stdout
    except subprocess.TimeoutExpired as time_err:
        raw_output = time_err.stdout
    output = raw_output.decode('utf-8')

    probabilities = {}
    zero_dict = {
        (('+' if clue_modifier >= 0 else '') + str(clue_modifier)): 0 for clue_modifier in range(-8, 9)
    }
    hundred_dict = {
        (('+' if clue_modifier >= 0 else '') + str(clue_modifier)): 100 for clue_modifier in range(-8, 9)
    }

    m = re.search(r"Specified draw pile size of ([0-9]+) cannot be reached with specified replay\.\nReplay ends at turn [0-9]+ with score of ([0-9]+)\.", output, re.M)
    if m:
        won = hundred_dict if m.group(2) == '25' else zero_dict
        for draw_pile_size in range(1, int(m.group(1)) + 1):
            probabilities[str(draw_pile_size)] = won

    for m in re.finditer('Probability with ([0-9]+) cards left in deck and [0-8] clues \((.[0-8])\).*: .*/.* ~ ([0-9.]*)', output):
        if m.group(1) not in probabilities.keys():
            probabilities[m.group(1)] = {}
        probabilities[m.group(1)][m.group(2)] = m.group(3)
    return probabilities


def full_analyze_game_cached(game_id: int, cheat: bool = False):
    cached = DATA.get('all', {}).get(str(game_id), None)
    if cached is not None:
        return cached
    result = full_analyze_game(str(GAMES_PATH / str(game_id)) + '-cheat' if cheat else '')
    key = 'all' if not cheat else 'all-cheat'
    if key not in DATA.keys():
        DATA[key] = {}
    DATA[key][game_id] = result
    save_cache()
    return result

def analyze_game_cached(game_id: int, cheat: bool = False):
    cached = DATA.get('normal', {}).get(str(game_id), None)
    if cached is not None:
        return cached
    result = analyze_game(str(GAMES_PATH / str(game_id)) + '-cheat' if cheat else '')
    key = 'normal' if not cheat else 'normal-cheat'
    if key not in DATA.keys():
        DATA[key] = {}
    DATA[key][game_id] = result
    save_cache()
    return result


def save_cache():
    with open(DATA_FILE, 'w') as f:
        f.writelines(json.dumps(DATA, indent=2))