1 /*---------------------------------------------------------------------------*
2   Project:  Horizon
3   File:     util_FlagsEnum.h
4 
5   Copyright (C)2009 Nintendo Co., Ltd.  All rights reserved.
6 
7   These coded instructions, statements, and computer programs contain
8   proprietary information of Nintendo of America Inc. and/or Nintendo
9   Company Ltd., and are protected by Federal copyright law.  They may
10   not be disclosed to third parties or copied or duplicated in any form,
11   in whole or in part, without the prior written consent of Nintendo.
12 
13   $Rev: 31068 $
14  *---------------------------------------------------------------------------*/
15 
16 #ifndef NN_UTIL_UTIL_FLAGSENUM_H_
17 #define NN_UTIL_UTIL_FLAGSENUM_H_
18 
19 #include <nn/types.h>
20 #include <nn/util/util_StaticAssert.h>
21 
22 #ifdef __cplusplus
23 
24 namespace nn { namespace util {
25 
26 template <typename StorageT, typename EnumT>
27 class FlagsEnum
28 {
29 private:
30     typedef FlagsEnum<StorageT, EnumT> Self;
31     NN_STATIC_ASSERT(sizeof(EnumT) <= sizeof(StorageT));
32 
33 private:
34     StorageT        m_FlagsValue;
35 
36 public:
EnumT()37     operator EnumT() const      { return Get(); }
38 
39     Self& operator =(bit32 v)   { Replace(v); return *this; }
40     Self& operator =(int v)     { Replace(v); return *this; }
41 
Replace(bit32 v)42     void Replace(bit32 v)       { m_FlagsValue = static_cast<StorageT>(v); }
Replace(int v)43     void Replace(int v)         { m_FlagsValue = static_cast<StorageT>(v); }
44 
Get()45     bit32 Get() const           { return m_FlagsValue; }
46 
Set(EnumT e)47     void Set(EnumT e)           { m_FlagsValue |= e; }
Clear(EnumT e)48     void Clear(EnumT e)         { m_FlagsValue &= ~e; }
IsSet(EnumT e)49     bool IsSet(EnumT e) const   { return (m_FlagsValue & e) != 0; }
50 };
51 
52 template <typename EnumT>
53 class FlagsEnum1 : public FlagsEnum<bit8, EnumT>
54 {
55 public:
FlagsEnum1()56     FlagsEnum1() : FlagsEnum<bit8, EnumT>() {}
FlagsEnum1(EnumT e)57     FlagsEnum1(EnumT e) : FlagsEnum<bit8, EnumT>(e) {}
58 
59     FlagsEnum1<EnumT>& operator =(bit32 v)   { this->Replace(v); return *this; }
60     FlagsEnum1<EnumT>& operator =(int v)     { this->Replace(v); return *this; }
61 };
62 
63 template <typename EnumT>
64 class FlagsEnum2 : public FlagsEnum<bit16, EnumT>
65 {
66 public:
FlagsEnum2()67     FlagsEnum2() : FlagsEnum<bit16, EnumT>() {}
FlagsEnum2(EnumT e)68     FlagsEnum2(EnumT e) : FlagsEnum<bit16, EnumT>(e) {}
69 
70     FlagsEnum2<EnumT>& operator =(bit32 v)   { this->Replace(v); return *this; }
71     FlagsEnum2<EnumT>& operator =(int v)     { this->Replace(v); return *this; }
72 };
73 
74 template <typename EnumT>
75 class FlagsEnum4 : public FlagsEnum<bit32, EnumT>
76 {
77 public:
FlagsEnum4()78     FlagsEnum4() : FlagsEnum<bit32, EnumT>() {}
FlagsEnum4(EnumT e)79     FlagsEnum4(EnumT e) : FlagsEnum<bit32, EnumT>(e) {}
80 
81     FlagsEnum4<EnumT>& operator =(bit32 v)   { this->Replace(v); return *this; }
82     FlagsEnum4<EnumT>& operator =(int v)     { this->Replace(v); return *this; }
83 };
84 
85 }}
86 
87 
88 #endif // __cplusplus
89 
90 #endif /* NN_UTIL_UTIL_FLAGSENUM_H_ */
91