turanszkij
97 lines
1.5 KiB
C++
97 lines
1.5 KiB
C++
#include "wiCpuInfo.h"
|
|
|
|
#ifndef WINSTORE_SUPPORT
|
|
|
|
bool wiCpuInfo::m_canReadCpu;
|
|
HQUERY wiCpuInfo::m_queryHandle;
|
|
HCOUNTER wiCpuInfo::m_counterHandle;
|
|
unsigned long wiCpuInfo::m_lastSampleTime;
|
|
long wiCpuInfo::m_cpuUsage;
|
|
|
|
wiCpuInfo::wiCpuInfo()
|
|
{
|
|
}
|
|
|
|
|
|
wiCpuInfo::~wiCpuInfo()
|
|
{
|
|
}
|
|
|
|
void wiCpuInfo::Initialize()
|
|
{
|
|
PDH_STATUS status;
|
|
|
|
|
|
// Initialize the flag indicating whether this object can read the system cpu usage or not.
|
|
m_canReadCpu = true;
|
|
|
|
// Create a query object to poll cpu usage.
|
|
status = PdhOpenQuery(NULL, 0, &m_queryHandle);
|
|
if(status != ERROR_SUCCESS)
|
|
{
|
|
m_canReadCpu = false;
|
|
}
|
|
|
|
// Set query object to poll all cpus in the system.
|
|
status = PdhAddCounter(m_queryHandle, TEXT("\\Processor(_Total)\\% processor time"), 0, &m_counterHandle);
|
|
if(status != ERROR_SUCCESS)
|
|
{
|
|
m_canReadCpu = false;
|
|
}
|
|
|
|
m_lastSampleTime = GetTickCount();
|
|
|
|
m_cpuUsage = 0;
|
|
|
|
return;
|
|
}
|
|
|
|
void wiCpuInfo::Shutdown()
|
|
{
|
|
if(m_canReadCpu)
|
|
{
|
|
PdhCloseQuery(m_queryHandle);
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
void wiCpuInfo::Frame()
|
|
{
|
|
PDH_FMT_COUNTERVALUE value;
|
|
|
|
if(m_canReadCpu)
|
|
{
|
|
if((m_lastSampleTime + 1000) < GetTickCount())
|
|
{
|
|
m_lastSampleTime = GetTickCount();
|
|
|
|
PdhCollectQueryData(m_queryHandle);
|
|
|
|
PdhGetFormattedCounterValue(m_counterHandle, PDH_FMT_LONG, NULL, &value);
|
|
|
|
m_cpuUsage = value.longValue;
|
|
}
|
|
}
|
|
|
|
return;
|
|
}
|
|
|
|
int wiCpuInfo::GetCpuPercentage()
|
|
{
|
|
int usage;
|
|
|
|
if(m_canReadCpu)
|
|
{
|
|
usage = (int)m_cpuUsage;
|
|
}
|
|
else
|
|
{
|
|
usage = -1;
|
|
}
|
|
|
|
return usage;
|
|
}
|
|
|
|
#endif
|