newgrf_debug_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 "core/backup_type.hpp"
#include "core/geometry_func.hpp"
#include "window_gui.h"
#include "window_func.h"
#include "random_access_file_type.h"
#include "spritecache.h"
#include "string_func.h"
#include "strings_func.h"
#include "textbuf_gui.h"
#include "vehicle_gui.h"
#include "zoom_func.h"
#include "core/string_consumer.hpp"
 
#include "engine_base.h"
#include "industry.h"
#include "object_base.h"
#include "station_base.h"
#include "town.h"
#include "vehicle_base.h"
#include "train.h"
#include "roadveh.h"
 
#include "newgrf_act5.h"
#include "newgrf_airport.h"
#include "newgrf_airporttiles.h"
#include "newgrf_badge.h"
#include "newgrf_debug.h"
#include "newgrf_object.h"
#include "newgrf_spritegroup.h"
#include "newgrf_station.h"
#include "newgrf_town.h"
#include "newgrf_railtype.h"
#include "newgrf_industries.h"
#include "newgrf_industrytiles.h"
 
#include "widgets/newgrf_debug_widget.h"
 
#include "table/strings.h"
 
#include "safeguards.h"

NewGrfDebugSpritePicker _newgrf_debug_sprite_picker = { SPM_NONE, nullptr, {} };

static inline uint GetFeatureIndex(uint window_number)
{
  return GB(window_number, 0, 24);
}

static inline uint GetInspectWindowNumber(GrfSpecFeature feature, uint index)
{
  assert((index >> 24) == 0);
  return (feature << 24) | index;
}
 
static inline uint GetInspectWindowNumber(GrfSpecFeature feature, ConvertibleThroughBase auto index) { return GetInspectWindowNumber(feature, index.base()); }

enum NIType : uint8_t {
  NIT_INT,   
  NIT_CARGO, 
};
 
using NIReadProc = uint32_t(const void *b);

struct NIProperty {
  std::string_view name;          
  NIReadProc *read_proc; 
  uint8_t prop;                 
  uint8_t type;
};
 

struct NICallback {
  std::string_view name;          
  NIReadProc *read_proc; 
  std::variant<
    std::monostate,
    VehicleCallbackMask,
    StationCallbackMask,
    RoadStopCallbackMask,
    HouseCallbackMask,
    CanalCallbackMask,
    CargoCallbackMask,
    IndustryCallbackMask,
    IndustryTileCallbackMask,
    ObjectCallbackMask,
    AirportTileCallbackMask> cb_bit; 
  uint16_t cb_id;              
};

struct NIVariable {
  std::string_view name;
  uint8_t var;
};

class NIHelper {
public:
  virtual ~NIHelper() = default;

  virtual bool IsInspectable(uint index) const = 0;

  virtual uint GetParent(uint index) const = 0;

  virtual const void *GetInstance(uint index) const = 0;

  virtual const void *GetSpec(uint index) const = 0;

  virtual std::string GetName(uint index) const = 0;

  virtual uint32_t GetGRFID(uint index) const = 0;

  virtual std::span<const BadgeID> GetBadges(uint index) const = 0;

  virtual uint Resolve(uint index, uint var, uint param, bool &avail) const = 0;

  virtual bool PSAWithParameter() const
  {
    return false;
  }

  virtual const std::span<int32_t> GetPSA([[maybe_unused]] uint index, [[maybe_unused]] uint32_t grfid) const
  {
    return {};
  }
};
 

struct NIFeature {
  std::span<const NIProperty> properties; 
  std::span<const NICallback> callbacks; 
  std::span<const NIVariable> variables; 
  std::unique_ptr<const NIHelper> helper; 
};
 
/* Load all the NewGRF debug data; externalised as it is just a huge bunch of tables. */
#include "table/newgrf_debug_data.h"

static inline GrfSpecFeature GetFeatureNum(uint window_number)
{
  return (GrfSpecFeature)GB(window_number, 24, 8);
}

static inline const NIFeature *GetFeature(uint window_number)
{
  GrfSpecFeature idx = GetFeatureNum(window_number);
  return idx < GSF_FAKE_END ? _nifeatures[idx] : nullptr;
}

static inline const NIHelper &GetFeatureHelper(uint window_number)
{
  return *GetFeature(window_number)->helper;
}

struct NewGRFInspectWindow : Window {
  static uint32_t var60params[GSF_FAKE_END][0x20];

  uint32_t caller_grfid = 0;

  uint chain_index = 0;

  uint8_t current_edit_param = 0;
 
  Scrollbar *vscroll = nullptr;

  static bool HasVariableParameter(uint variable)
  {
    return IsInsideBS(variable, 0x60, 0x20);
  }

  void SetCallerGRFID(uint32_t grfid)
  {
    this->caller_grfid = grfid;
    this->SetDirty();
  }

  bool HasChainIndex() const
  {
    GrfSpecFeature f = GetFeatureNum(this->window_number);
    return f == GSF_TRAINS || f == GSF_ROADVEHICLES;
  }

  uint GetFeatureIndex() const
  {
    uint index = ::GetFeatureIndex(this->window_number);
    if (this->chain_index > 0) {
      assert(this->HasChainIndex());
      const Vehicle *v = Vehicle::Get(index);
      v = v->Move(this->chain_index);
      if (v != nullptr) index = v->index.base();
    }
    return index;
  }

  void ValidateChainIndex()
  {
    if (this->chain_index == 0) return;
 
    assert(this->HasChainIndex());
 
    const Vehicle *v = Vehicle::Get(::GetFeatureIndex(this->window_number));
    v = v->Move(this->chain_index);
    if (v == nullptr) this->chain_index = 0;
  }
 
  NewGRFInspectWindow(WindowDesc &desc, WindowNumber wno) : Window(desc)
  {
    this->CreateNestedTree();
    this->vscroll = this->GetScrollbar(WID_NGRFI_SCROLLBAR);
    this->FinishInitNested(wno);
 
    this->vscroll->SetCount(0);
    this->SetWidgetDisabledState(WID_NGRFI_PARENT, GetFeatureHelper(this->window_number).GetParent(this->GetFeatureIndex()) == UINT32_MAX);
 
    this->OnInvalidateData(0, true);
  }
 
  std::string GetWidgetString(WidgetID widget, StringID stringid) const override
  {
    if (widget != WID_NGRFI_CAPTION) return this->Window::GetWidgetString(widget, stringid);
 
    return GetFeatureHelper(this->window_number).GetName(this->GetFeatureIndex());
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    switch (widget) {
      case WID_NGRFI_VEH_CHAIN: {
        assert(this->HasChainIndex());
        GrfSpecFeature f = GetFeatureNum(this->window_number);
        size.height = std::max(size.height, GetVehicleImageCellSize((VehicleType)(VEH_TRAIN + (f - GSF_TRAINS)), EIT_IN_DEPOT).height + 2 + WidgetDimensions::scaled.bevel.Vertical());
        break;
      }
 
      case WID_NGRFI_MAINPANEL:
        fill.height = resize.height = std::max(11, GetCharacterHeight(FS_NORMAL) + WidgetDimensions::scaled.vsep_normal);
        resize.width  = 1;
 
        size.height = 5 * resize.height + WidgetDimensions::scaled.frametext.Vertical();
        break;
    }
  }

  void DrawString(const Rect &r, int offset, const std::string &string) const
  {
    offset -= this->vscroll->GetPosition();
    if (offset < 0 || offset >= this->vscroll->GetCapacity()) return;
 
    ::DrawString(r.Shrink(WidgetDimensions::scaled.frametext).Shrink(0, offset * this->resize.step_height, 0, 0), string, TC_BLACK);
  }

  void DrawVehicleChainWidget(const Rect &r) const
  {
    const Vehicle *v = Vehicle::Get(this->GetFeatureIndex());
    int total_width = 0;
    int sel_start = 0;
    int sel_end = 0;
    for (const Vehicle *u = v->First(); u != nullptr; u = u->Next()) {
      if (u == v) sel_start = total_width;
      switch (u->type) {
        case VEH_TRAIN: total_width += Train::From(u)->GetDisplayImageWidth(); break;
        case VEH_ROAD:  total_width += RoadVehicle::From(u)->GetDisplayImageWidth(); break;
        default: NOT_REACHED();
      }
      if (u == v) sel_end = total_width;
    }
 
    Rect br = r.Shrink(WidgetDimensions::scaled.bevel);
    int width = br.Width();
    int skip = 0;
    if (total_width > width) {
      int sel_center = (sel_start + sel_end) / 2;
      if (sel_center > width / 2) skip = std::min(total_width - width, sel_center - width / 2);
    }
 
    GrfSpecFeature f = GetFeatureNum(this->window_number);
    int h = GetVehicleImageCellSize((VehicleType)(VEH_TRAIN + (f - GSF_TRAINS)), EIT_IN_DEPOT).height;
    int y = CentreBounds(br.top, br.bottom, h);
    DrawVehicleImage(v->First(), br, VehicleID::Invalid(), EIT_IN_DETAILS, skip);
 
    /* Highlight the articulated part (this is different to the whole-vehicle highlighting of DrawVehicleImage */
    if (_current_text_dir == TD_RTL) {
      DrawFrameRect(r.right - sel_end + skip, y, r.right - sel_start + skip, y + h, COLOUR_WHITE, FrameFlag::BorderOnly);
    } else {
      DrawFrameRect(r.left + sel_start - skip, y, r.left + sel_end - skip, y + h, COLOUR_WHITE, FrameFlag::BorderOnly);
    }
  }
 
  std::string GetPropertyString(const NIProperty &nip, uint value) const
  {
    switch (nip.type) {
      case NIT_INT: return GetString(STR_JUST_INT, value);
      case NIT_CARGO: return GetString(IsValidCargoType(value) ? CargoSpec::Get(value)->name : STR_QUANTITY_N_A);
      default: NOT_REACHED();
    }
  }

  void DrawMainPanelWidget(const Rect &r) const
  {
    uint index = this->GetFeatureIndex();
    const NIFeature *nif  = GetFeature(this->window_number);
    const NIHelper &nih   = *nif->helper;
    const void *base      = nih.GetInstance(index);
    const void *base_spec = nih.GetSpec(index);
 
    uint i = 0;
    if (!nif->variables.empty()) {
      this->DrawString(r, i++, "Variables:");
      for (const NIVariable &niv : nif->variables) {
        bool avail = true;
        uint param = HasVariableParameter(niv.var) ? NewGRFInspectWindow::var60params[GetFeatureNum(this->window_number)][niv.var - 0x60] : 0;
        uint value = nih.Resolve(index, niv.var, param, avail);
 
        if (!avail) continue;
 
        if (HasVariableParameter(niv.var)) {
          this->DrawString(r, i++, fmt::format("  {:02x}[{:02x}]: {:08x} ({})", niv.var, param, value, niv.name));
        } else {
          this->DrawString(r, i++, fmt::format("  {:02x}: {:08x} ({})", niv.var, value, niv.name));
        }
      }
    }
 
    auto psa = nih.GetPSA(index, this->caller_grfid);
    if (!psa.empty()) {
      if (nih.PSAWithParameter()) {
        this->DrawString(r, i++, fmt::format("Persistent storage [{:08X}]:", std::byteswap(this->caller_grfid)));
      } else {
        this->DrawString(r, i++, "Persistent storage:");
      }
      assert(psa.size() % 4 == 0);
      for (size_t j = 0; j < psa.size(); j += 4) {
        this->DrawString(r, i++, fmt::format("  {}: {} {} {} {}", j, psa[j], psa[j + 1], psa[j + 2], psa[j + 3]));
      }
    }
 
    auto badges = nih.GetBadges(index);
    if (!badges.empty()) {
      this->DrawString(r, i++, "Badges:");
      for (const BadgeID &badge_index : badges) {
        const Badge *badge = GetBadge(badge_index);
        this->DrawString(r, i++, fmt::format("  {}: {}", StrMakeValid(badge->label), GetString(badge->name)));
      }
    }
 
    if (!nif->properties.empty()) {
      this->DrawString(r, i++, "Properties:");
      for (const NIProperty &nip : nif->properties) {
        uint32_t value = nip.read_proc(base);
        this->DrawString(r, i++, fmt::format("  {:02x}: {} ({})", nip.prop, this->GetPropertyString(nip, value), nip.name));
      }
    }
 
    if (!nif->callbacks.empty()) {
      this->DrawString(r, i++, "Callbacks:");
      for (const NICallback &nic : nif->callbacks) {
        if (!std::holds_alternative<std::monostate>(nic.cb_bit)) {
          uint32_t value = nic.read_proc(base_spec);
 
          struct visitor {
            uint32_t value;
 
            bool operator()(const std::monostate &) { return false; }
            bool operator()(const VehicleCallbackMask &bit) { return static_cast<VehicleCallbackMasks>(this->value).Test(bit); }
            bool operator()(const StationCallbackMask &bit) { return static_cast<StationCallbackMasks>(this->value).Test(bit); }
            bool operator()(const RoadStopCallbackMask &bit) { return static_cast<RoadStopCallbackMasks>(this->value).Test(bit); }
            bool operator()(const HouseCallbackMask &bit) { return static_cast<HouseCallbackMasks>(this->value).Test(bit); }
            bool operator()(const CanalCallbackMask &bit) { return static_cast<CanalCallbackMasks>(this->value).Test(bit); }
            bool operator()(const CargoCallbackMask &bit) { return static_cast<CargoCallbackMasks>(this->value).Test(bit); }
            bool operator()(const IndustryCallbackMask &bit) { return static_cast<IndustryCallbackMasks>(this->value).Test(bit); }
            bool operator()(const IndustryTileCallbackMask &bit) { return static_cast<IndustryTileCallbackMasks>(this->value).Test(bit); }
            bool operator()(const ObjectCallbackMask &bit) { return static_cast<ObjectCallbackMasks>(this->value).Test(bit); }
            bool operator()(const AirportTileCallbackMask &bit) { return static_cast<AirportTileCallbackMasks>(this->value).Test(bit); }
          };
 
          if (!std::visit(visitor{value}, nic.cb_bit)) continue;
          this->DrawString(r, i++, fmt::format("  {:03x}: {}", nic.cb_id, nic.name));
        } else {
          this->DrawString(r, i++, fmt::format("  {:03x}: {} (unmasked)", nic.cb_id, nic.name));
        }
      }
    }
 
    /* Not nice and certainly a hack, but it beats duplicating
     * this whole function just to count the actual number of
     * elements. Especially because they need to be redrawn. */
    const_cast<NewGRFInspectWindow*>(this)->vscroll->SetCount(i);
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    switch (widget) {
      case WID_NGRFI_VEH_CHAIN:
        this->DrawVehicleChainWidget(r);
        break;
 
      case WID_NGRFI_MAINPANEL:
        this->DrawMainPanelWidget(r);
        break;
    }
  }
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    switch (widget) {
      case WID_NGRFI_PARENT: {
        const NIHelper &nih = GetFeatureHelper(this->window_number);
        uint index = nih.GetParent(this->GetFeatureIndex());
        ::ShowNewGRFInspectWindow(GetFeatureNum(index), ::GetFeatureIndex(index), nih.GetGRFID(this->GetFeatureIndex()));
        break;
      }
 
      case WID_NGRFI_VEH_PREV:
        if (this->chain_index > 0) {
          this->chain_index--;
          this->InvalidateData();
        }
        break;
 
      case WID_NGRFI_VEH_NEXT:
        if (this->HasChainIndex()) {
          uint index = this->GetFeatureIndex();
          Vehicle *v = Vehicle::Get(index);
          if (v != nullptr && v->Next() != nullptr) {
            this->chain_index++;
            this->InvalidateData();
          }
        }
        break;
 
      case WID_NGRFI_MAINPANEL: {
        /* Does this feature have variables? */
        const NIFeature *nif  = GetFeature(this->window_number);
        if (nif->variables.empty()) return;
 
        /* Get the line, make sure it's within the boundaries. */
        int32_t line = this->vscroll->GetScrolledRowFromWidget(pt.y, this, WID_NGRFI_MAINPANEL, WidgetDimensions::scaled.frametext.top);
        if (line == INT32_MAX) return;
 
        /* Find the variable related to the line */
        for (const NIVariable &niv : nif->variables) {
          if (--line != 0) continue; // 0 because of the "Variables:" line
 
          if (!HasVariableParameter(niv.var)) break;
 
          this->current_edit_param = niv.var;
          ShowQueryString({}, STR_NEWGRF_INSPECT_QUERY_CAPTION, 9, this, CS_HEXADECIMAL, {});
        }
      }
    }
  }
 
  void OnQueryTextFinished(std::optional<std::string> str) override
  {
    if (!str.has_value()) return;
 
    auto val = ParseInteger<uint32_t>(*str, 16, true);
    if (!val.has_value()) return;
    NewGRFInspectWindow::var60params[GetFeatureNum(this->window_number)][this->current_edit_param - 0x60] = *val;
    this->SetDirty();
  }
 
  void OnResize() override
  {
    this->vscroll->SetCapacityFromWidget(this, WID_NGRFI_MAINPANEL, WidgetDimensions::scaled.frametext.Vertical());
  }

  void OnInvalidateData([[maybe_unused]] int data = 0, [[maybe_unused]] bool gui_scope = true) override
  {
    if (!gui_scope) return;
    if (this->HasChainIndex()) {
      this->ValidateChainIndex();
      this->SetWidgetDisabledState(WID_NGRFI_VEH_PREV, this->chain_index == 0);
      Vehicle *v = Vehicle::Get(this->GetFeatureIndex());
      this->SetWidgetDisabledState(WID_NGRFI_VEH_NEXT, v == nullptr || v->Next() == nullptr);
    }
  }
};
 
/* static */ uint32_t NewGRFInspectWindow::var60params[GSF_FAKE_END][0x20] = { {0} }; // Use spec to have 0s in whole array
 
static constexpr std::initializer_list<NWidgetPart> _nested_newgrf_inspect_chain_widgets = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_NGRFI_CAPTION), SetStringTip(STR_NEWGRF_INSPECT_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
    NWidget(WWT_SHADEBOX, COLOUR_GREY),
    NWidget(WWT_DEFSIZEBOX, COLOUR_GREY),
    NWidget(WWT_STICKYBOX, COLOUR_GREY),
  EndContainer(),
  NWidget(WWT_PANEL, COLOUR_GREY),
    NWidget(NWID_HORIZONTAL),
      NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_NGRFI_VEH_PREV), SetArrowWidgetTypeTip(AWV_DECREASE),
      NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_NGRFI_VEH_NEXT), SetArrowWidgetTypeTip(AWV_INCREASE),
      NWidget(WWT_EMPTY, INVALID_COLOUR, WID_NGRFI_VEH_CHAIN), SetFill(1, 0), SetResize(1, 0),
    EndContainer(),
  EndContainer(),
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_PANEL, COLOUR_GREY, WID_NGRFI_MAINPANEL), SetMinimalSize(300, 0), SetScrollbar(WID_NGRFI_SCROLLBAR), EndContainer(),
    NWidget(NWID_VERTICAL),
      NWidget(NWID_VSCROLLBAR, COLOUR_GREY, WID_NGRFI_SCROLLBAR),
      NWidget(WWT_RESIZEBOX, COLOUR_GREY),
    EndContainer(),
  EndContainer(),
};
 
static constexpr std::initializer_list<NWidgetPart> _nested_newgrf_inspect_widgets = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_NGRFI_CAPTION), SetStringTip(STR_NEWGRF_INSPECT_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
    NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_NGRFI_PARENT), SetStringTip(STR_NEWGRF_INSPECT_PARENT_BUTTON, STR_NEWGRF_INSPECT_PARENT_TOOLTIP),
    NWidget(WWT_SHADEBOX, COLOUR_GREY),
    NWidget(WWT_DEFSIZEBOX, COLOUR_GREY),
    NWidget(WWT_STICKYBOX, COLOUR_GREY),
  EndContainer(),
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_PANEL, COLOUR_GREY, WID_NGRFI_MAINPANEL), SetMinimalSize(300, 0), SetScrollbar(WID_NGRFI_SCROLLBAR), EndContainer(),
    NWidget(NWID_VERTICAL),
      NWidget(NWID_VSCROLLBAR, COLOUR_GREY, WID_NGRFI_SCROLLBAR),
      NWidget(WWT_RESIZEBOX, COLOUR_GREY),
    EndContainer(),
  EndContainer(),
};
 
static WindowDesc _newgrf_inspect_chain_desc(
  WDP_AUTO, "newgrf_inspect_chain", 400, 300,
  WC_NEWGRF_INSPECT, WC_NONE,
  {},
  _nested_newgrf_inspect_chain_widgets
);
 
static WindowDesc _newgrf_inspect_desc(
  WDP_AUTO, "newgrf_inspect", 400, 300,
  WC_NEWGRF_INSPECT, WC_NONE,
  {},
  _nested_newgrf_inspect_widgets
);

void ShowNewGRFInspectWindow(GrfSpecFeature feature, uint index, const uint32_t grfid)
{
  if (!IsNewGRFInspectable(feature, index)) return;
 
  WindowNumber wno = GetInspectWindowNumber(feature, index);
  WindowDesc &desc = (feature == GSF_TRAINS || feature == GSF_ROADVEHICLES) ? _newgrf_inspect_chain_desc : _newgrf_inspect_desc;
  NewGRFInspectWindow *w = AllocateWindowDescFront<NewGRFInspectWindow, true>(desc, wno);
  w->SetCallerGRFID(grfid);
}

void InvalidateNewGRFInspectWindow(GrfSpecFeature feature, uint index)
{
  if (feature == GSF_INVALID) return;
 
  WindowNumber wno = GetInspectWindowNumber(feature, index);
  InvalidateWindowData(WC_NEWGRF_INSPECT, wno);
}

void DeleteNewGRFInspectWindow(GrfSpecFeature feature, uint index)
{
  if (feature == GSF_INVALID) return;
 
  WindowNumber wno = GetInspectWindowNumber(feature, index);
  CloseWindowById(WC_NEWGRF_INSPECT, wno);
 
  /* Reinitialise the land information window to remove the "debug" sprite if needed.
   * Note: Since we might be called from a command here, it is important to not execute
   * the invalidation immediately. The landinfo window tests commands itself. */
  InvalidateWindowData(WC_LAND_INFO, 0, 1);
}

bool IsNewGRFInspectable(GrfSpecFeature feature, uint index)
{
  const NIFeature *nif = GetFeature(GetInspectWindowNumber(feature, index));
  if (nif == nullptr) return false;
  return nif->helper->IsInspectable(index);
}

GrfSpecFeature GetGrfSpecFeature(TileIndex tile)
{
  switch (GetTileType(tile)) {
    default:              return GSF_INVALID;
    case TileType::Railway:      return GSF_RAILTYPES;
    case TileType::Road:         return IsLevelCrossing(tile) ? GSF_RAILTYPES : GSF_ROADTYPES;
    case TileType::House:        return GSF_HOUSES;
    case TileType::Industry:     return GSF_INDUSTRYTILES;
    case TileType::Object:       return GSF_OBJECTS;
 
    case TileType::Station:
      switch (GetStationType(tile)) {
        case StationType::Rail:    return GSF_STATIONS;
        case StationType::Airport: return GSF_AIRPORTTILES;
        case StationType::Bus:     return GSF_ROADSTOPS;
        case StationType::Truck:   return GSF_ROADSTOPS;
        default:              return GSF_INVALID;
      }
  }
}

GrfSpecFeature GetGrfSpecFeature(VehicleType type)
{
  switch (type) {
    case VEH_TRAIN:    return GSF_TRAINS;
    case VEH_ROAD:     return GSF_ROADVEHICLES;
    case VEH_SHIP:     return GSF_SHIPS;
    case VEH_AIRCRAFT: return GSF_AIRCRAFT;
    default:           return GSF_INVALID;
  }
}
 
 
/**** Sprite Aligner ****/

struct SpriteAlignerWindow : Window {
  typedef std::pair<int16_t, int16_t> XyOffs; 
 
  SpriteID current_sprite{}; 
  Scrollbar *vscroll = nullptr;
  std::map<SpriteID, XyOffs> offs_start_map{}; 
 
  static inline ZoomLevel zoom = ZoomLevel::End;
  static bool centre;
  static bool crosshair;
  const Action5Type *act5_type = nullptr; 
 
  SpriteAlignerWindow(WindowDesc &desc, WindowNumber wno) : Window(desc)
  {
    /* On first opening, set initial zoom to current zoom level. */
    if (SpriteAlignerWindow::zoom == ZoomLevel::End) SpriteAlignerWindow::zoom = _gui_zoom;
    SpriteAlignerWindow::zoom = Clamp(SpriteAlignerWindow::zoom, _settings_client.gui.zoom_min, _settings_client.gui.zoom_max);
 
    /* Oh yes, we assume there is at least one normal sprite! */
    while (GetSpriteType(this->current_sprite) != SpriteType::Normal) this->current_sprite++;
    this->SelectAction5Type();
 
    this->CreateNestedTree();
    this->vscroll = this->GetScrollbar(WID_SA_SCROLLBAR);
    this->vscroll->SetCount(_newgrf_debug_sprite_picker.sprites.size());
    this->FinishInitNested(wno);
 
    this->SetWidgetLoweredState(WID_SA_CENTRE, SpriteAlignerWindow::centre);
    this->SetWidgetLoweredState(WID_SA_CROSSHAIR, SpriteAlignerWindow::crosshair);
 
    this->InvalidateData(0, true);
  }
 
  std::string GetWidgetString(WidgetID widget, StringID stringid) const override
  {
    const Sprite *spr = GetSprite(this->current_sprite, SpriteType::Normal);
    switch (widget) {
      case WID_SA_CAPTION:
        if (this->act5_type != nullptr) {
          return GetString(STR_SPRITE_ALIGNER_CAPTION_ACTION5,
            this->act5_type - GetAction5Types().data(),
            this->current_sprite - this->act5_type->sprite_base,
            GetOriginFile(this->current_sprite)->GetSimplifiedFilename(),
            GetSpriteLocalID(this->current_sprite));
        }
        if (this->current_sprite < SPR_OPENTTD_BASE) {
          return GetString(STR_SPRITE_ALIGNER_CAPTION_ACTIONA,
            this->current_sprite,
            GetOriginFile(this->current_sprite)->GetSimplifiedFilename(),
            GetSpriteLocalID(this->current_sprite));
        }
        return GetString(STR_SPRITE_ALIGNER_CAPTION_NO_ACTION,
          GetOriginFile(this->current_sprite)->GetSimplifiedFilename(),
          GetSpriteLocalID(this->current_sprite));
 
      case WID_SA_OFFSETS_ABS:
        return GetString(STR_SPRITE_ALIGNER_OFFSETS_ABS, UnScaleByZoom(spr->x_offs, SpriteAlignerWindow::zoom), UnScaleByZoom(spr->y_offs, SpriteAlignerWindow::zoom));
 
      case WID_SA_OFFSETS_REL: {
        /* Relative offset is new absolute offset - starting absolute offset.
         * Show 0, 0 as the relative offsets if entry is not in the map (meaning they have not been changed yet).
         */
        const auto key_offs_pair = this->offs_start_map.find(this->current_sprite);
        if (key_offs_pair != this->offs_start_map.end()) {
          return GetString(STR_SPRITE_ALIGNER_OFFSETS_REL,
            UnScaleByZoom(spr->x_offs - key_offs_pair->second.first, SpriteAlignerWindow::zoom),
            UnScaleByZoom(spr->y_offs - key_offs_pair->second.second, SpriteAlignerWindow::zoom));
        }
 
        return GetString(STR_SPRITE_ALIGNER_OFFSETS_REL, 0, 0);
      }
 
      default:
        return this->Window::GetWidgetString(widget, stringid);
    }
  }
 
  void UpdateWidgetSize(WidgetID widget, Dimension &size, [[maybe_unused]] const Dimension &padding, [[maybe_unused]] Dimension &fill, [[maybe_unused]] Dimension &resize) override
  {
    switch (widget) {
      case WID_SA_SPRITE:
        size.height = ScaleGUITrad(200);
        break;
 
      case WID_SA_LIST: {
        Dimension d = {};
        for (const auto &spritefile : GetCachedSpriteFiles()) {
          d = maxdim(d, GetStringBoundingBox(GetString(STR_SPRITE_ALIGNER_SPRITE, spritefile->GetSimplifiedFilename(), GetParamMaxDigits(6))));
        }
        size.width = d.width + padding.width;
        fill.height = resize.height = GetCharacterHeight(FS_NORMAL) + padding.height;
        resize.width = 1;
        break;
      }
 
      default:
        break;
    }
  }
 
  void DrawWidget(const Rect &r, WidgetID widget) const override
  {
    switch (widget) {
      case WID_SA_SPRITE: {
        /* Center the sprite ourselves */
        const Sprite *spr = GetSprite(this->current_sprite, SpriteType::Normal);
        Rect ir = r.Shrink(WidgetDimensions::scaled.bevel);
        int x;
        int y;
        if (SpriteAlignerWindow::centre) {
          x = -UnScaleByZoom(spr->x_offs, SpriteAlignerWindow::zoom) + (ir.Width() - UnScaleByZoom(spr->width, SpriteAlignerWindow::zoom)) / 2;
          y = -UnScaleByZoom(spr->y_offs, SpriteAlignerWindow::zoom) + (ir.Height() - UnScaleByZoom(spr->height, SpriteAlignerWindow::zoom)) / 2;
        } else {
          x = ir.Width() / 2;
          y = ir.Height() / 2;
        }
 
        DrawPixelInfo new_dpi;
        if (!FillDrawPixelInfo(&new_dpi, ir)) break;
        AutoRestoreBackup dpi_backup(_cur_dpi, &new_dpi);
 
        DrawSprite(this->current_sprite, PAL_NONE, x, y, nullptr, SpriteAlignerWindow::zoom);
 
        Rect outline = {0, 0, UnScaleByZoom(spr->width, SpriteAlignerWindow::zoom) - 1, UnScaleByZoom(spr->height, SpriteAlignerWindow::zoom) - 1};
        outline = outline.Translate(x + UnScaleByZoom(spr->x_offs, SpriteAlignerWindow::zoom), y + UnScaleByZoom(spr->y_offs, SpriteAlignerWindow::zoom));
        DrawRectOutline(outline.Expand(1), PC_LIGHT_BLUE, 1, 1);
 
        if (SpriteAlignerWindow::crosshair) {
          GfxDrawLine(x, 0, x, ir.Height() - 1, PC_WHITE, 1, 1);
          GfxDrawLine(0, y, ir.Width() - 1, y, PC_WHITE, 1, 1);
        }
        break;
      }
 
      case WID_SA_LIST: {
        /* Don't redraw sprite list while it is still being filled by picker. */
        if (_newgrf_debug_sprite_picker.mode == SPM_REDRAW) break;
 
        const NWidgetBase *nwid = this->GetWidget<NWidgetBase>(widget);
        int step_size = nwid->resize_y;
 
        const FlatSet<SpriteID> &list = _newgrf_debug_sprite_picker.sprites;
 
        Rect ir = r.Shrink(WidgetDimensions::scaled.matrix);
        auto [first, last] = this->vscroll->GetVisibleRangeIterators(list);
        for (auto it = first; it != last; ++it) {
          const SpriteFile *file = GetOriginFile(*it);
          if (file == nullptr) {
            DrawString(ir, GetString(STR_JUST_COMMA, *it), *it == this->current_sprite ? TC_WHITE : (TC_GREY | TC_NO_SHADE), SA_RIGHT | SA_FORCE);
          } else {
            DrawString(ir, GetString(STR_SPRITE_ALIGNER_SPRITE, file->GetSimplifiedFilename(), GetSpriteLocalID(*it)), *it == this->current_sprite ? TC_WHITE : TC_BLACK);
          }
          ir.top += step_size;
        }
        break;
      }
    }
  }
 
  void OnClick([[maybe_unused]] Point pt, WidgetID widget, [[maybe_unused]] int click_count) override
  {
    switch (widget) {
      case WID_SA_PREVIOUS:
        do {
          this->current_sprite = (this->current_sprite == 0 ? GetMaxSpriteID() :  this->current_sprite) - 1;
        } while (GetSpriteType(this->current_sprite) != SpriteType::Normal);
        this->SelectAction5Type();
        this->SetDirty();
        break;
 
      case WID_SA_GOTO:
        ShowQueryString({}, STR_SPRITE_ALIGNER_GOTO_CAPTION, 7, this, CS_NUMERAL, {});
        break;
 
      case WID_SA_NEXT:
        do {
          this->current_sprite = (this->current_sprite + 1) % GetMaxSpriteID();
        } while (GetSpriteType(this->current_sprite) != SpriteType::Normal);
        this->SelectAction5Type();
        this->SetDirty();
        break;
 
      case WID_SA_PICKER:
        this->LowerWidget(WID_SA_PICKER);
        _newgrf_debug_sprite_picker.mode = SPM_WAIT_CLICK;
        this->SetDirty();
        break;
 
      case WID_SA_LIST: {
        auto it = this->vscroll->GetScrolledItemFromWidget(_newgrf_debug_sprite_picker.sprites, pt.y, this, widget);
        if (it != _newgrf_debug_sprite_picker.sprites.end()) {
          SpriteID spr = *it;
          if (GetSpriteType(spr) == SpriteType::Normal) this->current_sprite = spr;
        }
        this->SelectAction5Type();
        this->SetDirty();
        break;
      }
 
      case WID_SA_UP:
      case WID_SA_DOWN:
      case WID_SA_LEFT:
      case WID_SA_RIGHT: {
        /*
         * Yes... this is a hack.
         *
         * No... I don't think it is useful to make this less of a hack.
         *
         * If you want to align sprites, you just need the number. Generally
         * the sprite caches are big enough to not remove the sprite from the
         * cache. If that's not the case, just let the NewGRF developer
         * increase the cache size instead of storing thousands of offsets
         * for the incredibly small chance that it's actually going to be
         * used by someone and the sprite cache isn't big enough for that
         * particular NewGRF developer.
         */
        Sprite *spr = const_cast<Sprite *>(GetSprite(this->current_sprite, SpriteType::Normal));
 
        /* Remember the original offsets of the current sprite, if not already in mapping. */
        if (this->offs_start_map.count(this->current_sprite) == 0) {
          this->offs_start_map[this->current_sprite] = XyOffs(spr->x_offs, spr->y_offs);
        }
        int amt = ScaleByZoom(_ctrl_pressed ? 8 : 1, SpriteAlignerWindow::zoom);
        switch (widget) {
          /* Move eight units at a time if ctrl is pressed. */
          case WID_SA_UP:    spr->y_offs -= amt; break;
          case WID_SA_DOWN:  spr->y_offs += amt; break;
          case WID_SA_LEFT:  spr->x_offs -= amt; break;
          case WID_SA_RIGHT: spr->x_offs += amt; break;
        }
        /* Of course, we need to redraw the sprite, but where is it used?
         * Everywhere is a safe bet. */
        MarkWholeScreenDirty();
        break;
      }
 
      case WID_SA_RESET_REL:
        /* Reset the starting offsets for the current sprite. */
        this->offs_start_map.erase(this->current_sprite);
        this->SetDirty();
        break;
 
      case WID_SA_CENTRE:
        SpriteAlignerWindow::centre = !SpriteAlignerWindow::centre;
        this->SetWidgetLoweredState(widget, SpriteAlignerWindow::centre);
        this->SetDirty();
        break;
 
      case WID_SA_CROSSHAIR:
        SpriteAlignerWindow::crosshair = !SpriteAlignerWindow::crosshair;
        this->SetWidgetLoweredState(widget, SpriteAlignerWindow::crosshair);
        this->SetDirty();
        break;
 
      default:
        if (IsInsideBS(widget, WID_SA_ZOOM, to_underlying(ZoomLevel::End))) {
          SpriteAlignerWindow::zoom = static_cast<ZoomLevel>(widget - WID_SA_ZOOM);
          this->InvalidateData(0, true);
        }
        break;
    }
  }
 
  void OnQueryTextFinished(std::optional<std::string> str) override
  {
    if (!str.has_value()) return;
 
    auto value = ParseInteger(*str, 10, true);
    if (!value.has_value()) return;
    this->current_sprite = *value;
    if (this->current_sprite >= GetMaxSpriteID()) this->current_sprite = 0;
    while (GetSpriteType(this->current_sprite) != SpriteType::Normal) {
      this->current_sprite = (this->current_sprite + 1) % GetMaxSpriteID();
    }
    this->SelectAction5Type();
    this->SetDirty();
  }

  void OnInvalidateData([[maybe_unused]] int data = 0, [[maybe_unused]] bool gui_scope = true) override
  {
    if (!gui_scope) return;
    if (data == 1) {
      /* Sprite picker finished */
      this->RaiseWidget(WID_SA_PICKER);
      this->vscroll->SetCount(_newgrf_debug_sprite_picker.sprites.size());
    }
 
    SpriteAlignerWindow::zoom = Clamp(SpriteAlignerWindow::zoom, _settings_client.gui.zoom_min, _settings_client.gui.zoom_max);
    for (ZoomLevel z = ZoomLevel::Begin; z < ZoomLevel::End; z++) {
      this->SetWidgetsDisabledState(z < _settings_client.gui.zoom_min || z > _settings_client.gui.zoom_max, WID_SA_ZOOM + to_underlying(z));
      this->SetWidgetsLoweredState(SpriteAlignerWindow::zoom == z, WID_SA_ZOOM + to_underlying(z));
    }
  }
 
  void OnResize() override
  {
    this->vscroll->SetCapacityFromWidget(this, WID_SA_LIST);
  }
 
private:
  void SelectAction5Type()
  {
    const auto act5types = GetAction5Types();
    for (auto it = std::begin(act5types); it != std::end(act5types); ++it) {
      if (it->sprite_base <= this->current_sprite && this->current_sprite < it->sprite_base + it->max_sprites) {
        this->act5_type = &*it;
        return;
      }
    }
    this->act5_type = nullptr;
  }
};
 
bool SpriteAlignerWindow::centre = true;
bool SpriteAlignerWindow::crosshair = true;
 
static constexpr std::initializer_list<NWidgetPart> _nested_sprite_aligner_widgets = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_SA_CAPTION),
    NWidget(WWT_SHADEBOX, COLOUR_GREY),
    NWidget(WWT_STICKYBOX, COLOUR_GREY),
  EndContainer(),
  NWidget(WWT_PANEL, COLOUR_GREY),
    NWidget(NWID_HORIZONTAL), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0), SetPadding(WidgetDimensions::unscaled.sparse_resize),
      NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_sparse, 0),
        NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
          NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SA_PREVIOUS), SetStringTip(STR_SPRITE_ALIGNER_PREVIOUS_BUTTON, STR_SPRITE_ALIGNER_PREVIOUS_TOOLTIP), SetFill(1, 0), SetResize(1, 0),
          NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SA_GOTO), SetStringTip(STR_SPRITE_ALIGNER_GOTO_BUTTON, STR_SPRITE_ALIGNER_GOTO_TOOLTIP), SetFill(1, 0), SetResize(1, 0),
          NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SA_NEXT), SetStringTip(STR_SPRITE_ALIGNER_NEXT_BUTTON, STR_SPRITE_ALIGNER_NEXT_TOOLTIP), SetFill(1, 0), SetResize(1, 0),
        EndContainer(),
        NWidget(NWID_HORIZONTAL),
          NWidget(NWID_SPACER), SetFill(1, 1), SetResize(1, 0),
          NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, WID_SA_UP), SetSpriteTip(SPR_ARROW_UP, STR_SPRITE_ALIGNER_MOVE_TOOLTIP), SetResize(0, 0), SetMinimalSize(11, 11),
          NWidget(NWID_SPACER), SetFill(1, 1), SetResize(1, 0),
        EndContainer(),
        NWidget(NWID_HORIZONTAL_LTR), SetPIP(0, WidgetDimensions::unscaled.hsep_wide, 0),
          NWidget(NWID_VERTICAL),
            NWidget(NWID_SPACER), SetFill(1, 1), SetResize(0, 1),
            NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, WID_SA_LEFT), SetSpriteTip(SPR_ARROW_LEFT, STR_SPRITE_ALIGNER_MOVE_TOOLTIP), SetResize(0, 0), SetMinimalSize(11, 11),
            NWidget(NWID_SPACER), SetFill(1, 1), SetResize(0, 1),
          EndContainer(),
          NWidget(WWT_PANEL, COLOUR_DARK_BLUE, WID_SA_SPRITE), SetToolTip(STR_SPRITE_ALIGNER_SPRITE_TOOLTIP), SetResize(1, 1), SetFill(1, 1),
          EndContainer(),
          NWidget(NWID_VERTICAL),
            NWidget(NWID_SPACER), SetFill(1, 1), SetResize(0, 1),
            NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, WID_SA_RIGHT), SetSpriteTip(SPR_ARROW_RIGHT, STR_SPRITE_ALIGNER_MOVE_TOOLTIP), SetResize(0, 0), SetMinimalSize(11, 11),
            NWidget(NWID_SPACER), SetFill(1, 1), SetResize(0, 1),
          EndContainer(),
        EndContainer(),
        NWidget(NWID_HORIZONTAL),
          NWidget(NWID_SPACER), SetFill(1, 1), SetResize(1, 0),
          NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, WID_SA_DOWN), SetSpriteTip(SPR_ARROW_DOWN, STR_SPRITE_ALIGNER_MOVE_TOOLTIP), SetResize(0, 0), SetMinimalSize(11, 11),
          NWidget(NWID_SPACER), SetFill(1, 1), SetResize(1, 0),
        EndContainer(),
        NWidget(WWT_LABEL, INVALID_COLOUR, WID_SA_OFFSETS_ABS), SetFill(1, 0), SetResize(1, 0),
        NWidget(WWT_LABEL, INVALID_COLOUR, WID_SA_OFFSETS_REL), SetFill(1, 0), SetResize(1, 0),
        NWidget(NWID_HORIZONTAL, NWidContainerFlag::EqualSize), SetPIP(0, WidgetDimensions::unscaled.hsep_normal, 0),
          NWidget(WWT_TEXTBTN_2, COLOUR_GREY, WID_SA_CENTRE), SetStringTip(STR_SPRITE_ALIGNER_CENTRE_OFFSET), SetFill(1, 0), SetResize(1, 0),
          NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SA_RESET_REL), SetStringTip(STR_SPRITE_ALIGNER_RESET_BUTTON, STR_SPRITE_ALIGNER_RESET_TOOLTIP), SetFill(1, 0), SetResize(1, 0),
          NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SA_CROSSHAIR), SetStringTip(STR_SPRITE_ALIGNER_CROSSHAIR), SetFill(1, 0), SetResize(1, 0),
        EndContainer(),
      EndContainer(),
      NWidget(NWID_VERTICAL), SetPIP(0, WidgetDimensions::unscaled.vsep_sparse, 0),
        NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SA_PICKER), SetStringTip(STR_SPRITE_ALIGNER_PICKER_BUTTON, STR_SPRITE_ALIGNER_PICKER_TOOLTIP), SetFill(1, 0),
        NWidget(NWID_HORIZONTAL),
          NWidget(WWT_MATRIX, COLOUR_GREY, WID_SA_LIST), SetResize(1, 1), SetMatrixDataTip(1, 0), SetFill(1, 1), SetScrollbar(WID_SA_SCROLLBAR),
          NWidget(NWID_VSCROLLBAR, COLOUR_GREY, WID_SA_SCROLLBAR),
        EndContainer(),
        NWidget(NWID_VERTICAL),
          NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SA_ZOOM + to_underlying(ZoomLevel::In4x)), SetStringTip(STR_CONFIG_SETTING_ZOOM_LVL_MIN), SetFill(1, 0),
          NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SA_ZOOM + to_underlying(ZoomLevel::In2x)), SetStringTip(STR_CONFIG_SETTING_ZOOM_LVL_IN_2X), SetFill(1, 0),
          NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SA_ZOOM + to_underlying(ZoomLevel::Normal)), SetStringTip(STR_CONFIG_SETTING_ZOOM_LVL_NORMAL), SetFill(1, 0),
          NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SA_ZOOM + to_underlying(ZoomLevel::Out2x)), SetStringTip(STR_CONFIG_SETTING_ZOOM_LVL_OUT_2X), SetFill(1, 0),
          NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SA_ZOOM + to_underlying(ZoomLevel::Out4x)), SetStringTip(STR_CONFIG_SETTING_ZOOM_LVL_OUT_4X), SetFill(1, 0),
          NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SA_ZOOM + to_underlying(ZoomLevel::Out8x)), SetStringTip(STR_CONFIG_SETTING_ZOOM_LVL_OUT_8X), SetFill(1, 0),
        EndContainer(),
      EndContainer(),
    EndContainer(),
    NWidget(NWID_HORIZONTAL),
      NWidget(NWID_SPACER), SetFill(1, 0), SetResize(1, 0),
      NWidget(WWT_RESIZEBOX, COLOUR_GREY), SetResizeWidgetTypeTip(RWV_HIDE_BEVEL, STR_TOOLTIP_RESIZE),
    EndContainer(),
  EndContainer(),
};
 
static WindowDesc _sprite_aligner_desc(
  WDP_AUTO, "sprite_aligner", 400, 300,
  WC_SPRITE_ALIGNER, WC_NONE,
  {},
  _nested_sprite_aligner_widgets
);

void ShowSpriteAlignerWindow()
{
  AllocateWindowDescFront<SpriteAlignerWindow>(_sprite_aligner_desc, 0);
}