vehiclelist.h

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 <https://www.gnu.org/licenses/old-licenses/gpl-2.0>.
 */

 
#ifndef VEHICLELIST_H
#define VEHICLELIST_H
 
#include "company_type.h"
#include "group_type.h"
#include "order_type.h"
#include "station_type.h"
#include "tile_type.h"
#include "vehicle_type.h"
#include "window_type.h"

enum class VehicleListType : uint8_t {
  Company, 
  VehicleSharedOrders, 
  Station, 
  Depot, 
  Group, 
  End, 
};

struct VehicleListIdentifier {
  VehicleListType type; 
  VehicleType vtype;    
  CompanyID company;    
  uint32_t index;         
 
  WindowNumber ToWindowNumber() const;
 
  bool Valid() const { return this->type < VehicleListType::End; }
 
  constexpr CompanyID ToCompanyID() const { assert(this->type == VehicleListType::Company); return CompanyID(this->index); }
  constexpr DestinationID ToDestinationID() const { assert(this->type == VehicleListType::Depot); return DestinationID(this->index); }
  constexpr GroupID ToGroupID() const { assert(this->type == VehicleListType::Group); return GroupID(this->index); }
  constexpr StationID ToStationID() const { assert(this->type == VehicleListType::Station); return StationID(this->index); }
  constexpr VehicleID ToVehicleID() const { assert(this->type == VehicleListType::VehicleSharedOrders); return VehicleID(this->index); }
 
  constexpr void SetIndex(uint32_t index) { this->index = index; }
  constexpr void SetIndex(ConvertibleThroughBase auto index) { this->index = index.base(); }

  VehicleListIdentifier(VehicleListType type, VehicleType vtype, CompanyID company, uint index = 0) :
    type(type), vtype(vtype), company(company), index(index) {}
 
  VehicleListIdentifier(VehicleListType type, VehicleType vtype, CompanyID company, ConvertibleThroughBase auto index) :
    type(type), vtype(vtype), company(company), index(index.base()) {}
 
  VehicleListIdentifier() = default;
};

typedef std::vector<const Vehicle *> VehicleList;
 
bool GenerateVehicleSortList(VehicleList *list, const VehicleListIdentifier &identifier);
void BuildDepotVehicleList(VehicleType type, TileIndex tile, VehicleList *engine_list, VehicleList *wagon_list, bool individual_wagons = false);
uint GetUnitNumberDigits(VehicleList &vehicles);
 
#endif /* VEHICLELIST_H */