[STOBJECT]
-Added support for battery enumeration. -Now battery status is available. -Added icons and resources respectively for battery levels. -Now battery changes its icons dynamically as per its quantized levels. svn path=/branches/GSoC_2017/shellext/; revision=75367
|
@ -29,6 +29,7 @@ set_module_type(stobject win32dll UNICODE)
|
|||
target_link_libraries(stobject uuid wine atlnew)
|
||||
|
||||
add_importlibs(stobject
|
||||
setupapi
|
||||
advapi32
|
||||
winmm
|
||||
ole32
|
||||
|
|
|
@ -4,20 +4,32 @@
|
|||
* FILE: dll/shellext/stobject/power.cpp
|
||||
* PURPOSE: Power notification icon handler
|
||||
* PROGRAMMERS: Eric Kohl <eric.kohl@reactos.org>
|
||||
Shriraj Sawant a.k.a SR13 <sr.official@hotmail.com>
|
||||
* David Quintana <gigaherz@gmail.com>
|
||||
*/
|
||||
|
||||
#include <Windows.h>
|
||||
#include <SetupAPI.h>
|
||||
#include <devguid.h>
|
||||
#include <BatClass.h>
|
||||
|
||||
#include "precomp.h"
|
||||
#include "powrprof.h"
|
||||
|
||||
#include <mmsystem.h>
|
||||
#include <mmddk.h>
|
||||
|
||||
#define GBS_HASBATTERY 0x1
|
||||
#define GBS_ONBATTERY 0x2
|
||||
|
||||
#define GET_X_LPARAM(lp) ((int)(short)LOWORD(lp))
|
||||
#define GET_Y_LPARAM(lp) ((int)(short)HIWORD(lp))
|
||||
|
||||
WINE_DEFAULT_DEBUG_CHANNEL(stobject);
|
||||
|
||||
int br_icons[5] = { IDI_BATTCAP0, IDI_BATTCAP1, IDI_BATTCAP2, IDI_BATTCAP3, IDI_BATTCAP4 };
|
||||
int bc_icons[5] = { IDI_BATTCHA0, IDI_BATTCHA1, IDI_BATTCHA2, IDI_BATTCHA3, IDI_BATTCHA4 };
|
||||
|
||||
typedef struct _PWRSCHEMECONTEXT
|
||||
{
|
||||
HMENU hPopup;
|
||||
|
@ -25,66 +37,181 @@ typedef struct _PWRSCHEMECONTEXT
|
|||
UINT uiLast;
|
||||
} PWRSCHEMECONTEXT, *PPWRSCHEMECONTEXT;
|
||||
|
||||
//static HICON g_hIconBattery = NULL;
|
||||
static float g_batCap = 0;
|
||||
static HICON g_hIconBattery = NULL;
|
||||
static HICON g_hIconAC = NULL;
|
||||
|
||||
static BOOL g_IsRunning = FALSE;
|
||||
|
||||
/*** This function enumerates the available battery devices and provides the remaining capacity
|
||||
@param cap: if no error occurs, then this will contatin average remaining capacity
|
||||
@param dwResult: helps in making battery type checks
|
||||
{
|
||||
Returned value includes GBS_HASBATTERY if the system has a non-UPS battery, and GBS_ONBATTERY if the system is running on a battery.
|
||||
dwResult & GBS_ONBATTERY means we have not yet found AC power.
|
||||
dwResult & GBS_HASBATTERY means we have found a non-UPS battery.
|
||||
}
|
||||
@return : error checking
|
||||
*/
|
||||
static HRESULT GetBatteryState(float& cap, DWORD& dwResult)
|
||||
{
|
||||
cap = 0;
|
||||
dwResult = GBS_ONBATTERY;
|
||||
|
||||
HDEVINFO hdev = SetupDiGetClassDevs(&GUID_DEVCLASS_BATTERY, 0, 0, DIGCF_PRESENT | DIGCF_DEVICEINTERFACE);
|
||||
if (INVALID_HANDLE_VALUE == hdev)
|
||||
return E_HANDLE;
|
||||
|
||||
// Limit search to 100 batteries max
|
||||
for (int idev = 0, count = 0; idev < 100; idev++)
|
||||
{
|
||||
SP_DEVICE_INTERFACE_DATA did = { 0 };
|
||||
did.cbSize = sizeof(did);
|
||||
|
||||
if (SetupDiEnumDeviceInterfaces(hdev, 0, &GUID_DEVCLASS_BATTERY, idev, &did))
|
||||
{
|
||||
DWORD cbRequired = 0;
|
||||
|
||||
SetupDiGetDeviceInterfaceDetail(hdev, &did, 0, 0, &cbRequired, 0);
|
||||
if (ERROR_INSUFFICIENT_BUFFER == GetLastError())
|
||||
{
|
||||
PSP_DEVICE_INTERFACE_DETAIL_DATA pdidd = (PSP_DEVICE_INTERFACE_DETAIL_DATA)LocalAlloc(LPTR, cbRequired);
|
||||
if (pdidd)
|
||||
{
|
||||
pdidd->cbSize = sizeof(*pdidd);
|
||||
if (SetupDiGetDeviceInterfaceDetail(hdev, &did, pdidd, cbRequired, &cbRequired, 0))
|
||||
{
|
||||
// Enumerated a battery. Ask it for information.
|
||||
HANDLE hBattery = CreateFile(pdidd->DevicePath, GENERIC_READ | GENERIC_WRITE,
|
||||
FILE_SHARE_READ | FILE_SHARE_WRITE, NULL, OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
|
||||
|
||||
if (INVALID_HANDLE_VALUE != hBattery)
|
||||
{
|
||||
// Ask the battery for its tag.
|
||||
BATTERY_QUERY_INFORMATION bqi = { 0 };
|
||||
|
||||
DWORD dwWait = 0;
|
||||
DWORD dwOut;
|
||||
|
||||
if (DeviceIoControl(hBattery, IOCTL_BATTERY_QUERY_TAG, &dwWait, sizeof(dwWait), &bqi.BatteryTag,
|
||||
sizeof(bqi.BatteryTag), &dwOut, NULL) && bqi.BatteryTag)
|
||||
{
|
||||
// With the tag, you can query the battery info.
|
||||
BATTERY_INFORMATION bi = { 0 };
|
||||
bqi.InformationLevel = BatteryInformation;
|
||||
|
||||
if (DeviceIoControl(hBattery, IOCTL_BATTERY_QUERY_INFORMATION, &bqi, sizeof(bqi), &bi,
|
||||
sizeof(bi), &dwOut, NULL))
|
||||
{
|
||||
// Only non-UPS system batteries count
|
||||
if (bi.Capabilities & BATTERY_SYSTEM_BATTERY)
|
||||
{
|
||||
if (!(bi.Capabilities & BATTERY_IS_SHORT_TERM))
|
||||
dwResult |= GBS_HASBATTERY;
|
||||
|
||||
// Query the battery status.
|
||||
BATTERY_WAIT_STATUS bws = { 0 };
|
||||
bws.BatteryTag = bqi.BatteryTag;
|
||||
|
||||
BATTERY_STATUS bs;
|
||||
if (DeviceIoControl(hBattery, IOCTL_BATTERY_QUERY_STATUS, &bws, sizeof(bws),
|
||||
&bs, sizeof(bs), &dwOut, NULL))
|
||||
{
|
||||
if (bs.PowerState & BATTERY_POWER_ON_LINE)
|
||||
dwResult &= ~GBS_ONBATTERY;
|
||||
|
||||
// Take average of total capacity of batteries detected!
|
||||
cap = cap*(count)+(float)bs.Capacity / bi.FullChargedCapacity * 100;
|
||||
cap /= count + 1;
|
||||
count++;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
CloseHandle(hBattery);
|
||||
}
|
||||
}
|
||||
LocalFree(pdidd);
|
||||
}
|
||||
}
|
||||
}
|
||||
else if (ERROR_NO_MORE_ITEMS == GetLastError())
|
||||
{
|
||||
break; // Enumeration failed - perhaps we're out of items
|
||||
}
|
||||
}
|
||||
SetupDiDestroyDeviceInfoList(hdev);
|
||||
|
||||
// Final cleanup: If we didn't find a battery, then presume that we
|
||||
// are on AC power.
|
||||
|
||||
if (!(dwResult & GBS_HASBATTERY))
|
||||
dwResult &= ~GBS_ONBATTERY;
|
||||
|
||||
return S_OK;
|
||||
}
|
||||
|
||||
/*** This function quantizes the mentioned quantity to nearest level
|
||||
@param p: should be a quantity in percentage
|
||||
@param lvl: quantization level, default is 10
|
||||
@return : nearest quantized level
|
||||
*/
|
||||
static UINT Quantize(float p, UINT lvl = 10)
|
||||
{
|
||||
int i = 0;
|
||||
float f, q = (float)100 / lvl, d = q / 2;
|
||||
for (f = 0; f < p; f += q, i++);
|
||||
|
||||
if ((f - d) <= p)
|
||||
return i;
|
||||
else
|
||||
return i - 1;
|
||||
}
|
||||
|
||||
static HICON DynamicLoadIcon(HINSTANCE hinst)
|
||||
{
|
||||
HICON hBatIcon;
|
||||
float cap = 0;
|
||||
DWORD dw = 0;
|
||||
UINT index = -1;
|
||||
HRESULT hr = GetBatteryState(cap, dw);
|
||||
if (!FAILED(hr) && (dw & GBS_HASBATTERY))
|
||||
{
|
||||
index = Quantize(cap, 4);
|
||||
g_batCap = cap;
|
||||
}
|
||||
|
||||
if (dw & GBS_ONBATTERY)
|
||||
hBatIcon = LoadIcon(hinst, MAKEINTRESOURCE(index<0 ? IDI_BATTCAP_ERR : br_icons[index]));
|
||||
else
|
||||
hBatIcon = LoadIcon(hinst, MAKEINTRESOURCE(index<0 ? IDI_BATTCAP_ERR : bc_icons[index]));
|
||||
|
||||
return hBatIcon;
|
||||
}
|
||||
|
||||
HRESULT STDMETHODCALLTYPE Power_Init(_In_ CSysTray * pSysTray)
|
||||
{
|
||||
WCHAR strTooltip[128];
|
||||
HICON icon;
|
||||
|
||||
TRACE("Power_Init\n");
|
||||
|
||||
// g_hIconBattery = LoadIcon(g_hInstance, MAKEINTRESOURCE(IDI_BATTERY));
|
||||
g_hIconAC = LoadIcon(g_hInstance, MAKEINTRESOURCE(IDI_BATTERY));
|
||||
|
||||
|
||||
HICON icon;
|
||||
// if (g_IsMute)
|
||||
// icon = g_hIconBattery;
|
||||
// else
|
||||
icon = g_hIconAC;
|
||||
|
||||
LoadStringW(g_hInstance, IDS_PWR_AC, strTooltip, _countof(strTooltip));
|
||||
|
||||
g_hIconBattery = DynamicLoadIcon(g_hInstance);
|
||||
swprintf(strTooltip, L"%.2f %% Remaining", g_batCap);
|
||||
g_IsRunning = TRUE;
|
||||
|
||||
return pSysTray->NotifyIcon(NIM_ADD, ID_ICON_POWER, icon, strTooltip);
|
||||
return pSysTray->NotifyIcon(NIM_ADD, ID_ICON_POWER, g_hIconBattery, strTooltip);
|
||||
}
|
||||
|
||||
HRESULT STDMETHODCALLTYPE Power_Update(_In_ CSysTray * pSysTray)
|
||||
{
|
||||
// BOOL PrevState;
|
||||
|
||||
TRACE("Power_Update\n");
|
||||
|
||||
#if 0
|
||||
PrevState = g_IsMute;
|
||||
Volume_IsMute();
|
||||
|
||||
if (PrevState != g_IsMute)
|
||||
{
|
||||
WCHAR strTooltip[128];
|
||||
HICON icon;
|
||||
if (g_IsMute) {
|
||||
icon = g_hIconMute;
|
||||
LoadStringW(g_hInstance, IDS_VOL_MUTED, strTooltip, _countof(strTooltip));
|
||||
}
|
||||
else {
|
||||
icon = g_hIconVolume;
|
||||
LoadStringW(g_hInstance, IDS_VOL_VOLUME, strTooltip, _countof(strTooltip));
|
||||
}
|
||||
g_hIconBattery = DynamicLoadIcon(g_hInstance);
|
||||
swprintf(strTooltip, L"%.2f %% Remaining", g_batCap);
|
||||
|
||||
return pSysTray->NotifyIcon(NIM_MODIFY, ID_ICON_POWER, icon, strTooltip);
|
||||
}
|
||||
else
|
||||
{
|
||||
return S_OK;
|
||||
}
|
||||
#endif
|
||||
return S_OK;
|
||||
return pSysTray->NotifyIcon(NIM_MODIFY, ID_ICON_POWER, g_hIconBattery, strTooltip);
|
||||
}
|
||||
|
||||
HRESULT STDMETHODCALLTYPE Power_Shutdown(_In_ CSysTray * pSysTray)
|
||||
|
|
|
@ -39,4 +39,17 @@
|
|||
#define IDS_KEYS_MOUSE 331
|
||||
#define IDS_KEYS_FILTER 332
|
||||
|
||||
#define IDI_BATTCAP0 400
|
||||
#define IDI_BATTCAP1 401
|
||||
#define IDI_BATTCAP2 402
|
||||
#define IDI_BATTCAP3 403
|
||||
#define IDI_BATTCAP4 404
|
||||
#define IDI_BATTCAP5 405
|
||||
#define IDI_BATTCHA0 406
|
||||
#define IDI_BATTCHA1 407
|
||||
#define IDI_BATTCHA2 408
|
||||
#define IDI_BATTCHA3 409
|
||||
#define IDI_BATTCHA4 410
|
||||
#define IDI_BATTCAP_ERR 412
|
||||
|
||||
#define IDR_SYSTRAY 11001
|
||||
|
|
BIN
reactos/dll/shellext/stobject/resources/battery/0.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/1.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/2.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/3.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/4.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/5.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/charging0.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/charging1.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/charging2.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/charging3.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/charging4.ico
Normal file
After Width: | Height: | Size: 9.5 KiB |
BIN
reactos/dll/shellext/stobject/resources/battery/error.ico
Normal file
After Width: | Height: | Size: 25 KiB |
|
@ -11,6 +11,21 @@ IDI_EXTRACT ICON "resources/2.ico"
|
|||
IDI_VOLUME ICON "resources/3.ico"
|
||||
IDI_VOLMUTE ICON "resources/4.ico"
|
||||
|
||||
IDI_BATTCAP0 ICON "resources/battery/0.ico"
|
||||
IDI_BATTCAP1 ICON "resources/battery/1.ico"
|
||||
IDI_BATTCAP2 ICON "resources/battery/2.ico"
|
||||
IDI_BATTCAP3 ICON "resources/battery/3.ico"
|
||||
IDI_BATTCAP4 ICON "resources/battery/4.ico"
|
||||
IDI_BATTCAP5 ICON "resources/battery/5.ico"
|
||||
|
||||
IDI_BATTCHA0 ICON "resources/battery/charging0.ico"
|
||||
IDI_BATTCHA1 ICON "resources/battery/charging1.ico"
|
||||
IDI_BATTCHA2 ICON "resources/battery/charging2.ico"
|
||||
IDI_BATTCHA3 ICON "resources/battery/charging3.ico"
|
||||
IDI_BATTCHA4 ICON "resources/battery/charging4.ico"
|
||||
|
||||
IDI_BATTCAP_ERR ICON "resources/battery/error.ico"
|
||||
|
||||
IDR_SYSTRAY REGISTRY "resources/rgs/systray.rgs"
|
||||
|
||||
#include <reactos/manifest_dll.rc>
|
||||
|
|