agents.jl

#agent type definitions
@enum AgentDemographic begin
    Young = 1
    Adult
    Elderly
end
@enum AgentStatus begin
    Susceptible = 1
    Infected
    Recovered
    Immunized
end

struct AgentModel{GraphType,L1,L2}
    demographics::Vector{AgentDemographic}
    demographic_index_vectors::Vector{Vector{Int}}
    app_user::Vector{Bool}
    app_user_index::Vector{Int}
    base_network::GraphType
    ws_matrix_list::L1
    rest_matrix_list::L2
    function AgentModel(num_households)
        pop_list,base_network,index_vectors = generate_population(num_households)
        pop_sizes = length.(index_vectors)
        map_symmetrize(m_tuple) = map(md -> symmetrize_means(pop_sizes,md), m_tuple)

        workschool_contacts_mean_adjusted = map_symmetrize(workschool_mixing)
        rest_contacts_mean_adjusted = map_symmetrize(rest_mixing)

        app_user = rand(RNG,length(pop_list)) .< 0.1 #replace with parameter
        app_user_index = findall(==(true),app_user)

        return new{
            typeof(base_network),
            typeof(workschool_contacts_mean_adjusted),
            typeof(rest_contacts_mean_adjusted)
        }(
            pop_list,
            index_vectors,
            app_user,
            app_user_index,
            base_network,
            workschool_contacts_mean_adjusted,
            rest_contacts_mean_adjusted
        )
    end
end

#generate population with households distributed according to household_size_distribution
function generate_population(num_households)
    households_composition = sample_household_data(num_households)
    households = map( l -> [fill(AgentDemographic(i),n) for (i,n) in enumerate(l)],households_composition) |>
                    l -> reduce(vcat,l) |>
                    l -> reduce(vcat,l)

    total_household_pop = sum(sum.(households_composition))
    household_networks = map(LightGraphs.complete_graph, sum.(households_composition))

    population_list = reduce(vcat,households)
    network = reduce(blockdiag,household_networks; init = SimpleGraph())
    index_vectors = [findall(x -> x == AgentDemographic(i), population_list) for i in 1:(AgentDemographic.size-1)]
    return (;
        population_list,
        network,
        index_vectors
    )
end



# func(a) = println("agent")
function Base.show(io::IO, status::AgentStatus) 
    if status == Susceptible
        print(io, "S")
    elseif status == Infected
        print(io, "I")
    elseif status == Recovered
        print(io, "R")
    elseif status == Immunized
        print(io, "Vac")
    end
end

function Base.show(io::IO, status::AgentDemographic) 
    if status == Young
        print(io, "<25")
    elseif status == Adult
        print(io, "20-65")
    elseif status == Elderly
        print(io, ">65")
    end
end