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

 
#include "stdafx.h"
#include "gui.h"
#include "station_base.h"
#include "waypoint_base.h"
#include "window_gui.h"
#include "station_gui.h"
#include "terraform_gui.h"
#include "viewport_func.h"
#include "command_func.h"
#include "waypoint_func.h"
#include "newgrf_badge.h"
#include "newgrf_badge_gui.h"
#include "newgrf_station.h"
#include "company_base.h"
#include "strings_func.h"
#include "window_func.h"
#include "sound_func.h"
#include "company_func.h"
#include "dropdown_type.h"
#include "dropdown_func.h"
#include "tunnelbridge.h"
#include "tilehighlight_func.h"
#include "core/geometry_func.hpp"
#include "hotkeys.h"
#include "engine_base.h"
#include "vehicle_func.h"
#include "zoom_func.h"
#include "rail_gui.h"
#include "toolbar_gui.h"
#include "station_cmd.h"
#include "tunnelbridge_cmd.h"
#include "waypoint_cmd.h"
#include "rail_cmd.h"
#include "timer/timer.h"
#include "timer/timer_game_calendar.h"
#include "picker_gui.h"
 
#include "station_map.h"
#include "tunnelbridge_map.h"
 
#include "widgets/rail_widget.h"
 
#include "table/strings.h"
 
#include "safeguards.h"
 
 
static RailType _cur_railtype;               
static bool _remove_button_clicked;          
static DiagDirection _build_depot_direction; 
static bool _convert_signal_button;          
static SignalVariant _cur_signal_variant;    
static SignalType _cur_signal_type;          
 
struct WaypointPickerSelection {
  StationClassID sel_class; 
  uint16_t sel_type; 
};
static WaypointPickerSelection _waypoint_gui; 
 
struct StationPickerSelection {
  StationClassID sel_class; 
  uint16_t sel_type; 
  Axis axis; 
};
static StationPickerSelection _station_gui; 
 
 
static void HandleStationPlacement(TileIndex start, TileIndex end);
static void ShowBuildTrainDepotPicker(Window *parent);
static void ShowBuildWaypointPicker(Window *parent);
static Window *ShowStationBuilder(Window *parent);
static void ShowSignalBuilder(Window *parent);

static bool IsStationAvailable(const StationSpec *statspec)
{
  if (statspec == nullptr || !statspec->callback_mask.Test(StationCallbackMask::Avail)) return true;
 
  uint16_t cb_res = GetStationCallback(CBID_STATION_AVAILABILITY, 0, 0, statspec, nullptr, INVALID_TILE);
  if (cb_res == CALLBACK_FAILED) return true;
 
  return Convert8bitBooleanCallback(statspec->grf_prop.grffile, CBID_STATION_AVAILABILITY, cb_res);
}
 
void CcPlaySound_CONSTRUCTION_RAIL(Commands, const CommandCost &result, TileIndex tile)
{
  if (result.Succeeded() && _settings_client.sound.confirm) SndPlayTileFx(SND_20_CONSTRUCTION_RAIL, tile);
}
 
static void GenericPlaceRail(TileIndex tile, Track track)
{
  if (_remove_button_clicked) {
    Command<Commands::RemoveRail>::Post(STR_ERROR_CAN_T_REMOVE_RAILROAD_TRACK, CcPlaySound_CONSTRUCTION_RAIL,
        tile, track);
  } else {
    Command<Commands::BuildRail>::Post(STR_ERROR_CAN_T_BUILD_RAILROAD_TRACK, CcPlaySound_CONSTRUCTION_RAIL,
        tile, _cur_railtype, track, _settings_client.gui.auto_remove_signals);
  }
}

static void PlaceExtraDepotRail(TileIndex tile, DiagDirection dir, Track track)
{
  if (GetRailTileType(tile) == RailTileType::Depot) return;
  if (GetRailTileType(tile) == RailTileType::Signals && !_settings_client.gui.auto_remove_signals) return;
  if (!GetTrackBits(tile).Any(DiagdirReachesTracks(dir))) return;
 
  Command<Commands::BuildRail>::Post(tile, _cur_railtype, track, _settings_client.gui.auto_remove_signals);
}

static constexpr std::array<DiagDirectionIndexArray<Track>, 3> _place_depot_extra_track{{
  {Track::Left,  Track::Upper, Track::Upper, Track::Right}, // First additional track for directions 0..3
  {Track::X,     Track::Y,     Track::X,     Track::Y},     // Second additional track
  {Track::Lower, Track::Left,  Track::Right, Track::Lower}, // Third additional track
}};

static constexpr std::array<DiagDirectionIndexArray<DiagDirection>, 3> _place_depot_extra_dir{{
  {DiagDirection::SE, DiagDirection::SW, DiagDirection::SE, DiagDirection::SW}, // First additional track for directions 0..3
  {DiagDirection::SW, DiagDirection::NW, DiagDirection::NE, DiagDirection::SE}, // Second additional track
  {DiagDirection::NW, DiagDirection::NE, DiagDirection::NW, DiagDirection::NE}, // Third additional track
}};
 
void CcRailDepot(Commands, const CommandCost &result, TileIndex tile, RailType, DiagDirection dir)
{
  if (result.Failed()) return;
 
  if (_settings_client.sound.confirm) SndPlayTileFx(SND_20_CONSTRUCTION_RAIL, tile);
  if (!_settings_client.gui.persistent_buildingtools) ResetObjectToPlace();
 
  tile += TileOffsByDiagDir(dir);
 
  if (IsTileType(tile, TileType::Railway)) {
    PlaceExtraDepotRail(tile, _place_depot_extra_dir[0][dir], _place_depot_extra_track[0][dir]);
 
    /* Don't place the rail straight out of the depot of there is another depot across from it. */
    Tile double_depot_tile = tile + TileOffsByDiagDir(dir);
    bool is_double_depot = IsValidTile(double_depot_tile) && IsRailDepotTile(double_depot_tile);
    if (!is_double_depot) PlaceExtraDepotRail(tile, _place_depot_extra_dir[1][dir], _place_depot_extra_track[1][dir]);
 
    PlaceExtraDepotRail(tile, _place_depot_extra_dir[2][dir], _place_depot_extra_track[2][dir]);
  }
}

static void PlaceRail_Waypoint(TileIndex tile)
{
  if (_remove_button_clicked) {
    VpStartPlaceSizing(tile, VPM_X_AND_Y, DDSP_REMOVE_STATION);
    return;
  }
 
  Axis axis = GetAxisForNewRailWaypoint(tile);
  if (IsValidAxis(axis)) {
    /* Valid tile for waypoints */
    VpStartPlaceSizing(tile, axis == Axis::X ? VPM_X_LIMITED : VPM_Y_LIMITED, DDSP_BUILD_STATION);
    VpSetPlaceSizingLimit(_settings_game.station.station_spread);
  } else {
    /* Tile where we can't build rail waypoints. This is always going to fail,
     * but provides the user with a proper error message. */
    Command<Commands::BuildRailWaypoint>::Post(STR_ERROR_CAN_T_BUILD_RAIL_WAYPOINT , tile, Axis::X, 1, 1, STAT_CLASS_WAYP, 0, StationID::Invalid(), false);
  }
}
 
void CcStation(Commands, const CommandCost &result, TileIndex tile)
{
  if (result.Failed()) return;
 
  if (_settings_client.sound.confirm) SndPlayTileFx(SND_20_CONSTRUCTION_RAIL, tile);
  /* Only close the station builder window if the default station and non persistent building is chosen. */
  if (_station_gui.sel_class == STAT_CLASS_DFLT && _station_gui.sel_type == 0 && !_settings_client.gui.persistent_buildingtools) ResetObjectToPlace();
}

static void PlaceRail_Station(TileIndex tile)
{
  if (_remove_button_clicked) {
    VpStartPlaceSizing(tile, VPM_X_AND_Y_LIMITED, DDSP_REMOVE_STATION);
    VpSetPlaceSizingLimit(-1);
  } else if (_settings_client.gui.station_dragdrop) {
    VpStartPlaceSizing(tile, VPM_X_AND_Y_LIMITED, DDSP_BUILD_STATION);
    VpSetPlaceSizingLimit(_settings_game.station.station_spread);
  } else {
    int w = _settings_client.gui.station_numtracks;
    int h = _settings_client.gui.station_platlength;
    if (_station_gui.axis == Axis::X) std::swap(w, h);
 
    StationPickerSelection params = _station_gui;
    RailType rt = _cur_railtype;
    uint8_t numtracks = _settings_client.gui.station_numtracks;
    uint8_t platlength = _settings_client.gui.station_platlength;
    bool adjacent = _ctrl_pressed;
 
    auto proc = [=](bool test, StationID to_join) -> bool {
      if (test) {
        return Command<Commands::BuildRailStation>::Do(CommandFlagsToDCFlags(GetCommandFlags<Commands::BuildRailStation>()), tile, rt, params.axis, numtracks, platlength, params.sel_class, params.sel_type, StationID::Invalid(), adjacent).Succeeded();
      } else {
        return Command<Commands::BuildRailStation>::Post(STR_ERROR_CAN_T_BUILD_RAILROAD_STATION, CcStation, tile, rt, params.axis, numtracks, platlength, params.sel_class, params.sel_type, to_join, adjacent);
      }
    };
 
    ShowSelectStationIfNeeded(TileArea(tile, w, h), proc);
  }
}

static void GenericPlaceSignals(TileIndex tile)
{
  TrackBits trackbits = TrackdirBitsToTrackBits(GetTileTrackStatus(tile, TransportType::Rail, RoadTramType::Invalid).trackdirs);
 
  if (trackbits.Any(TRACK_BIT_VERT)) { // N-S direction
    trackbits = (_tile_fract_coords.x <= _tile_fract_coords.y) ? Track::Right : Track::Left;
  }
 
  if (trackbits.Any(TRACK_BIT_HORZ)) { // E-W direction
    trackbits = (_tile_fract_coords.x + _tile_fract_coords.y <= 15) ? Track::Upper : Track::Lower;
  }
 
  Track track = FindFirstTrack(trackbits);
 
  if (_remove_button_clicked) {
    Command<Commands::RemoveSignal>::Post(STR_ERROR_CAN_T_REMOVE_SIGNALS_FROM, CcPlaySound_CONSTRUCTION_RAIL, tile, track);
  } else {
    /* Which signals should we cycle through? */
    bool tile_has_signal = IsPlainRailTile(tile) && IsValidTrack(track) && HasSignalOnTrack(tile, track);
    SignalType cur_signal_on_tile = tile_has_signal ? GetSignalType(tile, track) : _cur_signal_type;
    SignalType cycle_start;
    SignalType cycle_end;
 
    /* Start with the least restrictive case: the player wants to cycle through all signals they can see. */
    if (_settings_client.gui.cycle_signal_types == SIGNAL_CYCLE_ALL) {
      cycle_start = _settings_client.gui.signal_gui_mode == SIGNAL_GUI_ALL ? SignalType::Block : SignalType::Path;
      cycle_end = SignalType::PathOneWay;
    } else {
      /* Only cycle through signals of the same group (block or path) as the current signal on the tile. */
      if (cur_signal_on_tile <= SignalType::Combo) {
        /* Block signals only. */
        cycle_start = SignalType::Block;
        cycle_end = SignalType::Combo;
      } else {
        /* Path signals only. */
        cycle_start = SignalType::Path;
        cycle_end = SignalType::PathOneWay;
      }
    }
 
    if (FindWindowById(WindowClass::BuildSignal, 0) != nullptr) {
      /* signal GUI is used */
      Command<Commands::BuildSignal>::Post(_convert_signal_button ? STR_ERROR_SIGNAL_CAN_T_CONVERT_SIGNALS_HERE : STR_ERROR_CAN_T_BUILD_SIGNALS_HERE, CcPlaySound_CONSTRUCTION_RAIL,
        tile, track, _cur_signal_type, _cur_signal_variant, _convert_signal_button, false, _ctrl_pressed, cycle_start, cycle_end, 0, 0);
    } else {
      SignalVariant sigvar = TimerGameCalendar::year < _settings_client.gui.semaphore_build_before ? SignalVariant::Semaphore : SignalVariant::Electric;
      Command<Commands::BuildSignal>::Post(STR_ERROR_CAN_T_BUILD_SIGNALS_HERE, CcPlaySound_CONSTRUCTION_RAIL,
        tile, track, _settings_client.gui.default_signal_type, sigvar, false, false, _ctrl_pressed, cycle_start, cycle_end, 0, 0);
 
    }
  }
}

static void PlaceRail_Bridge(TileIndex tile, Window *w)
{
  if (IsBridgeTile(tile)) {
    TileIndex other_tile = GetOtherTunnelBridgeEnd(tile);
    Point pt = {0, 0};
    w->OnPlaceMouseUp(VPM_X_OR_Y, DDSP_BUILD_BRIDGE, pt, other_tile, tile);
  } else {
    VpStartPlaceSizing(tile, VPM_X_OR_Y, DDSP_BUILD_BRIDGE);
  }
}

void CcBuildRailTunnel(Commands, const CommandCost &result, TileIndex tile)
{
  if (result.Succeeded()) {
    if (_settings_client.sound.confirm) SndPlayTileFx(SND_20_CONSTRUCTION_RAIL, tile);
    if (!_settings_client.gui.persistent_buildingtools) ResetObjectToPlace();
  } else {
    SetRedErrorSquare(_build_tunnel_endtile);
  }
}

static void ToggleRailButton_Remove(Window *w)
{
  CloseWindowById(WindowClass::JoinStation, 0);
  w->ToggleWidgetLoweredState(WID_RAT_REMOVE);
  w->SetWidgetDirty(WID_RAT_REMOVE);
  _remove_button_clicked = w->IsWidgetLowered(WID_RAT_REMOVE);
  SetSelectionRed(_remove_button_clicked);
}

static bool RailToolbar_CtrlChanged(Window *w)
{
  if (w->IsWidgetDisabled(WID_RAT_REMOVE)) return false;
 
  /* allow ctrl to switch remove mode only for these widgets */
  for (WidgetID i = WID_RAT_BUILD_NS; i <= WID_RAT_BUILD_STATION; i++) {
    if ((i <= WID_RAT_AUTORAIL || i >= WID_RAT_BUILD_WAYPOINT) && w->IsWidgetLowered(i)) {
      ToggleRailButton_Remove(w);
      return true;
    }
  }
 
  return false;
}
 

static void BuildRailClick_Remove(Window *w)
{
  if (w->IsWidgetDisabled(WID_RAT_REMOVE)) return;
  ToggleRailButton_Remove(w);
  SndClickBeep();
 
  /* handle station builder */
  if (w->IsWidgetLowered(WID_RAT_BUILD_STATION)) {
    if (_remove_button_clicked) {
      /* starting drag & drop remove */
      if (!_settings_client.gui.station_dragdrop) {
        SetTileSelectSize(1, 1);
      } else {
        VpSetPlaceSizingLimit(-1);
      }
    } else {
      /* starting station build mode */
      if (!_settings_client.gui.station_dragdrop) {
        int x = _settings_client.gui.station_numtracks;
        int y = _settings_client.gui.station_platlength;
        if (_station_gui.axis == Axis::X) std::swap(x, y);
        SetTileSelectSize(x, y);
      } else {
        VpSetPlaceSizingLimit(_settings_game.station.station_spread);
      }
    }
  }
}
 
static void DoRailroadTrack(Track track)
{
  if (_remove_button_clicked) {
    Command<Commands::RemoveRailLong>::Post(STR_ERROR_CAN_T_REMOVE_RAILROAD_TRACK, CcPlaySound_CONSTRUCTION_RAIL,
        TileVirtXY(_thd.selend.x, _thd.selend.y), TileVirtXY(_thd.selstart.x, _thd.selstart.y), track);
  } else {
    Command<Commands::BuildRailLong>::Post(STR_ERROR_CAN_T_BUILD_RAILROAD_TRACK, CcPlaySound_CONSTRUCTION_RAIL,
        TileVirtXY(_thd.selend.x, _thd.selend.y), TileVirtXY(_thd.selstart.x, _thd.selstart.y),  _cur_railtype, track, _settings_client.gui.auto_remove_signals, false);
  }
}
 
static void HandleAutodirPlacement()
{
  Track trackstat = static_cast<Track>( _thd.drawstyle & HT_DIR_MASK); // 0..5
 
  if (_thd.drawstyle & HT_RAIL) { // one tile case
    GenericPlaceRail(TileVirtXY(_thd.selend.x, _thd.selend.y), trackstat);
    return;
  }
 
  DoRailroadTrack(trackstat);
}

static void HandleAutoSignalPlacement()
{
  Track track = static_cast<Track>(_thd.drawstyle & HT_DIR_MASK); // 0..5
 
  if ((_thd.drawstyle & HT_DRAG_MASK) == HT_RECT) { // one tile case
    GenericPlaceSignals(TileVirtXY(_thd.selend.x, _thd.selend.y));
    return;
  }
 
  /* _settings_client.gui.drag_signals_density is given as a parameter such that each user
   * in a network game can specify their own signal density */
  if (_remove_button_clicked) {
    Command<Commands::RemoveSignalLong>::Post(STR_ERROR_CAN_T_REMOVE_SIGNALS_FROM, CcPlaySound_CONSTRUCTION_RAIL,
        TileVirtXY(_thd.selstart.x, _thd.selstart.y), TileVirtXY(_thd.selend.x, _thd.selend.y), track, _ctrl_pressed);
  } else {
    bool sig_gui = FindWindowById(WindowClass::BuildSignal, 0) != nullptr;
    SignalType sigtype = sig_gui ? _cur_signal_type : _settings_client.gui.default_signal_type;
    SignalVariant sigvar = sig_gui ? _cur_signal_variant : (TimerGameCalendar::year < _settings_client.gui.semaphore_build_before ? SignalVariant::Semaphore : SignalVariant::Electric);
    Command<Commands::BuildSignalLong>::Post(STR_ERROR_CAN_T_BUILD_SIGNALS_HERE, CcPlaySound_CONSTRUCTION_RAIL,
        TileVirtXY(_thd.selstart.x, _thd.selstart.y), TileVirtXY(_thd.selend.x, _thd.selend.y), track, sigtype, sigvar, false, _ctrl_pressed, !_settings_client.gui.drag_signals_fixed_distance, _settings_client.gui.drag_signals_density);
  }
}
 

struct BuildRailToolbarWindow : Window {
  RailType railtype = INVALID_RAILTYPE; 
  WidgetID last_user_action = INVALID_WIDGET; 
 
  BuildRailToolbarWindow(WindowDesc &desc, RailType railtype) : Window(desc), railtype(railtype)
  {
    this->CreateNestedTree();
    this->FinishInitNested(TransportType::Rail);
    this->DisableWidget(WID_RAT_REMOVE);
    this->OnInvalidateData();
 
    if (_settings_client.gui.link_terraform_toolbar) ShowTerraformToolbar(this);
  }
 
  void Close([[maybe_unused]] int data = 0) override
  {
    if (this->IsWidgetLowered(WID_RAT_BUILD_STATION)) SetViewportCatchmentStation(nullptr, true);
    if (this->IsWidgetLowered(WID_RAT_BUILD_WAYPOINT)) SetViewportCatchmentWaypoint(nullptr, true);
    if (_settings_client.gui.link_terraform_toolbar) CloseWindowById(WindowClass::ScenarioGenerateLandscape, 0, false);
    CloseWindowById(WindowClass::JoinStation, 0);
    this->Window::Close();
  }

  static inline const std::initializer_list<WidgetID> can_build_widgets = {
    WID_RAT_BUILD_NS, WID_RAT_BUILD_X, WID_RAT_BUILD_EW, WID_RAT_BUILD_Y, WID_RAT_AUTORAIL,
    WID_RAT_BUILD_DEPOT, WID_RAT_BUILD_WAYPOINT, WID_RAT_BUILD_STATION, WID_RAT_BUILD_SIGNALS,
    WID_RAT_BUILD_BRIDGE, WID_RAT_BUILD_TUNNEL, WID_RAT_CONVERT_RAIL,
  };
 
  void OnInvalidateData([[maybe_unused]] int data = 0, [[maybe_unused]] bool gui_scope = true) override
  {
    if (!gui_scope) return;
 
    if (_cur_railtype != this->railtype) {
      this->ModifyRailType(_cur_railtype);
 
      /* Update cursor and all sub windows. */
      if (_thd.GetCallbackWnd() == this) SetCursor(this->GetCursorForWidget(this->last_user_action), PAL_NONE);
      for (WindowClass cls : {WindowClass::BuildStation, WindowClass::BuildSignal, WindowClass::BuildWaypoint, WindowClass::BuildDepot}) {
        SetWindowDirty(cls, TransportType::Rail);
      }
    }
 
    if (!ValParamRailType(this->railtype)) {
      /* Close toolbar if rail type is not available. */
      this->Close();
      return;
    }
 
    bool can_build = CanBuildVehicleInfrastructure(VehicleType::Train);
    for (const WidgetID widget : can_build_widgets) this->SetWidgetDisabledState(widget, !can_build);
    if (!can_build) {
      CloseWindowById(WindowClass::BuildSignal, TransportType::Rail);
      CloseWindowById(WindowClass::BuildStation, TransportType::Rail);
      CloseWindowById(WindowClass::BuildDepot, TransportType::Rail);
      CloseWindowById(WindowClass::BuildWaypoint, TransportType::Rail);
      CloseWindowById(WindowClass::JoinStation, 0);
    }
  }
 
  bool OnTooltip([[maybe_unused]] Point pt, WidgetID widget, TooltipCloseCondition close_cond) override
  {
    bool can_build = CanBuildVehicleInfrastructure(VehicleType::Train);
    if (can_build) return false;
 
    if (std::ranges::find(can_build_widgets, widget) == std::end(can_build_widgets)) return false;
 
    GuiShowTooltips(this, GetEncodedString(STR_TOOLBAR_DISABLED_NO_VEHICLE_AVAILABLE), close_cond);
    return true;
  }
 
  void OnInit() override
  {
    /* Configure the rail toolbar for the railtype. */
    const RailTypeInfo *rti = GetRailTypeInfo(this->railtype);
    this->GetWidget<NWidgetCore>(WID_RAT_BUILD_NS)->SetSprite(rti->gui_sprites.build_ns_rail);
    this->GetWidget<NWidgetCore>(WID_RAT_BUILD_X)->SetSprite(rti->gui_sprites.build_x_rail);
    this->GetWidget<NWidgetCore>(WID_RAT_BUILD_EW)->SetSprite(rti->gui_sprites.build_ew_rail);
    this->GetWidget<NWidgetCore>(WID_RAT_BUILD_Y)->SetSprite(rti->gui_sprites.build_y_rail);
    this->GetWidget<NWidgetCore>(WID_RAT_AUTORAIL)->SetSprite(rti->gui_sprites.auto_rail);
    this->GetWidget<NWidgetCore>(WID_RAT_BUILD_DEPOT)->SetSprite(rti->gui_sprites.build_depot);
    this->GetWidget<NWidgetCore>(WID_RAT_CONVERT_RAIL)->SetSprite(rti->gui_sprites.convert_rail);
    this->GetWidget<NWidgetCore>(WID_RAT_BUILD_TUNNEL)->SetSprite(rti->gui_sprites.build_tunnel);
  }

  void ModifyRailType(RailType railtype)
  {
    this->railtype = railtype;
    this->ReInit();
  }
 
  void UpdateRemoveWidgetStatus(WidgetID clicked_widget)
  {
    switch (clicked_widget) {
      case WID_RAT_REMOVE:
        /* If it is the removal button that has been clicked, do nothing,
         * as it is up to the other buttons to drive removal status */
        return;
 
      case WID_RAT_BUILD_NS:
      case WID_RAT_BUILD_X:
      case WID_RAT_BUILD_EW:
      case WID_RAT_BUILD_Y:
      case WID_RAT_AUTORAIL:
      case WID_RAT_BUILD_WAYPOINT:
      case WID_RAT_BUILD_STATION:
      case WID_RAT_BUILD_SIGNALS:
        /* Removal button is enabled only if the rail/signal/waypoint/station
         * button is still lowered.  Once raised, it has to be disabled */
        this->SetWidgetDisabledState(WID_RAT_REMOVE, !this->IsWidgetLowered(clicked_widget));
        break;
 
      default:
        /* When any other buttons than rail/signal/waypoint/station, raise and
         * disable the removal button */
        this->DisableWidget(WID_RAT_REMOVE);
        this->RaiseWidget(WID_RAT_REMOVE);
        break;
    }
  }
 
  std::string GetWidgetString(WidgetID widget, StringID stringid) const override
  {
    if (widget == WID_RAT_CAPTION) {
      const RailTypeInfo *rti = GetRailTypeInfo(this->railtype);
      if (rti->max_speed > 0) {
        return GetString(STR_TOOLBAR_RAILTYPE_VELOCITY, rti->strings.toolbar_caption, PackVelocity(rti->max_speed, VehicleType::Train));
      }
      return GetString(rti->strings.toolbar_caption);
    }
 
    return this->Window::GetWidgetString(widget, stringid);
  }

  CursorID GetCursorForWidget(WidgetID widget)
  {
    switch (widget) {
      case WID_RAT_BUILD_NS: return GetRailTypeInfo(_cur_railtype)->cursor.rail_ns;
      case WID_RAT_BUILD_X: return GetRailTypeInfo(_cur_railtype)->cursor.rail_swne;
      case WID_RAT_BUILD_EW: return GetRailTypeInfo(_cur_railtype)->cursor.rail_ew;
      case WID_RAT_BUILD_Y: return GetRailTypeInfo(_cur_railtype)->cursor.rail_nwse;
      case WID_RAT_AUTORAIL: return GetRailTypeInfo(_cur_railtype)->cursor.autorail;
      case WID_RAT_DEMOLISH: return ANIMCURSOR_DEMOLISH;
      case WID_RAT_BUILD_DEPOT: return GetRailTypeInfo(_cur_railtype)->cursor.depot;
      case WID_RAT_BUILD_WAYPOINT: return SPR_CURSOR_WAYPOINT;
      case WID_RAT_BUILD_STATION: return SPR_CURSOR_RAIL_STATION;
      case WID_RAT_BUILD_SIGNALS: return ANIMCURSOR_BUILDSIGNALS;
      case WID_RAT_BUILD_BRIDGE: return SPR_CURSOR_BRIDGE;
      case WID_RAT_BUILD_TUNNEL: return GetRailTypeInfo(_cur_railtype)->cursor.tunnel;
      case WID_RAT_CONVERT_RAIL: return GetRailTypeInfo(_cur_railtype)->cursor.convert;
      default: NOT_REACHED();
    }
  }

  HighLightStyle GetHighLightStyleForWidget(WidgetID widget)
  {
    switch (widget) {
      case WID_RAT_BUILD_NS: return HT_LINE | HT_DIR_VL;
      case WID_RAT_BUILD_X: return HT_LINE | HT_DIR_X;
      case WID_RAT_BUILD_EW: return HT_LINE | HT_DIR_HL;
      case WID_RAT_BUILD_Y: return HT_LINE | HT_DIR_Y;
      case WID_RAT_AUTORAIL: return HT_RAIL;
      case WID_RAT_DEMOLISH: return HT_RECT | HT_DIAGONAL;
      case WID_RAT_BUILD_DEPOT: return HT_RECT;
      case WID_RAT_BUILD_WAYPOINT: return HT_RECT;
      case WID_RAT_BUILD_STATION: return HT_RECT;
      case WID_RAT_BUILD_SIGNALS: return HT_RECT;
      case WID_RAT_BUILD_BRIDGE: return HT_RECT;
      case WID_RAT_BUILD_TUNNEL: return HT_SPECIAL;
      case WID_RAT_CONVERT_RAIL: return HT_RECT | HT_DIAGONAL;
      default: NOT_REACHED();
    }
  }
 
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    if (widget < WID_RAT_BUILD_NS) return;
 
    _remove_button_clicked = false;
 
    if (widget == WID_RAT_REMOVE) {
      BuildRailClick_Remove(this);
      if (_ctrl_pressed) RailToolbar_CtrlChanged(this);
      return;
    }
 
    this->last_user_action = widget;
    bool started = HandlePlacePushButton(this, widget, this->GetCursorForWidget(widget), this->GetHighLightStyleForWidget(widget));
 
    switch (widget) {
      case WID_RAT_BUILD_DEPOT:
        if (started) {
          ShowBuildTrainDepotPicker(this);
        }
        break;
 
      case WID_RAT_BUILD_WAYPOINT:
        if (started) {
          ShowBuildWaypointPicker(this);
        }
        break;
 
      case WID_RAT_BUILD_STATION:
        if (started) {
          ShowStationBuilder(this);
        }
        break;
 
      case WID_RAT_BUILD_SIGNALS: {
        if (started != _ctrl_pressed) {
          ShowSignalBuilder(this);
        }
        break;
      }
    }
 
    this->UpdateRemoveWidgetStatus(widget);
    if (_ctrl_pressed) RailToolbar_CtrlChanged(this);
  }
 
  EventState OnHotkey(int hotkey) override
  {
    if (IsSpecialHotkey(hotkey)) return this->ChangeRailTypeOnHotkey(hotkey);
    MarkTileDirtyByTile(TileVirtXY(_thd.pos.x, _thd.pos.y)); // redraw tile selection
    return Window::OnHotkey(hotkey);
  }
 
  void OnPlaceObject([[maybe_unused]] Point pt, TileIndex tile) override
  {
    switch (this->last_user_action) {
      case WID_RAT_BUILD_NS:
        VpStartPlaceSizing(tile, VPM_FIX_VERTICAL | VPM_RAILDIRS, DDSP_PLACE_RAIL);
        break;
 
      case WID_RAT_BUILD_X:
        VpStartPlaceSizing(tile, VPM_FIX_Y | VPM_RAILDIRS, DDSP_PLACE_RAIL);
        break;
 
      case WID_RAT_BUILD_EW:
        VpStartPlaceSizing(tile, VPM_FIX_HORIZONTAL | VPM_RAILDIRS, DDSP_PLACE_RAIL);
        break;
 
      case WID_RAT_BUILD_Y:
        VpStartPlaceSizing(tile, VPM_FIX_X | VPM_RAILDIRS, DDSP_PLACE_RAIL);
        break;
 
      case WID_RAT_AUTORAIL:
        VpStartPlaceSizing(tile, VPM_RAILDIRS, DDSP_PLACE_RAIL);
        break;
 
      case WID_RAT_DEMOLISH:
        PlaceProc_DemolishArea(tile);
        break;
 
      case WID_RAT_BUILD_DEPOT:
        Command<Commands::BuildRailDepot>::Post(STR_ERROR_CAN_T_BUILD_TRAIN_DEPOT, CcRailDepot, tile, _cur_railtype, _build_depot_direction);
        break;
 
      case WID_RAT_BUILD_WAYPOINT:
        PlaceRail_Waypoint(tile);
        break;
 
      case WID_RAT_BUILD_STATION:
        PlaceRail_Station(tile);
        break;
 
      case WID_RAT_BUILD_SIGNALS:
        VpStartPlaceSizing(tile, VPM_SIGNALDIRS, DDSP_BUILD_SIGNALS);
        break;
 
      case WID_RAT_BUILD_BRIDGE:
        PlaceRail_Bridge(tile, this);
        break;
 
      case WID_RAT_BUILD_TUNNEL:
        Command<Commands::BuildTunnel>::Post(STR_ERROR_CAN_T_BUILD_TUNNEL_HERE, CcBuildRailTunnel, tile, TransportType::Rail, _cur_railtype, INVALID_ROADTYPE);
        break;
 
      case WID_RAT_CONVERT_RAIL:
        VpStartPlaceSizing(tile, VPM_X_AND_Y, DDSP_CONVERT_RAIL);
        break;
 
      default: NOT_REACHED();
    }
  }
 
  void OnPlaceDrag(ViewportPlaceMethod select_method, [[maybe_unused]] ViewportDragDropSelectionProcess select_proc, [[maybe_unused]] Point pt) override
  {
    /* no dragging if you have pressed the convert button */
    if (FindWindowById(WindowClass::BuildSignal, 0) != nullptr && _convert_signal_button && this->IsWidgetLowered(WID_RAT_BUILD_SIGNALS)) return;
 
    VpSelectTilesWithMethod(pt.x, pt.y, select_method);
  }
 
  Point OnInitialPosition(int16_t sm_width, [[maybe_unused]] int16_t sm_height, [[maybe_unused]] int window_number) override
  {
    return AlignInitialConstructionToolbar(sm_width);
  }
 
  void OnPlaceMouseUp([[maybe_unused]] ViewportPlaceMethod select_method, ViewportDragDropSelectionProcess select_proc, [[maybe_unused]] Point pt, TileIndex start_tile, TileIndex end_tile) override
  {
    if (pt.x != -1) {
      switch (select_proc) {
        default: NOT_REACHED();
        case DDSP_BUILD_BRIDGE:
          if (!_settings_client.gui.persistent_buildingtools) ResetObjectToPlace();
          ShowBuildBridgeWindow(start_tile, end_tile, TransportType::Rail, _cur_railtype, INVALID_ROADTYPE);
          break;
 
        case DDSP_PLACE_RAIL:
          HandleAutodirPlacement();
          break;
 
        case DDSP_BUILD_SIGNALS:
          HandleAutoSignalPlacement();
          break;
 
        case DDSP_DEMOLISH_AREA:
          GUIPlaceProcDragXY(select_proc, start_tile, end_tile);
          break;
 
        case DDSP_CONVERT_RAIL:
          Command<Commands::ConvertRail>::Post(STR_ERROR_CAN_T_CONVERT_RAIL, CcPlaySound_CONSTRUCTION_RAIL, end_tile, start_tile, _cur_railtype, _ctrl_pressed);
          break;
 
        case DDSP_REMOVE_STATION:
        case DDSP_BUILD_STATION:
          if (this->IsWidgetLowered(WID_RAT_BUILD_STATION)) {
            /* Station */
            if (_remove_button_clicked) {
              bool keep_rail = !_ctrl_pressed;
              Command<Commands::RemoveFromRailStation>::Post(STR_ERROR_CAN_T_REMOVE_PART_OF_STATION, CcPlaySound_CONSTRUCTION_RAIL, end_tile, start_tile, keep_rail);
            } else {
              HandleStationPlacement(start_tile, end_tile);
            }
          } else {
            /* Waypoint */
            if (_remove_button_clicked) {
              bool keep_rail = !_ctrl_pressed;
              Command<Commands::RemoveFromRailWaypoint>::Post(STR_ERROR_CAN_T_REMOVE_RAIL_WAYPOINT , CcPlaySound_CONSTRUCTION_RAIL, end_tile, start_tile, keep_rail);
            } else {
              TileArea ta(start_tile, end_tile);
              Axis axis = select_method == VPM_X_LIMITED ? Axis::X : Axis::Y;
              bool adjacent = _ctrl_pressed;
 
              auto proc = [=](bool test, StationID to_join) -> bool {
                if (test) {
                  return Command<Commands::BuildRailWaypoint>::Do(CommandFlagsToDCFlags(GetCommandFlags<Commands::BuildRailWaypoint>()), ta.tile, axis, ta.w, ta.h, _waypoint_gui.sel_class, _waypoint_gui.sel_type, StationID::Invalid(), adjacent).Succeeded();
                } else {
                  return Command<Commands::BuildRailWaypoint>::Post(STR_ERROR_CAN_T_BUILD_RAIL_WAYPOINT , CcPlaySound_CONSTRUCTION_RAIL, ta.tile, axis, ta.w, ta.h, _waypoint_gui.sel_class, _waypoint_gui.sel_type, to_join, adjacent);
                }
              };
 
              ShowSelectRailWaypointIfNeeded(ta, proc);
            }
          }
          break;
      }
    }
  }
 
  void OnPlaceObjectAbort() override
  {
    if (this->IsWidgetLowered(WID_RAT_BUILD_STATION)) SetViewportCatchmentStation(nullptr, true);
    if (this->IsWidgetLowered(WID_RAT_BUILD_WAYPOINT)) SetViewportCatchmentWaypoint(nullptr, true);
 
    this->RaiseButtons();
    this->DisableWidget(WID_RAT_REMOVE);
    this->SetWidgetDirty(WID_RAT_REMOVE);
 
    CloseWindowById(WindowClass::BuildSignal, TransportType::Rail);
    CloseWindowById(WindowClass::BuildStation, TransportType::Rail);
    CloseWindowById(WindowClass::BuildDepot, TransportType::Rail);
    CloseWindowById(WindowClass::BuildWaypoint, TransportType::Rail);
    CloseWindowById(WindowClass::JoinStation, 0);
    CloseWindowByClass(WindowClass::BuildBridge);
  }
 
  void OnPlacePresize([[maybe_unused]] Point pt, TileIndex tile) override
  {
    Command<Commands::BuildTunnel>::Do(DoCommandFlag::Auto, tile, TransportType::Rail, _cur_railtype, INVALID_ROADTYPE);
    VpSetPresizeRange(tile, _build_tunnel_endtile == INVALID_TILE ? tile : _build_tunnel_endtile);
  }
 
  EventState OnCTRLStateChange() override
  {
    /* do not toggle Remove button by Ctrl when placing station */
    if (!this->IsWidgetLowered(WID_RAT_BUILD_STATION) && !this->IsWidgetLowered(WID_RAT_BUILD_WAYPOINT) && RailToolbar_CtrlChanged(this)) return EventState::Handled;
    return EventState::NotHandled;
  }
 
  void OnRealtimeTick([[maybe_unused]] uint delta_ms) override
  {
    if (this->IsWidgetLowered(WID_RAT_BUILD_WAYPOINT)) CheckRedrawRailWaypointCoverage(this);
  }

  EventState ChangeRailTypeOnHotkey(int hotkey)
  {
    auto [index, step] = GetListIndexStep(SpecialListHotkeys(hotkey), _sorted_railtypes, this->railtype);
 
    while (!HasRailTypeAvail(_local_company, _sorted_railtypes[index])) {
      index = (index + step) % _sorted_railtypes.size();
    }
 
    _last_built_railtype = _cur_railtype = _sorted_railtypes[index];
    this->ModifyRailType(_last_built_railtype);
 
    /* Update cursor and all sub windows. */
    if (_thd.GetCallbackWnd() == this) SetCursor(this->GetCursorForWidget(this->last_user_action), PAL_NONE);
    for (WindowClass cls : {WindowClass::BuildStation, WindowClass::BuildSignal, WindowClass::BuildWaypoint, WindowClass::BuildDepot}) {
      SetWindowDirty(cls, TransportType::Rail);
    }
 
    return EventState::Handled;
  }

  static EventState RailToolbarGlobalHotkeys(int hotkey)
  {
    if (_game_mode != GameMode::Normal) return EventState::NotHandled;
    Window *w = ShowBuildRailToolbar(_last_built_railtype);
    if (w == nullptr) return EventState::NotHandled;
    return w->OnHotkey(hotkey);
  }
 
  static inline HotkeyList hotkeys{"railtoolbar", {
    Hotkey('1', "build_ns", WID_RAT_BUILD_NS),
    Hotkey('2', "build_x", WID_RAT_BUILD_X),
    Hotkey('3', "build_ew", WID_RAT_BUILD_EW),
    Hotkey('4', "build_y", WID_RAT_BUILD_Y),
    Hotkey({'5', 'A' | WKC_GLOBAL_HOTKEY}, "autorail", WID_RAT_AUTORAIL),
    Hotkey('6', "demolish", WID_RAT_DEMOLISH),
    Hotkey('7', "depot", WID_RAT_BUILD_DEPOT),
    Hotkey('8', "waypoint", WID_RAT_BUILD_WAYPOINT),
    Hotkey('9', "station", WID_RAT_BUILD_STATION),
    Hotkey('S', "signal", WID_RAT_BUILD_SIGNALS),
    Hotkey('B', "bridge", WID_RAT_BUILD_BRIDGE),
    Hotkey('T', "tunnel", WID_RAT_BUILD_TUNNEL),
    Hotkey('R', "remove", WID_RAT_REMOVE),
    Hotkey('C', "convert", WID_RAT_CONVERT_RAIL),
    Hotkey(WKC_L_BRACKET, "prev_railtype", to_underlying(SpecialListHotkeys::PreviousItem)),
    Hotkey(WKC_R_BRACKET, "next_railtype", to_underlying(SpecialListHotkeys::NextItem)),
    Hotkey(WKC_L_BRACKET | WKC_CTRL, "first_railtype", to_underlying(SpecialListHotkeys::FirstItem)),
    Hotkey(WKC_R_BRACKET | WKC_CTRL, "last_railtype", to_underlying(SpecialListHotkeys::LastItem)),
  }, RailToolbarGlobalHotkeys};
};
 
static constexpr std::initializer_list<NWidgetPart> _nested_build_rail_widgets = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, Colours::DarkGreen),
    NWidget(WWT_CAPTION, Colours::DarkGreen, WID_RAT_CAPTION), SetTextStyle(TextColour::White),
    NWidget(WWT_STICKYBOX, Colours::DarkGreen),
  EndContainer(),
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_BUILD_NS),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_RAIL_NS, STR_RAIL_TOOLBAR_TOOLTIP_BUILD_RAILROAD_TRACK),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_BUILD_X),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_RAIL_NE, STR_RAIL_TOOLBAR_TOOLTIP_BUILD_RAILROAD_TRACK),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_BUILD_EW),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_RAIL_EW, STR_RAIL_TOOLBAR_TOOLTIP_BUILD_RAILROAD_TRACK),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_BUILD_Y),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_RAIL_NW, STR_RAIL_TOOLBAR_TOOLTIP_BUILD_RAILROAD_TRACK),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_AUTORAIL),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_AUTORAIL, STR_RAIL_TOOLBAR_TOOLTIP_BUILD_AUTORAIL),
 
    NWidget(WWT_PANEL, Colours::DarkGreen), SetToolbarSpacerMinimalSize(), EndContainer(),
 
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_DEMOLISH),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_DYNAMITE, STR_TOOLTIP_DEMOLISH_BUILDINGS_ETC),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_BUILD_DEPOT),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_DEPOT_RAIL, STR_RAIL_TOOLBAR_TOOLTIP_BUILD_TRAIN_DEPOT_FOR_BUILDING),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_BUILD_WAYPOINT),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_WAYPOINT, STR_RAIL_TOOLBAR_TOOLTIP_CONVERT_RAIL_TO_WAYPOINT),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_BUILD_STATION),
            SetFill(0, 1), SetToolbarMinimalSize(2), SetSpriteTip(SPR_IMG_RAIL_STATION, STR_RAIL_TOOLBAR_TOOLTIP_BUILD_RAILROAD_STATION),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_BUILD_SIGNALS),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_RAIL_SIGNALS, STR_RAIL_TOOLBAR_TOOLTIP_BUILD_RAILROAD_SIGNALS),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_BUILD_BRIDGE),
            SetFill(0, 1), SetToolbarMinimalSize(2), SetSpriteTip(SPR_IMG_BRIDGE, STR_RAIL_TOOLBAR_TOOLTIP_BUILD_RAILROAD_BRIDGE),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_BUILD_TUNNEL),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_TUNNEL_RAIL, STR_RAIL_TOOLBAR_TOOLTIP_BUILD_RAILROAD_TUNNEL),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_REMOVE),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_REMOVE, STR_RAIL_TOOLBAR_TOOLTIP_TOGGLE_BUILD_REMOVE_FOR),
    NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_RAT_CONVERT_RAIL),
            SetFill(0, 1), SetToolbarMinimalSize(1), SetSpriteTip(SPR_IMG_CONVERT_RAIL, STR_RAIL_TOOLBAR_TOOLTIP_CONVERT_RAIL),
  EndContainer(),
};

static WindowDesc _build_rail_desc(
  WindowPosition::Manual, "toolbar_rail", 0, 0,
  WindowClass::BuildToolbar, WindowClass::None,
  WindowDefaultFlag::Construction,
  _nested_build_rail_widgets,
  &BuildRailToolbarWindow::hotkeys
);
 

Window *ShowBuildRailToolbar(RailType railtype)
{
  if (!Company::IsValidID(_local_company)) return nullptr;
  if (!ValParamRailType(railtype)) return nullptr;
 
  _cur_railtype = railtype;
  Window *w = BringWindowToFrontById(WindowClass::BuildToolbar, TransportType::Rail);
  if (w != nullptr) {
    w->OnInvalidateData();
    return w;
  }
 
  CloseWindowByClass(WindowClass::BuildToolbar);
  _remove_button_clicked = false;
  return new BuildRailToolbarWindow(_build_rail_desc, railtype);
}
 
/* TODO: For custom stations, respect their allowed platforms/lengths bitmasks!
 * --pasky */
 
static void HandleStationPlacement(TileIndex start, TileIndex end)
{
  TileArea ta(start, end);
  uint numtracks = ta.w;
  uint platlength = ta.h;
 
  if (_station_gui.axis == Axis::X) std::swap(numtracks, platlength);
 
  StationPickerSelection params = _station_gui;
  RailType rt = _cur_railtype;
  bool adjacent = _ctrl_pressed;
 
  auto proc = [=](bool test, StationID to_join) -> bool {
    if (test) {
      return Command<Commands::BuildRailStation>::Do(CommandFlagsToDCFlags(GetCommandFlags<Commands::BuildRailStation>()), ta.tile, rt, params.axis, numtracks, platlength, params.sel_class, params.sel_type, StationID::Invalid(), adjacent).Succeeded();
    } else {
      return Command<Commands::BuildRailStation>::Post(STR_ERROR_CAN_T_BUILD_RAILROAD_STATION, CcStation, ta.tile, rt, params.axis, numtracks, platlength, params.sel_class, params.sel_type, to_join, adjacent);
    }
  };
 
  ShowSelectStationIfNeeded(ta, proc);
}

static bool StationUsesDefaultType(const BaseStation *bst)
{
  for (TileIndex t : bst->train_station) {
    if (bst->TileBelongsToRailStation(t) && HasStationRail(t) && GetCustomStationSpecIndex(t) == 0) return true;
  }
  return false;
}
 
class StationPickerCallbacks : public PickerCallbacksNewGRFClass<StationClass> {
public:
  StationPickerCallbacks() : PickerCallbacksNewGRFClass<StationClass>("fav_stations") {}
 
  GrfSpecFeature GetFeature() const override { return GrfSpecFeature::Stations; }
 
  StringID GetClassTooltip() const override { return STR_PICKER_STATION_CLASS_TOOLTIP; }
  StringID GetTypeTooltip() const override { return STR_PICKER_STATION_TYPE_TOOLTIP; }
  StringID GetCollectionTooltip() const override { return STR_PICKER_STATION_COLLECTION_TOOLTIP; }
 
  bool IsActive() const override
  {
    for (const auto &cls : StationClass::Classes()) {
      if (IsWaypointClass(cls)) continue;
      for (const auto *spec : cls.Specs()) {
        if (spec != nullptr) return true;
      }
    }
    return false;
  }
 
  bool HasClassChoice() const override
  {
    return std::ranges::count_if(StationClass::Classes(), [](const auto &cls) { return !IsWaypointClass(cls); }) > 1;
  }
 
  int GetSelectedClass() const override { return _station_gui.sel_class.base(); }
  void SetSelectedClass(int id) const override { _station_gui.sel_class = this->GetClassIndex(id); }
 
  StringID GetClassName(int id) const override
  {
    const auto *sc = GetClass(id);
    if (IsWaypointClass(*sc)) return INVALID_STRING_ID;
    return sc->name;
  }
 
  int GetSelectedType() const override { return _station_gui.sel_type; }
  void SetSelectedType(int id) const override { _station_gui.sel_type = id; }
 
  StringID GetTypeName(int cls_id, int id) const override
  {
    const auto *spec = this->GetSpec(cls_id, id);
    return (spec == nullptr) ? STR_STATION_CLASS_DFLT_STATION : spec->name;
  }
 
  std::span<const BadgeID> GetTypeBadges(int cls_id, int id) const override
  {
    const auto *spec = this->GetSpec(cls_id, id);
    if (spec == nullptr) return {};
    return spec->badges;
  }
 
  bool IsTypeAvailable(int cls_id, int id) const override
  {
    return IsStationAvailable(this->GetSpec(cls_id, id));
  }
 
  void DrawType(int x, int y, int cls_id, int id) const override
  {
    if (!DrawStationTile(x, y, _cur_railtype, _station_gui.axis, this->GetClassIndex(cls_id), id)) {
      StationPickerDrawSprite(x, y, StationType::Rail, _cur_railtype, INVALID_ROADTYPE, 2 + to_underlying(_station_gui.axis));
    }
  }
 
  void FillUsedItems(std::set<PickerItem> &items) override
  {
    bool default_added = false;
    for (const Station *st : Station::Iterate()) {
      if (st->owner != _local_company) continue;
      if (!default_added && StationUsesDefaultType(st)) {
        items.insert({GrfID{}, 0, STAT_CLASS_DFLT.base(), 0});
        default_added = true;
      }
      for (const auto &sm : st->speclist) {
        if (sm.spec == nullptr) continue;
        items.insert({sm.grfid, sm.localidx, sm.spec->class_index.base(), sm.spec->index});
      }
    }
  }
 
  static StationPickerCallbacks instance;
};
/* static */ StationPickerCallbacks StationPickerCallbacks::instance;
 
struct BuildRailStationWindow : public PickerWindow {
private:
  uint coverage_height = 0; 

  void CheckSelectedSize(const StationSpec *statspec)
  {
    if (statspec == nullptr || _settings_client.gui.station_dragdrop) return;
 
    /* If current number of tracks is not allowed, make it as big as possible */
    if (HasBit(statspec->disallowed_platforms, _settings_client.gui.station_numtracks - 1)) {
      this->RaiseWidget(_settings_client.gui.station_numtracks + WID_BRAS_PLATFORM_NUM_BEGIN);
      _settings_client.gui.station_numtracks = 1;
      if (statspec->disallowed_platforms != UINT8_MAX) {
        while (HasBit(statspec->disallowed_platforms, _settings_client.gui.station_numtracks - 1)) {
          _settings_client.gui.station_numtracks++;
        }
        this->LowerWidget(_settings_client.gui.station_numtracks + WID_BRAS_PLATFORM_NUM_BEGIN);
      }
    }
 
    if (HasBit(statspec->disallowed_lengths, _settings_client.gui.station_platlength - 1)) {
      this->RaiseWidget(_settings_client.gui.station_platlength + WID_BRAS_PLATFORM_LEN_BEGIN);
      _settings_client.gui.station_platlength = 1;
      if (statspec->disallowed_lengths != UINT8_MAX) {
        while (HasBit(statspec->disallowed_lengths, _settings_client.gui.station_platlength - 1)) {
          _settings_client.gui.station_platlength++;
        }
        this->LowerWidget(_settings_client.gui.station_platlength + WID_BRAS_PLATFORM_LEN_BEGIN);
      }
    }
  }

  void SetSelectedSize()
  {
    if (_settings_client.gui.station_dragdrop) {
      SetTileSelectSize(1, 1);
    } else {
      int x = _settings_client.gui.station_numtracks;
      int y = _settings_client.gui.station_platlength;
      if (_station_gui.axis == Axis::X) std::swap(x, y);
      if (!_remove_button_clicked) {
        SetTileSelectSize(x, y);
      }
    }
 
    int rad = (_settings_game.station.modified_catchment) ? CA_TRAIN : CA_UNMODIFIED;
    if (_settings_client.gui.station_show_coverage) SetTileSelectBigSize(-rad, -rad, 2 * rad, 2 * rad);
  }
 
public:
  BuildRailStationWindow(WindowDesc &desc, Window *parent) : PickerWindow(desc, parent, TransportType::Rail, StationPickerCallbacks::instance)
  {
    this->coverage_height = 2 * GetCharacterHeight(FontSize::Normal) + WidgetDimensions::scaled.vsep_normal;
    this->ConstructWindow();
  }
 
  void OnInit() override
  {
    this->LowerWidget(WID_BRAS_PLATFORM_DIR_X + to_underlying(_station_gui.axis));
    if (_settings_client.gui.station_dragdrop) {
      this->LowerWidget(WID_BRAS_PLATFORM_DRAG_N_DROP);
    } else {
      this->LowerWidget(WID_BRAS_PLATFORM_NUM_BEGIN + _settings_client.gui.station_numtracks);
      this->LowerWidget(WID_BRAS_PLATFORM_LEN_BEGIN + _settings_client.gui.station_platlength);
    }
    this->SetWidgetLoweredState(WID_BRAS_HIGHLIGHT_OFF, !_settings_client.gui.station_show_coverage);
    this->SetWidgetLoweredState(WID_BRAS_HIGHLIGHT_ON, _settings_client.gui.station_show_coverage);
    this->SetSelectedSize();
 
    this->PickerWindow::OnInit();
  }
 
  void Close([[maybe_unused]] int data = 0) override
  {
    CloseWindowById(WindowClass::JoinStation, 0);
    this->PickerWindow::Close();
  }
 
  void OnInvalidateData([[maybe_unused]] int data = 0, [[maybe_unused]] bool gui_scope = true) override
  {
    if (gui_scope) {
      const StationSpec *statspec = StationClass::Get(_station_gui.sel_class)->GetSpec(_station_gui.sel_type);
      this->CheckSelectedSize(statspec);
    }
 
    this->PickerWindow::OnInvalidateData(data, gui_scope);
  }
 
  void OnPaint() override
  {
    const StationSpec *statspec = StationClass::Get(_station_gui.sel_class)->GetSpec(_station_gui.sel_type);
    int rad = (_settings_game.station.modified_catchment) ? CA_TRAIN : CA_UNMODIFIED;
 
    for (uint bits = 0; bits < 7; bits++) {
      bool disable = bits >= _settings_game.station.station_spread;
      if (statspec == nullptr) {
        this->SetWidgetDisabledState(bits + WID_BRAS_PLATFORM_NUM_1, disable);
        this->SetWidgetDisabledState(bits + WID_BRAS_PLATFORM_LEN_1, disable);
      } else {
        this->SetWidgetDisabledState(bits + WID_BRAS_PLATFORM_NUM_1, HasBit(statspec->disallowed_platforms, bits) || disable);
        this->SetWidgetDisabledState(bits + WID_BRAS_PLATFORM_LEN_1, HasBit(statspec->disallowed_lengths,   bits) || disable);
      }
    }
 
    this->DrawWidgets();
 
    if (this->IsShaded()) return;
    /* 'Accepts' and 'Supplies' texts. */
    Rect r = this->GetWidget<NWidgetBase>(WID_BRAS_COVERAGE_TEXTS)->GetCurrentRect();
    const int bottom = r.bottom;
    r.bottom = INT_MAX; // Allow overflow as we want to know the required height.
    if (statspec != nullptr) r.top = DrawBadgeNameList(r, statspec->badges, GrfSpecFeature::Stations);
    r.top = DrawStationCoverageAreaText(r, SCT_ALL, rad, false) + WidgetDimensions::scaled.vsep_normal;
    r.top = DrawStationCoverageAreaText(r, SCT_ALL, rad, true);
    /* Resize background if the window is too small.
     * Never make the window smaller to avoid oscillating if the size change affects the acceptance.
     * (This is the case, if making the window bigger moves the mouse into the window.) */
    if (r.top > bottom) {
      this->coverage_height += r.top - bottom;
      this->ReInit();
    }
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    switch (widget) {
      case WID_BRAS_PLATFORM_DIR_X:
      case WID_BRAS_PLATFORM_DIR_Y:
        size.width  = ScaleGUITrad(PREVIEW_WIDTH) + WidgetDimensions::scaled.fullbevel.Horizontal();
        size.height = ScaleGUITrad(PREVIEW_HEIGHT) + WidgetDimensions::scaled.fullbevel.Vertical();
        break;
 
      case WID_BRAS_COVERAGE_TEXTS:
        size.height = this->coverage_height;
        break;
 
      default:
        this->PickerWindow::UpdateWidgetSize(widget, size, padding, fill, resize);
        break;
    }
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    DrawPixelInfo tmp_dpi;
 
    switch (widget) {
      case WID_BRAS_PLATFORM_DIR_X: {
        /* Set up a clipping area for the '/' station preview */
        Rect ir = r.Shrink(WidgetDimensions::scaled.bevel);
        if (FillDrawPixelInfo(&tmp_dpi, ir)) {
          AutoRestoreBackup dpi_backup(_cur_dpi, &tmp_dpi);
          int x = (ir.Width()  - ScaleSpriteTrad(PREVIEW_WIDTH)) / 2 + ScaleSpriteTrad(PREVIEW_LEFT);
          int y = (ir.Height() + ScaleSpriteTrad(PREVIEW_HEIGHT)) / 2 - ScaleSpriteTrad(PREVIEW_BOTTOM);
          if (!DrawStationTile(x, y, _cur_railtype, Axis::X, _station_gui.sel_class, _station_gui.sel_type)) {
            StationPickerDrawSprite(x, y, StationType::Rail, _cur_railtype, INVALID_ROADTYPE, 2);
          }
        }
        break;
      }
 
      case WID_BRAS_PLATFORM_DIR_Y: {
        /* Set up a clipping area for the '\' station preview */
        Rect ir = r.Shrink(WidgetDimensions::scaled.bevel);
        if (FillDrawPixelInfo(&tmp_dpi, ir)) {
          AutoRestoreBackup dpi_backup(_cur_dpi, &tmp_dpi);
          int x = (ir.Width()  - ScaleSpriteTrad(PREVIEW_WIDTH)) / 2 + ScaleSpriteTrad(PREVIEW_LEFT);
          int y = (ir.Height() + ScaleSpriteTrad(PREVIEW_HEIGHT)) / 2 - ScaleSpriteTrad(PREVIEW_BOTTOM);
          if (!DrawStationTile(x, y, _cur_railtype, Axis::Y, _station_gui.sel_class, _station_gui.sel_type)) {
            StationPickerDrawSprite(x, y, StationType::Rail, _cur_railtype, INVALID_ROADTYPE, 3);
          }
        }
        break;
      }
 
      default:
        this->PickerWindow::DrawWidget(r, widget);
        break;
    }
  }
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    switch (widget) {
      case WID_BRAS_PLATFORM_DIR_X:
      case WID_BRAS_PLATFORM_DIR_Y:
        this->RaiseWidget(WID_BRAS_PLATFORM_DIR_X + to_underlying(_station_gui.axis));
        _station_gui.axis = static_cast<Axis>(widget - WID_BRAS_PLATFORM_DIR_X);
        this->LowerWidget(WID_BRAS_PLATFORM_DIR_X + to_underlying(_station_gui.axis));
 
        this->SetSelectedSize();
        SndClickBeep();
        this->SetDirty();
        CloseWindowById(WindowClass::JoinStation, 0);
        break;
 
      case WID_BRAS_PLATFORM_NUM_1:
      case WID_BRAS_PLATFORM_NUM_2:
      case WID_BRAS_PLATFORM_NUM_3:
      case WID_BRAS_PLATFORM_NUM_4:
      case WID_BRAS_PLATFORM_NUM_5:
      case WID_BRAS_PLATFORM_NUM_6:
      case WID_BRAS_PLATFORM_NUM_7: {
        this->RaiseWidget(WID_BRAS_PLATFORM_NUM_BEGIN + _settings_client.gui.station_numtracks);
        this->RaiseWidget(WID_BRAS_PLATFORM_DRAG_N_DROP);
 
        _settings_client.gui.station_numtracks = widget - WID_BRAS_PLATFORM_NUM_BEGIN;
        _settings_client.gui.station_dragdrop = false;
 
        const StationSpec *statspec = StationClass::Get(_station_gui.sel_class)->GetSpec(_station_gui.sel_type);
        if (statspec != nullptr && HasBit(statspec->disallowed_lengths, _settings_client.gui.station_platlength - 1)) {
          /* The previously selected number of platforms in invalid */
          for (uint i = 0; i < 7; i++) {
            if (!HasBit(statspec->disallowed_lengths, i)) {
              this->RaiseWidget(_settings_client.gui.station_platlength + WID_BRAS_PLATFORM_LEN_BEGIN);
              _settings_client.gui.station_platlength = i + 1;
              break;
            }
          }
        }
 
        this->LowerWidget(_settings_client.gui.station_numtracks + WID_BRAS_PLATFORM_NUM_BEGIN);
        this->LowerWidget(_settings_client.gui.station_platlength + WID_BRAS_PLATFORM_LEN_BEGIN);
        this->SetSelectedSize();
        SndClickBeep();
        this->SetDirty();
        CloseWindowById(WindowClass::JoinStation, 0);
        break;
      }
 
      case WID_BRAS_PLATFORM_LEN_1:
      case WID_BRAS_PLATFORM_LEN_2:
      case WID_BRAS_PLATFORM_LEN_3:
      case WID_BRAS_PLATFORM_LEN_4:
      case WID_BRAS_PLATFORM_LEN_5:
      case WID_BRAS_PLATFORM_LEN_6:
      case WID_BRAS_PLATFORM_LEN_7: {
        this->RaiseWidget(_settings_client.gui.station_platlength + WID_BRAS_PLATFORM_LEN_BEGIN);
        this->RaiseWidget(WID_BRAS_PLATFORM_DRAG_N_DROP);
 
        _settings_client.gui.station_platlength = widget - WID_BRAS_PLATFORM_LEN_BEGIN;
        _settings_client.gui.station_dragdrop = false;
 
        const StationSpec *statspec = StationClass::Get(_station_gui.sel_class)->GetSpec(_station_gui.sel_type);
        if (statspec != nullptr && HasBit(statspec->disallowed_platforms, _settings_client.gui.station_numtracks - 1)) {
          /* The previously selected number of tracks in invalid */
          for (uint i = 0; i < 7; i++) {
            if (!HasBit(statspec->disallowed_platforms, i)) {
              this->RaiseWidget(_settings_client.gui.station_numtracks + WID_BRAS_PLATFORM_NUM_BEGIN);
              _settings_client.gui.station_numtracks = i + 1;
              break;
            }
          }
        }
 
        this->LowerWidget(_settings_client.gui.station_numtracks + WID_BRAS_PLATFORM_NUM_BEGIN);
        this->LowerWidget(_settings_client.gui.station_platlength + WID_BRAS_PLATFORM_LEN_BEGIN);
        this->SetSelectedSize();
        SndClickBeep();
        this->SetDirty();
        CloseWindowById(WindowClass::JoinStation, 0);
        break;
      }
 
      case WID_BRAS_PLATFORM_DRAG_N_DROP: {
        _settings_client.gui.station_dragdrop ^= true;
 
        this->ToggleWidgetLoweredState(WID_BRAS_PLATFORM_DRAG_N_DROP);
 
        /* get the first allowed length/number of platforms */
        const StationSpec *statspec = StationClass::Get(_station_gui.sel_class)->GetSpec(_station_gui.sel_type);
        if (statspec != nullptr && HasBit(statspec->disallowed_lengths, _settings_client.gui.station_platlength - 1)) {
          for (uint i = 0; i < 7; i++) {
            if (!HasBit(statspec->disallowed_lengths, i)) {
              this->RaiseWidget(_settings_client.gui.station_platlength + WID_BRAS_PLATFORM_LEN_BEGIN);
              _settings_client.gui.station_platlength = i + 1;
              break;
            }
          }
        }
        if (statspec != nullptr && HasBit(statspec->disallowed_platforms, _settings_client.gui.station_numtracks - 1)) {
          for (uint i = 0; i < 7; i++) {
            if (!HasBit(statspec->disallowed_platforms, i)) {
              this->RaiseWidget(_settings_client.gui.station_numtracks + WID_BRAS_PLATFORM_NUM_BEGIN);
              _settings_client.gui.station_numtracks = i + 1;
              break;
            }
          }
        }
 
        this->SetWidgetLoweredState(_settings_client.gui.station_numtracks + WID_BRAS_PLATFORM_NUM_BEGIN, !_settings_client.gui.station_dragdrop);
        this->SetWidgetLoweredState(_settings_client.gui.station_platlength + WID_BRAS_PLATFORM_LEN_BEGIN, !_settings_client.gui.station_dragdrop);
        this->SetSelectedSize();
        SndClickBeep();
        this->SetDirty();
        CloseWindowById(WindowClass::JoinStation, 0);
        break;
      }
 
      case WID_BRAS_HIGHLIGHT_OFF:
      case WID_BRAS_HIGHLIGHT_ON:
        _settings_client.gui.station_show_coverage = (widget != WID_BRAS_HIGHLIGHT_OFF);
 
        this->SetWidgetLoweredState(WID_BRAS_HIGHLIGHT_OFF, !_settings_client.gui.station_show_coverage);
        this->SetWidgetLoweredState(WID_BRAS_HIGHLIGHT_ON, _settings_client.gui.station_show_coverage);
        this->SetSelectedSize();
        SndClickBeep();
        this->SetDirty();
        SetViewportCatchmentStation(nullptr, true);
        break;
 
      default:
        this->PickerWindow::OnClick(pt, widget, click_count);
        break;
    }
  }
 
  void OnRealtimeTick([[maybe_unused]] uint delta_ms) override
  {
    CheckRedrawStationCoverage(this);
  }

  static EventState BuildRailStationGlobalHotkeys(int hotkey)
  {
    if (_game_mode == GameMode::Menu) return EventState::NotHandled;
    Window *w = ShowStationBuilder(FindWindowById(WindowClass::BuildToolbar, TransportType::Rail));
    if (w == nullptr) return EventState::NotHandled;
    return w->OnHotkey(hotkey);
  }
 
  static inline HotkeyList hotkeys{"buildrailstation", {
    Hotkey('F', "focus_filter_box", PCWHK_FOCUS_FILTER_BOX),
  }, BuildRailStationGlobalHotkeys};
};
 
static constexpr std::initializer_list<NWidgetPart> _nested_station_builder_widgets = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, Colours::DarkGreen),
    NWidget(WWT_CAPTION, Colours::DarkGreen), SetStringTip(STR_STATION_BUILD_RAIL_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
    NWidget(WWT_SHADEBOX, Colours::DarkGreen),
    NWidget(WWT_DEFSIZEBOX, Colours::DarkGreen),
  EndContainer(),
  NWidget(NWID_HORIZONTAL),
    NWidget(NWID_VERTICAL),
      NWidgetFunction(MakePickerClassWidgets),
      NWidget(WWT_PANEL, Colours::DarkGreen),
        NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_picker, 0), SetPadding(WidgetDimensions::unscaled.picker),
          NWidget(WWT_LABEL, Colours::Invalid), SetMinimalSize(144, 11), SetFill(1, 0), SetStringTip(STR_STATION_BUILD_ORIENTATION),
          NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0), SetPIPRatio(1, 0, 1),
            NWidget(WWT_PANEL, Colours::Grey, WID_BRAS_PLATFORM_DIR_X), SetFill(0, 0), SetToolTip(STR_STATION_BUILD_RAILROAD_ORIENTATION_TOOLTIP), EndContainer(),
            NWidget(WWT_PANEL, Colours::Grey, WID_BRAS_PLATFORM_DIR_Y), SetFill(0, 0), SetToolTip(STR_STATION_BUILD_RAILROAD_ORIENTATION_TOOLTIP), EndContainer(),
          EndContainer(),
          NWidget(WWT_LABEL, Colours::Invalid), SetMinimalSize(144, 11), SetFill(1, 0), SetStringTip(STR_STATION_BUILD_NUMBER_OF_TRACKS),
          NWidget(NWID_HORIZONTAL), SetPIPRatio(1, 0, 1),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_NUM_1), SetAspect(1.25f), SetStringTip(STR_BLACK_1, STR_STATION_BUILD_NUMBER_OF_TRACKS_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_NUM_2), SetAspect(1.25f), SetStringTip(STR_BLACK_2, STR_STATION_BUILD_NUMBER_OF_TRACKS_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_NUM_3), SetAspect(1.25f), SetStringTip(STR_BLACK_3, STR_STATION_BUILD_NUMBER_OF_TRACKS_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_NUM_4), SetAspect(1.25f), SetStringTip(STR_BLACK_4, STR_STATION_BUILD_NUMBER_OF_TRACKS_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_NUM_5), SetAspect(1.25f), SetStringTip(STR_BLACK_5, STR_STATION_BUILD_NUMBER_OF_TRACKS_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_NUM_6), SetAspect(1.25f), SetStringTip(STR_BLACK_6, STR_STATION_BUILD_NUMBER_OF_TRACKS_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_NUM_7), SetAspect(1.25f), SetStringTip(STR_BLACK_7, STR_STATION_BUILD_NUMBER_OF_TRACKS_TOOLTIP),
          EndContainer(),
          NWidget(WWT_LABEL, Colours::Invalid), SetMinimalSize(144, 11), SetFill(1, 0), SetStringTip(STR_STATION_BUILD_PLATFORM_LENGTH),
          NWidget(NWID_HORIZONTAL), SetPIPRatio(1, 0, 1),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_LEN_1), SetAspect(1.25f), SetStringTip(STR_BLACK_1, STR_STATION_BUILD_PLATFORM_LENGTH_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_LEN_2), SetAspect(1.25f), SetStringTip(STR_BLACK_2, STR_STATION_BUILD_PLATFORM_LENGTH_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_LEN_3), SetAspect(1.25f), SetStringTip(STR_BLACK_3, STR_STATION_BUILD_PLATFORM_LENGTH_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_LEN_4), SetAspect(1.25f), SetStringTip(STR_BLACK_4, STR_STATION_BUILD_PLATFORM_LENGTH_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_LEN_5), SetAspect(1.25f), SetStringTip(STR_BLACK_5, STR_STATION_BUILD_PLATFORM_LENGTH_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_LEN_6), SetAspect(1.25f), SetStringTip(STR_BLACK_6, STR_STATION_BUILD_PLATFORM_LENGTH_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_LEN_7), SetAspect(1.25f), SetStringTip(STR_BLACK_7, STR_STATION_BUILD_PLATFORM_LENGTH_TOOLTIP),
          EndContainer(),
          NWidget(NWID_HORIZONTAL), SetPIPRatio(1, 0, 1),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_PLATFORM_DRAG_N_DROP), SetMinimalSize(75, 12), SetStringTip(STR_STATION_BUILD_DRAG_DROP, STR_STATION_BUILD_DRAG_DROP_TOOLTIP),
          EndContainer(),
          NWidget(WWT_LABEL, Colours::Invalid), SetStringTip(STR_STATION_BUILD_COVERAGE_AREA_TITLE), SetFill(1, 0),
          NWidget(NWID_HORIZONTAL), SetPIPRatio(1, 0, 1),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_HIGHLIGHT_OFF), SetMinimalSize(60, 12), SetStringTip(STR_STATION_BUILD_COVERAGE_OFF, STR_STATION_BUILD_COVERAGE_AREA_OFF_TOOLTIP),
            NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAS_HIGHLIGHT_ON), SetMinimalSize(60, 12), SetStringTip(STR_STATION_BUILD_COVERAGE_ON, STR_STATION_BUILD_COVERAGE_AREA_ON_TOOLTIP),
          EndContainer(),
          NWidget(WWT_EMPTY, Colours::Invalid, WID_BRAS_COVERAGE_TEXTS), SetFill(1, 1), SetResize(1, 0), SetMinimalTextLines(2, 0),
        EndContainer(),
      EndContainer(),
    EndContainer(),
    NWidgetFunction(MakePickerTypeWidgets),
  EndContainer(),
};

static WindowDesc _station_builder_desc(
  WindowPosition::Automatic, "build_station_rail", 0, 0,
  WindowClass::BuildStation, WindowClass::BuildToolbar,
  WindowDefaultFlag::Construction,
  _nested_station_builder_widgets,
  &BuildRailStationWindow::hotkeys
);

static Window *ShowStationBuilder(Window *parent)
{
  return new BuildRailStationWindow(_station_builder_desc, parent);
}
 
struct BuildSignalWindow : public PickerWindowBase {
private:
  Dimension sig_sprite_size{}; 
  int sig_sprite_bottom_offset = 0; 

  void DrawSignalSprite(const Rect &r, SpriteID image) const
  {
    Point offset;
    Dimension sprite_size = GetSpriteSize(image, &offset);
    Rect ir = r.Shrink(WidgetDimensions::scaled.imgbtn);
    int x = CentreBounds(ir.left, ir.right, sprite_size.width - offset.x) - offset.x; // centered
    int y = ir.top - sig_sprite_bottom_offset +
        (ir.Height() + sig_sprite_size.height) / 2; // aligned to bottom
 
    DrawSprite(image, PAL_NONE, x, y);
  }

  void SetSignalUIMode()
  {
    bool show_non_path_signals = (_settings_client.gui.signal_gui_mode == SIGNAL_GUI_ALL);
 
    this->GetWidget<NWidgetStacked>(WID_BS_BLOCK_SEL)->SetDisplayedPlane(show_non_path_signals ? 0 : SZSP_NONE);
    this->GetWidget<NWidgetStacked>(WID_BS_BLOCK_SPACER_SEL)->SetDisplayedPlane(show_non_path_signals ? 0 : SZSP_NONE);
  }
 
public:
  BuildSignalWindow(WindowDesc &desc, Window *parent) : PickerWindowBase(desc, parent)
  {
    this->CreateNestedTree();
    this->SetSignalUIMode();
    this->FinishInitNested(TransportType::Rail);
    this->OnInvalidateData();
  }
 
  void Close([[maybe_unused]] int data = 0) override
  {
    _convert_signal_button = false;
    this->PickerWindowBase::Close();
  }
 
  void OnInit() override
  {
    /* Calculate maximum signal sprite size. */
    this->sig_sprite_size.width = 0;
    this->sig_sprite_size.height = 0;
    this->sig_sprite_bottom_offset = 0;
    const RailTypeInfo *rti = GetRailTypeInfo(_cur_railtype);
    for (SignalType type : EnumRange(SignalType::End)) {
      for (SignalVariant variant : {SignalVariant::Electric, SignalVariant::Semaphore}) {
        for (SignalState state : {SignalState::Red, SignalState::Green}) {
          Point offset;
          Dimension sprite_size = GetSpriteSize(rti->gui_sprites.signals[type][variant][state], &offset);
          this->sig_sprite_bottom_offset = std::max<int>(this->sig_sprite_bottom_offset, sprite_size.height);
          this->sig_sprite_size.width = std::max<int>(this->sig_sprite_size.width, sprite_size.width - offset.x);
          this->sig_sprite_size.height = std::max<int>(this->sig_sprite_size.height, sprite_size.height - offset.y);
        }
      }
    }
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    if (widget == WID_BS_DRAG_SIGNALS_DENSITY_LABEL) {
      /* Two digits for signals density. */
      size.width = std::max(size.width, 2 * GetDigitWidth() + padding.width + WidgetDimensions::scaled.framerect.Horizontal());
    } else if (IsInsideMM(widget, WID_BS_SEMAPHORE_NORM, WID_BS_ELECTRIC_PBS_OWAY + 1)) {
      size.width = std::max(size.width, this->sig_sprite_size.width + padding.width);
      size.height = std::max(size.height, this->sig_sprite_size.height + padding.height);
    }
  }
 
  std::string GetWidgetString(WidgetID widget, StringID stringid) const override
  {
    switch (widget) {
      case WID_BS_DRAG_SIGNALS_DENSITY_LABEL:
        return GetString(STR_JUST_INT, _settings_client.gui.drag_signals_density);
 
      default:
        return this->Window::GetWidgetString(widget, stringid);
    }
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    if (IsInsideMM(widget, WID_BS_SEMAPHORE_NORM, WID_BS_ELECTRIC_PBS_OWAY + 1)) {
      /* Extract signal from widget number. */
      SignalType type = static_cast<SignalType>((widget - WID_BS_SEMAPHORE_NORM) % to_underlying(SignalType::End));
      SignalVariant var = static_cast<SignalVariant>(to_underlying(SignalVariant::Semaphore) - (widget - WID_BS_SEMAPHORE_NORM) / to_underlying(SignalType::End)); // SignalVariant order is reversed compared to the widgets.
      SpriteID sprite = GetRailTypeInfo(_cur_railtype)->gui_sprites.signals[type][var][static_cast<SignalState>(this->IsWidgetLowered(widget))];
 
      this->DrawSignalSprite(r, sprite);
    }
  }
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    switch (widget) {
      case WID_BS_SEMAPHORE_NORM:
      case WID_BS_SEMAPHORE_ENTRY:
      case WID_BS_SEMAPHORE_EXIT:
      case WID_BS_SEMAPHORE_COMBO:
      case WID_BS_SEMAPHORE_PBS:
      case WID_BS_SEMAPHORE_PBS_OWAY:
      case WID_BS_ELECTRIC_NORM:
      case WID_BS_ELECTRIC_ENTRY:
      case WID_BS_ELECTRIC_EXIT:
      case WID_BS_ELECTRIC_COMBO:
      case WID_BS_ELECTRIC_PBS:
      case WID_BS_ELECTRIC_PBS_OWAY:
        this->RaiseWidget((_cur_signal_variant == SignalVariant::Electric ? WID_BS_ELECTRIC_NORM : WID_BS_SEMAPHORE_NORM) + to_underlying(_cur_signal_type));
 
        _cur_signal_type = static_cast<SignalType>((uint)((widget - WID_BS_SEMAPHORE_NORM) % to_underlying(SignalType::End)));
        _cur_signal_variant = widget >= WID_BS_ELECTRIC_NORM ? SignalVariant::Electric : SignalVariant::Semaphore;
 
        /* Update default (last-used) signal type in config file. */
        _settings_client.gui.default_signal_type = _cur_signal_type;
 
        /* If 'remove' button of rail build toolbar is active, disable it. */
        if (_remove_button_clicked) {
          Window *w = FindWindowById(WindowClass::BuildToolbar, TransportType::Rail);
          if (w != nullptr) ToggleRailButton_Remove(w);
        }
 
        SndClickBeep();
        break;
 
      case WID_BS_CONVERT:
        _convert_signal_button = !_convert_signal_button;
        SndClickBeep();
        break;
 
      case WID_BS_DRAG_SIGNALS_DENSITY_DECREASE:
        if (_settings_client.gui.drag_signals_density > 1) {
          _settings_client.gui.drag_signals_density--;
        }
        break;
 
      case WID_BS_DRAG_SIGNALS_DENSITY_INCREASE:
        if (_settings_client.gui.drag_signals_density < 20) {
          _settings_client.gui.drag_signals_density++;
        }
        break;
 
      default: break;
    }
 
    this->InvalidateData();
  }

  void OnInvalidateData([[maybe_unused]] int data = 0, [[maybe_unused]] bool gui_scope = true) override
  {
    if (!gui_scope) return;
    this->LowerWidget((_cur_signal_variant == SignalVariant::Electric ? WID_BS_ELECTRIC_NORM : WID_BS_SEMAPHORE_NORM) + to_underlying(_cur_signal_type));
 
    this->SetWidgetLoweredState(WID_BS_CONVERT, _convert_signal_button);
 
    this->SetWidgetDisabledState(WID_BS_DRAG_SIGNALS_DENSITY_DECREASE, _settings_client.gui.drag_signals_density == 1);
    this->SetWidgetDisabledState(WID_BS_DRAG_SIGNALS_DENSITY_INCREASE, _settings_client.gui.drag_signals_density == 20);
  }
};

static constexpr std::initializer_list<NWidgetPart> _nested_signal_builder_widgets = {
  /* Title bar and buttons. */
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, Colours::DarkGreen),
    NWidget(WWT_CAPTION, Colours::DarkGreen, WID_BS_CAPTION), SetStringTip(STR_BUILD_SIGNAL_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
  EndContainer(),
 
  /* Container for both signal groups, spacers, and convert/autofill buttons. */
  NWidget(NWID_HORIZONTAL),
    /* Block signals (can be hidden). */
    NWidget(NWID_SELECTION, Colours::Invalid, WID_BS_BLOCK_SEL),
      NWidget(NWID_VERTICAL, NWidContainerFlag::EqualSize),
        /* Semaphore block signals. */
        NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize),
          NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_SEMAPHORE_NORM), SetToolTip(STR_BUILD_SIGNAL_SEMAPHORE_NORM_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
          NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_SEMAPHORE_ENTRY), SetToolTip(STR_BUILD_SIGNAL_SEMAPHORE_ENTRY_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
          NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_SEMAPHORE_EXIT), SetToolTip(STR_BUILD_SIGNAL_SEMAPHORE_EXIT_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
          NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_SEMAPHORE_COMBO), SetToolTip(STR_BUILD_SIGNAL_SEMAPHORE_COMBO_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
        EndContainer(),
        /* Electric block signals. */
        NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize),
          NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_ELECTRIC_NORM), SetToolTip(STR_BUILD_SIGNAL_ELECTRIC_NORM_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
          NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_ELECTRIC_ENTRY), SetToolTip(STR_BUILD_SIGNAL_ELECTRIC_ENTRY_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
          NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_ELECTRIC_EXIT), SetToolTip(STR_BUILD_SIGNAL_ELECTRIC_EXIT_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
          NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_ELECTRIC_COMBO), SetToolTip(STR_BUILD_SIGNAL_ELECTRIC_COMBO_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
        EndContainer(),
      EndContainer(),
    EndContainer(),
 
    /* Divider (only shown if block signals visible). */
    NWidget(NWID_SELECTION, Colours::Invalid, WID_BS_BLOCK_SPACER_SEL),
      NWidget(WWT_PANEL, Colours::DarkGreen), SetFill(0, 0), SetToolbarSpacerMinimalSize(), EndContainer(),
    EndContainer(),
 
    /* Path signals. */
    NWidget(NWID_VERTICAL, NWidContainerFlag::EqualSize),
      /* Semaphore path signals. */
      NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize),
        NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_SEMAPHORE_PBS), SetToolTip(STR_BUILD_SIGNAL_SEMAPHORE_PBS_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
        NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_SEMAPHORE_PBS_OWAY), SetToolTip(STR_BUILD_SIGNAL_SEMAPHORE_PBS_OWAY_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
      EndContainer(),
      /* Electric path signals. */
      NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize),
        NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_ELECTRIC_PBS), SetToolTip(STR_BUILD_SIGNAL_ELECTRIC_PBS_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
        NWidget(WWT_PANEL, Colours::DarkGreen, WID_BS_ELECTRIC_PBS_OWAY), SetToolTip(STR_BUILD_SIGNAL_ELECTRIC_PBS_OWAY_TOOLTIP), SetToolbarMinimalSize(1), EndContainer(),
      EndContainer(),
    EndContainer(),
 
    /* Convert/autofill buttons. */
    NWidget(NWID_VERTICAL, NWidContainerFlag::EqualSize),
      NWidget(WWT_IMGBTN, Colours::DarkGreen, WID_BS_CONVERT), SetSpriteTip(SPR_IMG_SIGNAL_CONVERT, STR_BUILD_SIGNAL_CONVERT_TOOLTIP), SetFill(0, 1),
      NWidget(WWT_PANEL, Colours::DarkGreen), SetToolTip(STR_BUILD_SIGNAL_DRAG_SIGNALS_DENSITY_TOOLTIP), SetFill(0, 1),
        NWidget(NWID_VERTICAL), SetPadding(2), SetPIPRatio(1, 0, 1),
          NWidget(WWT_LABEL, Colours::Invalid, WID_BS_DRAG_SIGNALS_DENSITY_LABEL), SetToolTip(STR_BUILD_SIGNAL_DRAG_SIGNALS_DENSITY_TOOLTIP), SetTextStyle(TextColour::Orange), SetFill(1, 1),
          NWidget(NWID_HORIZONTAL), SetPIPRatio(1, 0, 1),
            NWidget(WWT_PUSHARROWBTN, Colours::Grey, WID_BS_DRAG_SIGNALS_DENSITY_DECREASE), SetMinimalSize(9, 12), SetArrowWidgetTypeTip(ArrowWidgetType::Decrease, STR_BUILD_SIGNAL_DRAG_SIGNALS_DENSITY_DECREASE_TOOLTIP),
            NWidget(WWT_PUSHARROWBTN, Colours::Grey, WID_BS_DRAG_SIGNALS_DENSITY_INCREASE), SetMinimalSize(9, 12), SetArrowWidgetTypeTip(ArrowWidgetType::Increase, STR_BUILD_SIGNAL_DRAG_SIGNALS_DENSITY_INCREASE_TOOLTIP),
          EndContainer(),
        EndContainer(),
      EndContainer(),
    EndContainer(),
  EndContainer(),
};

static WindowDesc _signal_builder_desc(
  WindowPosition::Automatic, {}, 0, 0,
  WindowClass::BuildSignal, WindowClass::BuildToolbar,
  WindowDefaultFlag::Construction,
  _nested_signal_builder_widgets
);

static void ShowSignalBuilder(Window *parent)
{
  new BuildSignalWindow(_signal_builder_desc, parent);
}
 
struct BuildRailDepotWindow : public PickerWindowBase {
  BuildRailDepotWindow(WindowDesc &desc, Window *parent) : PickerWindowBase(desc, parent)
  {
    this->InitNested(TransportType::Rail);
    this->LowerWidget(WID_BRAD_DEPOT_NE + _build_depot_direction);
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    if (!IsInsideMM(widget, WID_BRAD_DEPOT_NE, WID_BRAD_DEPOT_NW + 1)) return;
 
    size.width  = ScaleGUITrad(64) + WidgetDimensions::scaled.fullbevel.Horizontal();
    size.height = ScaleGUITrad(48) + WidgetDimensions::scaled.fullbevel.Vertical();
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    if (!IsInsideMM(widget, WID_BRAD_DEPOT_NE, WID_BRAD_DEPOT_NW + 1)) return;
 
    DrawPixelInfo tmp_dpi;
    Rect ir = r.Shrink(WidgetDimensions::scaled.bevel);
    if (FillDrawPixelInfo(&tmp_dpi, ir)) {
      AutoRestoreBackup dpi_backup(_cur_dpi, &tmp_dpi);
      int x = (ir.Width()  - ScaleSpriteTrad(64)) / 2 + ScaleSpriteTrad(31);
      int y = (ir.Height() + ScaleSpriteTrad(48)) / 2 - ScaleSpriteTrad(31);
      DrawTrainDepotSprite(x, y, static_cast<DiagDirection>(widget - WID_BRAD_DEPOT_NE + to_underlying(DiagDirection::NE)), _cur_railtype);
    }
  }
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    switch (widget) {
      case WID_BRAD_DEPOT_NE:
      case WID_BRAD_DEPOT_SE:
      case WID_BRAD_DEPOT_SW:
      case WID_BRAD_DEPOT_NW:
        this->RaiseWidget(WID_BRAD_DEPOT_NE + _build_depot_direction);
        _build_depot_direction = (DiagDirection)(widget - WID_BRAD_DEPOT_NE);
        this->LowerWidget(WID_BRAD_DEPOT_NE + _build_depot_direction);
        SndClickBeep();
        this->SetDirty();
        break;
    }
  }
};

static constexpr std::initializer_list<NWidgetPart> _nested_build_depot_widgets = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, Colours::DarkGreen),
    NWidget(WWT_CAPTION, Colours::DarkGreen), SetStringTip(STR_BUILD_DEPOT_TRAIN_ORIENTATION_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
  EndContainer(),
  NWidget(WWT_PANEL, Colours::DarkGreen),
    NWidget(NWID_HORIZONTAL_LTR), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0), SetPIPRatio(1, 0, 1), SetPadding(WidgetDimensions::unscaled.picker),
      NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
        NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAD_DEPOT_NW), SetToolTip(STR_BUILD_DEPOT_TRAIN_ORIENTATION_TOOLTIP),
        NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAD_DEPOT_SW), SetToolTip(STR_BUILD_DEPOT_TRAIN_ORIENTATION_TOOLTIP),
      EndContainer(),
      NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_normal, 0),
        NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAD_DEPOT_NE), SetToolTip(STR_BUILD_DEPOT_TRAIN_ORIENTATION_TOOLTIP),
        NWidget(WWT_TEXTBTN, Colours::Grey, WID_BRAD_DEPOT_SE), SetToolTip(STR_BUILD_DEPOT_TRAIN_ORIENTATION_TOOLTIP),
      EndContainer(),
    EndContainer(),
  EndContainer(),
};

static WindowDesc _build_depot_desc(
  WindowPosition::Automatic, {}, 0, 0,
  WindowClass::BuildDepot, WindowClass::BuildToolbar,
  WindowDefaultFlag::Construction,
  _nested_build_depot_widgets
);
 
static void ShowBuildTrainDepotPicker(Window *parent)
{
  new BuildRailDepotWindow(_build_depot_desc, parent);
}
 
class WaypointPickerCallbacks : public PickerCallbacksNewGRFClass<StationClass> {
public:
  WaypointPickerCallbacks() : PickerCallbacksNewGRFClass<StationClass>("fav_waypoints") {}
 
  GrfSpecFeature GetFeature() const override { return GrfSpecFeature::Stations; }
 
  StringID GetClassTooltip() const override { return STR_PICKER_WAYPOINT_CLASS_TOOLTIP; }
  StringID GetTypeTooltip() const override { return STR_PICKER_WAYPOINT_TYPE_TOOLTIP; }
  StringID GetCollectionTooltip() const override { return STR_PICKER_WAYPOINT_COLLECTION_TOOLTIP; }
 
  bool IsActive() const override
  {
    for (const auto &cls : StationClass::Classes()) {
      if (!IsWaypointClass(cls)) continue;
      for (const auto *spec : cls.Specs()) {
        if (spec != nullptr) return true;
      }
    }
    return false;
  }
 
  bool HasClassChoice() const override
  {
    return std::ranges::count_if(StationClass::Classes(), [](const auto &cls) { return IsWaypointClass(cls); }) > 1;
  }
 
  void Close(int) override { ResetObjectToPlace(); }
  int GetSelectedClass() const override { return _waypoint_gui.sel_class.base(); }
  void SetSelectedClass(int id) const override { _waypoint_gui.sel_class = this->GetClassIndex(id); }
 
  StringID GetClassName(int id) const override
  {
    const auto *sc = GetClass(id);
    if (!IsWaypointClass(*sc)) return INVALID_STRING_ID;
    return sc->name;
  }
 
  int GetSelectedType() const override { return _waypoint_gui.sel_type; }
  void SetSelectedType(int id) const override { _waypoint_gui.sel_type = id; }
 
  StringID GetTypeName(int cls_id, int id) const override
  {
    const auto *spec = this->GetSpec(cls_id, id);
    return (spec == nullptr) ? STR_STATION_CLASS_WAYP_WAYPOINT : spec->name;
  }
 
  std::span<const BadgeID> GetTypeBadges(int cls_id, int id) const override
  {
    const auto *spec = this->GetSpec(cls_id, id);
    if (spec == nullptr) return {};
    return spec->badges;
  }
 
  bool IsTypeAvailable(int cls_id, int id) const override
  {
    return IsStationAvailable(this->GetSpec(cls_id, id));
  }
 
  void DrawType(int x, int y, int cls_id, int id) const override
  {
    DrawWaypointSprite(x, y, this->GetClassIndex(cls_id), id, _cur_railtype);
  }
 
  void FillUsedItems(std::set<PickerItem> &items) override
  {
    bool default_added = false;
    for (const Waypoint *wp : Waypoint::Iterate()) {
      if (wp->owner != _local_company || HasBit(wp->waypoint_flags, WPF_ROAD)) continue;
      if (!default_added && StationUsesDefaultType(wp)) {
        items.insert({GrfID{}, 0, STAT_CLASS_WAYP.base(), 0});
        default_added = true;
      }
      for (const auto &sm : wp->speclist) {
        if (sm.spec == nullptr) continue;
        items.insert({sm.grfid, sm.localidx, sm.spec->class_index.base(), sm.spec->index});
      }
    }
  }
 
  static WaypointPickerCallbacks instance;
};
/* static */ WaypointPickerCallbacks WaypointPickerCallbacks::instance;
 
struct BuildRailWaypointWindow : public PickerWindow {
  BuildRailWaypointWindow(WindowDesc &desc, Window *parent) : PickerWindow(desc, parent, TransportType::Rail, WaypointPickerCallbacks::instance)
  {
    this->ConstructWindow();
  }
 
  static inline HotkeyList hotkeys{"buildrailwaypoint", {
    Hotkey('F', "focus_filter_box", PCWHK_FOCUS_FILTER_BOX),
  }};
};

static constexpr std::initializer_list<NWidgetPart> _nested_build_waypoint_widgets = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, Colours::DarkGreen),
    NWidget(WWT_CAPTION, Colours::DarkGreen), SetStringTip(STR_WAYPOINT_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
    NWidget(WWT_SHADEBOX, Colours::DarkGreen),
    NWidget(WWT_DEFSIZEBOX, Colours::DarkGreen),
  EndContainer(),
  NWidget(NWID_HORIZONTAL),
    NWidgetFunction(MakePickerClassWidgets),
    NWidgetFunction(MakePickerTypeWidgets),
  EndContainer(),
};

static WindowDesc _build_waypoint_desc(
  WindowPosition::Automatic, "build_waypoint", 0, 0,
  WindowClass::BuildWaypoint, WindowClass::BuildToolbar,
  WindowDefaultFlag::Construction,
  _nested_build_waypoint_widgets,
  &BuildRailWaypointWindow::hotkeys
);
 
static void ShowBuildWaypointPicker(Window *parent)
{
  if (!WaypointPickerCallbacks::instance.IsActive()) return;
  new BuildRailWaypointWindow(_build_waypoint_desc, parent);
}

void InitializeRailGui()
{
  _build_depot_direction = DiagDirection::NW;
  _station_gui.sel_class = STAT_CLASS_DFLT;
  _station_gui.sel_type = 0;
  _waypoint_gui.sel_class = STAT_CLASS_WAYP;
  _waypoint_gui.sel_type = 0;
}

void ReinitGuiAfterToggleElrail(bool disable)
{
  if (disable && _last_built_railtype == RAILTYPE_ELECTRIC) {
    _last_built_railtype = _cur_railtype = RAILTYPE_RAIL;
    BuildRailToolbarWindow *w = dynamic_cast<BuildRailToolbarWindow *>(FindWindowById(WindowClass::BuildToolbar, TransportType::Rail));
    if (w != nullptr) w->ModifyRailType(_cur_railtype);
  }
  MarkWholeScreenDirty();
}

void SetDefaultRailGui()
{
  if (_local_company == COMPANY_SPECTATOR || !Company::IsValidID(_local_company)) return;
 
  RailType rt;
  switch (_settings_client.gui.default_rail_road_type) {
    case DefaultRailRoadType::MostUsed: {
      /* Find the most used rail type */
      const Company *c = Company::Get(_local_company);
 
      rt = static_cast<RailType>(std::distance(std::begin(c->infrastructure.rail), std::ranges::max_element(c->infrastructure.rail)));
      if (c->infrastructure.rail[rt] > 0) break;
 
      /* No rail, just get the first available one */
      [[fallthrough]];
    }
    case DefaultRailRoadType::FirstAvailable: {
      /* Use first available type */
      std::vector<RailType>::const_iterator it = std::find_if(_sorted_railtypes.begin(), _sorted_railtypes.end(),
          [](RailType r) { return HasRailTypeAvail(_local_company, r); });
      rt = it != _sorted_railtypes.end() ? *it : RAILTYPE_BEGIN;
      break;
    }
    case DefaultRailRoadType::LastAvailable: {
      /* Use last available type */
      std::vector<RailType>::const_reverse_iterator it = std::find_if(_sorted_railtypes.rbegin(), _sorted_railtypes.rend(),
          [](RailType r){ return HasRailTypeAvail(_local_company, r); });
      rt = it != _sorted_railtypes.rend() ? *it : RAILTYPE_BEGIN;
      break;
    }
    default:
      NOT_REACHED();
  }
 
  _last_built_railtype = _cur_railtype = rt;
}

void ResetSignalVariant(int32_t)
{
  SignalVariant new_variant = (TimerGameCalendar::year < _settings_client.gui.semaphore_build_before ? SignalVariant::Semaphore : SignalVariant::Electric);
 
  if (new_variant != _cur_signal_variant) {
    Window *w = FindWindowById(WindowClass::BuildSignal, 0);
    if (w != nullptr) {
      w->SetDirty();
      w->RaiseWidget((_cur_signal_variant == SignalVariant::Electric ? WID_BS_ELECTRIC_NORM : WID_BS_SEMAPHORE_NORM) + to_underlying(_cur_signal_type));
    }
    _cur_signal_variant = new_variant;
  }
}

static const IntervalTimer<TimerGameCalendar> _check_reset_signal{{TimerGameCalendar::Trigger::Year, TimerGameCalendar::Priority::None}, [](auto)
{
  if (TimerGameCalendar::year != _settings_client.gui.semaphore_build_before) return;
 
  ResetSignalVariant();
}};

void InitializeSignalGui()
{
  _convert_signal_button = false;
  _cur_signal_type = _settings_client.gui.default_signal_type;
  ResetSignalVariant();
}

DropDownList GetRailTypeDropDownList(bool for_replacement, bool all_option)
{
  RailTypes used_railtypes;
  RailTypes avail_railtypes;
 
  const Company *c = Company::Get(_local_company);
 
  /* Find the used railtypes. */
  if (for_replacement) {
    avail_railtypes = GetCompanyRailTypes(c->index, false);
    used_railtypes  = GetRailTypes(false);
  } else {
    avail_railtypes = c->avail_railtypes;
    used_railtypes  = GetRailTypes(true);
  }
 
  DropDownList list;
 
  if (all_option) {
    list.push_back(MakeDropDownListStringItem(STR_REPLACE_ALL_RAILTYPE, INVALID_RAILTYPE));
  }
 
  Dimension d = { 0, 0 };
  /* Get largest icon size, to ensure text is aligned on each menu item. */
  if (!for_replacement) {
    used_railtypes.Reset(_railtypes_hidden_mask);
    for (const auto &rt : _sorted_railtypes) {
      if (!used_railtypes.Test(rt)) continue;
      const RailTypeInfo *rti = GetRailTypeInfo(rt);
      d = maxdim(d, GetSpriteSize(rti->gui_sprites.build_x_rail));
    }
  }
 
  /* Shared list so that each item can take ownership. */
  auto badge_class_list = std::make_shared<GUIBadgeClasses>(GrfSpecFeature::RailTypes);
 
  for (const auto &rt : _sorted_railtypes) {
    /* If it's not used ever, don't show it to the user. */
    if (!used_railtypes.Test(rt)) continue;
 
    const RailTypeInfo *rti = GetRailTypeInfo(rt);
 
    if (for_replacement) {
      list.push_back(MakeDropDownListBadgeItem(badge_class_list, rti->badges, GrfSpecFeature::RailTypes, rti->introduction_date, GetString(rti->strings.replace_text), rt, !avail_railtypes.Test(rt)));
    } else {
      std::string str = rti->max_speed > 0
        ? GetString(STR_TOOLBAR_RAILTYPE_VELOCITY, rti->strings.menu_text, rti->max_speed)
        : GetString(rti->strings.menu_text);
      list.push_back(MakeDropDownListBadgeIconItem(badge_class_list, rti->badges, GrfSpecFeature::RailTypes, rti->introduction_date, RailBuildCost(rt), d, rti->gui_sprites.build_x_rail, PAL_NONE, std::move(str), rt, !avail_railtypes.Test(rt)));
    }
  }
 
  if (list.empty()) {
    /* Empty dropdowns are not allowed */
    list.push_back(MakeDropDownListStringItem(STR_NONE, INVALID_RAILTYPE, true));
  }
 
  return list;
}