yap-6.3/packages/CLPBN/clpbn/bp/VarNode.h

#ifndef HORUS_VARNODE_H
#define HORUS_VARNODE_H

#include "Horus.h"

using namespace std;

class VarNode
{
  public:
    VarNode (const VarNode*);
    VarNode (VarId, unsigned, int = NO_EVIDENCE);
    virtual ~VarNode (void) {};

    bool       isValidState (int);
    bool       isValidState (const string&);
    void       setEvidence (int);
    void       setEvidence (const string&);
    string     label (void) const;
    States     states (void) const;

    unsigned   varId (void) const        { return varId_; }
    unsigned   nrStates (void) const     { return nrStates_; }
    bool       hasEvidence (void) const  { return evidence_ != NO_EVIDENCE; }
    int        getEvidence (void) const  { return evidence_; }
    unsigned   getIndex (void) const     { return index_; }
    void       setIndex (unsigned idx)   { index_ = idx; }

    operator unsigned () const { return index_; }

    bool operator== (const VarNode& var) const
    {
      cout << "equal operator called" << endl;
      assert (!(varId_ == var.varId() && nrStates_ != var.nrStates()));
      return varId_ == var.varId();
    }

    bool operator!= (const VarNode& var) const
    {
      cout << "diff operator called" << endl;
      assert (!(varId_ == var.varId() && nrStates_ != var.nrStates()));
      return varId_ != var.varId();
    }

  private:
    VarId       varId_;
    unsigned    nrStates_;
    int         evidence_;
    unsigned    index_;

};

#endif // BP_VARNODE_H