Fixed a bug (introduced in commit 64db6bc3...) and another one in Factors2mx
[libdai.git] / include / dai / factor.h
index 70042ba..05f3e82 100644 (file)
 namespace dai {
 
 
-/// Function object similar to std::divides(), but different in that dividing by zero results in zero
-template<typename T> struct divides0 : public std::binary_function<T, T, T> {
-    /// Returns (\a j == 0 ? 0 : (\a i / \a j))
-    T operator()( const T &i, const T &j ) const {
-        if( j == (T)0 )
-            return (T)0;
-        else
-            return i / j;
-    }
-};
-
-
 /// Represents a (probability) factor.
 /** Mathematically, a \e factor is a function mapping joint states of some
  *  variables to the nonnegative real numbers.
@@ -61,29 +49,21 @@ template<typename T> struct divides0 : public std::binary_function<T, T, T> {
  *  ordering, which is defined by the one-to-one correspondence of a joint state
  *  in \f$\prod_{l\in L} X_l\f$ with a linear index in
  *  \f$\{0,1,\dots,\prod_{l\in L} S_l-1\}\f$ according to the mapping \f$\sigma\f$
- *  induced by VarSet::calcState(const std::map<Var,size_t> &).
+ *  induced by dai::calcLinearState().
  *
  *  \tparam T Should be a scalar that is castable from and to double and should support elementary arithmetic operations.
  *  \todo Define a better fileformat for .fg files (maybe using XML)?
  *  \todo Add support for sparse factors.
  */
-template <typename T> class TFactor {
+template <typename T>
+class TFactor {
     private:
         /// Stores the variables on which the factor depends
-        VarSet      _vs;
+        VarSet _vs;
         /// Stores the factor values
-        TProb<T>    _p;
+        TProb<T> _p;
 
     public:
-        /// Constant iterator over the values
-        typedef typename TProb<T>::const_iterator const_iterator;
-        /// Iterator over the values
-        typedef typename TProb<T>::iterator iterator;
-        /// Constant reverse iterator over the values
-        typedef typename TProb<T>::const_reverse_iterator const_reverse_iterator;
-        /// Reverse iterator over the values
-        typedef typename TProb<T>::reverse_iterator reverse_iterator;
-
     /// \name Constructors and destructors
     //@{
         /// Constructs factor depending on no variables with value \a p
@@ -93,10 +73,14 @@ template <typename T> class TFactor {
         TFactor( const Var &v ) : _vs(v), _p(v.states()) {}
 
         /// Constructs factor depending on variables in \a vars with uniform distribution
-        TFactor( const VarSet& vars ) : _vs(vars), _p(_vs.nrStates()) {}
+        TFactor( const VarSet& vars ) : _vs(vars), _p((size_t)_vs.nrStates()) {
+            DAI_ASSERT( _vs.nrStates() <= std::numeric_limits<std::size_t>::max() );
+        }
 
         /// Constructs factor depending on variables in \a vars with all values set to \a p
-        TFactor( const VarSet& vars, T p ) : _vs(vars), _p(_vs.nrStates(),p) {}
+        TFactor( const VarSet& vars, T p ) : _vs(vars), _p((size_t)_vs.nrStates(),p) {
+            DAI_ASSERT( _vs.nrStates() <= std::numeric_limits<std::size_t>::max() );
+        }
 
         /// Constructs factor depending on variables in \a vars, copying the values from a std::vector<>
         /** \tparam S Type of values of \a x
@@ -104,29 +88,45 @@ template <typename T> class TFactor {
          *  \param x Vector with values to be copied.
          */
         template<typename S>
-        TFactor( const VarSet& vars, const std::vector<S> &x ) : _vs(vars), _p(x.begin(), x.begin() + _vs.nrStates(), _vs.nrStates()) {
+        TFactor( const VarSet& vars, const std::vector<S> &x ) : _vs(vars), _p() {
             DAI_ASSERT( x.size() == vars.nrStates() );
+            _p = TProb<T>( x.begin(), x.end(), x.size() );
         }
 
         /// Constructs factor depending on variables in \a vars, copying the values from an array
         /** \param vars contains the variables that the new factor should depend on.
          *  \param p Points to array of values to be added.
          */
-        TFactor( const VarSet& vars, const T* p ) : _vs(vars), _p(p, p + _vs.nrStates(), _vs.nrStates()) {}
+        TFactor( const VarSet& vars, const T* p ) : _vs(vars), _p(p, p + (size_t)_vs.nrStates(), (size_t)_vs.nrStates()) {
+            DAI_ASSERT( _vs.nrStates() <= std::numeric_limits<std::size_t>::max() );
+        }
 
         /// Constructs factor depending on variables in \a vars, copying the values from \a p
         TFactor( const VarSet& vars, const TProb<T> &p ) : _vs(vars), _p(p) {
-            DAI_DEBASSERT( _vs.nrStates() == _p.size() );
+            DAI_ASSERT( _vs.nrStates() == _p.size() );
         }
 
         /// Constructs factor depending on variables in \a vars, permuting the values given in \a p accordingly
         TFactor( const std::vector<Var> &vars, const std::vector<T> &p ) : _vs(vars.begin(), vars.end(), vars.size()), _p(p.size()) {
+            size_t nrStates = 1;
+            for( size_t i = 0; i < vars.size(); i++ )
+                nrStates *= vars[i].states();
+            DAI_ASSERT( nrStates == p.size() );
             Permute permindex(vars);
             for( size_t li = 0; li < p.size(); ++li )
-                _p[permindex.convertLinearIndex(li)] = p[li];
+                _p.set( permindex.convertLinearIndex(li), p[li] );
         }
     //@}
 
+    /// \name Get/set individual entries
+    //@{
+        /// Sets \a i 'th entry to \a val
+        void set( size_t i, T val ) { _p.set( i, val ); }
+
+        /// Gets \a i 'th entry
+        T get( size_t i ) const { return _p[i]; }
+    //@}
+
     /// \name Queries
     //@{
         /// Returns constant reference to value vector
@@ -138,16 +138,16 @@ template <typename T> class TFactor {
         /// Returns a copy of the \a i 'th entry of the value vector
         T operator[] (size_t i) const { return _p[i]; }
 
-        /// Returns a reference to the \a i 'th entry of the value vector
-        T& operator[] (size_t i) { return _p[i]; }
-
         /// Returns constant reference to variable set (i.e., the variables on which the factor depends)
         const VarSet& vars() const { return _vs; }
 
+        /// Returns reference to variable set (i.e., the variables on which the factor depends)
+        VarSet& vars() { return _vs; }
+
         /// Returns the number of possible joint states of the variables on which the factor depends, \f$\prod_{l\in L} S_l\f$
         /** \note This is equal to the length of the value vector.
          */
-        size_t states() const { return _p.size(); }
+        size_t nrStates() const { return _p.size(); }
 
         /// Returns the Shannon entropy of \c *this, \f$-\sum_i p_i \log p_i\f$
         T entropy() const { return _p.entropy(); }
@@ -160,6 +160,9 @@ template <typename T> class TFactor {
 
         /// Returns sum of all values
         T sum() const { return _p.sum(); }
+        
+        /// Returns sum of absolute values
+        T sumAbs() const { return _p.sumAbs(); }
 
         /// Returns maximum absolute value of all values
         T maxAbs() const { return _p.maxAbs(); }
@@ -172,106 +175,102 @@ template <typename T> class TFactor {
 
         /// Returns strength of this factor (between variables \a i and \a j), as defined in eq. (52) of [\ref MoK07b]
         T strength( const Var &i, const Var &j ) const;
-    //@}
 
-    /// @name Iterator interface
-    //@{
-        /// Returns iterator that points to the first value
-        iterator begin() { return _p.begin(); }
-        /// Returns constant iterator that points to the first value
-        const_iterator begin() const { return _p.begin(); }
-
-        /// Returns iterator that points beyond the last value
-        iterator end() { return _p.end(); }
-        /// Returns constant iterator that points beyond the last value
-        const_iterator end() const { return _p.end(); }
-
-        /// Returns reverse iterator that points to the last value
-        reverse_iterator rbegin() { return _p.rbegin(); }
-        /// Returns constant reverse iterator that points to the last value
-        const_reverse_iterator rbegin() const { return _p.rbegin(); }
-
-        /// Returns reverse iterator that points beyond the first value
-        reverse_iterator rend() { return _p.rend(); }
-        /// Returns constant reverse iterator that points beyond the first value
-        const_reverse_iterator rend() const { return _p.rend(); }
+        /// Comparison
+        bool operator==( const TFactor<T>& y ) const {
+            return (_vs == y._vs) && (_p == y._p);
+        }
     //@}
 
     /// \name Unary transformations
     //@{
+        /// Returns negative of \c *this
+        TFactor<T> operator- () const { 
+            // Note: the alternative (shorter) way of implementing this,
+            //   return TFactor<T>( _vs, _p.abs() );
+            // is slower because it invokes the copy constructor of TProb<T>
+            TFactor<T> x;
+            x._vs = _vs;
+            x._p = -_p;
+            return x;
+        }
+
         /// Returns pointwise absolute value
         TFactor<T> abs() const {
-            TFactor<T> e;
-            e._vs = _vs;
-            e._p = _p.abs();
-            return e;
+            TFactor<T> x;
+            x._vs = _vs;
+            x._p = _p.abs();
+            return x;
         }
 
         /// Returns pointwise exponent
         TFactor<T> exp() const {
-            TFactor<T> e;
-            e._vs = _vs;
-            e._p = _p.exp();
-            return e;
+            TFactor<T> x;
+            x._vs = _vs;
+            x._p = _p.exp();
+            return x;
         }
 
         /// Returns pointwise logarithm
         /** If \a zero == \c true, uses <tt>log(0)==0</tt>; otherwise, <tt>log(0)==-Inf</tt>.
          */
         TFactor<T> log(bool zero=false) const {
-            TFactor<T> l;
-            l._vs = _vs;
-            l._p = _p.log(zero);
-            return l;
+            TFactor<T> x;
+            x._vs = _vs;
+            x._p = _p.log(zero);
+            return x;
         }
 
         /// Returns pointwise inverse
         /** If \a zero == \c true, uses <tt>1/0==0</tt>; otherwise, <tt>1/0==Inf</tt>.
          */
         TFactor<T> inverse(bool zero=true) const {
-            TFactor<T> inv;
-            inv._vs = _vs;
-            inv._p = _p.inverse(zero);
-            return inv;
+            TFactor<T> x;
+            x._vs = _vs;
+            x._p = _p.inverse(zero);
+            return x;
         }
 
         /// Returns normalized copy of \c *this, using the specified norm
-        TFactor<T> normalized( typename Prob::NormType norm=Prob::NORMPROB ) const {
-            TFactor<T> result;
-            result._vs = _vs;
-            result._p = _p.normalized( norm );
-            return result;
+        /** \throw NOT_NORMALIZABLE if the norm is zero
+         */
+        TFactor<T> normalized( ProbNormType norm=NORMPROB ) const {
+            TFactor<T> x;
+            x._vs = _vs;
+            x._p = _p.normalized( norm );
+            return x;
         }
     //@}
 
     /// \name Unary operations
     //@{
         /// Draws all values i.i.d. from a uniform distribution on [0,1)
-        TFactor<T> & randomize () { _p.randomize(); return *this; }
+        TFactor<T>& randomize() { _p.randomize(); return *this; }
 
         /// Sets all values to \f$1/n\f$ where \a n is the number of states
-        TFactor<T>& setUniform () { _p.setUniform(); return *this; }
+        TFactor<T>& setUniform() { _p.setUniform(); return *this; }
+
+        /// Applies absolute value pointwise
+        TFactor<T>& takeAbs() { _p.takeAbs(); return *this; }
+
+        /// Applies exponent pointwise
+        TFactor<T>& takeExp() { _p.takeExp(); return *this; }
+
+        /// Applies logarithm pointwise
+        /** If \a zero == \c true, uses <tt>log(0)==0</tt>; otherwise, <tt>log(0)==-Inf</tt>.
+         */
+        TFactor<T>& takeLog( bool zero = false ) { _p.takeLog(zero); return *this; }
 
         /// Normalizes factor using the specified norm
-        T normalize( typename Prob::NormType norm=Prob::NORMPROB ) { return _p.normalize( norm ); }
+        /** \throw NOT_NORMALIZABLE if the norm is zero
+         */
+        T normalize( ProbNormType norm=NORMPROB ) { return _p.normalize( norm ); }
     //@}
 
     /// \name Operations with scalars
     //@{
         /// Sets all values to \a x
-        TFactor<T> & fill (T x) { _p.fill( x ); return *this; }
-
-        // OBSOLETE
-        /// Sets values that are smaller (in absolute value) than \a epsilon to 0
-        /** \note Obsolete, to be removed soon
-         */
-        TFactor<T>& makeZero( T epsilon ) { _p.makeZero( epsilon ); return *this; }
-
-        // OBSOLETE
-        /// Sets values that are smaller than \a epsilon to \a epsilon
-        /** \note Obsolete, to be removed soon
-         */
-        TFactor<T>& makePositive( T epsilon ) { _p.makePositive( epsilon ); return *this; }
+        TFactor<T>& fill (T x) { _p.fill( x ); return *this; }
 
         /// Adds scalar \a x to each value
         TFactor<T>& operator+= (T x) { _p += x; return *this; }
@@ -293,29 +292,37 @@ template <typename T> class TFactor {
     //@{
         /// Returns sum of \c *this and scalar \a x
         TFactor<T> operator+ (T x) const {
-            TFactor<T> result(*this);
-            result._p += x;
+            // Note: the alternative (shorter) way of implementing this,
+            //   TFactor<T> result(*this);
+            //   result._p += x;
+            // is slower because it invokes the copy constructor of TFactor<T>
+            TFactor<T> result;
+            result._vs = _vs;
+            result._p = p() + x;
             return result;
         }
 
         /// Returns difference of \c *this and scalar \a x
         TFactor<T> operator- (T x) const {
-            TFactor<T> result(*this);
-            result._p -= x;
+            TFactor<T> result;
+            result._vs = _vs;
+            result._p = p() - x;
             return result;
         }
 
         /// Returns product of \c *this with scalar \a x
         TFactor<T> operator* (T x) const {
-            TFactor<T> result = *this;
-            result.p() *= x;
+            TFactor<T> result;
+            result._vs = _vs;
+            result._p = p() * x;
             return result;
         }
 
         /// Returns quotient of \c *this with scalar \a x
         TFactor<T> operator/ (T x) const {
-            TFactor<T> result = *this;
-            result.p() /= x;
+            TFactor<T> result;
+            result._vs = _vs;
+            result._p = p() / x;
             return result;
         }
 
@@ -323,93 +330,122 @@ template <typename T> class TFactor {
         TFactor<T> operator^ (T x) const {
             TFactor<T> result;
             result._vs = _vs;
-            result._p = _p^x;
+            result._p = p() ^ x;
             return result;
         }
     //@}
 
     /// \name Operations with other factors
     //@{
-        /// Adds \a f to \c *this
+        /// Applies binary operation \a op on two factors, \c *this and \a g
+        /** \tparam binOp Type of function object that accepts two arguments of type \a T and outputs a type \a T
+         *  \param g Right operand
+         *  \param op Operation of type \a binOp
+         */
+        template<typename binOp> TFactor<T>& binaryOp( const TFactor<T> &g, binOp op ) {
+            if( _vs == g._vs ) // optimize special case
+                _p.pwBinaryOp( g._p, op );
+            else {
+                TFactor<T> f(*this); // make a copy
+                _vs |= g._vs;
+                DAI_ASSERT( _vs.nrStates() < std::numeric_limits<std::size_t>::max() );
+                size_t N = (size_t)_vs.nrStates();
+
+                IndexFor i_f( f._vs, _vs );
+                IndexFor i_g( g._vs, _vs );
+
+                _p.p().clear();
+                _p.p().reserve( N );
+                for( size_t i = 0; i < N; i++, ++i_f, ++i_g )
+                    _p.p().push_back( op( f._p[i_f], g._p[i_g] ) );
+            }
+            return *this;
+        }
+
+        /// Adds \a g to \c *this
         /** The sum of two factors is defined as follows: if
          *  \f$f : \prod_{l\in L} X_l \to [0,\infty)\f$ and \f$g : \prod_{m\in M} X_m \to [0,\infty)\f$, then
          *  \f[f+g : \prod_{l\in L\cup M} X_l \to [0,\infty) : x \mapsto f(x_L) + g(x_M).\f]
          */
-        TFactor<T>& operator+= (const TFactor<T>& f) {
-            if( f._vs == _vs ) // optimize special case
-                _p += f._p;
-            else
-                *this = (*this + f);
-            return *this;
-        }
+        TFactor<T>& operator+= (const TFactor<T>& g) { return binaryOp( g, std::plus<T>() ); }
 
-        /// Subtracts \a f from \c *this
+        /// Subtracts \a g from \c *this
         /** The difference of two factors is defined as follows: if
          *  \f$f : \prod_{l\in L} X_l \to [0,\infty)\f$ and \f$g : \prod_{m\in M} X_m \to [0,\infty)\f$, then
          *  \f[f-g : \prod_{l\in L\cup M} X_l \to [0,\infty) : x \mapsto f(x_L) - g(x_M).\f]
          */
-        TFactor<T>& operator-= (const TFactor<T>& f) {
-            if( f._vs == _vs ) // optimize special case
-                _p -= f._p;
-            else
-                *this = (*this - f);
-            return *this;
-        }
+        TFactor<T>& operator-= (const TFactor<T>& g) { return binaryOp( g, std::minus<T>() ); }
 
-        /// Multiplies \c *this with \a f
+        /// Multiplies \c *this with \a g
         /** The product of two factors is defined as follows: if
          *  \f$f : \prod_{l\in L} X_l \to [0,\infty)\f$ and \f$g : \prod_{m\in M} X_m \to [0,\infty)\f$, then
          *  \f[fg : \prod_{l\in L\cup M} X_l \to [0,\infty) : x \mapsto f(x_L) g(x_M).\f]
          */
-        TFactor<T>& operator*= (const TFactor<T>& f) {
-            if( f._vs == _vs ) // optimize special case
-                _p *= f._p;
-            else
-                *this = (*this * f);
-            return *this;
-        }
+        TFactor<T>& operator*= (const TFactor<T>& g) { return binaryOp( g, std::multiplies<T>() ); }
 
-        /// Divides \c *this by \a f (where division by zero yields zero)
+        /// Divides \c *this by \a g (where division by zero yields zero)
         /** The quotient of two factors is defined as follows: if
          *  \f$f : \prod_{l\in L} X_l \to [0,\infty)\f$ and \f$g : \prod_{m\in M} X_m \to [0,\infty)\f$, then
          *  \f[\frac{f}{g} : \prod_{l\in L\cup M} X_l \to [0,\infty) : x \mapsto \frac{f(x_L)}{g(x_M)}.\f]
          */
-        TFactor<T>& operator/= (const TFactor<T>& f) {
-            if( f._vs == _vs ) // optimize special case
-                _p /= f._p;
-            else
-                *this = (*this / f);
-            return *this;
-        }
+        TFactor<T>& operator/= (const TFactor<T>& g) { return binaryOp( g, fo_divides0<T>() ); }
     //@}
 
     /// \name Transformations with other factors
     //@{
-        /// Returns sum of \c *this and \a f
+        /// Returns result of applying binary operation \a op on two factors, \c *this and \a g
+        /** \tparam binOp Type of function object that accepts two arguments of type \a T and outputs a type \a T
+         *  \param g Right operand
+         *  \param op Operation of type \a binOp
+         */
+        template<typename binOp> TFactor<T> binaryTr( const TFactor<T> &g, binOp op ) const {
+            // Note that to prevent a copy to be made, it is crucial 
+            // that the result is declared outside the if-else construct.
+            TFactor<T> result;
+            if( _vs == g._vs ) { // optimize special case
+                result._vs = _vs;
+                result._p = _p.pwBinaryTr( g._p, op );
+            } else {
+                result._vs = _vs | g._vs;
+                DAI_ASSERT( result._vs.nrStates() < std::numeric_limits<std::size_t>::max() );
+                size_t N = (size_t)result._vs.nrStates();
+
+                IndexFor i_f( _vs, result.vars() );
+                IndexFor i_g( g._vs, result.vars() );
+
+                result._p.p().clear();
+                result._p.p().reserve( N );
+                for( size_t i = 0; i < N; i++, ++i_f, ++i_g )
+                    result._p.p().push_back( op( _p[i_f], g[i_g] ) );
+            }
+            return result;
+        }
+
+        /// Returns sum of \c *this and \a g
         /** The sum of two factors is defined as follows: if
          *  \f$f : \prod_{l\in L} X_l \to [0,\infty)\f$ and \f$g : \prod_{m\in M} X_m \to [0,\infty)\f$, then
          *  \f[f+g : \prod_{l\in L\cup M} X_l \to [0,\infty) : x \mapsto f(x_L) + g(x_M).\f]
          */
-        TFactor<T> operator+ (const TFactor<T>& f) const {
-            return pointwiseOp(*this,f,std::plus<T>());
+        TFactor<T> operator+ (const TFactor<T>& g) const {
+            return binaryTr(g,std::plus<T>());
         }
 
-        /// Returns \c *this minus \a f
+        /// Returns \c *this minus \a g
         /** The difference of two factors is defined as follows: if
          *  \f$f : \prod_{l\in L} X_l \to [0,\infty)\f$ and \f$g : \prod_{m\in M} X_m \to [0,\infty)\f$, then
          *  \f[f-g : \prod_{l\in L\cup M} X_l \to [0,\infty) : x \mapsto f(x_L) - g(x_M).\f]
          */
-        TFactor<T> operator- (const TFactor<T>& f) const {
-            return pointwiseOp(*this,f,std::minus<T>());
+        TFactor<T> operator- (const TFactor<T>& g) const {
+            return binaryTr(g,std::minus<T>());
         }
 
-        /// Returns product of \c *this with \a f
+        /// Returns product of \c *this with \a g
         /** The product of two factors is defined as follows: if
          *  \f$f : \prod_{l\in L} X_l \to [0,\infty)\f$ and \f$g : \prod_{m\in M} X_m \to [0,\infty)\f$, then
          *  \f[fg : \prod_{l\in L\cup M} X_l \to [0,\infty) : x \mapsto f(x_L) g(x_M).\f]
          */
-        TFactor<T> operator* (const TFactor<T>& f) const {
-            return pointwiseOp(*this,f,std::multiplies<T>());
+        TFactor<T> operator* (const TFactor<T>& g) const {
+            return binaryTr(g,std::multiplies<T>());
         }
 
         /// Returns quotient of \c *this by \a f (where division by zero yields zero)
@@ -417,8 +453,8 @@ template <typename T> class TFactor {
          *  \f$f : \prod_{l\in L} X_l \to [0,\infty)\f$ and \f$g : \prod_{m\in M} X_m \to [0,\infty)\f$, then
          *  \f[\frac{f}{g} : \prod_{l\in L\cup M} X_l \to [0,\infty) : x \mapsto \frac{f(x_L)}{g(x_M)}.\f]
          */
-        TFactor<T> operator/ (const TFactor<T>& f) const {
-            return pointwiseOp(*this,f,divides0<T>());
+        TFactor<T> operator/ (const TFactor<T>& g) const {
+            return binaryTr(g,fo_divides0<T>());
         }
     //@}
 
@@ -426,7 +462,7 @@ template <typename T> class TFactor {
     //@{
         /// Returns a slice of \c *this, where the subset \a vars is in state \a varsState
         /** \pre \a vars sould be a subset of vars()
-         *  \pre \a varsState < vars.states()
+         *  \pre \a varsState < vars.nrStates()
          *
          *  The result is a factor that depends on the variables of *this except those in \a vars,
          *  obtained by setting the variables in \a vars to the joint state specified by the linear index
@@ -436,20 +472,7 @@ template <typename T> class TFactor {
          *  returned corresponds with the factor \f$g : \prod_{l \in L \setminus M} X_l \to [0,\infty)\f$
          *  defined by \f$g(\{x_l\}_{l\in L \setminus M}) = f(\{x_l\}_{l\in L \setminus M}, \{s(x_m)\}_{m\in M})\f$.
          */
-        TFactor<T> slice( const VarSet& vars, size_t varsState ) const {
-            DAI_ASSERT( vars << _vs );
-            VarSet varsrem = _vs / vars;
-            TFactor<T> result( varsrem, T(0) );
-
-            // OPTIMIZE ME
-            IndexFor i_vars (vars, _vs);
-            IndexFor i_varsrem (varsrem, _vs);
-            for( size_t i = 0; i < states(); i++, ++i_vars, ++i_varsrem )
-                if( (size_t)i_vars == varsState )
-                    result._p[i_varsrem] = _p[i];
-
-            return result;
-        }
+        TFactor<T> slice( const VarSet& vars, size_t varsState ) const; 
 
         /// Embeds this factor in a larger VarSet
         /** \pre vars() should be a subset of \a vars 
@@ -474,6 +497,22 @@ template <typename T> class TFactor {
 };
 
 
+template<typename T> TFactor<T> TFactor<T>::slice( const VarSet& vars, size_t varsState ) const {
+    DAI_ASSERT( vars << _vs );
+    VarSet varsrem = _vs / vars;
+    TFactor<T> result( varsrem, T(0) );
+
+    // OPTIMIZE ME
+    IndexFor i_vars (vars, _vs);
+    IndexFor i_varsrem (varsrem, _vs);
+    for( size_t i = 0; i < nrStates(); i++, ++i_vars, ++i_varsrem )
+        if( (size_t)i_vars == varsState )
+            result.set( i_varsrem, _p[i] );
+
+    return result;
+}
+
+
 template<typename T> TFactor<T> TFactor<T>::marginal(const VarSet &vars, bool normed) const {
     VarSet res_vars = vars & _vs;
 
@@ -481,10 +520,10 @@ template<typename T> TFactor<T> TFactor<T>::marginal(const VarSet &vars, bool no
 
     IndexFor i_res( res_vars, _vs );
     for( size_t i = 0; i < _p.size(); i++, ++i_res )
-        res._p[i_res] += _p[i];
+        res.set( i_res, res[i_res] + _p[i] );
 
     if( normed )
-        res.normalize( Prob::NORMPROB );
+        res.normalize( NORMPROB );
 
     return res;
 }
@@ -498,10 +537,10 @@ template<typename T> TFactor<T> TFactor<T>::maxMarginal(const VarSet &vars, bool
     IndexFor i_res( res_vars, _vs );
     for( size_t i = 0; i < _p.size(); i++, ++i_res )
         if( _p[i] > res._p[i_res] )
-            res._p[i_res] = _p[i];
+            res.set( i_res, _p[i] );
 
     if( normed )
-        res.normalize( Prob::NORMPROB );
+        res.normalize( NORMPROB );
 
     return res;
 }
@@ -536,36 +575,12 @@ template<typename T> T TFactor<T>::strength( const Var &i, const Var &j ) const
 }
 
 
-/// Apply binary operator pointwise on two factors
-/** \relates TFactor
- *  \tparam binaryOp Function object that accepts two arguments of type \a T and outputs a type \a T
- */
-template<typename T, typename binaryOp> TFactor<T> pointwiseOp( const TFactor<T> &f, const TFactor<T> &g, binaryOp op ) {
-    if( f.vars() == g.vars() ) { // optimizate special case
-        TFactor<T> result(f);
-        for( size_t i = 0; i < result.states(); i++ )
-            result[i] = op( result[i], g[i] );
-        return result;
-    } else {
-        TFactor<T> result( f.vars() | g.vars(), (T)0 );
-
-        IndexFor i1(f.vars(), result.vars());
-        IndexFor i2(g.vars(), result.vars());
-
-        for( size_t i = 0; i < result.states(); i++, ++i1, ++i2 )
-            result[i] = op( f[i1], g[i2] );
-
-        return result;
-    }
-}
-
-
 /// Writes a factor to an output stream
 /** \relates TFactor
  */
 template<typename T> std::ostream& operator<< (std::ostream& os, const TFactor<T>& f) {
     os << "(" << f.vars() << ", (";
-    for( size_t i = 0; i < f.states(); i++ )
+    for( size_t i = 0; i < f.nrStates(); i++ )
         os << (i == 0 ? "" : ", ") << f[i];
     os << "))";
     return os;
@@ -576,7 +591,7 @@ template<typename T> std::ostream& operator<< (std::ostream& os, const TFactor<T
 /** \relates TFactor
  *  \pre f.vars() == g.vars()
  */
-template<typename T> T dist( const TFactor<T> &f, const TFactor<T> &g, Prob::DistType dt ) {
+template<typename T> T dist( const TFactor<T> &f, const TFactor<T> &g, ProbDistType dt ) {
     if( f.vars().empty() || g.vars().empty() )
         return -1;
     else {
@@ -591,8 +606,8 @@ template<typename T> T dist( const TFactor<T> &f, const TFactor<T> &g, Prob::Dis
  *  \pre f.vars() == g.vars()
  */
 template<typename T> TFactor<T> max( const TFactor<T> &f, const TFactor<T> &g ) {
-    DAI_ASSERT( f._vs == g._vs );
-    return TFactor<T>( f._vs, max( f.p(), g.p() ) );
+    DAI_ASSERT( f.vars() == g.vars() );
+    return TFactor<T>( f.vars(), max( f.p(), g.p() ) );
 }
 
 
@@ -601,8 +616,8 @@ template<typename T> TFactor<T> max( const TFactor<T> &f, const TFactor<T> &g )
  *  \pre f.vars() == g.vars()
  */
 template<typename T> TFactor<T> min( const TFactor<T> &f, const TFactor<T> &g ) {
-    DAI_ASSERT( f._vs == g._vs );
-    return TFactor<T>( f._vs, min( f.p(), g.p() ) );
+    DAI_ASSERT( f.vars() == g.vars() );
+    return TFactor<T>( f.vars(), min( f.p(), g.p() ) );
 }
 
 
@@ -615,7 +630,7 @@ template<typename T> T MutualInfo(const TFactor<T> &f) {
     VarSet::const_iterator it = f.vars().begin();
     Var i = *it; it++; Var j = *it;
     TFactor<T> projection = f.marginal(i) * f.marginal(j);
-    return dist( f.normalized(), projection, Prob::DISTKL );
+    return dist( f.normalized(), projection, DISTKL );
 }
 
 
@@ -623,6 +638,52 @@ template<typename T> T MutualInfo(const TFactor<T> &f) {
 typedef TFactor<Real> Factor;
 
 
+/// Returns a binary unnormalized single-variable factor \f$ \exp(hx) \f$ where \f$ x = \pm 1 \f$
+/** \param x Variable (should be binary)
+ *  \param h Field strength
+ */
+Factor createFactorIsing( const Var &x, Real h );
+
+
+/// Returns a binary unnormalized pairwise factor \f$ \exp(J x_1 x_2) \f$ where \f$ x_1, x_2 = \pm 1 \f$
+/** \param x1 First variable (should be binary)
+ *  \param x2 Second variable (should be binary)
+ *  \param J Coupling strength
+ */
+Factor createFactorIsing( const Var &x1, const Var &x2, Real J );
+
+
+/// Returns a random factor on the variables \a vs with strength \a beta
+/** Each entry are set by drawing a normally distributed random with mean
+ *  0 and standard-deviation \a beta, and taking its exponent.
+ *  \param vs Variables
+ *  \param beta Factor strength (inverse temperature)
+ */
+Factor createFactorExpGauss( const VarSet &vs, Real beta );
+
+
+/// Returns a pairwise Potts factor \f$ \exp( J \delta_{x_1, x_2} ) \f$
+/** \param x1 First variable
+ *  \param x2 Second variable (should have the same number of states as \a x1)
+ *  \param J  Factor strength
+ */
+Factor createFactorPotts( const Var &x1, const Var &x2, Real J );
+
+
+/// Returns a Kronecker delta point mass
+/** \param v Variable
+ *  \param state The state of \a v that should get value 1
+ */
+Factor createFactorDelta( const Var &v, size_t state );
+
+
+/// Returns a Kronecker delta point mass
+/** \param vs Set of variables
+ *  \param state The state of \a vs that should get value 1
+ */
+Factor createFactorDelta( const VarSet& vs, size_t state );
+
+
 } // end of namespace dai