adding config for model_print

[satune.git] / src / AST / function.h

diff --git a/src/AST/function.h b/src/AST/function.h
index 1ade1368c924aab0fe22a69835bf0e71858e1cdc..b8d0c78267cd22d4850b342d93a71fed303d56fd 100644 (file)
--- a/src/AST/function.h
+++ b/src/AST/function.h
@@ -3,6 +3,7 @@
 #include "classlist.h"
 #include "mymemory.h"
 #include "ops.h"
+#include "astops.h"
 #include "structs.h"
 
 class Function {
@@ -11,7 +12,10 @@ public:
        FunctionType type;
        virtual ~Function() {}
        virtual Function *clone(CSolver *solver, CloneMap *map) {ASSERT(0); return NULL;}
-       MEMALLOC;
+       virtual void serialize(Serializer* serialiezr) =0;
+        virtual void print() = 0;
+       virtual Set * getRange() = 0;
+       CMEMALLOC;
 };
 
 class FunctionOperator : public Function {
@@ -24,7 +28,10 @@ public:
        uint64_t applyFunctionOperator(uint numVals, uint64_t *values);
        bool isInRangeFunction(uint64_t val);
        Function *clone(CSolver *solver, CloneMap *map);
-       MEMALLOC;
+       virtual void serialize(Serializer* serialiezr);
+        virtual void print();
+       Set * getRange() {return range;}
+       CMEMALLOC;
 };
 
 class FunctionTable : public Function {
@@ -33,7 +40,10 @@ public:
        UndefinedBehavior undefBehavior;
        FunctionTable (Table *table, UndefinedBehavior behavior);
        Function *clone(CSolver *solver, CloneMap *map);
-       MEMALLOC;
+       virtual void serialize(Serializer* serialiezr);
+        virtual void print();
+       Set * getRange();
+       CMEMALLOC;
 };
 
 #endif