cosmonarchy-bw-prerelease/mpq/lua/melee/zerg/general.lua

local Basic = require("melee.zerg.build.basic")
local Openers = require("melee.zerg.build.openers")
local Tier1 = require("melee.zerg.build.tier1")
local Tier2 = require("melee.zerg.build.tier2")
local Tier3 = require("melee.zerg.build.tier3")

local expansion_build_order = {
    Basic.build_expansion,
    Tier1.saturated_production,
    Tier3.general_expansion,
}
local attack_wave_cost_bias = 0.5

return {
    ---@type AIPersonality
    sleeping_giant = {
        name = "General: Sleeping Giant",
        unit_composition = {
            -- signature units
            [UnitId.ZergZethrokor] = 2,
            [UnitId.ZergUltrokor] = 2,
            [UnitId.ZergHydralisk] = 2,
            -- filler units
            [UnitId.ZergKalkalisk] = 1,
            [UnitId.ZergZoryusthaleth] = 1,
            [UnitId.ZergMutalisk] = 1,
            [UnitId.ZergNathrokor] = 1,
            [UnitId.ZergVorvrokor] = 1,
        },
        main_build_order = {
            Openers.sleeping_giant,
            Tier1.sleeping_giant,
            Tier2.giant_iroliris_caphrolosk_hachirosk_zorkiz_buildup,
            Tier3.giant_othstol_gathtelosk,
            Tier1.saturated_production,
        },
        expansion_build_order = expansion_build_order,
        build_anti_ground_defenses = Basic.build_anti_ground,
        build_anti_air_defenses = Basic.build_anti_air,
        attack_wave_cost_bias = attack_wave_cost_bias,
    },

    ---@type AIPersonality
    shrieking_swarm = {
        name = "General: Shrieking Swarm",
        unit_composition = {
            -- signature units (Vornath Pond)
            [UnitId.ZergVorvrokor] = 2,
            [UnitId.ZergNathrokor] = 2,
            -- filler units
            [UnitId.ZergUltrokor] = 1,
            [UnitId.ZergHydralisk] = 1,
            [UnitId.ZergLakizilisk] = 1,
            [UnitId.ZergSovroleth] = 1,
        },
        main_build_order = {
            Openers.shrieking_swarm,
            Tier1.shrieking_swarm,
            Tier2.swarm_iroliris_vornathkalkiir_zorkiz_caphrolosk_hachirosk,
            Tier3.acquire,
            Tier1.saturated_production,
        },
        expansion_build_order = expansion_build_order,
        build_anti_ground_defenses = Basic.build_anti_ground,
        build_anti_air_defenses = Basic.build_anti_air,
        attack_wave_cost_bias = attack_wave_cost_bias,
    },

    ---@type AIPersonality
    subjugation = {
        name = "General: Subjugation",
        unit_composition = {
            -- signature units (Skibact Scab)
            [UnitId.ZergSkithrokor] = 2,
            [UnitId.ZergBactalisk] = 2,
            [UnitId.ZergZethrokor] = 2,
            -- filler units
            [UnitId.ZergKeskathalor] = 1,
            [UnitId.ZergHydralisk] = 1,
            [UnitId.ZergZoryusthaleth] = 1,
            [UnitId.ZergVithrilisk] = 1,
        },
        main_build_order = {
            Openers.subjugation,
            Tier1.subjugation,
            Tier2.subjugation_quazeth_iroliris_buildup,
            Tier3.acquire,
            Tier1.saturated_production,
        },
        expansion_build_order = expansion_build_order,
        build_anti_ground_defenses = Basic.build_anti_ground,
        build_anti_air_defenses = Basic.build_anti_air,
        attack_wave_cost_bias = attack_wave_cost_bias,
    },

    ---@type AIPersonality
    before_the_storm = {
        name = "General: Before the Storm",
        unit_composition = {
            -- signature units
            [UnitId.ZergZethrokor] = 2,
            [UnitId.ZergKalkalisk] = 2,
            -- filler units
            [UnitId.ZergZoryusthaleth] = 1,
            [UnitId.ZergMutalisk] = 1,
            [UnitId.ZergGeszithalor] = 1,
            [UnitId.ZergNathrokor] = 1,
            [UnitId.ZergVorvrokor] = 1,
        },
        main_build_order = {
            Openers.before_the_storm,
            Tier2.storm_iroliris_hachirosk_vornathkalkiir,
            Tier3.acquire,
            Tier1.saturated_production,
        },
        expansion_build_order = expansion_build_order,
        build_anti_ground_defenses = Basic.build_anti_ground,
        build_anti_air_defenses = Basic.build_anti_air,
        attack_wave_cost_bias = attack_wave_cost_bias,
    },

    ---@type AIPersonality
    time_after_time = {
        name = "General: Time After Time",
        unit_composition = {
            -- signature units
            [UnitId.ZergZethrokor] = 2,
            [UnitId.ZergSkithrokor] = 2,
            [UnitId.ZergLakizilisk] = 2,
            -- filler units
            [UnitId.ZergLiiralisk] = 1,
            [UnitId.ZergCikralisk] = 1,
            [UnitId.ZergBactalisk] = 1,
            [UnitId.ZergNathrokor] = 1,
            [UnitId.ZergVorvrokor] = 1,
        },
        main_build_order = {
            Openers.time_after_time,
            Tier1.time_after_time,
            Tier2.time_iroliris_vornathkalkiir,
            Tier3.acquire,
            Tier1.saturated_production,
        },
        expansion_build_order = expansion_build_order,
        build_anti_ground_defenses = Basic.build_anti_ground,
        build_anti_air_defenses = Basic.build_anti_air,
        attack_wave_cost_bias = attack_wave_cost_bias,
    },

    ---@type AIPersonality
    constrained_coil = {
        name = "General: Constrained Coil",
        unit_composition = {
            -- signature units
            [UnitId.ZergZethrokor] = 2,
            [UnitId.ZergKalkalisk] = 2,
            -- filler units
            [UnitId.ZergZoryusthaleth] = 1,
            [UnitId.ZergMutalisk] = 1,
            [UnitId.ZergGeszithalor] = 1,
            [UnitId.ZergNathrokor] = 1,
            [UnitId.ZergVorvrokor] = 1,
        },
        main_build_order = {
            Openers.quazeth_pool_check_then_expand,
            Tier1.coil_buildup,
            Tier2.coil_iroliris_hachirosk,
            Tier3.acquire,
            Tier1.saturated_production,
        },
        expansion_build_order = expansion_build_order,
        build_anti_ground_defenses = Basic.build_anti_ground,
        build_anti_air_defenses = Basic.build_anti_air,
        attack_wave_cost_bias = attack_wave_cost_bias,
    },
}