cp_xvi.cpp

Go to the documentation of this file.

/*
Copyright (C) 2002-2020 UFO: Alien Invasion.

This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.

See the GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.

*/

#include "../../cl_shared.h"
#include "cp_campaign.h"
#include "cp_overlay.h"
#include "cp_geoscape.h"
#include "cp_xvi.h"
#include "cp_missions.h"
#include "save/save_xvi.h"

static technology_t* rsAlienXVI;

static bool xviNationInfectionNeedsUpdate = false;

static const float XVI_FACTOR = 15.0f;

static const int XVI_DECREASE_DAYS = 7;

#define XVI_EVENT_NAME "rs_alien_xvi_event"

void CP_SpreadXVIAtPos (const vec2_t pos)
{
    if (!CP_IsXVIStarted())
        return;

    CP_ChangeXVILevel(pos, XVI_FACTOR);

    xviNationInfectionNeedsUpdate = true;
}

void CP_ReduceXVIEverywhere (void)
{
    if (!CP_IsXVIStarted())
        return;

    /* Only decrease XVI if given days has passed */
    if (ccs.date.day % XVI_DECREASE_DAYS)
        return;

    CP_DecreaseXVILevelEverywhere();

    xviNationInfectionNeedsUpdate = true;
}

void CP_UpdateNationXVIInfection (void)
{
    /* No need to update XVI levels if the overlay didn't change */
    if (!xviNationInfectionNeedsUpdate)
        return;

    /* width in pixel of the XVI overlay */
    int width;
    /* height in pixel of the XVI overlay */
    int height;
    CP_GetXVIMapDimensions(&width, &height);

    const float heightPerDegree = height / 180.0f;
    const float widthPerDegree = width / 360.0f;
    /* parameter used to normalize nation XVI level.
     * decrease this factor to increase XVI level per nation */
    const float AREA_FACTOR = 650.0f;
    /* area used to normalized XVI infection level for each nation.
     * depend on overlay size so that if we change resolution of
     * overlay it doesn't impact nation XIInfection */
    const float normalizingArea = width * height / AREA_FACTOR;

    /* temporary array to store the XVI levels */
    float* xviInfection = Mem_PoolAllocTypeN(float, ccs.numNations, cp_campaignPool);
    OBJZERO(*xviInfection);

    for (int y = 0; y < height; y++) {
        const byte* previousNationColor;
        const nation_t* nation;
        /* current position (in latitude / longitude) */
        vec2_t currentPos;

        int* sum = Mem_PoolAllocTypeN(int, ccs.numNations, cp_campaignPool);
        OBJZERO(*sum);

        Vector2Set(currentPos, 180.0f, 90.0f - y / heightPerDegree);
        previousNationColor = GEO_GetColor(currentPos, MAPTYPE_NATIONS, nullptr);
        nation = GEO_GetNation(currentPos);

        for (int x = 0; x < width; x++) {
            const byte* nationColor;
            currentPos[0] = 180.0f - x / widthPerDegree;
            nationColor = GEO_GetColor(currentPos, MAPTYPE_NATIONS, nullptr);
            if (!VectorCompare(nationColor, previousNationColor)) {
                previousNationColor = nationColor;
                nation = GEO_GetNation(currentPos);
            }
            if (nation) {
                const int xviLevel = CP_GetXVILevel(x, y);
                if (xviLevel > 0)
                    sum[nation->idx] += xviLevel;
            }
        }
        /* divide the total XVI infection by the area of a pixel
         * because pixel are smaller as you go closer from the pole */
        for (int nationIdx = 0; nationIdx < ccs.numNations; nationIdx++)
            xviInfection[nationIdx] += ((float) sum[nationIdx]) / (cos(torad * currentPos[1]) * normalizingArea);
        cgi->Free(sum);
    }

    /* copy the new values of XVI infection level into nation array */
    int nationIdx = 0;
    NAT_Foreach(nation) {
        nation->stats[0].xviInfection = ceil(xviInfection[nationIdx]);
        nationIdx++;
    }
    cgi->Free(xviInfection);

    xviNationInfectionNeedsUpdate = false;
}

int CP_GetAverageXVIRate (void)
{
    assert(ccs.numNations);

    /* check for XVI infection rate */
    int xviRate = 0;
    NAT_Foreach(nation) {
        const nationInfo_t* stats = NAT_GetCurrentMonthInfo(nation);
        xviRate += stats->xviInfection;
    }
    xviRate /= ccs.numNations;
    return xviRate;
}

void CP_SpreadXVI (void)
{
    /* don't check if XVI spreading didn't start yet */
    if (!CP_IsXVIStarted())
        return;

    MIS_Foreach(mission) {
        if (mission->stage == STAGE_SPREAD_XVI)
            CP_SpreadXVIAtPos(mission->pos);
    }
}

bool CP_IsXVIVisible (void)
{
    return RS_IsResearched_ptr(rsAlienXVI);
}

void CP_XVIInit (void)
{
    rsAlienXVI = RS_GetTechByID(XVI_EVENT_NAME);
    if (!rsAlienXVI)
        cgi->Com_Error(ERR_DROP, "CP_XVIInit: Could not find tech definition for " XVI_EVENT_NAME);
}

bool XVI_SaveXML (xmlNode_t* p)
{
    int width;
    int height;

    CP_GetXVIMapDimensions(&width, &height);

    xmlNode_t* n = cgi->XML_AddNode(p, SAVE_XVI_XVI);
    cgi->XML_AddInt(n, SAVE_XVI_WIDTH, width);
    cgi->XML_AddInt(n, SAVE_XVI_HEIGHT, height);

    for (int y = 0; y < height; y++) {
        for (int x = 0; x < width; x++) {
            const int xviLevel = CP_GetXVILevel(x, y);
            /* That saves many bytes in the savegame */
            if (xviLevel > 0) {
                xmlNode_t* s = cgi->XML_AddNode(n, SAVE_XVI_ENTRY);
                cgi->XML_AddInt(s, SAVE_XVI_X, x);
                cgi->XML_AddInt(s, SAVE_XVI_Y, y);
                cgi->XML_AddInt(s, SAVE_XVI_LEVEL, xviLevel);
            }
        }
    }
    return true;
}

bool XVI_LoadXML (xmlNode_t* p)
{
    int width, height;
    xmlNode_t* s;
    xmlNode_t* n = cgi->XML_GetNode(p, SAVE_XVI_XVI);
    /* If there is no XVI, it will not be loaded */
    if (!n) {
        CP_InitializeXVIOverlay();
        return true;
    }

    width = cgi->XML_GetInt(n, SAVE_XVI_WIDTH, 0);
    height = cgi->XML_GetInt(n, SAVE_XVI_HEIGHT, 0);

    for (s = cgi->XML_GetNode(n, SAVE_XVI_ENTRY); s; s = cgi->XML_GetNextNode(s, n, SAVE_XVI_ENTRY)) {
        const int x = cgi->XML_GetInt(s, SAVE_XVI_X, 0);
        const int y = cgi->XML_GetInt(s, SAVE_XVI_Y, 0);
        const int level = cgi->XML_GetInt(s, SAVE_XVI_LEVEL, 0);

        if (x >= 0 && x < width && y >= 0 && y <= height)
            CP_SetXVILevel(x, y, level);
    }
    return true;
}

void CP_StartXVISpreading_f (void)
{
    int numAlienBases;
    const campaign_t* campaign = ccs.curCampaign;

    ccs.startXVI = true;

    /* Spawn a few alien bases depending on difficulty level */
    if (campaign->difficulty > 0)
        numAlienBases = 3;
    else if (campaign->difficulty < 0)
        numAlienBases = 1;
    else
        numAlienBases = 2;

    for (int i = 0; i < numAlienBases; i++)
        CP_CreateNewMission(INTERESTCATEGORY_BUILDING, false);
}

void CP_UpdateXVIMapButton (void)
{
    cgi->Cvar_SetValue("mn_xvimap", CP_IsXVIVisible());
}