aboutsummaryrefslogtreecommitdiff
path: root/pkg/utilities/nttools/lib/msort.x
blob: 2731351fb2d2acace23741b875cc782c8f94f804 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
include	"reloperr.h"

# MSORT -- General merge sort for arbitrary objects.  X is an integer array
# indexing the array to be sorted.  The user supplied COMPARE function is used
# to compare objects indexed by X:
# 
# 	-1,0,1 = compare (x1, x2)
# 
# where the value returned by COMPARE has the following significance:
# 
# 	-1	obj[x1]  < obj[x2]
# 	 0	obj[x1] == obj[x2]
# 	 1	obj[x1]  > obj[x2]
# 
# MSORT reorders the elements of the X array, which must be of type integer.
#
#	B.Simon		28-Sept-87	First Code

procedure msort (x, nx, nelem, compare)

int	x[ARB]			# array to be sorted
int	nx			# length of array x (Must be >= 2 * nelem)
int	nelem			# number of elements to be sorted
extern	compare()		# function to be called to compare elements
#--
bool	up
int	ielem, jelem, kelem, melem
int	runlen, ilen, jlen

int	compare()

begin
	if (2 * nelem > nx)
	    call error (BOUNDS, "Index array too small")

	# Merging two sorted runs creates a new sorted run twice the length
	# of the original run. Continue this process until the sorted run
	# length is equal to the array length.

	up = false
	for (runlen = 1; runlen < nelem; runlen = 2 * runlen) {

	    # The runs are stored in one of two halves of the x array.
	    # Set the array pointers according to the half the runs are
	    # located in now.

	    if (! up) {
		ielem = 1
		jelem = runlen + 1
		kelem = nx - nelem + 1
		melem = nelem
	    } else {
		ielem = nx - nelem + 1
		jelem = runlen + ielem
		kelem = 1
		melem = nx
	    }

	    # Loop over each pair of runs in the array

	    while (ielem <= melem) {
		ilen = min (runlen, melem-ielem+1)
		jlen = min (runlen, melem-jelem+1)

		# Merge the pair of runs into the other half of the x array

		while (ilen > 0 && jlen > 0) {
		    if (compare (x[ielem], x[jelem]) <= 0) {
			x[kelem] = x[ielem]
			ielem = ielem + 1
			kelem = kelem + 1
			ilen = ilen - 1
		    } else {
			x[kelem] = x[jelem]
			jelem = jelem + 1
			kelem = kelem + 1
			jlen = jlen - 1
		    }
		}

		# Copy the remaining elements from i when j is exhausted

		while (ilen > 0) {
		    x[kelem] = x[ielem]
		    ielem = ielem + 1
		    kelem = kelem + 1
		    ilen = ilen - 1
		}

		# Copy the remaining elements from j when i is exhausted

		while (jlen > 0) {
		    x[kelem] = x[jelem]
		    jelem = jelem + 1
		    kelem = kelem + 1
		    jlen = jlen - 1
		}

		# Set array pointers to next set of runs

		ielem = ielem + runlen
		jelem = jelem + runlen
	    }
	    up = ! up
	}

	# If result is in the upper end of x array, move it to the lower
	# end

	if (up)
	    call amovi (x[nx-nelem+1], x[1], nelem)

	end