MoleculeIdPool.h

#ifndef SRC_MOLECULES_MOLECULEIDPOOL_H_
#define SRC_MOLECULES_MOLECULEIDPOOL_H_
 
#include "utils/mardyn_assert.h"
 
class MoleculeIdPool {
    /* Local pool range is implemented by subdividing the total range of IDs into
     * numProcesses blocks of poolSize / numProcesses size assigning them consecutively
     * to the processes. */
public:
    MoleculeIdPool(unsigned long poolsize, int numProcs, int myProcRank) :
        _poolSize(poolsize), _numProcesses(numProcs), _myProcRank(myProcRank), _moleculesFromThisProcess(0) {}
 
    unsigned long getNewMoleculeId() {
        mardyn_assert(_moleculesFromThisProcess < localIdRangeSize());
        return myIDoffset() + _moleculesFromThisProcess++;
    }
    int getOwnerRank(unsigned long id) {
        return id / localIdRangeSize();
    }
    unsigned long localIdRangeSize(){
        static unsigned long localIdRangeSize = _poolSize / _numProcesses;
        return localIdRangeSize;
    }
 
private:
    unsigned long myIDoffset() {
        static unsigned long myIDoffset = localIdRangeSize() * _myProcRank;
        return myIDoffset;
    }
 
    unsigned long _poolSize;  
    int _numProcesses;  
    int _myProcRank;  
    unsigned long _moleculesFromThisProcess;  
};
 
#endif  // SRC_MOLECULES_MOLECULEIDPOOL_H_