X-Git-Url: https://bilbo.iut-bm.univ-fcomte.fr/and/gitweb/loba.git/blobdiff_plain/6f5ec5fdc42f96a8fe95f4b846b163d4dc92e0c8..035e2d971fc603174fa4c6883359f02d34b77226:/process.h?ds=inline diff --git a/process.h b/process.h index 586da55..053e348 100644 --- a/process.h +++ b/process.h @@ -1,25 +1,89 @@ #ifndef PROCESS_H #define PROCESS_H +#define USE_UNORDERED_MAP 1 +//#undef USE_UNORDERED_MAP + #include -#include +#ifdef USE_UNORDERED_MAP +# include +# define MAP_TEMPLATE std::tr1::unordered_map +#else +# include +# define MAP_TEMPLATE std::map +#endif +#include #include "communicator.h" #include "neighbor.h" class process { public: + static double get_total_load_init() { return total_load_init; } + static double get_total_load_exit() { return total_load_exit; } + process(int argc, char* argv[]); - ~process() { }; + virtual ~process(); + int run(); +protected: + typedef std::vector neigh_type; + typedef std::vector pneigh_type; + + pneigh_type pneigh; // list of pointers to neighbors that + // we are free to reorder + + // Returns the sum of "to_send" for all neighbors. + double sum_of_to_send() const; + private: - communicator comm; - std::vector neigh; - double load; - double expected_load; + static double total_load_init; + static double total_load_exit; + + typedef MAP_TEMPLATE rev_neigh_type; + neigh_type neigh; // list of neighbors (do not alter + // after construction!) + rev_neigh_type rev_neigh; // map m_host_t -> neighbor + + communicator comm; // communicator for this process + int ctrl_close_pending; // number of "close" messages to wait + // on ctrl channel + int data_close_pending; // number of "close" messages to wait + // on data channel - void receive(); + unsigned iter; // counter of iterations + + double prev_load_broadcast; // used to ensure that we do not send + // a same information messages + double load; // current load + double expected_load; // expected load in bookkeeping mode + + // The load balancing algorithm comes here... + // Parameter "my_load" is the load to take into account for myself + // (may be load or expected load). + // Returns the total load sent to neighbors. + virtual double load_balance(double my_load); + + // Virtually do some computation void compute(); + + // Send procedures, with helpers for bookkeeping mode or not + void send1_no_bookkeeping(neighbor& nb); + void send1_bookkeeping(neighbor& nb); + void send(); + + // Receive procedure: wait (or not) for a message to come. + // Returns false if some "close" message was received, returns true + // otherwise. + enum recv_wait_mode { NO_WAIT = 0, WAIT, WAIT_FOR_CLOSE }; + bool receive(recv_wait_mode wait); + + // Finalize sends a "close" message to each neighbor and wait for + // all of them to answer. + void finalize1(neighbor& nb); + void finalize(); + + // Print with given priority what we know about our neighbors' loads void print_loads(e_xbt_log_priority_t logp = xbt_log_priority_info); };