|
Packit |
67cb25 |
/*
|
|
Packit |
67cb25 |
* Implement Heap sort -- direct and indirect sorting
|
|
Packit |
67cb25 |
* Based on descriptions in Sedgewick "Algorithms in C"
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* Copyright (C) 1999 Thomas Walter
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* 18 February 2000: Modified for GSL by Brian Gough
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* This is free software; you can redistribute it and/or modify it
|
|
Packit |
67cb25 |
* under the terms of the GNU General Public License as published by the
|
|
Packit |
67cb25 |
* Free Software Foundation; either version 3, or (at your option) any
|
|
Packit |
67cb25 |
* later version.
|
|
Packit |
67cb25 |
*
|
|
Packit |
67cb25 |
* This source is distributed in the hope that it will be useful, but WITHOUT
|
|
Packit |
67cb25 |
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
Packit |
67cb25 |
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
Packit |
67cb25 |
* for more details.
|
|
Packit |
67cb25 |
*/
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static inline void FUNCTION (my, downheap) (BASE * data, const size_t stride, const size_t N, size_t k);
|
|
Packit |
67cb25 |
static inline void FUNCTION (my, downheap2) (BASE * data1, const size_t stride1, BASE * data2, const size_t stride2, const size_t N, size_t k);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static inline void
|
|
Packit |
67cb25 |
FUNCTION (my, downheap) (BASE * data, const size_t stride, const size_t N, size_t k)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
BASE v = data[k * stride];
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
while (k <= N / 2)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
size_t j = 2 * k;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (j < N && data[j * stride] < data[(j + 1) * stride])
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
j++;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (!(v < data[j * stride])) /* avoid infinite loop if nan */
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
break;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
data[k * stride] = data[j * stride];
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
k = j;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
data[k * stride] = v;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
static inline void
|
|
Packit |
67cb25 |
FUNCTION (my, downheap2) (BASE * data1, const size_t stride1, BASE * data2, const size_t stride2, const size_t N, size_t k)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
BASE v1 = data1[k * stride1];
|
|
Packit |
67cb25 |
BASE v2 = data2[k * stride2];
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
while (k <= N / 2)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
size_t j = 2 * k;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (j < N && data1[j * stride1] < data1[(j + 1) * stride1])
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
j++;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (!(v1 < data1[j * stride1])) /* avoid infinite loop if nan */
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
break;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
data1[k * stride1] = data1[j * stride1];
|
|
Packit |
67cb25 |
data2[k * stride2] = data2[j * stride2];
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
k = j;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
data1[k * stride1] = v1;
|
|
Packit |
67cb25 |
data2[k * stride2] = v2;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
void
|
|
Packit |
67cb25 |
TYPE (gsl_sort) (BASE * data, const size_t stride, const size_t n)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
size_t N;
|
|
Packit |
67cb25 |
size_t k;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (n == 0)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
return; /* No data to sort */
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* We have n_data elements, last element is at 'n_data-1', first at
|
|
Packit |
67cb25 |
'0' Set N to the last element number. */
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
N = n - 1;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
k = N / 2;
|
|
Packit |
67cb25 |
k++; /* Compensate the first use of 'k--' */
|
|
Packit |
67cb25 |
do
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
k--;
|
|
Packit |
67cb25 |
FUNCTION (my, downheap) (data, stride, N, k);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
while (k > 0);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
while (N > 0)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
/* first swap the elements */
|
|
Packit |
67cb25 |
BASE tmp = data[0 * stride];
|
|
Packit |
67cb25 |
data[0 * stride] = data[N * stride];
|
|
Packit |
67cb25 |
data[N * stride] = tmp;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* then process the heap */
|
|
Packit |
67cb25 |
N--;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
FUNCTION (my, downheap) (data, stride, N, 0);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
void
|
|
Packit |
67cb25 |
TYPE (gsl_sort_vector) (TYPE (gsl_vector) * v)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
TYPE (gsl_sort) (v->data, v->stride, v->size) ;
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
void
|
|
Packit |
67cb25 |
TYPE (gsl_sort2) (BASE * data1, const size_t stride1, BASE * data2, const size_t stride2, const size_t n)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
size_t N;
|
|
Packit |
67cb25 |
size_t k;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
if (n == 0)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
return; /* No data to sort */
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* We have n_data elements, last element is at 'n_data-1', first at
|
|
Packit |
67cb25 |
'0' Set N to the last element number. */
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
N = n - 1;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
k = N / 2;
|
|
Packit |
67cb25 |
k++; /* Compensate the first use of 'k--' */
|
|
Packit |
67cb25 |
do
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
k--;
|
|
Packit |
67cb25 |
FUNCTION (my, downheap2) (data1, stride1, data2, stride2, N, k);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
while (k > 0);
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
while (N > 0)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
/* first swap the elements */
|
|
Packit |
67cb25 |
BASE tmp;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
tmp = data1[0 * stride1];
|
|
Packit |
67cb25 |
data1[0 * stride1] = data1[N * stride1];
|
|
Packit |
67cb25 |
data1[N * stride1] = tmp;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
tmp = data2[0 * stride2];
|
|
Packit |
67cb25 |
data2[0 * stride2] = data2[N * stride2];
|
|
Packit |
67cb25 |
data2[N * stride2] = tmp;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
/* then process the heap */
|
|
Packit |
67cb25 |
N--;
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
FUNCTION (my, downheap2) (data1, stride1, data2, stride2, N, 0);
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
}
|
|
Packit |
67cb25 |
|
|
Packit |
67cb25 |
void
|
|
Packit |
67cb25 |
TYPE (gsl_sort_vector2) (TYPE (gsl_vector) * v1, TYPE (gsl_vector) * v2)
|
|
Packit |
67cb25 |
{
|
|
Packit |
67cb25 |
TYPE (gsl_sort2) (v1->data, v1->stride, v2->data, v2->stride, v1->size) ;
|
|
Packit |
67cb25 |
}
|