mainHelpers.jl

"""
Some common infrastructure for setting up and running the model.
"""



include("lib/loadLibsPath.jl")

# add various directories to Julia's model search path
addToLoadPath!(String(@__DIR__) * "/.", 
               String(@__DIR__) * "/lib",
               String(@__DIR__) * "/src",
               String(@__DIR__) * "/src/common",
               String(@__DIR__) * "/src/agents",
               String(@__DIR__) * "/src/agents/agent_modules",
               String(@__DIR__) * "/src/agents/interactions",
               String(@__DIR__) * "/src/simulate",
               String(@__DIR__) * "/src/fullModel"
               )

using ArgParse

using Utilities

using FullModelPars
using FullModel


include("src/fullModel/data.jl")

include("src/handleParams.jl")


"Generate and return a usable model object."
function setupModel(pars)
    datp = pars.datapars
    dir = datp.datadir

    demoData   = loadDemographyData(dir * "/" * datp.iniAgeFName,
                                dir * "/" * datp.pre51FertFName,
                                dir * "/" * datp.fertFName, 
                                dir * "/" * datp.pre51DeathsFName,
                                dir * "/" * datp.deathFFName,
                                dir * "/" * datp.deathMFName)
                                
    workData = loadWorkData(dir * "/" * datp.unemplFName, 
        dir * "/" * datp.wealthFName)

    model = createModel!(demoData, workData, pars)

    initializeModel!(model, pars.poppars, pars.workpars, pars.carepars, pars.taskcarepars, pars.mappars, pars.lhapars)
    
    model
end


"Prepare and return output file or nothing."
function setupLogging(simPars; FS = "\t")
    if simPars.logfile == ""
        return nothing
    end

    file = open(simPars.logfile, "w")

    print_header(file, Data; FS)

    file
end


"Run fully initialised model non-interactively from start to finish."
function runModel!(model, simPars, pars, logfile = nothing; FS = "\t")
    curTime = pars.poppars.startTime

    simPars.verbose ? setVerbose!() : unsetVerbose!()
    setDelay!(simPars.sleeptime)

    # no point in continuing with the simulation if we are not recording results
    finishTime = min(pars.poppars.finishTime, simPars.endLogTime)

    while curTime <= finishTime
        stepModel!(model, curTime, pars)

        if logfile != nothing && curTime >= simPars.startLogTime
            results = observe(Data, model, curTime, pars)
            log_results(logfile, results; FS)
        end
        
        #nWorkers = work_by_time(model.pop)
        
        #println(nWorkers)

        curTime += simPars.dt
    end
end