## ko-ppr-sweden-application.ssc


## control
ko.ppr.reset()


#### READ DATA

## the below function reads from Gunnars original data
## source("ko-ppr-sweden-origdataread.ssc")

## calculate nfx rates and mat rates
se.data <-
  ko.add.data(years = 1970:1999,
              age = 15:44,
              max.order = 4,
              period.cohort.data = T,
              last.rates.not.oerates = T,
              mat = list(
                order.1 = ko.matread("sweden\\se-mat-1.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=NULL,
                  set.na.to.zero=T),
                order.2 = ko.matread("sweden\\se-mat-2.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=1:2,
                  set.na.to.zero=T),
                order.3 = ko.matread("sweden\\se-mat-3.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=1:5,
                  set.na.to.zero=T)
                ),
              nfx = list(
                order.1 = ko.matread("sweden\\se-nfx-1.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=NULL,
                  set.na.to.zero=T),
                order.2 = ko.matread("sweden\\se-nfx-2.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=1:2,
                  set.na.to.zero=T),
                order.3 = ko.matread("sweden\\se-nfx-3.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=1:5,
                  set.na.to.zero=T),
                order.4 = ko.matread("sweden\\se-nfx-4plus.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=1:8,
                  set.na.to.zero=T)),
              par.dist = list(
                parity.0 = ko.matread("sweden\\se-pardist-0.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=NULL,
                  set.na.to.zero=T),
                parity.1 = ko.matread("sweden\\se-pardist-1.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=NULL,
                  set.na.to.zero=T),
                parity.2 = ko.matread("sweden\\se-pardist-2.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=NULL,
                  set.na.to.zero=T),
                parity.3 = ko.matread("sweden\\se-pardist-3plus.txt",
                  data.years=1961:1999,keep.years=1970:1999,set.rows=NULL,
                  set.na.to.zero=T)
                )
              )         

se.long.data <-
  ko.add.data(years = 1961:1999,
              age = 15:44,
              max.order = 4,
              period.cohort.data = T,
              last.rates.not.oerates = T,
              mat = list(
                order.1 = ko.matread("sweden\\se-mat-1.txt",
                  set.rows=NULL,
                  set.na.to.zero=T),
                order.2 = ko.matread("sweden\\se-mat-2.txt",
                  set.rows=1:2,
                  set.na.to.zero=T),
                order.3 = ko.matread("sweden\\se-mat-3.txt",
                  set.rows=1:5,
                  set.na.to.zero=T)
                ),
              nfx = list(
                order.1 = ko.matread("sweden\\se-nfx-1.txt",
                  set.rows=NULL,
                  set.na.to.zero=T),
                order.2 = ko.matread("sweden\\se-nfx-2.txt",
                  set.rows=1:2,
                  set.na.to.zero=T),
                order.3 = ko.matread("sweden\\se-nfx-3.txt",
                  set.rows=1:5,
                  set.na.to.zero=T),
                order.4 = ko.matread("sweden\\se-nfx-4plus.txt",
                  set.rows=1:8,
                  set.na.to.zero=T)),
              par.dist = list(
                parity.0 = ko.matread("sweden\\se-pardist-0.txt",
                  set.rows=NULL,
                  set.na.to.zero=T),
                parity.1 = ko.matread("sweden\\se-pardist-1.txt",
                  set.rows=NULL,
                  set.na.to.zero=T),
                parity.2 = ko.matread("sweden\\se-pardist-2.txt",
                  set.rows=NULL,
                  set.na.to.zero=T),
                parity.3 = ko.matread("sweden\\se-pardist-3plus.txt",
                  set.rows=NULL,
                  set.na.to.zero=T)
                )
              )

## Print summary of nfx
print(ko.asr.summary(se.data))
print(ko.asr.summary(se.data, what="mat"))

## calculate adjusted TFR
se.data <- ko.add.tfr.adjustment(se.data)

## estimate mean and variance changes in KO
se.data <- ko.ppr.add.fitted(se.data)

## calculate period parity progression measures
se.ppr.comparison <- ko.ppr.comparison(se.data)

## cohort summary for nfx cohort data
se.cohort.summary <-
  ko.calc.cohort.summary(se.long.data,
                         first.cohort = 1930,
                         use.initial.par.dist = T)

##complete fertility in the years 1975,1999
se.complete.fert.1975 <-
  ko.ppr.complete.fert.3.scenarios(se.data,1975,
       par.dist = 
         se.cohort.summary$par.dist[,se.cohort.summary$years==1975,],
       cum.nfx = 
         se.cohort.summary$cum.nfx[,se.cohort.summary$years==1975,],
       cum.mab = 
         se.cohort.summary$cum.mab[,se.cohort.summary$years==1975,]
       )
se.complete.fert.1999 <-
  ko.ppr.complete.fert.3.scenarios(se.data,1999,
       par.dist = 
         se.cohort.summary$par.dist[,se.cohort.summary$years==1999,],
       cum.nfx = 
         se.cohort.summary$cum.nfx[,se.cohort.summary$years==1999,],
       cum.mab = 
         se.cohort.summary$cum.mab[,se.cohort.summary$years==1999,]
       )



## print MAB info for Table 1
print(se.complete.fert.1999$stops$mab.by.order)
print(se.complete.fert.1999$continues$mab.by.order)
print(se.complete.fert.1999$observed$mab.by.order)
print(cbind(se.complete.fert.1999$stops$cohort.born,
            se.complete.fert.1999$stops$mab.overall,
            se.complete.fert.1999$continues$mab.overall,
            se.complete.fert.1999$observed$mab.overall))


#print gamma, delta, etc
print.what <- c("gamma","delta","mab","sdab")
for(tt in 1:length(print.what)){
  print(print.what[tt])
  print(se.data$ko.model[,print.what[tt],])
  }

## compare moments of adjusted
#print(ko.asr.moments(ko.ppr.predict.adjusted(se.data,
#                                             1,1998)$asrates,16:45))



## graphs for Tokyo paper
ko.per.data <- se.data
ko.ppr.data <- se.ppr.comparison
ko.coh.data <- se.cohort.summary
ko.refyrs <- c(1975,1999)
ko.coh.complete <- list(ref.yr.1975 = se.complete.fert.1975,
                        ref.yr.1999 = se.complete.fert.1999)
ko.p.opt <-
  list(
       ## lifetime birth prob graphs
       ltb.key.xpos=c(1970,1970,1970),
       ltb.key.ypos=c(.74,.8,.38),
       ## parity progression rates
       ppr.key.xpos=c(1970,1970),
       ppr.key.ypos=c(.93,.5),
       ## adjusted tfr graph
       tfr.1.key.xpos = 1970,
       tfr.1.key.ypos = .96,
       tfr.1.ylim =c(.6,1.1),
       tfr.a.key.xpos = 1970,
       tfr.a.key.ypos = 2.3,
       tfr.a.ylim = c(1.4,2.4),
       tfr.comp.1.key.xpos = 1970,
       tfr.comp.1.key.ypos = .45,
       tfr.comp.a.key.xpos = 1977,
       tfr.comp.a.key.ypos = 1,
       ## projections
       proj.complete.age = 44,
       proj.cohort.lim =c(1930,1985),
       proj.ylim.1 = c(.08,.28),
       proj.ylim.2 = c(1.4,2.1),
       proj.ctext.ypos = list(c(.27,.25,1.48,1.43),
         c(.27,.25,1.48,1.43)),
       ## mean variance changes
       mvc.mablim = list(c(25.5,31.5),c(25.5,30.5),c(25.5,31.5)),
       mvc.mab.key.xpos = 1970,
       mvc.sdlim = list(c(4.85,5.45),c(4.85,5.45),c(4.6,5.2)),
       mvc.gammalim = list(c(-.05,.25),c(-.05,.25),c(-.05,.25)),
       mvc.deltalim = list(c(-.005,.01),c(-.005,.01),c(-.005,.001))
       )

source("ko-ppr-tokyo-figures.ssc")

## graphs for ko-ppr-paper
#source("ko-ppr-sweden-graphs-pprpaper.ssc")




##data.dump(ls(),"ko-ppr-sweden-dumpdata-011106.txt")
##data.restore("bfsynthetic-dumpdata-001124.txt")