#include "precomp.h"
#include <devguid.h>
#include <winioctl.h>
#include <powrprof.h>
#include <windows.h>
#include <batclass.h>

Include dependency graph for power.cpp:

Go to the source code of this file.

Classes
struct	_PWRSCHEMECONTEXT

Typedefs
typedef struct _PWRSCHEMECONTEXT	PWRSCHEMECONTEXT

typedef struct _PWRSCHEMECONTEXT *	PPWRSCHEMECONTEXT

Functions
static UINT	Quantize (BYTE p)

static HICON	DynamicLoadIcon (HINSTANCE hinst)

HRESULT STDMETHODCALLTYPE	Power_Init (_In_ CSysTray *pSysTray)

HRESULT STDMETHODCALLTYPE	Power_Update (_In_ CSysTray *pSysTray)

HRESULT STDMETHODCALLTYPE	Power_Shutdown (_In_ CSysTray *pSysTray)

static void	_RunPower ()

static void	_ShowContextMenu (CSysTray *pSysTray)

static BOOLEAN CALLBACK	PowerSchemesEnumProc (UINT uiIndex, DWORD dwName, LPWSTR sName, DWORD dwDesc, LPWSTR sDesc, PPOWER_POLICY pp, LPARAM lParam)

static VOID	ShowPowerSchemesPopupMenu (CSysTray *pSysTray)

HRESULT STDMETHODCALLTYPE	Power_Message (_In_ CSysTray *pSysTray, UINT uMsg, WPARAM wParam, LPARAM lParam, LRESULT &lResult)

Variables
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 }

CString	g_strTooltip

static HICON	g_hIconBattery = NULL

Typedef Documentation

◆ PPWRSCHEMECONTEXT

typedef struct _PWRSCHEMECONTEXT * PPWRSCHEMECONTEXT

◆ PWRSCHEMECONTEXT

typedef struct _PWRSCHEMECONTEXT PWRSCHEMECONTEXT

Function Documentation

◆ _RunPower()

static void _RunPower ( )

static

Definition at line 144 of file power.cpp.

{
    ShellExecuteW(NULL, NULL, L"powercfg.cpl", NULL, NULL, SW_SHOWNORMAL);
}

Referenced by _ShowContextMenu(), and Power_Message().

◆ _ShowContextMenu()

static void _ShowContextMenu ( CSysTray * pSysTray )

static

Definition at line 149 of file power.cpp.

{
    CString strOpen((LPCSTR)IDS_PWR_PROPERTIES);
    HMENU hPopup = CreatePopupMenu();
    AppendMenuW(hPopup, MF_STRING, IDS_PWR_PROPERTIES, strOpen);
    SetMenuDefaultItem(hPopup, IDS_PWR_PROPERTIES, FALSE);
 
    SetForegroundWindow(pSysTray->GetHWnd());
    DWORD flags = TPM_RETURNCMD | TPM_NONOTIFY | TPM_RIGHTALIGN | TPM_BOTTOMALIGN;
    POINT pt;
    GetCursorPos(&pt);
 
    DWORD id = TrackPopupMenuEx(hPopup, flags,
        pt.x, pt.y,
        pSysTray->GetHWnd(), NULL);
 
    switch (id)
    {
        case IDS_PWR_PROPERTIES:
            _RunPower();
            break;
    }
    DestroyMenu(hPopup);
}

Referenced by Power_Message().

◆ DynamicLoadIcon()

static HICON DynamicLoadIcon ( HINSTANCE hinst )

static

Definition at line 77 of file power.cpp.

{
    SYSTEM_POWER_STATUS PowerStatus;
    HICON hBatIcon;
    UINT index = -1;
 
    if (!GetSystemPowerStatus(&PowerStatus) ||
        PowerStatus.ACLineStatus == AC_LINE_UNKNOWN ||
        PowerStatus.BatteryFlag == BATTERY_FLAG_UNKNOWN)
    {
        hBatIcon = LoadIcon(hinst, MAKEINTRESOURCE(IDI_BATTCAP_ERR));
        g_strTooltip.LoadStringW(IDS_PWR_UNKNOWN_REMAINING);
        return hBatIcon;
    }
 
    if (((PowerStatus.BatteryFlag & BATTERY_FLAG_NO_BATTERY) == 0) &&
        (PowerStatus.BatteryLifePercent == BATTERY_PERCENTAGE_UNKNOWN))
    {
        hBatIcon = LoadIcon(hinst, MAKEINTRESOURCE(IDI_BATTCAP_ERR));
        g_strTooltip.LoadStringW(IDS_PWR_UNKNOWN_REMAINING);
    }
    else if (((PowerStatus.BatteryFlag & BATTERY_FLAG_NO_BATTERY) == 0) &&
        ((PowerStatus.BatteryFlag & BATTERY_FLAG_CHARGING) == BATTERY_FLAG_CHARGING))
    {
        index = Quantize(PowerStatus.BatteryLifePercent);
        hBatIcon = LoadIcon(hinst, MAKEINTRESOURCE(bc_icons[index]));
        g_strTooltip.Format(IDS_PWR_CHARGING, PowerStatus.BatteryLifePercent);
    }
    else if (((PowerStatus.BatteryFlag & BATTERY_FLAG_NO_BATTERY) == 0) &&
             ((PowerStatus.BatteryFlag & BATTERY_FLAG_CHARGING) == 0))
    {
        index = Quantize(PowerStatus.BatteryLifePercent);
        hBatIcon = LoadIcon(hinst, MAKEINTRESOURCE(br_icons[index]));
        g_strTooltip.Format(IDS_PWR_PERCENT_REMAINING, PowerStatus.BatteryLifePercent);
    }
    else
    {
        hBatIcon = LoadIcon(hinst, MAKEINTRESOURCE(IDI_POWER_AC));
        g_strTooltip.LoadStringW(IDS_PWR_AC);
    }
 
    return hBatIcon;
}

Referenced by Power_Init(), and Power_Update().

◆ Power_Init()

HRESULT STDMETHODCALLTYPE Power_Init ( _In_ CSysTray * pSysTray )

Definition at line 121 of file power.cpp.

{
    TRACE("Power_Init\n");
    g_hIconBattery = DynamicLoadIcon(g_hInstance);
 
    return pSysTray->NotifyIcon(NIM_ADD, ID_ICON_POWER, g_hIconBattery, g_strTooltip);
}

Referenced by Power_Message().

◆ Power_Message()

HRESULT STDMETHODCALLTYPE Power_Message	(	_In_ CSysTray *	pSysTray,
		UINT	uMsg,
		WPARAM	wParam,
		LPARAM	lParam,
		LRESULT &	lResult
	)

Definition at line 236 of file power.cpp.

{
    TRACE("Power_Message uMsg=%d, wParam=%x, lParam=%x\n", uMsg, wParam, lParam);
 
    switch (uMsg)
    {
        case WM_USER + 220:
            TRACE("Power_Message: WM_USER+220\n");
            if (wParam == POWER_SERVICE_FLAG)
            {
                if (lParam)
                {
                    pSysTray->EnableService(POWER_SERVICE_FLAG, TRUE);
                    return Power_Init(pSysTray);
                }
                else
                {
                    pSysTray->EnableService(POWER_SERVICE_FLAG, FALSE);
                    return Power_Shutdown(pSysTray);
                }
            }
            return S_FALSE;
 
        case WM_USER + 221:
            TRACE("Power_Message: WM_USER+221\n");
            if (wParam == POWER_SERVICE_FLAG)
            {
                lResult = (LRESULT)pSysTray->IsServiceEnabled(POWER_SERVICE_FLAG);
                return S_OK;
            }
            return S_FALSE;
 
        case WM_TIMER:
            if (wParam == POWER_TIMER_ID)
            {
                KillTimer(pSysTray->GetHWnd(), POWER_TIMER_ID);
                ShowPowerSchemesPopupMenu(pSysTray);
            }
            break;
 
        case ID_ICON_POWER:
            Power_Update(pSysTray);
 
            switch (lParam)
            {
                case WM_LBUTTONDOWN:
                    SetTimer(pSysTray->GetHWnd(), POWER_TIMER_ID, GetDoubleClickTime(), NULL);
                    break;
 
                case WM_LBUTTONUP:
                    break;
 
                case WM_LBUTTONDBLCLK:
                    KillTimer(pSysTray->GetHWnd(), POWER_TIMER_ID);
                    _RunPower();
                    break;
 
                case WM_RBUTTONDOWN:
                    break;
 
                case WM_RBUTTONUP:
                    _ShowContextMenu(pSysTray);
                    break;
 
                case WM_RBUTTONDBLCLK:
                    break;
 
                case WM_MOUSEMOVE:
                    break;
            }
            return S_OK;
 
        default:
            TRACE("Power_Message received for unknown ID %d, ignoring.\n");
            return S_FALSE;
    }
 
    return S_FALSE;
}

◆ Power_Shutdown()

HRESULT STDMETHODCALLTYPE Power_Shutdown ( _In_ CSysTray * pSysTray )

Definition at line 137 of file power.cpp.

{
    TRACE("Power_Shutdown\n");
 
    return pSysTray->NotifyIcon(NIM_DELETE, ID_ICON_POWER, NULL, NULL);
}

Referenced by Power_Message().

◆ Power_Update()

HRESULT STDMETHODCALLTYPE Power_Update ( _In_ CSysTray * pSysTray )

Definition at line 129 of file power.cpp.

{
    TRACE("Power_Update\n");
    g_hIconBattery = DynamicLoadIcon(g_hInstance);
 
    return pSysTray->NotifyIcon(NIM_MODIFY, ID_ICON_POWER, g_hIconBattery, g_strTooltip);
}

Referenced by Power_Message().

◆ PowerSchemesEnumProc()

static BOOLEAN CALLBACK PowerSchemesEnumProc	(	UINT	uiIndex,
		DWORD	dwName,
		LPWSTR	sName,
		DWORD	dwDesc,
		LPWSTR	sDesc,
		PPOWER_POLICY	pp,
		LPARAM	lParam
	)

static

Definition at line 177 of file power.cpp.

{
    PPWRSCHEMECONTEXT PowerSchemeContext = (PPWRSCHEMECONTEXT)lParam;
 
    if (AppendMenuW(PowerSchemeContext->hPopup, MF_STRING, uiIndex + 1, sName))
    {
        if (PowerSchemeContext->uiFirst == 0)
            PowerSchemeContext->uiFirst = uiIndex + 1;
 
        PowerSchemeContext->uiLast = uiIndex + 1;
    }
 
    return TRUE;
}

Referenced by ShowPowerSchemesPopupMenu().

◆ Quantize()

static UINT Quantize ( BYTE p )

static

Definition at line 51 of file power.cpp.

{
    if (p <= 12)
        return 0;
    else if (p > 12 && p <= 37)
        return 1;
    else if (p > 37 && p <= 62)
        return 2;
    else if (p > 62 && p <= 87)
        return 3;
    else
        return 4;
}

Referenced by DynamicLoadIcon().

◆ ShowPowerSchemesPopupMenu()

static VOID ShowPowerSchemesPopupMenu ( CSysTray * pSysTray )

static

Definition at line 201 of file power.cpp.

{
    PWRSCHEMECONTEXT PowerSchemeContext = {NULL, 0, 0};
    UINT uiActiveScheme;
    DWORD id;
    POINT pt;
    PowerSchemeContext.hPopup = CreatePopupMenu();
    EnumPwrSchemes(PowerSchemesEnumProc, (LPARAM)&PowerSchemeContext);
 
    if (GetActivePwrScheme(&uiActiveScheme))
    {
        CheckMenuRadioItem(PowerSchemeContext.hPopup,
                           PowerSchemeContext.uiFirst,
                           PowerSchemeContext.uiLast,
                           uiActiveScheme + 1,
                           MF_BYCOMMAND);
    }
 
    SetForegroundWindow(pSysTray->GetHWnd());
    GetCursorPos(&pt);
 
    id = TrackPopupMenuEx(PowerSchemeContext.hPopup,
                          TPM_RETURNCMD | TPM_NONOTIFY | TPM_RIGHTALIGN | TPM_BOTTOMALIGN,
                          pt.x,
                          pt.y,
                          pSysTray->GetHWnd(),
                          NULL);
 
    DestroyMenu(PowerSchemeContext.hPopup);
 
    if (id != 0)
        SetActivePwrScheme(id - 1, NULL, NULL);
}