FloatArrayWrapper.java

/**
 * Copyright (c) 2012 Michael Kutschke.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    Michael Kutschke - initial API and implementation.
 */
package org.eclipse.recommenders.jayes.factor.arraywrapper;

import java.util.Arrays;
import java.util.Iterator;
import java.util.NoSuchElementException;

import org.eclipse.recommenders.internal.jayes.util.ArrayUtils;

public class FloatArrayWrapper implements IArrayWrapper {

    private float[] array;

    public FloatArrayWrapper(float... array) {
        this.array = array;
    }

    @Override
    public void setArray(double... array) {
        setArray(ArrayUtils.toFloatArray(array));
    }

    @Override
    public void setArray(float... array) {
        this.array = array;
    }

    @Override
    public double[] toDoubleArray() {
        return ArrayUtils.toDoubleArray(array);
    }

    @Override
    public float[] toFloatArray() {
        return array;
    }

    @Override
    public void set(int index, double d) {
        array[index] = (float) d;
    }

    @Override
    public void set(int index, float d) {
        array[index] = d;
    }

    @Override
    public void mulAssign(int index, double d) {
        array[index] *= (float) d;
    }

    @Override
    public void mulAssign(int index, float d) {
        array[index] *= d;
    }

    @Override
    public void mulAssign(int index, IArrayWrapper arg, int argIndex) {
        array[index] *= arg.getFloat(argIndex);
    }

    @Override
    public void addAssign(int index, double d) {
        array[index] += (float) d;
    }

    @Override
    public void addAssign(int index, float d) {
        array[index] += d;
    }

    @Override
    public void addAssign(int index, IArrayWrapper arg, int argIndex) {
        array[index] += arg.getFloat(argIndex);
    }

    @Override
    public double getDouble(int index) {
        return array[index];
    }

    @Override
    public float getFloat(int index) {
        return array[index];
    }

    @Override
    public int length() {
        return array.length;
    }

    @Override
    public void copy(double... array) {
        setArray(array);
    }

    @Override
    public void copy(float... array) {
        this.array = array.clone();
    }

    @Override
    public void copy(IArrayWrapper array) {
        copy(array.toFloatArray());
    }

    @Override
    public void fill(double d) {
        Arrays.fill(array, (float) d);
    }

    @Override
    public void fill(float d) {
        Arrays.fill(array, d);
    }

    @Override
    public void arrayCopy(IArrayWrapper src, int srcOffset, int destOffset, int length) {
        System.arraycopy(src.toFloatArray(), srcOffset, array, destOffset, length);
    }

    @Override
    public void newArray(int capacity) {
        array = new float[capacity];
    }

    @Override
    public FloatArrayWrapper clone() {
        FloatArrayWrapper clone;
        try {
            clone = (FloatArrayWrapper) super.clone();
            clone.array = array.clone();
            return clone;
        } catch (CloneNotSupportedException e) {
            throw new AssertionError(e);
        }
    }

    @Override
    public int sizeOfElement() {
        return 4;
    }

    @Override
    public Iterator<Number> iterator() {
        return new Iterator<Number>() {

            private int index = 0;

            @Override
            public boolean hasNext() {
                return index < array.length;
            }

            @Override
            public Number next() {
                if (index >= array.length) {
                    throw new NoSuchElementException();
                }

                return array[index++];
            }

            @Override
            public void remove() {
                throw new UnsupportedOperationException();
            }
        };
    }
}