com.meliorbis.numerics.generic.primitives.impl.DoubleNaryOpCallable Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of Numerics Show documentation

A library for working with large multi-dimensional arrays and the functions they represent

There is a newer version: 1.2

package com.meliorbis.numerics.generic.primitives.impl;

import com.meliorbis.numerics.generic.SettableIterator;
import com.meliorbis.numerics.generic.impl.NaryOpCallable;
import com.meliorbis.numerics.generic.primitives.DoubleNaryOp;
import com.meliorbis.numerics.generic.primitives.DoubleSettableIterator;

/**
 * Specialisation of NaryOpCallable for primitive double arrays
 */
public class DoubleNaryOpCallable extends NaryOpCallable
{
    private final double[] _currentVals;

    public DoubleNaryOpCallable(DoubleNaryOp op_, SettableIterator[] inputs_, SettableIterator result_,
                                boolean iterateResult_)
    {
        super(op_, inputs_, result_, iterateResult_);

        _currentVals = new double[inputs_.length];
    }

    @SuppressWarnings("unchecked")
	@Override
    protected void computeNextStep() throws E
    {
        for (int i = 0; i < _iterators.length; i++) {
            _currentVals[i] = ((DoubleSettableIterator)_iterators[i]).nextDouble();
        }

        // Only iterate the result if requested (because the result could be the same as one of the inputs and hence be
        // already iterated
        if(_iterateResult)
        {
            ((DoubleSettableIterator)_resultIter).nextDouble();
        }

        ((DoubleSettableIterator)_resultIter).set(((DoubleNaryOp)_op).perform(_currentVals));
    }
}