benchmark/ms-queue/main.c

   1 #include <stdlib.h>
   2 #include <stdio.h>
   3 #include <threads.h>
   4
   5 #include "my_queue.h"
   6 #include "model-assert.h"
   7
   8 static int procs = 2;
   9 static queue_t *queue;
  10 static thrd_t *threads;
  11 static unsigned int *input;
  12 static unsigned int *output;
  13 static int num_threads;
  14
  15 int get_thread_num()
  16 {
  17         thrd_t curr = thrd_current();
  18         int i;
  19         for (i = 0; i < num_threads; i++)
  20                 if (curr.priv == threads[i].priv)
  21                         return i;
  22         //MODEL_ASSERT(0);
  23         return -1;
  24 }
  25
  26 static void main_task(void *param)
  27 {
  28         unsigned int val;
  29         int pid = *((int *)param);
  30
  31         if (!pid) {
  32                 input[0] = 17;
  33                 //enqueue(queue, input[0]);
  34                 enqueue(queue, input[0]);
  35                 //output[0] = dequeue(queue);
  36         } else {
  37                 input[1] = 37;
  38                 //enqueue(queue, input[1]);
  39                 //output[1] = dequeue(queue);
  40                 //output[0] = dequeue(queue);
  41                 bool succ = dequeue(queue, &output[0]);
  42         }
  43 }
  44
  45 int user_main(int argc, char **argv)
  46 {
  47         /**
  48                 @Begin
  49                 @Entry_point
  50                 @End
  51         */
  52         int i;
  53         int *param;
  54         unsigned int in_sum = 0, out_sum = 0;
  55
  56         queue = calloc(1, sizeof(*queue));
  57         //MODEL_ASSERT(queue);
  58
  59         num_threads = procs;
  60         threads = malloc(num_threads * sizeof(thrd_t));
  61         param = malloc(num_threads * sizeof(*param));
  62         input = calloc(num_threads, sizeof(*input));
  63         output = calloc(num_threads, sizeof(*output));
  64
  65         init_queue(queue, num_threads);
  66         for (i = 0; i < num_threads; i++) {
  67                 param[i] = i;
  68                 thrd_create(&threads[i], main_task, &param[i]);
  69         }
  70         for (i = 0; i < num_threads; i++)
  71                 thrd_join(threads[i]);
  72 /*
  73         for (i = 0; i < num_threads; i++) {
  74                 in_sum += input[i];
  75                 out_sum += output[i];
  76         }
  77         for (i = 0; i < num_threads; i++)
  78                 printf("input[%d] = %u\n", i, input[i]);
  79         for (i = 0; i < num_threads; i++)
  80                 printf("output[%d] = %u\n", i, output[i]);
  81         //MODEL_ASSERT(in_sum == out_sum);
  82         */
  83
  84         free(param);
  85         free(threads);
  86         free(queue);
  87
  88         return 0;
  89 }