LiftedVe: move internal classes to source file
This commit is contained in:
parent
7edfe44576
commit
77dfe50acb
@ -1,8 +1,10 @@
|
|||||||
#include <cassert>
|
#include <cassert>
|
||||||
|
|
||||||
|
#include <vector>
|
||||||
#include <set>
|
#include <set>
|
||||||
#include <queue>
|
#include <queue>
|
||||||
#include <algorithm>
|
#include <algorithm>
|
||||||
|
#include <string>
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
#include <sstream>
|
#include <sstream>
|
||||||
|
|
||||||
@ -14,6 +16,141 @@
|
|||||||
|
|
||||||
namespace Horus {
|
namespace Horus {
|
||||||
|
|
||||||
|
class LiftedOperator {
|
||||||
|
public:
|
||||||
|
virtual ~LiftedOperator (void) { }
|
||||||
|
|
||||||
|
virtual double getLogCost (void) = 0;
|
||||||
|
|
||||||
|
virtual void apply (void) = 0;
|
||||||
|
|
||||||
|
virtual std::string toString (void) = 0;
|
||||||
|
|
||||||
|
static std::vector<LiftedOperator*> getValidOps (
|
||||||
|
ParfactorList&, const Grounds&);
|
||||||
|
|
||||||
|
static void printValidOps (ParfactorList&, const Grounds&);
|
||||||
|
|
||||||
|
static std::vector<ParfactorList::iterator> getParfactorsWithGroup (
|
||||||
|
ParfactorList&, PrvGroup group);
|
||||||
|
|
||||||
|
private:
|
||||||
|
DISALLOW_ASSIGN (LiftedOperator);
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class ProductOperator : public LiftedOperator {
|
||||||
|
public:
|
||||||
|
ProductOperator (
|
||||||
|
ParfactorList::iterator g1,
|
||||||
|
ParfactorList::iterator g2,
|
||||||
|
ParfactorList& pfList)
|
||||||
|
: g1_(g1), g2_(g2), pfList_(pfList) { }
|
||||||
|
|
||||||
|
double getLogCost (void);
|
||||||
|
|
||||||
|
void apply (void);
|
||||||
|
|
||||||
|
static std::vector<ProductOperator*> getValidOps (ParfactorList&);
|
||||||
|
|
||||||
|
std::string toString (void);
|
||||||
|
|
||||||
|
private:
|
||||||
|
static bool validOp (Parfactor*, Parfactor*);
|
||||||
|
|
||||||
|
ParfactorList::iterator g1_;
|
||||||
|
ParfactorList::iterator g2_;
|
||||||
|
ParfactorList& pfList_;
|
||||||
|
|
||||||
|
DISALLOW_COPY_AND_ASSIGN (ProductOperator);
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class SumOutOperator : public LiftedOperator {
|
||||||
|
public:
|
||||||
|
SumOutOperator (PrvGroup group, ParfactorList& pfList)
|
||||||
|
: group_(group), pfList_(pfList) { }
|
||||||
|
|
||||||
|
double getLogCost (void);
|
||||||
|
|
||||||
|
void apply (void);
|
||||||
|
|
||||||
|
static std::vector<SumOutOperator*> getValidOps (
|
||||||
|
ParfactorList&, const Grounds&);
|
||||||
|
|
||||||
|
std::string toString (void);
|
||||||
|
|
||||||
|
private:
|
||||||
|
static bool validOp (PrvGroup, ParfactorList&, const Grounds&);
|
||||||
|
|
||||||
|
static bool isToEliminate (Parfactor*, PrvGroup, const Grounds&);
|
||||||
|
|
||||||
|
PrvGroup group_;
|
||||||
|
ParfactorList& pfList_;
|
||||||
|
|
||||||
|
DISALLOW_COPY_AND_ASSIGN (SumOutOperator);
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class CountingOperator : public LiftedOperator {
|
||||||
|
public:
|
||||||
|
CountingOperator (
|
||||||
|
ParfactorList::iterator pfIter,
|
||||||
|
LogVar X,
|
||||||
|
ParfactorList& pfList)
|
||||||
|
: pfIter_(pfIter), X_(X), pfList_(pfList) { }
|
||||||
|
|
||||||
|
double getLogCost (void);
|
||||||
|
|
||||||
|
void apply (void);
|
||||||
|
|
||||||
|
static std::vector<CountingOperator*> getValidOps (ParfactorList&);
|
||||||
|
|
||||||
|
std::string toString (void);
|
||||||
|
|
||||||
|
private:
|
||||||
|
static bool validOp (Parfactor*, LogVar);
|
||||||
|
|
||||||
|
ParfactorList::iterator pfIter_;
|
||||||
|
LogVar X_;
|
||||||
|
ParfactorList& pfList_;
|
||||||
|
|
||||||
|
DISALLOW_COPY_AND_ASSIGN (CountingOperator);
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class GroundOperator : public LiftedOperator {
|
||||||
|
public:
|
||||||
|
GroundOperator (
|
||||||
|
PrvGroup group,
|
||||||
|
unsigned lvIndex,
|
||||||
|
ParfactorList& pfList)
|
||||||
|
: group_(group), lvIndex_(lvIndex), pfList_(pfList) { }
|
||||||
|
|
||||||
|
double getLogCost (void);
|
||||||
|
|
||||||
|
void apply (void);
|
||||||
|
|
||||||
|
static std::vector<GroundOperator*> getValidOps (ParfactorList&);
|
||||||
|
|
||||||
|
std::string toString (void);
|
||||||
|
|
||||||
|
private:
|
||||||
|
std::vector<std::pair<PrvGroup, unsigned>> getAffectedFormulas (void);
|
||||||
|
|
||||||
|
PrvGroup group_;
|
||||||
|
unsigned lvIndex_;
|
||||||
|
ParfactorList& pfList_;
|
||||||
|
|
||||||
|
DISALLOW_COPY_AND_ASSIGN (GroundOperator);
|
||||||
|
};
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
std::vector<LiftedOperator*>
|
std::vector<LiftedOperator*>
|
||||||
LiftedOperator::getValidOps (
|
LiftedOperator::getValidOps (
|
||||||
ParfactorList& pfList,
|
ParfactorList& pfList,
|
||||||
|
@ -1,148 +1,13 @@
|
|||||||
#ifndef YAP_PACKAGES_CLPBN_HORUS_LIFTEDVE_H_
|
#ifndef YAP_PACKAGES_CLPBN_HORUS_LIFTEDVE_H_
|
||||||
#define YAP_PACKAGES_CLPBN_HORUS_LIFTEDVE_H_
|
#define YAP_PACKAGES_CLPBN_HORUS_LIFTEDVE_H_
|
||||||
|
|
||||||
#include <vector>
|
|
||||||
#include <string>
|
|
||||||
|
|
||||||
#include "LiftedSolver.h"
|
#include "LiftedSolver.h"
|
||||||
#include "ParfactorList.h"
|
#include "ParfactorList.h"
|
||||||
|
|
||||||
|
|
||||||
namespace Horus {
|
namespace Horus {
|
||||||
|
|
||||||
class LiftedOperator {
|
class LiftedOperator;
|
||||||
public:
|
|
||||||
virtual ~LiftedOperator (void) { }
|
|
||||||
|
|
||||||
virtual double getLogCost (void) = 0;
|
|
||||||
|
|
||||||
virtual void apply (void) = 0;
|
|
||||||
|
|
||||||
virtual std::string toString (void) = 0;
|
|
||||||
|
|
||||||
static std::vector<LiftedOperator*> getValidOps (
|
|
||||||
ParfactorList&, const Grounds&);
|
|
||||||
|
|
||||||
static void printValidOps (ParfactorList&, const Grounds&);
|
|
||||||
|
|
||||||
static std::vector<ParfactorList::iterator> getParfactorsWithGroup (
|
|
||||||
ParfactorList&, PrvGroup group);
|
|
||||||
|
|
||||||
private:
|
|
||||||
DISALLOW_ASSIGN (LiftedOperator);
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class ProductOperator : public LiftedOperator {
|
|
||||||
public:
|
|
||||||
ProductOperator (
|
|
||||||
ParfactorList::iterator g1,
|
|
||||||
ParfactorList::iterator g2,
|
|
||||||
ParfactorList& pfList)
|
|
||||||
: g1_(g1), g2_(g2), pfList_(pfList) { }
|
|
||||||
|
|
||||||
double getLogCost (void);
|
|
||||||
|
|
||||||
void apply (void);
|
|
||||||
|
|
||||||
static std::vector<ProductOperator*> getValidOps (ParfactorList&);
|
|
||||||
|
|
||||||
std::string toString (void);
|
|
||||||
|
|
||||||
private:
|
|
||||||
static bool validOp (Parfactor*, Parfactor*);
|
|
||||||
|
|
||||||
ParfactorList::iterator g1_;
|
|
||||||
ParfactorList::iterator g2_;
|
|
||||||
ParfactorList& pfList_;
|
|
||||||
|
|
||||||
DISALLOW_COPY_AND_ASSIGN (ProductOperator);
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class SumOutOperator : public LiftedOperator {
|
|
||||||
public:
|
|
||||||
SumOutOperator (PrvGroup group, ParfactorList& pfList)
|
|
||||||
: group_(group), pfList_(pfList) { }
|
|
||||||
|
|
||||||
double getLogCost (void);
|
|
||||||
|
|
||||||
void apply (void);
|
|
||||||
|
|
||||||
static std::vector<SumOutOperator*> getValidOps (
|
|
||||||
ParfactorList&, const Grounds&);
|
|
||||||
|
|
||||||
std::string toString (void);
|
|
||||||
|
|
||||||
private:
|
|
||||||
static bool validOp (PrvGroup, ParfactorList&, const Grounds&);
|
|
||||||
|
|
||||||
static bool isToEliminate (Parfactor*, PrvGroup, const Grounds&);
|
|
||||||
|
|
||||||
PrvGroup group_;
|
|
||||||
ParfactorList& pfList_;
|
|
||||||
|
|
||||||
DISALLOW_COPY_AND_ASSIGN (SumOutOperator);
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class CountingOperator : public LiftedOperator {
|
|
||||||
public:
|
|
||||||
CountingOperator (
|
|
||||||
ParfactorList::iterator pfIter,
|
|
||||||
LogVar X,
|
|
||||||
ParfactorList& pfList)
|
|
||||||
: pfIter_(pfIter), X_(X), pfList_(pfList) { }
|
|
||||||
|
|
||||||
double getLogCost (void);
|
|
||||||
|
|
||||||
void apply (void);
|
|
||||||
|
|
||||||
static std::vector<CountingOperator*> getValidOps (ParfactorList&);
|
|
||||||
|
|
||||||
std::string toString (void);
|
|
||||||
|
|
||||||
private:
|
|
||||||
static bool validOp (Parfactor*, LogVar);
|
|
||||||
|
|
||||||
ParfactorList::iterator pfIter_;
|
|
||||||
LogVar X_;
|
|
||||||
ParfactorList& pfList_;
|
|
||||||
|
|
||||||
DISALLOW_COPY_AND_ASSIGN (CountingOperator);
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class GroundOperator : public LiftedOperator {
|
|
||||||
public:
|
|
||||||
GroundOperator (
|
|
||||||
PrvGroup group,
|
|
||||||
unsigned lvIndex,
|
|
||||||
ParfactorList& pfList)
|
|
||||||
: group_(group), lvIndex_(lvIndex), pfList_(pfList) { }
|
|
||||||
|
|
||||||
double getLogCost (void);
|
|
||||||
|
|
||||||
void apply (void);
|
|
||||||
|
|
||||||
static std::vector<GroundOperator*> getValidOps (ParfactorList&);
|
|
||||||
|
|
||||||
std::string toString (void);
|
|
||||||
|
|
||||||
private:
|
|
||||||
std::vector<std::pair<PrvGroup, unsigned>> getAffectedFormulas (void);
|
|
||||||
|
|
||||||
PrvGroup group_;
|
|
||||||
unsigned lvIndex_;
|
|
||||||
ParfactorList& pfList_;
|
|
||||||
|
|
||||||
DISALLOW_COPY_AND_ASSIGN (GroundOperator);
|
|
||||||
};
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class LiftedVe : public LiftedSolver {
|
class LiftedVe : public LiftedSolver {
|
||||||
|
Reference in New Issue
Block a user