from datetime import datetime as d
import matplotlib.pyplot as plt

def lb_to_kg(lb):
  return lb*0.45359237

# Data (unknown day -> 1)
p_raw = [
(d(1964,  9, 1), lb_to_kg(710)), # 322.1
(d(1964,  9, 1), lb_to_kg(715)), # 324.4
(d(1965,  9, 1), lb_to_kg(725)), # 328.9
(d(1965,  9, 1), lb_to_kg(740)), # 335.6
(d(1967,  9, 1), lb_to_kg(784)), # 355.6
(d(1969,  8, 1), lb_to_kg(800)), # 362.9
(d(1971, 11, 1), lb_to_kg(820)), # 371.9
(d(1972,  9, 1), lb_to_kg(820)), # 371.9
(d(1972, 11, 1), lb_to_kg(845)), # 383.3
(d(1974, 11, 1), 385.0),
(d(1976,  8, 1), lb_to_kg(855)), # 387.8
(d(1976,  8, 1), lb_to_kg(860)), # 390.1
(d(1995,  7, 1), lb_to_kg(903)), # 409.6
(d(2005, 11, 1), 426.0),
(d(2005, 11, 1), 440.0),
(d(2010, 10, 1), 442.5),
(d(2011,  4, 1), lb_to_kg(1015)), #460.4
(d(2022,  3, 1), 465.0),
(d(2022,  7, 1), 467.5),
(d(2022,  7, 1), 487.5),
] 

p_eqp = [
(d(1972, 10, 1), lb_to_kg(882.5)), # 400.3
(d(1975,  5, 1), lb_to_kg(885.5)), # 401.6
(d(1981, 11, 1), lb_to_kg(886.0)), # 401.9
(d(1982, 12, 1), lb_to_kg(904.0)), # 410.0
(d(1986,  4, 1), lb_to_kg(904.0)), # 410.0
(d(1990,  7, 1), 417.5),
(d(1992,  3, 1), lb_to_kg(925.0)), # 419.6
(d(2002, 11, 1), 422.5),
(d(2003,  3, 1), 423.0),
(d(2005,  6, 1), 425.0),
(d(2005, 11, 1), 427.5),
(d(2006,  3, 1), 440.5),
(d(2006, 11, 1), 455.0),
(d(2009,  4, 1), 457.5),
]

s_all = [
(d(1996, 10, 1), 410.0),
(d(2005,  1, 1), 410.0),
(d(2008,  7, 1), 410.0),
(d(2011,  3, 1), 412.5),
(d(2011,  3, 1), 420.0),
(d(2011,  3, 1), 430.0),
(d(2011,  9, 1), 435.0),
(d(2011,  9, 1), 440.0),
(d(2013,  8, 1), 442.5),
(d(2014,  3, 1), 445.0),
(d(2014,  8, 1), 446.0),
(d(2014,  8, 1), 461.0),
(d(2015,  3, 1), 462.0),
(d(2015,  7, 1), 463.0),
(d(2016,  7, 1), 465.0),
(d(2016,  7, 1), 500.0),
(d(2020,  5, 1), 501.0),
(d(2025,  7, 1), 505.0),
]

# Plot
records = [p_raw, p_eqp, s_all]
colors  = ['tab:blue', 'tab:green', 'tab:red']
labels  = ['Powerlifting, raw', 'Powerlifting, equipped', 'Strongman']

fig = plt.figure()
fig.set_size_inches(12, 8) # aspect ratio = 2/3

for i, r in enumerate(records):
  # Append today as last value (for lines)
  today_kg = r[-1][1]
  today_date = d.now()
  r.append((today_date, today_kg))

  # Plot step line with where=post
  plt.step(*zip(*r), where='post', color=colors[i], label=labels[i])

  # Plot markers (not for today point)
  plt.scatter(*zip(*r[:-1]), color=colors[i], s=15)

plt.legend(framealpha=1.0)

plt.ylabel("Weight [Kg]")

plt.grid(axis='y', linestyle = '--')

plt.tight_layout()
plt.savefig('Progression of the deadlift world record.svg') #plt.show()