ObjectWindowOverrides.cpp

#include "ObjectWindowOverrides.h"
#include "MiscWindowOverrides.h"
#include "DialogImposterManager.h"
#include "HallOfFame.h"

namespace cse
{
	namespace uiManager
	{

		std::string ObjectWindowFormListGetColumnText(HWND FormList, TESForm* Form, UInt32 ColumnIndex)
		{
			char Buffer[0x500];

			NMLVDISPINFO CallbackInfo;
			CallbackInfo.hdr.hwndFrom = FormList;
			CallbackInfo.hdr.idFrom = GetDlgCtrlID(FormList);
			CallbackInfo.hdr.code = LVN_GETDISPINFO;

			CallbackInfo.item.mask = LVIF_TEXT;
			CallbackInfo.item.lParam = reinterpret_cast<LPARAM>(Form);
			CallbackInfo.item.pszText = Buffer;
			CallbackInfo.item.cchTextMax = sizeof(Buffer);
			CallbackInfo.item.iSubItem = ColumnIndex;

			TESObjectWindow::TreeEntryInfo Dummy;
			Dummy.formType = Form->formType;

			Dummy.ListViewGetDispInfoCallback(&CallbackInfo);

			return Buffer;
		}

		LRESULT CALLBACK ObjectWindowSubclassProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam,
												  bgsee::WindowSubclassProcCollection::SubclassProcExtraParams* SubclassParams)
		{
			LRESULT DlgProcResult = FALSE;

			if (ObjectWindowImposterManager::Instance.IsImposter(hWnd))
			{
				// the imposter needs to handle its own messages exclusively
				return DlgProcResult;
			}

			HWND FilterEditBox = GetDlgItem(hWnd, IDC_CSEFILTERABLEFORMLIST_FILTEREDIT);
			HWND FormList = GetDlgItem(hWnd, TESObjectWindow::kFormListCtrlID);
			HWND TreeList = GetDlgItem(hWnd, TESObjectWindow::kTreeViewCtrlID);

			switch (uMsg)
			{
			case TESObjectWindow::kWindowMessage_Reload:
				SubclassParams->Out.MarkMessageAsHandled = true;

				// the window subclasser's windows hook has an out-sized performance penalty for
				// insertions in tree view controls (probably also has to do with the way it's used in this dialog)
				// so we have to temporarily suspend the hook when this code is executing
				BGSEEUI->GetInvalidationManager()->Push(TreeList);
				SubclassParams->In.Subclasser->SuspendHooks();
				{
					DlgProcResult = SubclassParams->In.Subclasser->TunnelMessageToOrgWndProc(hWnd, uMsg, wParam, lParam, true);
				}
				SubclassParams->In.Subclasser->ResumeHooks();
				BGSEEUI->GetInvalidationManager()->Pop(TreeList);
				BGSEEUI->GetInvalidationManager()->Redraw(TreeList);

				break;
			case WM_NOTIFY:
				{
					bool Skip = true;

					if (((NMHDR*)lParam)->code == NM_KEYDOWN && ((NMKEY*)lParam)->nVKey == VK_F5)
						Skip = false;
					else if (((NMHDR*)lParam)->code == LVN_KEYDOWN && ((NMLVKEYDOWN*)lParam)->wVKey == VK_F5)
						Skip = false;
					else if (((NMHDR*)lParam)->code == TVN_KEYDOWN && ((NMTVKEYDOWN*)lParam)->wVKey == VK_F5)
						Skip = false;

					if (Skip)
						break;

					SubclassParams->Out.MarkMessageAsHandled = true;
					ObjectWindowImposterManager::Instance.RefreshImposters();
					SendMessage(hWnd, TESDialog::kWindowMessage_Refresh, NULL, NULL);
				}
			case WM_ACTIVATE:
				ObjectWindowImposterManager::Instance.HandleObjectWindowActivating(hWnd, uMsg, wParam, lParam);
				SubclassParams->Out.MarkMessageAsHandled = true;

				break;
			case WM_CLOSE:
				SendMessage(*TESCSMain::WindowHandle, WM_COMMAND, TESCSMain::kMainMenu_View_ObjectWindow, NULL);
				SubclassParams->Out.MarkMessageAsHandled = true;

				break;
			case TESDialog::kWindowMessage_Destroy:
			case WM_DESTROY:
				{
					FilterableFormListManager::Instance.Unregister(FilterEditBox);
					ObjectWindowImposterManager::Instance.DestroyImposters();
					TESObjectWindow::PrimaryObjectWindowHandle = nullptr;
				}

				break;
			case WM_INITDIALOG:
				{
					SME_ASSERT(FilterEditBox);

					uiManager::FilterableFormListManager::InitParams Params;
					Params.ParentWindow = hWnd;
					Params.FormListView = FormList;
					Params.FilterEditBox = FilterEditBox;
					Params.FilterLabel = GetDlgItem(hWnd, IDC_CSEFILTERABLEFORMLIST_FILTERLBL);
					Params.ColumnTextCallback = ObjectWindowFormListGetColumnText;
					uiManager::FilterableFormListManager::Instance.Register(Params);

					std::string WndTitle = "Object Window";
					if (settings::general::kShowHallOfFameMembersInTitleBar().i != hallOfFame::kDisplayESMember_None)
					{
						hallOfFame::GetRandomESMember(WndTitle);
						WndTitle += " Object Window";
					}
					SetWindowText(hWnd, WndTitle.c_str());

					TESObjectWindow::PrimaryObjectWindowHandle = hWnd;
				}

				break;
			case WM_SIZE:
				ObjectWindowImposterManager::Instance.HandleObjectWindowSizing(hWnd, uMsg, wParam, lParam);
				SubclassParams->Out.MarkMessageAsHandled = true;

				break;
			}

			if (SubclassParams->Out.MarkMessageAsHandled == false && FilterableFormListManager::Instance.HandleMessages(FilterEditBox, uMsg, wParam, lParam))
			{
				HTREEITEM Selection = TreeView_GetSelection(TreeList);

				TreeView_SelectItem(TreeList, nullptr);
				TreeView_SelectItem(TreeList, Selection);
			}

			return DlgProcResult;
		}

		void InitializeObjectWindowOverrides()
		{
			BGSEEUI->GetSubclasser()->RegisterSubclassForDialogResourceTemplate(TESDialog::kDialogTemplate_ObjectWindow,
				uiManager::ObjectWindowSubclassProc);
			BGSEEUI->GetSubclasser()->RegisterSubclassForDialogResourceTemplate(TESDialog::kDialogTemplate_ObjectWindow,
				uiManager::CommonDialogExtraFittingsSubClassProc);

			if (settings::dialogs::kShowMainWindowsInTaskbar.GetData().i)
			{
				bgsee::WindowStyler::StyleData RegularAppWindow = { 0 };
				RegularAppWindow.Extended = WS_EX_APPWINDOW;
				RegularAppWindow.ExtendedOp = bgsee::WindowStyler::StyleData::kOperation_OR;

				BGSEEUI->GetWindowStyler()->RegisterStyle(TESDialog::kDialogTemplate_ObjectWindow, RegularAppWindow);
			}
		}
	}
}