src/script/api/script_companymode.cpp

   1 /*
   2  * This file is part of OpenTTD.
   3  * 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.
   4  * 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.
   5  * 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/>.
   6  */
   7
   8 /** @file script_companymode.cpp Implementation of ScriptCompanyMode. */
   9
  10 #include "../../stdafx.h"
  11 #include "../../company_base.h"
  12 #include "script_companymode.hpp"
  13
  14 #include "../../safeguards.h"
  15
  16 ScriptCompanyMode::ScriptCompanyMode(SQInteger company)
  17 {
  18         if (company < OWNER_BEGIN || company >= MAX_COMPANIES) company = INVALID_COMPANY;
  19         if (!::Company::IsValidID(company)) company = INVALID_COMPANY;
  20
  21         this->last_company = ScriptObject::GetCompany();
  22         ScriptObject::SetCompany((::CompanyID)company);
  23 }
  24
  25 ScriptCompanyMode::~ScriptCompanyMode()
  26 {
  27         ScriptObject::SetCompany(this->last_company);
  28 }
  29
  30 /* static */ bool ScriptCompanyMode::IsValid()
  31 {
  32         return ::Company::IsValidID(ScriptObject::GetCompany());
  33 }
  34
  35 /* static */ bool ScriptCompanyMode::IsDeity()
  36 {
  37         return ScriptObject::GetCompany() == OWNER_DEITY;
  38 }