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 <http://www.gnu.org/licenses/>.
 */
 
#include "stdafx.h"
#include <stdarg.h>
#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 "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_airporttiles.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, std::vector<SpriteID>() };
 
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;
}
 
enum NIType {
  NIT_INT,   
  NIT_CARGO, 
};
 
typedef const void *NIOffsetProc(const void *b);
 
struct NIProperty {
  const char *name;          
  NIOffsetProc *offset_proc; 
  byte read_size;            
  byte prop;                 
  byte type;
};
 
 
struct NICallback {
  const char *name;          
  NIOffsetProc *offset_proc; 
  byte read_size;            
  byte cb_bit;               
  uint16 cb_id;              
};
static const int CBM_NO_BIT = UINT8_MAX;
 
struct NIVariable {
  const char *name;
  byte var;
};
 
class NIHelper {
public:
  virtual ~NIHelper() {}
 
  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 void SetStringParameters(uint index) const = 0;
 
  virtual uint32 GetGRFID(uint index) const = 0;
 
  virtual uint Resolve(uint index, uint var, uint param, bool *avail) const = 0;
 
  virtual bool PSAWithParameter() const
  {
    return false;
  }
 
  virtual uint GetPSASize(uint index, uint32 grfid) const
  {
    return 0;
  }
 
  virtual const int32 *GetPSAFirstPosition(uint index, uint32 grfid) const
  {
    return nullptr;
  }
 
protected:
  void SetSimpleStringParameters(StringID string, uint32 index) const
  {
    SetDParam(0, string);
    SetDParam(1, index);
  }
 
 
  void SetObjectAtStringParameters(StringID string, uint32 index, TileIndex tile) const
  {
    SetDParam(0, STR_NEWGRF_INSPECT_CAPTION_OBJECT_AT);
    SetDParam(1, string);
    SetDParam(2, index);
    SetDParam(3, tile);
  }
};
 
 
struct NIFeature {
  const NIProperty *properties; 
  const NICallback *callbacks;  
  const NIVariable *variables;  
  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 var60params[GSF_FAKE_END][0x20];
 
  uint32 caller_grfid;
 
  uint chain_index;
 
  byte current_edit_param;
 
  Scrollbar *vscroll;
 
  static bool HasVariableParameter(uint variable)
  {
    return IsInsideBS(variable, 0x60, 0x20);
  }
 
  void SetCallerGRFID(uint32 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;
    }
    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);
  }
 
  void SetStringParameters(int widget) const override
  {
    if (widget != WID_NGRFI_CAPTION) return;
 
    GetFeatureHelper(this->window_number)->SetStringParameters(this->GetFeatureIndex());
  }
 
  void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, 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:
        resize->height = std::max(11, FONT_HEIGHT_NORMAL + WidgetDimensions::scaled.vsep_normal);
        resize->width  = 1;
 
        size->height = 5 * resize->height + WidgetDimensions::scaled.frametext.Vertical();
        break;
    }
  }
 
  void WARN_FORMAT(4, 5) DrawString(const Rect &r, int offset, const char *format, ...) const
  {
    char buf[1024];
 
    va_list va;
    va_start(va, format);
    vseprintf(buf, lastof(buf), format, va);
    va_end(va);
 
    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), buf, TC_BLACK);
  }
 
  void DrawWidget(const Rect &r, int widget) const override
  {
    switch (widget) {
      case WID_NGRFI_VEH_CHAIN: {
        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 = CenterBounds(br.top, br.bottom, h);
        DrawVehicleImage(v->First(), br, INVALID_VEHICLE, 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, FR_BORDERONLY);
        } else {
          DrawFrameRect(r.left  + sel_start - skip, y, r.left  + sel_end   - skip, y + h, COLOUR_WHITE, FR_BORDERONLY);
        }
        break;
      }
    }
 
    if (widget != WID_NGRFI_MAINPANEL) return;
 
    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 != nullptr) {
      this->DrawString(r, i++, "Variables:");
      for (const NIVariable *niv = nif->variables; niv->name != nullptr; niv++) {
        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++, "  %02x[%02x]: %08x (%s)", niv->var, param, value, niv->name);
        } else {
          this->DrawString(r, i++, "  %02x: %08x (%s)", niv->var, value, niv->name);
        }
      }
    }
 
    uint psa_size = nih->GetPSASize(index, this->caller_grfid);
    const int32 *psa = nih->GetPSAFirstPosition(index, this->caller_grfid);
    if (psa_size != 0 && psa != nullptr) {
      if (nih->PSAWithParameter()) {
        this->DrawString(r, i++, "Persistent storage [%08X]:", BSWAP32(this->caller_grfid));
      } else {
        this->DrawString(r, i++, "Persistent storage:");
      }
      assert(psa_size % 4 == 0);
      for (uint j = 0; j < psa_size; j += 4, psa += 4) {
        this->DrawString(r, i++, "  %i: %i %i %i %i", j, psa[0], psa[1], psa[2], psa[3]);
      }
    }
 
    if (nif->properties != nullptr) {
      this->DrawString(r, i++, "Properties:");
      for (const NIProperty *nip = nif->properties; nip->name != nullptr; nip++) {
        const void *ptr = nip->offset_proc(base);
        uint value;
        switch (nip->read_size) {
          case 1: value = *(const uint8  *)ptr; break;
          case 2: value = *(const uint16 *)ptr; break;
          case 4: value = *(const uint32 *)ptr; break;
          default: NOT_REACHED();
        }
 
        StringID string;
        SetDParam(0, value);
        switch (nip->type) {
          case NIT_INT:
            string = STR_JUST_INT;
            break;
 
          case NIT_CARGO:
            string = value != INVALID_CARGO ? CargoSpec::Get(value)->name : STR_QUANTITY_N_A;
            break;
 
          default:
            NOT_REACHED();
        }
 
        char buffer[64];
        GetString(buffer, string, lastof(buffer));
        this->DrawString(r, i++, "  %02x: %s (%s)", nip->prop, buffer, nip->name);
      }
    }
 
    if (nif->callbacks != nullptr) {
      this->DrawString(r, i++, "Callbacks:");
      for (const NICallback *nic = nif->callbacks; nic->name != nullptr; nic++) {
        if (nic->cb_bit != CBM_NO_BIT) {
          const void *ptr = nic->offset_proc(base_spec);
          uint value;
          switch (nic->read_size) {
            case 1: value = *(const uint8  *)ptr; break;
            case 2: value = *(const uint16 *)ptr; break;
            case 4: value = *(const uint32 *)ptr; break;
            default: NOT_REACHED();
          }
 
          if (!HasBit(value, nic->cb_bit)) continue;
          this->DrawString(r, i++, "  %03x: %s", nic->cb_id, nic->name);
        } else {
          this->DrawString(r, i++, "  %03x: %s (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 OnClick(Point pt, int widget, 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 == nullptr) return;
 
        /* Get the line, make sure it's within the boundaries. */
        int line = this->vscroll->GetScrolledRowFromWidget(pt.y, this, WID_NGRFI_MAINPANEL, WidgetDimensions::scaled.frametext.top);
        if (line == INT_MAX) return;
 
        /* Find the variable related to the line */
        for (const NIVariable *niv = nif->variables; niv->name != nullptr; niv++, line--) {
          if (line != 1) continue; // 1 because of the "Variables:" line
 
          if (!HasVariableParameter(niv->var)) break;
 
          this->current_edit_param = niv->var;
          ShowQueryString(STR_EMPTY, STR_NEWGRF_INSPECT_QUERY_CAPTION, 9, this, CS_HEXADECIMAL, QSF_NONE);
        }
      }
    }
  }
 
  void OnQueryTextFinished(char *str) override
  {
    if (StrEmpty(str)) return;
 
    NewGRFInspectWindow::var60params[GetFeatureNum(this->window_number)][this->current_edit_param - 0x60] = strtol(str, nullptr, 16);
    this->SetDirty();
  }
 
  void OnResize() override
  {
    this->vscroll->SetCapacityFromWidget(this, WID_NGRFI_MAINPANEL, WidgetDimensions::scaled.frametext.Vertical());
  }
 
  void OnInvalidateData(int data = 0, 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 NewGRFInspectWindow::var60params[GSF_FAKE_END][0x20] = { {0} }; // Use spec to have 0s in whole array
 
static const NWidgetPart _nested_newgrf_inspect_chain_widgets[] = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_NGRFI_CAPTION), SetDataTip(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), SetDataTip(AWV_DECREASE, STR_NULL),
      NWidget(WWT_PUSHARROWBTN, COLOUR_GREY, WID_NGRFI_VEH_NEXT), SetDataTip(AWV_INCREASE, STR_NULL),
      NWidget(WWT_EMPTY, COLOUR_GREY, 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 const NWidgetPart _nested_newgrf_inspect_widgets[] = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_NGRFI_CAPTION), SetDataTip(STR_NEWGRF_INSPECT_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
    NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_NGRFI_PARENT), SetDataTip(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,
  0,
  _nested_newgrf_inspect_chain_widgets, lengthof(_nested_newgrf_inspect_chain_widgets)
);
 
static WindowDesc _newgrf_inspect_desc(
  WDP_AUTO, "newgrf_inspect", 400, 300,
  WC_NEWGRF_INSPECT, WC_NONE,
  0,
  _nested_newgrf_inspect_widgets, lengthof(_nested_newgrf_inspect_widgets)
);
 
void ShowNewGRFInspectWindow(GrfSpecFeature feature, uint index, const uint32 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>(desc, wno, true);
  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 MP_RAILWAY:      return GSF_RAILTYPES;
    case MP_ROAD:         return IsLevelCrossing(tile) ? GSF_RAILTYPES : GSF_INVALID;
    case MP_HOUSE:        return GSF_HOUSES;
    case MP_INDUSTRY:     return GSF_INDUSTRYTILES;
    case MP_OBJECT:       return GSF_OBJECTS;
 
    case MP_STATION:
      switch (GetStationType(tile)) {
        case STATION_RAIL:    return GSF_STATIONS;
        case STATION_AIRPORT: return GSF_AIRPORTTILES;
        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, int16> XyOffs;    
 
  SpriteID current_sprite;                   
  Scrollbar *vscroll;
  SmallMap<SpriteID, XyOffs> offs_start_map; 
 
  static bool centre;
  static bool crosshair;
 
  SpriteAlignerWindow(WindowDesc *desc, WindowNumber wno) : Window(desc)
  {
    this->CreateNestedTree();
    this->vscroll = this->GetScrollbar(WID_SA_SCROLLBAR);
    this->FinishInitNested(wno);
 
    this->SetWidgetLoweredState(WID_SA_CENTRE, SpriteAlignerWindow::centre);
    this->SetWidgetLoweredState(WID_SA_CROSSHAIR, SpriteAlignerWindow::crosshair);
 
    /* Oh yes, we assume there is at least one normal sprite! */
    while (GetSpriteType(this->current_sprite) != ST_NORMAL) this->current_sprite++;
  }
 
  void SetStringParameters(int widget) const override
  {
    const Sprite *spr = GetSprite(this->current_sprite, ST_NORMAL);
    switch (widget) {
      case WID_SA_CAPTION:
        SetDParam(0, this->current_sprite);
        SetDParamStr(1, GetOriginFile(this->current_sprite)->GetSimplifiedFilename());
        break;
 
      case WID_SA_OFFSETS_ABS:
        SetDParam(0, spr->x_offs);
        SetDParam(1, spr->y_offs);
        break;
 
      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()) {
          SetDParam(0, spr->x_offs - key_offs_pair->second.first);
          SetDParam(1, spr->y_offs - key_offs_pair->second.second);
        } else {
          SetDParam(0, 0);
          SetDParam(1, 0);
        }
        break;
      }
 
      default:
        break;
    }
  }
 
  void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize) override
  {
    switch (widget) {
      case WID_SA_SPRITE:
        size->height = ScaleGUITrad(200);
        break;
      case WID_SA_LIST:
        SetDParamMaxDigits(0, 6);
        size->width = GetStringBoundingBox(STR_BLACK_COMMA).width + padding.width;
        resize->height = FONT_HEIGHT_NORMAL + padding.height;
        resize->width  = 1;
        fill->height = resize->height;
        break;
      default:
        break;
    }
  }
 
  void DrawWidget(const Rect &r, int widget) const override
  {
    switch (widget) {
      case WID_SA_SPRITE: {
        /* Center the sprite ourselves */
        const Sprite *spr = GetSprite(this->current_sprite, ST_NORMAL);
        Rect ir = r.Shrink(WidgetDimensions::scaled.bevel);
        int x;
        int y;
        if (SpriteAlignerWindow::centre) {
          x = -UnScaleGUI(spr->x_offs) + (ir.Width() - UnScaleGUI(spr->width)) / 2;
          y = -UnScaleGUI(spr->y_offs) + (ir.Height() - UnScaleGUI(spr->height)) / 2;
        } else {
          x = ir.Width() / 2;
          y = ir.Height() / 2;
        }
 
        DrawPixelInfo new_dpi;
        if (!FillDrawPixelInfo(&new_dpi, ir.left, ir.top, ir.Width(), ir.Height())) break;
        DrawPixelInfo *old_dpi = _cur_dpi;
        _cur_dpi = &new_dpi;
 
        DrawSprite(this->current_sprite, PAL_NONE, x, y, nullptr, ZOOM_LVL_GUI);
        if (this->crosshair) {
          GfxDrawLine(x, 0, x, ir.Height() - 1, PC_WHITE, 1, 1);
          GfxDrawLine(0, y, ir.Width() - 1, y, PC_WHITE, 1, 1);
        }
 
        _cur_dpi = old_dpi;
 
        break;
      }
 
      case WID_SA_LIST: {
        const NWidgetBase *nwid = this->GetWidget<NWidgetBase>(widget);
        int step_size = nwid->resize_y;
 
        std::vector<SpriteID> &list = _newgrf_debug_sprite_picker.sprites;
        int max = std::min<int>(this->vscroll->GetPosition() + this->vscroll->GetCapacity(), (uint)list.size());
 
        Rect ir = r.Shrink(WidgetDimensions::scaled.matrix);
        for (int i = this->vscroll->GetPosition(); i < max; i++) {
          SetDParam(0, list[i]);
          DrawString(ir, STR_BLACK_COMMA, TC_FROMSTRING, SA_RIGHT | SA_FORCE);
          ir.top += step_size;
        }
        break;
      }
    }
  }
 
  void OnClick(Point pt, int widget, 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) != ST_NORMAL);
        this->SetDirty();
        break;
 
      case WID_SA_GOTO:
        ShowQueryString(STR_EMPTY, STR_SPRITE_ALIGNER_GOTO_CAPTION, 7, this, CS_NUMERAL, QSF_NONE);
        break;
 
      case WID_SA_NEXT:
        do {
          this->current_sprite = (this->current_sprite + 1) % GetMaxSpriteID();
        } while (GetSpriteType(this->current_sprite) != ST_NORMAL);
        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: {
        const NWidgetBase *nwid = this->GetWidget<NWidgetBase>(widget);
        int step_size = nwid->resize_y;
 
        uint i = this->vscroll->GetPosition() + (pt.y - nwid->pos_y) / step_size;
        if (i < _newgrf_debug_sprite_picker.sprites.size()) {
          SpriteID spr = _newgrf_debug_sprite_picker.sprites[i];
          if (GetSpriteType(spr) == ST_NORMAL) this->current_sprite = spr;
        }
        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, ST_NORMAL));
 
        /* Remember the original offsets of the current sprite, if not already in mapping. */
        if (!(this->offs_start_map.Contains(this->current_sprite))) {
          this->offs_start_map.Insert(this->current_sprite, XyOffs(spr->x_offs, spr->y_offs));
        }
        switch (widget) {
          /* Move eight units at a time if ctrl is pressed. */
          case WID_SA_UP:    spr->y_offs -= _ctrl_pressed ? 8 : 1; break;
          case WID_SA_DOWN:  spr->y_offs += _ctrl_pressed ? 8 : 1; break;
          case WID_SA_LEFT:  spr->x_offs -= _ctrl_pressed ? 8 : 1; break;
          case WID_SA_RIGHT: spr->x_offs += _ctrl_pressed ? 8 : 1; 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;
    }
  }
 
  void OnQueryTextFinished(char *str) override
  {
    if (StrEmpty(str)) return;
 
    this->current_sprite = atoi(str);
    if (this->current_sprite >= GetMaxSpriteID()) this->current_sprite = 0;
    while (GetSpriteType(this->current_sprite) != ST_NORMAL) {
      this->current_sprite = (this->current_sprite + 1) % GetMaxSpriteID();
    }
    this->SetDirty();
  }
 
  void OnInvalidateData(int data = 0, bool gui_scope = true) override
  {
    if (!gui_scope) return;
    if (data == 1) {
      /* Sprite picker finished */
      this->RaiseWidget(WID_SA_PICKER);
      this->vscroll->SetCount((uint)_newgrf_debug_sprite_picker.sprites.size());
    }
  }
 
  void OnResize() override
  {
    this->vscroll->SetCapacityFromWidget(this, WID_SA_LIST);
  }
};
 
bool SpriteAlignerWindow::centre = true;
bool SpriteAlignerWindow::crosshair = true;
 
static const NWidgetPart _nested_sprite_aligner_widgets[] = {
  NWidget(NWID_HORIZONTAL),
    NWidget(WWT_CLOSEBOX, COLOUR_GREY),
    NWidget(WWT_CAPTION, COLOUR_GREY, WID_SA_CAPTION), SetDataTip(STR_SPRITE_ALIGNER_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
    NWidget(WWT_SHADEBOX, COLOUR_GREY),
    NWidget(WWT_STICKYBOX, COLOUR_GREY),
  EndContainer(),
  NWidget(WWT_PANEL, COLOUR_GREY),
    NWidget(NWID_HORIZONTAL), SetPIP(0, 0, 10),
      NWidget(NWID_VERTICAL), SetPIP(10, 5, 10),
        NWidget(NWID_HORIZONTAL, NC_EQUALSIZE), SetPIP(10, 5, 10),
          NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SA_PREVIOUS), SetDataTip(STR_SPRITE_ALIGNER_PREVIOUS_BUTTON, STR_SPRITE_ALIGNER_PREVIOUS_TOOLTIP), SetFill(1, 0),
          NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SA_GOTO), SetDataTip(STR_SPRITE_ALIGNER_GOTO_BUTTON, STR_SPRITE_ALIGNER_GOTO_TOOLTIP), SetFill(1, 0),
          NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SA_NEXT), SetDataTip(STR_SPRITE_ALIGNER_NEXT_BUTTON, STR_SPRITE_ALIGNER_NEXT_TOOLTIP), SetFill(1, 0),
        EndContainer(),
        NWidget(NWID_HORIZONTAL), SetPIP(10, 5, 10),
          NWidget(NWID_SPACER), SetFill(1, 1),
          NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, WID_SA_UP), SetDataTip(SPR_ARROW_UP, STR_SPRITE_ALIGNER_MOVE_TOOLTIP), SetResize(0, 0), SetMinimalSize(11, 11),
          NWidget(NWID_SPACER), SetFill(1, 1),
        EndContainer(),
        NWidget(NWID_HORIZONTAL_LTR), SetPIP(10, 5, 10),
          NWidget(NWID_VERTICAL),
            NWidget(NWID_SPACER), SetFill(1, 1),
            NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, WID_SA_LEFT), SetDataTip(SPR_ARROW_LEFT, STR_SPRITE_ALIGNER_MOVE_TOOLTIP), SetResize(0, 0), SetMinimalSize(11, 11),
            NWidget(NWID_SPACER), SetFill(1, 1),
          EndContainer(),
          NWidget(WWT_PANEL, COLOUR_DARK_BLUE, WID_SA_SPRITE), SetDataTip(STR_NULL, STR_SPRITE_ALIGNER_SPRITE_TOOLTIP),
          EndContainer(),
          NWidget(NWID_VERTICAL),
            NWidget(NWID_SPACER), SetFill(1, 1),
            NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, WID_SA_RIGHT), SetDataTip(SPR_ARROW_RIGHT, STR_SPRITE_ALIGNER_MOVE_TOOLTIP), SetResize(0, 0), SetMinimalSize(11, 11),
            NWidget(NWID_SPACER), SetFill(1, 1),
          EndContainer(),
        EndContainer(),
        NWidget(NWID_HORIZONTAL), SetPIP(10, 5, 10),
          NWidget(NWID_SPACER), SetFill(1, 1),
          NWidget(WWT_PUSHIMGBTN, COLOUR_GREY, WID_SA_DOWN), SetDataTip(SPR_ARROW_DOWN, STR_SPRITE_ALIGNER_MOVE_TOOLTIP), SetResize(0, 0), SetMinimalSize(11, 11),
          NWidget(NWID_SPACER), SetFill(1, 1),
        EndContainer(),
        NWidget(WWT_LABEL, COLOUR_GREY, WID_SA_OFFSETS_ABS), SetDataTip(STR_SPRITE_ALIGNER_OFFSETS_ABS, STR_NULL), SetFill(1, 0), SetPadding(0, 10, 0, 10),
        NWidget(WWT_LABEL, COLOUR_GREY, WID_SA_OFFSETS_REL), SetDataTip(STR_SPRITE_ALIGNER_OFFSETS_REL, STR_NULL), SetFill(1, 0), SetPadding(0, 10, 0, 10),
        NWidget(NWID_HORIZONTAL, NC_EQUALSIZE), SetPIP(10, 5, 10),
          NWidget(WWT_TEXTBTN_2, COLOUR_GREY, WID_SA_CENTRE), SetDataTip(STR_SPRITE_ALIGNER_CENTRE_OFFSET, STR_NULL), SetFill(1, 0),
          NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_SA_RESET_REL), SetDataTip(STR_SPRITE_ALIGNER_RESET_BUTTON, STR_SPRITE_ALIGNER_RESET_TOOLTIP), SetFill(1, 0),
          NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SA_CROSSHAIR), SetDataTip(STR_SPRITE_ALIGNER_CROSSHAIR, STR_NULL), SetFill(1, 0),
        EndContainer(),
      EndContainer(),
      NWidget(NWID_VERTICAL), SetPIP(10, 5, 10),
        NWidget(WWT_TEXTBTN, COLOUR_GREY, WID_SA_PICKER), SetDataTip(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, STR_NULL), SetFill(1, 1), SetScrollbar(WID_SA_SCROLLBAR),
          NWidget(NWID_VSCROLLBAR, COLOUR_GREY, WID_SA_SCROLLBAR),
        EndContainer(),
      EndContainer(),
    EndContainer(),
  EndContainer(),
};
 
static WindowDesc _sprite_aligner_desc(
  WDP_AUTO, "sprite_aligner", 400, 300,
  WC_SPRITE_ALIGNER, WC_NONE,
  0,
  _nested_sprite_aligner_widgets, lengthof(_nested_sprite_aligner_widgets)
);
 
void ShowSpriteAlignerWindow()
{
  AllocateWindowDescFront<SpriteAlignerWindow>(&_sprite_aligner_desc, 0);
}