#ifndef HORUS_CFACTORGRAPH_H #define HORUS_CFACTORGRAPH_H #include #include "FactorGraph.h" #include "Factor.h" #include "Util.h" #include "Horus.h" class VarCluster; class FacCluster; class VarSignatureHash; class FacSignatureHash; typedef long Color; typedef vector Colors; typedef vector> VarSignature; typedef vector FacSignature; typedef unordered_map DistColorMap; typedef unordered_map VarColorMap; typedef unordered_map VarSignMap; typedef unordered_map FacSignMap; typedef vector VarClusters; typedef vector FacClusters; typedef unordered_map VarId2VarCluster; struct VarSignatureHash { size_t operator() (const VarSignature &sig) const { size_t val = hash()(sig.size()); for (unsigned i = 0; i < sig.size(); i++) { val ^= hash()(sig[i].first); val ^= hash()(sig[i].second); } return val; } }; struct FacSignatureHash { size_t operator() (const FacSignature &sig) const { size_t val = hash()(sig.size()); for (unsigned i = 0; i < sig.size(); i++) { val ^= hash()(sig[i]); } return val; } }; class VarCluster { public: VarCluster (const VarNodes& vs) : members_(vs) { } const VarNode* first (void) const { return members_.front(); } const VarNodes& members (void) const { return members_; } VarNode* representative (void) const { return repr_; } void setRepresentative (VarNode* vn) { repr_ = vn; } private: VarNodes members_; VarNode* repr_; }; class FacCluster { public: FacCluster (const FacNodes& fcs, const VarClusters& vcs) : members_(fcs), varClusters_(vcs) { } const FacNode* first (void) const { return members_.front(); } const FacNodes& members (void) const { return members_; } VarClusters& varClusters (void) { return varClusters_; } FacNode* representative (void) const { return repr_; } void setRepresentative (FacNode* fn) { repr_ = fn; } private: FacNodes members_; VarClusters varClusters_; FacNode* repr_; }; class CFactorGraph { public: CFactorGraph (const FactorGraph&); ~CFactorGraph (void); const VarClusters& varClusters (void) { return varClusters_; } const FacClusters& facClusters (void) { return facClusters_; } VarNode* getEquivalent (VarId vid) { VarCluster* vc = vid2VarCluster_.find (vid)->second; return vc->representative(); } FactorGraph* getGroundFactorGraph (void); unsigned getEdgeCount (const FacCluster*, const VarCluster*, unsigned index) const; static bool checkForIdenticalFactors; private: Color getFreeColor (void) { ++ freeColor_; return freeColor_ - 1; } Color getColor (const VarNode* vn) const { return varColors_[vn->getIndex()]; } Color getColor (const FacNode* fn) const { return facColors_[fn->getIndex()]; } void setColor (const VarNode* vn, Color c) { varColors_[vn->getIndex()] = c; } void setColor (const FacNode* fn, Color c) { facColors_[fn->getIndex()] = c; } void findIdenticalFactors (void); void setInitialColors (void); void createGroups (void); void createClusters (const VarSignMap&, const FacSignMap&); VarSignature getSignature (const VarNode*); FacSignature getSignature (const FacNode*); void printGroups (const VarSignMap&, const FacSignMap&) const; Color freeColor_; Colors varColors_; Colors facColors_; VarClusters varClusters_; FacClusters facClusters_; VarId2VarCluster vid2VarCluster_; const FactorGraph* groundFg_; }; #endif // HORUS_CFACTORGRAPH_H