import sys
import os
sys.path.append(os.path.join(os.path.dirname(__file__), '..'))

from src.cfr.info_set import InfoSet

def demo_strategy():

    info_set = InfoSet(act_cnt=2)

    turn = [
        ([10.0, 5.0], 1.0, "BT 0"),
        ([15.0, 20.0], 1.0, "CL 0"),
        ([9.0, 25.0], 1.0, "CL 1"),
        ([5.0, 30.0], 1.0, "CL 1"),
    ]

    for reg, wgt, str in turn:
        info_set.regret_sum = reg[:]
        cur_stra = info_set.get_strat(wgt)
        avg_stra = info_set.get_avg_strat()

    print(f"{str}")
    print(f"  sum_regret: {reg}")
    print(f"  cur_strategy: [CK/FD: {cur_stra[0]:.3f}, BT/CL: {cur_stra[1]:.3f}]")
    print(f"  avg_strategy: [CK/FD: {avg_stra[0]:.3f}, BT/CL: {avg_stra[1]:.3f}]")
    print()

if __name__ == "__main__":
    demo_strategy()