diff --git a/src/ai/ai_gui.cpp b/src/ai/ai_gui.cpp new file mode 100644 --- /dev/null +++ b/src/ai/ai_gui.cpp @@ -0,0 +1,257 @@ +/* $Id$ */ + +/** @file ai_gui.cpp Window for configuring the AIs */ + +#include "../stdafx.h" +#include "../openttd.h" +#include "../gui.h" +#include "../window_gui.h" +#include "../company_func.h" +#include "../company_base.h" +#include "../company_gui.h" +#include "../economy_func.h" +#include "../variables.h" +#include "../cargotype.h" +#include "../strings_func.h" +#include "../core/alloc_func.hpp" +#include "../window_func.h" +#include "../date_func.h" +#include "../gfx_func.h" +#include "../debug.h" +#include "../command_func.h" +#include "../network/network.h" + +#include "ai.hpp" +#include "api/ai_types.hpp" +#include "api/ai_controller.hpp" +#include "api/ai_object.hpp" +#include "api/ai_log.hpp" +#include "ai_info.hpp" + +#include "table/strings.h" +#include "../table/sprites.h" + +struct AIDebugWindow : public Window { + enum AIDebugWindowWidgets { + AID_WIDGET_CLOSEBOX = 0, + AID_WIDGET_CAPTION, + AID_WIDGET_VIEW, + AID_WIDGET_NAME_TEXT, + AID_WIDGET_RELOAD_TOGGLE, + AID_WIDGET_LOG_PANEL, + AID_WIDGET_SCROLLBAR, + AID_WIDGET_UNUSED_1, + AID_WIDGET_UNUSED_2, + AID_WIDGET_UNUSED_3, + AID_WIDGET_UNUSED_4, + AID_WIDGET_UNUSED_5, + AID_WIDGET_UNUSED_6, + AID_WIDGET_UNUSED_7, + + AID_WIDGET_COMPANY_BUTTON_START, + AID_WIDGET_COMPANY_BUTTON_END = AID_WIDGET_COMPANY_BUTTON_START + 14, + }; + + static CompanyID ai_debug_company; + int redraw_timer; + + AIDebugWindow(const WindowDesc *desc, WindowNumber number) : Window(desc, number) + { + /* Disable the companies who are not active or not an AI */ + for (CompanyID i = COMPANY_FIRST; i < MAX_COMPANIES; i++) { + this->SetWidgetDisabledState(i + AID_WIDGET_COMPANY_BUTTON_START, !IsValidCompanyID(i) || !GetCompany(i)->is_ai); + } + this->DisableWidget(AID_WIDGET_RELOAD_TOGGLE); + + this->vscroll.cap = 14; + this->vscroll.pos = 0; + this->resize.step_height = 12; + + if (ai_debug_company != INVALID_COMPANY) this->LowerWidget(ai_debug_company + AID_WIDGET_COMPANY_BUTTON_START); + + this->FindWindowPlacementAndResize(desc); + } + + virtual void OnPaint() + { + /* Check if the currently selected company is still active. */ + if (ai_debug_company == INVALID_COMPANY || !IsValidCompanyID(ai_debug_company)) { + if (ai_debug_company != INVALID_COMPANY) { + /* Raise and disable the widget for the previous selection. */ + this->RaiseWidget(ai_debug_company + AID_WIDGET_COMPANY_BUTTON_START); + this->DisableWidget(ai_debug_company + AID_WIDGET_COMPANY_BUTTON_START); + + ai_debug_company = INVALID_COMPANY; + } + + for (CompanyID i = COMPANY_FIRST; i < MAX_COMPANIES; i++) { + if (IsValidCompanyID(i) && GetCompany(i)->is_ai) { + /* Lower the widget corresponding to this company. */ + this->LowerWidget(i + AID_WIDGET_COMPANY_BUTTON_START); + + ai_debug_company = i; + break; + } + } + } + + /* Update "Reload AI" button */ + this->SetWidgetDisabledState(AID_WIDGET_RELOAD_TOGGLE, ai_debug_company == INVALID_COMPANY); + + /* Draw standard stuff */ + this->DrawWidgets(); + + /* If there are no active companies, don't display anything else. */ + if (ai_debug_company == INVALID_COMPANY) return; + + /* Paint the company icons */ + for (CompanyID i = COMPANY_FIRST; i < MAX_COMPANIES; i++) { + if (!IsValidCompanyID(i) || !GetCompany(i)->is_ai) { + /* Check if we have the company as an active company */ + if (!this->IsWidgetDisabled(i + AID_WIDGET_COMPANY_BUTTON_START)) { + /* Bah, company gone :( */ + this->DisableWidget(i + AID_WIDGET_COMPANY_BUTTON_START); + + /* We need a repaint */ + this->SetDirty(); + } + continue; + } + + /* Check if we have the company marked as inactive */ + if (this->IsWidgetDisabled(i + AID_WIDGET_COMPANY_BUTTON_START)) { + /* New AI! Yippie :p */ + this->EnableWidget(i + AID_WIDGET_COMPANY_BUTTON_START); + + /* We need a repaint */ + this->SetDirty(); + } + + byte x = (i == ai_debug_company) ? 1 : 0; + DrawCompanyIcon(i, (i % 8) * 37 + 13 + x, (i < 8 ? 0 : 13) + 16 + x); + } + + /* Draw the AI name */ + AIInfo *info = GetCompany(ai_debug_company)->ai_info; + assert(info != NULL); + DoDrawString(info->GetName(), 7, 47, TC_BLACK); + + CompanyID old_company = _current_company; + _current_company = ai_debug_company; + AILog::LogData *log = (AILog::LogData *)AIObject::GetLogPointer(); + _current_company = old_company; + + SetVScrollCount(this, (log == NULL) ? 0 : log->used); + this->InvalidateWidget(AID_WIDGET_SCROLLBAR); + if (log == NULL) return; + + int y = 6; + for (int i = this->vscroll.pos; i < (this->vscroll.cap + this->vscroll.pos); i++) { + uint pos = (log->count + log->pos - i) % log->count; + if (log->lines[pos] == NULL) break; + + uint colour; + switch (log->type[pos]) { + case AILog::LOG_SQ_INFO: colour = TC_BLACK; break; + case AILog::LOG_SQ_ERROR: colour = TC_RED; break; + case AILog::LOG_INFO: colour = TC_BLACK; break; + case AILog::LOG_WARNING: colour = TC_YELLOW; break; + case AILog::LOG_ERROR: colour = TC_RED; break; + default: colour = TC_BLACK; break; + } + + DoDrawStringTruncated(log->lines[pos], 7, this->widget[AID_WIDGET_LOG_PANEL].top + y, colour, this->widget[AID_WIDGET_LOG_PANEL].right - this->widget[AID_WIDGET_LOG_PANEL].left - 14); + y += 12; + } + } + + virtual void OnClick(Point pt, int widget) + { + /* Check which button is clicked */ + if (IsInsideMM(widget, AID_WIDGET_COMPANY_BUTTON_START, AID_WIDGET_COMPANY_BUTTON_END + 1)) { + /* Is it no on disable? */ + if (!this->IsWidgetDisabled(widget)) { + this->RaiseWidget(ai_debug_company + AID_WIDGET_COMPANY_BUTTON_START); + ai_debug_company = (CompanyID)(widget - AID_WIDGET_COMPANY_BUTTON_START); + this->LowerWidget(ai_debug_company + AID_WIDGET_COMPANY_BUTTON_START); + this->SetDirty(); + } + } + if (widget == AID_WIDGET_RELOAD_TOGGLE && !this->IsWidgetDisabled(widget)) { + /* First kill the company of the AI, then start a new one. This should start the current AI again */ + DoCommandP(0, 2, ai_debug_company, CMD_COMPANY_CTRL); + DoCommandP(0, 1, 0, CMD_COMPANY_CTRL); + } + } + + virtual void OnTimeout() + { + this->RaiseWidget(AID_WIDGET_RELOAD_TOGGLE); + this->SetDirty(); + } + + virtual void OnInvalidateData(int data = 0) + { + if (data == -1 || ai_debug_company == data) this->SetDirty(); + } + + virtual void OnResize(Point new_size, Point delta) + { + this->vscroll.cap += delta.y / (int)this->resize.step_height; + } +}; + +CompanyID AIDebugWindow::ai_debug_company = INVALID_COMPANY; + +static const Widget _ai_debug_widgets[] = { +{ WWT_CLOSEBOX, RESIZE_NONE, COLOUR_GREY, 0, 10, 0, 13, STR_00C5, STR_018B_CLOSE_WINDOW}, // AID_WIDGET_CLOSEBOX +{ WWT_CAPTION, RESIZE_RIGHT, COLOUR_GREY, 11, 298, 0, 13, STR_AI_DEBUG, STR_018C_WINDOW_TITLE_DRAG_THIS}, // AID_WIDGET_CAPTION +{ WWT_PANEL, RESIZE_RIGHT, COLOUR_GREY, 0, 298, 14, 40, 0x0, STR_NULL}, // AID_WIDGET_VIEW + +{ WWT_PANEL, RESIZE_RIGHT, COLOUR_GREY, 0, 149, 41, 60, 0x0, STR_AI_DEBUG_NAME_TIP}, // AID_WIDGET_NAME_TEXT +{ WWT_PUSHTXTBTN, RESIZE_LR, COLOUR_GREY, 150, 298, 41, 60, STR_AI_DEBUG_RELOAD, STR_AI_DEBUG_RELOAD_TIP}, // AID_WIDGET_RELOAD_TOGGLE +{ WWT_PANEL, RESIZE_RB, COLOUR_GREY, 0, 286, 61, 240, 0x0, STR_NULL}, // AID_WIDGET_LOG_PANEL +{ WWT_SCROLLBAR, RESIZE_LRB, COLOUR_GREY, 287, 298, 61, 228, STR_NULL, STR_0190_SCROLL_BAR_SCROLLS_LIST}, // AID_WIDGET_SCROLLBAR +/* As this is WIP, leave the next few so we can work a bit with the GUI */ +{ WWT_EMPTY, RESIZE_NONE, COLOUR_GREY, 0, 298, 101, 120, 0x0, STR_NULL}, // AID_WIDGET_UNUSED_1 +{ WWT_EMPTY, RESIZE_NONE, COLOUR_GREY, 0, 298, 121, 140, 0x0, STR_NULL}, // AID_WIDGET_UNUSED_2 +{ WWT_EMPTY, RESIZE_NONE, COLOUR_GREY, 0, 298, 141, 160, 0x0, STR_NULL}, // AID_WIDGET_UNUSED_3 +{ WWT_EMPTY, RESIZE_NONE, COLOUR_GREY, 0, 298, 161, 180, 0x0, STR_NULL}, // AID_WIDGET_UNUSED_4 +{ WWT_EMPTY, RESIZE_NONE, COLOUR_GREY, 0, 298, 181, 200, 0x0, STR_NULL}, // AID_WIDGET_UNUSED_5 +{ WWT_EMPTY, RESIZE_NONE, COLOUR_GREY, 0, 298, 201, 220, 0x0, STR_NULL}, // AID_WIDGET_UNUSED_6 +{ WWT_EMPTY, RESIZE_NONE, COLOUR_GREY, 0, 298, 221, 240, 0x0, STR_NULL}, // AID_WIDGET_UNUSED_7 + +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 2, 38, 14, 26, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, // AID_WIDGET_COMPANY_BUTTON_START +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 39, 75, 14, 26, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 76, 112, 14, 26, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 113, 149, 14, 26, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 150, 186, 14, 26, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 187, 223, 14, 26, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 224, 260, 14, 26, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 261, 297, 14, 26, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 2, 38, 27, 39, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 39, 75, 27, 39, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 76, 112, 27, 39, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 113, 149, 27, 39, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 150, 186, 27, 39, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 187, 223, 27, 39, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, +{ WWT_PANEL, RESIZE_NONE, COLOUR_GREY, 224, 260, 27, 39, 0x0, STR_704F_CLICK_HERE_TO_TOGGLE_COMPANY}, // AID_WIDGET_COMPANY_BUTTON_END +{ WWT_RESIZEBOX, RESIZE_LRTB, COLOUR_GREY, 287, 298, 229, 240, STR_NULL, STR_RESIZE_BUTTON}, +{ WIDGETS_END}, +}; + +static const WindowDesc _ai_debug_desc = { + WDP_AUTO, WDP_AUTO, 299, 241, 299, 241, + WC_AI_DEBUG, WC_NONE, + WDF_STD_TOOLTIPS | WDF_STD_BTN | WDF_DEF_WIDGET | WDF_RESIZABLE, + _ai_debug_widgets +}; + +void ShowAIDebugWindow() +{ + if (!_networking || _network_server) { + AllocateWindowDescFront(&_ai_debug_desc, 0); + } else { + ShowErrorMessage(INVALID_STRING_ID, STR_AI_DEBUG_SERVER_ONLY, 0, 0); + } +}