All Downloads are FREE. Search and download functionalities are using the official Maven repository.

nom.tam.util.ArrayFuncs Maven / Gradle / Ivy

package nom.tam.util;

/*
 * #%L
 * nom.tam FITS library
 * %%
 * Copyright (C) 2004 - 2024 nom-tam-fits
 * %%
 * This is free and unencumbered software released into the public domain.
 *
 * Anyone is free to copy, modify, publish, use, compile, sell, or
 * distribute this software, either in source code form or as a compiled
 * binary, for any purpose, commercial or non-commercial, and by any
 * means.
 *
 * In jurisdictions that recognize copyright laws, the author or authors
 * of this software dedicate any and all copyright interest in the
 * software to the public domain. We make this dedication for the benefit
 * of the public at large and to the detriment of our heirs and
 * successors. We intend this dedication to be an overt act of
 * relinquishment in perpetuity of all present and future rights to this
 * software under copyright law.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
 * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
 * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
 * IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 * #L%
 */

import java.lang.reflect.Array;
import java.util.Arrays;
import java.util.logging.Level;
import java.util.logging.Logger;

import nom.tam.util.array.MultiArrayCopier;
import nom.tam.util.type.ElementType;

/**
 * (for internal use) Varioys static functions for handling arrays. Generally these routines attempt to complete
 * without throwing errors by ignoring data they cannot understand.
 */
public final class ArrayFuncs {

    private static final Logger LOG = Logger.getLogger(ArrayFuncs.class.getName());

    private ArrayFuncs() {
    }

    /**
     * Retuens a copy of the input array with the order of elements reversed.
     *
     * @param  index the input array
     *
     * @return       a copy of the input array in reversed order
     */
    public static int[] getReversed(int... index) {
        int[] rev = new int[index.length];
        int nm1 = index.length - 1;
        for (int i = index.length; --i >= 0;) {
            rev[i] = index[nm1 - i];
        }
        return rev;
    }

    /**
     * Perform an array copy with an API similar to System.arraycopy(), specifying the number of values to jump to the
     * next read.
     *
     * @param src     The source array.
     * @param srcPos  Starting position in the source array.
     * @param dest    The destination array.
     * @param destPos Starting position in the destination data.
     * @param length  The number of array elements to be read.
     * @param step    The number of jumps to the next read.
     *
     * @since         1.18
     */
    public static void copy(Object src, int srcPos, Object dest, int destPos, int length, int step) {
        if (!dest.getClass().equals(src.getClass())) {
            throw new IllegalArgumentException("Mismatched types: src " + src.getClass() + ", dst " + dest.getClass());
        }

        if (src instanceof Object[]) {
            final Object[] from = (Object[]) src;
            final Object[] to = (Object[]) dest;
            int toIndex = 0;
            for (int index = srcPos; index < srcPos + length; index += step) {
                ArrayFuncs.copy(from[index], srcPos, to[toIndex++], destPos, length, step);
            }
        } else if (step == 1) {
            System.arraycopy(src, srcPos, dest, destPos, length);
        } else {
            final int srcLength = Array.getLength(src);
            final int loopLength = Math.min(srcLength, srcPos + length);
            for (int i = srcPos; i < loopLength; i += step) {
                Array.set(dest, destPos++, Array.get(src, i));
            }
        }
    }

    /**
     * @return   a description of an array (presumed rectangular).
     *
     * @param  o The array to be described.
     */
    public static String arrayDescription(Object o) {
        Class base = getBaseClass(o);
        if (base == Void.TYPE) {
            return "NULL";
        }
        return base.getSimpleName() + Arrays.toString(getDimensions(o));
    }

    /**
     * @deprecated   Use {@link FitsEncoder#computeSize(Object)} instead.
     *
     * @param      o the object
     *
     * @return       the number of bytes in the FITS binary representation of the object or 0 if the object has no FITS
     *                   representation. (Also elements not known to FITS will count as 0 sized).
     */
    @Deprecated
    public static long computeLSize(Object o) {
        return FitsEncoder.computeSize(o);
    }

    /**
     * @deprecated   Use {@link FitsEncoder#computeSize(Object)} instead.
     *
     * @param      o the object
     *
     * @return       the number of bytes in the FITS binary representation of the object or 0 if the object has no FITS
     *                   representation. (Also elements not known to FITS will count as 0 sized).
     */
    @Deprecated
    public static int computeSize(Object o) {
        return (int) computeLSize(o);
    }

    /**
     * Converts a numerical array to a specified element type. This method supports conversions only among the primitive
     * numeric types, and {@link ComplexValue} types (as of version 1.20). When converting primitive arrays to complex
     * values, the trailing dimension must be 2, corresponding to the real and imaginary components of the complex
     * values stored.
     * 
     * @param  array   a numerical array of one or more dimensions
     * @param  newType the desired output type. This should be one of the class descriptors for primitive numeric data,
     *                     e.g., double.class, or else a {@link ComplexValue} type (also supported as of
     *                     1.20).
     * 
     * @return         a new array with the requested element type, which may also be composed of {@link ComplexValue}
     *                     types as of version 1.20.
     * 
     * @see            #convertArray(Object, Class, boolean)
     */
    public static Object convertArray(Object array, Class newType) {
        if (newType.equals(getBaseClass(array))) {

            Object copy = Array.newInstance(newType, getDimensions(array));
            copyArray(array, copy);
            return copy;
        }
        return convertArray(array, newType, null);
    }

    /**
     * Converts a numerical array to a specified element type, returning the original if type conversion is not needed.
     * This method supports conversions only among the primitive numeric types originally. Support for
     * {@link ComplexValue} types was added as of version 1.20. When converting primitive arrays to complex values, the
     * trailing dimension must be 2, corresponding to the real and imaginary components of the complex values stored.
     *
     * @param  array   a numerical array of one or more dimensions
     * @param  newType the desired output type. This should be one of the class descriptors for primitive numeric data,
     *                     e.g., double.class r else a {@link ComplexValue} type (also supported as of
     *                     1.20).
     * @param  reuse   If the original (rather than a copy) should be returned when possible for the same type.
     * 
     * @return         a new array with the requested element type, or possibly the original array if it readily matches
     *                     the type and reuse is enabled.
     * 
     * @see            #convertArray(Object, Class)
     * @see            #convertArray(Object, Class, Quantizer)
     */
    public static Object convertArray(Object array, Class newType, boolean reuse) {
        if (reuse) {
            return convertArray(array, newType, null);
        }
        return convertArray(array, newType);
    }

    /**
     * Copy one array into another. This function copies the contents of one array into a previously allocated array.
     * The arrays must agree in type and size.
     *
     * @param      original                 The array to be copied.
     * @param      copy                     The array to be copied into. This array must already be fully allocated.
     *
     * @throws     IllegalArgumentException if the two arrays do not match in type or size.
     * 
     * @deprecated                          (for internal use)
     */
    @Deprecated
    public static void copyArray(Object original, Object copy) throws IllegalArgumentException {
        Class cl = original.getClass();
        if (!cl.isArray()) {
            throw new IllegalArgumentException("original is not an array");
        }

        if (!copy.getClass().equals(cl)) {
            throw new IllegalArgumentException("mismatch of types: " + cl.getName() + " vs " + copy.getClass().getName());
        }

        int length = Array.getLength(original);

        if (Array.getLength(copy) != length) {
            throw new IllegalArgumentException("mismatch of sizes: " + length + " vs " + Array.getLength(copy));
        }

        if (original instanceof Object[]) {
            Object[] from = (Object[]) original;
            Object[] to = (Object[]) copy;
            for (int index = 0; index < length; index++) {
                copyArray(from[index], to[index]);
            }
        } else {
            System.arraycopy(original, 0, copy, 0, length);
        }
    }

    /**
     * Copy an array into an array of a different type. The dimensions and dimensionalities of the two arrays should be
     * the same.
     *
     * @param array The original array.
     * @param mimic The array mimicking the original.
     */
    public static void copyInto(Object array, Object mimic) {
        MultiArrayCopier.copyInto(array, mimic);
    }

    /**
     * Curl an input array up into a multi-dimensional array.
     *
     * @param  input                 The one dimensional array to be curled.
     * @param  dimens                The desired dimensions
     *
     * @return                       The curled array.
     * 
     * @throws IllegalStateException if the size of the input does not match the specified dimensions.
     */
    public static Object curl(Object input, int... dimens) throws IllegalStateException {
        if (input == null) {
            return null;
        }
        if (!input.getClass().isArray()) {
            throw new IllegalArgumentException("Attempt to curl a non-array: " + input.getClass());
        }
        if (input.getClass().getComponentType().isArray()) {
            throw new IllegalArgumentException("Attempt to curl non-1D array: " + input.getClass());
        }
        int size = Array.getLength(input);
        int test = 1;
        for (int dimen : dimens) {
            test *= dimen;
        }
        if (test != size) {
            throw new IllegalStateException("Curled array does not fit desired dimensions: " + size + ", expected " + test
                    + " (" + getBaseClass(input) + ")");
        }
        Object newArray = ArrayFuncs.newInstance(getBaseClass(input), dimens);
        MultiArrayCopier.copyInto(input, newArray);
        return newArray;

    }

    /**
     * Returns a deep clone of an array (in one or more dimensions) or a standard clone of a scalar. The object may
     * comprise arrays of any primitive type or any Object type which implements Cloneable. However, if the Object is
     * some kind of collection, such as a {@link java.util.List}, then only a shallow copy of that object is made. I.e.,
     * deep refers only to arrays.
     *
     * @return   a new object, with a copy of the original.
     * 
     * @param  o The object (usually an array) to be copied.
     */
    public static Object deepClone(Object o) {
        if (o == null) {
            return null;
        }

        if (!o.getClass().isArray()) {
            return genericClone(o);
        }

        if (o instanceof Object[]) {
            Object[] array = (Object[]) o;
            Object[] copy = (Object[]) ArrayFuncs.newInstance(array.getClass().getComponentType(), array.length);
            // Now fill in the next level down by recursion.
            for (int i = 0; i < array.length; i++) {
                copy[i] = deepClone(array[i]);
            }
            return copy;
        }

        // Must be primitive array...
        // Check if this is a 1D primitive array.
        int length = Array.getLength(o);
        Object copy = Array.newInstance(o.getClass().getComponentType(), length);
        System.arraycopy(o, 0, copy, 0, length);
        return copy;
    }

    /**
     * Given an array of arbitrary dimensionality .
     *
     * @return       the array flattened into a single dimension.
     *
     * @param  input The input array.
     */
    public static Object flatten(Object input) {
        if (input == null) {
            return null;
        }

        if (!input.getClass().isArray()) {
            return input;
        }

        int[] dimens = getDimensions(input);
        if (dimens.length <= 1) {
            return input;
        }
        int size = 1;
        for (int dimen : dimens) {
            size *= dimen;
        }
        Object flat = ArrayFuncs.newInstance(getBaseClass(input), size);
        MultiArrayCopier.copyInto(input, flat);
        return flat;
    }

    /**
     * Clone an Object if possible. This method returns an Object which is a clone of the input object. It checks if the
     * method implements the Cloneable interface and then uses reflection to invoke the clone method. This can't be done
     * directly since as far as the compiler is concerned the clone method for Object is protected and someone could
     * implement Cloneable but leave the clone method protected. The cloning can fail in a variety of ways which are
     * trapped so that it returns null instead. This method will generally create a shallow clone. If you wish a deep
     * copy of an array the method deepClone should be used.
     *
     * @param  o The object to be cloned.
     *
     * @return   the clone
     */
    public static Object genericClone(Object o) {
        if (o.getClass().isArray()) {
            return deepClone(o);
        }
        if (!(o instanceof Cloneable)) {
            LOG.log(Level.SEVERE, "generic clone called on a non clonable type");
            return null;
        }
        try {
            return o.getClass().getMethod("clone").invoke(o);
        } catch (Exception e) {
            LOG.log(Level.WARNING, "Implements cloneable, but does not apparently make clone public.", e);
            return null;
        }
    }

    /**
     * This routine returns the base array of a multi-dimensional array. I.e., a one-d array of whatever the array is
     * composed of. Note that arrays are not guaranteed to be rectangular, so this returns o[0][0]....
     *
     * @param  o the multi-dimensional array
     *
     * @return   base array of a multi-dimensional array.
     */
    public static Object getBaseArray(Object o) {
        if (o instanceof Object[]) {
            return getBaseArray(((Object[]) o)[0]);
        }
        return o;
    }

    /**
     * This routine returns the base class of an object. This is just the class of the object for non-arrays.
     *
     * @param  o array to get the base class from
     *
     * @return   the base class of an array
     */
    public static Class getBaseClass(Object o) {
        if (o == null) {
            return Void.TYPE;
        }
        Class clazz = o.getClass();
        while (clazz.isArray()) {
            clazz = clazz.getComponentType();
        }
        return clazz;
    }

    /**
     * This routine returns the size of the base element of an array.
     *
     * @param  o The array object whose base length is desired.
     *
     * @return   the size of the object in bytes, 0 if null, or -1 if not a primitive array.
     */
    public static int getBaseLength(Object o) {
        if (o == null) {
            return 0;
        }
        ElementType type = ElementType.forClass(getBaseClass(o));
        return type.size();
    }

    /**
     * Find the dimensions of an object. This method returns an integer array with the dimensions of the object o which
     * should usually be an array. It returns an array of dimension 0 for scalar objects and it returns -1 for dimension
     * which have not been allocated, e.g., int[][][] x = new
     * int[100][][]; should return [100,-1,-1].
     *
     * @param  o The object to get the dimensions of.
     *
     * @return   the dimensions of an object
     */
    public static int[] getDimensions(Object o) {
        if (o == null) {
            return null;
        }
        Object object = o;
        Class clazz = o.getClass();
        int ndim = 0;
        while (clazz.isArray()) {
            clazz = clazz.getComponentType();
            ndim++;
        }
        clazz = o.getClass();
        int[] dimens = new int[ndim];
        ndim = 0;
        while (clazz.isArray()) {
            dimens[ndim] = -1;
            if (object != null) {
                int length = Array.getLength(object);
                if (length > 0) {
                    dimens[ndim] = length;
                    object = Array.get(object, 0);
                } else {
                    dimens[ndim] = 0;
                    object = null;
                }
            }
            clazz = clazz.getComponentType();
            ndim++;
        }
        return dimens;
    }

    /**
     * Create an array of a type given by new type with the dimensionality given in array.
     *
     * @return         the new array with same dimensions
     *
     * @param  array   A possibly multidimensional array to be converted.
     * @param  newType The desired output type. This should be one of the class descriptors for primitive numeric data,
     *                     e.g., double.type.
     */
    public static Object mimicArray(Object array, Class newType) {
        int dims = 0;
        Class arrayClass = array.getClass();
        while (arrayClass.isArray()) {
            arrayClass = arrayClass.getComponentType();
            dims++;
        }

        if (dims <= 1) {
            return ArrayFuncs.newInstance(newType, Array.getLength(array));
        }

        Object[] xarray = (Object[]) array;
        int[] dimens = new int[dims];
        dimens[0] = xarray.length; // Leave other dimensions at 0.

        Object mimic = ArrayFuncs.newInstance(newType, dimens);

        for (int i = 0; i < xarray.length; i++) {
            Object temp = mimicArray(xarray[i], newType);
            ((Object[]) mimic)[i] = temp;
        }

        return mimic;
    }

    /**
     * Convenience method to check a generic Array object.
     *
     * @param  o The Array to check.
     *
     * @return   True if it's empty, False otherwise.
     *
     * @since    1.18
     */
    public static boolean isEmpty(final Object o) {
        return (o == null || Array.getLength(o) == 0);
    }

    /**
     * @return       Count the number of elements in an array.
     *
     * @param      o the array to count the elements
     *
     * @deprecated   May silently underestimate size if number is > 2 G.
     */
    @Deprecated
    public static int nElements(Object o) {
        return (int) nLElements(o);
    }

    /**
     * Allocate an array dynamically. The Array.newInstance method does not throw an error and silently returns a
     * null.throws an OutOfMemoryError if insufficient space is available.
     *
     * @param  cl   The class of the array.
     * @param  dims The dimensions of the array.
     *
     * @return      The allocated array.
     */
    public static Object newInstance(Class cl, int... dims) {
        if (dims.length == 0) {
            // Treat a scalar as a 1-d array of length 1
            dims = new int[] {1};
        }
        return Array.newInstance(cl, dims);
    }

    /**
     * @return       Count the number of elements in an array.
     *
     * @param      o the array to count elements in
     *
     * @deprecated   Use the more aptly named {@link #countElements(Object)} instead.
     */
    public static long nLElements(Object o) {
        return countElements(o);
    }

    /**
     * @return   Count the number of elements in an array.
     *
     * @param  o the array to count elements in
     * 
     * @since    1.18
     */
    public static long countElements(Object o) {
        if (o == null) {
            return 0;
        }

        if (o instanceof Object[]) {
            long count = 0;
            for (Object e : (Object[]) o) {
                count += countElements(e);
            }
            return count;
        }

        if (o.getClass().isArray()) {
            return Array.getLength(o);
        }

        return 1;
    }

    /**
     * Reverse an integer array. This can be especially useful when dealing with an array of indices in FITS order that
     * you wish to have in Java order.
     *
     * @return         the reversed array.
     *
     * @param  indices the array to reverse
     */
    public static int[] reverseIndices(int... indices) {
        int[] result = new int[indices.length];
        int len = indices.length;
        for (int i = 0; i < indices.length; i++) {
            result[len - i - 1] = indices[i];
        }
        return result;
    }

    /**
     * Checks that an array has a regular structure, with a consistent shape and element types, and returns the regular
     * array size or else throws an exeption. Optionally, it will also throw an exception if any or all all elements are
     * null.
     * 
     * @param  o                        An array object
     * @param  allowNulls               If we should tolerate null entries.
     * 
     * @return                          the regular shape of the array with sizes along each array dimension.
     * 
     * @throws NullPointerException     if the argument is null.
     * @throws IllegalArgumentException if the array contains mismatched elements in size, or contains null
     *                                      values.
     * @throws ClassCastException       if the array contain a heterogeneous collection of different element types.
     * 
     * @since                           1.18
     */
    public static int[] checkRegularArray(Object o, boolean allowNulls)
            throws NullPointerException, IllegalArgumentException, ClassCastException {
        if (!o.getClass().isArray()) {
            throw new IllegalArgumentException("Not an array: " + o.getClass());
        }

        if (o.getClass().getComponentType().isPrimitive()) {
            return new int[] {Array.getLength(o)};
        }

        int[] dim = getDimensions(o);
        if (dim[0] == 0) {
            return dim;
        }

        Class type = null;
        Object[] array = (Object[]) o;

        for (int i = 0; i < dim[0]; i++) {
            Object e = array[i];

            if (e == null) {
                if (allowNulls) {
                    continue;
                }
                throw new IllegalArgumentException("Entry at index " + i + " is null");
            }

            if (type == null) {
                type = e.getClass();
            } else if (!e.getClass().equals(type)) {
                throw new ClassCastException(
                        "Mismatched component type at index " + i + ": " + e.getClass() + ", expected " + type);
            }

            if (e.getClass().isArray()) {
                int[] sub = checkRegularArray(e, allowNulls);

                if (sub.length + 1 != dim.length) {
                    throw new IllegalArgumentException("Mismatched component dimension at index " + i + ": " + sub.length
                            + ", expected " + (dim.length - 1));
                }

                if (sub[0] != dim[1]) {
                    throw new IllegalArgumentException(
                            "Mismatched component size at index " + i + ": " + sub[0] + ", expected " + dim[0]);
                }
            }
        }

        return dim;
    }

    /**
     * Converts complex value(s) count to float[2] or double[2] or arrays thereof, which
     * maintain the shape of the original input array (if applicable).
     * 
     * @param  o                        one of more complex values
     * @param  decimalType              float.class or double.class (all other values default
     *                                      to as if double.class was used.
     * 
     * @return                          an array of float[2] or double[2], or arrays thereof.
     * 
     * @throws IllegalArgumentException if the argument is not suitable for conversion to complex values.
     * 
     * @see                             #decimalsToComplex(Object)
     * 
     * @since                           1.18
     */
    public static Object complexToDecimals(Object o, Class decimalType) {

        if (o instanceof ComplexValue) {
            ComplexValue z = (ComplexValue) o;
            if (float.class.equals(decimalType)) {
                return new float[] {(float) z.re(), (float) z.im()};
            }
            return new double[] {z.re(), z.im()};
        }

        if (o instanceof ComplexValue[]) {
            ComplexValue[] z = (ComplexValue[]) o;

            if (float.class.equals(decimalType)) {
                float[][] f = new float[z.length][];
                for (int i = 0; i < z.length; i++) {
                    f[i] = (float[]) complexToDecimals(z[i], decimalType);
                }
                return f;
            }

            double[][] d = new double[z.length][];
            for (int i = 0; i < z.length; i++) {
                d[i] = (double[]) complexToDecimals(z[i], decimalType);
            }
            return d;
        }

        if (o instanceof Object[]) {
            Object[] array = (Object[]) o;
            Object[] z = null;

            for (int i = 0; i < array.length; i++) {
                Object e = complexToDecimals(array[i], decimalType);
                if (z == null) {
                    z = (Object[]) Array.newInstance(e.getClass(), array.length);
                }
                z[i] = e;
            }

            return z;
        }

        throw new IllegalArgumentException("Cannot convert to complex values: " + o.getClass().getName());
    }

    /**
     * Converts real-valued arrays of even element count to a {@link ComplexValue} or arrays thereof. The size and shape
     * is otherwise maintained, apart from coalescing pairs of real values into ComplexValue objects.
     * 
     * @param  array                    an array of float or double elements containing an
     *                                      even number of elements at the last dimension
     * 
     * @return                          one of more complex values
     * 
     * @throws IllegalArgumentException if the argument is not suitable for conversion to complex values.
     * 
     * @see                             #decimalsToComplex(Object, Object)
     * @see                             #complexToDecimals(Object, Class)
     * 
     * @since                           1.18
     */
    public static Object decimalsToComplex(Object array) throws IllegalArgumentException {
        if (array instanceof float[]) {
            float[] f = (float[]) array;
            if (f.length == 2) {
                return new ComplexValue.Float(f[0], f[1]);
            }
            if (f.length % 2 == 1) {
                throw new IllegalArgumentException("Odd number floats for complex conversion: " + f.length);
            }
            ComplexValue[] z = new ComplexValue[f.length >>> 1];
            for (int i = 0; i < f.length; i += 2) {
                z[i >> 1] = new ComplexValue(f[i], f[i + 1]);
            }
            return z;
        }

        if (array instanceof double[]) {
            double[] d = (double[]) array;
            if (d.length == 2) {
                return new ComplexValue(d[0], d[1]);
            }
            if (d.length % 2 == 1) {
                throw new IllegalArgumentException("Odd number floats for complex conversion: " + d.length);
            }
            ComplexValue[] z = new ComplexValue[d.length >>> 1];
            for (int i = 0; i < d.length; i += 2) {
                z[i >> 1] = new ComplexValue(d[i], d[i + 1]);
            }
            return z;
        }

        if (array instanceof Object[]) {
            Object[] o = (Object[]) array;
            Object[] z = null;

            for (int i = 0; i < o.length; i++) {
                Object e = decimalsToComplex(o[i]);
                if (z == null) {
                    z = (Object[]) Array.newInstance(e.getClass(), o.length);
                }
                z[i] = e;
            }

            return z;
        }

        throw new IllegalArgumentException("Cannot convert to complex values: " + array.getClass().getName());
    }

    /**
     * Converts separate but matched real valued arrays, containing the real an imaginary parts respectively, to
     * {@link ComplexValue} or arrays thereof. The size and shape of the matching input arrays is otherwise maintained,
     * apart from coalescing pairs of real values into ComplexValue objects.
     * 
     * @param  re                       an array of float or double elements containing the
     *                                      real parts of the complex values.
     * @param  im                       a matching array of the same type and shape as the real parts which contains the
     *                                      imaginary parts of the complex values.
     * 
     * @return                          one of more complex values
     * 
     * @throws IllegalArgumentException if the argument is not suitable for conversion to complex values.
     * 
     * @see                             #decimalsToComplex(Object)
     * 
     * @since                           1.20
     */
    public static Object decimalsToComplex(Object re, Object im) {
        if (!re.getClass().equals(im.getClass())) {
            throw new IllegalArgumentException(
                    "Mismatched components: " + re.getClass().getName() + " vs " + re.getClass().getName());
        }

        if (re instanceof float[]) {
            float[] fre = (float[]) re;
            float[] fim = (float[]) im;
            ComplexValue.Float[] z = new ComplexValue.Float[fre.length];
            for (int i = fre.length; --i >= 0;) {
                z[i] = new ComplexValue.Float(fre[i], fim[i]);
            }
            return z;
        }
        if (re instanceof double[]) {
            double[] dre = (double[]) re;
            double[] dim = (double[]) im;
            ComplexValue[] z = new ComplexValue[dre.length];
            for (int i = dre.length; --i >= 0;) {
                z[i] = new ComplexValue(dre[i], dim[i]);
            }
            return z;
        }
        if (re instanceof Object[]) {
            Object[] ore = (Object[]) re;
            Object[] oim = (Object[]) im;
            Object[] z = null;
            for (int i = ore.length; --i >= 0;) {
                Object e = decimalsToComplex(ore[i], oim[i]);
                if (z == null) {
                    z = (Object[]) Array.newInstance(e.getClass(), ore.length);
                }
                z[i] = e;
            }
            return z;
        }
        throw new IllegalArgumentException("Cannot convert components to complex: " + re.getClass().getName());
    }

    private static void decimalToInteger(Object from, Object to, Quantizer q) {
        if (from instanceof Object[]) {
            Object[] a = (Object[]) from;
            Object[] b = (Object[]) to;
            for (int i = 0; i < a.length; i++) {
                decimalToInteger(a[i], b[i], q);
            }
        } else {
            for (int i = Array.getLength(from); --i >= 0;) {
                long l = q.toLong(from instanceof double[] ? ((double[]) from)[i] : ((float[]) from)[i]);

                if (to instanceof byte[]) {
                    ((byte[]) to)[i] = (byte) l;
                } else if (to instanceof short[]) {
                    ((short[]) to)[i] = (short) l;
                } else if (to instanceof int[]) {
                    ((int[]) to)[i] = (int) l;
                } else {
                    ((long[]) to)[i] = l;
                }
            }
        }
    }

    private static void integerToDecimal(Object from, Object to, Quantizer q) {
        if (from instanceof Object[]) {
            Object[] a = (Object[]) from;
            Object[] b = (Object[]) to;
            for (int i = 0; i < a.length; i++) {
                integerToDecimal(a[i], b[i], q);
            }
        } else {
            for (int i = Array.getLength(from); --i >= 0;) {
                double d = q.toDouble(Array.getLong(from, i));

                if (to instanceof float[]) {
                    ((float[]) to)[i] = (float) d;
                } else {
                    ((double[]) to)[i] = d;
                }
            }
        }
    }

    /**
     * Converts a numerical array to a specified element type, returning the original if type conversion is not needed.
     * If the conversion is from decimal to integer type, or vice-versa, an optional quantization may be supplied to to
     * perform the integer-decimal conversion of the elements. This method supports conversions only among the primitive
     * numeric types and also {@link ComplexValue} type.
     *
     * @param  array                    a numerical array of one or more dimensions
     * @param  newType                  the desired output type. This should be one of the class descriptors for
     *                                      primitive numeric data, e.g., double.class, or else a
     *                                      {@link ComplexValue} or {@link ComplexValue.Float}.
     * @param  quant                    optional qunatizer for integer-decimal conversion, or null to use
     *                                      simply rounding.
     * 
     * @return                          a new array with the requested element type, or possibly the original array if
     *                                      it readily matches the type and reuse is enabled.
     * 
     * @throws IllegalArgumentException if the input is not an array, or its elements are not a supported type or if the
     *                                      new type is not supported.
     * 
     * @see                             #convertArray(Object, Class)
     * @see                             #convertArray(Object, Class, Quantizer)
     * 
     * @since                           1.20
     */
    public static Object convertArray(Object array, Class newType, Quantizer quant) throws IllegalArgumentException {

        if (!array.getClass().isArray()) {
            throw new IllegalArgumentException("Not an array: " + array.getClass().getName());
        }

        Class fromType = getBaseClass(array);

        if (newType.isAssignableFrom(fromType)) {
            return array;
        }

        boolean toComplex = ComplexValue.class.isAssignableFrom(newType);
        if (toComplex) {
            newType = (ComplexValue.Float.class.isAssignableFrom(newType) ? float.class : double.class);
        }

        if (!newType.isPrimitive() || newType == boolean.class || newType == char.class) {
            throw new IllegalArgumentException("Not a supported numerical type: " + newType.getName());
        }

        boolean toInteger = (newType != float.class && newType != double.class);

        if (ComplexValue.class.isAssignableFrom(fromType)) {
            fromType = (ComplexValue.Float.class.isAssignableFrom(fromType) ? float.class : double.class);
            array = complexToDecimals(array, fromType);
        }

        boolean fromInteger = (fromType != float.class && fromType != double.class);

        Object t = Array.newInstance(newType, getDimensions(array));

        if (quant != null) {
            if (fromInteger && !toInteger) {
                integerToDecimal(array, t, quant);
            } else if (!fromInteger && toInteger) {
                decimalToInteger(array, t, quant);
            } else {
                MultiArrayCopier.copyInto(array, t);
            }
        } else {
            MultiArrayCopier.copyInto(array, t);
        }

        if (toComplex) {
            t = decimalsToComplex(t);
        }

        return t;
    }

    /**
     * Obtains a sparse sampling of from an array of one or more dimensions.
     * 
     * @param  orig                      the original array
     * @param  step                      the sampling step size along all dimensions for a subsampled slice. A negative
     *                                       value indicates that the sampling should proceed in the reverse direction
     *                                       along every axis.
     * 
     * @return                           the requested sampling from the original. The returned array may share data
     *                                       with the original, and so modifications to either may affect the other. The
     *                                       orginal object is returned if it is not an array.
     * 
     * @throws IndexOutOfBoundsException if any of the indices for the requested slice are out of bounds for the
     *                                       original. That is if the original does not fully contain the requested
     *                                       slice. Or, if the from and size arguments have differing lengths.
     * 
     * @since                            1.20
     * 
     * @author                           Attila Kovacs
     * 
     * @see                              #sample(Object, int[])
     * @see                              #sample(Object, int[], int[], int[])
     */
    public static Object sample(Object orig, int step) throws IndexOutOfBoundsException {
        int[] n = getDimensions(orig);
        Arrays.fill(n, step);
        return sample(orig, n);
    }

    /**
     * Obtains a sparse sampling of from an array of one or more dimensions.
     * 
     * @param  orig                      the original array
     * @param  step                      the sampling step size along each dimension for a subsampled slice. Negative
     *                                       values indicate that the sampling should proceed in the reverse direction
     *                                       along the given axis.
     * 
     * @return                           the requested sampling from the original. The returned array may share data
     *                                       with the original, and so modifications to either may affect the other. The
     *                                       orginal object is returned if it is not an array.
     * 
     * @throws IndexOutOfBoundsException if any of the indices for the requested slice are out of bounds for the
     *                                       original. That is if the original does not fully contain the requested
     *                                       slice. Or, if the from and size arguments have differing lengths.
     * 
     * @since                            1.20
     * 
     * @author                           Attila Kovacs
     * 
     * @see                              #sample(Object, int)
     * @see                              #sample(Object, int[], int[], int[])
     */
    public static Object sample(Object orig, int[] step) throws IndexOutOfBoundsException {
        return sample(orig, null, null, step);
    }

    /**
     * Obtains a slice (subarray) from an array of one or more dimensions.
     * 
     * @param  orig                      the original array
     * @param  from                      the starting indices for the slice in the original array. It should have at
     *                                       most as many elements as there are array dimensions, but it can also have
     *                                       fewer.
     * 
     * @return                           the requested slice from the original. The returned array may share data with
     *                                       the original, and so modifications to either may affect the other. The
     *                                       orginal object is returned if it is not an array.
     * 
     * @throws IndexOutOfBoundsException if any of the indices for the requested slice are out of bounds for the
     *                                       original. That is if the original does not fully contain the requested
     *                                       slice. Or, if the from and size arguments have differing lengths.
     * 
     * @since                            1.20
     * 
     * @author                           Attila Kovacs
     * 
     * @see                              #slice(Object, int[], int[])
     * @see                              #sample(Object, int[], int[], int[])
     */
    public static Object slice(Object orig, int[] from) throws IndexOutOfBoundsException {
        return slice(orig, from, null);
    }

    /**
     * Obtains a slice (subarray) from an array of one or more dimensions.
     * 
     * @param  orig                      the original array
     * @param  from                      the starting indices for the slice in the original array. It should have at
     *                                       most as many elements as there are array dimensions, but it can also have
     *                                       fewer.
     * @param  size                      the size of the slice. Negative values can indicate moving backwards in the
     *                                       original array (but forward in the slice -- resulting in a flipped axis). A
     *                                       null size argument can be used to sample the full original.
     *                                       The slice will end at index from[k] + size[k] in dimension
     *                                       k in the original (not including the ending index). It should
     *                                       have the same number of elements as the from argument.
     * 
     * @return                           the requested slice from the original. The returned array may share data with
     *                                       the original, and so modifications to either may affect the other. The
     *                                       orginal object is returned if it is not an array.
     * 
     * @throws IndexOutOfBoundsException if any of the indices for the requested slice are out of bounds for the
     *                                       original. That is if the original does not fully contain the requested
     *                                       slice. Or, if the from and size arguments have differing lengths.
     * 
     * @since                            1.20
     * 
     * @author                           Attila Kovacs
     * 
     * @see                              #slice(Object, int[])
     * @see                              #sample(Object, int[], int[], int[])
     */
    public static Object slice(Object orig, int[] from, int[] size) throws IndexOutOfBoundsException {
        int[] step = null;

        if (size != null) {
            step = new int[size.length];
            for (int i = 0; i < size.length; i++) {
                step[i] = size[i] < 0 ? -1 : 1;
            }
        }

        return sample(orig, from, size, step, 0);
    }

    /**
     * Obtains a sparse sampling from an array of one or more dimensions.
     * 
     * @param  orig                      the original array
     * @param  from                      the starting indices in the original array at which to start sampling. It
     *                                       should have at most as many elements as there are array dimensions, but it
     *                                       can also have fewer. A null argument can be used to sample
     *                                       from the start or end of the array (depending on the direction).
     * @param  size                      the size of the sampled area in the original along each dimension. The
     *                                       signature of the values is irrelevant as the direction of sampling is
     *                                       determined by the step argument. Zero entries can be used to indicate that
     *                                       the full array should be sampled along the given dimension, while a
     *                                       null argument will sample the full array in all dimensions.
     * @param  step                      the sampling step size along each dimension for a subsampled slice. Negative
     *                                       values indicate sampling the original in the reverse direction along the
     *                                       given dimension. 0 values are are automatically bumped to 1 (full
     *                                       sampling), and a null argument is understood to mean full
     *                                       sampling along all dimensions.
     * 
     * @return                           the requested sampling from the original. The returned array may share data
     *                                       with the original, and so modifications to either may affect the other. The
     *                                       orginal object is returned if it is not an array.
     * 
     * @throws IndexOutOfBoundsException if any of the indices for the requested slice are out of bounds for the
     *                                       original. That is if the original does not fully contain the requested
     *                                       slice. Or, if the from and size arguments have differing lengths.
     * 
     * @since                            1.20
     * 
     * @author                           Attila Kovacs
     * 
     * @see                              #sample(Object, int)
     * @see                              #sample(Object, int[])
     * @see                              #slice(Object, int[], int[])
     */
    public static Object sample(Object orig, int[] from, int[] size, int[] step) throws IndexOutOfBoundsException {
        return sample(orig, from, size, step, 0);
    }

    private static Object sample(Object orig, int[] from, int[] size, int[] step, int idx)
            throws IndexOutOfBoundsException {

        // If leaf, return it as is...
        if (!orig.getClass().isArray() || (from != null && idx == from.length)) {
            return orig;
        }

        int l = Array.getLength(orig);
        int ndim = from == null ? getDimensions(orig).length : from.length;

        // Check if reverse sampling
        boolean isReversed = (step != null && step[idx] < 0);

        int ifrom = from == null ? (isReversed ? l - 1 : 0) : from[idx];

        int isize = size == null ? 0 : Math.abs(size[idx]);
        if (isize == 0) {
            isize = l - ifrom;
        }

        int ito = ifrom + (isReversed ? -isize : isize);
        if (ifrom < 0 || ito < -1 || ifrom >= l || ito > l) {
            throw new IndexOutOfBoundsException("Sampled bounds are out of range for original array");
        }

        int istep = step == null ? 1 : step[idx];
        if (istep == 0) {
            istep = 1;
        }

        int astep = Math.abs(istep);
        int n = Math.abs((isize + astep - 1) / astep);

        Object slice = Array.newInstance(orig.getClass().getComponentType(), n);

        if (!isReversed && ndim == 1 && istep == 1) {
            // Special case for fast in-order slicing along last dim...
            System.arraycopy(orig, ifrom, slice, 0, isize);
        } else {
            // Generic sampling with the parameters...
            for (int i = 0; i < n; i++) {
                Object efrom = Array.get(orig, ifrom + i * istep);
                Array.set(slice, i, sample(efrom, from, size, step, idx + 1));
            }
        }

        return slice;
    }

}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy