properties.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_properties_h
#define __defined_libdai_properties_h


#include <iostream>
#include <sstream>
#include <boost/any.hpp>
#include <map>
#include <vector>
#include <typeinfo>
#include <dai/exceptions.h>
#include <dai/util.h>
#include <boost/lexical_cast.hpp>


namespace dai {


typedef std::string PropertyKey;

typedef boost::any  PropertyValue;

typedef std::pair<PropertyKey, PropertyValue> Property;



std::ostream& operator<< ( std::ostream & os, const Property &p );



class PropertySet : private std::map<PropertyKey, PropertyValue> {
    public:

        PropertySet() {}


        PropertySet( const std::string& s ) {
            std::stringstream ss;
            ss << s;
            ss >> *this;
        }


        PropertySet& set( const PropertyKey& key, const PropertyValue& val ) { 
            this->operator[](key) = val; 
            return *this; 
        }

        PropertySet& set( const PropertySet& newProps ) {
            const std::map<PropertyKey, PropertyValue> *m = &newProps;
            bforeach(value_type i, *m)
                set( i.first, i.second );
            return *this;
        }


        PropertySet operator()( const PropertyKey& key, const PropertyValue& val ) const { 
            PropertySet copy = *this; 
            return copy.set(key,val); 
        }


        template<typename ValueType>
        PropertySet& setAsString( const PropertyKey& key, const ValueType& val ) {
            try {
                return set( key, boost::lexical_cast<std::string>(val) );
            } catch( boost::bad_lexical_cast & ) {
                DAI_THROWE(IMPOSSIBLE_TYPECAST,"Cannot cast value of property '" + key + "' to string.");
            }
        }


        template<typename ValueType>
        void convertTo( const PropertyKey& key ) { 
            PropertyValue val = get(key);
            if( val.type() != typeid(ValueType) ) {
                DAI_ASSERT( val.type() == typeid(std::string) );
                try {
                    set(key, boost::lexical_cast<ValueType>(getAs<std::string>(key)));
                } catch(boost::bad_lexical_cast &) {
                    DAI_THROWE(IMPOSSIBLE_TYPECAST,"Cannot cast value of property '" + key + "' from string to desired type.");
                }
            }
        }



        size_t size() const {
            return std::map<PropertyKey, PropertyValue>::size();
        }

        void clear() {
            std::map<PropertyKey, PropertyValue>::clear();
        }

        size_t erase( const PropertyKey &key ) {
            return std::map<PropertyKey, PropertyValue>::erase( key );
        }

        bool hasKey( const PropertyKey& key ) const { 
            PropertySet::const_iterator x = find(key); 
            return (x != this->end()); 
        }

        std::set<PropertyKey> keys() const {
            std::set<PropertyKey> res;
            const_iterator i;
            for( i = begin(); i != end(); i++ )
                res.insert( i->first );
            return res;
        }


        const PropertyValue& get( const PropertyKey& key ) const {
            PropertySet::const_iterator x = find(key);
            if( x == this->end() )
                DAI_THROWE(OBJECT_NOT_FOUND,"PropertySet::get cannot find property '" + key + "'");
            return x->second;
        }


        template<typename ValueType>
        ValueType getAs( const PropertyKey& key ) const {
            try {
                return boost::any_cast<ValueType>(get(key));
            } catch( const boost::bad_any_cast & ) {
                DAI_THROWE(IMPOSSIBLE_TYPECAST,"Cannot cast value of property '" + key + "' to desired type.");
                return ValueType();
            }
        }


        template<typename ValueType>
        ValueType getStringAs( const PropertyKey& key ) const { 
            PropertyValue val = get(key);
            if( val.type() == typeid(ValueType) ) {
                return boost::any_cast<ValueType>(val);
            } else if( val.type() == typeid(std::string) ) {
                try {
                    return boost::lexical_cast<ValueType>(getAs<std::string>(key));
                } catch(boost::bad_lexical_cast &) {
                    DAI_THROWE(IMPOSSIBLE_TYPECAST,"Cannot cast value of property '" + key + "' from string to desired type.");
                }
            } else
                DAI_THROWE(IMPOSSIBLE_TYPECAST,"Cannot cast value of property '" + key + "' from string to desired type.");
            return ValueType();
        }



        friend std::ostream& operator<< ( std::ostream& os, const PropertySet& ps );

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


        friend std::istream& operator>> ( std::istream& is, PropertySet& ps );
        
        void fromString( const std::string& s ) {
            std::stringstream ss( s );
            ss >> *this;
        }
};


} // end of namespace dai


#endif