#ifndef INCLUDED_TASK_MANAGER
#define INCLUDED_TASK_MANAGER

#include "TaskManagerImpl.h"
#include "Random.h"
#include "MemList.h"
#include "Scheduler.h"

class TaskManager {
public:
    /* constructor */
    TaskManager(int num = 1); // The number of threads
    ~TaskManager(void);

    /* variables */
    TaskManagerImpl *m_impl;
    void (*tm_end)(void);
    RandomPtr gen_random;

    /* user function */
    HTaskPtr create_task(int cmd);
    void run(void);
    void *allocate(int size);
    void set_TMend(void (*endf)(void));
    int get_cpuNum(void);
    int get_random(void);
    Scheduler *get_scheduler();

    /* functions */
    void init(void);
    void finish(void);

private:
    int machineNum;
};

#endif

extern TaskManager *manager;