CPowerThread更新至v1.1.0

//========================================================================
//TITLE:
//    CPowerThread更新至v1.1.0
//AUTHOR:
//    norains
//DATE:
//    Tuesday  25-December-2007
//Environment:
//    EVC4.0 + Windows CE 5.0 Standard SDK
//    EVC4.0 + SDK-WINCE5.0-MIPSII
//    VS2005 + SDK-WINCE5.0-MIPSII   
//========================================================================

    相对于v1.0.0版本,有如下变动:
   
    1.去掉GetInstance()函数,不用获取对象实例,可以直接声明对象实例使用
   
    2.改进该类以令其在继承之后能正常工作.
   
    3.增加OnNotifyPower()函数,可以不设置回调函数而通过重载该函数监视电源的变化.
   
    v1.0.0版本:http://blog.csdn.net/norains/archive/2007/07/20/1700980.aspx 

//////////////////////////////////////////////////////////////////////
// PowerThread.h: interface for the CPowerThread class.
//
//Version:
//    1.1.0
//Date:
//    2007.11.15
//////////////////////////////////////////////////////////////////////

#ifndef POWERTHREAD_H
#define POWERTHREAD_H



#include "Pm.h"



//-----------------------------------------------------------------
//Enum data type
enum PowerStatusType
{
    POW_UNKNOW,        //Unknow the status
    POW_CHARGING,    //It's charging now
    POW_CHARGEFULL,    //Full charge
    POW_VLOW,        //The battery level is very low
    POW_LOW,
    POW_NORMAL,
    POW_HIGH,
    POW_VHIGH        //The battery level is very high
};
//----------------------------------------------------------------

class CPowerThread  
{
public:
    BOOL ResumnSign(); //Not support now
    void GetCallbackFunction(void (* *pCallbackFunc)(PowerStatusType powStatus, int iBatteryPercent));
    void SetCallbackFunction(void (*pCallbackFunc)(PowerStatusType powStatus, int iBatteryPercent));
    void StopCapture();
    BOOL StartCapture();
    BOOL GetRunStatus();
    void SetTimeout(ULONG ulTime);
    
    CPowerThread();
    virtual ~CPowerThread();


protected:
    virtual void OnNotifyPower(PowerStatusType powStatus, int iBatteryPercent);
        
    
    //The critical section function
    inline void    InitLock()    { InitializeCriticalSection(&m_csLock); }
    inline void LockThis()    { EnterCriticalSection(&m_csLock);      }    
    inline void    UnLockThis()  { LeaveCriticalSection(&m_csLock);      }
    inline void    DelLock()     { DeleteCriticalSection(&m_csLock);     }


private:
    PowerStatusType GetPowerStatus(PPOWER_BROADCAST pPowerInfo, int *piPercent);
    static DWORD WINAPI PowerThread(PVOID pArg);

    BOOL m_bExitThread;
    ULONG m_ulWaitTime;
    BOOL m_bRunning;
    CRITICAL_SECTION m_csLock;

    //This is for callback function.
    void (*m_pNotifyPower)(PowerStatusType powStatus, int iBatteryPercent);
};

#endif //#ifndef POWERTHREAD_H

 
//////////////////////////////////////////////////////////////////////   
// PowerThread.cpp: implementation of the CPowerThread class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
#include "PowerThread.h"
#include "Msgqueue.h"


//--------------------------------------------------------------------
//Macro define
#define DEFAULT_TIMEOUT        1000//1000ms

//----------------------------------------------------------------------
//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CPowerThread::CPowerThread():
m_bExitThread(TRUE),
m_ulWaitTime(DEFAULT_TIMEOUT),
m_bRunning(FALSE),
m_pNotifyPower(NULL)
{
    InitLock();
}

CPowerThread::~CPowerThread()
{
    DelLock();
}



//------------------------------------------------------------------
//Description:
//    Get the level of power from the PPOWER_BROADCAST struct
//
//Parameters:
//    pPowerInfo:[in] The struct includes the power information
//    piPercent:[out] The battery life percent.
//
//Return Values:
//    The power status
//----------------------------------------------------------------
PowerStatusType CPowerThread::GetPowerStatus(PPOWER_BROADCAST pPowerInfo, int *piPercent)
{
    PowerStatusType powStatus = POW_UNKNOW;

    if ( !pPowerInfo )
    {
        return POW_UNKNOW;
    }
    
    PPOWER_BROADCAST_POWER_INFO ppbpi = (PPOWER_BROADCAST_POWER_INFO) pPowerInfo->SystemPowerState;
    if ( !ppbpi )
    {
        return POW_UNKNOW;
    }
    
    *piPercent = ppbpi->bBatteryLifePercent; 
    
    if(ppbpi->bACLineStatus == AC_LINE_ONLINE)
    {        
        if(ppbpi->bBatteryFlag == BATTERY_FLAG_CHARGING)
        {
            //Charging
            powStatus = POW_CHARGING;
        }
        else
        {
            //May be full charging,or may be no battery
            powStatus = POW_CHARGEFULL;
        }
    }
    else
    {
        //Use battery
    
        if(0 <= ppbpi->bBatteryLifePercent && ppbpi->bBatteryLifePercent <= 20)
        {
            powStatus = POW_VLOW;
        }
        else if(20 < ppbpi->bBatteryLifePercent && ppbpi->bBatteryLifePercent <= 40)
        {
            powStatus = POW_LOW;
        }
        else if(40 < ppbpi->bBatteryLifePercent && ppbpi->bBatteryLifePercent <=60)
        {
            powStatus = POW_NORMAL;
        }
        else if(60 < ppbpi->bBatteryLifePercent && ppbpi->bBatteryLifePercent <=80)
        {
            powStatus = POW_HIGH;
        }
        else if(80 < ppbpi->bBatteryLifePercent && ppbpi->bBatteryLifePercent <= 100)
        {
            powStatus = POW_VHIGH;
        }
        else
        {
            powStatus = POW_UNKNOW;
        }
    }

    return powStatus;
}




//------------------------------------------------------------------
//Description:
//    Thread to get the power status
//----------------------------------------------------------------
DWORD WINAPI CPowerThread::PowerThread(PVOID pArg)
{
    CPowerThread *pObject = (CPowerThread *) pArg;

    pObject->m_bRunning = TRUE;

    BYTE pbMsgBuf[sizeof(POWER_BROADCAST) + sizeof(POWER_BROADCAST_POWER_INFO)];
    PPOWER_BROADCAST ppb = (PPOWER_BROADCAST) pbMsgBuf;
    MSGQUEUEOPTIONS msgopts;
    
    // Create our message queue
    memset(&msgopts, 0, sizeof(msgopts));
    msgopts.dwSize = sizeof(msgopts);
    msgopts.dwFlags = 0;
    msgopts.dwMaxMessages = 0;
    msgopts.cbMaxMessage = sizeof(pbMsgBuf);
    msgopts.bReadAccess = TRUE;
    
    HANDLE rghWaits[1] = { NULL };
    rghWaits[0] = CreateMsgQueue(NULL, &msgopts);
    if (!rghWaits[0])
    {
        //erro
        return 0x10;
    
    }
    
    HANDLE hReq = NULL;
    // Request notifications
    hReq = RequestPowerNotifications(rghWaits[0], PBT_POWERINFOCHANGE);
    if (!hReq)
    {
        CloseHandle( rghWaits[ 0 ] );
        //erro
        return 0x15;

    }      
    


    while(pObject->m_bExitThread == FALSE)
    {
        

        DWORD dwWaitCode = MsgWaitForMultipleObjectsEx( 1, rghWaits, pObject->m_ulWaitTime, QS_ALLINPUT, MWMO_INPUTAVAILABLE );
        if ( dwWaitCode == WAIT_OBJECT_0 )
        {              
            DWORD dwSize, dwFlags;
            BOOL bReadResult = ReadMsgQueue(rghWaits[0], ppb, sizeof(pbMsgBuf), &dwSize, 0, &dwFlags);
            if (bReadResult == TRUE)
            {
                int iPowPercent;
                PowerStatusType powStatus = pObject->GetPowerStatus(ppb,&iPowPercent);

                pObject->LockThis();
                if(pObject->m_pNotifyPower != NULL)
                {
                    pObject->m_pNotifyPower(powStatus,iPowPercent);
                }
                pObject->OnNotifyPower(powStatus,iPowPercent);
                pObject->UnLockThis();

            }
            else
            {
                // We should never get here
                break;                
            }
        }


    }
    pObject->m_bRunning = FALSE;
    
    return 0;
}




//------------------------------------------------------------------
//Description:
//    Set the timeout for the wait thread.  It is only for the MsgWaitForMultipleObjectsEx()
//The default value is DEFAULT_TIMEOUT
//----------------------------------------------------------------
void CPowerThread::SetTimeout(ULONG ulTime)
{
    m_ulWaitTime = ulTime;
}


//------------------------------------------------------------------
//Description:
//    Get the status of thread
//
//Return Values:
//    TRUE: The thread is running for capturing the power status.
//    FALSE: No thread running.
//----------------------------------------------------------------
BOOL CPowerThread::GetRunStatus()
{
    return m_bRunning;
}


//------------------------------------------------------------------
//Description:
//    start capturing the power status.If there is thread running,
//it will return FALSE;
//
//------------------------------------------------------------------
BOOL CPowerThread::StartCapture()
{
    if(m_bRunning == TRUE)
    {
        return FALSE;
    }

    m_bExitThread = FALSE;

    //Create the thread for batter sampled    
    DWORD dwPwrThdID;
    HANDLE hdThrd = CreateThread(NULL,0,PowerThread,(void *)this,0,&dwPwrThdID);
    if(hdThrd == NULL)
    {
        return FALSE;
    }
    CloseHandle(hdThrd);

    return TRUE;
}



//-----------------------------------------------------------------------------
//Description:
//    Stop capturing.
//
//--------------------------------------------------------------------------------
void CPowerThread::StopCapture()
{
    m_bExitThread = TRUE;
}


//------------------------------------------------------------------
//Description:
//    Set the callback function for receive the power status
//------------------------------------------------------------------
void CPowerThread::SetCallbackFunction(void (*pCallbackFunc)(PowerStatusType powStatus, int iBatteryPercent))
{
    LockThis();
    m_pNotifyPower = pCallbackFunc;
    UnLockThis();
}



//------------------------------------------------------------------
//Description:
//    Get the callback function
//------------------------------------------------------------------
void CPowerThread::GetCallbackFunction(void (* *pCallbackFunc)(PowerStatusType powStatus, int iBatteryPercent))
{
    LockThis();
    *pCallbackFunc = m_pNotifyPower;
    UnLockThis();
}


//------------------------------------------------------------------
//Description:
//    Resumn the sign order to get the current power status,or the power
//status will return untill the power status changed.
//------------------------------------------------------------------
BOOL CPowerThread::ResumnSign()
{
    /*
    BYTE pbMsgBuf[sizeof(POWER_BROADCAST) + sizeof(POWER_BROADCAST_POWER_INFO)];
    PPOWER_BROADCAST ppb = (PPOWER_BROADCAST) pbMsgBuf;
    MSGQUEUEOPTIONS msgopts;
    
    // Create our message queue
    memset(&msgopts, 0, sizeof(msgopts));
    msgopts.dwSize = sizeof(msgopts);
    msgopts.dwFlags = 0;
    msgopts.dwMaxMessages = 0;
    msgopts.cbMaxMessage = sizeof(pbMsgBuf);
    msgopts.bReadAccess = TRUE;
    
    HANDLE rghWaits[1] = { NULL };
    rghWaits[0] = CreateMsgQueue(NULL, &msgopts);
    if (!rghWaits[0])
    {
        //erro
        return FALSE;
    
    }
    

    // Request notifications
    HANDLE hReq = RequestPowerNotifications(rghWaits[0], PBT_POWERINFOCHANGE);
    if (!hReq)
    {
        CloseHandle( rghWaits[ 0 ] );
        //erro
        return FALSE;

    }      




    return TRUE;
    */


    return FALSE;
}



//------------------------------------------------------------------
//Description:
//    Notify the power. It's used as call back funtion.
//------------------------------------------------------------------
void CPowerThread::OnNotifyPower(PowerStatusType powStatus, int iBatteryPercent)
{
    return;
}