// Copyright Epic Games, Inc. All Rights Reserved.

#pragma once

#include <Windows/WindowsPlatformTime.h>

namespace UE::StylusInput::RealTimeStylus
{
	class FPacketStats
	{
	public:
		void NewPacket()
		{
			const uint64 Timestamp = FPlatformTime::Cycles64();

			if (LatestTimestamp - EarliestTimestamp >= CyclesPerSecond)
			{
				PacketsPerSecond = EarliestTimestamp != LatestTimestamp ? NumPackets / ((LatestTimestamp - EarliestTimestamp) / CyclesPerSecond) : 0.0f;
				EarliestTimestamp = Timestamp;
				NumPackets = -1;
			}

			LatestTimestamp = Timestamp;
			++NumPackets;
		}

		float GetPacketsPerSecond() const
		{
			const uint64 Timestamp = FPlatformTime::Cycles64();
			return Timestamp - LatestTimestamp <= CyclesPerSecond ? PacketsPerSecond : 0.0f;
		}

	private:
		uint64 EarliestTimestamp = 0;
		uint64 LatestTimestamp = 0;
		uint32 NumPackets = 0;
		float PacketsPerSecond = 0.0f;
		const double CyclesPerSecond = 1.0 / FPlatformTime::GetSecondsPerCycle64();
	};
}