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

#ifndef YAP_PACKAGES_CLPBN_HORUS_VAR_H_
#define YAP_PACKAGES_CLPBN_HORUS_VAR_H_

#include <cassert>

#include <unordered_map>
#include <string>

#include "Util.h"
#include "Horus.h"


namespace Horus {

struct VarInfo {
  VarInfo (std::string l, const States& sts)
      : label(l), states(sts) { }
  std::string label;
  States states;
};


class Var {
  public:
    Var (const Var*);

    Var (VarId, unsigned, int = Constants::unobserved);

    virtual ~Var (void) { };

    VarId varId (void) const { return varId_; }

    unsigned range (void) const { return range_; }

    int getEvidence (void) const  { return evidence_; }

    size_t getIndex (void) const { return index_; }

    void setIndex (size_t idx) { index_ = idx; }

    bool hasEvidence (void) const;

    operator size_t (void) const;

    bool operator== (const Var& var) const;

    bool operator!= (const Var& var) const;

    bool isValidState (int);

    void setEvidence (int);

    std::string label (void) const;

    States states (void) const;

    static void addVarInfo (
        VarId vid, std::string label, const States& states);

    static VarInfo getVarInfo (VarId vid);

    static bool varsHaveInfo (void);

    static void clearVarsInfo (void);

  private:
    VarId     varId_;
    unsigned  range_;
    int       evidence_;
    size_t    index_;

    static std::unordered_map<VarId, VarInfo> varsInfo_;

    DISALLOW_COPY_AND_ASSIGN(Var);
};



inline bool
Var::hasEvidence (void) const
{
  return evidence_ != Constants::unobserved;
}



inline
Var::operator size_t (void) const
{
  return index_;
}



inline bool
Var::operator== (const Var& var) const
{
  assert (!(varId_ == var.varId() && range_ != var.range()));
  return varId_ == var.varId();
}



inline bool
Var::operator!= (const Var& var) const
{
  return !(*this == var);
}

}  // namespace Horus

#endif  // YAP_PACKAGES_CLPBN_HORUS_VAR_H_