rainmeter-studio/Plugins/PluginPerfMon/PerfData.cpp

251 lines
5.9 KiB
C++
Raw Normal View History

2009-02-10 18:37:48 +00:00
/*
Copyright (C) 2004 Kimmo Pekkola
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
2009-02-10 18:37:48 +00:00
*/
#include <windows.h>
#include "PerfData.h"
#include <string>
#include <map>
#include <crtdbg.h>
2011-02-03 18:09:24 +00:00
#include "../../Library/Export.h" // Rainmeter's exported functions
2009-02-10 18:37:48 +00:00
2011-02-03 18:09:24 +00:00
#include "../../Library/DisableThreadLibraryCalls.h" // contains DllMain entry point
2009-02-10 18:37:48 +00:00
ULONGLONG GetPerfData(PCTSTR ObjectName, PCTSTR InstanceName, PCTSTR CounterName);
struct PerfMeasure
{
std::wstring ObjectName;
std::wstring CounterName;
std::wstring InstanceName;
bool Difference;
ULONGLONG OldValue;
bool FirstTime;
};
static CPerfTitleDatabase g_CounterTitles( PERF_TITLE_COUNTER );
static std::map<UINT, PerfMeasure*> g_Measures;
/*
This function is called when the measure is initialized.
2011-03-29 19:21:57 +00:00
The function must return the maximum value that can be measured.
2009-02-10 18:37:48 +00:00
The return value can also be 0, which means that Rainmeter will
track the maximum value automatically. The parameters for this
function are:
instance The instance of this DLL
iniFile The name of the ini-file (usually Rainmeter.ini)
section The name of the section in the ini-file for this measure
id The identifier for the measure. This is used to identify the measures that use the same plugin.
*/
UINT Initialize(HMODULE instance, LPCTSTR iniFile, LPCTSTR section, UINT id)
{
PerfMeasure* measure = new PerfMeasure;
measure->Difference = false;
measure->FirstTime = true;
measure->OldValue = 0;
LPCTSTR buffer = ReadConfigString(section, L"PerfMonObject", L"");
if (buffer)
{
measure->ObjectName = buffer;
}
buffer = ReadConfigString(section, L"PerfMonCounter", L"");
if (buffer)
{
measure->CounterName = buffer;
}
buffer = ReadConfigString(section, L"PerfMonInstance", L"");
if (buffer)
{
measure->InstanceName = buffer;
}
buffer = ReadConfigString(section, L"PerfMonDifference", L"1");
if (buffer)
{
measure->Difference = 1==_wtoi(buffer);
}
// Store the measure
g_Measures[id] = measure;
int maxVal = 0;
buffer = ReadConfigString(section, L"PerfMonMaxValue", L"0");
if (buffer)
{
maxVal = _wtoi(buffer);
}
return maxVal;
}
/*
This function is called when new value should be measured.
The function returns the new value.
*/
double Update2(UINT id)
2009-02-10 18:37:48 +00:00
{
double value = 0;
2009-02-10 18:37:48 +00:00
std::map<UINT, PerfMeasure*>::iterator i = g_Measures.find(id);
2011-03-29 19:21:57 +00:00
if (i != g_Measures.end())
2009-02-10 18:37:48 +00:00
{
PerfMeasure* measure = (*i).second;
2011-03-29 19:21:57 +00:00
if (measure)
2009-02-10 18:37:48 +00:00
{
2011-02-20 23:03:15 +00:00
ULONGLONG longvalue;
2011-03-29 19:21:57 +00:00
longvalue = GetPerfData(measure->ObjectName.c_str(),
measure->InstanceName.c_str(),
2011-02-20 23:03:15 +00:00
measure->CounterName.c_str());
2009-02-10 18:37:48 +00:00
2011-03-29 19:21:57 +00:00
if (measure->Difference)
2011-02-20 23:03:15 +00:00
{
// Compare with the old value
2011-03-29 19:21:57 +00:00
if (!measure->FirstTime)
2009-02-10 18:37:48 +00:00
{
2011-02-20 23:03:15 +00:00
value = (double)(longvalue - measure->OldValue);
2009-02-10 18:37:48 +00:00
}
2011-02-20 23:03:15 +00:00
measure->OldValue = longvalue;
measure->FirstTime = false;
2009-02-10 18:37:48 +00:00
}
else
{
2011-02-20 23:03:15 +00:00
value = (double)longvalue;
2009-02-10 18:37:48 +00:00
}
}
}
return value;
}
/*
If the measure needs to free resources before quitting.
The plugin can export Finalize function, which is called
when Rainmeter quits (or refreshes).
*/
void Finalize(HMODULE instance, UINT id)
{
// delete the measure
std::map<UINT, PerfMeasure*>::iterator i = g_Measures.find(id);
2011-03-29 19:21:57 +00:00
if (i != g_Measures.end())
2009-02-10 18:37:48 +00:00
{
delete (*i).second;
g_Measures.erase(i);
}
CPerfSnapshot::CleanUp();
}
/*
This method gets value of the given perfmon counter.
*/
ULONGLONG GetPerfData(PCTSTR ObjectName, PCTSTR InstanceName, PCTSTR CounterName)
{
CPerfObject* pPerfObj;
CPerfObjectInstance* pObjInst;
CPerfCounter* pPerfCntr;
BYTE data[256];
WCHAR name[256];
ULONGLONG value = 0;
2011-03-29 19:21:57 +00:00
if (ObjectName == NULL || CounterName == NULL || wcslen(ObjectName) == 0 || wcslen(CounterName) == 0)
2009-02-10 18:37:48 +00:00
{
// Unable to continue
return 0;
}
CPerfSnapshot snapshot(&g_CounterTitles);
CPerfObjectList objList(&snapshot, &g_CounterTitles);
2011-03-29 19:21:57 +00:00
if (snapshot.TakeSnapshot(ObjectName))
2009-02-10 18:37:48 +00:00
{
pPerfObj = objList.GetPerfObject(ObjectName);
2011-03-29 19:21:57 +00:00
if (pPerfObj)
2009-02-10 18:37:48 +00:00
{
2011-03-29 19:21:57 +00:00
for (pObjInst = pPerfObj->GetFirstObjectInstance();
2009-02-10 18:37:48 +00:00
pObjInst != NULL;
pObjInst = pPerfObj->GetNextObjectInstance())
{
if (InstanceName != NULL && wcslen(InstanceName) > 0)
{
2011-03-29 19:21:57 +00:00
if (pObjInst->GetObjectInstanceName(name, 256))
2009-02-10 18:37:48 +00:00
{
2011-03-29 19:21:57 +00:00
if (_wcsicmp(InstanceName, name) != 0)
2009-02-10 18:37:48 +00:00
{
delete pObjInst;
continue;
}
}
else
{
delete pObjInst;
continue;
}
}
pPerfCntr = pObjInst->GetCounterByName(CounterName);
2011-03-29 19:21:57 +00:00
if (pPerfCntr != NULL)
2009-02-10 18:37:48 +00:00
{
pPerfCntr->GetData(data, 256, NULL);
2011-03-29 19:21:57 +00:00
if (pPerfCntr->GetSize() == 1)
2009-02-10 18:37:48 +00:00
{
value = *(BYTE*)data;
2011-03-29 19:21:57 +00:00
}
else if (pPerfCntr->GetSize() == 2)
2009-02-10 18:37:48 +00:00
{
value = *(WORD*)data;
}
2011-03-29 19:21:57 +00:00
else if (pPerfCntr->GetSize() == 4)
2009-02-10 18:37:48 +00:00
{
value = *(DWORD*)data;
}
2011-03-29 19:21:57 +00:00
else if (pPerfCntr->GetSize() == 8)
2009-02-10 18:37:48 +00:00
{
value = *(ULONGLONG*)data;
}
delete pPerfCntr;
delete pObjInst;
break; // No need to continue
}
delete pObjInst;
}
delete pPerfObj;
}
}
return value;
}
UINT GetPluginVersion()
{
return 1002;
}
LPCTSTR GetPluginAuthor()
{
return L"Rainy (rainy@iki.fi)";
}