widget_type.h

Go to the documentation of this file.

/*
 * This file is part of OpenTTD.
 * OpenTTD is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation, version 2.
 * OpenTTD is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with OpenTTD. If not, see <http://www.gnu.org/licenses/>.
 */
 
#ifndef WIDGET_TYPE_H
#define WIDGET_TYPE_H
 
#include "core/alloc_type.hpp"
#include "core/bitmath_func.hpp"
#include "core/math_func.hpp"
#include "strings_type.h"
#include "gfx_type.h"
#include "window_type.h"
 
/* Number of column bits of the WWT_MATRIX widget data. */
static constexpr uint8_t MAT_COL_START = 0; 
static constexpr uint8_t MAT_COL_BITS = 8; 
 
/* Number of row bits of the WWT_MATRIX widget data. */
static constexpr uint8_t MAT_ROW_START = 8; 
static constexpr uint8_t MAT_ROW_BITS = 8; 
 
enum ArrowWidgetValues {
  AWV_DECREASE, 
  AWV_INCREASE, 
  AWV_LEFT,     
  AWV_RIGHT,    
};
 
enum ResizeWidgetValues {
  RWV_SHOW_BEVEL, 
  RWV_HIDE_BEVEL, 
};
 
enum WidgetType {
  /* Window widget types. */
  WWT_EMPTY,      
 
  WWT_PANEL,      
  WWT_INSET,      
  WWT_IMGBTN,     
  WWT_IMGBTN_2,   
  WWT_ARROWBTN,   
  WWT_TEXTBTN,    
  WWT_TEXTBTN_2,  
  WWT_LABEL,      
  WWT_TEXT,       
  WWT_MATRIX,     
  WWT_FRAME,      
  WWT_CAPTION,    
 
  WWT_DEBUGBOX,   
  WWT_SHADEBOX,   
  WWT_DEFSIZEBOX, 
  WWT_STICKYBOX,  
 
  WWT_RESIZEBOX,  
  WWT_CLOSEBOX,   
  WWT_DROPDOWN,   
  WWT_EDITBOX,    
  WWT_LAST,       
 
  /* Nested widget types. */
  NWID_HORIZONTAL,      
  NWID_HORIZONTAL_LTR,  
  NWID_VERTICAL,        
  NWID_MATRIX,          
  NWID_SPACER,          
  NWID_SELECTION,       
  NWID_VIEWPORT,        
  NWID_BUTTON_DROPDOWN, 
  NWID_HSCROLLBAR,      
  NWID_VSCROLLBAR,      
  NWID_CUSTOM,          
 
  /* Nested widget part types. */
  WPT_ATTRIBUTE_BEGIN, 
  WPT_RESIZE,       
  WPT_MINSIZE,      
  WPT_MINTEXTLINES, 
  WPT_FILL,         
  WPT_DATATIP,      
  WPT_PADDING,      
  WPT_PIPSPACE,     
  WPT_PIPRATIO,     
  WPT_TEXTSTYLE,    
  WPT_ALIGNMENT,    
  WPT_SCROLLBAR,    
  WPT_ASPECT,       
  WPT_ATTRIBUTE_END, 
 
  WPT_FUNCTION, 
  WPT_ENDCONTAINER, 
 
  /* Pushable window widget types. */
  WWT_MASK = 0x7F,
 
  WWB_PUSHBUTTON    = 1 << 7,
 
  WWT_PUSHBTN       = WWT_PANEL    | WWB_PUSHBUTTON,    
  WWT_PUSHTXTBTN    = WWT_TEXTBTN  | WWB_PUSHBUTTON,    
  WWT_PUSHIMGBTN    = WWT_IMGBTN   | WWB_PUSHBUTTON,    
  WWT_PUSHARROWBTN  = WWT_ARROWBTN | WWB_PUSHBUTTON,    
  NWID_PUSHBUTTON_DROPDOWN = NWID_BUTTON_DROPDOWN | WWB_PUSHBUTTON,
};
 
enum SizingType {
  ST_SMALLEST, 
  ST_RESIZE,   
};
 
enum class AspectFlags : uint8_t {
  ResizeX = 1U << 0,
  ResizeY = 1U << 1,
  ResizeXY = ResizeX | ResizeY,
};
DECLARE_ENUM_AS_BIT_SET(AspectFlags)
 
/* Forward declarations. */
class NWidgetCore;
class Scrollbar;
 
using WidgetLookup = std::map<WidgetID, class NWidgetBase *>;
 
class NWidgetBase : public ZeroedMemoryAllocator {
public:
  NWidgetBase(WidgetType tp);
 
  void ApplyAspectRatio();
  virtual void AdjustPaddingForZoom();
  virtual void SetupSmallestSize(Window *w) = 0;
  virtual void AssignSizePosition(SizingType sizing, int x, int y, uint given_width, uint given_height, bool rtl) = 0;
 
  virtual void FillWidgetLookup(WidgetLookup &widget_lookup) = 0;
 
  virtual NWidgetCore *GetWidgetFromPos(int x, int y) = 0;
  virtual NWidgetBase *GetWidgetOfType(WidgetType tp);
 
  template <class NWID>
  NWID *GetParentWidget()
  {
    for (NWidgetBase *nwid_parent = this->parent; nwid_parent != nullptr; nwid_parent = nwid_parent->parent) {
      if (NWID *nwid = dynamic_cast<NWID *>(nwid_parent); nwid != nullptr) return nwid;
    }
    return nullptr;
  }
 
  template <class NWID>
  const NWID *GetParentWidget() const
  {
    for (const NWidgetBase *nwid_parent = this->parent; nwid_parent != nullptr; nwid_parent = nwid_parent->parent) {
      if (const NWID *nwid = dynamic_cast<const NWID *>(nwid_parent); nwid != nullptr) return nwid;
    }
    return nullptr;
  }
 
  virtual bool IsHighlighted() const { return false; }
  virtual TextColour GetHighlightColour() const { return TC_INVALID; }
  virtual void SetHighlighted([[maybe_unused]] TextColour highlight_colour) {}
 
  inline void SetPadding(uint8_t top, uint8_t right, uint8_t bottom, uint8_t left)
  {
    this->uz_padding.top = top;
    this->uz_padding.right = right;
    this->uz_padding.bottom = bottom;
    this->uz_padding.left = left;
    this->AdjustPaddingForZoom();
  }
 
  inline void SetPadding(const RectPadding &padding)
  {
    this->uz_padding = padding;
    this->AdjustPaddingForZoom();
  }
 
  inline uint GetHorizontalStepSize(SizingType sizing) const;
  inline uint GetVerticalStepSize(SizingType sizing) const;
 
  virtual void Draw(const Window *w) = 0;
  virtual void SetDirty(const Window *w) const;
 
  Rect GetCurrentRect() const
  {
    Rect r;
    r.left = this->pos_x;
    r.top = this->pos_y;
    r.right = this->pos_x + this->current_x - 1;
    r.bottom = this->pos_y + this->current_y - 1;
    return r;
  }
 
  WidgetType type;      
  uint fill_x;          
  uint fill_y;          
  uint resize_x;        
  uint resize_y;        
  /* Size of the widget in the smallest window possible.
   * Computed by #SetupSmallestSize() followed by #AssignSizePosition().
   */
  uint smallest_x;      
  uint smallest_y;      
  /* Current widget size (that is, after resizing). */
  uint current_x;       
  uint current_y;       
  float aspect_ratio = 0; 
  AspectFlags aspect_flags = AspectFlags::ResizeX; 
 
  int pos_x;            
  int pos_y;            
 
  RectPadding padding;    
  RectPadding uz_padding; 
 
  NWidgetBase *parent; 
 
protected:
  inline void StoreSizePosition(SizingType sizing, int x, int y, uint given_width, uint given_height);
};
 
inline uint NWidgetBase::GetHorizontalStepSize(SizingType sizing) const
{
  return (sizing == ST_RESIZE) ? this->resize_x : this->fill_x;
}
 
inline uint NWidgetBase::GetVerticalStepSize(SizingType sizing) const
{
  return (sizing == ST_RESIZE) ? this->resize_y : this->fill_y;
}
 
inline void NWidgetBase::StoreSizePosition(SizingType sizing, int x, int y, uint given_width, uint given_height)
{
  this->pos_x = x;
  this->pos_y = y;
  if (sizing == ST_SMALLEST) {
    this->smallest_x = given_width;
    this->smallest_y = given_height;
  }
  this->current_x = given_width;
  this->current_y = given_height;
}
 
 
class NWidgetResizeBase : public NWidgetBase {
public:
  NWidgetResizeBase(WidgetType tp, uint fill_x, uint fill_y);
 
  void AdjustPaddingForZoom() override;
  void SetMinimalSize(uint min_x, uint min_y);
  void SetMinimalSizeAbsolute(uint min_x, uint min_y);
  void SetMinimalTextLines(uint8_t min_lines, uint8_t spacing, FontSize size);
  void SetFill(uint fill_x, uint fill_y);
  void SetResize(uint resize_x, uint resize_y);
  void SetAspect(float ratio, AspectFlags flags = AspectFlags::ResizeX);
  void SetAspect(int x_ratio, int y_ratio, AspectFlags flags = AspectFlags::ResizeX);
 
  bool UpdateMultilineWidgetSize(const std::string &str, int max_lines);
  bool UpdateSize(uint min_x, uint min_y);
  bool UpdateVerticalSize(uint min_y);
 
  void AssignSizePosition(SizingType sizing, int x, int y, uint given_width, uint given_height, bool rtl) override;
 
  uint min_x; 
  uint min_y; 
 
  bool absolute; 
  uint uz_min_x; 
  uint uz_min_y; 
 
  uint8_t uz_text_lines;   
  uint8_t uz_text_spacing; 
  FontSize uz_text_size; 
};
 
enum NWidgetDisplay {
  /* Generic. */
  NDB_LOWERED         = 0, 
  NDB_DISABLED        = 1, 
  /* Viewport widget. */
  NDB_NO_TRANSPARENCY = 2, 
  NDB_SHADE_GREY      = 3, 
  NDB_SHADE_DIMMED    = 4, 
  /* Button dropdown widget. */
  NDB_DROPDOWN_ACTIVE = 5, 
  /* Scrollbar widget. */
  NDB_SCROLLBAR_UP    = 6, 
  NDB_SCROLLBAR_DOWN  = 7, 
  /* Generic. */
  NDB_HIGHLIGHT       = 8, 
  NDB_DROPDOWN_CLOSED = 9, 
 
  ND_LOWERED  = 1 << NDB_LOWERED,                
  ND_DISABLED = 1 << NDB_DISABLED,               
  ND_HIGHLIGHT = 1 << NDB_HIGHLIGHT,             
  ND_NO_TRANSPARENCY = 1 << NDB_NO_TRANSPARENCY, 
  ND_SHADE_GREY      = 1 << NDB_SHADE_GREY,      
  ND_SHADE_DIMMED    = 1 << NDB_SHADE_DIMMED,    
  ND_DROPDOWN_ACTIVE = 1 << NDB_DROPDOWN_ACTIVE, 
  ND_SCROLLBAR_UP    = 1 << NDB_SCROLLBAR_UP,    
  ND_SCROLLBAR_DOWN  = 1 << NDB_SCROLLBAR_DOWN,  
  ND_SCROLLBAR_BTN   = ND_SCROLLBAR_UP | ND_SCROLLBAR_DOWN, 
  ND_DROPDOWN_CLOSED = 1 << NDB_DROPDOWN_CLOSED, 
};
DECLARE_ENUM_AS_BIT_SET(NWidgetDisplay)
 
 
class NWidgetCore : public NWidgetResizeBase {
public:
  NWidgetCore(WidgetType tp, Colours colour, WidgetID index, uint fill_x, uint fill_y, uint32_t widget_data, StringID tool_tip);
 
  void SetDataTip(uint32_t widget_data, StringID tool_tip);
  void SetToolTip(StringID tool_tip);
  void SetTextStyle(TextColour colour, FontSize size);
  void SetAlignment(StringAlignment align);
 
  inline void SetLowered(bool lowered);
  inline bool IsLowered() const;
  inline void SetDisabled(bool disabled);
  inline bool IsDisabled() const;
 
  void FillWidgetLookup(WidgetLookup &widget_lookup) override;
  NWidgetCore *GetWidgetFromPos(int x, int y) override;
  bool IsHighlighted() const override;
  TextColour GetHighlightColour() const override;
  void SetHighlighted(TextColour highlight_colour) override;
 
  NWidgetDisplay disp_flags; 
  Colours colour;            
  const WidgetID index;      
  uint32_t widget_data;        
  StringID tool_tip;         
  WidgetID scrollbar_index;  
  TextColour highlight_colour; 
  TextColour text_colour;    
  FontSize text_size;        
  StringAlignment align;     
};
 
inline void NWidgetCore::SetHighlighted(TextColour highlight_colour)
{
  this->disp_flags = highlight_colour != TC_INVALID ? SETBITS(this->disp_flags, ND_HIGHLIGHT) : CLRBITS(this->disp_flags, ND_HIGHLIGHT);
  this->highlight_colour = highlight_colour;
}
 
inline bool NWidgetCore::IsHighlighted() const
{
  return HasBit(this->disp_flags, NDB_HIGHLIGHT);
}
 
inline TextColour NWidgetCore::GetHighlightColour() const
{
  return this->highlight_colour;
}
 
inline void NWidgetCore::SetLowered(bool lowered)
{
  this->disp_flags = lowered ? SETBITS(this->disp_flags, ND_LOWERED) : CLRBITS(this->disp_flags, ND_LOWERED);
}
 
inline bool NWidgetCore::IsLowered() const
{
  return HasBit(this->disp_flags, NDB_LOWERED);
}
 
inline void NWidgetCore::SetDisabled(bool disabled)
{
  this->disp_flags = disabled ? SETBITS(this->disp_flags, ND_DISABLED) : CLRBITS(this->disp_flags, ND_DISABLED);
}
 
inline bool NWidgetCore::IsDisabled() const
{
  return HasBit(this->disp_flags, NDB_DISABLED);
}
 
 
class NWidgetContainer : public NWidgetBase {
public:
  NWidgetContainer(WidgetType tp) : NWidgetBase(tp) { }
 
  void AdjustPaddingForZoom() override;
  void Add(std::unique_ptr<NWidgetBase> &&wid);
  void FillWidgetLookup(WidgetLookup &widget_lookup) override;
 
  void Draw(const Window *w) override;
  NWidgetCore *GetWidgetFromPos(int x, int y) override;
 
  inline bool IsEmpty() { return this->children.empty(); }
 
  NWidgetBase *GetWidgetOfType(WidgetType tp) override;
 
protected:
  std::vector<std::unique_ptr<NWidgetBase>> children; 
};
 
enum StackedZeroSizePlanes {
  SZSP_VERTICAL = INT_MAX / 2, 
  SZSP_HORIZONTAL,             
  SZSP_NONE,                   
 
  SZSP_BEGIN = SZSP_VERTICAL,  
};
 
class NWidgetStacked : public NWidgetContainer {
public:
  NWidgetStacked(WidgetID index);
 
  void AdjustPaddingForZoom() override;
  void SetupSmallestSize(Window *w) override;
  void AssignSizePosition(SizingType sizing, int x, int y, uint given_width, uint given_height, bool rtl) override;
  void FillWidgetLookup(WidgetLookup &widget_lookup) override;
 
  void Draw(const Window *w) override;
  NWidgetCore *GetWidgetFromPos(int x, int y) override;
 
  bool SetDisplayedPlane(int plane);
 
  int shown_plane; 
  const WidgetID index; 
private:
  WidgetLookup *widget_lookup; 
};
 
enum NWidContainerFlags {
  NCB_EQUALSIZE = 0, 
  NCB_BIGFIRST  = 1, 
 
  NC_NONE = 0,                       
  NC_EQUALSIZE = 1 << NCB_EQUALSIZE, 
  NC_BIGFIRST  = 1 << NCB_BIGFIRST,  
};
DECLARE_ENUM_AS_BIT_SET(NWidContainerFlags)
 
 
class NWidgetPIPContainer : public NWidgetContainer {
public:
  NWidgetPIPContainer(WidgetType tp, NWidContainerFlags flags = NC_NONE);
 
  void AdjustPaddingForZoom() override;
  void SetPIP(uint8_t pip_pre, uint8_t pip_inter, uint8_t pip_post);
  void SetPIPRatio(uint8_t pip_ratio_pre, uint8_t pip_ratio_inter, uint8_t pip_rato_post);
 
protected:
  NWidContainerFlags flags; 
  uint8_t pip_pre;            
  uint8_t pip_inter;          
  uint8_t pip_post;           
  uint8_t pip_ratio_pre;      
  uint8_t pip_ratio_inter;    
  uint8_t pip_ratio_post;     
 
  uint8_t uz_pip_pre;         
  uint8_t uz_pip_inter;       
  uint8_t uz_pip_post;        
 
  uint8_t gaps; 
};
 
class NWidgetHorizontal : public NWidgetPIPContainer {
public:
  NWidgetHorizontal(NWidContainerFlags flags = NC_NONE);
 
  void SetupSmallestSize(Window *w) override;
  void AssignSizePosition(SizingType sizing, int x, int y, uint given_width, uint given_height, bool rtl) override;
};
 
class NWidgetHorizontalLTR : public NWidgetHorizontal {
public:
  NWidgetHorizontalLTR(NWidContainerFlags flags = NC_NONE);
 
  void AssignSizePosition(SizingType sizing, int x, int y, uint given_width, uint given_height, bool rtl) override;
};
 
class NWidgetVertical : public NWidgetPIPContainer {
public:
  NWidgetVertical(NWidContainerFlags flags = NC_NONE);
 
  void SetupSmallestSize(Window *w) override;
  void AssignSizePosition(SizingType sizing, int x, int y, uint given_width, uint given_height, bool rtl) override;
};
 
class NWidgetMatrix : public NWidgetPIPContainer {
public:
  NWidgetMatrix(Colours colour, WidgetID index);
 
  void SetClicked(int clicked);
  void SetCount(int count);
  void SetScrollbar(Scrollbar *sb);
  int GetCurrentElement() const;
 
  void SetupSmallestSize(Window *w) override;
  void AssignSizePosition(SizingType sizing, int x, int y, uint given_width, uint given_height, bool rtl) override;
  void FillWidgetLookup(WidgetLookup &widget_lookup) override;
 
  NWidgetCore *GetWidgetFromPos(int x, int y) override;
  void Draw(const Window *w) override;
protected:
  const WidgetID index; 
  Colours colour; 
  int clicked;    
  int count;      
  int current_element; 
  Scrollbar *sb;  
private:
  int widget_w;   
  int widget_h;   
  int widgets_x;  
  int widgets_y;  
 
  void GetScrollOffsets(int &start_x, int &start_y, int &base_offs_x, int &base_offs_y);
};
 
 
class NWidgetSpacer : public NWidgetResizeBase {
public:
  NWidgetSpacer(int width, int height);
 
  void SetupSmallestSize(Window *w) override;
  void FillWidgetLookup(WidgetLookup &widget_lookup) override;
 
  void Draw(const Window *w) override;
  void SetDirty(const Window *w) const override;
  NWidgetCore *GetWidgetFromPos(int x, int y) override;
};
 
class NWidgetBackground : public NWidgetCore {
public:
  NWidgetBackground(WidgetType tp, Colours colour, WidgetID index, std::unique_ptr<NWidgetPIPContainer> &&child = nullptr);
 
  void Add(std::unique_ptr<NWidgetBase> &&nwid);
  void SetPIP(uint8_t pip_pre, uint8_t pip_inter, uint8_t pip_post);
  void SetPIPRatio(uint8_t pip_ratio_pre, uint8_t pip_ratio_inter, uint8_t pip_ratio_post);
 
  void AdjustPaddingForZoom() override;
  void SetupSmallestSize(Window *w) override;
  void AssignSizePosition(SizingType sizing, int x, int y, uint given_width, uint given_height, bool rtl) override;
 
  void FillWidgetLookup(WidgetLookup &widget_lookup) override;
 
  void Draw(const Window *w) override;
  NWidgetCore *GetWidgetFromPos(int x, int y) override;
  NWidgetBase *GetWidgetOfType(WidgetType tp) override;
 
private:
  std::unique_ptr<NWidgetPIPContainer> child; 
};
 
class NWidgetViewport : public NWidgetCore {
public:
  NWidgetViewport(WidgetID index);
 
  void SetupSmallestSize(Window *w) override;
  void Draw(const Window *w) override;
 
  void InitializeViewport(Window *w, std::variant<TileIndex, VehicleID> focus, ZoomLevel zoom);
  void UpdateViewportCoordinates(Window *w);
};
 
class Scrollbar {
public:
  using size_type = int32_t;
  static constexpr size_type max_size_type = std::numeric_limits<size_type>::max();
  static constexpr size_type npos = max_size_type;
private:
  const bool is_vertical; 
  size_type count; 
  size_type cap; 
  size_type pos; 
  size_type stepsize; 
 
public:
  enum ScrollbarStepping {
    SS_RAW,             
    SS_SMALL,           
    SS_BIG,             
  };
 
  Scrollbar(bool is_vertical) : is_vertical(is_vertical), stepsize(1)
  {
  }
 
  inline size_type GetCount() const
  {
    return this->count;
  }
 
  inline size_type GetCapacity() const
  {
    return this->cap;
  }
 
  inline size_type GetPosition() const
  {
    return this->pos;
  }
 
  inline bool IsVisible(size_type item) const
  {
    return IsInsideBS(item, this->GetPosition(), this->GetCapacity());
  }
 
  inline bool IsVertical() const
  {
    return this->is_vertical;
  }
 
  void SetStepSize(size_t stepsize)
  {
    assert(stepsize > 0);
 
    this->stepsize = ClampTo<size_type>(stepsize);
  }
 
  void SetCount(size_t num)
  {
    assert(num < Scrollbar::max_size_type);
 
    this->count = ClampTo<size_type>(num);
    /* Ensure position is within bounds */
    this->SetPosition(this->pos);
  }
 
  void SetCapacity(size_t capacity)
  {
    assert(capacity < Scrollbar::max_size_type);
 
    this->cap = ClampTo<size_type>(capacity);
    /* Ensure position is within bounds */
    this->SetPosition(this->pos);
  }
 
  void SetCapacityFromWidget(Window *w, WidgetID widget, int padding = 0);
 
  bool SetPosition(size_type position)
  {
    size_type old_pos = this->pos;
    this->pos = Clamp(position, 0, std::max(this->count - this->cap, 0));
    return this->pos != old_pos;
  }
 
  bool UpdatePosition(int difference, ScrollbarStepping unit = SS_SMALL)
  {
    if (difference == 0) return false;
    switch (unit) {
      case SS_SMALL: difference *= this->stepsize; break;
      case SS_BIG:   difference *= this->cap; break;
      default: break;
    }
    return this->SetPosition(this->pos + difference);
  }
 
  void ScrollTowards(size_type position)
  {
    if (position < this->GetPosition()) {
      /* scroll up to the item */
      this->SetPosition(position);
    } else if (position >= this->GetPosition() + this->GetCapacity()) {
      /* scroll down so that the item is at the bottom */
      this->SetPosition(position - this->GetCapacity() + 1);
    }
  }
 
  size_type GetScrolledRowFromWidget(int clickpos, const Window * const w, WidgetID widget, int padding = 0, int line_height = -1) const;
 
  template <typename Tcontainer>
  auto GetVisibleRangeIterators(Tcontainer &container) const
  {
    assert(static_cast<size_t>(this->GetCount()) == container.size()); // Scrollbar and container size must match.
    auto first = std::next(std::begin(container), this->GetPosition());
    auto last = std::next(first, std::min<size_t>(this->GetCapacity(), this->GetCount() - this->GetPosition()));
    return std::make_pair(first, last);
  }
 
  template <typename Tcontainer>
  auto GetScrolledItemFromWidget(Tcontainer &container, int clickpos, const Window * const w, WidgetID widget, int padding = 0, int line_height = -1) const
  {
    assert(static_cast<size_t>(this->GetCount()) == container.size()); // Scrollbar and container size must match.
    size_type row = this->GetScrolledRowFromWidget(clickpos, w, widget, padding, line_height);
    if (row == Scrollbar::npos) return std::end(container);
 
    return std::next(std::begin(container), row);
  }
 
  EventState UpdateListPositionOnKeyPress(int &list_position, uint16_t keycode) const;
};
 
class NWidgetScrollbar : public NWidgetCore, public Scrollbar {
public:
  NWidgetScrollbar(WidgetType tp, Colours colour, WidgetID index);
 
  void SetupSmallestSize(Window *w) override;
  void Draw(const Window *w) override;
 
  static void InvalidateDimensionCache();
  static Dimension GetVerticalDimension();
  static Dimension GetHorizontalDimension();
 
private:
  static Dimension vertical_dimension;   
  static Dimension horizontal_dimension; 
};
 
class NWidgetLeaf : public NWidgetCore {
public:
  NWidgetLeaf(WidgetType tp, Colours colour, WidgetID index, uint32_t data, StringID tip);
 
  void SetupSmallestSize(Window *w) override;
  void Draw(const Window *w) override;
 
  bool ButtonHit(const Point &pt);
 
  static void InvalidateDimensionCache();
 
  static Dimension dropdown_dimension;  
  static Dimension resizebox_dimension; 
  static Dimension closebox_dimension;  
private:
  static Dimension shadebox_dimension;  
  static Dimension debugbox_dimension;  
  static Dimension defsizebox_dimension; 
  static Dimension stickybox_dimension; 
};
 
inline uint ComputeMaxSize(uint base, uint max_space, uint step)
{
  if (base >= max_space || step == 0) return base;
  if (step == 1) return max_space;
  uint increment = max_space - base;
  increment -= increment % step;
  return base + increment;
}
 
struct NWidgetPartDataTip {
  uint32_t data;      
  StringID tooltip; 
};
 
struct NWidgetPartWidget {
  Colours colour; 
  WidgetID index; 
};
 
struct NWidgetPartPaddings : RectPadding {
};
 
struct NWidgetPartPIP {
  uint8_t pre, inter, post; 
};
 
struct NWidgetPartTextLines {
  uint8_t lines;   
  uint8_t spacing; 
  FontSize size; 
};
 
struct NWidgetPartTextStyle {
  TextColour colour; 
  FontSize size; 
};
 
struct NWidgetPartAlignment {
  StringAlignment align; 
};
 
struct NWidgetPartAspect {
  float ratio;
  AspectFlags flags;
};
 
typedef std::unique_ptr<NWidgetBase> NWidgetFunctionType();
 
struct NWidgetPart {
  WidgetType type;                         
  union NWidgetPartUnion {
    Point xy;                        
    NWidgetPartDataTip data_tip;     
    NWidgetPartWidget widget;        
    NWidgetPartPaddings padding;     
    NWidgetPartPIP pip;              
    NWidgetPartTextLines text_lines; 
    NWidgetPartTextStyle text_style; 
    NWidgetPartAlignment align;      
    NWidgetFunctionType *func_ptr;   
    NWidContainerFlags cont_flags;   
    NWidgetPartAspect aspect; 
 
    /* Constructors for each NWidgetPartUnion data type. */
    constexpr NWidgetPartUnion() : xy() {}
    constexpr NWidgetPartUnion(Point xy) : xy(xy) {}
    constexpr NWidgetPartUnion(NWidgetPartDataTip data_tip) : data_tip(data_tip) {}
    constexpr NWidgetPartUnion(NWidgetPartWidget widget) : widget(widget) {}
    constexpr NWidgetPartUnion(NWidgetPartPaddings padding) : padding(padding) {}
    constexpr NWidgetPartUnion(NWidgetPartPIP pip) : pip(pip) {}
    constexpr NWidgetPartUnion(NWidgetPartTextLines text_lines) : text_lines(text_lines) {}
    constexpr NWidgetPartUnion(NWidgetPartTextStyle text_style) : text_style(text_style) {}
    constexpr NWidgetPartUnion(NWidgetPartAlignment align) : align(align) {}
    constexpr NWidgetPartUnion(NWidgetFunctionType *func_ptr) : func_ptr(func_ptr) {}
    constexpr NWidgetPartUnion(NWidContainerFlags cont_flags) : cont_flags(cont_flags) {}
    constexpr NWidgetPartUnion(NWidgetPartAspect aspect) : aspect(aspect) {}
  } u;
 
  /* Constructors for each NWidgetPart data type. */
  explicit constexpr NWidgetPart(WidgetType type) : type(type), u() {}
  constexpr NWidgetPart(WidgetType type, Point xy) : type(type), u(xy) {}
  constexpr NWidgetPart(WidgetType type, NWidgetPartDataTip data_tip) : type(type), u(data_tip) {}
  constexpr NWidgetPart(WidgetType type, NWidgetPartWidget widget) : type(type), u(widget) {}
  constexpr NWidgetPart(WidgetType type, NWidgetPartPaddings padding) : type(type), u(padding) {}
  constexpr NWidgetPart(WidgetType type, NWidgetPartPIP pip) : type(type), u(pip) {}
  constexpr NWidgetPart(WidgetType type, NWidgetPartTextLines text_lines) : type(type), u(text_lines) {}
  constexpr NWidgetPart(WidgetType type, NWidgetPartTextStyle text_style) : type(type), u(text_style) {}
  constexpr NWidgetPart(WidgetType type, NWidgetPartAlignment align) : type(type), u(align) {}
  constexpr NWidgetPart(WidgetType type, NWidgetFunctionType *func_ptr) : type(type), u(func_ptr) {}
  constexpr NWidgetPart(WidgetType type, NWidContainerFlags cont_flags) : type(type), u(cont_flags) {}
  constexpr NWidgetPart(WidgetType type, NWidgetPartAspect aspect) : type(type), u(aspect) {}
};
 
constexpr NWidgetPart SetResize(int16_t dx, int16_t dy)
{
  return NWidgetPart{WPT_RESIZE, Point{dx, dy}};
}
 
constexpr NWidgetPart SetMinimalSize(int16_t x, int16_t y)
{
  return NWidgetPart{WPT_MINSIZE, Point{x, y}};
}
 
constexpr NWidgetPart SetMinimalTextLines(uint8_t lines, uint8_t spacing, FontSize size = FS_NORMAL)
{
  return NWidgetPart{WPT_MINTEXTLINES, NWidgetPartTextLines{lines, spacing, size}};
}
 
constexpr NWidgetPart SetTextStyle(TextColour colour, FontSize size = FS_NORMAL)
{
  return NWidgetPart{WPT_TEXTSTYLE, NWidgetPartTextStyle{colour, size}};
}
 
constexpr NWidgetPart SetAlignment(StringAlignment align)
{
  return NWidgetPart{WPT_ALIGNMENT, NWidgetPartAlignment{align}};
}
 
constexpr NWidgetPart SetFill(uint16_t fill_x, uint16_t fill_y)
{
  return NWidgetPart{WPT_FILL, Point{fill_x, fill_y}};
}
 
constexpr NWidgetPart EndContainer()
{
  return NWidgetPart{WPT_ENDCONTAINER};
}
 
constexpr NWidgetPart SetDataTip(uint32_t data, StringID tip)
{
  return NWidgetPart{WPT_DATATIP, NWidgetPartDataTip{data, tip}};
}
 
constexpr NWidgetPart SetMatrixDataTip(uint8_t cols, uint8_t rows, StringID tip)
{
  return SetDataTip((rows << MAT_ROW_START) | (cols << MAT_COL_START), tip);
}
 
constexpr NWidgetPart SetPadding(uint8_t top, uint8_t right, uint8_t bottom, uint8_t left)
{
  return NWidgetPart{WPT_PADDING, NWidgetPartPaddings{left, top, right, bottom}};
}
 
constexpr NWidgetPart SetPadding(const RectPadding &padding)
{
  return NWidgetPart{WPT_PADDING, NWidgetPartPaddings{padding}};
}
 
constexpr NWidgetPart SetPadding(uint8_t padding)
{
  return SetPadding(padding, padding, padding, padding);
}
 
constexpr NWidgetPart SetPIP(uint8_t pre, uint8_t inter, uint8_t post)
{
  return NWidgetPart{WPT_PIPSPACE, NWidgetPartPIP{pre, inter, post}};
}
 
constexpr NWidgetPart SetPIPRatio(uint8_t ratio_pre, uint8_t ratio_inter, uint8_t ratio_post)
{
  return NWidgetPart{WPT_PIPRATIO, NWidgetPartPIP{ratio_pre, ratio_inter, ratio_post}};
}
 
constexpr NWidgetPart SetScrollbar(WidgetID index)
{
  return NWidgetPart{WPT_SCROLLBAR, NWidgetPartWidget{INVALID_COLOUR, index}};
}
 
constexpr NWidgetPart SetAspect(float ratio, AspectFlags flags = AspectFlags::ResizeX)
{
  return NWidgetPart{WPT_ASPECT, NWidgetPartAspect{ratio, flags}};
}
 
constexpr NWidgetPart NWidget(WidgetType tp, Colours col, WidgetID idx = -1)
{
  return NWidgetPart{tp, NWidgetPartWidget{col, idx}};
}
 
constexpr NWidgetPart NWidget(WidgetType tp, NWidContainerFlags cont_flags = NC_NONE)
{
  return NWidgetPart{tp, NWidContainerFlags{cont_flags}};
}
 
constexpr NWidgetPart NWidgetFunction(NWidgetFunctionType *func_ptr)
{
  return NWidgetPart{WPT_FUNCTION, func_ptr};
}
 
bool IsContainerWidgetType(WidgetType tp);
std::unique_ptr<NWidgetBase> MakeNWidgets(std::span<const NWidgetPart> nwid_parts, std::unique_ptr<NWidgetBase> &&container);
std::unique_ptr<NWidgetBase> MakeWindowNWidgetTree(std::span<const NWidgetPart> nwid_parts, NWidgetStacked **shade_select);
 
std::unique_ptr<NWidgetBase> MakeCompanyButtonRows(WidgetID widget_first, WidgetID widget_last, Colours button_colour, int max_length, StringID button_tooltip, bool resizable = true);
 
void SetupWidgetDimensions();
 
#endif /* WIDGET_TYPE_H */