More SAT Encoder
[satune.git] / src / Backend / satencoder.c
1 #include "satencoder.h"
2 #include "structs.h"
3 #include "csolver.h"
4 #include "boolean.h"
5 #include "constraint.h"
6
7 SATEncoder * allocSATEncoder() {
8         SATEncoder *This=ourmalloc(sizeof (SATEncoder));
9         This->varcount=1;
10         return This;
11 }
12
13 void deleteSATEncoder(SATEncoder *This) {
14         ourfree(This);
15 }
16
17 void encodeAllSATEncoder(SATEncoder * This, CSolver *csolver) {
18         VectorBoolean *constraints=csolver->constraints;
19         uint size=getSizeVectorBoolean(constraints);
20         for(uint i=0;i<size;i++) {
21                 Boolean *constraint=getVectorBoolean(constraints, i);
22                 encodeConstraintSATEncoder(This, constraint);
23         }
24 }
25
26 Constraint * encodeConstraintSATEncoder(SATEncoder *This, Boolean *constraint) {
27         switch(constraint->btype) {
28         case ORDERCONST:
29                 return encodeOrderSATEncoder(This, (BooleanOrder *) constraint);
30         case BOOLEANVAR:
31                 return encodeVarSATEncoder(This, (BooleanVar *) constraint);
32         case LOGICOP:
33                 return encodeLogicSATEncoder(This, (BooleanLogic *) constraint);
34         case COMPARE:
35                 return encodeCompareSATEncoder(This, (BooleanComp *) constraint);
36         }
37 }
38
39 Constraint * encodeOrderSATEncoder(SATEncoder *This, BooleanOrder * constraint) {
40         return NULL;
41 }
42
43 Constraint * getNewVarSATEncoder(SATEncoder *This) {
44         Constraint * var=allocVarConstraint(VAR, This->varcount);
45         Constraint * varneg=allocVarConstraint(NOTVAR, This->varcount++);
46         setNegConstraint(var, varneg);
47         setNegConstraint(varneg, var);
48         return var;
49 }
50
51 Constraint * encodeVarSATEncoder(SATEncoder *This, BooleanVar * constraint) {
52         if (constraint->var == NULL) {
53                 constraint->var=getNewVarSATEncoder(This);
54         }
55         return constraint->var;
56 }
57
58 Constraint * encodeLogicSATEncoder(SATEncoder *This, BooleanLogic * constraint) {
59         Constraint *left=encodeConstraintSATEncoder(This, constraint->left);
60         Constraint *right=NULL;
61         if (constraint->right!=NULL)
62                 right=encodeConstraintSATEncoder(This, constraint->right);
63         switch(constraint->op) {
64         case L_AND:
65                 return allocConstraint(AND, left, right);
66         case L_OR:
67                 return allocConstraint(OR, left, right);
68         case L_NOT:
69                 return negateConstraint(allocConstraint(OR, left, NULL));
70         case L_XOR: {
71                 Constraint * nleft=negateConstraint(cloneConstraint(left));
72                 Constraint * nright=negateConstraint(cloneConstraint(right));
73                 return allocConstraint(OR,
74                                                                                                          allocConstraint(AND, left, nright),
75                                                                                                          allocConstraint(AND, nleft, right));
76         }
77         case L_IMPLIES:
78                 return allocConstraint(IMPLIES, left, right);
79         }
80         return NULL;
81 }
82
83 Constraint * encodeCompareSATEncoder(SATEncoder *This, BooleanComp * constraint) {
84         return NULL;
85 }