Examples with Range edu.rit.util.Range used on opensource projects

Example 76 with Range

use of edu.rit.util.Range in project ffx by mjschnie.

the class GradientSchedule method defineRanges.

private void defineRanges() {
    double totalWeight = totalWeight();
    /**
     * Infrequent edge case where the total weight is less than or equal to
     * the number of threads.
     */
    if (totalWeight <= nThreads) {
        Range temp = new Range(0, nAtoms - 1);
        ranges = temp.subranges(nThreads);
        return;
    }
    /**
     * Handle the case where we only have a single thread, which will
     * receive all the atoms.
     */
    if (nThreads == 1) {
        ranges[0] = new Range(0, nAtoms - 1);
        return;
    }
    double targetWeight = (totalWeight / nThreads);
    int lastAtom = nAtoms - 1;
    int currentAtom = 0;
    lowerBounds[0] = 0;
    int currentThread = 0;
    while (currentThread < nThreads) {
        int threadWeight = 0;
        while (threadWeight < targetWeight && currentAtom < lastAtom) {
            threadWeight += weights[currentAtom];
            currentAtom++;
        }
        currentThread++;
        if (currentAtom < lastAtom) {
            lowerBounds[currentThread] = currentAtom;
        } else {
            lowerBounds[currentThread] = lastAtom;
            break;
        }
    }
    int lastThread = currentThread;
    /**
     * Loop over all threads that will receive work except the final one.
     */
    for (currentThread = 0; currentThread < lastThread - 1; currentThread++) {
        ranges[currentThread] = new Range(lowerBounds[currentThread], lowerBounds[currentThread + 1] - 1);
    // logger.info(String.format("Range for thread %d %s.", currentThread, ranges[currentThread]));
    }
    /**
     * Final range for the last thread that will receive work.
     */
    ranges[lastThread - 1] = new Range(lowerBounds[lastThread - 1], lastAtom);
    /**
     * Left-over threads with null ranges.
     */
    for (int it = lastThread; it < nThreads; it++) {
        ranges[it] = null;
    }
}