(svn r27796) -Fix [FS#6545]: 32bpp-anim blitters assumed that pitch and width of...

[openttd-github.git] / src / graph_gui.cpp

/* $Id$ */

/*
 * 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/>.
 */

/** @file graph_gui.cpp GUI that shows performance graphs. */

#include "stdafx.h"
#include "graph_gui.h"
#include "window_gui.h"
#include "company_base.h"
#include "company_gui.h"
#include "economy_func.h"
#include "cargotype.h"
#include "strings_func.h"
#include "window_func.h"
#include "date_func.h"
#include "gfx_func.h"
#include "sortlist_type.h"
#include "core/geometry_func.hpp"
#include "currency.h"

#include "widgets/graph_widget.h"

#include "table/strings.h"
#include "table/sprites.h"
#include <math.h>

#include "safeguards.h"

/* Bitmasks of company and cargo indices that shouldn't be drawn. */
static uint _legend_excluded_companies;
static uint _legend_excluded_cargo;

/* Apparently these don't play well with enums. */
static const OverflowSafeInt64 INVALID_DATAPOINT(INT64_MAX); // Value used for a datapoint that shouldn't be drawn.
static const uint INVALID_DATAPOINT_POS = UINT_MAX;  // Used to determine if the previous point was drawn.

/****************/
/* GRAPH LEGEND */
/****************/

struct GraphLegendWindow : Window {
        GraphLegendWindow(WindowDesc *desc, WindowNumber window_number) : Window(desc)
        {
                this->InitNested(window_number);

                for (CompanyID c = COMPANY_FIRST; c < MAX_COMPANIES; c++) {
                        if (!HasBit(_legend_excluded_companies, c)) this->LowerWidget(c + WID_GL_FIRST_COMPANY);

                        this->OnInvalidateData(c);
                }
        }

        virtual void DrawWidget(const Rect &r, int widget) const
        {
                if (!IsInsideMM(widget, WID_GL_FIRST_COMPANY, MAX_COMPANIES + WID_GL_FIRST_COMPANY)) return;

                CompanyID cid = (CompanyID)(widget - WID_GL_FIRST_COMPANY);

                if (!Company::IsValidID(cid)) return;

                bool rtl = _current_text_dir == TD_RTL;

                Dimension d = GetSpriteSize(SPR_COMPANY_ICON);
                DrawCompanyIcon(cid, rtl ? r.right - d.width - 2 : r.left + 2, r.top + (r.bottom - r.top - d.height) / 2);

                SetDParam(0, cid);
                SetDParam(1, cid);
                DrawString(r.left + (rtl ? (uint)WD_FRAMERECT_LEFT : (d.width + 4)), r.right - (rtl ? (d.width + 4) : (uint)WD_FRAMERECT_RIGHT), r.top + (r.bottom - r.top + 1 - FONT_HEIGHT_NORMAL) / 2, STR_COMPANY_NAME_COMPANY_NUM, HasBit(_legend_excluded_companies, cid) ? TC_BLACK : TC_WHITE);
        }

        virtual void OnClick(Point pt, int widget, int click_count)
        {
                if (!IsInsideMM(widget, WID_GL_FIRST_COMPANY, MAX_COMPANIES + WID_GL_FIRST_COMPANY)) return;

                ToggleBit(_legend_excluded_companies, widget - WID_GL_FIRST_COMPANY);
                this->ToggleWidgetLoweredState(widget);
                this->SetDirty();
                InvalidateWindowData(WC_INCOME_GRAPH, 0);
                InvalidateWindowData(WC_OPERATING_PROFIT, 0);
                InvalidateWindowData(WC_DELIVERED_CARGO, 0);
                InvalidateWindowData(WC_PERFORMANCE_HISTORY, 0);
                InvalidateWindowData(WC_COMPANY_VALUE, 0);
        }

        /**
         * Some data on this window has become invalid.
         * @param data Information about the changed data.
         * @param gui_scope Whether the call is done from GUI scope. You may not do everything when not in GUI scope. See #InvalidateWindowData() for details.
         */
        virtual void OnInvalidateData(int data = 0, bool gui_scope = true)
        {
                if (!gui_scope) return;
                if (Company::IsValidID(data)) return;

                SetBit(_legend_excluded_companies, data);
                this->RaiseWidget(data + WID_GL_FIRST_COMPANY);
        }
};

/**
 * Construct a vertical list of buttons, one for each company.
 * @param biggest_index Storage for collecting the biggest index used in the returned tree.
 * @return Panel with company buttons.
 * @post \c *biggest_index contains the largest used index in the tree.
 */
static NWidgetBase *MakeNWidgetCompanyLines(int *biggest_index)
{
        NWidgetVertical *vert = new NWidgetVertical();
        uint line_height = max<uint>(GetSpriteSize(SPR_COMPANY_ICON).height, FONT_HEIGHT_NORMAL) + WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;

        for (int widnum = WID_GL_FIRST_COMPANY; widnum <= WID_GL_LAST_COMPANY; widnum++) {
                NWidgetBackground *panel = new NWidgetBackground(WWT_PANEL, COLOUR_GREY, widnum);
                panel->SetMinimalSize(246, line_height);
                panel->SetFill(1, 0);
                panel->SetDataTip(0x0, STR_GRAPH_KEY_COMPANY_SELECTION_TOOLTIP);
                vert->Add(panel);
        }
        *biggest_index = WID_GL_LAST_COMPANY;
        return vert;
}

static const NWidgetPart _nested_graph_legend_widgets[] = {
        NWidget(NWID_HORIZONTAL),
                NWidget(WWT_CLOSEBOX, COLOUR_GREY),
                NWidget(WWT_CAPTION, COLOUR_GREY), SetDataTip(STR_GRAPH_KEY_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
                NWidget(WWT_SHADEBOX, COLOUR_GREY),
                NWidget(WWT_STICKYBOX, COLOUR_GREY),
        EndContainer(),
        NWidget(WWT_PANEL, COLOUR_GREY, WID_GL_BACKGROUND),
                NWidget(NWID_SPACER), SetMinimalSize(0, 2),
                NWidget(NWID_HORIZONTAL),
                        NWidget(NWID_SPACER), SetMinimalSize(2, 0),
                        NWidgetFunction(MakeNWidgetCompanyLines),
                        NWidget(NWID_SPACER), SetMinimalSize(2, 0),
                EndContainer(),
        EndContainer(),
};

static WindowDesc _graph_legend_desc(
        WDP_AUTO, "graph_legend", 0, 0,
        WC_GRAPH_LEGEND, WC_NONE,
        0,
        _nested_graph_legend_widgets, lengthof(_nested_graph_legend_widgets)
);

static void ShowGraphLegend()
{
        AllocateWindowDescFront<GraphLegendWindow>(&_graph_legend_desc, 0);
}

/** Contains the interval of a graph's data. */
struct ValuesInterval {
        OverflowSafeInt64 highest; ///< Highest value of this interval. Must be zero or greater.
        OverflowSafeInt64 lowest;  ///< Lowest value of this interval. Must be zero or less.
};

/******************/
/* BASE OF GRAPHS */
/*****************/

struct BaseGraphWindow : Window {
protected:
        static const int GRAPH_MAX_DATASETS     =  32;
        static const int GRAPH_AXIS_LINE_COLOUR = PC_BLACK;
        static const int GRAPH_NUM_MONTHS       =  24; ///< Number of months displayed in the graph.

        static const int MIN_GRAPH_NUM_LINES_Y  =   9; ///< Minimal number of horizontal lines to draw.
        static const int MIN_GRID_PIXEL_SIZE    =  20; ///< Minimum distance between graph lines.

        uint excluded_data; ///< bitmask of the datasets that shouldn't be displayed.
        byte num_dataset;
        byte num_on_x_axis;
        byte num_vert_lines;
        static const TextColour graph_axis_label_colour = TC_BLACK; ///< colour of the graph axis label.

        /* The starting month and year that values are plotted against. If month is
         * 0xFF, use x_values_start and x_values_increment below instead. */
        byte month;
        Year year;

        /* These values are used if the graph is being plotted against values
         * rather than the dates specified by month and year. */
        uint16 x_values_start;
        uint16 x_values_increment;

        int graph_widget;
        StringID format_str_y_axis;
        byte colours[GRAPH_MAX_DATASETS];
        OverflowSafeInt64 cost[GRAPH_MAX_DATASETS][GRAPH_NUM_MONTHS]; ///< Stored costs for the last #GRAPH_NUM_MONTHS months

        /**
         * Get the interval that contains the graph's data. Excluded data is ignored to show smaller values in
         * better detail when disabling higher ones.
         * @param num_hori_lines Number of horizontal lines to be drawn.
         * @return Highest and lowest values of the graph (ignoring disabled data).
         */
        ValuesInterval GetValuesInterval(int num_hori_lines) const
        {
                assert(num_hori_lines > 0);

                ValuesInterval current_interval;
                current_interval.highest = INT64_MIN;
                current_interval.lowest  = INT64_MAX;

                for (int i = 0; i < this->num_dataset; i++) {
                        if (HasBit(this->excluded_data, i)) continue;
                        for (int j = 0; j < this->num_on_x_axis; j++) {
                                OverflowSafeInt64 datapoint = this->cost[i][j];

                                if (datapoint != INVALID_DATAPOINT) {
                                        current_interval.highest = max(current_interval.highest, datapoint);
                                        current_interval.lowest  = min(current_interval.lowest, datapoint);
                                }
                        }
                }

                /* Prevent showing values too close to the graph limits. */
                current_interval.highest = (11 * current_interval.highest) / 10;
                current_interval.lowest =  (11 * current_interval.lowest) / 10;

                /* Always include zero in the shown range. */
                double abs_lower  = (current_interval.lowest > 0) ? 0 : (double)abs(current_interval.lowest);
                double abs_higher = (current_interval.highest < 0) ? 0 : (double)current_interval.highest;

                int num_pos_grids;
                int64 grid_size;

                if (abs_lower != 0 || abs_higher != 0) {
                        /* The number of grids to reserve for the positive part is: */
                        num_pos_grids = (int)floor(0.5 + num_hori_lines * abs_higher / (abs_higher + abs_lower));

                        /* If there are any positive or negative values, force that they have at least one grid. */
                        if (num_pos_grids == 0 && abs_higher != 0) num_pos_grids++;
                        if (num_pos_grids == num_hori_lines && abs_lower != 0) num_pos_grids--;

                        /* Get the required grid size for each side and use the maximum one. */
                        int64 grid_size_higher = (abs_higher > 0) ? ((int64)abs_higher + num_pos_grids - 1) / num_pos_grids : 0;
                        int64 grid_size_lower = (abs_lower > 0) ? ((int64)abs_lower + num_hori_lines - num_pos_grids - 1) / (num_hori_lines - num_pos_grids) : 0;
                        grid_size = max(grid_size_higher, grid_size_lower);
                } else {
                        /* If both values are zero, show an empty graph. */
                        num_pos_grids = num_hori_lines / 2;
                        grid_size = 1;
                }

                current_interval.highest = num_pos_grids * grid_size;
                current_interval.lowest = -(num_hori_lines - num_pos_grids) * grid_size;
                return current_interval;
        }

        /**
         * Get width for Y labels.
         * @param current_interval Interval that contains all of the graph data.
         * @param num_hori_lines Number of horizontal lines to be drawn.
         */
        uint GetYLabelWidth(ValuesInterval current_interval, int num_hori_lines) const
        {
                /* draw text strings on the y axis */
                int64 y_label = current_interval.highest;
                int64 y_label_separation = (current_interval.highest - current_interval.lowest) / num_hori_lines;

                uint max_width = 0;

                for (int i = 0; i < (num_hori_lines + 1); i++) {
                        SetDParam(0, this->format_str_y_axis);
                        SetDParam(1, y_label);
                        Dimension d = GetStringBoundingBox(STR_GRAPH_Y_LABEL);
                        if (d.width > max_width) max_width = d.width;

                        y_label -= y_label_separation;
                }

                return max_width;
        }

        /**
         * Actually draw the graph.
         * @param r the rectangle of the data field of the graph
         */
        void DrawGraph(Rect r) const
        {
                uint x, y;               ///< Reused whenever x and y coordinates are needed.
                ValuesInterval interval; ///< Interval that contains all of the graph data.
                int x_axis_offset;       ///< Distance from the top of the graph to the x axis.

                /* the colours and cost array of GraphDrawer must accommodate
                 * both values for cargo and companies. So if any are higher, quit */
                assert_compile(GRAPH_MAX_DATASETS >= (int)NUM_CARGO && GRAPH_MAX_DATASETS >= (int)MAX_COMPANIES);
                assert(this->num_vert_lines > 0);

                byte grid_colour = _colour_gradient[COLOUR_GREY][4];

                /* Rect r will be adjusted to contain just the graph, with labels being
                 * placed outside the area. */
                r.top    += 5 + GetCharacterHeight(FS_SMALL) / 2;
                r.bottom -= (this->month == 0xFF ? 1 : 3) * GetCharacterHeight(FS_SMALL) + 4;
                r.left   += 9;
                r.right  -= 5;

                /* Initial number of horizontal lines. */
                int num_hori_lines = 160 / MIN_GRID_PIXEL_SIZE;
                /* For the rest of the height, the number of horizontal lines will increase more slowly. */
                int resize = (r.bottom - r.top - 160) / (2 * MIN_GRID_PIXEL_SIZE);
                if (resize > 0) num_hori_lines += resize;

                interval = GetValuesInterval(num_hori_lines);

                int label_width = GetYLabelWidth(interval, num_hori_lines);

                r.left += label_width;

                int x_sep = (r.right - r.left) / this->num_vert_lines;
                int y_sep = (r.bottom - r.top) / num_hori_lines;

                /* Redetermine right and bottom edge of graph to fit with the integer
                 * separation values. */
                r.right = r.left + x_sep * this->num_vert_lines;
                r.bottom = r.top + y_sep * num_hori_lines;

                OverflowSafeInt64 interval_size = interval.highest + abs(interval.lowest);
                /* Where to draw the X axis. Use floating point to avoid overflowing and results of zero. */
                x_axis_offset = (int)((r.bottom - r.top) * (double)interval.highest / (double)interval_size);

                /* Draw the vertical grid lines. */

                /* Don't draw the first line, as that's where the axis will be. */
                x = r.left + x_sep;

                for (int i = 0; i < this->num_vert_lines; i++) {
                        GfxFillRect(x, r.top, x, r.bottom, grid_colour);
                        x += x_sep;
                }

                /* Draw the horizontal grid lines. */
                y = r.bottom;

                for (int i = 0; i < (num_hori_lines + 1); i++) {
                        GfxFillRect(r.left - 3, y, r.left - 1, y, GRAPH_AXIS_LINE_COLOUR);
                        GfxFillRect(r.left, y, r.right, y, grid_colour);
                        y -= y_sep;
                }

                /* Draw the y axis. */
                GfxFillRect(r.left, r.top, r.left, r.bottom, GRAPH_AXIS_LINE_COLOUR);

                /* Draw the x axis. */
                y = x_axis_offset + r.top;
                GfxFillRect(r.left, y, r.right, y, GRAPH_AXIS_LINE_COLOUR);

                /* Find the largest value that will be drawn. */
                if (this->num_on_x_axis == 0) return;

                assert(this->num_on_x_axis > 0);
                assert(this->num_dataset > 0);

                /* draw text strings on the y axis */
                int64 y_label = interval.highest;
                int64 y_label_separation = abs(interval.highest - interval.lowest) / num_hori_lines;

                y = r.top - GetCharacterHeight(FS_SMALL) / 2;

                for (int i = 0; i < (num_hori_lines + 1); i++) {
                        SetDParam(0, this->format_str_y_axis);
                        SetDParam(1, y_label);
                        DrawString(r.left - label_width - 4, r.left - 4, y, STR_GRAPH_Y_LABEL, graph_axis_label_colour, SA_RIGHT);

                        y_label -= y_label_separation;
                        y += y_sep;
                }

                /* draw strings on the x axis */
                if (this->month != 0xFF) {
                        x = r.left;
                        y = r.bottom + 2;
                        byte month = this->month;
                        Year year  = this->year;
                        for (int i = 0; i < this->num_on_x_axis; i++) {
                                SetDParam(0, month + STR_MONTH_ABBREV_JAN);
                                SetDParam(1, month + STR_MONTH_ABBREV_JAN + 2);
                                SetDParam(2, year);
                                DrawStringMultiLine(x, x + x_sep, y, this->height, month == 0 ? STR_GRAPH_X_LABEL_MONTH_YEAR : STR_GRAPH_X_LABEL_MONTH, graph_axis_label_colour);

                                month += 3;
                                if (month >= 12) {
                                        month = 0;
                                        year++;
                                }
                                x += x_sep;
                        }
                } else {
                        /* Draw the label under the data point rather than on the grid line. */
                        x = r.left;
                        y = r.bottom + 2;
                        uint16 label = this->x_values_start;

                        for (int i = 0; i < this->num_on_x_axis; i++) {
                                SetDParam(0, label);
                                DrawString(x + 1, x + x_sep - 1, y, STR_GRAPH_Y_LABEL_NUMBER, graph_axis_label_colour, SA_HOR_CENTER);

                                label += this->x_values_increment;
                                x += x_sep;
                        }
                }

                /* draw lines and dots */
                uint linewidth = _settings_client.gui.graph_line_thickness;
                uint pointoffs1 = (linewidth + 1) / 2;
                uint pointoffs2 = linewidth + 1 - pointoffs1;
                for (int i = 0; i < this->num_dataset; i++) {
                        if (!HasBit(this->excluded_data, i)) {
                                /* Centre the dot between the grid lines. */
                                x = r.left + (x_sep / 2);

                                byte colour  = this->colours[i];
                                uint prev_x = INVALID_DATAPOINT_POS;
                                uint prev_y = INVALID_DATAPOINT_POS;

                                for (int j = 0; j < this->num_on_x_axis; j++) {
                                        OverflowSafeInt64 datapoint = this->cost[i][j];

                                        if (datapoint != INVALID_DATAPOINT) {
                                                /*
                                                 * Check whether we need to reduce the 'accuracy' of the
                                                 * datapoint value and the highest value to split overflows.
                                                 * And when 'drawing' 'one million' or 'one million and one'
                                                 * there is no significant difference, so the least
                                                 * significant bits can just be removed.
                                                 *
                                                 * If there are more bits needed than would fit in a 32 bits
                                                 * integer, so at about 31 bits because of the sign bit, the
                                                 * least significant bits are removed.
                                                 */
                                                int mult_range = FindLastBit(x_axis_offset) + FindLastBit(abs(datapoint));
                                                int reduce_range = max(mult_range - 31, 0);

                                                /* Handle negative values differently (don't shift sign) */
                                                if (datapoint < 0) {
                                                        datapoint = -(abs(datapoint) >> reduce_range);
                                                } else {
                                                        datapoint >>= reduce_range;
                                                }
                                                y = r.top + x_axis_offset - ((r.bottom - r.top) * datapoint) / (interval_size >> reduce_range);

                                                /* Draw the point. */
                                                GfxFillRect(x - pointoffs1, y - pointoffs1, x + pointoffs2, y + pointoffs2, colour);

                                                /* Draw the line connected to the previous point. */
                                                if (prev_x != INVALID_DATAPOINT_POS) GfxDrawLine(prev_x, prev_y, x, y, colour, linewidth);

                                                prev_x = x;
                                                prev_y = y;
                                        } else {
                                                prev_x = INVALID_DATAPOINT_POS;
                                                prev_y = INVALID_DATAPOINT_POS;
                                        }

                                        x += x_sep;
                                }
                        }
                }
        }


        BaseGraphWindow(WindowDesc *desc, int widget, StringID format_str_y_axis) :
                        Window(desc),
                        format_str_y_axis(format_str_y_axis)
        {
                SetWindowDirty(WC_GRAPH_LEGEND, 0);
                this->num_vert_lines = 24;
                this->graph_widget = widget;
        }

        void InitializeWindow(WindowNumber number)
        {
                /* Initialise the dataset */
                this->UpdateStatistics(true);

                this->InitNested(number);
        }

public:
        virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
        {
                if (widget != this->graph_widget) return;

                uint x_label_width = 0;

                if (this->month != 0xFF) {
                        byte month = this->month;
                        Year year  = this->year;
                        for (int i = 0; i < this->num_on_x_axis; i++) {
                                SetDParam(0, month + STR_MONTH_ABBREV_JAN);
                                SetDParam(1, month + STR_MONTH_ABBREV_JAN + 2);
                                SetDParam(2, year);
                                x_label_width = max(x_label_width, GetStringBoundingBox(month == 0 ? STR_GRAPH_X_LABEL_MONTH_YEAR : STR_GRAPH_X_LABEL_MONTH).width);

                                month += 3;
                                if (month >= 12) {
                                        month = 0;
                                        year++;
                                }
                        }
                } else {
                        /* Draw the label under the data point rather than on the grid line. */
                        SetDParamMaxValue(0, this->x_values_start + this->num_on_x_axis * this->x_values_increment, 0, FS_SMALL);
                        x_label_width = GetStringBoundingBox(STR_GRAPH_Y_LABEL_NUMBER).width;
                }

                SetDParam(0, this->format_str_y_axis);
                SetDParam(1, INT64_MAX);
                uint y_label_width = GetStringBoundingBox(STR_GRAPH_Y_LABEL).width;

                size->width  = max<uint>(size->width,  5 + y_label_width + this->num_on_x_axis * (x_label_width + 5) + 9);
                size->height = max<uint>(size->height, 5 + (1 + MIN_GRAPH_NUM_LINES_Y * 2 + (this->month != 0xFF ? 3 : 1)) * FONT_HEIGHT_SMALL + 4);
                size->height = max<uint>(size->height, size->width / 3);
        }

        virtual void DrawWidget(const Rect &r, int widget) const
        {
                if (widget != this->graph_widget) return;

                DrawGraph(r);
        }

        virtual OverflowSafeInt64 GetGraphData(const Company *c, int j)
        {
                return INVALID_DATAPOINT;
        }

        virtual void OnClick(Point pt, int widget, int click_count)
        {
                /* Clicked on legend? */
                if (widget == WID_CV_KEY_BUTTON) ShowGraphLegend();
        }

        virtual void OnTick()
        {
                this->UpdateStatistics(false);
        }

        /**
         * Some data on this window has become invalid.
         * @param data Information about the changed data.
         * @param gui_scope Whether the call is done from GUI scope. You may not do everything when not in GUI scope. See #InvalidateWindowData() for details.
         */
        virtual void OnInvalidateData(int data = 0, bool gui_scope = true)
        {
                if (!gui_scope) return;
                this->UpdateStatistics(true);
        }

        /**
         * Update the statistics.
         * @param initialize Initialize the data structure.
         */
        void UpdateStatistics(bool initialize)
        {
                uint excluded_companies = _legend_excluded_companies;

                /* Exclude the companies which aren't valid */
                for (CompanyID c = COMPANY_FIRST; c < MAX_COMPANIES; c++) {
                        if (!Company::IsValidID(c)) SetBit(excluded_companies, c);
                }

                byte nums = 0;
                const Company *c;
                FOR_ALL_COMPANIES(c) {
                        nums = min(this->num_vert_lines, max(nums, c->num_valid_stat_ent));
                }

                int mo = (_cur_month / 3 - nums) * 3;
                int yr = _cur_year;
                while (mo < 0) {
                        yr--;
                        mo += 12;
                }

                if (!initialize && this->excluded_data == excluded_companies && this->num_on_x_axis == nums &&
                                this->year == yr && this->month == mo) {
                        /* There's no reason to get new stats */
                        return;
                }

                this->excluded_data = excluded_companies;
                this->num_on_x_axis = nums;
                this->year = yr;
                this->month = mo;

                int numd = 0;
                for (CompanyID k = COMPANY_FIRST; k < MAX_COMPANIES; k++) {
                        c = Company::GetIfValid(k);
                        if (c != NULL) {
                                this->colours[numd] = _colour_gradient[c->colour][6];
                                for (int j = this->num_on_x_axis, i = 0; --j >= 0;) {
                                        this->cost[numd][i] = (j >= c->num_valid_stat_ent) ? INVALID_DATAPOINT : GetGraphData(c, j);
                                        i++;
                                }
                        }
                        numd++;
                }

                this->num_dataset = numd;
        }
};


/********************/
/* OPERATING PROFIT */
/********************/

struct OperatingProfitGraphWindow : BaseGraphWindow {
        OperatingProfitGraphWindow(WindowDesc *desc, WindowNumber window_number) :
                        BaseGraphWindow(desc, WID_CV_GRAPH, STR_JUST_CURRENCY_SHORT)
        {
                this->InitializeWindow(window_number);
        }

        virtual OverflowSafeInt64 GetGraphData(const Company *c, int j)
        {
                return c->old_economy[j].income + c->old_economy[j].expenses;
        }
};

static const NWidgetPart _nested_operating_profit_widgets[] = {
        NWidget(NWID_HORIZONTAL),
                NWidget(WWT_CLOSEBOX, COLOUR_GREY),
                NWidget(WWT_CAPTION, COLOUR_GREY), SetDataTip(STR_GRAPH_OPERATING_PROFIT_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
                NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CV_KEY_BUTTON), SetMinimalSize(50, 0), SetMinimalTextLines(1, WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM + 2), SetDataTip(STR_GRAPH_KEY_BUTTON, STR_GRAPH_KEY_TOOLTIP),
                NWidget(WWT_SHADEBOX, COLOUR_GREY),
                NWidget(WWT_DEFSIZEBOX, COLOUR_GREY),
                NWidget(WWT_STICKYBOX, COLOUR_GREY),
        EndContainer(),
        NWidget(WWT_PANEL, COLOUR_GREY, WID_CV_BACKGROUND),
                NWidget(NWID_HORIZONTAL),
                        NWidget(WWT_EMPTY, COLOUR_GREY, WID_CV_GRAPH), SetMinimalSize(576, 160), SetFill(1, 1), SetResize(1, 1),
                        NWidget(NWID_VERTICAL),
                                NWidget(NWID_SPACER), SetFill(0, 1), SetResize(0, 1),
                                NWidget(WWT_RESIZEBOX, COLOUR_GREY, WID_CV_RESIZE),
                        EndContainer(),
                EndContainer(),
        EndContainer(),
};

static WindowDesc _operating_profit_desc(
        WDP_AUTO, "graph_operating_profit", 0, 0,
        WC_OPERATING_PROFIT, WC_NONE,
        0,
        _nested_operating_profit_widgets, lengthof(_nested_operating_profit_widgets)
);


void ShowOperatingProfitGraph()
{
        AllocateWindowDescFront<OperatingProfitGraphWindow>(&_operating_profit_desc, 0);
}


/****************/
/* INCOME GRAPH */
/****************/

struct IncomeGraphWindow : BaseGraphWindow {
        IncomeGraphWindow(WindowDesc *desc, WindowNumber window_number) :
                        BaseGraphWindow(desc, WID_CV_GRAPH, STR_JUST_CURRENCY_SHORT)
        {
                this->InitializeWindow(window_number);
        }

        virtual OverflowSafeInt64 GetGraphData(const Company *c, int j)
        {
                return c->old_economy[j].income;
        }
};

static const NWidgetPart _nested_income_graph_widgets[] = {
        NWidget(NWID_HORIZONTAL),
                NWidget(WWT_CLOSEBOX, COLOUR_GREY),
                NWidget(WWT_CAPTION, COLOUR_GREY), SetDataTip(STR_GRAPH_INCOME_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
                NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CV_KEY_BUTTON), SetMinimalSize(50, 0), SetMinimalTextLines(1, WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM + 2), SetDataTip(STR_GRAPH_KEY_BUTTON, STR_GRAPH_KEY_TOOLTIP),
                NWidget(WWT_SHADEBOX, COLOUR_GREY),
                NWidget(WWT_DEFSIZEBOX, COLOUR_GREY),
                NWidget(WWT_STICKYBOX, COLOUR_GREY),
        EndContainer(),
        NWidget(WWT_PANEL, COLOUR_GREY, WID_CV_BACKGROUND),
                NWidget(NWID_HORIZONTAL),
                        NWidget(WWT_EMPTY, COLOUR_GREY, WID_CV_GRAPH), SetMinimalSize(576, 128), SetFill(1, 1), SetResize(1, 1),
                        NWidget(NWID_VERTICAL),
                                NWidget(NWID_SPACER), SetFill(0, 1), SetResize(0, 1),
                                NWidget(WWT_RESIZEBOX, COLOUR_GREY, WID_CV_RESIZE),
                        EndContainer(),
                EndContainer(),
        EndContainer(),
};

static WindowDesc _income_graph_desc(
        WDP_AUTO, "graph_income", 0, 0,
        WC_INCOME_GRAPH, WC_NONE,
        0,
        _nested_income_graph_widgets, lengthof(_nested_income_graph_widgets)
);

void ShowIncomeGraph()
{
        AllocateWindowDescFront<IncomeGraphWindow>(&_income_graph_desc, 0);
}

/*******************/
/* DELIVERED CARGO */
/*******************/

struct DeliveredCargoGraphWindow : BaseGraphWindow {
        DeliveredCargoGraphWindow(WindowDesc *desc, WindowNumber window_number) :
                        BaseGraphWindow(desc, WID_CV_GRAPH, STR_JUST_COMMA)
        {
                this->InitializeWindow(window_number);
        }

        virtual OverflowSafeInt64 GetGraphData(const Company *c, int j)
        {
                return c->old_economy[j].delivered_cargo.GetSum<OverflowSafeInt64>();
        }
};

static const NWidgetPart _nested_delivered_cargo_graph_widgets[] = {
        NWidget(NWID_HORIZONTAL),
                NWidget(WWT_CLOSEBOX, COLOUR_GREY),
                NWidget(WWT_CAPTION, COLOUR_GREY), SetDataTip(STR_GRAPH_CARGO_DELIVERED_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
                NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CV_KEY_BUTTON), SetMinimalSize(50, 0), SetMinimalTextLines(1, WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM + 2), SetDataTip(STR_GRAPH_KEY_BUTTON, STR_GRAPH_KEY_TOOLTIP),
                NWidget(WWT_SHADEBOX, COLOUR_GREY),
                NWidget(WWT_DEFSIZEBOX, COLOUR_GREY),
                NWidget(WWT_STICKYBOX, COLOUR_GREY),
        EndContainer(),
        NWidget(WWT_PANEL, COLOUR_GREY, WID_CV_BACKGROUND),
                NWidget(NWID_HORIZONTAL),
                        NWidget(WWT_EMPTY, COLOUR_GREY, WID_CV_GRAPH), SetMinimalSize(576, 128), SetFill(1, 1), SetResize(1, 1),
                        NWidget(NWID_VERTICAL),
                                NWidget(NWID_SPACER), SetFill(0, 1), SetResize(0, 1),
                                NWidget(WWT_RESIZEBOX, COLOUR_GREY, WID_CV_RESIZE),
                        EndContainer(),
                EndContainer(),
        EndContainer(),
};

static WindowDesc _delivered_cargo_graph_desc(
        WDP_AUTO, "graph_delivered_cargo", 0, 0,
        WC_DELIVERED_CARGO, WC_NONE,
        0,
        _nested_delivered_cargo_graph_widgets, lengthof(_nested_delivered_cargo_graph_widgets)
);

void ShowDeliveredCargoGraph()
{
        AllocateWindowDescFront<DeliveredCargoGraphWindow>(&_delivered_cargo_graph_desc, 0);
}

/***********************/
/* PERFORMANCE HISTORY */
/***********************/

struct PerformanceHistoryGraphWindow : BaseGraphWindow {
        PerformanceHistoryGraphWindow(WindowDesc *desc, WindowNumber window_number) :
                        BaseGraphWindow(desc, WID_PHG_GRAPH, STR_JUST_COMMA)
        {
                this->InitializeWindow(window_number);
        }

        virtual OverflowSafeInt64 GetGraphData(const Company *c, int j)
        {
                return c->old_economy[j].performance_history;
        }

        virtual void OnClick(Point pt, int widget, int click_count)
        {
                if (widget == WID_PHG_DETAILED_PERFORMANCE) ShowPerformanceRatingDetail();
                this->BaseGraphWindow::OnClick(pt, widget, click_count);
        }
};

static const NWidgetPart _nested_performance_history_widgets[] = {
        NWidget(NWID_HORIZONTAL),
                NWidget(WWT_CLOSEBOX, COLOUR_GREY),
                NWidget(WWT_CAPTION, COLOUR_GREY), SetDataTip(STR_GRAPH_COMPANY_PERFORMANCE_RATINGS_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
                NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_PHG_DETAILED_PERFORMANCE), SetMinimalSize(50, 0), SetMinimalTextLines(1, WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM + 2), SetDataTip(STR_PERFORMANCE_DETAIL_KEY, STR_GRAPH_PERFORMANCE_DETAIL_TOOLTIP),
                NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_PHG_KEY), SetMinimalSize(50, 0), SetMinimalTextLines(1, WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM + 2), SetDataTip(STR_GRAPH_KEY_BUTTON, STR_GRAPH_KEY_TOOLTIP),
                NWidget(WWT_SHADEBOX, COLOUR_GREY),
                NWidget(WWT_DEFSIZEBOX, COLOUR_GREY),
                NWidget(WWT_STICKYBOX, COLOUR_GREY),
        EndContainer(),
        NWidget(WWT_PANEL, COLOUR_GREY, WID_PHG_BACKGROUND),
                NWidget(NWID_HORIZONTAL),
                        NWidget(WWT_EMPTY, COLOUR_GREY, WID_PHG_GRAPH), SetMinimalSize(576, 224), SetFill(1, 1), SetResize(1, 1),
                        NWidget(NWID_VERTICAL),
                                NWidget(NWID_SPACER), SetFill(0, 1), SetResize(0, 1),
                                NWidget(WWT_RESIZEBOX, COLOUR_GREY, WID_PHG_RESIZE),
                        EndContainer(),
                EndContainer(),
        EndContainer(),
};

static WindowDesc _performance_history_desc(
        WDP_AUTO, "graph_performance", 0, 0,
        WC_PERFORMANCE_HISTORY, WC_NONE,
        0,
        _nested_performance_history_widgets, lengthof(_nested_performance_history_widgets)
);

void ShowPerformanceHistoryGraph()
{
        AllocateWindowDescFront<PerformanceHistoryGraphWindow>(&_performance_history_desc, 0);
}

/*****************/
/* COMPANY VALUE */
/*****************/

struct CompanyValueGraphWindow : BaseGraphWindow {
        CompanyValueGraphWindow(WindowDesc *desc, WindowNumber window_number) :
                        BaseGraphWindow(desc, WID_CV_GRAPH, STR_JUST_CURRENCY_SHORT)
        {
                this->InitializeWindow(window_number);
        }

        virtual OverflowSafeInt64 GetGraphData(const Company *c, int j)
        {
                return c->old_economy[j].company_value;
        }
};

static const NWidgetPart _nested_company_value_graph_widgets[] = {
        NWidget(NWID_HORIZONTAL),
                NWidget(WWT_CLOSEBOX, COLOUR_GREY),
                NWidget(WWT_CAPTION, COLOUR_GREY), SetDataTip(STR_GRAPH_COMPANY_VALUES_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
                NWidget(WWT_PUSHTXTBTN, COLOUR_GREY, WID_CV_KEY_BUTTON), SetMinimalSize(50, 0), SetMinimalTextLines(1, WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM + 2), SetDataTip(STR_GRAPH_KEY_BUTTON, STR_GRAPH_KEY_TOOLTIP),
                NWidget(WWT_SHADEBOX, COLOUR_GREY),
                NWidget(WWT_DEFSIZEBOX, COLOUR_GREY),
                NWidget(WWT_STICKYBOX, COLOUR_GREY),
        EndContainer(),
        NWidget(WWT_PANEL, COLOUR_GREY, WID_CV_BACKGROUND),
                NWidget(NWID_HORIZONTAL),
                        NWidget(WWT_EMPTY, COLOUR_GREY, WID_CV_GRAPH), SetMinimalSize(576, 224), SetFill(1, 1), SetResize(1, 1),
                        NWidget(NWID_VERTICAL),
                                NWidget(NWID_SPACER), SetFill(0, 1), SetResize(0, 1),
                                NWidget(WWT_RESIZEBOX, COLOUR_GREY, WID_CV_RESIZE),
                        EndContainer(),
                EndContainer(),
        EndContainer(),
};

static WindowDesc _company_value_graph_desc(
        WDP_AUTO, "graph_company_value", 0, 0,
        WC_COMPANY_VALUE, WC_NONE,
        0,
        _nested_company_value_graph_widgets, lengthof(_nested_company_value_graph_widgets)
);

void ShowCompanyValueGraph()
{
        AllocateWindowDescFront<CompanyValueGraphWindow>(&_company_value_graph_desc, 0);
}

/*****************/
/* PAYMENT RATES */
/*****************/

struct PaymentRatesGraphWindow : BaseGraphWindow {
        bool first_init; ///< This value is true until the first initialization of the window has finished.
        PaymentRatesGraphWindow(WindowDesc *desc, WindowNumber window_number) :
                        BaseGraphWindow(desc, WID_CPR_GRAPH, STR_JUST_CURRENCY_SHORT)
        {
                this->first_init = true;
                this->num_on_x_axis = 20;
                this->num_vert_lines = 20;
                this->month = 0xFF;
                this->x_values_start     = 10;
                this->x_values_increment = 10;

                /* Initialise the dataset */
                this->OnHundredthTick();

                this->InitNested(window_number);

                this->UpdateLoweredWidgets();
        }

        virtual void OnInit()
        {
                /* UpdateLoweredWidgets needs to be called after a language or NewGRF change, but it can't be called before
                 * InitNested is done. On the first init these functions are called in the correct order by the constructor. */
                if (!this->first_init) {
                        /* Initialise the dataset */
                        this->OnHundredthTick();
                        this->UpdateLoweredWidgets();
                }
                this->first_init = false;
        }

        void UpdateExcludedData()
        {
                this->excluded_data = 0;

                int i = 0;
                const CargoSpec *cs;
                FOR_ALL_SORTED_STANDARD_CARGOSPECS(cs) {
                        if (HasBit(_legend_excluded_cargo, cs->Index())) SetBit(this->excluded_data, i);
                        i++;
                }
        }

        void UpdateLoweredWidgets()
        {
                for (int i = 0; i < _sorted_standard_cargo_specs_size; i++) {
                        this->SetWidgetLoweredState(WID_CPR_CARGO_FIRST + i, !HasBit(this->excluded_data, i));
                }
        }

        virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
        {
                if (widget < WID_CPR_CARGO_FIRST) {
                        BaseGraphWindow::UpdateWidgetSize(widget, size, padding, fill, resize);
                        return;
                }

                const CargoSpec *cs = _sorted_cargo_specs[widget - WID_CPR_CARGO_FIRST];
                SetDParam(0, cs->name);
                Dimension d = GetStringBoundingBox(STR_GRAPH_CARGO_PAYMENT_CARGO);
                d.width += 14; // colour field
                d.width += WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT;
                d.height += WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM;
                *size = maxdim(d, *size);
        }

        virtual void DrawWidget(const Rect &r, int widget) const
        {
                if (widget < WID_CPR_CARGO_FIRST) {
                        BaseGraphWindow::DrawWidget(r, widget);
                        return;
                }

                const CargoSpec *cs = _sorted_cargo_specs[widget - WID_CPR_CARGO_FIRST];
                bool rtl = _current_text_dir == TD_RTL;

                /* Since the buttons have no text, no images,
                 * both the text and the coloured box have to be manually painted.
                 * clk_dif will move one pixel down and one pixel to the right
                 * when the button is clicked */
                byte clk_dif = this->IsWidgetLowered(widget) ? 1 : 0;
                int x = r.left + WD_FRAMERECT_LEFT;
                int y = r.top;

                int rect_x = clk_dif + (rtl ? r.right - 12 : r.left + WD_FRAMERECT_LEFT);

                GfxFillRect(rect_x, y + clk_dif, rect_x + 8, y + 5 + clk_dif, PC_BLACK);
                GfxFillRect(rect_x + 1, y + 1 + clk_dif, rect_x + 7, y + 4 + clk_dif, cs->legend_colour);
                SetDParam(0, cs->name);
                DrawString(rtl ? r.left : x + 14 + clk_dif, (rtl ? r.right - 14 + clk_dif : r.right), y + clk_dif, STR_GRAPH_CARGO_PAYMENT_CARGO);
        }

        virtual void OnClick(Point pt, int widget, int click_count)
        {
                switch (widget) {
                        case WID_CPR_ENABLE_CARGOES:
                                /* Remove all cargoes from the excluded lists. */
                                _legend_excluded_cargo = 0;
                                this->excluded_data = 0;
                                this->UpdateLoweredWidgets();
                                this->SetDirty();
                                break;

                        case WID_CPR_DISABLE_CARGOES: {
                                /* Add all cargoes to the excluded lists. */
                                int i = 0;
                                const CargoSpec *cs;
                                FOR_ALL_SORTED_STANDARD_CARGOSPECS(cs) {
                                        SetBit(_legend_excluded_cargo, cs->Index());
                                        SetBit(this->excluded_data, i);
                                        i++;
                                }
                                this->UpdateLoweredWidgets();
                                this->SetDirty();
                                break;
                        }

                        default:
                                if (widget >= WID_CPR_CARGO_FIRST) {
                                        int i = widget - WID_CPR_CARGO_FIRST;
                                        ToggleBit(_legend_excluded_cargo, _sorted_cargo_specs[i]->Index());
                                        this->ToggleWidgetLoweredState(widget);
                                        this->UpdateExcludedData();
                                        this->SetDirty();
                                }
                                break;
                }
        }

        virtual void OnTick()
        {
                /* Override default OnTick */
        }

        /**
         * Some data on this window has become invalid.
         * @param data Information about the changed data.
         * @param gui_scope Whether the call is done from GUI scope. You may not do everything when not in GUI scope. See #InvalidateWindowData() for details.
         */
        virtual void OnInvalidateData(int data = 0, bool gui_scope = true)
        {
                if (!gui_scope) return;
                this->OnHundredthTick();
        }

        virtual void OnHundredthTick()
        {
                this->UpdateExcludedData();

                int i = 0;
                const CargoSpec *cs;
                FOR_ALL_SORTED_STANDARD_CARGOSPECS(cs) {
                        this->colours[i] = cs->legend_colour;
                        for (uint j = 0; j != 20; j++) {
                                this->cost[i][j] = GetTransportedGoodsIncome(10, 20, j * 4 + 4, cs->Index());
                        }
                        i++;
                }
                this->num_dataset = i;
        }
};

/** Construct the row containing the digit keys. */
static NWidgetBase *MakeCargoButtons(int *biggest_index)
{
        NWidgetVertical *ver = new NWidgetVertical;

        for (int i = 0; i < _sorted_standard_cargo_specs_size; i++) {
                NWidgetBackground *leaf = new NWidgetBackground(WWT_PANEL, COLOUR_ORANGE, WID_CPR_CARGO_FIRST + i, NULL);
                leaf->tool_tip = STR_GRAPH_CARGO_PAYMENT_TOGGLE_CARGO;
                leaf->SetFill(1, 0);
                leaf->SetLowered(true);
                ver->Add(leaf);
        }
        *biggest_index = WID_CPR_CARGO_FIRST + _sorted_standard_cargo_specs_size - 1;
        return ver;
}


static const NWidgetPart _nested_cargo_payment_rates_widgets[] = {
        NWidget(NWID_HORIZONTAL),
                NWidget(WWT_CLOSEBOX, COLOUR_GREY),
                NWidget(WWT_CAPTION, COLOUR_GREY), SetDataTip(STR_GRAPH_CARGO_PAYMENT_RATES_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, WID_CPR_BACKGROUND), SetMinimalSize(568, 128),
                NWidget(NWID_HORIZONTAL),
                        NWidget(NWID_SPACER), SetFill(1, 0), SetResize(1, 0),
                        NWidget(WWT_TEXT, COLOUR_GREY, WID_CPR_HEADER), SetMinimalSize(0, 6), SetPadding(2, 0, 2, 0), SetDataTip(STR_GRAPH_CARGO_PAYMENT_RATES_TITLE, STR_NULL),
                        NWidget(NWID_SPACER), SetFill(1, 0), SetResize(1, 0),
                EndContainer(),
                NWidget(NWID_HORIZONTAL),
                        NWidget(WWT_EMPTY, COLOUR_GREY, WID_CPR_GRAPH), SetMinimalSize(495, 0), SetFill(1, 1), SetResize(1, 1),
                        NWidget(NWID_VERTICAL),
                                NWidget(NWID_SPACER), SetMinimalSize(0, 24), SetFill(0, 0), SetResize(0, 1),
                                NWidget(WWT_PUSHTXTBTN, COLOUR_ORANGE, WID_CPR_ENABLE_CARGOES), SetDataTip(STR_GRAPH_CARGO_ENABLE_ALL, STR_GRAPH_CARGO_TOOLTIP_ENABLE_ALL), SetFill(1, 0),
                                NWidget(WWT_PUSHTXTBTN, COLOUR_ORANGE, WID_CPR_DISABLE_CARGOES), SetDataTip(STR_GRAPH_CARGO_DISABLE_ALL, STR_GRAPH_CARGO_TOOLTIP_DISABLE_ALL), SetFill(1, 0),
                                NWidget(NWID_SPACER), SetMinimalSize(0, 4),
                                NWidgetFunction(MakeCargoButtons),
                                NWidget(NWID_SPACER), SetMinimalSize(0, 24), SetFill(0, 1), SetResize(0, 1),
                        EndContainer(),
                        NWidget(NWID_SPACER), SetMinimalSize(5, 0), SetFill(0, 1), SetResize(0, 1),
                EndContainer(),
                NWidget(NWID_HORIZONTAL),
                        NWidget(NWID_SPACER), SetMinimalSize(WD_RESIZEBOX_WIDTH, 0), SetFill(1, 0), SetResize(1, 0),
                        NWidget(WWT_TEXT, COLOUR_GREY, WID_CPR_FOOTER), SetMinimalSize(0, 6), SetPadding(2, 0, 2, 0), SetDataTip(STR_GRAPH_CARGO_PAYMENT_RATES_X_LABEL, STR_NULL),
                        NWidget(NWID_SPACER), SetFill(1, 0), SetResize(1, 0),
                        NWidget(WWT_RESIZEBOX, COLOUR_GREY, WID_CPR_RESIZE),
                EndContainer(),
        EndContainer(),
};

static WindowDesc _cargo_payment_rates_desc(
        WDP_AUTO, "graph_cargo_payment_rates", 0, 0,
        WC_PAYMENT_RATES, WC_NONE,
        0,
        _nested_cargo_payment_rates_widgets, lengthof(_nested_cargo_payment_rates_widgets)
);


void ShowCargoPaymentRates()
{
        AllocateWindowDescFront<PaymentRatesGraphWindow>(&_cargo_payment_rates_desc, 0);
}

/************************/
/* COMPANY LEAGUE TABLE */
/************************/

static const StringID _performance_titles[] = {
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_ENGINEER,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_ENGINEER,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_TRAFFIC_MANAGER,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_TRAFFIC_MANAGER,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_TRANSPORT_COORDINATOR,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_TRANSPORT_COORDINATOR,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_ROUTE_SUPERVISOR,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_ROUTE_SUPERVISOR,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_DIRECTOR,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_DIRECTOR,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_CHIEF_EXECUTIVE,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_CHIEF_EXECUTIVE,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_CHAIRMAN,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_CHAIRMAN,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_PRESIDENT,
        STR_COMPANY_LEAGUE_PERFORMANCE_TITLE_TYCOON,
};

static inline StringID GetPerformanceTitleFromValue(uint value)
{
        return _performance_titles[minu(value, 1000) >> 6];
}

class CompanyLeagueWindow : public Window {
private:
        GUIList<const Company*> companies;
        uint ordinal_width; ///< The width of the ordinal number
        uint text_width;    ///< The width of the actual text
        uint icon_width;    ///< The width of the company icon
        int line_height;    ///< Height of the text lines

        /**
         * (Re)Build the company league list
         */
        void BuildCompanyList()
        {
                if (!this->companies.NeedRebuild()) return;

                this->companies.Clear();

                const Company *c;
                FOR_ALL_COMPANIES(c) {
                        *this->companies.Append() = c;
                }

                this->companies.Compact();
                this->companies.RebuildDone();
        }

        /** Sort the company league by performance history */
        static int CDECL PerformanceSorter(const Company * const *c1, const Company * const *c2)
        {
                return (*c2)->old_economy[0].performance_history - (*c1)->old_economy[0].performance_history;
        }

public:
        CompanyLeagueWindow(WindowDesc *desc, WindowNumber window_number) : Window(desc)
        {
                this->InitNested(window_number);
                this->companies.ForceRebuild();
                this->companies.NeedResort();
        }

        virtual void OnPaint()
        {
                this->BuildCompanyList();
                this->companies.Sort(&PerformanceSorter);

                this->DrawWidgets();
        }

        virtual void DrawWidget(const Rect &r, int widget) const
        {
                if (widget != WID_CL_BACKGROUND) return;

                int icon_y_offset = 1 + (FONT_HEIGHT_NORMAL - this->line_height) / 2;
                uint y = r.top + WD_FRAMERECT_TOP - icon_y_offset;

                bool rtl = _current_text_dir == TD_RTL;
                uint ordinal_left  = rtl ? r.right - WD_FRAMERECT_LEFT - this->ordinal_width : r.left + WD_FRAMERECT_LEFT;
                uint ordinal_right = rtl ? r.right - WD_FRAMERECT_LEFT : r.left + WD_FRAMERECT_LEFT + this->ordinal_width;
                uint icon_left     = r.left + WD_FRAMERECT_LEFT + WD_FRAMERECT_RIGHT + (rtl ? this->text_width : this->ordinal_width);
                uint text_left     = rtl ? r.left + WD_FRAMERECT_LEFT : r.right - WD_FRAMERECT_LEFT - this->text_width;
                uint text_right    = rtl ? r.left + WD_FRAMERECT_LEFT + this->text_width : r.right - WD_FRAMERECT_LEFT;

                for (uint i = 0; i != this->companies.Length(); i++) {
                        const Company *c = this->companies[i];
                        DrawString(ordinal_left, ordinal_right, y, i + STR_ORDINAL_NUMBER_1ST, i == 0 ? TC_WHITE : TC_YELLOW);

                        DrawCompanyIcon(c->index, icon_left, y + icon_y_offset);

                        SetDParam(0, c->index);
                        SetDParam(1, c->index);
                        SetDParam(2, GetPerformanceTitleFromValue(c->old_economy[0].performance_history));
                        DrawString(text_left, text_right, y, STR_COMPANY_LEAGUE_COMPANY_NAME);
                        y += this->line_height;
                }
        }

        virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
        {
                if (widget != WID_CL_BACKGROUND) return;

                this->ordinal_width = 0;
                for (uint i = 0; i < MAX_COMPANIES; i++) {
                        this->ordinal_width = max(this->ordinal_width, GetStringBoundingBox(STR_ORDINAL_NUMBER_1ST + i).width);
                }
                this->ordinal_width += 5; // Keep some extra spacing

                uint widest_width = 0;
                uint widest_title = 0;
                for (uint i = 0; i < lengthof(_performance_titles); i++) {
                        uint width = GetStringBoundingBox(_performance_titles[i]).width;
                        if (width > widest_width) {
                                widest_title = i;
                                widest_width = width;
                        }
                }

                Dimension d = GetSpriteSize(SPR_COMPANY_ICON);
                this->icon_width = d.width + 2;
                this->line_height = max<int>(d.height + 2, FONT_HEIGHT_NORMAL);

                const Company *c;
                FOR_ALL_COMPANIES(c) {
                        SetDParam(0, c->index);
                        SetDParam(1, c->index);
                        SetDParam(2, _performance_titles[widest_title]);
                        widest_width = max(widest_width, GetStringBoundingBox(STR_COMPANY_LEAGUE_COMPANY_NAME).width);
                }

                this->text_width = widest_width + 30; // Keep some extra spacing

                size->width = WD_FRAMERECT_LEFT + this->ordinal_width + WD_FRAMERECT_RIGHT + this->icon_width + WD_FRAMERECT_LEFT + this->text_width + WD_FRAMERECT_RIGHT;
                size->height = WD_FRAMERECT_TOP + this->line_height * MAX_COMPANIES + WD_FRAMERECT_BOTTOM;
        }


        virtual void OnTick()
        {
                if (this->companies.NeedResort()) {
                        this->SetDirty();
                }
        }

        /**
         * Some data on this window has become invalid.
         * @param data Information about the changed data.
         * @param gui_scope Whether the call is done from GUI scope. You may not do everything when not in GUI scope. See #InvalidateWindowData() for details.
         */
        virtual void OnInvalidateData(int data = 0, bool gui_scope = true)
        {
                if (data == 0) {
                        /* This needs to be done in command-scope to enforce rebuilding before resorting invalid data */
                        this->companies.ForceRebuild();
                } else {
                        this->companies.ForceResort();
                }
        }
};

static const NWidgetPart _nested_company_league_widgets[] = {
        NWidget(NWID_HORIZONTAL),
                NWidget(WWT_CLOSEBOX, COLOUR_GREY),
                NWidget(WWT_CAPTION, COLOUR_GREY), SetDataTip(STR_COMPANY_LEAGUE_TABLE_CAPTION, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
                NWidget(WWT_SHADEBOX, COLOUR_GREY),
                NWidget(WWT_STICKYBOX, COLOUR_GREY),
        EndContainer(),
        NWidget(WWT_PANEL, COLOUR_GREY, WID_CL_BACKGROUND), SetMinimalSize(400, 0), SetMinimalTextLines(15, WD_FRAMERECT_TOP + WD_FRAMERECT_BOTTOM),
};

static WindowDesc _company_league_desc(
        WDP_AUTO, "league", 0, 0,
        WC_COMPANY_LEAGUE, WC_NONE,
        0,
        _nested_company_league_widgets, lengthof(_nested_company_league_widgets)
);

void ShowCompanyLeagueTable()
{
        AllocateWindowDescFront<CompanyLeagueWindow>(&_company_league_desc, 0);
}

/*****************************/
/* PERFORMANCE RATING DETAIL */
/*****************************/

struct PerformanceRatingDetailWindow : Window {
        static CompanyID company;
        int timeout;

        PerformanceRatingDetailWindow(WindowDesc *desc, WindowNumber window_number) : Window(desc)
        {
                this->UpdateCompanyStats();

                this->InitNested(window_number);
                this->OnInvalidateData(INVALID_COMPANY);
        }

        void UpdateCompanyStats()
        {
                /* Update all company stats with the current data
                 * (this is because _score_info is not saved to a savegame) */
                Company *c;
                FOR_ALL_COMPANIES(c) {
                        UpdateCompanyRatingAndValue(c, false);
                }

                this->timeout = DAY_TICKS * 5;
        }

        uint score_info_left;
        uint score_info_right;
        uint bar_left;
        uint bar_right;
        uint bar_width;
        uint bar_height;
        uint score_detail_left;
        uint score_detail_right;

        virtual void UpdateWidgetSize(int widget, Dimension *size, const Dimension &padding, Dimension *fill, Dimension *resize)
        {
                switch (widget) {
                        case WID_PRD_SCORE_FIRST:
                                this->bar_height = FONT_HEIGHT_NORMAL + 4;
                                size->height = this->bar_height + 2 * WD_MATRIX_TOP;

                                uint score_info_width = 0;
                                for (uint i = SCORE_BEGIN; i < SCORE_END; i++) {
                                        score_info_width = max(score_info_width, GetStringBoundingBox(STR_PERFORMANCE_DETAIL_VEHICLES + i).width);
                                }
                                SetDParamMaxValue(0, 1000);
                                score_info_width += GetStringBoundingBox(STR_BLACK_COMMA).width + WD_FRAMERECT_LEFT;

                                SetDParamMaxValue(0, 100);
                                this->bar_width = GetStringBoundingBox(STR_PERFORMANCE_DETAIL_PERCENT).width + 20; // Wide bars!

                                /* At this number we are roughly at the max; it can become wider,
                                 * but then you need at 1000 times more money. At that time you're
                                 * not that interested anymore in the last few digits anyway.
                                 * The 500 is because 999 999 500 to 999 999 999 are rounded to
                                 * 1 000 M, and not 999 999 k. Use negative numbers to account for
                                 * the negative income/amount of money etc. as well. */
                                int max = -(999999999 - 500);

                                /* Scale max for the display currency. Prior to rendering the value
                                 * is converted into the display currency, which may cause it to
                                 * raise significantly. We need to compensate for that since {{CURRCOMPACT}}
                                 * is used, which can produce quite short renderings of very large
                                 * values. Otherwise the calculated width could be too narrow.
                                 * Note that it doesn't work if there was a currency with an exchange
                                 * rate greater than max.
                                 * When the currency rate is more than 1000, the 999 999 k becomes at
                                 * least 999 999 M which roughly is equally long. Furthermore if the
                                 * exchange rate is that high, 999 999 k is usually not enough anymore
                                 * to show the different currency numbers. */
                                if (_currency->rate < 1000) max /= _currency->rate;
                                SetDParam(0, max);
                                SetDParam(1, max);
                                uint score_detail_width = GetStringBoundingBox(STR_PERFORMANCE_DETAIL_AMOUNT_CURRENCY).width;

                                size->width = 7 + score_info_width + 5 + this->bar_width + 5 + score_detail_width + 7;
                                uint left  = 7;
                                uint right = size->width - 7;

                                bool rtl = _current_text_dir == TD_RTL;
                                this->score_info_left  = rtl ? right - score_info_width : left;
                                this->score_info_right = rtl ? right : left + score_info_width;

                                this->score_detail_left  = rtl ? left : right - score_detail_width;
                                this->score_detail_right = rtl ? left + score_detail_width : right;

                                this->bar_left  = left + (rtl ? score_detail_width : score_info_width) + 5;
                                this->bar_right = this->bar_left + this->bar_width;
                                break;
                }
        }

        virtual void DrawWidget(const Rect &r, int widget) const
        {
                /* No need to draw when there's nothing to draw */
                if (this->company == INVALID_COMPANY) return;

                if (IsInsideMM(widget, WID_PRD_COMPANY_FIRST, WID_PRD_COMPANY_LAST + 1)) {
                        if (this->IsWidgetDisabled(widget)) return;
                        CompanyID cid = (CompanyID)(widget - WID_PRD_COMPANY_FIRST);
                        int offset = (cid == this->company) ? 1 : 0;
                        Dimension sprite_size = GetSpriteSize(SPR_COMPANY_ICON);
                        DrawCompanyIcon(cid, (r.left + r.right - sprite_size.width) / 2 + offset, (r.top + r.bottom - sprite_size.height) / 2 + offset);
                        return;
                }

                if (!IsInsideMM(widget, WID_PRD_SCORE_FIRST, WID_PRD_SCORE_LAST + 1)) return;

                ScoreID score_type = (ScoreID)(widget - WID_PRD_SCORE_FIRST);

                /* The colours used to show how the progress is going */
                int colour_done = _colour_gradient[COLOUR_GREEN][4];
                int colour_notdone = _colour_gradient[COLOUR_RED][4];

                /* Draw all the score parts */
                int val    = _score_part[company][score_type];
                int needed = _score_info[score_type].needed;
                int score  = _score_info[score_type].score;

                /* SCORE_TOTAL has his own rules ;) */
                if (score_type == SCORE_TOTAL) {
                        for (ScoreID i = SCORE_BEGIN; i < SCORE_END; i++) score += _score_info[i].score;
                        needed = SCORE_MAX;
                }

                uint bar_top  = r.top + WD_MATRIX_TOP;
                uint text_top = bar_top + 2;

                DrawString(this->score_info_left, this->score_info_right, text_top, STR_PERFORMANCE_DETAIL_VEHICLES + score_type);

                /* Draw the score */
                SetDParam(0, score);
                DrawString(this->score_info_left, this->score_info_right, text_top, STR_BLACK_COMMA, TC_FROMSTRING, SA_RIGHT);

                /* Calculate the %-bar */
                uint x = Clamp(val, 0, needed) * this->bar_width / needed;
                bool rtl = _current_text_dir == TD_RTL;
                if (rtl) {
                        x = this->bar_right - x;
                } else {
                        x = this->bar_left + x;
                }

                /* Draw the bar */
                if (x != this->bar_left)  GfxFillRect(this->bar_left, bar_top, x, bar_top + this->bar_height, rtl ? colour_notdone : colour_done);
                if (x != this->bar_right) GfxFillRect(x, bar_top, this->bar_right, bar_top + this->bar_height, rtl ? colour_done : colour_notdone);

                /* Draw it */
                SetDParam(0, Clamp(val, 0, needed) * 100 / needed);
                DrawString(this->bar_left, this->bar_right, text_top, STR_PERFORMANCE_DETAIL_PERCENT, TC_FROMSTRING, SA_HOR_CENTER);

                /* SCORE_LOAN is inversed */
                if (score_type == SCORE_LOAN) val = needed - val;

                /* Draw the amount we have against what is needed
                 * For some of them it is in currency format */
                SetDParam(0, val);
                SetDParam(1, needed);
                switch (score_type) {
                        case SCORE_MIN_PROFIT:
                        case SCORE_MIN_INCOME:
                        case SCORE_MAX_INCOME:
                        case SCORE_MONEY:
                        case SCORE_LOAN:
                                DrawString(this->score_detail_left, this->score_detail_right, text_top, STR_PERFORMANCE_DETAIL_AMOUNT_CURRENCY);
                                break;
                        default:
                                DrawString(this->score_detail_left, this->score_detail_right, text_top, STR_PERFORMANCE_DETAIL_AMOUNT_INT);
                }
        }

        virtual void OnClick(Point pt, int widget, int click_count)
        {
                /* Check which button is clicked */
                if (IsInsideMM(widget, WID_PRD_COMPANY_FIRST, WID_PRD_COMPANY_LAST + 1)) {
                        /* Is it no on disable? */
                        if (!this->IsWidgetDisabled(widget)) {
                                this->RaiseWidget(this->company + WID_PRD_COMPANY_FIRST);
                                this->company = (CompanyID)(widget - WID_PRD_COMPANY_FIRST);
                                this->LowerWidget(this->company + WID_PRD_COMPANY_FIRST);
                                this->SetDirty();
                        }
                }
        }

        virtual void OnTick()
        {
                if (_pause_mode != PM_UNPAUSED) return;

                /* Update the company score every 5 days */
                if (--this->timeout == 0) {
                        this->UpdateCompanyStats();
                        this->SetDirty();
                }
        }

        /**
         * Some data on this window has become invalid.
         * @param data the company ID of the company that is going to be removed
         * @param gui_scope Whether the call is done from GUI scope. You may not do everything when not in GUI scope. See #InvalidateWindowData() for details.
         */
        virtual void OnInvalidateData(int data = 0, bool gui_scope = true)
        {
                if (!gui_scope) return;
                /* Disable the companies who are not active */
                for (CompanyID i = COMPANY_FIRST; i < MAX_COMPANIES; i++) {
                        this->SetWidgetDisabledState(i + WID_PRD_COMPANY_FIRST, !Company::IsValidID(i));
                }

                /* Check if the currently selected company is still active. */
                if (this->company != INVALID_COMPANY && !Company::IsValidID(this->company)) {
                        /* Raise the widget for the previous selection. */
                        this->RaiseWidget(this->company + WID_PRD_COMPANY_FIRST);
                        this->company = INVALID_COMPANY;
                }

                if (this->company == INVALID_COMPANY) {
                        const Company *c;
                        FOR_ALL_COMPANIES(c) {
                                this->company = c->index;
                                break;
                        }
                }

                /* Make sure the widget is lowered */
                this->LowerWidget(this->company + WID_PRD_COMPANY_FIRST);
        }
};

CompanyID PerformanceRatingDetailWindow::company = INVALID_COMPANY;

/**
 * Make a vertical list of panels for outputting score details.
 * @param biggest_index Storage for collecting the biggest index used in the returned tree.
 * @return Panel with performance details.
 * @post \c *biggest_index contains the largest used index in the tree.
 */
static NWidgetBase *MakePerformanceDetailPanels(int *biggest_index)
{
        const StringID performance_tips[] = {
                STR_PERFORMANCE_DETAIL_VEHICLES_TOOLTIP,
                STR_PERFORMANCE_DETAIL_STATIONS_TOOLTIP,
                STR_PERFORMANCE_DETAIL_MIN_PROFIT_TOOLTIP,
                STR_PERFORMANCE_DETAIL_MIN_INCOME_TOOLTIP,
                STR_PERFORMANCE_DETAIL_MAX_INCOME_TOOLTIP,
                STR_PERFORMANCE_DETAIL_DELIVERED_TOOLTIP,
                STR_PERFORMANCE_DETAIL_CARGO_TOOLTIP,
                STR_PERFORMANCE_DETAIL_MONEY_TOOLTIP,
                STR_PERFORMANCE_DETAIL_LOAN_TOOLTIP,
                STR_PERFORMANCE_DETAIL_TOTAL_TOOLTIP,
        };

        assert_compile(lengthof(performance_tips) == SCORE_END - SCORE_BEGIN);

        NWidgetVertical *vert = new NWidgetVertical(NC_EQUALSIZE);
        for (int widnum = WID_PRD_SCORE_FIRST; widnum <= WID_PRD_SCORE_LAST; widnum++) {
                NWidgetBackground *panel = new NWidgetBackground(WWT_PANEL, COLOUR_GREY, widnum);
                panel->SetFill(1, 1);
                panel->SetDataTip(0x0, performance_tips[widnum - WID_PRD_SCORE_FIRST]);
                vert->Add(panel);
        }
        *biggest_index = WID_PRD_SCORE_LAST;
        return vert;
}

/** Make a number of rows with buttons for each company for the performance rating detail window. */
NWidgetBase *MakeCompanyButtonRowsGraphGUI(int *biggest_index)
{
        return MakeCompanyButtonRows(biggest_index, WID_PRD_COMPANY_FIRST, WID_PRD_COMPANY_LAST, 8, STR_PERFORMANCE_DETAIL_SELECT_COMPANY_TOOLTIP);
}

static const NWidgetPart _nested_performance_rating_detail_widgets[] = {
        NWidget(NWID_HORIZONTAL),
                NWidget(WWT_CLOSEBOX, COLOUR_GREY),
                NWidget(WWT_CAPTION, COLOUR_GREY), SetDataTip(STR_PERFORMANCE_DETAIL, STR_TOOLTIP_WINDOW_TITLE_DRAG_THIS),
                NWidget(WWT_SHADEBOX, COLOUR_GREY),
                NWidget(WWT_STICKYBOX, COLOUR_GREY),
        EndContainer(),
        NWidget(WWT_PANEL, COLOUR_GREY),
                NWidgetFunction(MakeCompanyButtonRowsGraphGUI), SetPadding(0, 1, 1, 2),
        EndContainer(),
        NWidgetFunction(MakePerformanceDetailPanels),
};

static WindowDesc _performance_rating_detail_desc(
        WDP_AUTO, "league_details", 0, 0,
        WC_PERFORMANCE_DETAIL, WC_NONE,
        0,
        _nested_performance_rating_detail_widgets, lengthof(_nested_performance_rating_detail_widgets)
);

void ShowPerformanceRatingDetail()
{
        AllocateWindowDescFront<PerformanceRatingDetailWindow>(&_performance_rating_detail_desc, 0);
}