The Problem
C++ does not have a facility to allow one enum type to be extended by inheritance as it does for classes and structures. Sometimes it is important to have this functionality. Suppose you had:
enum Fruit { Orange, Mango, Banana };
#include "Fruit.h"
void eat(Fruit fruit);
If you wanted to add another fruit type, you could extend Fruit as follows:
enum Fruit { Orange, Mango, Banana, Apple, Pear };
You could also have another function that handles the extra fruit:
void consume(Fruit fruit);
There are two problems with this approach:
- Fruit.h may be a library file that you don't necessarily want to change.
void eat(Fruit)
is implemented in some library that you can't change and as a result it might not handle 'Apple' properly but will still compile without errors. Your consume(Fruit)
function may know about 'Apple' and handle it properly but library users, even though it is not your intention, may still call eat(Apple)
with undefined behaviour.
In summary, the results of calls to eat()
and consume()
are as follows:
eat( Orange );
consume( Orange );
eat( Apple );
consume( Apple );
The Solution
InheritEnum
solves this problem by allowing you to leave the first enum declaration as is and add another enum declaration with new enum types.
Following on from our example, to handle new fruits as well as the first set of fruits, we will then have:
#include "Fruit.h"
#include "InheritEnum.h"
enum NewFruits { Apple, Pear };
typedef InheritEnum< NewFruit, Fruit > MyFruit;
Now our consume()
declaration becomes:
void consume(MyFruit myfruit);
Now, our call summary looks as follows:
eat( Orange );
consume( Orange );
eat( Apple );
consume( Apple );
The Code
template <typename EnumT, typename BaseEnumT>
class InheritEnum
{
public:
InheritEnum() {}
InheritEnum(EnumT e)
: enum_(e)
{}
InheritEnum(BaseEnumT e)
: baseEnum_(e)
{}
explicit InheritEnum( int val )
: enum_(static_cast<EnumT>(val))
{}
operator EnumT() const { return enum_; }
private:
union
{
EnumT enum_;
BaseEnumT baseEnum_;
};
};
Thank you.