/*********************************************************************************************************************/
/*********************************************************************************************************************/

clear all
set more off
global outdata "../data"
global outplot "../plots"
global indata "../../../../input_data"

/*
global ster "$local_path/Estimates/PSID"

adopath ++ "$local_path/Estimation"
adopath ++ "$local_path/hyp_mata"
adopath ++ "$local_path/utilities"
adopath ++ "$local_path/Makedata/HRS"
*/

use "$indata/psid_transition.dta", clear

* Number of birth events

global birthevents births 

/*
*** McArthur Feb 2014 Specification 
global allvars_births black hispan  ed_0 ed_2 ed_3 ed_4 mthred_1 mthred_2 mthred_3 mthred_4 ///
		lyrsnclastkid4p lnumbiokids1 lnumbiokids2 lnumbiokids3p lmstat_newmar lmstat_newcohab more1mb lwork ///
		lage25l lage2530 lage31p lage25l_black lage2530_black lage31p_black lage25l_hispan lage2530_hispan lage31p_hispan 
*/

gen lage25l_ed0=lage25l*ed_0
gen lage25l_ed2=lage25l*ed_2
gen lage25l_ed3=lage25l*ed_3
gen lage25l_ed4=lage25l*ed_4

gen lage2530_ed0=lage2530*ed_0
gen lage2530_ed2=lage2530*ed_2
gen lage2530_ed3=lage2530*ed_3
gen lage2530_ed4=lage2530*ed_4

gen lage31p_ed0=lage31p*ed_0
gen lage31p_ed2=lage31p*ed_2
gen lage31p_ed3=lage31p*ed_3
gen lage31p_ed4=lage31p*ed_4

label var lyrsnclastkid "Years since last birth (lag)"

global allvars_births black hispan  ed_0 ed_2 ed_3 ed_4 mthred_1 mthred_2 mthred_3 mthred_4 lyrsnclastkid lnumbiokids1 lnumbiokids2 ///
		lnumbiokids3p lmstat_newmar lmstat_newcohab lwork lage25l lage2530 lage31p lage25l_ed0 lage2530_ed0 lage31p_ed0 ///
		lage25l_ed2 lage2530_ed2 lage31p_ed2 lage25l_ed3 lage2530_ed3 lage31p_ed3 lage25l_ed4 lage2530_ed4 lage31p_ed4


/*********************************************************************/
* ESTIMATE ORDERED OUTCOMES
/*********************************************************************/
* estimate birth event models with population limited to women less than 45
foreach n in $birthevents {
	local x = "allvars_`n'"
	oprobit `n' $`x' if `n'!=-2&`n'!=9 & male==0 & age<45
	gen e_`n' = e(sample)
        mfx2, stub(o_`n')
	est save "$outdata/`n'.ster", replace
	est store `n'
	matrix m`n' = e(b)
	predict simu_`n' if e_`n' == 1
}

esttab births using "$outdata/childbearing.csv", compress se scalars(N aic ll chi2 p r2_p cmd) label title("Childbearing Model") replace
xml_tab o_births_mfx, save("$outdata/childbearing_marg.xml") replace title("Order Probit Model for Childbearing - Marginal Effects")
xml_tab births, save("$outdata/childbearing.xml") replace title("Order Probit Model for Childbearing")