Binary encoding for ElementSet and table-based ElementFunction

[satune.git] / src / Encoders / elementencoding.c

diff --git a/src/Encoders/elementencoding.c b/src/Encoders/elementencoding.c
index 4a4bc89670583334c61bfd7adf8208bbd4fb401d..02a9f7f100731a0144d4e32b252fb612f3afc00a 100644 (file)
--- a/src/Encoders/elementencoding.c
+++ b/src/Encoders/elementencoding.c
@@ -1,4 +1,7 @@
 #include "elementencoding.h"
+#include "common.h"
+#include "naiveencoder.h"
+#include "element.h"
 
 void initElementEncoding(ElementEncoding * This, Element *element) {
        This->element=element;
@@ -26,6 +29,30 @@ void allocInUseArrayElement(ElementEncoding *This, uint size) {
        This->inUseArray=ourcalloc(1, size >> 6);
 }
 
+void allocElementConstraintVariables(ElementEncoding* This, uint numVars){
+       This->numVars = numVars;
+       This->variables = ourmalloc(sizeof(Constraint*) * numVars);
+}
+
 void setElementEncodingType(ElementEncoding* This, ElementEncodingType type){
        This->type = type;
 }
+
+void generateBinaryIndexEncodingVars(NaiveEncoder* encoder, ElementEncoding* This){
+       ASSERT(This->type==BINARYINDEX);
+       uint size = getElementSize(This->element);
+       allocElementConstraintVariables(This, NUMBITS(size-1));
+       getArrayNewVars(encoder, This->numVars, This->variables);
+}
+
+void generateElementEncodingVariables(NaiveEncoder* encoder, ElementEncoding* This){
+       ASSERT(This->type!=ELEM_UNASSIGNED);
+       ASSERT(This->variables==NULL);
+       switch(This->type){
+               case BINARYINDEX:
+                       generateBinaryIndexEncodingVars(encoder, This);
+                       break;
+               default:
+                       ASSERT(0);
+       }
+}
\ No newline at end of file