NS/main/source/mod/AvHAIConfig.cpp

#include "AvHAIConfig.h"
#include "AvHAIMath.h"

#include "AvHServerUtil.h"

#include <unordered_map>

BotFillTiming CurrentBotFillTiming = FILLTIMING_ALLHUMANS;

std::unordered_map<std::string, TeamSizeDefinitions> TeamSizeMap;

bot_skill BotSkillLevels[4];

AvHMessageID ChamberSequence[3] = { ALIEN_BUILD_DEFENSE_CHAMBER, ALIEN_BUILD_MOVEMENT_CHAMBER, ALIEN_BUILD_SENSORY_CHAMBER };

char BotPrefix[32] = "";

extern cvar_t avh_botskill;
extern cvar_t avh_botallowlerk;
extern cvar_t avh_botallowfade;
extern cvar_t avh_botallowonos;
extern cvar_t avh_botcommanderwait;
extern cvar_t avh_botlerkcooldown;
extern cvar_t avh_botmaxstucktime;

float CONFIG_GetCommanderWaitTime()
{
    return avh_botcommanderwait.value;
}

float CONFIG_GetLerkCooldown()
{
    return avh_botlerkcooldown.value;
}

bool CONFIG_IsLerkAllowed()
{
    return avh_botallowlerk.value > 0;
}

bool CONFIG_IsFadeAllowed()
{
    return avh_botallowfade.value > 0;
}

bool CONFIG_IsOnosAllowed()
{
    return avh_botallowonos.value > 0;
}

float CONFIG_GetMaxStuckTime()
{
    return avh_botmaxstucktime.value;
}

string CONFIG_GetBotPrefix()
{
    return string(BotPrefix);
}

int CONFIG_GetTeamASizeForMap(const char* MapName)
{
    std::string s = MapName;
    std::unordered_map<std::string, TeamSizeDefinitions>::const_iterator got = TeamSizeMap.find(s);

    if (got == TeamSizeMap.end())
    {
        return TeamSizeMap["default"].TeamASize;
    }
    else
    {
        return got->second.TeamASize;
    }
}

int CONFIG_GetTeamBSizeForMap(const char* MapName)
{
    std::string s = MapName;
    std::unordered_map<std::string, TeamSizeDefinitions>::const_iterator got = TeamSizeMap.find(s);

    if (got == TeamSizeMap.end())
    {
        return TeamSizeMap["default"].TeamBSize;
    }
    else
    {
        return got->second.TeamBSize;
    }
}

AvHMessageID CONFIG_GetHiveTechAtIndex(const int Index)
{
    if (Index < 0 || Index > 2) { return MESSAGE_NULL; }

    return ChamberSequence[Index];
}

bot_skill CONFIG_GetBotSkillLevel()
{
    int index = clampi((int)avh_botskill.value, 0, 3);

    return BotSkillLevels[index];
}

void CONFIG_ParseConfigFile()
{

    BotSkillLevels[0].marine_bot_reaction_time = 0.4f;
    BotSkillLevels[0].marine_bot_aim_skill = 0.1f;
    BotSkillLevels[0].marine_bot_motion_tracking_skill = 0.1f;
    BotSkillLevels[0].marine_bot_view_speed = 0.5f;

    BotSkillLevels[0].alien_bot_reaction_time = 0.4f;
    BotSkillLevels[0].alien_bot_aim_skill = 0.2f;
    BotSkillLevels[0].alien_bot_motion_tracking_skill = 0.2f;    
    BotSkillLevels[0].alien_bot_view_speed = 0.75f;

    BotSkillLevels[1].marine_bot_reaction_time = 0.2f;
    BotSkillLevels[1].marine_bot_aim_skill = 0.5f;
    BotSkillLevels[1].marine_bot_motion_tracking_skill = 0.4f;
    BotSkillLevels[1].marine_bot_view_speed = 1.0f;

    BotSkillLevels[1].alien_bot_reaction_time = 0.2f;
    BotSkillLevels[1].alien_bot_aim_skill = 0.5f;
    BotSkillLevels[1].alien_bot_motion_tracking_skill = 0.5f;
    BotSkillLevels[1].alien_bot_view_speed = 1.3f;

    BotSkillLevels[2].marine_bot_reaction_time = 0.2f;
    BotSkillLevels[2].marine_bot_aim_skill = 0.6f;
    BotSkillLevels[2].marine_bot_motion_tracking_skill = 0.6f;
    BotSkillLevels[2].marine_bot_view_speed = 1.5f;

    BotSkillLevels[2].alien_bot_reaction_time = 0.2f;
    BotSkillLevels[2].alien_bot_aim_skill = 0.8f;
    BotSkillLevels[2].alien_bot_motion_tracking_skill = 0.8f;
    BotSkillLevels[2].alien_bot_view_speed = 1.5f;

    BotSkillLevels[3].marine_bot_reaction_time = 0.1f;
    BotSkillLevels[3].marine_bot_aim_skill = 1.0f;
    BotSkillLevels[3].marine_bot_motion_tracking_skill = 1.0f;
    BotSkillLevels[3].marine_bot_view_speed = 2.0f;

    BotSkillLevels[3].alien_bot_reaction_time = 0.1f;
    BotSkillLevels[3].alien_bot_aim_skill = 1.0f;
    BotSkillLevels[3].alien_bot_motion_tracking_skill = 1.0f;
    BotSkillLevels[3].alien_bot_view_speed = 2.0f;


    string BotConfigFile = string(getModDirectory()) + "/nsbots.ini";

    const char* filename = BotConfigFile.c_str();

    std::ifstream cFile(filename);
    if (cFile.is_open())
    {
        std::string line;
        int CurrSkillIndex = 0;

        while (getline(cFile, line))
        {
            line.erase(std::remove_if(line.begin(), line.end(), ::isspace),
                line.end());
            if (line[0] == '#' || line.empty())
                continue;
            auto delimiterPos = line.find("=");
            auto key = line.substr(0, delimiterPos);
            auto value = line.substr(delimiterPos + 1);

            if (key.compare("TeamSize") == 0)
            {
                auto mapDelimiterPos = value.find(":");

                if (mapDelimiterPos == std::string::npos)
                {
                    continue;
                }

                auto mapName = value.substr(0, mapDelimiterPos);
                auto teamSizes = value.substr(mapDelimiterPos + 1);
                auto sizeDelimiterPos = teamSizes.find("/");
                if (sizeDelimiterPos == std::string::npos)
                {
                    continue;
                }
                auto marineSize = teamSizes.substr(0, sizeDelimiterPos);
                auto alienSize = teamSizes.substr(sizeDelimiterPos + 1);

                int iMarineSize = atoi(marineSize.c_str());
                int iAlienSize = atoi(alienSize.c_str());

                if (iMarineSize >= 0 && iMarineSize <= 32 && iAlienSize >= 0 && iAlienSize <= 32)
                {
                    TeamSizeMap[mapName].TeamASize = atoi(marineSize.c_str());
                    TeamSizeMap[mapName].TeamBSize = atoi(alienSize.c_str());
                }

                continue;
            }

            if (key.compare("prefix") == 0)
            {
                sprintf(BotPrefix, value.c_str());

                continue;
            }

            if (key.compare("BotFillTiming") == 0)
            {
                int FillSetting = atoi(value.c_str());
                FillSetting = clampi(FillSetting, 0, 2);
                CurrentBotFillTiming = (BotFillTiming)FillSetting;
                continue;
            }

            if (key.compare("BotSkillLevel") == 0)
            {
                CurrSkillIndex = std::stoi(value.c_str());
                CurrSkillIndex = clampi(CurrSkillIndex, 0, 3);
                continue;
            }

            if (key.compare("MarineReactionTime") == 0)
            {
                float NewValue = std::stof(value.c_str());

                BotSkillLevels[CurrSkillIndex].marine_bot_reaction_time = clampf(NewValue, 0.0f, 1.0f);

                continue;
            }

            if (key.compare("AlienReactionTime") == 0)
            {

                float NewValue = std::stof(value.c_str());

                BotSkillLevels[CurrSkillIndex].alien_bot_reaction_time = clampf(NewValue, 0.0f, 1.0f);

                continue;
            }

            if (key.compare("MarineAimSkill") == 0)
            {

                float NewValue = std::stof(value.c_str());

                BotSkillLevels[CurrSkillIndex].marine_bot_aim_skill = clampf(NewValue, 0.0f, 1.0f);

                continue;
            }

            if (key.compare("AlienAimSkill") == 0)
            {

                float NewValue = std::stof(value.c_str());

                BotSkillLevels[CurrSkillIndex].alien_bot_aim_skill = clampf(NewValue, 0.0f, 1.0f);

                continue;
            }

            if (key.compare("MarineMovementTracking") == 0)
            {

                float NewValue = std::stof(value.c_str());

                BotSkillLevels[CurrSkillIndex].marine_bot_motion_tracking_skill = clampf(NewValue, 0.0f, 1.0f);

                continue;
            }

            if (key.compare("AlienMovementTracking") == 0)
            {

                float NewValue = std::stof(value.c_str());

                BotSkillLevels[CurrSkillIndex].alien_bot_motion_tracking_skill = clampf(NewValue, 0.0f, 1.0f);

                continue;
            }

            if (key.compare("MarineViewSpeed") == 0)
            {

                float NewValue = std::stof(value.c_str());

                BotSkillLevels[CurrSkillIndex].marine_bot_view_speed = clampf(NewValue, 0.0f, 5.0f);

                continue;
            }

            if (key.compare("AlienViewSpeed") == 0)
            {

                float NewValue = std::stof(value.c_str());

                BotSkillLevels[CurrSkillIndex].alien_bot_view_speed = clampf(NewValue, 0.0f, 5.0f);

                continue;
            }

            if (key.compare("ChamberSequence") == 0)
            {
                AvHMessageID HiveOneTech = MESSAGE_NULL;
                AvHMessageID HiveTwoTech = MESSAGE_NULL;
                AvHMessageID HiveThreeTech = MESSAGE_NULL;

                std::vector<AvHMessageID> AvailableTechs = { ALIEN_BUILD_DEFENSE_CHAMBER, ALIEN_BUILD_MOVEMENT_CHAMBER, ALIEN_BUILD_SENSORY_CHAMBER };

                auto firstTechDelimiter = value.find("/");

                if (firstTechDelimiter == std::string::npos)
                {
                    continue;
                }

                auto FirstTech = value.substr(0, firstTechDelimiter);
                auto NextTechs = value.substr(firstTechDelimiter + 1);

                auto SecondTechDelimiter = NextTechs.find("/");

                if (SecondTechDelimiter == std::string::npos)
                {
                    continue;
                }

                auto SecondTech = NextTechs.substr(0, SecondTechDelimiter);
                auto ThirdTech = NextTechs.substr(SecondTechDelimiter + 1);

                if (FirstTech.compare("movement") == 0)
                {
                    HiveOneTech = ALIEN_BUILD_MOVEMENT_CHAMBER;

                    AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_MOVEMENT_CHAMBER), AvailableTechs.end());

                }
                else if (FirstTech.compare("defense") == 0)
                {
                    HiveOneTech = ALIEN_BUILD_DEFENSE_CHAMBER;

                    AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_DEFENSE_CHAMBER), AvailableTechs.end());
                }
                else if (FirstTech.compare("sensory") == 0)
                {
                    HiveOneTech = ALIEN_BUILD_SENSORY_CHAMBER;

                    AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_SENSORY_CHAMBER), AvailableTechs.end());
                }

                if (SecondTech.compare("movement") == 0)
                {
                    if (std::find(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_MOVEMENT_CHAMBER) != AvailableTechs.end())
                    {
                        HiveTwoTech = ALIEN_BUILD_MOVEMENT_CHAMBER;
                        AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_MOVEMENT_CHAMBER), AvailableTechs.end());
                    }
                }
                else if (SecondTech.compare("defense") == 0)
                {
                    if (std::find(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_DEFENSE_CHAMBER) != AvailableTechs.end())
                    {
                        HiveTwoTech = ALIEN_BUILD_DEFENSE_CHAMBER;
                        AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_DEFENSE_CHAMBER), AvailableTechs.end());
                    }
                }
                else if (SecondTech.compare("sensory") == 0)
                {
                    if (std::find(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_SENSORY_CHAMBER) != AvailableTechs.end())
                    {
                        HiveTwoTech = ALIEN_BUILD_SENSORY_CHAMBER;
                        AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_SENSORY_CHAMBER), AvailableTechs.end());
                    }
                }

                if (ThirdTech.compare("movement") == 0)
                {
                    if (std::find(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_MOVEMENT_CHAMBER) != AvailableTechs.end())
                    {
                        HiveThreeTech = ALIEN_BUILD_MOVEMENT_CHAMBER;
                        AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_MOVEMENT_CHAMBER), AvailableTechs.end());
                    }
                }
                else if (ThirdTech.compare("defense") == 0)
                {
                    if (std::find(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_DEFENSE_CHAMBER) != AvailableTechs.end())
                    {
                        HiveThreeTech = ALIEN_BUILD_DEFENSE_CHAMBER;
                        AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_DEFENSE_CHAMBER), AvailableTechs.end());
                    }
                }
                else if (ThirdTech.compare("sensory") == 0)
                {
                    if (std::find(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_SENSORY_CHAMBER) != AvailableTechs.end())
                    {
                        HiveThreeTech = ALIEN_BUILD_SENSORY_CHAMBER;
                        AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), ALIEN_BUILD_SENSORY_CHAMBER), AvailableTechs.end());
                    }
                }

                if (HiveOneTech == MESSAGE_NULL)
                {
                    int random = rand() % AvailableTechs.size();
                    HiveOneTech = AvailableTechs[random];

                    AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), HiveOneTech), AvailableTechs.end());
                }

                if (HiveTwoTech == MESSAGE_NULL)
                {
                    int random = rand() % AvailableTechs.size();
                    HiveTwoTech = AvailableTechs[random];

                    AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), HiveTwoTech), AvailableTechs.end());
                }

                if (HiveThreeTech == MESSAGE_NULL)
                {
                    int random = rand() % AvailableTechs.size();
                    HiveThreeTech = AvailableTechs[random];

                    AvailableTechs.erase(std::remove(AvailableTechs.begin(), AvailableTechs.end(), HiveTwoTech), AvailableTechs.end());
                }

                ChamberSequence[0] = HiveOneTech;
                ChamberSequence[1] = HiveTwoTech;
                ChamberSequence[2] = HiveThreeTech;

                continue;
            }
        }
    }
    else
    {
        ALERT(at_console, "nsbots.ini was not found in the NS mod folder. You can regenerate it with the console command 'sv_regenbotini'");
    }
}

BotFillTiming CONFIG_GetBotFillTiming()
{
    return CurrentBotFillTiming;
}

void CONFIG_RegenerateIniFile()
{
    string BotConfigFile = string(getModDirectory()) + "/nsbots.ini";

    const char* filename = BotConfigFile.c_str();

    FILE* NewConfigFile = fopen(filename, "w+");

    if (!NewConfigFile)
    {
        ALERT(at_console, "Unable to write to %s, please ensure the user has privileges\n", filename);
        return;
    }

    fprintf(NewConfigFile, "### General bot settings ###\n\n");

    fprintf(NewConfigFile, "# What prefix to put in front of a bot's name (can leave blank)\n");
    fprintf(NewConfigFile, "prefix=[BOT]\n\n");

    fprintf(NewConfigFile, "# When should the server start adding bots? Note: bots will always be added after round start regardless\n");
    fprintf(NewConfigFile, "# 0 = On map load (after 5 second grace period)\n");
    fprintf(NewConfigFile, "# 1 = When all humans have joined a team (i.e. no more humans left in ready room)\n");
    fprintf(NewConfigFile, "# 2 = When the round has started (after countdown)\n");
    fprintf(NewConfigFile, "BotFillTiming = 1\n\n\n");


    fprintf(NewConfigFile, "### Skill Settings ###\n\n");

    fprintf(NewConfigFile, "# Bot skill settings. You can define as many settings as you like and reference them by name\n");
    fprintf(NewConfigFile, "# Format is BotSkillName = name, followed by one of the following:\n");
    fprintf(NewConfigFile, "# ReactionTime = How quickly in seconds the bot will react to sighting enemies\n");
    fprintf(NewConfigFile, "# AimSkill = How accurately the bot can lock sights on you after seeing you (0.0 - 1.0)\n");
    fprintf(NewConfigFile, "# MovementTracking = How accurately the bot can follow a moving target (0.0 - 1.0)\n");
    fprintf(NewConfigFile, "# ViewSpeed = How fast the bot can swivel its view (0.1 - 2.0)\n");
    fprintf(NewConfigFile, "# Set the difficulty using the 'mp_botskill' cvar (0 - 3)\n\n");

    fprintf(NewConfigFile, "BotSkillLevel=0\n");
    fprintf(NewConfigFile, "MarineReactionTime=0.5\n");
    fprintf(NewConfigFile, "MarineAimSkill=0.1\n");
    fprintf(NewConfigFile, "MarineMovementTracking=0.1\n");
    fprintf(NewConfigFile, "MarineViewSpeed=0.5\n");
    fprintf(NewConfigFile, "AlienReactionTime=0.5\n");
    fprintf(NewConfigFile, "AlienAimSkill=0.2\n");
    fprintf(NewConfigFile, "AlienMovementTracking=0.2\n");
    fprintf(NewConfigFile, "AlienViewSpeed=0.75\n\n");

    fprintf(NewConfigFile, "BotSkillLevel=1\n");
    fprintf(NewConfigFile, "MarineReactionTime=0.2\n");
    fprintf(NewConfigFile, "MarineAimSkill=0.5\n");
    fprintf(NewConfigFile, "MarineMovementTracking=0.4\n");
    fprintf(NewConfigFile, "MarineViewSpeed=1.0\n");
    fprintf(NewConfigFile, "AlienReactionTime=0.2\n");
    fprintf(NewConfigFile, "AlienAimSkill=0.5\n");
    fprintf(NewConfigFile, "AlienMovementTracking=0.5\n");
    fprintf(NewConfigFile, "AlienViewSpeed=1.3\n\n");

    fprintf(NewConfigFile, "BotSkillLevel=2\n");
    fprintf(NewConfigFile, "MarineReactionTime=0.2\n");
    fprintf(NewConfigFile, "MarineAimSkill=0.6\n");
    fprintf(NewConfigFile, "MarineMovementTracking=0.6\n");
    fprintf(NewConfigFile, "MarineViewSpeed=1.5\n");
    fprintf(NewConfigFile, "AlienReactionTime=0.2\n");
    fprintf(NewConfigFile, "AlienAimSkill=0.8\n");
    fprintf(NewConfigFile, "AlienMovementTracking=0.8\n");
    fprintf(NewConfigFile, "AlienViewSpeed=1.5\n\n");

    fprintf(NewConfigFile, "BotSkillLevel=3\n");
    fprintf(NewConfigFile, "MarineReactionTime=0.1\n");
    fprintf(NewConfigFile, "MarineAimSkill=1.0\n");
    fprintf(NewConfigFile, "MarineMovementTracking=1.0\n");
    fprintf(NewConfigFile, "MarineViewSpeed=2.0\n");
    fprintf(NewConfigFile, "AlienReactionTime=0.1\n");
    fprintf(NewConfigFile, "AlienAimSkill=1.0\n");
    fprintf(NewConfigFile, "AlienMovementTracking=1.0\n");
    fprintf(NewConfigFile, "AlienViewSpeed=2.0\n\n");

    fprintf(NewConfigFile, "# Desired team sizes. Only used if bot fill mode is 'fillteams'\n");
    fprintf(NewConfigFile, "# Format is TeamSize=mapname:nummarines/numaliens\n");
    fprintf(NewConfigFile, "# 'default' will be used if playing a map not listed below\n");
    fprintf(NewConfigFile, "TeamSize=default:7/7\n");
    fprintf(NewConfigFile, "TeamSize=ns_machina:8/8\n");
    fprintf(NewConfigFile, "TeamSize=ns_ragnarok:8/8\n");
    fprintf(NewConfigFile, "TeamSize=co_faceoff:4/4\n");
    fprintf(NewConfigFile, "TeamSize=co_core:4/4\n");
    fprintf(NewConfigFile, "TeamSize=co_pulse:6/6\n");
    fprintf(NewConfigFile, "TeamSize=co_ulysses:6/6\n");
    fprintf(NewConfigFile, "TeamSize=co_niveus:5/5\n");
    fprintf(NewConfigFile, "TeamSize=co_kestrel:5/5\n\n\n");


    fprintf(NewConfigFile, "### Alien Settings ###\n\n");

    fprintf(NewConfigFile, "# Preferred chamber sequence. Valid entries are 'defense', 'movement' and 'sensory'. Separate sequence with forward slash\n");
    fprintf(NewConfigFile, "# You can also use ? for random, so if you want movement always first but then defense and sensory at random, use\n");
    fprintf(NewConfigFile, "# ChamberSequence:movement/?/?\n");
    fprintf(NewConfigFile, "# Or if you want sensory always last, but movement and defence random, use\n");
    fprintf(NewConfigFile, "# ChamberSequence=?/?/sensory\n");
    fprintf(NewConfigFile, "ChamberSequence=defense/movement/sensory\n");

    fflush(NewConfigFile);
    fclose(NewConfigFile);

}