varset.h

Go to the documentation of this file.

/*  This file is part of libDAI - http://www.libdai.org/
 *
 *  Copyright (c) 2006-2011, The libDAI authors. All rights reserved.
 *
 *  Use of this source code is governed by a BSD-style license that can be found in the LICENSE file.
 */




#ifndef __defined_libdai_varset_h
#define __defined_libdai_varset_h


#include <vector>
#include <map>
#include <ostream>
#include <dai/var.h>
#include <dai/util.h>
#include <dai/smallset.h>


namespace dai {


// Predefine for definitions of calcLinearState() and calcState()
class VarSet;



size_t calcLinearState( const VarSet &vs, const std::map<Var, size_t> &state );



std::map<Var, size_t> calcState( const VarSet &vs, size_t linearState );



class VarSet : public SmallSet<Var> {
    public:

        VarSet() : SmallSet<Var>() {}

        VarSet( const SmallSet<Var> &x ) : SmallSet<Var>(x) {}

        VarSet( const Var &v ) : SmallSet<Var>(v) {}

        VarSet( const Var &v1, const Var &v2 ) : SmallSet<Var>(v1,v2) {}


        template <typename VarIterator>
        VarSet( VarIterator begin, VarIterator end, size_t sizeHint=0 ) : SmallSet<Var>(begin,end,sizeHint) {}



        BigInt nrStates() const {
            BigInt states = 1;
            for( VarSet::const_iterator n = begin(); n != end(); n++ )
                states *= (BigInt)n->states();
            return states;
        }


        friend std::ostream& operator<<( std::ostream &os, const VarSet &vs )  {
            os << "{";
            for( VarSet::const_iterator v = vs.begin(); v != vs.end(); v++ )
                os << (v != vs.begin() ? ", " : "") << *v;
            os << "}";
            return( os );
        }

        std::string toString() const {
            std::stringstream ss;
            ss << *this;
            return ss.str();
        }
};


} // end of namespace dai


#endif