source: proiecte/swift/trunk/test/test_deque.c @ 176

/*
 * swift_context.c
 *
 * (c) 2009 Ionut Rosoiu <ionut.rosoiu@gmail.com>
 *
 */

#include <stdlib.h>
#include <stdio.h>
#include <unistd.h>
#include "swift.h"

#define NUM_THREADS     2
#define MAX_VAL                 100

swift_deque_t deque;
int vals[MAX_VAL];
swift_frame_t frames[MAX_VAL];

void* pusher(void *arg) {
        swift_status_t status;
        int i, j = 0;

        printf("pusher started.\n");
        for (i=0; i<MAX_VAL; i++) {
                do {
                        swift_deque_push(&deque, &frames[j++], &status);
                } while (status == SWIFT_DEQUE_FULL);
        }
        printf("pusher finished.\n");

        return arg;
}

void* popper(void *arg) {
        swift_status_t status;
        swift_frame_t *frame;
        int i;

        printf("popper started.\n");
        for (i=0; i<MAX_VAL; i++) {
                frame = swift_deque_pop(&deque, &status);
                if (frame) {
                        vals[frame->dependencies_no] = 1;
                }
        }
        printf("popper finished.\n");

        return arg;
}

void* stealer(void *arg) {
        swift_status_t status;
        swift_thread_t *thread = arg;
        swift_frame_t *frame;
        int i;

        printf("stealer %d started\n", thread->id);
        for (i=0; i<MAX_VAL + 1; i++) {
                do {
                        frame = swift_deque_steal(&deque, &status);
                } while (status == SWIFT_DEQUE_ABORT);

                if (frame) {
                        printf("[S]    pop %d (%p)\n", frame->dependencies_no, (void*)frame);
                        vals[frame->dependencies_no] = 1;
                }
        }
        printf("stealer %d finished.\n", thread->id);

        return arg;
}

swift_entrypoint_t funcs[4] = {
                pusher, // must be first
                popper,
                stealer,
                stealer
};

int main() {
        swift_thread_t threads[NUM_THREADS];
        int i;

        //swift_assert(NUM_THREADS == 4);

        /* initialize the deque */
        swift_deque_init(&threads[0], &deque, MAX_VAL);

        /* initialize the value vector */
        for (i=0; i<MAX_VAL; i++) {
                frames[i].dependencies_no = i;
                vals[i] = 0;
        }

        /* start the threads */
        for (i=0; i<NUM_THREADS; i++) {
                threads[i].id = i;

                swift_thread_start(&threads[i], funcs[i]);
        }

        printf("cleaning-up...\n");

        /* clean-up */
        for (i=0; i<NUM_THREADS; i++) {
                swift_thread_wait(&threads[i]);
                printf("thread %d done.\n", i);
        }

        for (i=0; i<MAX_VAL; i++) {
                printf("%d ", i);
                swift_assert(vals[i]);
        }

        swift_deque_destroy(&threads[0], &deque);

        return 0;
}