diff options
Diffstat (limited to 'src/script/api/script_companymode.cpp')
-rw-r--r-- | src/script/api/script_companymode.cpp | 30 |
1 files changed, 30 insertions, 0 deletions
diff --git a/src/script/api/script_companymode.cpp b/src/script/api/script_companymode.cpp new file mode 100644 index 000000000..d260053e1 --- /dev/null +++ b/src/script/api/script_companymode.cpp @@ -0,0 +1,30 @@ +/* $Id$ */ + +/* + * This file is part of OpenTTD. + * OpenTTD 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, version 2. + * OpenTTD 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 OpenTTD. If not, see <http://www.gnu.org/licenses/>. + */ + +/** @file script_companymode.cpp Implementation of ScriptCompanyMode. */ + +#include "../../stdafx.h" +#include "script_companymode.hpp" +#include "../../company_base.h" +#include "../../company_func.h" +#include "../script_instance.hpp" +#include "../script_fatalerror.hpp" + +ScriptCompanyMode::ScriptCompanyMode(int company) +{ + if (company < OWNER_BEGIN || company >= MAX_COMPANIES) company = INVALID_COMPANY; + + this->last_company = ScriptObject::GetCompany(); + ScriptObject::SetCompany((CompanyID)company); +} + +ScriptCompanyMode::~ScriptCompanyMode() +{ + ScriptObject::SetCompany(this->last_company); +} |