def _run_and_plot(self, plot_index, num_plots, name, match_maker, environment):
engine = Engine(match_maker, environment)
for i in range(self.num_rounds):
engine.one_round()
target_players = engine.players_with_mmr_between(self.min_mmr, self.max_mmr)
stats = engine.statistics(target_players)
Runner._print_stats(name, stats)
Runner._plot(plot_index, num_plots, stats)
class Demo(OnGameFinishedListener, OnLobbyFoundListener):
def __init__(self, width=800, height=600, bar_height=10, wait_ms=100, bg_color=(50, 50, 50), text_color=(255, 255, 255)):
pygame.init()
self.width = width
self.height = height
self.bar_height = bar_height
self.screen = pygame.display.set_mode((self.width, self.height))
self.engine = None
self.environment = None
self.wait_ms = wait_ms
self.bg_color = bg_color
self.font = pygame.font.Font(None, 22)
self.text_color = text_color
self.num_playing = 0
self._currently_skipping_rounds = False
def on_lobby_found(self, team_1: List[Queuer], team_2: List[Queuer]) -> None:
self.num_playing += TEAM_SIZE * 2
avg_mmr_diff = int(avg_mmr([q.player for q in team_2]) - avg_mmr([q.player for q in team_1]))
max_mmr_d = max_mmr_diff([q.player for q in team_1], [q.player for q in team_2])
names = [q.player.name for q in team_1 + team_2]
skills = [self.environment.get_player_skill(name) for name in names]
max_skill_d = max(skills) - min(skills)
if not self._currently_skipping_rounds:
print("New game: max diff: " + str(max_mmr_d) +
", avg diff: " + str(avg_mmr_diff) +
", max skill diff: " + str(max_skill_d) +
", avg wait: " + str(avg([q.waited for q in team_1 + team_2])))
def on_game_finished(self, game: Game) -> None:
self.num_playing -= TEAM_SIZE * 2
def run(self, match_maker: MatchMaker, mmr_engine: MmrEngine, environment: Environment, skip_rounds: int = 0):
self.engine = Engine(match_maker, mmr_engine, environment)
self.engine._on_game_finished_listeners.append(self)
self.engine._on_lobby_found_listeners.append(self)
self.environment = environment
self._main_loop(skip_rounds)
def _main_loop(self, skip_rounds: int):
self._currently_skipping_rounds = True
for i in range(skip_rounds):
self.engine.one_round()
self._currently_skipping_rounds = False
self._render()
pygame.time.wait(1000)
while True:
for event in pygame.event.get():
if event.type == pygame.QUIT:
sys.exit()
elif event.type == pygame.KEYDOWN:
if event.key == pygame.K_DOWN:
if self.wait_ms == 0:
self.wait_ms = 2
else:
self.wait_ms = int(self.wait_ms * 1.5)
elif event.key == pygame.K_UP:
self.wait_ms = int(self.wait_ms / 1.5)
self.engine.one_round()
pygame.time.wait(self.wait_ms)
self._render()
def _render(self):
pygame.draw.rect(self.screen, self.bg_color, (0, 0, self.width, self.height))
for i, queuer in enumerate(self.engine.queue()):
width_divider = 8
w = queuer.player.mmr / width_divider
line_x = self.environment.get_player_skill(queuer.player.name) / width_divider
line_color = (255, 255, 255)
x = 10
space = 1
top_space = 50
y = top_space + i * (self.bar_height + space)
rect = (x, y, w, self.bar_height)
if queuer.waited < 255:
color = (queuer.waited, 0, 255-queuer.waited)
else:
color = (255, 0, 0)
pygame.draw.rect(self.screen, color, rect)
pygame.draw.rect(self.screen, line_color, (x + line_x, y, 3, self.bar_height))
longest_wait = 0
if len(self.engine.queue()) > 0:
longest_wait = self.engine.queue()[0].waited
self._render_text(str(len(self.engine.queue())) + " queueing", 10)
self._render_text("longest: " + str(longest_wait) + "s", 150)
self._render_text(str(len(self.engine._data_store.replays)) + " games played", 310)
self._render_text(str(self.wait_ms) + "ms / round", 450)
self._render_text(str(self.num_playing) + " playing", 610)
pygame.display.flip()
def _render_text(self, msg, x):
text = self.font.render(msg, 1, self.text_color)
textpos = text.get_rect()
textpos.x = x
textpos.y = 10
self.screen.blit(text, textpos)
