Skip to content

Instantly share code, notes, and snippets.

@liwt31
Created October 30, 2018 09:06
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save liwt31/da9fb67e5c98c60a661ce73f55ff805e to your computer and use it in GitHub Desktop.
Save liwt31/da9fb67e5c98c60a661ce73f55ff805e to your computer and use it in GitHub Desktop.
Comparision between 2 tim sort
/* Copyright (c) 2010-2014 Christopher Swenson. */
/* Copyright (c) 2012 Google Inc. All Rights Reserved. */
#define _XOPEN_SOURCE
#include <sys/time.h>
#define SORT_NAME sorter
#define SORT_TYPE int64_t
#define SORT_CMP(x, y) ((x) - (y))
#include "sort.h"
#define SORT_NAME stable
#define SORT_TYPE int*
#define SORT_CMP(x, y) (*(x) - *(y))
#include "sort.h"
#define SORT_NAME osorter
#define SORT_TYPE int64_t
#define SORT_CMP(x, y) ((x) - (y))
#include "osort.h"
#define SORT_NAME ostable
#define SORT_TYPE int*
#define SORT_CMP(x, y) (*(x) - *(y))
#include "osort.h"
/* Used to control the stress test */
#define SEED 123
#define MAXSIZE 16000
#define TESTS 1000
#define RAND_RANGE(__n, __min, __max) \
(__n) = (__min) + (long) ((double) ( (double) (__max) - (__min) + 1.0) * ((__n) / (RAND_MAX + 1.0)))
enum {
FILL_RANDOM,
FILL_SAME,
FILL_SORTED,
FILL_SORTED_10,
FILL_SORTED_100,
FILL_SORTED_10000,
FILL_SWAPPED_N2,
FILL_SWAPPED_N8,
FILL_LAST_ELEMENT
};
char *test_names[FILL_LAST_ELEMENT] = {
"random numbers",
"same number",
"sorted numbers",
"sorted blocks of length 10",
"sorted blocks of length 100",
"sorted blocks of length 10000",
"swapped size/2 pairs",
"swapped size/8 pairs"
};
/* used for stdlib */
static __inline int simple_cmp(const void *a, const void *b) {
const int64_t da = *((const int64_t *) a);
const int64_t db = *((const int64_t *) b);
return (da < db) ? -1 : (da == db) ? 0 : 1;
}
static __inline double utime() {
struct timeval t;
gettimeofday(&t, NULL);
return (1000000.0 * t.tv_sec + t.tv_usec);
}
/* helper functions */
int verify(int64_t *dst, const int size) {
int i;
for (i = 1; i < size; i++) {
if (dst[i - 1] > dst[i]) {
printf("Verify failed! at %d", i);
return 0;
}
}
return 1;
}
static void fill_random(int64_t *dst, const int size) {
int i;
for (i = 0; i < size; i++) {
dst[i] = lrand48();
}
}
static void fill_same(int64_t *dst, const int size) {
int i;
for (i = 0; i < size; i++) {
dst[i] = 0;
}
}
static void fill_sorted(int64_t *dst, const int size) {
int i;
for (i = 0; i < size; i++) {
dst[i] = i;
}
}
static void fill_sorted_blocks(int64_t *dst, const int size, const int block_size) {
int i, filled, this_block_size;
filled = 0;
for (i = 0; i < size; i += block_size) {
this_block_size = (filled + block_size) < size ? block_size : (size - filled);
fill_random(dst + filled, this_block_size);
qsort(dst + filled, this_block_size, sizeof(int64_t), simple_cmp);
filled += this_block_size;
}
}
static void fill_swapped(int64_t *dst, const int size, const int swapped_cnt) {
int i, tmp;
size_t ind1 = 0;
size_t ind2 = 0;
fill_sorted(dst, size);
for (i = 0; i < swapped_cnt; i++) {
ind1 = lrand48();
RAND_RANGE(ind1, 0, size);
ind2 = lrand48();
RAND_RANGE(ind2, 0, size);
}
tmp = dst[ind1];
dst[ind1] = dst[ind2];
dst[ind2] = tmp;
}
static void fill(int64_t *dst, const int size, int type) {
switch (type) {
case FILL_SORTED:
fill_sorted(dst, size);
break;
case FILL_SORTED_10:
fill_sorted_blocks(dst, size, 10);
break;
case FILL_SORTED_100:
fill_sorted_blocks(dst, size, 100);
break;
case FILL_SORTED_10000:
fill_sorted_blocks(dst, size, 10000);
break;
case FILL_SWAPPED_N2:
fill_swapped(dst, size, size / 2);
break;
case FILL_SWAPPED_N8:
fill_swapped(dst, size, size / 8);
break;
case FILL_SAME:
fill_same(dst, size);
break;
case FILL_RANDOM:
default:
fill_random(dst, size);
break;
}
}
#define TEST_STDLIB(name) do { \
res = 0; \
diff = 0; \
printf("%-29s", "stdlib " #name ); \
for (test = 0; test < sizes_cnt; test++) { \
int64_t size = sizes[test]; \
int64_t dst[MAXSIZE]; \
fill(dst, size, type); \
usec1 = utime(); \
name (dst, size, sizeof(int64_t), simple_cmp); \
usec2 = utime(); \
res = verify(dst, size); \
if (!res) { \
break; \
} \
diff += usec2 - usec1; \
} \
printf(" - %s, %10.1f usec\n", res ? "ok" : "FAILED", diff); \
if (!res) return 0; \
} while (0)
#define TEST_SORT_H(head, name) do { \
res = 0; \
diff = 0; \
printf("%-29s", "sort.h " #head #name); \
for (test = 0; test < sizes_cnt; test++) { \
int64_t size = sizes[test]; \
int64_t dst[MAXSIZE]; \
fill(dst, size, type); \
usec1 = utime(); \
head ## _ ## name (dst, size); \
usec2 = utime(); \
res = verify(dst, size); \
if (!res) { \
break; \
} \
diff += usec2 - usec1; \
} \
printf(" - %s, %10.1f usec\n", res ? "ok" : "FAILED", diff); \
if (!res) return 0; \
} while (0)
int run_tests(int64_t *sizes, int sizes_cnt, int type) {
int test, res;
double usec1, usec2, diff;
printf("-------\nRunning tests with %s:\n-------\n", test_names[type]);
TEST_SORT_H(sorter, tim_sort);
TEST_SORT_H(osorter, tim_sort);
return 0;
}
/* stability testing functions */
/* cheap hack to keep a copy to compare against */
static int **original;
static int first_original = 0;
/* make a int* array */
void make_intp_array(int **array, int64_t size, int num_values) {
int64_t i;
if (first_original == 0) {
first_original = 1;
original = malloc(sizeof(int *) * size);
}
for (i = 0; i < size; i++) {
array[i] = original[i] = malloc(sizeof(int));
*(array[i]) = lrand48() % num_values;
}
}
/* free all the pointers */
void clean_intp_array(int **array, int64_t size) {
int64_t i;
for (i = 0; i < size; i++) {
free(array[i]);
}
}
/* find the first instance of an element in an array of pointers */
int64_t find_next(int **array, int64_t start, int64_t last, int value) {
int64_t i;
for (i = start; i < last; i++) {
if (*(array[i]) == value) {
break;
}
}
return i;
}
/* verify that the given list is stable */
int verify_stable(int **array, int64_t size, int num_values) {
int value;
int64_t i = 0;
int64_t j = 0;
for (value = 0; value < num_values; value++) {
while (1) {
i = find_next(original, i, size, value);
/* unlikely, but possible */
if (i == size) {
break;
}
j = find_next(array, j, size, value);
if (j == size) {
return 0;
}
if (original[i] != array[j]) {
return 0;
}
i++;
j++;
}
}
return 1;
}
/* Checks that given sort function is stable. */
void check_stable(char *name, void (*sort_fun)(int **arr, size_t size), int size, int num_values) {
int **array = malloc(sizeof(int *) * size);
make_intp_array(array, size, num_values);
sort_fun(array, size);
printf("%21s -- %s\n", name, verify_stable(array, size, num_values) ? "stable" : "UNSTABLE");
clean_intp_array(array, size);
free(array);
}
/* Check which sorts are stable. */
void stable_tests() {
int size = 100000;
int num_values = 1000;
check_stable("tim sort", stable_tim_sort, size, num_values);
check_stable("otim sort", ostable_tim_sort, size, num_values);
}
int main(void) {
int i = 0;
int64_t sizes[TESTS];
srand48(SEED);
stable_tests();
fill_random(sizes, TESTS);
for (i = 0; i < TESTS; i++) {
RAND_RANGE(sizes[i], 0, MAXSIZE);
}
for (i = 0; i < FILL_LAST_ELEMENT; i++) {
int result = run_tests(sizes, TESTS, i);
if (result) {
return 1;
}
}
return 0;
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment