bracket.py

import re
from math import ceil, floor

bracket = eval(file("teams.dat").read())

teamre = '\s*(\d+) <a href="[^"]+">([^<]+)</a>\s*<a href=.*?</a></a>\s*(\d+-\d+)\s*(\.\d+)\s*([\d.]*)\/(\d+)\s*([\d.]*)\/(\d+)'
kenpom = [x.groups() for x in [re.search(teamre, line) for line in file("kenpom.html")] if x]
#kenpom is a list of tuples:
#(rank, name, record, pythagorean score, adjusted O, adj O rank, adjusted D, adj D rank)
kenpom = dict((x[1], x) for x in kenpom)

#make sure we agree on all our team names
for r in bracket:
  for t in bracket[r].values():
    if type(t) == type([]):
      for t2 in t:
        assert(t2 in kenpom)
    else:
      assert(t in kenpom)

#let's build a dictionary {"region" -> {seed -> [name, pyth, adjo, adjd]}}
merged = {}
for region in ("Midwest", "West", "East", "South"):
  merged[region] = {}
  for seed, teamname in bracket[region].items():
    if type(teamname) == type([]):
      t1, t2 = teamname 
      rank, name, record, pyth, adjo, adjorank, adjd, adjdrank = kenpom[t1]
      merged[region][16] = [name, pyth, adjo, adjd]
      rank, name, record, pyth, adjo, adjorank, adjd, adjdrank = kenpom[t2]
      merged[region][17] = [name, pyth, adjo, adjd]
    else:
      rank, name, record, pyth, adjo, adjorank, adjd, adjdrank = kenpom[teamname]
      merged[region][seed] = [name, pyth, adjo, adjd]

class Game(object):
  def __init__(self, region, round, gameno, seed1=None, team1=None, seed2=None, team2=None):
    self.region = region
    self.round = round
    self.gameno = gameno
    self.seed1 = seed1 or ""
    self.team1 = team1 or [""]
    self.seed2 = seed2 or ""
    self.team2 = team2 or [""]
    self.child = None
    self.rows = [None, None]

  def __repr__(self):
    return "%s vs %s round %s game %s region %s rows %s" % (self.team1[0],
      self.team2[0], self.round, self.gameno, self.region, self.rows)

games = []
game = 1
for region in merged:
  for seed in (1,8,5,4,6,3,7,2):
    oppseed = 17-seed
    t1 = merged[region][seed]
    t2 = merged[region][oppseed]
    games.append(Game(region, 1, game, seed, t1, oppseed, t2))
    game += 1

for round in (2,3,4):
  i = iter(g for g in games if g.round==round-1)
  roundg = []
  for g in i:
    g2 = i.next()
    gg = Game(g.region, round, game)
    roundg.append(gg)
    g.child = g2.child = gg
    game += 1
  games.extend(roundg)

rf1, rf2 = [g for g in games if g.round==4 and g.region in ("Midwest", "West")]
rf3, rf4 = [g for g in games if g.round==4 and g.region in ("East", "South")]
ff1 = Game("final four", 5, game)
ff2 = Game("final four", 5, game+1)
games.append(ff1)
games.append(ff2)
rf1.child = rf2.child = ff1
rf3.child = rf4.child = ff2

c = Game("final four", 6, game+2)
games.append(c)
ff1.child = ff2.child = c

winner = Game("final four", 7, game+3)
games.append(winner)
c.child = winner

class Table(object):
  def __init__(self):
    self.data = [[]]

  def __getitem__(self, (x,y)):
    return self.data[y][x]

  def __setitem__(self, (x,y), val):
    try:
      self.data[y][x] = val
    except IndexError:
      self.resize(x, y)
      self.data[y][x] = val
    except:
      print y, x
      raise

  #we're fine with *horrible* efficiency in this particular app
  def resize(self, x, y):
    for i in range(y - len(self.data) + 1):
      self.data.append([])
    for row in self.data:
      for i in range(x - len(row) + 1):
        row.append(None)

  def __repr__(self):
    return self.data.__repr__()

  def tableize(self):
    o = []
    for y, row in enumerate(self.data):
      o.append("<tr>")
      for x, elt in enumerate(row):
        if x == 0:
          #TODO: 16v17
          pass
        elif not elt: o.append('<td class="none"></td>')
        elif elt.rows[0] == y: 
          if x == 1:
            o.append('''<td class="top" width="200">%s. <span class="round-%s game-%s"
                            >%s (%s, %s, %s)</span></td>''' % (
              elt.seed1, elt.round, elt.gameno, elt.team1[0], elt.team1[1],
              elt.team1[2], elt.team1[3]))
          else:
            o.append('''<td class="top" width="180">&nbsp;<span
                            class="round-%s game-%s"></span>''' % (
              elt.round, elt.gameno))
        elif elt.rows[1] == y:
          if x in (0,1):
            o.append('''<td class="bottom">%s. <span class="round-%s game-%s"
                            >%s (%s, %s, %s)</span></td>''' % (
              elt.seed2, elt.round, elt.gameno, elt.team2[0], elt.team2[1],
              elt.team2[2], elt.team2[3]))
          else:
            o.append('''<td class="bottom">&nbsp;<span class="round-%s game-%s"></span>''' % (
              elt.round, elt.gameno))
        else:
          o.append('<td class="middle">&nbsp;</td>')
    return '\n'.join(o)
    
t = Table()

#insert the games into the table in the proper spot
row = 0
for region in ("Midwest", "West", "East", "South"):
  for g in [x for x in games if x.round==1 and x.region==region]:
    g.rows = [row, row+2]
    t[1, row] = g
    t[1, row+1] = g
    t[1, row+2] = g
    g.child.rows[1 if g.child.rows[0] else 0] = row+1
    row += 3

  for round in (2,3,4):
    for g in [x for x in games if x.round==round and x.region==region]:
      miny, maxy = g.rows
      for i in range(0, maxy - miny + 1):
        t[round, miny + i] = g
        g.child.rows[1 if g.child.rows[0] else 0] = miny + int(ceil(float(maxy-miny)/2))

ff1.rows = [12, 36]
for i in range(12,37): t[5, i] = ff1
ff2.rows = [60, 84]
for i in range(60,85): t[5, i] = ff2
c.rows = [25, 72]
for i in range(25,73): t[6, i] = c
winner.rows = [50, 51]
t[7, 50] = winner

out = file("out.html", "w")
#TODO: if you change a team in an early game, update later games
#TODO: color teams based on pythagorean diff
#TODO: better randomizer
out.write("""
<html><head>
<script src="jquery-1.3.2.min.js" type="text/javascript"></script>
<script>
rounds = {0:0, 1: 32, 2: 48, 3:56, 4:60, 5:62, 6:63};

function handleClick(that) {
  var atts = that.attr("class");
  var game = parseFloat(atts.match(/game-(\d+)/)[1]);
  var round = parseFloat(atts.match(/round-(\d+)/)[1]);
  var game_parity = (game - rounds[round-1]) % 2 == 0 ? 1 : 0;
  if (round < 4) {
    var nextgame = (Math.ceil((game-rounds[round-1])/2) + rounds[round]).toString();
    var firstorlast = game_parity ? ":last" : ":first";
  }
  else {
    if (game == 59 || game==57) {
      var nextgame = 61;
      var firstorlast = game == 59 ? ":first" : ":last";
    }
    else if (game==58 || game==60) {
      var nextgame = 62;
      var firstorlast = game == 58 ? ":first" : ":last";
    }
    else if (game == 61 || game == 62) {
      var nextgame = 63;
      var firstorlast = game == 61 ? ":first" : ":last";
    }
    else if (game == 63) {
      var nextgame = 64;
      var firstorlast = ":first";
    }
  }
  that.click(function() {
    //console.log("game, round, nextgame, (g-r[r-1]) ", game, round, nextgame, game - rounds[round-1], firstorlast);
    $(".game-" + nextgame + firstorlast).html(that.html());
  });
}

function randomize() {
  for (i=0; i < 7; i++) {
    $("td.top > span.round-" + i).each(function(_) {
      var that = $(this);
      var atts = that.attr("class");
      var game = atts.match(/game-\d+/)[0];
      var opp = $("." + game + ":last");
      function parsepoints(obj) {
        p = obj.html().match(/(.*?) \((.\d+), (\d+\.\d+), (\d+\.\d+)/);
        return [p[1], parseFloat(p[2]), parseFloat(p[3]), parseFloat(p[4])]
      }
      var topp = parsepoints(that);
      var oppp = parsepoints(opp);
      var a = topp[1];
      var b = oppp[1];

      //use the log5 formula
      var log5 = (a - a * b) / (a + b - 2 * a * b);

      if (log5 > .5) {
        that.css("color", "#00" + parseInt(255 * log5).toString(16) + "00");
        opp.css("color", "#" + parseInt(255 * log5).toString(16) + "0000");
      } else {
        opp.css("color", "#00" + parseInt(255 * log5).toString(16) + "00");
        that.css("color", "#" + parseInt(255 * log5).toString(16) + "0000");
      }

      console.log(topp[0] + " vs " + oppp[0] + " %: ", log5);

      log5 > Math.random() ? that.click() : opp.click();
    });
  }
}

$(document).ready(function() {
  for (i=0; i < 8; i++) {
    $(".round-" + i).each(function(i) { handleClick($(this)); });
  }
  $("td.top > span.round-1").each(function(i) { $(this).parent().css("height", "30px").css("vertical-align", "bottom"); })
  $("#randomize").click(function() { randomize(); });
});
</script>
<style>
.top {  border-bottom: 1px solid #aaaaaa; padding: 0px 5px 0px 5px; }
.bottom { border-bottom: 1px solid #aaaaaa; border-right: 1px solid #aaaaaa; padding: 0px 5px 0px 5px; }
.middle { border-right: 1px solid #aaaaaa; padding: 0px 5px 0px 5px; }
tr { padding-bottom: 10px; font-size: 12px; }
span { cursor: pointer; }
</style>
</head>
<body><input type="submit" id="randomize" value="randomize"><table cellspacing=0 width=1200 style='table-layout:fixed'>
""")
out.write(t.tableize())
out.write("</table></body></html>")
out.close()