#!/usr/bin/env python3

from dataclasses import dataclass
from typing import Dict

from type.rate import Rate


# Used to compute RMDs.  Source:
# https://www.irs.gov/publications/p590b#en_US_2018_publink1000231258
IRS_ACTUARY_TABLES = {
    70: 27.4,
    71: 26.5,
    72: 25.6,
    73: 24.7,
    74: 23.8,
    75: 22.9,
    76: 22.0,
    77: 21.2,
    78: 20.3,
    79: 19.5,
    80: 18.7,
    81: 17.9,
    82: 17.1,
    83: 16.3,
    84: 15.5,
    85: 14.8,
    86: 14.1,
    87: 13.4,
    88: 12.7,
    89: 12.0,
    90: 11.4,
    91: 10.8,
    92: 10.2,
    93: 9.6,
    94: 9.1,
    95: 8.6,
    96: 8.1,
    97: 7.6,
    98: 7.1,
    99: 6.7,
    100: 6.3,
    101: 5.9,
    102: 5.5,
    103: 5.2,
    104: 4.9,
    105: 4.5,
    106: 4.2,
    107: 3.9,
    108: 3.7,
    109: 3.4,
    110: 3.1,
    111: 2.9,
    112: 2.6,
    113: 2.4,
    114: 2.1,
}


def get_actuary_number_years_to_live(age: int):
    if age < 70:
        return 27.4 + age
    if age > 114:
        return 1.9
    return IRS_ACTUARY_TABLES[age]


def make_multiplier(percent: float) -> Rate:
    percent /= 100
    if percent > 0:
        percent += 1.0
    else:
        percent = 1.0 - percent
    return percent


#                        year       s&p    10y    infla-
#                                   500    bond    tion
#
# source: https://www.multpl.com/inflation/table/by-year
# source: http://pages.stern.nyu.edu/~adamodar/New_Home_Page/datafile/histretSP.html

@dataclass
class StocksBondsAndInflation:
    snp_500_return: Rate
    us_10y_bond_return: Rate
    inflation: Rate

    def __init__(self, snp: Rate, bonds: Rate, inflation: Rate):
        self.snp_500_return = snp
        self.us_10y_bond_return = bonds
        self.inflation = inflation


ROIs_AND_INFLATION: Dict[int, StocksBondsAndInflation] = {
    2020: StocksBondsAndInflation(Rate(percent_change=18.01), Rate(percent_change=11.33),  Rate(percent_change=2.49)),
    2019: StocksBondsAndInflation(Rate(percent_change=31.21),  Rate(percent_change=9.64),   Rate(percent_change=1.55)),
    2018: StocksBondsAndInflation(Rate(percent_change=-4.23),  Rate(percent_change=-0.02),  Rate(percent_change=2.07)),
    2017: StocksBondsAndInflation(Rate(percent_change=21.61),  Rate(percent_change=2.80),   Rate(percent_change=2.50)),
    2016: StocksBondsAndInflation(Rate(percent_change=11.77),  Rate(percent_change=0.69),   Rate(percent_change=1.37)),
    2015: StocksBondsAndInflation(Rate(percent_change=1.38),   Rate(percent_change=1.28),   Rate(percent_change=-0.09)),
    2014: StocksBondsAndInflation(Rate(percent_change=13.52),  Rate(percent_change=10.75),  Rate(percent_change=1.58)),
    2013: StocksBondsAndInflation(Rate(percent_change=32.15),  Rate(percent_change=-9.10),  Rate(percent_change=1.59)),
    2012: StocksBondsAndInflation(Rate(percent_change=15.89),  Rate(percent_change=2.97),   Rate(percent_change=2.93)),
    2011: StocksBondsAndInflation(Rate(percent_change=2.10),   Rate(percent_change=16.04),  Rate(percent_change=1.63)),
    2010: StocksBondsAndInflation(Rate(percent_change=14.82),  Rate(percent_change=8.46),   Rate(percent_change=2.63)),
    2009: StocksBondsAndInflation(Rate(percent_change=25.94),  Rate(percent_change=-11.12), Rate(percent_change=0.03)),
    2008: StocksBondsAndInflation(Rate(percent_change=-36.55), Rate(percent_change=20.10),  Rate(percent_change=4.28)),
    2007: StocksBondsAndInflation(Rate(percent_change=5.48),   Rate(percent_change=10.21),  Rate(percent_change=2.08)),
    2006: StocksBondsAndInflation(Rate(percent_change=15.61),  Rate(percent_change=1.96),   Rate(percent_change=3.99)),
    2005: StocksBondsAndInflation(Rate(percent_change=4.83),   Rate(percent_change=2.87),   Rate(percent_change=2.97)),
    2004: StocksBondsAndInflation(Rate(percent_change=10.74),  Rate(percent_change=4.49),   Rate(percent_change=1.93)),
    2003: StocksBondsAndInflation(Rate(percent_change=28.36),  Rate(percent_change=0.38),   Rate(percent_change=2.60)),
    2002: StocksBondsAndInflation(Rate(percent_change=-21.97), Rate(percent_change=15.12),  Rate(percent_change=1.14)),
    2001: StocksBondsAndInflation(Rate(percent_change=-11.85), Rate(percent_change=5.57),   Rate(percent_change=3.73)),
    2000: StocksBondsAndInflation(Rate(percent_change=-9.03),  Rate(percent_change=16.66),  Rate(percent_change=2.74)),
    1999: StocksBondsAndInflation(Rate(percent_change=20.89),  Rate(percent_change=-8.25),  Rate(percent_change=1.67)),
    1998: StocksBondsAndInflation(Rate(percent_change=28.34),  Rate(percent_change=14.92),  Rate(percent_change=1.57)),
    1997: StocksBondsAndInflation(Rate(percent_change=33.10),  Rate(percent_change=9.94),   Rate(percent_change=3.04)),
    1996: StocksBondsAndInflation(Rate(percent_change=22.68),  Rate(percent_change=1.43),   Rate(percent_change=2.73)),
    1995: StocksBondsAndInflation(Rate(percent_change=37.20),  Rate(percent_change=23.48),  Rate(percent_change=2.80)),
    1994: StocksBondsAndInflation(Rate(percent_change=1.33),   Rate(percent_change=-8.04),  Rate(percent_change=2.52)),
    1993: StocksBondsAndInflation(Rate(percent_change=9.97),   Rate(percent_change=14.21),  Rate(percent_change=3.26)),
    1992: StocksBondsAndInflation(Rate(percent_change=7.49),   Rate(percent_change=9.36),   Rate(percent_change=2.60)),
    1991: StocksBondsAndInflation(Rate(percent_change=30.23),  Rate(percent_change=15.00),  Rate(percent_change=5.65)),
    1990: StocksBondsAndInflation(Rate(percent_change=-3.06),  Rate(percent_change=6.24),   Rate(percent_change=5.20)),
    1989: StocksBondsAndInflation(Rate(percent_change=31.48),  Rate(percent_change=17.69),  Rate(percent_change=4.67)),
    1988: StocksBondsAndInflation(Rate(percent_change=16.54),  Rate(percent_change=8.22),   Rate(percent_change=4.05)),
    1987: StocksBondsAndInflation(Rate(percent_change=5.81),   Rate(percent_change=-4.96),  Rate(percent_change=1.46)),
    1986: StocksBondsAndInflation(Rate(percent_change=18.49),  Rate(percent_change=24.28),  Rate(percent_change=3.89)),
    1985: StocksBondsAndInflation(Rate(percent_change=31.24),  Rate(percent_change=25.71),  Rate(percent_change=3.53)),
    1984: StocksBondsAndInflation(Rate(percent_change=6.15),   Rate(percent_change=13.73),  Rate(percent_change=4.19)),
    1983: StocksBondsAndInflation(Rate(percent_change=22.34),  Rate(percent_change=3.20),   Rate(percent_change=3.71)),
    1982: StocksBondsAndInflation(Rate(percent_change=20.42),  Rate(percent_change=32.81),  Rate(percent_change=8.39)),
    1981: StocksBondsAndInflation(Rate(percent_change=-4.70),  Rate(percent_change=8.20),   Rate(percent_change=11.83)),
    1980: StocksBondsAndInflation(Rate(percent_change=31.74),  Rate(percent_change=-2.99),  Rate(percent_change=13.91)),
    1979: StocksBondsAndInflation(Rate(percent_change=18.52),  Rate(percent_change=0.67),   Rate(percent_change=9.28)),
    1978: StocksBondsAndInflation(Rate(percent_change=6.51),   Rate(percent_change=-0.78),  Rate(percent_change=6.84)),
    1977: StocksBondsAndInflation(Rate(percent_change=-6.98),  Rate(percent_change=1.29),   Rate(percent_change=5.22)),
    1976: StocksBondsAndInflation(Rate(percent_change=23.83),  Rate(percent_change=15.98),  Rate(percent_change=6.72)),
    1975: StocksBondsAndInflation(Rate(percent_change=37.00),  Rate(percent_change=3.61),   Rate(percent_change=11.80)),
    1974: StocksBondsAndInflation(Rate(percent_change=-25.90), Rate(percent_change=1.99),   Rate(percent_change=9.39)),
    1973: StocksBondsAndInflation(Rate(percent_change=-14.31), Rate(percent_change=3.66),   Rate(percent_change=3.65)),
    1972: StocksBondsAndInflation(Rate(percent_change=18.76),  Rate(percent_change=2.82),   Rate(percent_change=3.27)),
    1971: StocksBondsAndInflation(Rate(percent_change=14.22),  Rate(percent_change=9.79),   Rate(percent_change=5.29)),
    1970: StocksBondsAndInflation(Rate(percent_change=3.56),   Rate(percent_change=16.75),  Rate(percent_change=6.18)),
    1969: StocksBondsAndInflation(Rate(percent_change=-8.24),  Rate(percent_change=-5.01),  Rate(percent_change=4.40)),
    1968: StocksBondsAndInflation(Rate(percent_change=10.81),  Rate(percent_change=3.27),   Rate(percent_change=3.65)),
    1967: StocksBondsAndInflation(Rate(percent_change=23.80),  Rate(percent_change=-1.58),  Rate(percent_change=3.46)),
    1966: StocksBondsAndInflation(Rate(percent_change=-9.97),  Rate(percent_change=2.91),   Rate(percent_change=1.92)),
    1965: StocksBondsAndInflation(Rate(percent_change=12.40),  Rate(percent_change=0.72),   Rate(percent_change=0.97)),
    1964: StocksBondsAndInflation(Rate(percent_change=16.42),  Rate(percent_change=3.73),   Rate(percent_change=1.64)),
    1963: StocksBondsAndInflation(Rate(percent_change=22.61),  Rate(percent_change=1.68),   Rate(percent_change=1.33)),
    1962: StocksBondsAndInflation(Rate(percent_change=-8.81),  Rate(percent_change=5.69),   Rate(percent_change=0.67)),
    1961: StocksBondsAndInflation(Rate(percent_change=26.64),  Rate(percent_change=2.06),   Rate(percent_change=1.71)),
    1960: StocksBondsAndInflation(Rate(percent_change=0.34),   Rate(percent_change=11.64),  Rate(percent_change=1.03)),
    1959: StocksBondsAndInflation(Rate(percent_change=12.06),  Rate(percent_change=-2.65),  Rate(percent_change=1.40)),
    1958: StocksBondsAndInflation(Rate(percent_change=43.72),  Rate(percent_change=-2.10),  Rate(percent_change=3.62)),
    1957: StocksBondsAndInflation(Rate(percent_change=-10.46), Rate(percent_change=6.80),   Rate(percent_change=2.99)),
    1956: StocksBondsAndInflation(Rate(percent_change=7.44),   Rate(percent_change=-2.26),  Rate(percent_change=0.37)),
    1955: StocksBondsAndInflation(Rate(percent_change=32.60),  Rate(percent_change=-1.34),  Rate(percent_change=-0.74)),
    1954: StocksBondsAndInflation(Rate(percent_change=52.56),  Rate(percent_change=3.29),   Rate(percent_change=1.13)),
    1953: StocksBondsAndInflation(Rate(percent_change=-1.21),  Rate(percent_change=4.14),   Rate(percent_change=0.38)),
    1952: StocksBondsAndInflation(Rate(percent_change=18.15),  Rate(percent_change=2.27),   Rate(percent_change=4.33)),
    1951: StocksBondsAndInflation(Rate(percent_change=23.68),  Rate(percent_change=-0.30),  Rate(percent_change=8.09)),
    1950: StocksBondsAndInflation(Rate(percent_change=30.81),  Rate(percent_change=0.43),   Rate(percent_change=-2.08)),
    1949: StocksBondsAndInflation(Rate(percent_change=18.30),  Rate(percent_change=4.66),   Rate(percent_change=1.27)),
    1948: StocksBondsAndInflation(Rate(percent_change=5.70),   Rate(percent_change=1.95),   Rate(percent_change=10.23)),
    1947: StocksBondsAndInflation(Rate(percent_change=5.20),   Rate(percent_change=0.92),   Rate(percent_change=18.13)),
    1946: StocksBondsAndInflation(Rate(percent_change=-8.43),  Rate(percent_change=3.13),   Rate(percent_change=2.25)),
    1945: StocksBondsAndInflation(Rate(percent_change=35.82),  Rate(percent_change=3.80),   Rate(percent_change=2.30)),
    1944: StocksBondsAndInflation(Rate(percent_change=19.03),  Rate(percent_change=2.58),   Rate(percent_change=2.96)),
    1943: StocksBondsAndInflation(Rate(percent_change=25.06),  Rate(percent_change=2.49),   Rate(percent_change=7.64)),
    1942: StocksBondsAndInflation(Rate(percent_change=19.17),  Rate(percent_change=2.29),   Rate(percent_change=11.35)),
    1941: StocksBondsAndInflation(Rate(percent_change=-12.77), Rate(percent_change=-2.02),  Rate(percent_change=1.44)),
    1940: StocksBondsAndInflation(Rate(percent_change=-10.67), Rate(percent_change=5.40),   Rate(percent_change=-0.71)),
    1939: StocksBondsAndInflation(Rate(percent_change=-1.10),  Rate(percent_change=4.41),   Rate(percent_change=-1.41)),
    1938: StocksBondsAndInflation(Rate(percent_change=29.28),  Rate(percent_change=4.21),   Rate(percent_change=0.71)),
    1937: StocksBondsAndInflation(Rate(percent_change=-35.34), Rate(percent_change=1.38),   Rate(percent_change=2.17)),
    1936: StocksBondsAndInflation(Rate(percent_change=31.94),  Rate(percent_change=5.02),   Rate(percent_change=1.47)),
    1935: StocksBondsAndInflation(Rate(percent_change=46.74),  Rate(percent_change=4.47),   Rate(percent_change=3.03)),
    1934: StocksBondsAndInflation(Rate(percent_change=-1.19),  Rate(percent_change=7.96),   Rate(percent_change=2.33)),
    1933: StocksBondsAndInflation(Rate(percent_change=49.98),  Rate(percent_change=1.86),   Rate(percent_change=-9.79)),
    1932: StocksBondsAndInflation(Rate(percent_change=-8.64),  Rate(percent_change=8.79),   Rate(percent_change=-10.06)),
    1931: StocksBondsAndInflation(Rate(percent_change=-43.84), Rate(percent_change=-2.56),  Rate(percent_change=-7.02)),
    1930: StocksBondsAndInflation(Rate(percent_change=-25.12), Rate(percent_change=4.54),   Rate(percent_change=0.00)),
    1929: StocksBondsAndInflation(Rate(percent_change=-8.30),  Rate(percent_change=4.20),   Rate(percent_change=-1.16)),
    1928: StocksBondsAndInflation(Rate(percent_change=43.81),  Rate(percent_change=0.84),   Rate(percent_change=-1.14)),
}