ADDs replaced by BDDs

This commit is contained in:
Fabrizio Riguzzi 2010-07-27 22:51:05 +02:00
parent 63e0523a43
commit a52d33aa5e
3 changed files with 56 additions and 31 deletions

View File

@ -63,4 +63,5 @@ DdNode * retFactor(DdManager * mgr, factor f, variables v);
double Prob(DdNode *node, variables vars,GHashTable * nodes); double Prob(DdNode *node, variables vars,GHashTable * nodes);
double ProbBool(DdNode *node, int bits, int nBit,int posBVar,variable v, double ProbBool(DdNode *node, int bits, int nBit,int posBVar,variable v,
variables vars,GHashTable * nodes); int mVarIndex,
variables vars,GHashTable * nodes, int comp);

View File

@ -117,26 +117,31 @@ nodes is used to store nodes for which the probability has alread been computed
so that it is not recomputed so that it is not recomputed
*/ */
{ {
int index,mVarIndex,nBit; int index,mVarIndex,nBit,comp;
variable v; variable v;
double res; double res;
double value;
double * value_p; double * value_p;
DdNode **key; DdNode **key,*nodereg;
double *rp; double *rp;
index=node->index; index=Cudd_NodeReadIndex(node);
comp=Cudd_IsComplement(node);
if (Cudd_IsConstant(node)) if (Cudd_IsConstant(node))
{ {
value=node->type.value; if (comp)
return value; return 0.0;
else
return 1.0;
} }
else else
{ {
nodereg=Cudd_Regular(node);
value_p=g_hash_table_lookup(nodes,&node); value_p=g_hash_table_lookup(nodes,&nodereg);
if (value_p!=NULL) if (value_p!=NULL)
{ {
if (comp)
return 1-*value_p;
else
return *value_p; return *value_p;
} }
else else
@ -144,24 +149,29 @@ so that it is not recomputed
mVarIndex=vars.bVar2mVar[index]; mVarIndex=vars.bVar2mVar[index];
v=vars.varar[mVarIndex]; v=vars.varar[mVarIndex];
nBit=v.nBit; nBit=v.nBit;
res=ProbBool(node,0,nBit,0,v,vars,nodes); res=ProbBool(node,0,nBit,0,v,mVarIndex,vars,nodes,0);
key=(DdNode **)malloc(sizeof(DdNode *)); key=(DdNode **)malloc(sizeof(DdNode *));
*key=node; *key=nodereg;
rp=(double *)malloc(sizeof(double)); rp=(double *)malloc(sizeof(double));
*rp=res; *rp=res;
g_hash_table_insert(nodes, key, rp); g_hash_table_insert(nodes, key, rp);
if (comp)
return 1-res;
else
return res; return res;
} }
} }
} }
double ProbBool(DdNode *node, int bits, int nBit,int posBVar,variable v, double ProbBool(DdNode *node, int bits, int nBit,int posBVar,variable v,
variables vars, GHashTable * nodes) int mVarIndex,
variables vars, GHashTable * nodes,int comp)
/* explores a group of binary variables making up the multivalued variable v */ /* explores a group of binary variables making up the multivalued variable v */
{ {
DdNode *T,*F; DdNode *T,*F;
double p,res; double p,res;
double * probs; double * probs;
int comp1,comp2,index,indexF,mVarIndexF;
probs=v.probabilities; probs=v.probabilities;
if (nBit==0) if (nBit==0)
@ -171,28 +181,41 @@ variables vars, GHashTable * nodes)
else else
{ {
p=probs[bits]; p=probs[bits];
if (comp)
res=p*(1-Prob(node,vars,nodes));
else
res=p*Prob(node,vars,nodes); res=p*Prob(node,vars,nodes);
return res; return res;
} }
} }
else else
{ {
if (correctPosition(node->index,v,node,posBVar)) index=Cudd_NodeReadIndex(node);
if (correctPosition(index,v,node,posBVar))
{ {
T = node->type.kids.T; T = Cudd_T(node);
F = node->type.kids.E; F = Cudd_E(node);
bits=bits<<1; bits=bits<<1;
comp1=Cudd_IsComplement(F);
res=ProbBool(T,bits+1,nBit-1,posBVar+1,v,mVarIndex,vars,nodes,comp);
indexF=Cudd_NodeReadIndex(F);
if (Cudd_IsConstant(F))
mVarIndexF=-1;
else
mVarIndexF=vars.bVar2mVar[indexF];
if (mVarIndexF==mVarIndex)
comp2=(comp1 && !comp) || (!comp1 && comp);
else
comp2=comp;
res=res+ ProbBool(F,bits,nBit-1,posBVar+1,v,mVarIndex,vars,nodes,comp2);
res=ProbBool(T,bits+1,nBit-1,posBVar+1,v,vars,nodes)+
ProbBool(F,bits,nBit-1,posBVar+1,v,vars,nodes);
return res; return res;
} }
else else
{ {
bits=bits<<1; bits=bits<<1;
res=ProbBool(node,bits+1,nBit-1,posBVar+1,v,vars,nodes)+ res=ProbBool(node,bits+1,nBit-1,posBVar+1,v,mVarIndex,vars,nodes,comp)+
ProbBool(node,bits,nBit-1,posBVar+1,v,vars,nodes); ProbBool(node,bits,nBit-1,posBVar+1,v,mVarIndex,vars,nodes,comp);
return res; return res;
} }
} }
@ -203,7 +226,8 @@ int correctPosition(int index,variable v, DdNode * node,int posBVar)
currently explored by ProbBool */ currently explored by ProbBool */
{ {
DdNode * bvar; DdNode * bvar;
int ind;
bvar=v.booleanVars[posBVar]; bvar=v.booleanVars[posBVar];
return bvar->index==index; ind=Cudd_NodeReadIndex(bvar);
return ind==index;
} }

View File

@ -139,7 +139,7 @@ static int compute_prob(void)
YAP_Term out,arg1,arg2,arg3,arg4; YAP_Term out,arg1,arg2,arg3,arg4;
variables vars; variables vars;
expr expression; expr expression;
DdNode * function, * add; DdNode * function;
DdManager * mgr; DdManager * mgr;
int nBVar,i,j,intBits,create_dot; int nBVar,i,j,intBits,create_dot;
FILE * file; FILE * file;
@ -179,7 +179,7 @@ static int compute_prob(void)
/* the BDD build by retFunction is converted to an ADD (algebraic decision diagram) /* the BDD build by retFunction is converted to an ADD (algebraic decision diagram)
because it is easier to interpret and to print */ because it is easier to interpret and to print */
add=Cudd_BddToAdd(mgr,function); //add=Cudd_BddToAdd(mgr,function);
//Cudd_PrintInfo(mgr,stderr); //Cudd_PrintInfo(mgr,stderr);
if (create_dot) if (create_dot)
@ -188,7 +188,7 @@ static int compute_prob(void)
nBVar=vars.nBVar; nBVar=vars.nBVar;
for(i=0;i<nBVar;i++) for(i=0;i<nBVar;i++)
names[i]=inames[i]; names[i]=inames[i];
array[0]=add; array[0]=function;
onames[0]="Out"; onames[0]="Out";
file = open_file("cpl.dot", "w"); file = open_file("cpl.dot", "w");
Cudd_DumpDot(mgr,1,array,names,onames,file); Cudd_DumpDot(mgr,1,array,names,onames,file);
@ -204,7 +204,7 @@ static int compute_prob(void)
{ {
dividend=(dividend<<1)+1; dividend=(dividend<<1)+1;
} }
prob=Prob(add,vars,nodes); prob=Prob(function,vars,nodes);
out=YAP_MkFloatTerm(prob); out=YAP_MkFloatTerm(prob);
g_hash_table_foreach (nodes,dealloc,NULL); g_hash_table_foreach (nodes,dealloc,NULL);
g_hash_table_destroy(nodes); g_hash_table_destroy(nodes);