This repository has been archived on 2023-08-20. You can view files and clone it, but cannot push or open issues or pull requests.
yap-6.3/packages/CLPBN/clpbn/bp/BpNode.h
2011-02-23 15:31:56 +00:00

57 lines
2.0 KiB
C++

#ifndef BP_BP_NODE_H
#define BP_BP_NODE_H
#include <vector>
#include <map>
#include <deque>
#include <string>
#include "BayesianNode.h"
using namespace std;
class BpNode : public BayesianNode
{
public:
// constructs
BpNode (string, vector<BayesianNode*>, Distribution* dist, int = -1);
// destruct
~BpNode (void);
// methods
static void enableParallelSchedule (void);
void allocateMemory (void);
double* getPiValues (void) const;
double getPiValue (int) const;
void setPiValue (int, double);
double* getLambdaValues (void) const;
double getLambdaValue (int) const;
void setLambdaValue (int, double);
double* getPiMessages (BpNode*) const;
double getPiMessage (BpNode*, int) const;
void setPiMessage (BpNode*, int, double);
double* getLambdaMessages (BpNode*) const;
double getLambdaMessage (BpNode*, int) const;
void setLambdaMessage (BpNode*, int, double);
double* getBeliefs (void);
double getBeliefChange (void);
void normalizeMessages (void);
void swapMessages (void);
private:
BpNode (const BpNode&); // disallow copy
void operator= (const BpNode&); // disallow assign
// members
double* lambdaValues_;
double* piValues_;
map<BpNode*, double*> piMessages_;
map<BpNode*, double*> lambdaMessages_;
map<BpNode*, double*>* newPiMessages_;
map<BpNode*, double*>* newLambdaMessages_;
double* oldBeliefs_;
static bool parallelSchedule_;
static const double MAX_CHANGE_ = 1.0;
};
#endif // BP_BP_NODE_H