import matplotlib.pyplot as plt


# Premissas
qtd_anos = 25 #mais ou menos o tempo de um titulo do tesouro IPCA
ano_original = 2022 #ano dos dados coletados

inflacao = 0.0579 #https://agenciadenoticias.ibge.gov.br/agencia-noticias/2012-agencia-de-noticias/noticias/36051-inflacao-sobe-0-62-em-dezembro-e-fecha-2022-com-alta-de-5-79
pib_original = 9.9 #https://cultura.uol.com.br/economia/noticias/2023/03/03/237_brasil-volta-a-ser-12-maior-economia-do-mundo-em-2022.html
populacao_original = 203 #220_000_000  # https://educa.ibge.gov.br/jovens/materias-especiais/22005-censo-2022-o-retrato-atualizado-do-brasil.html
riqueza_percentual_original_do_1_porcento = 0.484 #https://istoedinheiro.com.br/riqueza-no-brasil-1-mais-ricos-global-wealth/
taxa_crescimento_real_pib = 0.029 #https://agenciadenoticias.ibge.gov.br/agencia-sala-de-imprensa/2013-agencia-de-noticias/releases/36371-pib-cresce-2-9-em-2022-e-fecha-o-ano-em-r-9-9-trilhoes


ganho_real_ipca = 0.0642 #https://www.tesourodireto.com.br/titulos/precos-e-taxas.htm
taxa_crescimento_populacional = 0.0052 #https://agenciadenoticias.ibge.gov.br/agencia-noticias/2012-agencia-de-noticias/noticias/37237-de-2010-a-2022-populacao-brasileira-cresce-6-5-e-chega-a-203-1-milhoes
taxa_crescimento_do_pib_teorico = inflacao + 0.029 #https://agenciadenoticias.ibge.gov.br/agencia-sala-de-imprensa/2013-agencia-de-noticias/releases/36371-pib-cresce-2-9-em-2022-e-fecha-o-ano-em-r-9-9-trilhoes

taxa_crescimento_do_pib = inflacao + taxa_crescimento_real_pib
taxa_crescimento_do_1_porcento = inflacao + ganho_real_ipca
taxa_crescimento_do_99_porcento = inflacao



riqueza_absoluta_original_1_porcento = pib_original * riqueza_percentual_original_do_1_porcento
riqueza_absoluta_original_99_porcento = pib_original - riqueza_absoluta_original_1_porcento

#inicio dos vetores
anos =[]
riqueza_absoluta_1_porcento = []
riqueza_absoluta_99_porcento = []
pib = []
pib_teorico = []
riqueza_percentual_pib_1_porcento = []
riqueza_percentual_pib_99_porcento = []
populacao = []
populacao_1_percento = []
populacao_99_percento = []
pib_per_capita = []
pib_per_capita_1_percent = []
pib_per_capita_99_percent = []

for a in range(ano_original, ano_original + qtd_anos):
    #adicionando os anos ao vetor de anos para mostrar nos graficos
    anos.append(a)

    #fazendo os calculo pertinentes ao problema
    tempo = a - ano_original
    r1 = riqueza_absoluta_original_1_porcento * (1 + taxa_crescimento_do_1_porcento) ** tempo
    r99 = riqueza_absoluta_original_99_porcento * (1 + taxa_crescimento_do_99_porcento) ** tempo
    pb = r1 + r99
    pbt = pib_original * (1 + taxa_crescimento_do_pib) ** tempo
    pp = populacao_original * (1 + taxa_crescimento_populacional) ** tempo
    p1 = pp * 0.01
    p99 = pp - p1
    ppc = (pb * 1_000_000_000_000) / (pp * 1_000_000)/1000
    ppc1 = (r1 * 1_000_000_000_000) / (p1 * 1_000_000)/1000
    ppc99 = (r99 * 1_000_000_000_000) / (p99 * 1_000_000)/1000

    #adicionando os valores aos vetores
    riqueza_absoluta_1_porcento.append(r1)
    riqueza_absoluta_99_porcento.append(r99)
    pib.append(pb)
    pib_teorico.append(pbt)
    riqueza_percentual_pib_1_porcento.append(r1/pb)
    riqueza_percentual_pib_99_porcento.append(r99/pb)
    populacao.append(pp)
    populacao_1_percento.append(p1)
    populacao_99_percento.append(p99)
    pib_per_capita.append(ppc)
    pib_per_capita_1_percent.append(ppc1)
    pib_per_capita_99_percent.append(ppc99)



# Plotando gráfico de linha para crescimento absoluto da riqueza
plt.figure(figsize=(12, 6))
plt.plot(anos, pib, label='PIB', color='green', linestyle='--')
plt.plot(anos, pib_teorico, label='PIB ({:.2f}%)'.format(taxa_crescimento_real_pib * 100), color='orange', linestyle='dotted')
plt.plot(anos, riqueza_absoluta_1_porcento, label='1% mais rico ({:.2f}%)'.format(taxa_crescimento_do_1_porcento * 100), color='red')
plt.plot(anos, riqueza_absoluta_99_porcento, label='99% menos ricos ({:.2f}%)'.format(taxa_crescimento_do_99_porcento * 100), color='blue')
plt.xlabel('Anos')
plt.ylabel('Riqueza em Trilhoes de R$')
plt.title('Crescimento Absoluto Comparativo da Riqueza e PIB')
plt.legend()
plt.grid(True)
plt.show()


# Plotando gráfico de barras empilhadas para percentuais de riqueza
plt.figure(figsize=(12, 6))
plt.bar(anos, riqueza_percentual_pib_99_porcento, label='99% menos ricos ({:.2f}%)'.format(taxa_crescimento_do_99_porcento * 100), color='blue')
plt.bar(anos, riqueza_percentual_pib_1_porcento, bottom=riqueza_percentual_pib_99_porcento, label='1% mais rico ({:.2f}%)'.format(taxa_crescimento_do_1_porcento * 100), color='red')
plt.xlabel('Anos')
plt.ylabel('Percentual da Riqueza / PIB')
plt.title('Evolução Percentual da Distribuição de Riqueza ao Longo dos Anos')
plt.legend()
plt.grid(True)
plt.ylim(0, 1)  # Define o limite do eixo y para 100%
plt.show()


# Plotando gráfico de linha para PIB per capita
plt.figure(figsize=(12, 6))
plt.plot(anos, pib_per_capita, label='PIB per capita total', color='green', linestyle='--')
plt.plot(anos, pib_per_capita_1_percent, label='PIB per capita 1% mais rico', color='red')
plt.plot(anos, pib_per_capita_99_percent, label='PIB per capita 99% menos ricos', color='blue')
plt.xlabel('Anos')
plt.ylabel('PIB per capita (mil R$)')
plt.title('Evolução do PIB per capita Total e por Faixa de Riqueza')
plt.legend()
plt.grid(True)
plt.show()