windirstat/WorkLimiter.cpp


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56

#include "stdafx.h"
#include "WorkLimiter.h"

CWorkLimiter::CWorkLimiter()
    : m_done()
    , m_tickLimit()
    , m_prevTicks()
{
}

CWorkLimiter::~CWorkLimiter()
{
}

void CWorkLimiter::Start(ULONGLONG ticks)
{
    ULONGLONG start = CWorkLimiter::Now();
    m_done = false;
    m_tickLimit = start + ticks;
    m_prevTicks = start;
}

bool CWorkLimiter::IsDone() const
{
    if (m_done) return true;

    // check remaining ticks
    ULONGLONG now = CWorkLimiter::Now();
    // signed subtraction to deal with overflow
    ULONGLONG remaining = m_tickLimit - now;
    if (remaining <= 0)
    {
        m_done = true;
        return true;
    }

    // check if there are any pending window messages
    ULONGLONG elapsed = now - m_prevTicks;
    if (elapsed > 10)
    {
        m_prevTicks = now;
        MSG msg;
        if (PeekMessage(&msg, NULL, 0, 0, PM_NOREMOVE | PM_QS_INPUT))
        {
            m_done = true;
            return true;
        }
    }

    return false;
}

inline ULONGLONG CWorkLimiter::Now()
{
    return _GetTickCount64();
}