029e50127d4da346a789cbaaa43e22b23c2fd14f
[charm.git] / src / conv-core / futures.c
1 #include "converse.h"
2
3 typedef struct Cfuture_data_s
4 {
5   void      *value;
6   int        ready;
7   CthThread  waiters;
8 }
9 *futdata;
10
11 typedef struct CfutureValue_s
12 {
13   char core[CmiMsgHeaderSizeBytes];
14   struct Cfuture_data_s *data;
15   int valsize;
16   double rest[1];
17 }
18 *CfutureValue;
19
20 #define field_offset(t, f) ((CMK_SIZE_T)(((t)0)->f))
21 #define void_to_value(v) ((CfutureValue)(((char*)v)-field_offset(CfutureValue,rest)))
22
23 CpvDeclare(int, CfutureStoreIndex);
24
25 Cfuture CfutureCreate()
26 {
27   futdata data = (futdata)malloc(sizeof(struct Cfuture_data_s));
28   Cfuture result;
29   data->value = 0;
30   data->ready = 0;
31   data->waiters = 0;
32   result.pe = CmiMyPe();
33   result.data = data;
34   return result;
35 }
36
37 static void CfutureAwaken(futdata data, CfutureValue val)
38 {
39   CthThread t;
40   data->value = val;
41   data->ready = 1;
42   for (t=data->waiters; t; t=CthGetNext(t))
43     CthAwaken(t);
44   data->waiters=0;
45 }
46
47 static void CfutureStore(CfutureValue m)
48 {
49   CmiGrabBuffer((void **)&m);
50   CfutureAwaken(m->data, m);
51 }
52
53 void *CfutureCreateBuffer(int bytes)
54 {
55   int valsize = sizeof(struct CfutureValue_s) + bytes;
56   CfutureValue m = (CfutureValue)CmiAlloc(valsize);
57   CmiSetHandler(m, CpvAccess(CfutureStoreIndex));
58   m->valsize = valsize;
59   return (void*)(m->rest);
60 }
61
62 void CfutureDestroyBuffer(void *v)
63 {
64   CfutureValue value = void_to_value(v);
65   CmiFree(v);
66 }
67
68 void CfutureStoreBuffer(Cfuture f, void *value)
69 {
70   CfutureValue m = void_to_value(value);
71   futdata data;
72   if (f.pe == CmiMyPe()) {
73     CfutureAwaken(f.data, m);
74   } else {
75     m->data = f.data;
76     CmiSyncSendAndFree(f.pe, m->valsize, m);
77   }
78 }
79
80 void CfutureSet(Cfuture f, void *value, int len)
81 {
82   void *copy = CfutureCreateBuffer(len);
83   memcpy(copy, value, len);
84   CfutureStoreBuffer(f, copy);
85 }
86
87 void *CfutureWait(Cfuture f)
88 {
89   CthThread self; CfutureValue value; futdata data;
90   if (f.pe != CmiMyPe()) {
91     CmiPrintf("error: CfutureWait: future not local.\n");
92     exit(1);
93   }
94   data = f.data;
95   if (data->ready == 0) {
96     self = CthSelf();
97     CthSetNext(self, data->waiters);
98     data->waiters = self;
99     CthSuspend();
100   }
101   value = data->value;
102   return (void*)(value->rest);
103 }
104
105 void CfutureDestroy(Cfuture f)
106 {
107   if (f.pe != CmiMyPe()) {
108     CmiPrintf("error: CfutureDestroy: future not local.\n");
109     exit(1);
110   }
111   if (f.data->waiters) {
112     CmiPrintf("error: CfutureDestroy: destroying an active future.\n");
113     exit(1);
114   }
115   if (f.data->value) CmiFree(f.data->value);
116   free(f.data);
117 }
118
119 void CfutureModuleInit()
120 {
121   CpvInitialize(int, CfutureStoreIndex);
122   CpvAccess(CfutureStoreIndex) = CmiRegisterHandler(CfutureStore);
123 }