esp8266/cores/esp8266/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 void init_lists()
{
    if (sCount != 0) {
        return;
    }
    while (sCount < SCHEDULED_FN_INITIAL_COUNT) {
        scheduled_fn_t* it = new scheduled_fn_t;
        if (sCount == 0) {
            sFirstUnused = it;
        }
        else {
            sLastUnused->mNext = it;
        }
        sLastUnused = it;
        ++sCount;
    }
    sLastUnused->mNext = NULL;
}

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()
{
    while (sFirst) {
        scheduled_fn_t* item = sFirst;
        sFirst = item->mNext;
        if (sFirst == NULL) {
            sLast = NULL;
        }
        item->mFunc();
        item->mFunc = std::function<void(void)>();
        recycle_fn(item);
    }
}