org.ejml.ops.SortCoupledArray_F32 Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of ejml-core Show documentation
A fast and easy to use dense and sparse matrix linear algebra library written in Java.
The newest version!
/*
 * Copyright (c) 2020, Peter Abeles. All Rights Reserved.
 *
 * This file is part of Efficient Java Matrix Library (EJML).
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.ejml.ops;

import javax.annotation.Generated;
/**
 * @author Peter Abeles
 */
@Generated("org.ejml.ops.SortCoupledArray_F64")
public class SortCoupledArray_F32 {

    int[] tmp = new int[0];

    int[] copyA = new int[0];
    float[] copyB = new float[0];

    QuickSort_S32 quicksort = new QuickSort_S32();

    public void quick( int[] segments, int length, int[] valuesA, float[] valuesB ) {
        for (int i = 1; i < length; i++) {
            int x0 = segments[i - 1];
            int x1 = segments[i];

            quick(x0, x1 - x0, valuesA, valuesB);
        }
    }

    private void quick( int offset, int length, int[] valuesA, float[] valuesB ) {

        if (length <= 1)
            return;

        if (tmp.length < length) {
            int l = length*2 + 1;
            tmp = new int[l];
            copyA = new int[l];
            copyB = new float[l];
        }

        System.arraycopy(valuesA, offset, copyA, 0, length);
        System.arraycopy(valuesB, offset, copyB, 0, length);

        if (length > 50)
            quicksort.sort(copyA, length, tmp);
        else
            shellSort(copyA, 0, length, tmp);

        for (int i = 0; i < length; i++) {
            valuesA[offset + i] = copyA[tmp[i]];
            valuesB[offset + i] = copyB[tmp[i]];
        }
    }

    public static void shellSort( int[] data, int offset, int length, int[] indexes ) {
        for (int i = 0; i < length; i++) {
            indexes[i] = offset + i;
        }

        int i, j;
        int inc = 1;
        int v;

        do {
            inc *= 3;
            inc++;
        } while (inc <= length);

        do {
            inc /= 3;

            for (i = inc; i < length; i++) {
                v = data[indexes[i]];
                int idx_i = indexes[i];
                j = i;
                while (data[indexes[j - inc]] > v) {
                    indexes[j] = indexes[j - inc];
                    j -= inc;
                    if (j < inc) break;
                }
                indexes[j] = idx_i;
            }
        } while (inc > 1);
    }
}