wand125/rating-simulator.py

## rating-simulator.py
import math
import argparse

def calc_win_rate(a,b):
    return (1.0 / (math.pow(10,(a-b) / -400) + 1))

def calc_win_rate_final(a,b):
    p = calc_win_rate(a,b)
    return p * p * p + 3 * p * p * (1-p)

def calc_tournament(rating_tree, win_rate_func):
    def calc_info(r):
      return [{ 'rating': r, 'probability': 1.0 }] if isinstance(r, int) else calc_tournament(r,calc_win_rate)
    left = calc_info(rating_tree[0])
    right = calc_info(rating_tree[1])
    def calc_expected_value(p , opponents):
        return { 'rating': p['rating'], 'probability': sum([p['probability'] * o['probability'] * win_rate_func(p['rating'], o['rating']) for o in opponents])}
    return [calc_expected_value(l, right) for l in left] + [calc_expected_value(r, left) for r in right]

def __main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--ryuoh', '-r', action='store_true', default=False,
             help='set flag to ryuoh tournament mode')
    arg = parser.parse_args()
    l = eval(input())
    print("\n".join(["{0:d}: {1:.2f}%".format(e['rating'],e['probability'] * 100) for e in calc_tournament(l, calc_win_rate_final if arg.ryuoh else calc_win_rate)]))

if __name__ == '__main__':
    __main()
	import math
	import argparse

	def calc_win_rate(a,b):
	return (1.0 / (math.pow(10,(a-b) / -400) + 1))

	def calc_win_rate_final(a,b):
	p = calc_win_rate(a,b)
	return p * p * p + 3 * p * p * (1-p)

	def calc_tournament(rating_tree, win_rate_func):
	def calc_info(r):
	return [{ 'rating': r, 'probability': 1.0 }] if isinstance(r, int) else calc_tournament(r,calc_win_rate)
	left = calc_info(rating_tree[0])
	right = calc_info(rating_tree[1])
	def calc_expected_value(p , opponents):
	return { 'rating': p['rating'], 'probability': sum([p['probability'] * o['probability'] * win_rate_func(p['rating'], o['rating']) for o in opponents])}
	return [calc_expected_value(l, right) for l in left] + [calc_expected_value(r, left) for r in right]

	def __main():
	parser = argparse.ArgumentParser()
	parser.add_argument('--ryuoh', '-r', action='store_true', default=False,
	help='set flag to ryuoh tournament mode')
	arg = parser.parse_args()
	l = eval(input())
	print("\n".join(["{0:d}: {1:.2f}%".format(e['rating'],e['probability'] * 100) for e in calc_tournament(l, calc_win_rate_final if arg.ryuoh else calc_win_rate)]))

	if __name__ == '__main__':
	__main()