MTDistributedIncomingSpikeScheduler.cpp

Go to the documentation of this file.

#include "MTDistributedIncomingSpikeScheduler.h"

MTDistributedIncomingSpikeScheduler::MTDistributedIncomingSpikeScheduler(int numThreads,
        ThreadPool &threadPool,
        MPIInputBufferVector &inputBuffers,
        GlobalMultiTargetDelayMap &globalDelayMap,
        vector<PropagatedSpikeBuffer*> &spikeBuffers,
        int cycleSteps) :
        nThreads(numThreads), thr_pool(threadPool)
{
    incomingSpikeSchedulers.reserve(nThreads);
    jobs.reserve(nThreads-1);
    for (int i = 0 ; i < nThreads; ++i) {
        STDistributedIncomingSpikeScheduler *stIncomingSpikeScheduler =
            new STDistributedIncomingSpikeScheduler(inputBuffers,
                                                    globalDelayMap[i],
                                                    *spikeBuffers[i],
                                                    cycleSteps);
        incomingSpikeSchedulers.push_back(stIncomingSpikeScheduler);
        if (i < nThreads - 1) {
            IncomingSpikeSchedulerThreadPoolJob *job = new IncomingSpikeSchedulerThreadPoolJob(
                        stIncomingSpikeScheduler);
            jobs.push_back(job);
        }
    }
}

MTDistributedIncomingSpikeScheduler::~MTDistributedIncomingSpikeScheduler()
{
    for (int i = 0 ; i < nThreads; ++i) {
        delete incomingSpikeSchedulers[i];
        if (i < nThreads-1)
            delete jobs[i];
    }
}

void MTDistributedIncomingSpikeScheduler::processMPIInputSpikeBuffers()
{
    for (int i = 0 ; i < nThreads - 1 ; ++i) {
        thr_pool.dispatch(i, *jobs[i]);
    }
    incomingSpikeSchedulers[nThreads-1]->processMPIInputSpikeBuffers();
    thr_pool.waitAll();
}

---