Changeset 40928

Timestamp:

Oct 8, 2019, 9:19:57 PM (7 years ago)

Author:

eugene

Message:

add an int index-only sort

Location:

branches/eam_branches/ohana.20190329/src/libohana

Files:

• include/ohana_sort.h (modified) (1 diff)
• src/sorts.c (modified) (1 diff)

branches/eam_branches/ohana.20190329/src/libohana/include/ohana_sort.h

@@ -84 +84 @@
 
 void dsort_indexonly (double *X, off_t *S, off_t N);
+void dsort_int_indexonly (double *X, int *S, int N);
 
 #endif

branches/eam_branches/ohana.20190329/src/libohana/src/sorts.c

@@ -211 +211 @@
 }
 
-
+/* sort the index of a vector (vector stays unsorted) */
+void dsort_int_indexonly (double *X, int *S, int N) {
+  
+# define SWAPFUNC(A,B){ int itmp; \
+  itmp = S[A]; S[A] = S[B]; S[B] = itmp; \
+}
+# define COMPARE(A,B)((!isfinite(X[S[A]]) && isfinite(X[S[B]])) || (X[S[A]] < X[S[B]]))
+
+  OHANA_SORT (N, COMPARE, SWAPFUNC);
+
+# undef SWAPFUNC
+# undef COMPARE
+
+}
+
+