MultistepLB_SFC.h

#ifndef _MULTISTEPLB_SFC_H_
#define _MULTISTEPLB_SFC_H_
 
#include "MultistepLB_SFC.decl.h"
#include "Vector3D.h"
#include "cosmoType.h"
#include "SFC.h"
#include "CentralLB.h"
 
void Orb_PrintLBStats(BaseLB::LDStats *stats, int numobjs);
void write_LB_particles(BaseLB::LDStats* stats, const char *achFileName, bool bFrom);

 
class MultistepLB_SFC : public CBase_MultistepLB_SFC {
private:
    void init();
    bool QueryBalanceNow(int step);
 
    decltype(BaseLB::LDStats::to_proc) *mapping;
    decltype(BaseLB::LDStats::from_proc) *from;
    double dTotalLoad;
    double maxPieceProc;
 
public:
    MultistepLB_SFC(const CkLBOptions &);
    MultistepLB_SFC(CkMigrateMessage *m) : CBase_MultistepLB_SFC(m) {
        init();
    }
 
    class SFCObject
    {
    public:
        int lbindex;
        Vector3D<cosmoType> centroid;
        SFC::Key key;
        double load;
 
        SFCObject() : lbindex(-1), load(0) {}
        SFCObject(int _lbindex, double _load) :
            lbindex(_lbindex),
            load(_load)
        {
        }
        bool operator<(const SFCObject &o) const{
            return key < o.key;
        }
    };
 
    void work(BaseLB::LDStats* stats);
    void work2(BaseLB::LDStats* stats);
    void sfcPrepare(std::vector<SFCObject> &tp_array,
                    int nObjs, BaseLB::LDStats * stats,
                    bool node_partition=false);
    void sfcPartition(int nProcs, std::vector<SFCObject> & tp,
                      BaseLB::LDStats *stats, bool node_partition=false);
    void pup(PUP::er &p);
};
 
 
#endif /* _MultistepLB_notopo */