Schedule.cpp

#include "Schedule.h"

struct scheduled_fn_t
{
    scheduled_fn_t* mNext;
    std::function<void(void)> mFunc;
};

static scheduled_fn_t* sFirst = 0;
static scheduled_fn_t* sLast = 0;

static scheduled_fn_t* sFirstUnused = 0;
static scheduled_fn_t* sLastUnused = 0;

static int sCount = 0;

static scheduled_fn_t* get_fn()
{
    scheduled_fn_t* result = NULL;
    // try to get an item from unused items list
    if (sFirstUnused)
    {
        result = sFirstUnused;
        sFirstUnused = result->mNext;
        if (sFirstUnused == NULL)
        {
            sLastUnused = NULL;
        }
    }
    // if no unused items, and count not too high, allocate a new one
    else if (sCount != SCHEDULED_FN_MAX_COUNT)
    {
        result = new scheduled_fn_t;
        result->mNext = NULL;
        ++sCount;
    }
    return result;
}

static void recycle_fn(scheduled_fn_t* fn)
{
    if (!sLastUnused)
    {
        sFirstUnused = fn;
    }
    else
    {
        sLastUnused->mNext = fn;
    }
    fn->mNext = NULL;
    sLastUnused = fn;
}

bool schedule_function(std::function<void(void)> fn)
{
    scheduled_fn_t* item = get_fn();
    if (!item)
    {
        return false;
    }
    item->mFunc = fn;
    item->mNext = NULL;
    if (!sFirst)
    {
        sFirst = item;
    }
    else
    {
        sLast->mNext = item;
    }
    sLast = item;
    return true;
}

void run_scheduled_functions()
{
    scheduled_fn_t* rFirst = sFirst;
    sFirst = NULL;
    sLast  = NULL;
    while (rFirst)
    {
        scheduled_fn_t* item = rFirst;
        rFirst = item->mNext;
        item->mFunc();
        item->mFunc = std::function<void(void)>();
        recycle_fn(item);
    }
}