36 lines
760 B
C++
Executable File
36 lines
760 B
C++
Executable File
#ifndef BP_FACTORGRAPH_H
|
|
#define BP_FACTORGRAPH_H
|
|
|
|
#include <vector>
|
|
#include <string>
|
|
|
|
#include "GraphicalModel.h"
|
|
#include "Shared.h"
|
|
|
|
using namespace std;
|
|
|
|
class FgVarNode;
|
|
class Factor;
|
|
|
|
class FactorGraph : public GraphicalModel
|
|
{
|
|
public:
|
|
FactorGraph (const char* fileName);
|
|
~FactorGraph (void);
|
|
|
|
FgVarSet getFgVarNodes (void) const;
|
|
vector<Factor*> getFactors (void) const;
|
|
VarSet getVariables (void) const;
|
|
FgVarNode* getVariableById (unsigned) const;
|
|
FgVarNode* getVariableByLabel (string) const;
|
|
void printFactorGraph (void) const;
|
|
|
|
private:
|
|
DISALLOW_COPY_AND_ASSIGN (FactorGraph);
|
|
|
|
FgVarSet varNodes_;
|
|
vector<Factor*> factors_;
|
|
};
|
|
|
|
#endif
|