Merge branch 'master' of ssh://demsky.eecs.uci.edu/home/git/constraint_compiler into...
[satune.git] / src / AST / function.c
1 #include "function.h"
2 #include "table.h"
3 #include "set.h"
4
5
6 Function* allocFunctionOperator(ArithOp op, Set ** domain, uint numDomain, Set * range, OverFlowBehavior overflowbehavior) {
7         FunctionOperator* This = (FunctionOperator*) ourmalloc(sizeof(FunctionOperator));
8         GETFUNCTIONTYPE(This)=OPERATORFUNC;
9         initArrayInitSet(&This->domains, domain, numDomain);
10         This->op=op;
11         This->overflowbehavior = overflowbehavior;
12         This->range=range;
13         return &This->base;
14 }
15
16 Function* allocFunctionTable (Table* table){
17         FunctionTable* This = (FunctionTable*) ourmalloc(sizeof(FunctionTable));
18         GETFUNCTIONTYPE(This)=TABLEFUNC;
19         This->table = table;
20         return &This->base;
21 }
22
23 uint64_t applyFunctionOperator(FunctionOperator* This, uint64_t var1, uint64_t var2) {
24         uint64_t result = 0;
25         switch(This->op){
26                 case ADD:
27                         result = var1+ var2;
28                         break;
29                 case SUB:
30                         result = var1 - var2;
31                         break;
32                 default:
33                         ASSERT(0);
34         }
35         return result;
36 }
37
38 bool isInRangeFunction(FunctionOperator *This, uint64_t val) {
39         return existsInSet(This->range, val);
40 }
41
42 void deleteFunction(Function* This){
43         switch(GETFUNCTIONTYPE(This)){
44         case TABLEFUNC:
45                 break;
46         case OPERATORFUNC:
47                 deleteInlineArraySet(&((FunctionOperator*) This)->domains);
48                 break;
49         default:
50                 ASSERT(0);
51         }
52         ourfree(This);
53 }