/* hrs_analytic_recoded */
cap log close
log using "hrs2016_mi.log", replace

***
* product: hrs2016_mi_full.dta; hrs2016_mi_short.dta
* generate descriptive statistics for HRS 2016 TICS with MI
* indata: HRS analytic recoded;
* outdata: HRS 2016; whole sample (only age, year and iwstat restrictions)
* exhibits: Table3 HRS; Graph1 HRS
* dependent files: none
***

use ../../../input_data/hrs_analytic_recoded.dta, clear
merge 1:1 hhidpn wave using "../../../input_data/tics.dta", keepusing(totcog cogstate) nogen keep(master match)
keep if wave==13
keep if iwstat==1
keep if age >= 63

gen mi_totcog = 1 if missing(totcog)
replace mi_totcog = 0 if !missing(totcog)

gen male_hsless = male*hsless 
gen male_black = male*black 
gen male_hispan = male*hispan 

/* Create age spline indicator variables */
gen l2age = age - 2
gen l2age65l = 63 if 63 < l2age
replace l2age65l = l2age if l2age <= 63
gen temp_max = 0 if l2age-63<0
replace temp_max = l2age-63 if l2age-63>0
gen l2age6574 = 10 if 10 < temp_max
replace l2age6574 = temp_max if temp_max <= 10
gen l2age75p = 0 if 0 > l2age-73
replace l2age75p = l2age-73 if l2age-73 >= 0


mdesc totcog black hispan hsless college male male_hsless male_black male_hispan l2age65l l2age6574 l2age75p hearte stroke cancre hibpe diabe lunge hearta work widowed smokev
global totcog_var black hispan hsless college male male_hsless male_black male_hispan l2age65l l2age6574 l2age75p hearte stroke cancre hibpe diabe lunge hearta work widowed smokev
foreach x of varlist $totcog_var {
    drop if missing(`x')
}
mi set wide
mi misstable summarize totcog
mi register imputed totcog

mi impute ologit totcog = black hispan hsless college male male_hsless male_black male_hispan l2age65l l2age6574 l2age75p hearte stroke cancre hibpe diabe lunge hearta work widowed smokev, add(500) rseed(2020)

mi convert mlong, clear
drop if _mi_miss==1
replace wtrespe=wtrespe/500 if _mi_m != 0
gen mi_flag_16 = 0
replace mi_flag_16 = 1 if _mi_m != 0
rename totcog totcog_16
rename wtrespe weight_16
rename cogstate cogstate_16

save hrs2016_mi_full.dta, replace

keep hhidpn totcog_16 weight_16 cogstate_16 mi_flag_16
gen cogcat_16 = .
replace cogcat_16 = 1 if totcog_16 <= 6 
replace cogcat_16 = 2 if totcog_16 >= 7 & totcog_16 <= 11
replace cogcat_16 = 3 if totcog_16 >= 12 & totcog_16 <= 27

save hrs2016_mi_short.dta, replace

tab cogcat_16 [aweight=weight_16]
tab cogstate_16 [aweight=weight_16]
sum totcog_16 [aw=weight_16], de 
tab totcog_16 [aw=weight_16]

log close


/*

/* stock_hrs-2006 */
cap log close
log using "hrs2016_mi.log", replace

use ../../../input_data/hrs_selected, clear
keep if wave==13
merge 1:1 hhidpn using "../../../input_data/stock_hrs_2006.dta", keepusing(hhidpn)
keep if _m==3
drop _m
keep if iwstat==1
keep if age >= 63

gen mi_totcog = 1 if missing(totcog)
replace mi_totcog = 0 if !missing(totcog)

gen male_hsless = male*hsless 
gen male_black = male*black 
gen male_hispan = male*hispan 

/* Create age spline indicator variables */
gen l2age = age - 2
gen l2age65l = 63 if 63 < l2age
replace l2age65l = l2age if l2age <= 63
gen temp_max = 0 if l2age-63<0
replace temp_max = l2age-63 if l2age-63>0
gen l2age6574 = 10 if 10 < temp_max
replace l2age6574 = temp_max if temp_max <= 10
gen l2age75p = 0 if 0 > l2age-73
replace l2age75p = l2age-73 if l2age-73 >= 0


mdesc totcog black hispan hsless college male male_hsless male_black male_hispan l2age65l l2age6574 l2age75p hearte stroke cancre hibpe diabe lunge hearta work widowed smokev
global totcog_var black hispan hsless college male male_hsless male_black male_hispan l2age65l l2age6574 l2age75p hearte stroke cancre hibpe diabe lunge hearta work widowed smokev
foreach x of varlist $totcog_var {
    drop if missing(`x')
}
mi set wide
mi misstable summarize totcog
mi register imputed totcog

mi impute ologit totcog = black hispan hsless college male male_hsless male_black male_hispan l2age65l l2age6574 l2age75p hearte stroke cancre hibpe diabe lunge hearta work widowed smokev, add(500) rseed(2020)

mi convert mlong, clear
drop if _mi_miss==1
replace weight=weight/500 if _mi_m != 0
gen mi_flag_16 = 0
replace mi_flag_16 = 1 if _mi_m != 0
rename totcog totcog_16
rename weight weight_16
rename cogstate cogstate_16

save hrs2016_mi_full.dta, replace

keep hhidpn totcog_16 weight_16 cogstate_16 mi_flag_16
gen cogcat_16 = .
replace cogcat_16 = 1 if totcog_16 <= 6 
replace cogcat_16 = 2 if totcog_16 >= 7 & totcog_16 <= 11
replace cogcat_16 = 3 if totcog_16 >= 12 & totcog_16 <= 27

save hrs2016_mi_short.dta, replace

tab cogcat_16 [aweight=weight_16]
tab cogstate_16 [aweight=weight_16]
sum totcog_16 [aw=weight_16], de 
tab totcog_16 [aw=weight_16]

log close