mirror of
https://gitlab.com/hashborgir/d2tweaks-rnd2k.git
synced 2025-09-19 01:52:08 +00:00
Initial commit
This commit is contained in:
Hash Borgir
@@ -0,0 +1,410 @@
|
||||
#include <d2tweaks/client/modules/damage_display/damage_display_client.h>
|
||||
|
||||
#include <d2tweaks/common/protocol.h>
|
||||
#include <d2tweaks/client/client.h>
|
||||
#include <spdlog/spdlog.h>
|
||||
#include <diablo2/utils/screen.h>
|
||||
#include <diablo2/d2common.h>
|
||||
#include <diablo2/structures/unit.h>
|
||||
#include <diablo2/d2client.h>
|
||||
#include <d2tweaks/ui/ui_manager.h>
|
||||
#include <diablo2/d2win.h>
|
||||
|
||||
#include <windows.h>
|
||||
|
||||
#include <iostream> // std::cout
|
||||
#include <string> // std::string, std::to_string
|
||||
|
||||
|
||||
#include <fw/pool.h>
|
||||
|
||||
#include <diablo2/structures/gfxdata.h>
|
||||
#include <diablo2/structures/gfxcell.h>
|
||||
#include <diablo2/d2cmp.h>
|
||||
#include <common/hooking.h>
|
||||
|
||||
MODULE_INIT(damage_display)
|
||||
|
||||
struct damage_label {
|
||||
bool screen_space;
|
||||
diablo2::ui_color_t color;
|
||||
uint32_t x;
|
||||
uint32_t y;
|
||||
uint32_t unit_width;
|
||||
uint32_t unit_height;
|
||||
uint64_t start;
|
||||
int32_t damage;
|
||||
int32_t currentHp; // New field for current hit points
|
||||
int32_t maxHp; // New field for maximum hit points
|
||||
int32_t text_width;
|
||||
wchar_t text[64];
|
||||
|
||||
damage_label(uint32_t x, uint32_t y, uint32_t uw, uint32_t uh, int32_t damage, int32_t hp, int32_t maxHp)
|
||||
: screen_space(false), color(diablo2::ui_color_t::UI_COLOR_WHITE), x(x), y(y),
|
||||
unit_width(uw), unit_height(uh), damage(damage), currentHp(hp), maxHp(maxHp), text_width(42) {
|
||||
start = GetTickCount64();
|
||||
memset(text, 0x00, sizeof text);
|
||||
}
|
||||
|
||||
damage_label()
|
||||
: screen_space(false), color(diablo2::UI_COLOR_WHITE), x(0), y(0),
|
||||
unit_width(0), unit_height(0), damage(0), currentHp(0), maxHp(0), text_width(0) {
|
||||
start = GetTickCount64();
|
||||
memset(text, 0x00, sizeof text);
|
||||
}
|
||||
};
|
||||
|
||||
static growing_object_pool<damage_label> g_label_pool([]() {
|
||||
return new damage_label();
|
||||
});
|
||||
|
||||
static const size_t DAMAGE_LABELS_SIZE = 1024;
|
||||
static damage_label* g_labels[DAMAGE_LABELS_SIZE]{ nullptr };
|
||||
static size_t g_labels_count = 0;
|
||||
static uint32_t DISPLAY_TIME = 500;
|
||||
|
||||
static float lerp(const float v0, const float v1, const float t) {
|
||||
return v0 + t * (v1 - v0);
|
||||
}
|
||||
|
||||
static bool add_label(damage_label* label) {
|
||||
for (size_t i = 0; i < DAMAGE_LABELS_SIZE; i++) {
|
||||
if (g_labels[i] != nullptr)
|
||||
continue;
|
||||
|
||||
g_labels[i] = label;
|
||||
g_labels_count++;
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
static void remove_label(damage_label* label) {
|
||||
for (size_t i = 0; i < DAMAGE_LABELS_SIZE; i++) {
|
||||
if (g_labels[i] != label)
|
||||
continue;
|
||||
|
||||
g_labels[i] = nullptr;
|
||||
g_labels_count--;
|
||||
return;
|
||||
}
|
||||
}
|
||||
|
||||
static void remove_label(size_t index) {
|
||||
if (index < 0 || index >= DAMAGE_LABELS_SIZE)
|
||||
return;
|
||||
|
||||
g_labels[index] = nullptr;
|
||||
g_labels_count--;
|
||||
}
|
||||
|
||||
static unsigned int g_font_enemy = 0;
|
||||
static unsigned int g_font_player = 1;
|
||||
static unsigned int g_player_label_posx = 70;
|
||||
static unsigned int g_player_label_posy = 500;
|
||||
|
||||
static void draw_damage_labels() {
|
||||
const auto player = diablo2::d2_client::get_local_player();
|
||||
|
||||
if (!player)
|
||||
return;
|
||||
|
||||
if (g_labels_count == 0)
|
||||
return;
|
||||
|
||||
const auto currentTime = GetTickCount64();
|
||||
|
||||
size_t updatedLabels = 0;
|
||||
auto mx = 0, my = 0;
|
||||
|
||||
const auto font = diablo2::d2_win::get_current_font();
|
||||
|
||||
/*for (size_t i = 0; i < DAMAGE_LABELS_SIZE; i++) {
|
||||
const auto label = g_labels[i];
|
||||
|
||||
if (label == nullptr) {
|
||||
auto next = g_labels[i + 1];
|
||||
|
||||
if (updatedLabels == g_labels_count)
|
||||
break;
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
const auto delta = currentTime - label->start;
|
||||
|
||||
if (delta >= DISPLAY_TIME) {
|
||||
remove_label(i);
|
||||
g_label_pool.put(label);
|
||||
continue;
|
||||
}
|
||||
|
||||
updatedLabels++;
|
||||
|
||||
if (label->screen_space) {
|
||||
mx = label->x;
|
||||
my = label->y;
|
||||
diablo2::d2_win::set_current_font((diablo2::ui_font_t)g_font_player);
|
||||
} else {
|
||||
diablo2::utils::screen::world_to_screen(label->x, label->y, mx, my);
|
||||
diablo2::d2_win::set_current_font((diablo2::ui_font_t)g_font_enemy);
|
||||
}
|
||||
|
||||
const auto offset = static_cast<int32_t>(lerp(static_cast<float>(label->unit_height) + 5.f,
|
||||
static_cast<float>(label->unit_height) + 30.f,
|
||||
static_cast<float>(delta) / static_cast<float>(DISPLAY_TIME)));
|
||||
|
||||
mx -= label->text_width / 2;
|
||||
mx -= label->unit_width / 2;
|
||||
my -= offset;
|
||||
|
||||
diablo2::d2_win::draw_text(label->text,
|
||||
mx,
|
||||
my,
|
||||
label->color, 0);
|
||||
}*/
|
||||
|
||||
for (size_t i = 0; i < DAMAGE_LABELS_SIZE; i++) {
|
||||
const auto label = g_labels[i];
|
||||
|
||||
if (label == nullptr) {
|
||||
auto next = g_labels[i + 1];
|
||||
|
||||
if (updatedLabels == g_labels_count)
|
||||
break;
|
||||
|
||||
continue;
|
||||
}
|
||||
|
||||
const auto delta = currentTime - label->start;
|
||||
|
||||
if (delta >= DISPLAY_TIME) {
|
||||
remove_label(i);
|
||||
g_label_pool.put(label);
|
||||
continue;
|
||||
}
|
||||
|
||||
updatedLabels++;
|
||||
|
||||
int32_t mx = label->x;
|
||||
int32_t my = label->y;
|
||||
|
||||
if (label->screen_space) {
|
||||
diablo2::utils::screen::world_to_screen(label->x, label->y, mx, my);
|
||||
}
|
||||
else {
|
||||
diablo2::utils::screen::world_to_screen(label->x, label->y, mx, my);
|
||||
}
|
||||
|
||||
mx -= label->text_width / 2;
|
||||
mx -= label->unit_width / 2;
|
||||
my -= label->unit_height;
|
||||
|
||||
// Draw different labels based on screen_space flag
|
||||
if (label->screen_space) {
|
||||
const auto offset = static_cast<int32_t>(lerp(static_cast<float>(label->unit_height) + 5.f, static_cast<float>(label->unit_height) + 30.f, static_cast<float>(delta) / static_cast<float>(DISPLAY_TIME)));
|
||||
mx -= 10;
|
||||
my -= 50;
|
||||
my -= offset;
|
||||
|
||||
// Draw player label
|
||||
std::wstring dmgText = std::to_wstring(label->damage);
|
||||
const wchar_t* dmgTextPtr = dmgText.c_str();
|
||||
diablo2::d2_win::set_current_font((diablo2::ui_font_t)g_font_player);
|
||||
diablo2::d2_win::draw_text(const_cast<wchar_t*>(dmgTextPtr), mx, my, label->color, 0);
|
||||
}
|
||||
else {
|
||||
// Draw monster healthbar
|
||||
if (label->currentHp > 0) {
|
||||
// Calculate the percentage of currentHp relative to maxHp
|
||||
float healthPercentage = (label->maxHp != 0) ? static_cast<float>(label->currentHp) / static_cast<float>(label->maxHp) : 0.0f;
|
||||
|
||||
// Determine the color based on healthPercentage
|
||||
diablo2::ui_color_t textColor;
|
||||
if (healthPercentage >= 0.67f) {
|
||||
textColor = diablo2::UI_COLOR_GREEN;
|
||||
}
|
||||
else if (healthPercentage <= 0.33f) {
|
||||
textColor = diablo2::UI_COLOR_RED;
|
||||
}
|
||||
else {
|
||||
textColor = diablo2::UI_COLOR_YELLOW;
|
||||
}
|
||||
|
||||
int bMaxWidth = 100;
|
||||
|
||||
// Calculate the width of the health bar based on the percentage (max width is 10 characters)
|
||||
uint32_t barWidth = static_cast<uint32_t>(healthPercentage * 10.0f);
|
||||
barWidth = (barWidth > bMaxWidth) ? bMaxWidth : barWidth; // Ensure barWidth doesn't exceed 10
|
||||
barWidth = (label->currentHp > 0 && barWidth == 0) ? 1 : barWidth; // Ensure at least one '#' if currentHp is not 0
|
||||
|
||||
// Construct the health bar string representation
|
||||
std::wstring barText;
|
||||
for (uint32_t i = 0; i < barWidth; ++i) {
|
||||
barText.append(L"!"); // Use '#' to represent filled portion of the bar
|
||||
}
|
||||
for (uint32_t i = barWidth; i < bMaxWidth; ++i) {
|
||||
barText.append(L""); // Use '-' to represent empty portion of the bar
|
||||
}
|
||||
|
||||
// Construct the health fraction string
|
||||
std::wstring fractionStr = std::to_wstring(label->currentHp) + L"/" + std::to_wstring(label->maxHp);
|
||||
|
||||
// Combine the strings for health percentage and bar text
|
||||
std::wstring combinedText = std::to_wstring(static_cast<int>(healthPercentage * 100.0f)) + L"% " + barText;
|
||||
|
||||
const WCHAR* combinedTextPtr = combinedText.c_str();
|
||||
|
||||
// Calculate text position for the combined text
|
||||
uint32_t textX = mx;
|
||||
uint32_t textY = my;
|
||||
|
||||
|
||||
// Draw the combined text (health percentage and bar text)
|
||||
//diablo2::d2_win::draw_text(const_cast<wchar_t*>(combinedText.c_str()), textX, textY, textColor, 0);
|
||||
//diablo2::d2_win::draw_boxed_text(const_cast<wchar_t*>(fractionStr.c_str()), textX + label->unit_width/2, textY - 12, 0, 0, textColor);
|
||||
|
||||
diablo2::d2_win::set_current_font((diablo2::ui_font_t)g_font_enemy); // Set font to FONT16
|
||||
diablo2::d2_win::draw_text(const_cast<wchar_t*>(fractionStr.c_str()), textX + diablo2::d2_win::get_text_pixel_width(const_cast<wchar_t*>(combinedTextPtr)) / 2, textY - 12, textColor, 0);
|
||||
|
||||
diablo2::d2_win::set_current_font((diablo2::ui_font_t)g_font_enemy); // Set font to FONT6
|
||||
diablo2::d2_win::draw_text(const_cast<wchar_t*>(combinedText.c_str()), textX, textY, textColor, 0);
|
||||
|
||||
//diablo2::d2_win::draw_boxed_text(const_cast<wchar_t*>(combinedText.c_str()), textX, textY, 0, 0, textColor);
|
||||
|
||||
|
||||
const auto offset = static_cast<int32_t>(lerp(static_cast<float>(label->unit_height) + 5.f, static_cast<float>(label->unit_height) + 30.f, static_cast<float>(delta) / static_cast<float>(DISPLAY_TIME)));
|
||||
my -= offset;
|
||||
|
||||
// Draw damage label
|
||||
std::wstring dmgText = L" " + std::to_wstring(label->damage) + L" ";
|
||||
const wchar_t* dmgTextPtr = dmgText.c_str();
|
||||
diablo2::d2_win::set_current_font((diablo2::ui_font_t)g_font_enemy);
|
||||
diablo2::d2_win::draw_text(const_cast<wchar_t*>(dmgTextPtr), textX + diablo2::d2_win::get_text_pixel_width(const_cast<wchar_t*>(combinedTextPtr)) / 2, my + label->unit_height / 2, textColor, 0);
|
||||
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
||||
diablo2::d2_win::set_current_font(font);
|
||||
}
|
||||
|
||||
static void(__fastcall* g_draw_game_ui)(int32_t a1);
|
||||
static void __fastcall draw_game_ui(int32_t a1) {
|
||||
draw_damage_labels();
|
||||
g_draw_game_ui(a1);
|
||||
}
|
||||
|
||||
static diablo2::ui_color_t damage_type_to_color(d2_tweaks::common::damage_type_t type) {
|
||||
switch (type) {
|
||||
case d2_tweaks::common::DAMAGE_TYPE_PHYSICAL: return diablo2::UI_COLOR_GREY;
|
||||
case d2_tweaks::common::DAMAGE_TYPE_COLD: return diablo2::UI_COLOR_BLUE;
|
||||
case d2_tweaks::common::DAMAGE_TYPE_FIRE: return diablo2::UI_COLOR_RED;
|
||||
case d2_tweaks::common::DAMAGE_TYPE_LIGHTNING: return diablo2::UI_COLOR_YELLOW;
|
||||
case d2_tweaks::common::DAMAGE_TYPE_POISON: return diablo2::UI_COLOR_GREEN;
|
||||
case d2_tweaks::common::DAMAGE_TYPE_MAGIC: return diablo2::UI_COLOR_PURPLE;
|
||||
default: return diablo2::UI_COLOR_BLACK;
|
||||
}
|
||||
}
|
||||
|
||||
void d2_tweaks::client::modules::damage_display::init_early() {
|
||||
|
||||
}
|
||||
|
||||
void d2_tweaks::client::modules::damage_display::init() {
|
||||
char acPathToIni[MAX_PATH] = { 0 };
|
||||
const char* pcIniFile = "\\d2tweaks.ini";
|
||||
|
||||
GetCurrentDirectory(MAX_PATH, acPathToIni);
|
||||
lstrcat(acPathToIni, pcIniFile);
|
||||
|
||||
if (GetPrivateProfileInt("modules", "DamageDisplay", 1, acPathToIni) != FALSE) {
|
||||
g_font_enemy = GetPrivateProfileInt("DamageDisplay", "EnemyDamageFont", 0, acPathToIni);
|
||||
g_font_player = GetPrivateProfileInt("DamageDisplay", "PlayerDamageFont", 1, acPathToIni);
|
||||
g_player_label_posx = GetPrivateProfileInt("DamageDisplay", "PlayerDamagePosx", 70, acPathToIni);
|
||||
g_player_label_posy = GetPrivateProfileInt("DamageDisplay", "PlayerDamagePosy", 500, acPathToIni);
|
||||
DISPLAY_TIME = GetPrivateProfileInt("DamageDisplay", "DisplayTime", 1000, acPathToIni);
|
||||
singleton<client>::instance().register_packet_handler(common::MESSAGE_TYPE_DAMAGE_INFO, this);
|
||||
hooking::hook(diablo2::d2_client::get_base() + 0x80A30, draw_game_ui, reinterpret_cast<void**>(&g_draw_game_ui));
|
||||
}
|
||||
}
|
||||
|
||||
void d2_tweaks::client::modules::damage_display::handle_packet(common::packet_header* packet) {
|
||||
static auto& instance = singleton<client>::instance();
|
||||
static diablo2::structures::gfxdata gfxdata;
|
||||
const auto player = diablo2::d2_client::get_local_player();
|
||||
const auto info = static_cast<common::damage_info_sc*>(packet);
|
||||
|
||||
if (player == nullptr)
|
||||
return;
|
||||
|
||||
if (info->unit_type == 0x00 && info->guid == player->guid) {
|
||||
const auto label = g_label_pool.get();
|
||||
label->screen_space = true;
|
||||
label->color = damage_type_to_color(info->damage_type);
|
||||
label->unit_width = 0;
|
||||
label->unit_height = 0;
|
||||
label->x = g_player_label_posx;
|
||||
label->y = g_player_label_posy;
|
||||
label->damage = info->damage;
|
||||
label->start = GetTickCount64();
|
||||
swprintf_s(label->text, L"%u", label->damage);
|
||||
label->text_width = diablo2::d2_win::get_text_pixel_width(label->text);
|
||||
|
||||
if (add_label(label))
|
||||
return;
|
||||
|
||||
g_label_pool.put(label); //prevent memory leak if there's no room for another label
|
||||
return;
|
||||
}
|
||||
|
||||
if (info->unit_type != 0x01)
|
||||
return;
|
||||
|
||||
const auto monsterUnit = instance.get_client_unit(0x01, info->guid);
|
||||
|
||||
if (monsterUnit == nullptr)
|
||||
return;
|
||||
|
||||
const auto mx = diablo2::d2_common::get_unit_x(monsterUnit->path);
|
||||
const auto my = diablo2::d2_common::get_unit_y(monsterUnit->path);
|
||||
|
||||
const auto label = g_label_pool.get();
|
||||
|
||||
label->color = damage_type_to_color(info->damage_type);
|
||||
|
||||
memset(&gfxdata, 0x00, sizeof gfxdata);
|
||||
|
||||
int32_t index;
|
||||
if (diablo2::d2_client::cache_gfx_data(&gfxdata, monsterUnit, nullptr, 0, 0, &index, 1, -1) &&
|
||||
diablo2::structures::d2_cmp::init_gfx_data(&gfxdata) &&
|
||||
gfxdata.cell_init) {
|
||||
label->unit_width = gfxdata.cell_init->width + gfxdata.cell_init->offset_x;
|
||||
label->unit_height = gfxdata.cell_init->height - gfxdata.cell_init->offset_y;
|
||||
} else {
|
||||
label->unit_width = 0;
|
||||
label->unit_height = 0;
|
||||
}
|
||||
|
||||
label->screen_space = false;
|
||||
label->x = mx;
|
||||
label->y = my;
|
||||
label->damage = info->damage;
|
||||
label->start = GetTickCount64();
|
||||
swprintf_s(label->text, L"%i", label->damage);
|
||||
label->text_width = diablo2::d2_win::get_text_pixel_width(label->text);
|
||||
|
||||
if (add_label(label))
|
||||
return;
|
||||
|
||||
g_label_pool.put(label); //prevent memory leak if there's no room for another label
|
||||
}
|
||||
|
||||
void d2_tweaks::client::modules::damage_display::tick() {
|
||||
|
||||
}
|
||||
Reference in New Issue
Block a user