rail.cpp

Go to the documentation of this file.

/*
 * 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/>.
 */
 
#include "stdafx.h"
#include "station_map.h"
#include "tunnelbridge_map.h"
#include "timer/timer_game_calendar.h"
#include "company_func.h"
#include "company_base.h"
#include "engine_base.h"
 
#include "table/track_data.h"
 
#include "safeguards.h"
 
RailType GetTileRailType(Tile tile)
{
  switch (GetTileType(tile)) {
    case MP_RAILWAY:
      return GetRailType(tile);
 
    case MP_ROAD:
      /* rail/road crossing */
      if (IsLevelCrossing(tile)) return GetRailType(tile);
      break;
 
    case MP_STATION:
      if (HasStationRail(tile)) return GetRailType(tile);
      break;
 
    case MP_TUNNELBRIDGE:
      if (GetTunnelBridgeTransportType(tile) == TRANSPORT_RAIL) return GetRailType(tile);
      break;
 
    default:
      break;
  }
  return INVALID_RAILTYPE;
}
 
bool HasRailTypeAvail(const CompanyID company, const RailType railtype)
{
  return !_railtypes_hidden_mask.Test(railtype) && Company::Get(company)->avail_railtypes.Test(railtype);
}
 
bool HasAnyRailTypesAvail(const CompanyID company)
{
  RailTypes avail = Company::Get(company)->avail_railtypes;
  avail.Reset(_railtypes_hidden_mask);
  return avail.Any();
}
 
bool ValParamRailType(const RailType rail)
{
  return rail < RAILTYPE_END && HasRailTypeAvail(_current_company, rail);
}
 
RailTypes AddDateIntroducedRailTypes(RailTypes current, TimerGameCalendar::Date date)
{
  RailTypes rts = current;
 
  for (RailType rt = RAILTYPE_BEGIN; rt != RAILTYPE_END; rt++) {
    const RailTypeInfo *rti = GetRailTypeInfo(rt);
    /* Unused rail type. */
    if (rti->label == 0) continue;
 
    /* Not date introduced. */
    if (!IsInsideMM(rti->introduction_date, 0, CalendarTime::MAX_DATE.base())) continue;
 
    /* Not yet introduced at this date. */
    if (rti->introduction_date > date) continue;
 
    /* Have we introduced all required railtypes? */
    RailTypes required = rti->introduction_required_railtypes;
    if (!rts.All(required)) continue;
 
    rts.Set(rti->introduces_railtypes);
  }
 
  /* When we added railtypes we need to run this method again; the added
   * railtypes might enable more rail types to become introduced. */
  return rts == current ? rts : AddDateIntroducedRailTypes(rts, date);
}
 
RailTypes GetCompanyRailTypes(CompanyID company, bool introduces)
{
  RailTypes rts{};
 
  for (const Engine *e : Engine::IterateType(VEH_TRAIN)) {
    const EngineInfo *ei = &e->info;
 
    if (ei->climates.Test(_settings_game.game_creation.landscape) &&
        (e->company_avail.Test(company) || TimerGameCalendar::date >= e->intro_date + CalendarTime::DAYS_IN_YEAR)) {
      const RailVehicleInfo *rvi = &e->u.rail;
 
      if (rvi->railveh_type != RAILVEH_WAGON) {
        assert(rvi->railtypes.Any());
        if (introduces) {
          rts.Set(GetAllIntroducesRailTypes(rvi->railtypes));
        } else {
          rts.Set(rvi->railtypes);
        }
      }
    }
  }
 
  if (introduces) return AddDateIntroducedRailTypes(rts, TimerGameCalendar::date);
  return rts;
}
 
RailTypes GetRailTypes(bool introduces)
{
  RailTypes rts{};
 
  for (const Engine *e : Engine::IterateType(VEH_TRAIN)) {
    const EngineInfo *ei = &e->info;
    if (!ei->climates.Test(_settings_game.game_creation.landscape)) continue;
 
    const RailVehicleInfo *rvi = &e->u.rail;
    if (rvi->railveh_type != RAILVEH_WAGON) {
      assert(rvi->railtypes.Any());
      if (introduces) {
        rts.Set(GetAllIntroducesRailTypes(rvi->railtypes));
      } else {
        rts.Set(rvi->railtypes);
      }
    }
  }
 
  if (introduces) return AddDateIntroducedRailTypes(rts, CalendarTime::MAX_DATE);
  return rts;
}
 
RailType GetRailTypeByLabel(RailTypeLabel label, bool allow_alternate_labels)
{
  if (label == 0) return INVALID_RAILTYPE;
 
  /* Loop through each rail type until the label is found */
  for (RailType r = RAILTYPE_BEGIN; r != RAILTYPE_END; r++) {
    const RailTypeInfo *rti = GetRailTypeInfo(r);
    if (rti->label == label) return r;
  }
 
  if (allow_alternate_labels) {
    /* Test if any rail type defines the label as an alternate. */
    for (RailType r = RAILTYPE_BEGIN; r != RAILTYPE_END; r++) {
      const RailTypeInfo *rti = GetRailTypeInfo(r);
      if (std::ranges::find(rti->alternate_labels, label) != rti->alternate_labels.end()) return r;
    }
  }
 
  /* No matching label was found, so it is invalid */
  return INVALID_RAILTYPE;
}