Home Research Papers Posters Manuals Talks Download People Help Internal

ck-ldb/GreedyLB.h

Go to the documentation of this file.

#ifndef _GREEDYLB_H_
#define _GREEDYLB_H_

#include "CentralLB.h"
#include "GreedyLB.decl.h"

void CreateGreedyLB();
BaseLB * AllocateGreedyLB();

class GreedyLB : public CentralLB {

public:
  struct HeapData {
    double load;
    int    pe;
    int    id;
  };

  GreedyLB(const CkLBOptions &);
  GreedyLB(CkMigrateMessage *m):CentralLB(m) { lbname = "GreedyLB"; }
  void work(LDStats* stats);
private:
        enum           HeapCmp {GT = '>', LT = '<'};
        void           Heapify(HeapData*, int, int, HeapCmp);
        void           HeapSort(HeapData*, int, HeapCmp);
        void           BuildHeap(HeapData*, int, HeapCmp);
        CmiBool        Compare(double, double, HeapCmp);
        HeapData*      BuildCpuArray(BaseLB::LDStats*, int, int *);  
        HeapData*      BuildObjectArray(BaseLB::LDStats*, int, int *);      
        CmiBool        QueryBalanceNow(int step);
};

#endif /* _HEAPCENTLB_H_ */

---