Added HUDExtension, fully functional

hudextension/HUDExtension.h

@@ -0,0 +1,193 @@
+#pragma once
+
+#include "skse64/GameMenus.h"
+#include "skse64/Hooks_UI.h"
+
+#include <set>
+
+class ObjectWidget
+{
+public:
+	bool operator<(const ObjectWidget & rhs) const	{ return formId < rhs.formId; }
+
+	enum Flags
+	{
+		kFlag_None					= 0,
+		kFlag_UpdatePercent			= (1 << 0),
+		kFlag_UseLineOfSight		= (1 << 1),
+		kFlag_RemoveOnDeath			= (1 << 2),
+		kFlag_RemoveOutOfCombat		= (1 << 3),
+		kFlag_HideOnDeath			= (1 << 4),
+		kFlag_HideOutOfCombat		= (1 << 5),
+		kFlag_ShowInCombat			= (1 << 6),
+		kFlag_Friendly				= (1 << 7),
+		kFlag_UseHostility			= (1 << 8),
+		kFlag_HideOnInvisibility	= (1 << 9),
+		kFlag_HideName				= (1 << 10)
+	};
+
+	enum FillModes
+	{
+		kFillMode_Left = 0,
+		kFillMode_Right,
+		kFillMode_Both
+	};
+
+	enum PropertyTypes
+	{
+		kPropertyType_Flags,
+		kPropertyType_CurrentValue,
+		kPropertyType_MaximumValue,
+		kPropertyType_PrimaryColor,
+		kPropertyType_SecondaryColor,
+		kPropertyType_FlashColor,
+		kPropertyType_PrimaryFriendlyColor,
+		kPropertyType_SecondaryFriendlyColor,
+		kPropertyType_FlashFriendlyColor,
+		kPropertyType_FillMode,
+		kPropertyType_StartFlash,
+		kPropertyType_Name,
+		kPropertyType_NumProperties
+	};
+
+
+	enum Properties
+	{
+		kProperty_CurrentValue = 0,
+		kProperty_MaximumValue,
+		kProperty_PrimaryColor,
+		kProperty_SecondaryColor,
+		kProperty_FlashColor,
+		kProperty_PrimaryFriendlyColor,
+		kProperty_SecondaryFriendlyColor,
+		kProperty_FlashFriendlyColor,
+		kProperty_FillMode,
+		kProperty_Name,
+		kProperty_NumProperties
+	};
+
+	enum StateContext
+	{
+		kContext_None = 0,
+		kContext_LeaveCombat = 1,
+		kContext_Death = 2,
+		kContext_Friendly = 4
+	};
+
+	ObjectWidget::ObjectWidget()
+	{
+		flags |= kFlag_RemoveOnDeath | kFlag_RemoveOutOfCombat | kFlag_UpdatePercent | kFlag_UseLineOfSight;
+		for(UInt32 i = 0; i < kProperty_NumProperties; i++)
+			params[i].SetUndefined();
+	}
+
+	UInt32		formId;
+	UInt32		flags;
+	GFxValue	params[kProperty_NumProperties];
+	GFxValue	object;
+
+	double GetProperty(UInt32 type);
+	void SetProperty(UInt32 type, double value);
+	void UpdateProperty(UInt32 type);
+
+	void QueryState(TESObjectREFR * reference, bool * visible, bool * hostile);
+	bool IsFriendly() const;
+
+	void UpdateFlags();
+	void UpdateValues();
+	void UpdateColors();
+	void UpdateFillMode();
+	void UpdateFlash();
+	void UpdateText();
+	void UpdateComponent(GFxMovieView * view, float * depth);
+};
+
+class ObjectWidgets : public SafeDataHolder<std::set<ObjectWidget>>
+{
+	typedef std::set<ObjectWidget> HealthbarSet;
+public:
+	void AddGFXMeter(GFxMovieView * view, ObjectWidget * objectMeter, float current, float max, UInt32 flags, UInt32 fillMode, UInt32 colors[]);
+	void RemoveGFXMeter(GFxMovieView * view, ObjectWidget * objectMeter);
+
+	bool AddMeter(GFxMovieView * view, UInt32 formId, float current, float max, UInt32 flags, UInt32 fillMode, UInt32 colors[]);
+	bool RemoveMeter(GFxMovieView * view, UInt32 formId, UInt32 context);
+
+	void UpdateMeterProperty(UInt32 formId, UInt32 type);
+	double GetMeterProperty(UInt32 formId, UInt32 type);
+	void SetMeterProperty(UInt32 formId, UInt32 type, double value);
+
+	void RemoveAllHealthbars(GFxMovieView * view);
+	void UpdateComponents(GFxMovieView * view);
+};
+
+class HUDExtension : public HUDObject
+{
+public:
+	HUDExtension(GFxMovieView* view);
+	virtual ~HUDExtension();
+
+	bool AddMeter(UInt32 formId, float current, float max, UInt32 flags, UInt32 fillMode, UInt32 colors[]);
+	bool RemoveMeter(UInt32 formId, UInt32 context);
+
+	void UpdateMeterProperty(UInt32 formId, UInt32 type);
+	double GetMeterProperty(UInt32 formId, UInt32 type);
+	void SetMeterProperty(UInt32 formId, UInt32 type, double value);
+
+	void RemoveAllHealthbars();
+
+	virtual void Update();
+
+	enum
+	{
+		kFlags_None				= 0,
+		kFlags_HideEnemies		= (1 << 0),
+		kFlags_HideAllies		= (1 << 1),
+		kFlags_HideFriendly		= (1 << 2),
+		kFlags_HideNonHostile	= (1 << 3),
+		kFlags_HideAtFull		= (1 << 4),
+		kFlags_HideName			= (1 << 5),
+	};
+
+	UInt32 hudFlags;
+
+private:
+	ObjectWidgets	m_components;
+};
+
+
+class AddRemoveWidgetTask : public UIDelegate_v1
+{
+public:
+	AddRemoveWidgetTask::AddRemoveWidgetTask(UInt32 formId, float current, float max, UInt32 state, UInt32 context) : m_formId(formId), m_current(current), m_max(max), m_state(state), m_context(context){}
+	virtual void Run();
+	virtual void Dispose()
+	{
+		delete this;
+	}
+
+private:
+	UInt32			m_formId;
+	float			m_current;
+	float			m_max;
+	UInt32			m_state;
+	UInt32			m_context;
+};
+
+class SetWidgetPropertyTask : public UIDelegate_v1
+{
+public:
+	SetWidgetPropertyTask(UInt32 formId, UInt32 type, double value);
+
+	virtual void Run();
+	virtual void Dispose()
+	{
+		delete this;	
+	};
+
+private:
+	UInt32			m_formId;
+	UInt32			m_type;
+	double			m_value;
+};
+
+extern HUDExtension * g_hudExtension;

hudextension/Hooks.cpp

@@ -0,0 +1,40 @@
+#include "Hooks.h"
+#include "HUDExtension.h"
+#include <float.h>
+
+#include "skse64/GameMenus.h"
+
+const _Fn85FD10 Fn85FD10 = (_Fn85FD10)0x0085FD10;
+
+void __stdcall InstallHudComponents(HUDMenu * menu)
+{
+	GFxValue hudComponent;
+	GFxValue args[2];
+	args[0].SetString("hudExtension");
+	menu->view->Invoke("_root.getNextHighestDepth", &args[1], NULL, 0);
+	menu->view->Invoke("_root.createEmptyMovieClip", &hudComponent, args, 2);
+	args[0].SetString("hud_extension.swf");
+	hudComponent.Invoke("loadMovie", NULL, &args[0], 1);
+	g_hudExtension = new HUDExtension(menu->view);
+	g_hudExtension->object = hudComponent;
+	menu->hudComponents.Push((HUDObject*&)g_hudExtension);
+}
+
+const UInt32 kInstallRaceMenuHook_Base = 0x008652E0 + 0x4F5;
+const UInt32 kInstallRaceMenuHook_Entry_retn = kInstallRaceMenuHook_Base + 0x05;
+
+/*
+__declspec(naked) void InstallRaceMenuHook_Entry(void)
+{
+	__asm
+	{
+		call	Fn85FD10
+		pushad
+		push	edi
+		call	InstallHudComponents
+		popad
+
+		jmp		[kInstallRaceMenuHook_Entry_retn]
+	}
+}
+*/

hudextension/Hooks.h

@@ -0,0 +1,14 @@
+#pragma once
+
+#include "skse64/GameTypes.h"
+
+class HUDMenu;
+
+extern const UInt32 kInstallRaceMenuHook_Base;
+extern const UInt32 kInstallRaceMenuHook_Entry_retn;
+
+void __stdcall InstallHudComponents(HUDMenu * menu);
+void InstallRaceMenuHook_Entry(void);
+
+typedef void (* _Fn85FD10)();
+extern const _Fn85FD10 Fn85FD10;

hudextension/exports.def

@@ -0,0 +1,4 @@
+LIBRARY	"hudextension"
+EXPORTS
+SKSEPlugin_Query
+SKSEPlugin_Load