Annotation of libaitio/example/rb.c, revision 1.2.30.1
1.2 misho 1: #include <stdio.h>
2: #include <stdlib.h>
3: #include <string.h>
4: #include <unistd.h>
5: #include <atree.h>
6: #include <sys/time.h>
1.2.30.1! misho 7: #include <sys/queue.h>
1.2 misho 8:
9: struct blah {
10: int n;
11: RB_ENTRY(blah) node;
12: SPLAY_ENTRY(blah) node1;
13: AVL_ENTRY(blah) node2;
1.2.30.1! misho 14: SLIST_ENTRY(blah) node3;
! 15: TAILQ_ENTRY(blah) node4;
1.2 misho 16: };
17: RB_HEAD(rb_head, blah) rbh;
18: SPLAY_HEAD(sp_head, blah) sph;
19: AVL_HEAD(avl_head, blah) avlh;
1.2.30.1! misho 20: SLIST_HEAD(, blah) slh;
! 21: TAILQ_HEAD(, blah) tqh;
1.2 misho 22:
23: static int
24: rb_cmp(struct blah *a, struct blah *b)
25: {
26: int n = (a ? a->n : 0) - (b ? b->n : 0);
27:
28: if (!n)
29: return 0;
30: else if (n < 0)
31: return -1;
32: else
33: return 1;
34: }
35:
36: RB_PROTOTYPE(rb_head, blah, node, rb_cmp);
37: RB_GENERATE(rb_head, blah, node, rb_cmp);
38: SPLAY_PROTOTYPE(sp_head, blah, node1, rb_cmp);
39: SPLAY_GENERATE(sp_head, blah, node1, rb_cmp);
40: AVL_PROTOTYPE(avl_head, blah, node2, rb_cmp)
41: AVL_GENERATE(avl_head, blah, node2, rb_cmp)
42:
43:
44: int
45: main()
46: {
47: struct blah *b, *b2;
48: register int i;
49: struct timeval before, after;
50:
51: RB_INIT(&rbh);
52: SPLAY_INIT(&sph);
1.2.30.1! misho 53: AVL_INIT(&avlh);
! 54: SLIST_INIT(&slh);
! 55: TAILQ_INIT(&tqh);
1.2 misho 56: srandom(getpid());
57:
58: gettimeofday(&before, NULL);
59: for (i = 0; i < 1000000; i++) {
60: b = malloc(sizeof(struct blah));
61: memset(b, 0, sizeof(struct blah));
62: b->n = random();
63: if (RB_INSERT(rb_head, &rbh, b))
64: free(b);
65: }
66: gettimeofday(&after, NULL);
67: printf("RB::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
68: gettimeofday(&before, NULL);
69: for (i = 0; i < 1000000; i++) {
70: b = malloc(sizeof(struct blah));
71: memset(b, 0, sizeof(struct blah));
72: b->n = random();
73: if (SPLAY_INSERT(sp_head, &sph, b))
74: free(b);
75: }
76: gettimeofday(&after, NULL);
77: printf("SP::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
78:
79: gettimeofday(&before, NULL);
80: for (i = 0; i < 1000000; i++) {
81: b = malloc(sizeof(struct blah));
82: memset(b, 0, sizeof(struct blah));
83: b->n = random();
84: if (AVL_INSERT(avl_head, &avlh, b))
85: free(b);
86: }
87: gettimeofday(&after, NULL);
88: printf("AVL::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
89:
1.2.30.1! misho 90: gettimeofday(&before, NULL);
! 91: for (i = 0; i < 1000000; i++) {
! 92: b = malloc(sizeof(struct blah));
! 93: memset(b, 0, sizeof(struct blah));
! 94: b->n = random();
! 95: SLIST_INSERT_HEAD(&slh, b, node3);
! 96: }
! 97: gettimeofday(&after, NULL);
! 98: printf("SLIST::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
! 99: gettimeofday(&before, NULL);
! 100: for (i = 0; i < 1000000; i++) {
! 101: b = malloc(sizeof(struct blah));
! 102: memset(b, 0, sizeof(struct blah));
! 103: b->n = random();
! 104: TAILQ_INSERT_HEAD(&tqh, b, node4);
! 105: }
! 106: gettimeofday(&after, NULL);
! 107: printf("TAILQ::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
! 108:
1.2 misho 109: printf("---\n");
110:
111: b2 = malloc(sizeof(struct blah));
112: gettimeofday(&before, NULL);
113: for (i = 0; i < 10000; i++) {
114: memset(b2, 0, sizeof(struct blah));
115: b2->n = random();
116: b = SPLAY_FIND(sp_head, &sph, b2);
117: if (b)
118: printf("SP::Found key %d(%s) -> %d\n", i, RB_COLOR(b, node) == RB_BLACK ? "black" : "red", b->n);
119: }
120: gettimeofday(&after, NULL);
121: printf("SP::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
122:
123: printf("---\n");
124:
125: gettimeofday(&before, NULL);
126: for (i = 0; i < 10000; i++) {
127: memset(b2, 0, sizeof(struct blah));
128: b2->n = random();
129: b = RB_FIND(rb_head, &rbh, b2);
130: if (b)
131: printf("RB::Found key %d(%s) -> %d\n", i, RB_COLOR(b, node) == RB_BLACK ? "black" : "red", b->n);
132: }
133: gettimeofday(&after, NULL);
134: printf("RB::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
135:
136: printf("---\n");
137:
138: gettimeofday(&before, NULL);
139: for (i = 0; i < 10000; i++) {
140: memset(b2, 0, sizeof(struct blah));
141: b2->n = random();
142: b = AVL_FIND(avl_head, &avlh, b2);
143: if (b)
144: printf("AVL::Found key %d %d-> %d next=%d prev=%d\n", i, b2->n, b->n,
145: AVL_NEXT(avl_head, &avlh, b) ? AVL_NEXT(avl_head, &avlh, b)->n : 0,
146: AVL_PREV(avl_head, &avlh, b) ? AVL_PREV(avl_head, &avlh, b)->n : 0);
147: }
148: gettimeofday(&after, NULL);
149: printf("AVL::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
150:
151: printf("---\n");
152:
1.2.30.1! misho 153: free(b2);
! 154:
! 155: gettimeofday(&before, NULL);
! 156: while ((b = TAILQ_FIRST(&tqh))) {
! 157: TAILQ_REMOVE(&tqh, b, node4);
! 158: free(b);
! 159: }
! 160: gettimeofday(&after, NULL);
! 161: printf("TAILQ::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
! 162: gettimeofday(&before, NULL);
! 163: while ((b = SLIST_FIRST(&slh))) {
! 164: SLIST_REMOVE(&slh, b, blah, node3);
! 165: free(b);
! 166: }
! 167: gettimeofday(&after, NULL);
! 168: printf("SLIST::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
! 169:
1.2 misho 170: gettimeofday(&before, NULL);
171: AVL_FOREACH_SAFE(b, avl_head, &avlh, b2) {
172: /*
173: printf("aaa %p[%d] %p[%d] --- ", b, b ? b->n : 0, b2, b2 ? b2->n : 0);
174: printf("b:%p %p /// ", b ? AVL_LEFT(b, node2) : b, b ? AVL_RIGHT(b, node2) : b);
175: printf("b2:%p %p\n", b2 ? AVL_LEFT(b2, node2) : b2, b2 ? AVL_RIGHT(b2, node2) : b2); fflush(stdout);
176: */
177: AVL_REMOVE(avl_head, &avlh, b);
178: free(b);
179: }
180: gettimeofday(&after, NULL);
181: printf("AVL::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
1.2.30.1! misho 182:
1.2 misho 183: gettimeofday(&before, NULL);
184: for (b = SPLAY_MIN(sp_head, &sph); b; b = b2) {
185: b2 = SPLAY_NEXT(sp_head, &sph, b);
186: SPLAY_REMOVE(sp_head, &sph, b);
187: }
188: gettimeofday(&after, NULL);
189: printf("SP::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
190:
191: gettimeofday(&before, NULL);
192: RB_FOREACH_SAFE(b, rb_head, &rbh, b2) {
193: RB_REMOVE(rb_head, &rbh, b);
194: free(b);
195: }
196: gettimeofday(&after, NULL);
197: printf("RB::Time %f\n", (after.tv_sec - before.tv_sec) + (after.tv_usec - before.tv_usec) / 1e6);
198: return !AVL_EMPTY(&avlh);
199: }
FreeBSD-CVSweb <freebsd-cvsweb@FreeBSD.org>