FunctionThread.h

Go to the documentation of this file.

#ifndef __FUNCTION_THREAD_H
#define __FUNCTION_THREAD_H

#include "SingleProducerConsumer.h"
#include "ThreadPool.h"
#include "RakMemoryOverride.h"
#include "Export.h"

namespace RakNet
{

// Forward declarations
class Functor;

class RAK_DLL_EXPORT FunctionThread
{
public:
        FunctionThread();
        ~FunctionThread();

        struct FunctorAndContext
        {
                Functor *functor;
                void *context;
        };

        void StartThreads(int numThreads);

        void StopThreads(bool blockOnCurrentProcessing);

        void Push(Functor *functor, void *context=0);

        void CallResultHandlers(void);

        void CancelFunctorsWithContext(bool (*cancelThisFunctor)(FunctorAndContext func, void *userData), void *userData);

        void SetPostResultFunction(void (*postResult)(FunctorAndContext func));


protected:
        void CancelInput(void);
        ThreadPool<FunctorAndContext, FunctorAndContext> threadPool;

        void (*pr)(FunctorAndContext func);
};

class Functor
{
public:
        Functor() {}
        virtual ~Functor() {}

        virtual void Process(void *context)=0;
        virtual void HandleResult(bool wasCancelled, void *context)=0;
};

class RAK_DLL_EXPORT FunctionThreadDependentClass
{
public:
        FunctionThreadDependentClass();
        virtual ~FunctionThreadDependentClass();

        virtual void AssignFunctionThread(FunctionThread *ft);

        FunctionThread *GetFunctionThread(void) const;

        bool GetFunctionThreadWasAllocated(void) const;

        virtual void PushFunctor(Functor *functor, void *context=0);
protected:
        void StartFunctionThread();
        FunctionThread *functionThread;
        bool functionThreadWasAllocated;
};

}

#endif