#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>

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

	{

		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;