enum.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_enum_h
#define __defined_libdai_enum_h


#include <cstring>
#include <iostream>
#include <dai/exceptions.h>



#define DAI_ENUM(x,val0,...) class x {\
    public:\
        enum value {val0,__VA_ARGS__};\
\
        x() : v(val0) {}\
\
        x(value w) : v(w) {}\
\
        x(char const *w) {\
            static char const* labelstring = #val0 "," #__VA_ARGS__;\
            size_t pos_begin = 0;\
            size_t i = 0;\
            for( size_t pos_end = 0; ; pos_end++ ) {\
                if( (labelstring[pos_end] == ',') || (labelstring[pos_end] == '\0') ) {\
                    if( (strlen( w ) == pos_end - pos_begin) && (strncmp( labelstring + pos_begin, w, pos_end - pos_begin ) == 0) ) {\
                        v = (value)i;\
                        return;\
                    } else {\
                        i++;\
                        pos_begin = pos_end + 1;\
                    }\
                }\
                if( labelstring[pos_end] == '\0' )\
                    break;\
            }\
            DAI_THROWE(UNKNOWN_ENUM_VALUE,"'" + std::string(w) + "' is not in [" + std::string(labelstring) + "]");\
        }\
\
        operator value() const { return v; }\
\
        operator size_t() const { return (size_t)v; }\
\
        operator char const*() const {\
            static char labelstring[] = #val0 "," #__VA_ARGS__;\
            size_t pos_begin = 0;\
            size_t i = 0;\
            for( size_t pos_end = 0; ; pos_end++ )\
                if( (labelstring[pos_end] == ',') || (labelstring[pos_end] == '\0') ) {\
                    if( (size_t)v == i ) {\
                        labelstring[pos_end] = '\0';\
                        return labelstring + pos_begin;\
                    } else {\
                        i++;\
                        pos_begin = pos_end + 1;\
                    }\
                }\
        }\
\
        operator wchar_t const*() const {\
            static wchar_t labelstring[] = #val0 L"," #__VA_ARGS__;\
            size_t pos_begin = 0;\
            size_t i = 0;\
            for( size_t pos_end = 0; ; pos_end++ )\
                if( (labelstring[pos_end] == L',') || (labelstring[pos_end] == L'\0') ) {\
                    if( (size_t)v == i ) {\
                        labelstring[pos_end] = L'\0';\
                        return labelstring + pos_begin;\
                    } else {\
                        i++;\
                        pos_begin = pos_end + 1;\
                    }\
                }\
        }\
\
        friend std::istream& operator >> (std::istream& is, x& y) {\
            std::string s;\
            is >> s;\
            y = x(s.c_str());\
            return is;\
        }\
\
        friend std::wistream& operator >> (std::wistream& wis, x& y) {\
            std::wstring s;\
            wis >> s;\
            y = x((const char *)s.c_str());\
            return wis;\
        }\
\
        friend std::ostream& operator << (std::ostream& os, const x& y) {\
            os << (const char *)y;\
            return os;\
        }\
\
        friend std::wostream& operator << (std::wostream& wos, const x& y) {\
            wos << (const wchar_t *)y;\
            return wos;\
        }\
\
    protected:\
        value v;\
};


#endif