1173 lines
		
	
	
		
			52 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			1173 lines
		
	
	
		
			52 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
| /******************************************************************************\
 | |
| *                                                                              *
 | |
| *    SimpleCUDD library (www.cs.kuleuven.be/~theo/tools/simplecudd.html)       *
 | |
| *  SimpleCUDD was developed at Katholieke Universiteit Leuven(www.kuleuven.be) *
 | |
| *                                                                              *
 | |
| *  Copyright Katholieke Universiteit Leuven 2008                               *
 | |
| *                                                                              *
 | |
| *  Author: Theofrastos Mantadelis, Angelika Kimmig, Bernd Gutmann              *
 | |
| *  File: ProblogBDD.c                                                          *
 | |
| *                                                                              *
 | |
| ********************************************************************************
 | |
| *                                                                              *
 | |
| * Artistic License 2.0                                                         *
 | |
| *                                                                              *
 | |
| * Copyright (c) 2000-2006, The Perl Foundation.                                *
 | |
| *                                                                              *
 | |
| * Everyone is permitted to copy and distribute verbatim copies of this license *
 | |
| * document, but changing it is not allowed.                                    *
 | |
| *                                                                              *
 | |
| * Preamble                                                                     *
 | |
| *                                                                              *
 | |
| * This license establishes the terms under which a given free software Package *
 | |
| * may be copied, modified, distributed, and/or redistributed. The intent is    *
 | |
| * that the Copyright Holder maintains some artistic control over the           *
 | |
| * development of that Package while still keeping the Package available as     *
 | |
| * open source and free software.                                               *
 | |
| *                                                                              *
 | |
| * You are always permitted to make arrangements wholly outside of this license *
 | |
| * directly with the Copyright Holder of a given Package. If the terms of this  *
 | |
| * license do not permit the full use that you propose to make of the Package,  *
 | |
| * you should contact the Copyright Holder and seek a different licensing       *
 | |
| * arrangement.                                                                 *
 | |
| * Definitions                                                                  *
 | |
| *                                                                              *
 | |
| * "Copyright Holder" means the individual(s) or organization(s) named in the   *
 | |
| * copyright notice for the entire Package.                                     *
 | |
| *                                                                              *
 | |
| * "Contributor" means any party that has contributed code or other material to *
 | |
| * the Package, in accordance with the Copyright Holder's procedures.           *
 | |
| *                                                                              *
 | |
| * "You" and "your" means any person who would like to copy, distribute, or     *
 | |
| * modify the Package.                                                          *
 | |
| *                                                                              *
 | |
| * "Package" means the collection of files distributed by the Copyright Holder, *
 | |
| * and derivatives of that collection and/or of those files. A given Package    *
 | |
| * may consist of either the Standard Version, or a Modified Version.           *
 | |
| *                                                                              *
 | |
| * "Distribute" means providing a copy of the Package or making it accessible   *
 | |
| * to anyone else, or in the case of a company or organization, to others       *
 | |
| * outside of your company or organization.                                     *
 | |
| *                                                                              *
 | |
| * "Distributor Fee" means any fee that you charge for Distributing this        *
 | |
| * Package or providing support for this Package to another party. It does not  *
 | |
| * mean licensing fees.                                                         *
 | |
| *                                                                              *
 | |
| * "Standard Version" refers to the Package if it has not been modified, or has *
 | |
| * been modified only in ways explicitly requested by the Copyright Holder.     *
 | |
| *                                                                              *
 | |
| * "Modified Version" means the Package, if it has been changed, and such       *
 | |
| * changes were not explicitly requested by the Copyright Holder.               *
 | |
| *                                                                              *
 | |
| * "Original License" means this Artistic License as Distributed with the       *
 | |
| * Standard Version of the Package, in its current version or as it may be      *
 | |
| * modified by The Perl Foundation in the future.                               *
 | |
| *                                                                              *
 | |
| * "Source" form means the source code, documentation source, and configuration *
 | |
| * files for the Package.                                                       *
 | |
| *                                                                              *
 | |
| * "Compiled" form means the compiled bytecode, object code, binary, or any     *
 | |
| * other form resulting from mechanical transformation or translation of the    *
 | |
| * Source form.                                                                 *
 | |
| * Permission for Use and Modification Without Distribution                     *
 | |
| *                                                                              *
 | |
| * (1) You are permitted to use the Standard Version and create and use         *
 | |
| * Modified Versions for any purpose without restriction, provided that you do  *
 | |
| * not Distribute the Modified Version.                                         *
 | |
| * Permissions for Redistribution of the Standard Version                       *
 | |
| *                                                                              *
 | |
| * (2) You may Distribute verbatim copies of the Source form of the Standard    *
 | |
| * Version of this Package in any medium without restriction, either gratis or  *
 | |
| * for a Distributor Fee, provided that you duplicate all of the original       *
 | |
| * copyright notices and associated disclaimers. At your discretion, such       *
 | |
| * verbatim copies may or may not include a Compiled form of the Package.       *
 | |
| *                                                                              *
 | |
| * (3) You may apply any bug fixes, portability changes, and other              *
 | |
| * modifications made available from the Copyright Holder. The resulting        *
 | |
| * Package will still be considered the Standard Version, and as such will be   *
 | |
| * subject to the Original License.                                             *
 | |
| * Distribution of Modified Versions of the Package as Source                   *
 | |
| *                                                                              *
 | |
| * (4) You may Distribute your Modified Version as Source (either gratis or for *
 | |
| * a Distributor Fee, and with or without a Compiled form of the Modified       *
 | |
| * Version) provided that you clearly document how it differs from the Standard *
 | |
| * Version, including, but not limited to, documenting any non-standard         *
 | |
| * features, executables, or modules, and provided that you do at least ONE of  *
 | |
| * the following:                                                               *
 | |
| *                                                                              *
 | |
| * (a) make the Modified Version available to the Copyright Holder of the       *
 | |
| * Standard Version, under the Original License, so that the Copyright Holder   *
 | |
| * may include your modifications in the Standard Version.                      *
 | |
| * (b) ensure that installation of your Modified Version does not prevent the   *
 | |
| * user installing or running the Standard Version. In addition, the Modified   *
 | |
| * Version must bear a name that is different from the name of the Standard     *
 | |
| * Version.                                                                     *
 | |
| * (c) allow anyone who receives a copy of the Modified Version to make the     *
 | |
| * Source form of the Modified Version available to others under                *
 | |
| * (i) the Original License or                                                  *
 | |
| * (ii) a license that permits the licensee to freely copy, modify and          *
 | |
| * redistribute the Modified Version using the same licensing terms that apply  *
 | |
| * to the copy that the licensee received, and requires that the Source form of *
 | |
| * the Modified Version, and of any works derived from it, be made freely       *
 | |
| * available in that license fees are prohibited but Distributor Fees are       *
 | |
| * allowed.                                                                     *
 | |
| * Distribution of Compiled Forms of the Standard Version or Modified Versions  *
 | |
| * without the Source                                                           *
 | |
| *                                                                              *
 | |
| * (5) You may Distribute Compiled forms of the Standard Version without the    *
 | |
| * Source, provided that you include complete instructions on how to get the    *
 | |
| * Source of the Standard Version. Such instructions must be valid at the time  *
 | |
| * of your distribution. If these instructions, at any time while you are       *
 | |
| * carrying out such distribution, become invalid, you must provide new         *
 | |
| * instructions on demand or cease further distribution. If you provide valid   *
 | |
| * instructions or cease distribution within thirty days after you become aware *
 | |
| * that the instructions are invalid, then you do not forfeit any of your       *
 | |
| * rights under this license.                                                   *
 | |
| *                                                                              *
 | |
| * (6) You may Distribute a Modified Version in Compiled form without the       *
 | |
| * Source, provided that you comply with Section 4 with respect to the Source   *
 | |
| * of the Modified Version.                                                     *
 | |
| * Aggregating or Linking the Package                                           *
 | |
| *                                                                              *
 | |
| * (7) You may aggregate the Package (either the Standard Version or Modified   *
 | |
| * Version) with other packages and Distribute the resulting aggregation        *
 | |
| * provided that you do not charge a licensing fee for the Package. Distributor *
 | |
| * Fees are permitted, and licensing fees for other components in the           *
 | |
| * aggregation are permitted. The terms of this license apply to the use and    *
 | |
| * Distribution of the Standard or Modified Versions as included in the         *
 | |
| * aggregation.                                                                 *
 | |
| *                                                                              *
 | |
| * (8) You are permitted to link Modified and Standard Versions with other      *
 | |
| * works, to embed the Package in a larger work of your own, or to build        *
 | |
| * stand-alone binary or bytecode versions of applications that include the     *
 | |
| * Package, and Distribute the result without restriction, provided the result  *
 | |
| * does not expose a direct interface to the Package.                           *
 | |
| * Items That are Not Considered Part of a Modified Version                     *
 | |
| *                                                                              *
 | |
| * (9) Works (including, but not limited to, modules and scripts) that merely   *
 | |
| * extend or make use of the Package, do not, by themselves, cause the Package  *
 | |
| * to be a Modified Version. In addition, such works are not considered parts   *
 | |
| * of the Package itself, and are not subject to the terms of this license.     *
 | |
| * General Provisions                                                           *
 | |
| *                                                                              *
 | |
| * (10) Any use, modification, and distribution of the Standard or Modified     *
 | |
| * Versions is governed by this Artistic License. By using, modifying or        *
 | |
| * distributing the Package, you accept this license. Do not use, modify, or    *
 | |
| * distribute the Package, if you do not accept this license.                   *
 | |
| *                                                                              *
 | |
| * (11) If your Modified Version has been derived from a Modified Version made  *
 | |
| * by someone other than you, you are nevertheless required to ensure that your *
 | |
| * Modified Version complies with the requirements of this license.             *
 | |
| *                                                                              *
 | |
| * (12) This license does not grant you the right to use any trademark, service *
 | |
| * mark, tradename, or logo of the Copyright Holder.                            *
 | |
| *                                                                              *
 | |
| * (13) This license includes the non-exclusive, worldwide, free-of-charge      *
 | |
| * patent license to make, have made, use, offer to sell, sell, import and      *
 | |
| * otherwise transfer the Package with respect to any patent claims licensable  *
 | |
| * by the Copyright Holder that are necessarily infringed by the Package. If    *
 | |
| * you institute patent litigation (including a cross-claim or counterclaim)    *
 | |
| * against any party alleging that the Package constitutes direct or            *
 | |
| * contributory patent infringement, then this Artistic License to you shall    *
 | |
| * terminate on the date that such litigation is filed.                         *
 | |
| *                                                                              *
 | |
| * (14) Disclaimer of Warranty: THE PACKAGE IS PROVIDED BY THE COPYRIGHT HOLDER *
 | |
| * AND CONTRIBUTORS "AS IS' AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES. THE  *
 | |
| * IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE, OR  *
 | |
| * NON-INFRINGEMENT ARE DISCLAIMED TO THE EXTENT PERMITTED BY YOUR LOCAL LAW.   *
 | |
| * UNLESS REQUIRED BY LAW, NO COPYRIGHT HOLDER OR CONTRIBUTOR WILL BE LIABLE    *
 | |
| * FOR ANY DIRECT, INDIRECT, INCIDENTAL, OR CONSEQUENTIAL DAMAGES ARISING IN    *
 | |
| * ANY WAY OUT OF THE USE OF THE PACKAGE, EVEN IF ADVISED OF THE POSSIBILITY OF *
 | |
| * SUCH DAMAGE.                                                                 *
 | |
| *                                                                              *
 | |
| *         The End                                                              *
 | |
| *                                                                              *
 | |
| \******************************************************************************/
 | |
| 
 | |
| #include "simplecudd.h"
 | |
| #include "problogmath.h"
 | |
| #include "pqueue.h"
 | |
| #include "iqueue.h"
 | |
| #include <signal.h>
 | |
| #include <stdarg.h>
 | |
| #define  VERSION "2.0.0"
 | |
| 
 | |
| typedef struct _parameters {
 | |
|   int loadfile;
 | |
|   int savedfile;
 | |
|   int exportfile;
 | |
|   int inputfile;
 | |
|   int debug;
 | |
|   int errorcnt;
 | |
|   int *error;
 | |
|   int method;
 | |
|   int queryid;
 | |
|   int timeout;
 | |
|   double sigmoid_slope;
 | |
|   int online;
 | |
|   int maxbufsize;
 | |
|   char *ppid;
 | |
|   int orderfile;
 | |
| } parameters;
 | |
| 
 | |
| typedef struct _gradientpair {
 | |
|   double probability;
 | |
|   double gradient;
 | |
| } gradientpair;
 | |
| 
 | |
| typedef struct _extmanager {
 | |
|   DdManager *manager;
 | |
|   DdNode *t, *f;
 | |
|   hisqueue *his;
 | |
|   namedvars varmap;
 | |
| } extmanager;
 | |
| 
 | |
| int argtype(const char *arg);
 | |
| void printhelp(int argc, char **arg);
 | |
| parameters loadparam(int argc, char **arg);
 | |
| parameters params;
 | |
| 
 | |
| void handler(int num);
 | |
| void pidhandler(int num);
 | |
| void termhandler(int num);
 | |
| 
 | |
| void myexpand(extmanager MyManager, DdNode *Current);
 | |
| double CalcProbability(extmanager MyManager, DdNode *Current);
 | |
| double CalcProbabilitySigmoid(extmanager MyManager, DdNode *Current);
 | |
| gradientpair CalcGradient(extmanager MyManager, DdNode *Current, int TargetVar, char *TargetPattern, int type);
 | |
| double CalcExpectedCountsUp(extmanager * MyManager, DdNode *Current, char *query_id) ;
 | |
| double CalcExpectedCountsDown(extmanager * MyManager, DdNode *Current, char *query_id);
 | |
| double CalcExpectedCounts(extmanager * MyManager, DdNode *Current, char *query_id, int calcdown_needed);
 | |
| int patterncalculated(char *pattern, extmanager MyManager, int loc);
 | |
| char * extractpattern(char *thestr);
 | |
| 
 | |
| int main(int argc, char **arg) {
 | |
|   extmanager MyManager;
 | |
|   DdNode *bdd, **forest, *bakbdd;
 | |
|   bddfileheader fileheader;
 | |
|   int i, ivarcnt, code, curbdd;
 | |
|   gradientpair tvalue;
 | |
|   double probability = -1.0;
 | |
|   char *varpattern;
 | |
|   varpattern = NULL;
 | |
|   code = -1;
 | |
|   params = loadparam(argc, arg);
 | |
| 
 | |
|   if (params.errorcnt > 0) {
 | |
|     printhelp(argc, arg);
 | |
|     for (i = 0; i < params.errorcnt; i++) {
 | |
|       fprintf(stderr, "Error: not known or error at parameter %s.\n", arg[params.error[i]]);
 | |
|     }
 | |
|     return -1;
 | |
|   }
 | |
| 
 | |
|   if (params.online == 0 && params.loadfile == -1) {
 | |
|     printhelp(argc, arg);
 | |
|     fprintf(stderr, "Error: you must specify a loading file.\n");
 | |
|     return -1;
 | |
|   }
 | |
| 
 | |
|   if (params.method != 0 && arg[params.method][0] != 'g' && arg[params.method][0] != 'p' && arg[params.method][0] != 'o' && arg[params.method][0] != 'l' && arg[params.method][0] != 'e' && arg[params.method][0] != 'd') {
 | |
|     printhelp(argc, arg);
 | |
|     fprintf(stderr, "Error: you must choose a calculation method beetween [p]robability, [g]radient, [l]ine search, [o]nline, [e]xpected counts, probability with [d]eterministic nodes.\n");
 | |
|     return -1;
 | |
|   }
 | |
| 
 | |
|   if (params.method != 0 && (arg[params.method][0] == 'g' || arg[params.method][0] == 'p' || arg[params.method][0] == 'l'|| arg[params.method][0] == 'e'|| arg[params.method][0] == 'd') && params.inputfile == -1) {
 | |
|     printhelp(argc, arg);
 | |
|     fprintf(stderr, "Error: an input file is necessary for probability, gradient, line search calculation or expected counts methods.\n");
 | |
|     return -1;
 | |
|   }
 | |
| 
 | |
|   if (params.debug) DEBUGON;
 | |
|   RAPIDLOADON;
 | |
|   SETMAXBUFSIZE(params.maxbufsize);
 | |
| 
 | |
|   signal(SIGINT, termhandler);
 | |
|   if (params.ppid != NULL) {
 | |
|     signal(SIGALRM, pidhandler);
 | |
|     alarm(5);
 | |
|   } else {
 | |
|     signal(SIGALRM, handler);
 | |
|     alarm(params.timeout);
 | |
|   }
 | |
|   
 | |
|   if (params.online) {
 | |
|     MyManager.manager = simpleBDDinit(0);
 | |
|     MyManager.t = HIGH(MyManager.manager);
 | |
|     MyManager.f = LOW(MyManager.manager);
 | |
|     MyManager.varmap = InitNamedVars(1, 0);
 | |
|     bdd = OnlineGenerateBDD(MyManager.manager, &MyManager.varmap);
 | |
|     ivarcnt = GetVarCount(MyManager.manager);
 | |
|   } else {
 | |
|     //fprintf(stderr,"reading file \n");
 | |
|     fileheader = ReadFileHeader(arg[params.loadfile]);
 | |
|     switch(fileheader.filetype) {
 | |
|       case BDDFILE_SCRIPT:
 | |
| 	//	fprintf(stderr," ..... %i \n",fileheader.varcnt);
 | |
|         MyManager.manager = simpleBDDinit(fileheader.varcnt);
 | |
|         MyManager.t = HIGH(MyManager.manager);
 | |
|         MyManager.f = LOW(MyManager.manager);
 | |
|         MyManager.varmap = InitNamedVars(fileheader.varcnt, fileheader.varstart);
 | |
|         if (fileheader.version > 1) {
 | |
|           forest = FileGenerateBDDForest(MyManager.manager, MyManager.varmap, fileheader);
 | |
|           bdd = forest[0];
 | |
|           bakbdd = bdd;
 | |
|         } else {
 | |
|           forest = NULL;
 | |
|           bdd = FileGenerateBDD(MyManager.manager, MyManager.varmap, fileheader);
 | |
|           bakbdd = bdd;
 | |
|         }
 | |
|         ivarcnt = fileheader.varcnt;
 | |
|         break;
 | |
|       case BDDFILE_NODEDUMP:
 | |
|         MyManager.manager = simpleBDDinit(fileheader.varcnt);
 | |
|         MyManager.t = HIGH(MyManager.manager);
 | |
|         MyManager.f = LOW(MyManager.manager);
 | |
|         MyManager.varmap = InitNamedVars(fileheader.varcnt, fileheader.varstart);
 | |
|         bdd = LoadNodeDump(MyManager.manager, MyManager.varmap, fileheader.inputfile);
 | |
|         ivarcnt = fileheader.varcnt;
 | |
|         break;
 | |
|       default:
 | |
|         fprintf(stderr, "Error: not a valid file format to load.\n");
 | |
|         return -1;
 | |
|         break;
 | |
|     }
 | |
|     //    fprintf(stderr,"bdd built\n");
 | |
|   }
 | |
| 
 | |
| 
 | |
|   alarm(0);
 | |
| 
 | |
|   // problem specifics
 | |
| 
 | |
|   if (bdd != NULL) {
 | |
|     ivarcnt = RepairVarcnt(&MyManager.varmap);
 | |
|     code = 0;
 | |
|     if (params.inputfile != -1) {
 | |
|       if (LoadVariableData(MyManager.varmap, arg[params.inputfile]) == -1) return -1;
 | |
|       //     if (!all_loaded(MyManager.varmap, 1)) return -1;
 | |
|       all_loaded_for_deterministic_variables(MyManager.varmap, 1);
 | |
|     }
 | |
|     // impose a predifined order good for debugging
 | |
|     // can be used with a partial number of variables to impose ordering at beggining of BDD
 | |
|     if (params.orderfile != -1) {
 | |
|       ImposeOrder(MyManager.manager, MyManager.varmap, GetVariableOrder(arg[params.orderfile], MyManager.varmap.varcnt));
 | |
|     }
 | |
|     curbdd = 0;
 | |
|     do {
 | |
|       MyManager.his = InitHistory(ivarcnt);
 | |
|       if (params.method != 0) {
 | |
|         switch(arg[params.method][0]) {
 | |
|           case 'g':
 | |
|             for (i = 0; i < MyManager.varmap.varcnt; i++) {
 | |
|               if (MyManager.varmap.vars[i] != NULL) {
 | |
| 
 | |
|                  // check whether this is a continues fact
 | |
|                 if (MyManager.varmap.dynvalue[i] == NULL) {  // nope, regular fact
 | |
|                   varpattern = extractpattern(MyManager.varmap.vars[i]);
 | |
|                   if ((varpattern == NULL) || (!patterncalculated(varpattern, MyManager, i))) {
 | |
|                     tvalue = CalcGradient(MyManager, bdd, i + MyManager.varmap.varstart, varpattern, 0);
 | |
|                     probability = tvalue.probability;
 | |
|                     if (varpattern == NULL) {
 | |
|                       printf("query_gradient(%s,%s,p,%e).\n", arg[params.queryid], MyManager.varmap.vars[i], tvalue.gradient);
 | |
|                     } else {
 | |
|                       varpattern[strlen(varpattern) - 2] = '\0';
 | |
|                       printf("query_gradient(%s,%s,p,%e).\n", arg[params.queryid], varpattern, tvalue.gradient);
 | |
|                     }
 | |
|                     ReInitHistory(MyManager.his, MyManager.varmap.varcnt);
 | |
|                     if (varpattern != NULL) free(varpattern);
 | |
|                   }
 | |
|                 } else { // it is! let's do the Hybrid Problog Magic
 | |
|                   // first for mu
 | |
|                   varpattern = extractpattern(MyManager.varmap.vars[i]);
 | |
|                   if ((varpattern == NULL) || (!patterncalculated(varpattern, MyManager, i))) {
 | |
|                     tvalue = CalcGradient(MyManager, bdd, i + MyManager.varmap.varstart, varpattern, 1);
 | |
|                     probability = tvalue.probability;
 | |
|                     if (varpattern == NULL) {
 | |
|                       printf("query_gradient(%s,%s,mu,%e).\n", arg[params.queryid], MyManager.varmap.vars[i], tvalue.gradient);
 | |
|                     } else {
 | |
|                       varpattern[strlen(varpattern) - 2] = '\0';
 | |
|                       printf("query_gradient(%s,%s,mu,%e).\n", arg[params.queryid], varpattern, tvalue.gradient);
 | |
|                     }
 | |
|                   }
 | |
|                   ReInitHistory(MyManager.his, MyManager.varmap.varcnt);
 | |
|                   if (varpattern != NULL) free(varpattern);
 | |
| 
 | |
|                   // then for sigma
 | |
|                   varpattern = extractpattern(MyManager.varmap.vars[i]);
 | |
|                   if ((varpattern == NULL) || (!patterncalculated(varpattern, MyManager, i))) {
 | |
|                     tvalue = CalcGradient(MyManager, bdd, i + MyManager.varmap.varstart, varpattern, 2);
 | |
|                     probability = tvalue.probability;
 | |
|                     if (varpattern == NULL) {
 | |
|                       printf("query_gradient(%s,%s,sigma,%e).\n", arg[params.queryid], MyManager.varmap.vars[i], tvalue.gradient);
 | |
|                     } else {
 | |
|                       varpattern[strlen(varpattern) - 2] = '\0';
 | |
|                       printf("query_gradient(%s,%s,sigma,%e).\n", arg[params.queryid], varpattern, tvalue.gradient);
 | |
|                     }
 | |
|                   }
 | |
|                   ReInitHistory(MyManager.his, MyManager.varmap.varcnt);
 | |
|                   if (varpattern != NULL) free(varpattern);
 | |
|                 }
 | |
| 
 | |
|               } else {
 | |
|                 fprintf(stderr, "Error: no variable name given for parameter.\n");
 | |
|               }
 | |
|             }
 | |
|             if (probability < 0.0) {
 | |
|               // no nodes, so we have to calculate probability ourself
 | |
|               tvalue = CalcGradient(MyManager, bdd, 0 + MyManager.varmap.varstart, NULL, 0);
 | |
|               probability = tvalue.probability;
 | |
|             }
 | |
|             printf("query_probability(%s,%e).\n", arg[params.queryid], probability);
 | |
|             break;
 | |
|          case 'l':
 | |
|             tvalue = CalcGradient(MyManager, bdd, 0 + MyManager.varmap.varstart, NULL, 0);
 | |
|             probability = tvalue.probability;
 | |
|             printf("query_probability(%s,%e).\n", arg[params.queryid], probability);
 | |
|             break;
 | |
|           case 'e':
 | |
|             //fprintf(stderr,"start calc exp count\n");
 | |
| 	    printf("query_probability(%s,%30.30e).\n", arg[params.queryid],CalcExpectedCounts(&MyManager, bdd,arg[params.queryid],1));
 | |
|             break;
 | |
|           case 'd':
 | |
|             //fprintf(stderr,"start calc exp count\n");
 | |
| 	    printf("query_probability(%s,%30.30e).\n", arg[params.queryid],CalcExpectedCounts(&MyManager, bdd,arg[params.queryid],0));
 | |
|             break;
 | |
|           case 'p':
 | |
| 	    printf("query_probability(%s,%e).\n", arg[params.queryid],  CalcProbability(MyManager, bdd));
 | |
|             break;
 | |
|           case 'o':
 | |
|             onlinetraverse(MyManager.manager, MyManager.varmap, MyManager.his, bdd);
 | |
|             break;
 | |
|           default:
 | |
|             myexpand(MyManager, bdd);
 | |
|             break;
 | |
|         }
 | |
|       } else {
 | |
|         myexpand(MyManager, bdd);
 | |
|       }
 | |
|       if (forest != NULL) {
 | |
|         curbdd++;
 | |
|         bdd = forest[curbdd];
 | |
|       } else {
 | |
|         bdd = NULL;
 | |
|       }
 | |
|       ReInitHistory(MyManager.his, MyManager.varmap.varcnt);
 | |
|     } while(bdd != NULL);
 | |
|     bdd = bakbdd;
 | |
|     if (params.savedfile > -1) SaveNodeDump(MyManager.manager, MyManager.varmap, bdd, arg[params.savedfile]);
 | |
|     if (params.exportfile > -1) simpleNamedBDDtoDot(MyManager.manager, MyManager.varmap, bdd, arg[params.exportfile]);
 | |
|     free(MyManager.his);
 | |
|   }
 | |
|   if (MyManager.manager != NULL) {
 | |
|     KillBDD(MyManager.manager);
 | |
|     free(MyManager.varmap.dvalue);
 | |
|     free(MyManager.varmap.ivalue);
 | |
|     if (MyManager.varmap.dynvalue != NULL) {
 | |
|       for(i = 0; i < MyManager.varmap.varcnt; i++)
 | |
|         if (MyManager.varmap.dynvalue[i] != NULL) {
 | |
|           free(MyManager.varmap.dynvalue[i]);
 | |
|         }
 | |
|       free(MyManager.varmap.dynvalue);
 | |
|     }
 | |
|     for (i = 0; i < MyManager.varmap.varcnt; i++)
 | |
|       free(MyManager.varmap.vars[i]);
 | |
|     free(MyManager.varmap.vars);
 | |
|   }
 | |
|   if (params.error != NULL) free(params.error);
 | |
| 
 | |
|   return code;
 | |
| 
 | |
| }
 | |
| 
 | |
| /* Shell Parameters handling */
 | |
| 
 | |
| int argtype(const char *arg) {
 | |
|   if (strcmp(arg, "-l") == 0 || strcmp(arg, "--load") == 0) return 0;
 | |
|   if (strcmp(arg, "-e") == 0 || strcmp(arg, "--export") == 0) return 2;
 | |
|   if (strcmp(arg, "-m") == 0 || strcmp(arg, "--method") == 0) return 3;
 | |
|   if (strcmp(arg, "-i") == 0 || strcmp(arg, "--input") == 0) return 4;
 | |
|   if (strcmp(arg, "-h") == 0 || strcmp(arg, "--help") == 0) return 5;
 | |
|   if (strcmp(arg, "-d") == 0 || strcmp(arg, "--debug") == 0) return 6;
 | |
|   if (strcmp(arg, "-id") == 0 || strcmp(arg, "--queryid") == 0) return 7;
 | |
|   if (strcmp(arg, "-t") == 0 || strcmp(arg, "--timeout") == 0) return 8;
 | |
|   if (strcmp(arg, "-sd") == 0 || strcmp(arg, "--savedump") == 0) return 9;
 | |
|   if (strcmp(arg, "-sl") == 0 || strcmp(arg, "--slope") == 0) return 10;
 | |
|   if (strcmp(arg, "-o") == 0 || strcmp(arg, "--online") == 0) return 11;
 | |
|   if (strcmp(arg, "-bs") == 0 || strcmp(arg, "--bufsize") == 0) return 12;
 | |
|   if (strcmp(arg, "-pid") == 0 || strcmp(arg, "--pid") == 0) return 13;
 | |
|   if (strcmp(arg, "-ord") == 0 || strcmp(arg, "--order") == 0) return 14;
 | |
|   
 | |
|   return -1;
 | |
| }
 | |
| 
 | |
| void printhelp(int argc, char **arg) {
 | |
|   fprintf(stderr, "\n\nProbLogBDD Tool Version: %s\n\n", VERSION);
 | |
|   fprintf(stderr, "SimpleCUDD library (www.cs.kuleuven.be/~theo/tools/simplecudd.html)\n");
 | |
|   fprintf(stderr, "SimpleCUDD was developed at Katholieke Universiteit Leuven(www.kuleuven.be)\n");
 | |
|   fprintf(stderr, "Copyright Katholieke Universiteit Leuven 2008\n");
 | |
|   fprintf(stderr, "Authors: Theofrastos Mantadelis, Angelika Kimmig, Bernd Gutmann\n");
 | |
|   fprintf(stderr, "This package falls under the: Artistic License 2.0\n");
 | |
|   fprintf(stderr, "\nUsage: %s -l [filename] -i [filename] -o (-s(d) [filename] -e [filename] -m [method] -id [queryid] -sl [double]) (-t [seconds] -d -h)\n", arg[0]);
 | |
|   fprintf(stderr, "Generates and traverses a BDD\nMandatory parameters:\n");
 | |
|   fprintf(stderr, "\t-l [filename]\t->\tfilename to load supports two formats:\n\t\t\t\t\t\t1. script with generation instructions\n\t\t\t\t\t\t2. node dump saved file\n");
 | |
|   fprintf(stderr, "\t-i [filename]\t->\tfilename to input problem specifics (mandatory with file formats 1, 2)\n");
 | |
|   fprintf(stderr, "\t-o\t\t->\tgenerates the BDD in online mode instead from a file can be used instead of -l\n");
 | |
|   fprintf(stderr, "Optional parameters:\n");
 | |
|   fprintf(stderr, "\t-sd [filename]\t->\tfilename to save generated BDD in node dump format (fast loading, traverse valid only)\n");
 | |
|   fprintf(stderr, "\t-e [filename]\t->\tfilename to export generated BDD in dot format\n");
 | |
|   fprintf(stderr, "\t-m [method]\t->\tthe calculation method to be used: none(default), [p]robability, [g]radient, [l]ine search, [o]nline, [e]xpexted counts, prob. with [d]eterministic nodes\n");
 | |
|   fprintf(stderr, "\t-id [queryid]\t->\tthe queries identity name (used by gradient) default: %s\n", arg[0]);
 | |
|   fprintf(stderr, "\t-sl [double]\t->\tthe sigmoid slope (used by gradient) default: 1.0\n");
 | |
|   fprintf(stderr, "Extra parameters:\n");
 | |
|   fprintf(stderr, "\t-t [seconds]\t->\tthe seconds (int) for BDD generation timeout default 0 = no timeout\n");
 | |
|   fprintf(stderr, "\t-pid [pid]\t->\ta process id (int) to check for termination default 0 = no process to check\n");
 | |
|   fprintf(stderr, "\t-bs [bytes]\t->\tthe bytes (int) to use as a maximum buffer size to read files default 0 = no max\n");
 | |
|   fprintf(stderr, "\t-ord [filename]\t->\tUse the [filename] to define a specific BDD variable order\n");
 | |
|   fprintf(stderr, "\t-d\t\t->\tRun in debug mode (gives extra messages in stderr)\n");
 | |
|   fprintf(stderr, "\t-h\t\t->\tHelp (displays this message)\n");
 | |
|   fprintf(stderr, "Extra notes:\nSupports a forest of BDDs in one shared BDD.\nSelected computational methods will be applied to each BDD seperately.\nFile operations will be applied only to the first BDD.\n");
 | |
|   fprintf(stderr, "\nExample: %s -l testbdd -i input.txt -m g -id testbdd\n", arg[0]);
 | |
| }
 | |
| 
 | |
| parameters loadparam(int argc, char **arg) {
 | |
|   int i;
 | |
|   parameters params;
 | |
|   params.loadfile = -1;
 | |
|   params.savedfile = -1;
 | |
|   params.exportfile = -1;
 | |
|   params.method = 0;
 | |
|   params.inputfile = -1;
 | |
|   params.debug = 0;
 | |
|   params.errorcnt = 0;
 | |
|   params.queryid = 0;
 | |
|   params.timeout = 0;
 | |
|   params.sigmoid_slope = 1.0;
 | |
|   params.online = 0;
 | |
|   params.maxbufsize = 0;
 | |
|   params.ppid = NULL;
 | |
|   params.orderfile = -1;
 | |
|   params.error = (int *) malloc(argc * sizeof(int));
 | |
|   for (i = 1; i < argc; i++) {
 | |
|     switch(argtype(arg[i])) {
 | |
|       case 0:
 | |
|         if (argc > i + 1) {
 | |
|           i++;
 | |
|           params.loadfile = i;
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       case 2:
 | |
|         if (argc > i + 1) {
 | |
|           i++;
 | |
|           params.exportfile = i;
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       case 3:
 | |
|         if (argc > i + 1) {
 | |
|           i++;
 | |
|           params.method = i;
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       case 4:
 | |
|         if (argc > i + 1) {
 | |
|           i++;
 | |
|           params.inputfile = i;
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       case 5:
 | |
|         printhelp(argc, arg);
 | |
|         break;
 | |
|       case 6:
 | |
|         params.debug = 1;
 | |
|         break;
 | |
|       case 7:
 | |
|         if (argc > i + 1) {
 | |
|           i++;
 | |
|           params.queryid = i;
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       case 8:
 | |
|         if ((argc > i + 1) && (IsPosNumber(arg[i + 1]))) {
 | |
|           i++;
 | |
|           params.timeout = atoi(arg[i]);
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       case 9:
 | |
|         if (argc > i + 1) {
 | |
|           i++;
 | |
|           params.savedfile = i;
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       case 10:
 | |
|         if ((argc > i + 1) && (IsRealNumber(arg[i + 1]))) {
 | |
|           i++;
 | |
|           params.sigmoid_slope = atof(arg[i]);
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       case 11:
 | |
|         params.online = 1;
 | |
|         break;
 | |
|       case 12:
 | |
|         if ((argc > i + 1) && (IsPosNumber(arg[i + 1]))) {
 | |
|           i++;
 | |
|           params.maxbufsize = atoi(arg[i]);
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       case 13:
 | |
|         if ((argc > i + 1) && (IsPosNumber(arg[i + 1]))) {
 | |
|           i++;
 | |
|           params.ppid = (char *) malloc(sizeof(char) * (strlen(arg[i]) + 1));
 | |
|           strcpy(params.ppid, arg[i]);
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       case 14:
 | |
|         if (argc > i + 1) {
 | |
|           i++;
 | |
|           params.orderfile = i;
 | |
|         } else {
 | |
|           params.error[params.errorcnt] = i;
 | |
|           params.errorcnt++;
 | |
|         }
 | |
|         break;
 | |
|       default:
 | |
|         params.error[params.errorcnt] = i;
 | |
|         params.errorcnt++;
 | |
|         break;
 | |
|     }
 | |
|   }
 | |
|   return params;
 | |
| }
 | |
| 
 | |
| /* Error Handlers */
 | |
| 
 | |
| void handler(int num) {
 | |
|   fprintf(stderr, "Error: Timeout %i exceeded.\n", params.timeout);
 | |
|   exit(-1);
 | |
| }
 | |
| 
 | |
| void pidhandler(int num) {
 | |
|   char *s;
 | |
|   if (params.timeout > 0) {
 | |
|     params.timeout -= 5;
 | |
|     if (params.timeout <= 0) {
 | |
|       fprintf(stderr, "Error: Timeout exceeded.\n");
 | |
|       exit(-1);
 | |
|     }
 | |
|   }
 | |
|   s = (char *) malloc(sizeof(char) * (19 + strlen(params.ppid)));
 | |
|   strcpy(s, "ps "); strcat(s, params.ppid); strcat(s, " >/dev/null");
 | |
|   if (system(s) != 0) exit(4);
 | |
|   signal(SIGALRM, pidhandler);
 | |
|   alarm(5);
 | |
|   free(s);
 | |
| }
 | |
| 
 | |
| void termhandler(int num) {
 | |
|   exit(3);
 | |
| }
 | |
| 
 | |
| /* Debugging traverse function */
 | |
| 
 | |
| void myexpand(extmanager MyManager, DdNode *Current) {
 | |
|   DdNode *h, *l;
 | |
|   hisnode *Found;
 | |
|   char *curnode;
 | |
|   curnode = GetNodeVarNameDisp(MyManager.manager, MyManager.varmap, Current);
 | |
|   printf("%s\n", curnode);
 | |
|   if ((Current != MyManager.t) && (Current != MyManager.f) &&
 | |
|       ((Found = GetNode(MyManager.his, MyManager.varmap.varstart, Current)) == NULL)) {
 | |
|     l = LowNodeOf(MyManager.manager, Current);
 | |
|     h = HighNodeOf(MyManager.manager, Current);
 | |
|     printf("l(%s)->", curnode);
 | |
|     myexpand(MyManager, l);
 | |
|     printf("h(%s)->", curnode);
 | |
|     myexpand(MyManager, h);
 | |
|     AddNode(MyManager.his, MyManager.varmap.varstart, Current, 0.0, 0, NULL);
 | |
|   }
 | |
| }
 | |
| 
 | |
| /* Angelicas Algorithm */
 | |
| 
 | |
| double CalcProbability(extmanager MyManager, DdNode *Current) {
 | |
|   DdNode *h, *l;
 | |
|   hisnode *Found;
 | |
|   char *curnode, *dynvalue;
 | |
|   double lvalue, hvalue, tvalue;
 | |
|   density_integral dynvalue_parsed;
 | |
|   if (params.debug) {
 | |
|     curnode = GetNodeVarNameDisp(MyManager.manager, MyManager.varmap, Current);
 | |
|     fprintf(stderr, "%s\n", curnode);
 | |
|   }
 | |
|   if (Current == MyManager.t) return 1.0;
 | |
|   if (Current == MyManager.f) return 0.0;
 | |
| 
 | |
|   if ((Found = GetNode(MyManager.his, MyManager.varmap.varstart, Current)) != NULL) return Found->dvalue;
 | |
|   l = LowNodeOf(MyManager.manager, Current);
 | |
|   h = HighNodeOf(MyManager.manager, Current);
 | |
|   if (params.debug) fprintf(stderr, "l(%s)->", curnode);
 | |
|   lvalue = CalcProbability(MyManager, l);
 | |
|   if (params.debug) fprintf(stderr, "h(%s)->", curnode);
 | |
|   hvalue = CalcProbability(MyManager, h);
 | |
| 
 | |
| 
 | |
|   tvalue = MyManager.varmap.dvalue[GetIndex(Current) - MyManager.varmap.varstart];
 | |
| 
 | |
|   tvalue = tvalue * hvalue + lvalue * (1.0 - tvalue);
 | |
|   AddNode(MyManager.his, MyManager.varmap.varstart, Current, tvalue, 0, NULL);
 | |
|   return tvalue;
 | |
| }
 | |
| 
 | |
| 
 | |
| double CalcExpectedCounts(extmanager * MyManager, DdNode *Current, char *query_id, int calcdown_needed) {
 | |
| 
 | |
|   // fprintf(stderr,"%%calcing up\n");
 | |
|   double ret = CalcExpectedCountsUp(MyManager, Current, query_id);  
 | |
|   //   fprintf(stderr,"%%result is %e\n",ret);
 | |
|   // fprintf(stderr,"%%calcing down\n");
 | |
|   
 | |
|   if (calcdown_needed != 0) {
 | |
|     double retd=CalcExpectedCountsDown(MyManager,Current, query_id);
 | |
|   }
 | |
| /*   if(1 != retd){ */
 | |
| /*     fprintf(stderr,"down %e != up %e/%e\n",ret,retd,ret); */
 | |
| /*     exit(1); */
 | |
| /*   } */
 | |
|   return ret;
 | |
| }
 | |
| 
 | |
| 
 | |
| /* ComparisonFunction compare_nodes (  extmanager MyManager) */
 | |
| /* { */
 | |
| /*   //fprintf(stderr,"creating comparator for  %p\n",MyManager); */
 | |
| 
 | |
| /*   int comparator(void *av, void *bv){ */
 | |
| /*     fprintf(stderr,"========================> %p \n",MyManager); */
 | |
| /*     fprintf(stderr,"xxxxx \n"); */
 | |
| /*     DdNode* a = (DdNode*)av; */
 | |
| /*     DdNode* b = (DdNode*)bv; */
 | |
| /*     extmanager MyManager; */
 | |
| /*     int aindex,bindex, aperm, bperm; */
 | |
| /*     aindex=GetIndex(a); */
 | |
| /*     bindex=GetIndex(b); */
 | |
| /*     aperm=Cudd_IsConstant(a) ? CUDD_CONST_INDEX : Cudd_ReadPerm(MyManager.manager,aindex); */
 | |
| /*     Cudd_ReadPerm(MyManager.manager,bindex); */
 | |
| /*     bperm=Cudd_IsConstant(b) ? CUDD_CONST_INDEX : Cudd_ReadPerm(MyManager.manager,bindex); */
 | |
| /*     int temp = aperm-bperm; */
 | |
| /* //-Cudd_ReadPerm(MyManager.manager,(*b).index);//-Cudd_ReadPerm(MyManager,b);// - Cudd_ReadPerm(b); */
 | |
| /*     //fprintf(stderr,"comparing3 %p %p %p\n",a,b,MyManager); */
 | |
| /*     //  return -1; */
 | |
| /*     if (temp < 0) */
 | |
| /*       return 1; */
 | |
| /*     else if (temp > 0) */
 | |
| /*       return -1; */
 | |
| /*     //    else //never return zero otherwise one is pruned away, or(?) */
 | |
| /*     //      return 0; */
 | |
| /*   } */
 | |
| 
 | |
| /*   //  return (a<b) ? 1 : -1; */
 | |
| /*   return *comparator; */
 | |
| /* } */
 | |
| 
 | |
| #define NODE_VALUE 1001
 | |
| #define LOG_EXPECTED  0
 | |
| 
 | |
| 
 | |
| 
 | |
| void PrintNodeQueue(Queue q , extmanager MyManager){
 | |
| 
 | |
|     QueueIterator qiter = QueueIteratorNew(q, 1);
 | |
|     fprintf(stderr,"Queue %p is [", *q);
 | |
| 
 | |
|     while (qiter->currentItem != NULL) {
 | |
|       DdNode* val = (DdNode*) qiter->currentItem->element;
 | |
|       QueueIteratorAdvance(qiter);
 | |
|       fprintf(stderr," %s %s", GetNodeVarNameDisp(MyManager.manager, MyManager.varmap, val),
 | |
| 	      (qiter->currentItem !=NULL)?",":"]\n");
 | |
|     }
 | |
| 
 | |
| }
 | |
|  /** also nesting in CalcExpected seems to not work (must be here nested only valid within function frame)*/
 | |
| /* will be changed at later stage */
 | |
| static extmanager * ineedtostorethatsomehow;
 | |
| int comparator(void *av, void *bv){
 | |
|     int ret;
 | |
|     DdNode* a = (DdNode*)av;
 | |
|     DdNode* b = (DdNode*)bv;
 | |
|     int aindex,bindex, aperm, bperm;
 | |
|     aindex=GetIndex(a);
 | |
|     bindex=GetIndex(b);
 | |
|     aperm=Cudd_IsConstant(a) ? CUDD_CONST_INDEX : Cudd_ReadPerm(ineedtostorethatsomehow->manager,aindex);
 | |
|     Cudd_ReadPerm(ineedtostorethatsomehow->manager,bindex);
 | |
|     bperm=Cudd_IsConstant(b) ? CUDD_CONST_INDEX : Cudd_ReadPerm(ineedtostorethatsomehow->manager,bindex);
 | |
|     int temp = -aperm+bperm;
 | |
|     if (temp < 0)
 | |
|       ret= 1;
 | |
|     else if (temp > 0)
 | |
|       ret= -1;    
 | |
|     //    else //never return zero otherwise one is pruned away, or(?)
 | |
|     //      return 0;
 | |
|     if(LOG_EXPECTED){
 | |
|       fprintf(stderr,"perm(%s,%i)=%i perm(%s,%i)=%i => %i\n",GetNodeVarNameDisp(ineedtostorethatsomehow->manager, ineedtostorethatsomehow->varmap, a),
 | |
| 	      aindex,
 | |
| 	      aperm,
 | |
| 	      GetNodeVarNameDisp(ineedtostorethatsomehow->manager, ineedtostorethatsomehow->varmap, b),
 | |
| 	      bindex,bperm,
 | |
| 			     ret);}
 | |
| 
 | |
|     return ret;
 | |
| }
 | |
| 
 | |
| /** output information for skipped nodes **/
 | |
| skip_nodes(extmanager * MyManager, double (*counts)[] , DdNode* node, DdNode* l,double dprob, char *query_id){
 | |
|   int skipcnt;
 | |
|   skipcnt = Cudd_ReadPerm(MyManager->manager,GetIndex(node))+1;
 | |
|   if(LOG_EXPECTED){fprintf(stderr,">> skipper >> %s=%i@%i of %i -> %i@%i %i\n",
 | |
| 			   MyManager->varmap.dynvalue[GetIndex(node) - MyManager->varmap.varstart],
 | |
| 			   GetIndex(node),
 | |
| 			   Cudd_ReadPerm(MyManager->manager,GetIndex(node)),
 | |
| 			   Cudd_ReadSize( MyManager->manager),
 | |
| 			   GetIndex(l),
 | |
| 			   Cudd_ReadPerm(MyManager->manager,GetIndex(l)),
 | |
| 			   Cudd_IsConstant(l)
 | |
| 			   );
 | |
|   }
 | |
|   return skip_nodes_cnt( MyManager, counts,  skipcnt,  l, dprob, query_id);
 | |
| 
 | |
| }
 | |
| skip_nodes_cnt(extmanager * MyManager, double (*counts)[] , int skipcnt, DdNode* l,double dprob, char *query_id){
 | |
|   if(LOG_EXPECTED) fprintf(stderr,"====================\n");
 | |
|   double p;
 | |
|   int ivalue;
 | |
|   //  fprintf(stderr, " skip (:%i) \n",__LINE__);
 | |
|   while(Cudd_IsConstant(l) ?
 | |
| 	skipcnt < Cudd_ReadSize( MyManager->manager)  // the terminals/leafs/constants will be ignored
 | |
| 	:
 | |
| 	skipcnt < Cudd_ReadPerm(MyManager->manager,GetIndex(l) )
 | |
| 	){
 | |
|     skipcnt++;
 | |
|     if(LOG_EXPECTED){    fprintf(stderr,"skipcnt %i\n",skipcnt-1);}
 | |
|     int idx=Cudd_ReadInvPerm(MyManager->manager,skipcnt-1);
 | |
|     if(LOG_EXPECTED){fprintf(stderr,"index %i %i\n",idx,MyManager->varmap.varstart);}
 | |
|     //fprintf(stdout,"%i %s.\n",skipcnt,MyManager->varmap.dynvalue[GetIndex(node) - MyManager->varmap.varstart]);
 | |
|     if(LOG_EXPECTED){fprintf(stderr,"Node skipped level %i index: %i name: %s (dprob is %e)\n",
 | |
| 			     skipcnt,idx,
 | |
|                              MyManager->varmap.vars[idx - MyManager->varmap.varstart],
 | |
| 			     dprob);}
 | |
|     //notiz
 | |
|     ivalue = MyManager->varmap.ivalue[idx - MyManager->varmap.varstart];
 | |
|     //+ new{
 | |
|     //double tvalue; // probability of prob fact corresp to node
 | |
|     //tvalue = MyManager->varmap.dvalue[idx - MyManager->varmap.varstart];
 | |
|     //}
 | |
|     if(ivalue==1){
 | |
|       p=dprob*MyManager->varmap.dvalue[idx - MyManager->varmap.varstart];
 | |
|       //+ new{
 | |
|       //p=dprob*MyManager->varmap.dvalue[idx - MyManager->varmap.varstart] *tvalue;
 | |
|       //}
 | |
|       if(p>0){// probability is zero, don't follow this branch
 | |
| 	(*counts)[idx - MyManager->varmap.varstart]+=p;
 | |
| 	//	fprintf(stdout,"oec(%s,%s,%e). %%2\n",query_id,MyManager->varmap.vars[idx - MyManager->varmap.varstart],p);
 | |
| 	if(LOG_EXPECTED) fprintf(stderr,"ec -> %s,%s,%e . %%2_1\n",query_id,MyManager->varmap.vars[idx - MyManager->varmap.varstart],p);
 | |
|       }else{
 | |
| 	if(LOG_EXPECTED){fprintf(stdout,"%% ec(%s,%s,%30.30e). %%2_2\n",query_id,MyManager->varmap.vars[idx - MyManager->varmap.varstart],p);}
 | |
|       }
 | |
|     }
 | |
|   }
 | |
|   //  fprintf(stderr, " skip %i \n",__LINE__);
 | |
|   if(LOG_EXPECTED){fprintf(stderr,"skipped\n");}
 | |
| }
 | |
| 
 | |
| 
 | |
| double CalcExpectedCountsDown(extmanager * MyManager, DdNode *Current, char *query_id) {
 | |
|   ineedtostorethatsomehow=MyManager;
 | |
|   Queue q = QueueNew();
 | |
|   //fprintf(stderr", =====> queue is: %p \n",q);
 | |
|   int i;
 | |
|   char *curnode, *curh, *curl,*dynvalue;
 | |
|   DdNode *h, *l, *node;
 | |
|   ComparisonFunction fun;
 | |
|   hisnode *Found,*lfound, *hfound;
 | |
|   double dprob; //downward probability of current node
 | |
|   double tvalue; // probability of prob fact corresp to node
 | |
|   int ivalue;
 | |
|   double retval; //last value of true
 | |
| 
 | |
|   double counts[MyManager->varmap.varcnt] ;
 | |
|   double (*pcnt)[MyManager->varmap.varcnt];
 | |
|   pcnt = &counts;
 | |
|   for( i = 0 ;i< MyManager->varmap.varcnt ; i++){
 | |
|     (*pcnt)[i]=0;
 | |
|   }
 | |
|   // skip everything before the first node:
 | |
|   skip_nodes_cnt(MyManager,pcnt,0,Current,1,query_id);
 | |
| 
 | |
| 
 | |
|   fun = *comparator;
 | |
|   if(LOG_EXPECTED){fprintf(stderr," ##############################\n");}
 | |
|   if(LOG_EXPECTED){  fprintf(stderr," ##############################\n fun is %p\n",fun);}
 | |
|   if(!Cudd_IsConstant(Current)){
 | |
|       QueuePutOnPriority(q, Current, NODE_VALUE,fun);
 | |
|       Found = GetNode(MyManager->his, MyManager->varmap.varstart, Current);
 | |
|       (*Found).dvalue2=1.0/((*Found).dvalue);
 | |
|       dynvalue = (*Found).dynvalue;
 | |
|   }
 | |
|   Current= NULL; // not used anymore or should not be
 | |
|   retval=0;
 | |
| 
 | |
|   while(QueueSize(q)>0){
 | |
|     if(LOG_EXPECTED){fprintf(stderr,"\n");}
 | |
|     if(LOG_EXPECTED){PrintNodeQueue(q,*MyManager);}
 | |
|     node=QueueGet(q);
 | |
|     curnode = GetNodeVarNameDisp(MyManager->manager, MyManager->varmap, node);
 | |
|     int level = Cudd_ReadPerm(MyManager->manager,GetIndex(node));
 | |
|     if(!Cudd_IsConstant(node)){
 | |
|     tvalue = MyManager->varmap.dvalue[GetIndex(node) - MyManager->varmap.varstart];
 | |
|     ivalue = MyManager->varmap.ivalue[GetIndex(node) - MyManager->varmap.varstart];
 | |
|     dynvalue = MyManager->varmap.vars[GetIndex(node) - MyManager->varmap.varstart];
 | |
|     Found = GetNode(MyManager->his, MyManager->varmap.varstart, node);
 | |
|     dprob=(*Found).dvalue2;
 | |
|     l = LowNodeOf(MyManager->manager, node);
 | |
|     h = HighNodeOf(MyManager->manager, node);
 | |
|     lfound = GetNode(MyManager->his, MyManager->varmap.varstart, l);
 | |
|     hfound = GetNode(MyManager->his, MyManager->varmap.varstart, h) ;
 | |
|     curh = GetNodeVarNameDisp(MyManager->manager, MyManager->varmap, h);
 | |
|     curl = GetNodeVarNameDisp(MyManager->manager, MyManager->varmap, l);
 | |
|     
 | |
|       if(LOG_EXPECTED){fprintf(stderr, "%s (%i)-->  %s %s\n", curnode,(*node).index,curh,curl);}
 | |
|       /** low node */
 | |
|       if((*lfound).dvalue2<-0.1){ //only if not seen before == dvalue2=0 (almost) otherwise requing does not harm
 | |
| 	if(LOG_EXPECTED){fprintf(stderr,"queueing l(%s)=%s \n",curnode,curl);}
 | |
| 	QueuePutOnPriority(q, l, NODE_VALUE,fun);
 | |
| 	(*lfound).dvalue2=0;
 | |
|       }
 | |
|       ((*lfound).dvalue2)=((*lfound).dvalue2)+(ivalue==0? dprob : dprob*(1-tvalue));
 | |
|       if(LOG_EXPECTED){fprintf(stderr, "l(%s)=%s %e \n", curnode,curl,(*lfound).dvalue2);}
 | |
|       if(LOG_EXPECTED){fprintf(stderr, "l(%s)=%s %e %e %e\n", curnode,curl,(*lfound).dvalue2,tvalue,dprob);}
 | |
| 
 | |
|       /** high node */
 | |
|       if((*hfound).dvalue2<-0.1){ //only if not seen before == dvalue2=0 (almost) otherwise requing does not harm
 | |
| 	fun = *comparator;
 | |
| 	(*fun)(l,l);
 | |
| 	if(LOG_EXPECTED){
 | |
| 	  PrintNodeQueue(q,*MyManager);
 | |
| 	  fprintf(stderr,"-> %p\n",h);
 | |
| 	}
 | |
| 	QueuePutOnPriority(q, h, NODE_VALUE,fun);	
 | |
| 	(*hfound).dvalue2=0;
 | |
|       }
 | |
|       (*hfound).dvalue2=(*hfound).dvalue2+(ivalue==0? dprob : (dprob*(tvalue)));
 | |
|       if(LOG_EXPECTED){fprintf(stderr, "h(%s)=%s %e %e %e\n", curnode,curh,(*hfound).dvalue2,tvalue,dprob);}
 | |
|       /** output expected counts current node */     
 | |
|       if(ivalue==1){
 | |
| 	(*pcnt)[GetIndex(node) - MyManager->varmap.varstart]+=dprob * tvalue * (*hfound).dvalue;
 | |
| 	//fprintf(stdout,"oec(%s,%s,%e). %% 1_1\n",query_id,dynvalue,dprob * tvalue * (*hfound).dvalue);
 | |
| 	if(LOG_EXPECTED) fprintf(stderr,"ec -> %s,%s,%e . %% 1_1\n",query_id,dynvalue,dprob * tvalue * (*hfound).dvalue);
 | |
|       }else{
 | |
| 	(*pcnt)[GetIndex(node) - MyManager->varmap.varstart]+=dprob * tvalue * (*hfound).dvalue;
 | |
| 	if(LOG_EXPECTED) fprintf(stderr,"ec -> %s,%s,%e . %% 1_2\n",query_id,dynvalue,dprob * tvalue * (*hfound).dvalue);
 | |
|       }
 | |
|       /** output expected counts of skipped nodes for low branch*/
 | |
|       skip_nodes(MyManager,pcnt,node,l,dprob*((ivalue==0)?1:(1-tvalue))*(*lfound).dvalue,query_id);
 | |
|       skip_nodes(MyManager,pcnt,node,h,dprob*((ivalue==0)?1:(tvalue))*(*hfound).dvalue,query_id);
 | |
|     }else{
 | |
|       if(LOG_EXPECTED){fprintf(stderr,"here: retval %s %e=>%e\n",curnode,retval,(*Found).dvalue2);}
 | |
|       if(node==(MyManager->t)){
 | |
| 	if(LOG_EXPECTED){fprintf(stderr,"updating retval %e=>%e\n",retval,(*Found).dvalue2);}
 | |
| 	retval=(*Found).dvalue2;
 | |
|       }
 | |
| 
 | |
|     }
 | |
|   }
 | |
|   for( i = 0 ;i< MyManager->varmap.varcnt ; i++){
 | |
|     ivalue = MyManager->varmap.ivalue[i];
 | |
|     /* fprintf(stderr,"Node  level %i index: %i name: %s (dprob is %e)\n", */
 | |
|     /* 	    i,idx, */
 | |
|     /* 	    MyManager->varmap.vars[idx - MyManager->varmap.varstart], */
 | |
|     /* 	    dprob); */
 | |
|     //fprintf(stderr,"Node idx: %i level: %i \n",i,Cudd_ReadPerm(MyManager->manager,i));
 | |
|     if(ivalue==0){
 | |
|       fprintf(stdout,"%% det: ec(%s,%s,%30.30e).\n",query_id,MyManager->varmap.vars[i],(counts)[i]);
 | |
|     }else{
 | |
|       fprintf(stdout,"ec(%s,%s,%30.30e).\n",query_id,MyManager->varmap.vars[i],(counts)[i]);
 | |
| 
 | |
|     }
 | |
|   }
 | |
|   // free(counts);
 | |
|   if(LOG_EXPECTED){
 | |
| fprintf(stderr,"retval is %e\n",retval);
 | |
| }
 | |
|   return retval;
 | |
| }
 | |
| 
 | |
| double CalcExpectedCountsUp(extmanager * MyManager, DdNode *Current, char *query_id) {
 | |
|   //  fprintf(stderr,"--------------------- the manager 2 %p \n",&MyManager);
 | |
| 
 | |
|   DdNode *h, *l;
 | |
|   hisnode *Found;
 | |
|   char *curnode;
 | |
|   double lvalue, hvalue, tvalue;
 | |
|   //  tvalue=0.0;
 | |
|   int ivalue;
 | |
|   if (params.debug) {
 | |
|     curnode = GetNodeVarNameDisp(MyManager->manager, MyManager->varmap, Current);
 | |
|     fprintf(stderr, "%s\n", curnode);
 | |
|   }
 | |
| 
 | |
|   if (Current == MyManager->t){
 | |
|     //    if ((Found = GetNode(MyManager->his, MyManager->varmap.varstart, Current)) == NULL) {
 | |
|     //    fprintf(stderr,"adding true \n");
 | |
|     AddNode(MyManager->his, MyManager->varmap.varstart, MyManager->t, 1, 0, NULL);//}//needed in down
 | |
|     return 1.0;
 | |
|   }
 | |
|   if (Current == MyManager->f){ 
 | |
|     //    fprintf(stderr,"adding false \n");
 | |
|     //    if ((Found = GetNode(MyManager->his, MyManager->varmap.varstart, Current)) == NULL) {
 | |
|     AddNode(MyManager->his, MyManager->varmap.varstart, MyManager->f, 0, 0, NULL);//}//needed in down
 | |
|     return 0.0;
 | |
|   }
 | |
| 
 | |
|   if ((Found = GetNode(MyManager->his, MyManager->varmap.varstart, Current)) != NULL) return Found->dvalue;
 | |
|   l = LowNodeOf(MyManager->manager, Current);
 | |
|   h = HighNodeOf(MyManager->manager, Current);
 | |
|   if (params.debug) fprintf(stderr, "l(%s)->", curnode);
 | |
|   lvalue = CalcExpectedCountsUp(MyManager, l,query_id);
 | |
|   if (params.debug) fprintf(stderr, "h(%s)->", curnode);
 | |
|   hvalue = CalcExpectedCountsUp(MyManager, h,query_id);
 | |
| 
 | |
| 
 | |
|   tvalue = MyManager->varmap.dvalue[GetIndex(Current) - MyManager->varmap.varstart];
 | |
|   //notiz
 | |
|   ivalue = MyManager->varmap.ivalue[GetIndex(Current) - MyManager->varmap.varstart];
 | |
| 
 | |
|   if(ivalue == 1){
 | |
|     tvalue = tvalue * hvalue + lvalue * (1.0 - tvalue);
 | |
|   }else if (ivalue == 0){
 | |
|     tvalue = hvalue + lvalue ;
 | |
|   }
 | |
|   //  fprintf(stderr," ---> %e \n",tvalue);
 | |
|   AddNode(MyManager->his, MyManager->varmap.varstart, Current, tvalue, 0, NULL);
 | |
|   return tvalue;
 | |
| }
 | |
| 
 | |
| 
 | |
| /* Bernds Algorithm */
 | |
| // type=0  regular probabilistic fact
 | |
| // type=1  derive gradient for mu
 | |
| // type=2  derive gradient for sigma
 | |
| gradientpair CalcGradient(extmanager MyManager, DdNode *Current, int TargetVar, char *TargetPattern, int type) {
 | |
|   DdNode *h, *l;
 | |
|   hisnode *Found;
 | |
|   char *curnode, *dynvalue;
 | |
|   gradientpair lowvalue, highvalue, tvalue;
 | |
|   double this_probability;
 | |
|   double *gradient;
 | |
|   density_integral dynvalue_parsed;
 | |
| 
 | |
|   if (params.debug) {
 | |
|     curnode = GetNodeVarNameDisp(MyManager.manager, MyManager.varmap, Current);
 | |
|     fprintf(stderr, "%s\n", curnode);
 | |
|   }
 | |
|   // base cases
 | |
|   if (Current == MyManager.t) {
 | |
|     tvalue.probability = 1.0;
 | |
|     tvalue.gradient = 0.0;
 | |
|     return tvalue;
 | |
|   }
 | |
|   if (Current == MyManager.f) {
 | |
|     tvalue.probability = 0.0;
 | |
|     tvalue.gradient = 0.0;
 | |
|     return tvalue;
 | |
|   }
 | |
|   //node is in cache
 | |
|   if ((Found = GetNode(MyManager.his, MyManager.varmap.varstart, Current)) != NULL) {
 | |
|     tvalue.probability = Found->dvalue;
 | |
|     tvalue.gradient = *((double *) Found->dynvalue);
 | |
|     return tvalue;
 | |
|   }
 | |
| 
 | |
|   //inductive case
 | |
|   l = LowNodeOf(MyManager.manager, Current);
 | |
|   h = HighNodeOf(MyManager.manager, Current);
 | |
|   if (params.debug) fprintf(stderr, "l(%s)->", curnode);
 | |
|   lowvalue = CalcGradient(MyManager, l, TargetVar, TargetPattern,type);
 | |
|   if (params.debug) fprintf(stderr, "h(%s)->", curnode);
 | |
|   highvalue = CalcGradient(MyManager, h, TargetVar, TargetPattern,type);
 | |
|   dynvalue = (char*) MyManager.varmap.dynvalue[GetIndex(Current) - MyManager.varmap.varstart];
 | |
|   if (dynvalue == NULL) { // no dynvalue, it's a regular probabilistic fact
 | |
|     this_probability = sigmoid(MyManager.varmap.dvalue[GetIndex(Current) - MyManager.varmap.varstart], params.sigmoid_slope);
 | |
|   } else { // there is a dynvalue, it's a continuous fact! let's do the hybrid ProbLog magic here
 | |
|     curnode = GetNodeVarNameDisp(MyManager.manager, MyManager.varmap, Current);
 | |
|     dynvalue_parsed = parse_density_integral_string(dynvalue, curnode);
 | |
|     this_probability=cumulative_normal(dynvalue_parsed.low,dynvalue_parsed.high,dynvalue_parsed.mu,dynvalue_parsed.sigma);
 | |
|   }
 | |
| 
 | |
|   tvalue.probability = this_probability * highvalue.probability + (1 - this_probability) * lowvalue.probability;
 | |
|   tvalue.gradient = this_probability * highvalue.gradient + (1 - this_probability) * lowvalue.gradient;
 | |
|   if ((GetIndex(Current) == TargetVar) ||
 | |
|       ((TargetPattern != NULL) && patternmatch(TargetPattern, MyManager.varmap.vars[GetIndex(Current)]))) {
 | |
|     if (type == 0) {  // current node is normal probabilistic fact
 | |
|       tvalue.gradient += (highvalue.probability - lowvalue.probability) * this_probability * (1 - this_probability) * params.sigmoid_slope;
 | |
|     } else if (type == 1) { // it's a continues fact and we need d/dmu
 | |
|       tvalue.gradient += cumulative_normal_dmu(dynvalue_parsed.low, dynvalue_parsed.high, dynvalue_parsed.mu, dynvalue_parsed.sigma) * (highvalue.probability + lowvalue.probability);
 | |
|     } else if (type == 2) { // it's a continues fact and we need d/dsigma
 | |
|       tvalue.gradient += cumulative_normal_dsigma(dynvalue_parsed.low, dynvalue_parsed.high, dynvalue_parsed.mu, dynvalue_parsed.sigma) * (highvalue.probability + lowvalue.probability);
 | |
|     }
 | |
|   }
 | |
|   gradient = (double *) malloc(sizeof(double));
 | |
|   *gradient = tvalue.gradient;
 | |
|   AddNode(MyManager.his, MyManager.varmap.varstart, Current, tvalue.probability, 0, gradient);
 | |
|   return tvalue;
 | |
| }
 | |
| 
 | |
| char * extractpattern(char *thestr) {
 | |
|   char *p;
 | |
|   int i = 0, sl = strlen(thestr);
 | |
|   while((thestr[i] != '_') && (i < sl)) i++;
 | |
|   if (i == sl) return NULL;
 | |
|   i++;
 | |
|   p = (char *) malloc(sizeof(char) * (i + 2));
 | |
|   strncpy(p, thestr, i);
 | |
|   p[i] = '*';
 | |
|   p[i + 1] = '\0';
 | |
|   return p;
 | |
| }
 | |
| 
 | |
| int patterncalculated(char *pattern, extmanager MyManager, int loc) {
 | |
|   int i;
 | |
|   if (pattern == NULL) return 0;
 | |
|   for (i = loc - 1; i > -1; i--)
 | |
|     if (patternmatch(pattern, MyManager.varmap.vars[i])) return 1;
 | |
|   return 0;
 | |
| }
 |