source("~/delfinverse/dinput/R/utils.R")

options(readr.show_col_types = FALSE)

setwd("~/data/appl-wb/20_staff/kjo/fhh/2025-05-20T1646_u80874371_ahv_basis_kjo")

PARAM_GLOBAL <- 
  read_delim("PARAM_GLOBAL.csv")

BEVOELKERUNG <- 
  read_delim("BEVOELKERUNG.csv")

TAUX_MORTALITE <- 
  read_delim("TAUX_MORTALITE.csv")

RR <- 
  read_delim("RR_AVS.csv")

EF <-
  read_delim("EPRC_PROJECTION_FLEX.csv") %>%
  filter(age_ret > 61, gpr == "rvieillesse_simple") %>%
  mutate(coh = jahr - alt) %>%
  dplyr::summarize(eprc = sum(eprc, na.rm = TRUE),
                   .by = c("coh", "jahr", "sex", "nat", "dom", "alt")) %>%
  mutate(type = "EF")

EF <- 
  EPRC_PROJECTION_FLEX %>% 
  filter(gpr == "rvieillesse_simple") %>%
  mutate(coh = jahr - alt) %>% 
  dplyr::summarize(eprc = sum(eprc, na.rm = TRUE), 
                   .by = c("coh", "jahr", "sex", "nat", "dom", "alt")) %>% 
  mutate(type = "EF")

EE <- 
  read_delim("EPRC_ESTIMATION.csv") %>% 
  mutate(coh = jahr - alt) %>% 
  filter(alt >= 62) %>% 
  dplyr::summarize(eprc = sum(eprc, na.rm = TRUE), 
                   .by = c("coh", "jahr", "sex", "nat", "dom", "alt")) %>% 
  mutate(type = "EE")

viz <- 
  bind_rows(EE, EF) %>% 
  filter(jahr <= 2040) %>% 
  dplyr::summarize(eprc = sum(eprc, na.rm = TRUE), .by = c("jahr", "type"))

ggplot(viz, aes(x = jahr, y = eprc, col = type)) +
  geom_vline(xintercept = 2032.5) +
  geom_shadowpoint()

# x <- RR %>% 
#   filter(gpr == "rvieillesse_simple") %>% 
#   dplyr::summarize(eprc = sum(eprc), 
#                    .by = c("jahr", "sex", "nat", "dom", "age_ret", "alt")) %>% 
#   mutate(coh = jahr - alt) %>% 
#   select(coh, jahr, sex, nat, dom, age_ret, alt, eprc)
# 
# viz <- 
#   bind_rows(x, EF) %>% 
#   filter(jahr <= 2040)

# ggplot(filter(viz, coh == 1965, dom == "ch", sex == "f", age_ret == 64,
#               jahr <= 2040), 
#        aes(x = jahr, y = eprc, col = nat)) +
#   geom_vline(xintercept = 2024.5) +
#   geom_shadowpoint()

# viz2 <- viz %>% 
#   dplyr::summarize(eprc = sum(eprc, na.rm = TRUE), .by = "jahr")
# 
# ggplot(viz2, aes(x = jahr, y = eprc)) +
#   geom_vline(xintercept = 2032.5) +
#   geom_shadowpoint()
