yap-6.3/packages/CLPBN/horus/LiftedCircuit.h

#ifndef HORUS_LIFTEDCIRCUIT_H
#define HORUS_LIFTEDCIRCUIT_H

#include "LiftedWCNF.h"


enum CircuitNodeType {
  OR_NODE,
  AND_NODE,
  SET_OR_NODE,
  SET_AND_NODE,
  INC_EXC_NODE,
  LEAF_NODE,
  SMOOTH_NODE,
  TRUE_NODE,
  FAIL_NODE
};



class CircuitNode
{
  public:
    CircuitNode (const Clauses& clauses, string explanation = "")
        : clauses_(clauses), explanation_(explanation) { }
    
    const Clauses& clauses (void) { return clauses_; }
    
    virtual double weight (void) const { return 0; }
    
    string explanation (void) const { return explanation_; }
        
  private:    
    Clauses  clauses_;
    string   explanation_;
};



class OrNode : public CircuitNode
{
  public:
    OrNode (const Clauses& clauses, string explanation = "")
        : CircuitNode (clauses, explanation),
          leftBranch_(0), rightBranch_(0) { }

    CircuitNode** leftBranch  (void) { return &leftBranch_; }
    CircuitNode** rightBranch (void) { return &rightBranch_; }
  private:
    CircuitNode* leftBranch_;
    CircuitNode* rightBranch_;
};



class AndNode : public CircuitNode
{
  public:
    AndNode (const Clauses& clauses, string explanation = "")
        : CircuitNode (clauses, explanation),
          leftBranch_(0), rightBranch_(0) { }
          
    AndNode (
        const Clauses& clauses,
        CircuitNode* leftBranch,
        CircuitNode* rightBranch,
        string explanation = "")
        : CircuitNode (clauses, explanation),
          leftBranch_(leftBranch), rightBranch_(rightBranch) { }
          
   AndNode (
        CircuitNode* leftBranch,
        CircuitNode* rightBranch,
        string explanation = "")
        : CircuitNode ({}, explanation),
          leftBranch_(leftBranch), rightBranch_(rightBranch) { }

    CircuitNode** leftBranch  (void) { return &leftBranch_; }
    CircuitNode** rightBranch (void) { return &rightBranch_; }
  private:
    CircuitNode* leftBranch_;
    CircuitNode* rightBranch_;
};



class SetAndNode : public CircuitNode
{
  public:
  private:
    CircuitNode* follow_;
};



class SetOrNode	: public CircuitNode
{
  public:
  private:
    CircuitNode* follow_;
};



class IncExclNode : public CircuitNode
{
  public:
  private:
    CircuitNode* xFollow_;
    CircuitNode* yFollow_;
    CircuitNode* zFollow_;
};



class LeafNode : public CircuitNode
{
  public:
    LeafNode (const Clause& clause) : CircuitNode ({clause}) { }
};



class SmoothNode : public CircuitNode
{
  public:
    SmoothNode (const Clauses& clauses) : CircuitNode (clauses) { }
};



class TrueNode : public CircuitNode
{
  public:
    TrueNode () : CircuitNode ({}) { }
};




class FailNode : public CircuitNode
{
  public:
    FailNode (const Clauses& clauses) : CircuitNode (clauses) { }
};



class LiftedCircuit
{
  public:
    LiftedCircuit (const LiftedWCNF* lwcnf);
    
    void smoothCircuit (void);
    
    void exportToGraphViz (const char*);

  private:

    void compile (CircuitNode** follow, const Clauses& clauses);

    bool tryUnitPropagation (CircuitNode** follow, const Clauses& clauses);
    bool tryIndependence    (CircuitNode** follow, const Clauses& clauses);
    bool tryShannonDecomp   (CircuitNode** follow, const Clauses& clauses);
        
    TinySet<LiteralId> smoothCircuit (CircuitNode* node);
    
    CircuitNodeType getCircuitNodeType (const CircuitNode* node) const;
     
    string escapeNode (const CircuitNode* node) const;
    
    void exportToGraphViz (CircuitNode* node, ofstream&);

    CircuitNode*       root_;
    const LiftedWCNF*  lwcnf_;
};

#endif // HORUS_LIFTEDCIRCUIT_H