h2-mod/src/component/fps.cpp

#include <stdinc.hpp>
#include "loader/component_loader.hpp"

#include "game/game.hpp"
#include "game/dvars.hpp"

#include "chat.hpp"
#include "scheduler.hpp"

#include <utils/string.hpp>
#include <utils/hook.hpp>

#define fps_font game::R_RegisterFont("fonts/fira_mono_regular.ttf", 25)

namespace fps
{
	namespace
	{
		std::chrono::nanoseconds frametime;
		auto lastframe = std::chrono::high_resolution_clock::now();

		game::dvar_t* cg_drawfps;

		float fps_color_good[4] = {0.6f, 1.0f, 0.0f, 1.0f};
		float fps_color_ok[4] = {1.0f, 0.7f, 0.3f, 1.0f};
		float fps_color_bad[4] = {1.0f, 0.3f, 0.3f, 1.0f};

		float screen_max[2];

		int history_count = 20;
		int history[20] = { 0 };
		int index;

		void check_resize()
		{
			screen_max[0] = game::ScrPlace_GetViewPlacement()->realViewportSize[0];
			screen_max[1] = game::ScrPlace_GetViewPlacement()->realViewportSize[1];
		}

		int average_fps()
		{
			auto total = 0;

			for (auto i = 0; i < history_count; i++)
			{
				total += history[i];
			}

			return total / history_count;
		}

		void draw()
		{
			check_resize();

			if (cg_drawfps->current.integer >= 1)
			{
				const auto now = std::chrono::high_resolution_clock::now();
				const auto frametime = now - lastframe;
				lastframe = now;

				const int fps = 1000000000 / frametime.count();

				if (index >= history_count)
				{
					index = 0;
				}

				history[index++] = fps;

				const auto fps_string = utils::string::va("%i", average_fps());
				const auto x = screen_max[0] - 15.f - game::R_TextWidth(fps_string, 0x7FFFFFFF, fps_font);

				game::R_AddCmdDrawText(fps_string, 0x7FFFFFFF, fps_font, x, 35.f, 1.0f, 1.0f, 0.0f,
					fps >= 60 ? fps_color_good : (fps >= 30 ? fps_color_ok : fps_color_bad), 0);
			}

			if (game::CL_IsCgameInitialized() && cg_drawfps->current.integer >= 2)
			{
				const auto font = fps_font;

				const auto pos_string = utils::string::va("%f %f %f",
														  game::g_entities[0].origin[0],
														  game::g_entities[0].origin[1],
														  game::g_entities[0].origin[2]);

				const auto x = screen_max[0] - 15.f - game::R_TextWidth(pos_string, 0x7FFFFFFF, font);

				game::R_AddCmdDrawText(pos_string, 0x7FFFFFFF, font, x,
					60.f, 1.0f, 1.0f, 0.0f, fps_color_ok, 0);
			}
		}
	}

	class component final : public component_interface
	{
	public:
		void post_unpack() override
		{
			cg_drawfps = game::Dvar_RegisterInt(game::generateHashValue("cg_drawfps"), "", 0, 0, 4, game::DVAR_FLAG_SAVED);

			scheduler::loop(draw, scheduler::pipeline::renderer);
		}
	};
}

REGISTER_COMPONENT(fps::component)