(svn r19914) -Codechange: Wrap a helper class around temporary assignments of _current_company to ensure proper restoration.

This commit is contained in:
frosch
2010-05-31 20:22:57 +00:00
parent 1d05fbebbe
commit 9db3cde73a
19 changed files with 256 additions and 117 deletions

View File

@@ -10,6 +10,7 @@
/** @file ai_core.cpp Implementation of AI. */
#include "../stdafx.h"
#include "../core/backup_type.hpp"
#include "../core/bitmath_func.hpp"
#include "../company_base.h"
#include "../company_func.h"
@@ -68,13 +69,15 @@
assert(_settings_game.difficulty.competitor_speed <= 4);
if ((AI::frame_counter & ((1 << (4 - _settings_game.difficulty.competitor_speed)) - 1)) != 0) return;
Backup<CompanyByte> cur_company(_current_company);
const Company *c;
FOR_ALL_COMPANIES(c) {
if (c->is_ai) {
_current_company = c->index;
cur_company.Change(c->index);
c->ai_instance->GameLoop();
}
}
cur_company.Restore();
/* Occasionally collect garbage; every 255 ticks do one company.
* Effectively collecting garbage once every two months per AI. */
@@ -82,8 +85,6 @@
CompanyID cid = (CompanyID)GB(AI::frame_counter, 8, 4);
if (Company::IsValidAiID(cid)) Company::Get(cid)->ai_instance->CollectGarbage();
}
_current_company = OWNER_NONE;
}
/* static */ uint AI::GetTick()
@@ -95,14 +96,13 @@
{
if (_networking && !_network_server) return;
CompanyID old_company = _current_company;
_current_company = company;
Backup<CompanyByte> cur_company(_current_company, company);
Company *c = Company::Get(company);
delete c->ai_instance;
c->ai_instance = NULL;
_current_company = old_company;
cur_company.Restore();
InvalidateWindowData(WC_AI_DEBUG, 0, -1);
DeleteWindowById(WC_AI_SETTINGS, company);
@@ -112,11 +112,10 @@
{
if (_networking && !_network_server) return;
CompanyID old_company = _current_company;
_current_company = company;
Backup<CompanyByte> cur_company(_current_company, company);
Company::Get(company)->ai_instance->Suspend();
_current_company = old_company;
cur_company.Restore();
}
/* static */ void AI::KillAll()
@@ -202,10 +201,9 @@
}
/* Queue the event */
CompanyID old_company = _current_company;
_current_company = company;
Backup<CompanyByte> cur_company(_current_company, company);
AIEventController::InsertEvent(event);
_current_company = old_company;
cur_company.Restore();
event->Release();
}
@@ -249,10 +247,9 @@ void CcAI(const CommandCost &result, TileIndex tile, uint32 p1, uint32 p2)
Company *c = Company::GetIfValid(company);
assert(c != NULL && c->ai_instance != NULL);
CompanyID old_company = _current_company;
_current_company = company;
Backup<CompanyByte> cur_company(_current_company, company);
c->ai_instance->Save();
_current_company = old_company;
cur_company.Restore();
} else {
AIInstance::SaveEmpty();
}
@@ -264,10 +261,9 @@ void CcAI(const CommandCost &result, TileIndex tile, uint32 p1, uint32 p2)
Company *c = Company::GetIfValid(company);
assert(c != NULL && c->ai_instance != NULL);
CompanyID old_company = _current_company;
_current_company = company;
Backup<CompanyByte> cur_company(_current_company, company);
c->ai_instance->Load(version);
_current_company = old_company;
cur_company.Restore();
} else {
/* Read, but ignore, the load data */
AIInstance::LoadEmpty();