-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathEnumIO.h
95 lines (84 loc) · 2.47 KB
/
EnumIO.h
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
/**
* \file EnumIO.h
* \brief
*
* \review
*/
#include <StdStream/StdStream.h>
#include <StdTest/StdTest.h>
#include <Stl.h>
//-------------------------------------------------------------------------------------------------
// This is the type that will hold all the strings.
// Each enumeration type will declare its own specialization.
// Any enum that does not have a specialization will generate a compiler error
// indicating that there is no definition of this variable (as there should be
// be no definition of a generic version).
template<typename T>
struct enumStrings
{
static char const* data[];
};
//-------------------------------------------------------------------------------------------------
// This is a utility type.
// Created automatically. Should not be used directly.
template<typename T>
struct enumRefHolder
{
T& enumVal;
enumRefHolder(T& enumVal) :
enumVal(enumVal)
{
}
};
template<typename T>
struct enumConstRefHolder
{
T const& enumVal;
enumConstRefHolder(T const& enumVal) :
enumVal(enumVal)
{
}
};
//-------------------------------------------------------------------------------------------------
// The next two functions do the actual work of reading/writing an
// enum as a string.
template<typename T>
std::ostream &
operator << (std::ostream& str, enumConstRefHolder<T> const& data)
{
return str << enumStrings<T>::data[data.enumVal];
}
template<typename T>
std::istream &
operator >> (std::istream& str, enumRefHolder<T> const& data)
{
std::string value;
str >> value;
// These two can be made easier to read in C++11
// using std::begin() and std::end()
//
static auto begin = std::begin(enumStrings<T>::data);
static auto end = std::end(enumStrings<T>::data);
auto find = std::find(begin, end, value);
if (find != end) {
data.enumVal = static_cast<T>(std::distance(begin, find));
}
return str;
}
//-------------------------------------------------------------------------------------------------
// This is the public interface:
// use the ability of function to deduce their template type without
// being explicitly told to create the correct type of enumRefHolder<T>
template<typename T>
enumConstRefHolder<T>
enumToString(T const& e)
{
return enumConstRefHolder<T>(e);
}
template<typename T>
enumRefHolder<T>
enumFromString(T& e)
{
return enumRefHolder<T>(e);
}
//-------------------------------------------------------------------------------------------------