#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) const { return clauses_; }
    
    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) { }
          
    double weight (void) const;

    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) { }
          
    double weight (void) const;

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



class SetOrNode	: public CircuitNode
{
  public:
    SetOrNode (const Clauses& clauses, string explanation = "")
        : CircuitNode (clauses, explanation), follow_(0) { }
        
    double weight (void) const;
                            
    CircuitNode** follow (void) { return &follow_; }
  private:
    CircuitNode* follow_;
};



class SetAndNode : public CircuitNode
{
  public:
    SetAndNode (unsigned nrGroundings, const Clauses& clauses)
        : CircuitNode (clauses, " IPG"), nrGroundings_(nrGroundings), 
          follow_(0) { }
        
    double weight (void) const;
                    
    CircuitNode** follow (void) { return &follow_; }
  private:
    unsigned      nrGroundings_;
    CircuitNode*  follow_;
};



class IncExcNode : public CircuitNode
{
  public:
    IncExcNode (const Clauses& clauses)
        : CircuitNode (clauses), plus1Branch_(0),
        plus2Branch_(0), minusBranch_(0) { }
      
    CircuitNode** plus1Branch (void) { return &plus1Branch_; }
    CircuitNode** plus2Branch (void) { return &plus2Branch_; }
    CircuitNode** minusBranch (void) { return &minusBranch_; }
  private:
    CircuitNode* plus1Branch_;
    CircuitNode* plus2Branch_;
    CircuitNode* minusBranch_;
};



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



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



class TrueNode : public CircuitNode
{
  public:
    TrueNode (void) : CircuitNode ({}) { }
    
    double weight (void) const;
};




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



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

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

    bool tryUnitPropagation (CircuitNode** follow, Clauses& clauses);
    bool tryIndependence    (CircuitNode** follow, Clauses& clauses);
    bool tryShannonDecomp   (CircuitNode** follow, Clauses& clauses);
    bool tryInclusionExclusion (CircuitNode** follow, Clauses& clauses);
    bool tryIndepPartialGrounding (CircuitNode** follow, Clauses& clauses);
    bool tryIndepPartialGroundingAux (Clauses& clauses, ConstraintTree& ct,
        vector<unsigned>& indices);
    bool tryGrounding       (CircuitNode** follow, 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