loadbalance.h

Go to the documentation of this file.

#ifndef LOADBALANCE_H
#define LOADBALANCE_H

#include "condition.inc"
#include "mutex.inc"
#include "thread.h"




// Load balancing utils
// There is no guarantee that all the load clients will be run in a 
// processing operation.


class LoadServer;



class LoadPackage
{
public:
        LoadPackage();
        virtual ~LoadPackage();
        
        Condition *completion_lock;
// Range to search in the total scan area
//      int pixel1, pixel2;
};


class LoadClient : public Thread
{
public:
        LoadClient(LoadServer *server);
        LoadClient();
        virtual ~LoadClient();
        
        void run();
        virtual void process_package(LoadPackage *package) {};
        int get_package_number();
        LoadServer* get_server();

        int done;
        int package_number;
        Condition *input_lock;
        Condition *completion_lock;
        LoadServer *server;
};




class LoadServer
{
public:
        LoadServer(int total_clients, int total_packages);
        virtual ~LoadServer();

        friend class LoadClient;

// Called first in process_packages.  Should also initialize clients.
        virtual void init_packages() {};
        virtual LoadClient* new_client() { return 0; };
        virtual LoadPackage* new_package() { return 0; };

// User calls this to do an iteration
        void process_packages();


        int get_total_packages();
        int get_total_clients();
        LoadPackage* get_package(int number);
        LoadClient* get_client(int number);
        void set_package_count(int total_packages);



        void delete_clients();
        void create_clients();
        void delete_packages();
        void create_packages();





        int current_package;
        LoadPackage **packages;
        int total_packages;
        LoadClient **clients;
        int total_clients;
        Mutex *client_lock;
};



#endif

---