¿Escribe Flags de bits Safe (R) en C ++?
-
26-09-2019 - |
Pregunta
Mientras revisaba un código antiguo de C ++, me encontré con varios bitflags definido como enums.
enum FooFlags
{
FooFlag1 = 1 << 0,
FooFlag2 = 1 << 1,
FooFlag3 = 1 << 2
// etc...
};
Esto no es raro, pero me molestó que tan pronto como comiences a combinar banderas, pierdes la información de tipo.
int flags = FooFlag1 | FooFlag2; // We've lost the information that this is a set of flags relating to *Foo*
Algunos que buscaban así demostró que no soy el solamente una molesto por esto.
Una alternativa es declarar las banderas como #defines o integrales const, por lo que las operaciones bit a bit no transformarían el tipo (probablemente). El problema con esto es que permite que nuestro conjunto de bits se mezcle con banderas no relacionadas, a través de INTS u otros enums.
Estoy familiarizado con std :: bitset y Boost :: Dynamic_Bitset, pero ninguno está diseñado para abordar mi problema. Lo que estoy buscando es algo como C# Flagsattribute.
Mi pregunta es, ¿qué otras soluciones hay para un conjunto (más) de tipo seguro de bitflags?
Publicaré mi propia solución a continuación.
Solución 2
Aquí está mi propia solución, usando elementos de C ++ 0x que la versión actual de VS2010 permite:
#include <iostream>
#include <numeric>
#include <string>
#include <initializer_list>
template <typename enumT>
class FlagSet
{
public:
typedef enumT enum_type;
typedef decltype(enumT()|enumT()) store_type;
// Default constructor (all 0s)
FlagSet() : FlagSet(store_type(0))
{
}
// Initializer list constructor
FlagSet(const std::initializer_list<enum_type>& initList)
{
// This line didn't work in the initializer list like I thought it would. It seems to dislike the use of the lambda. Forbidden, or a compiler bug?
flags_ = std::accumulate(initList.begin(), initList.end(), store_type(0), [](enum_type x, enum_type y) { return x | y; })
}
// Value constructor
explicit FlagSet(store_type value) : flags_(value)
{
}
// Explicit conversion operator
operator store_type() const
{
return flags_;
}
operator std::string() const
{
return to_string();
}
bool operator [] (enum_type flag) const
{
return test(flag);
}
std::string to_string() const
{
std::string str(size(), '0');
for(size_t x = 0; x < size(); ++x)
{
str[size()-x-1] = (flags_ & (1<<x) ? '1' : '0');
}
return str;
}
FlagSet& set()
{
flags_ = ~store_type(0);
return *this;
}
FlagSet& set(enum_type flag, bool val = true)
{
flags_ = (val ? (flags_|flag) : (flags_&~flag));
return *this;
}
FlagSet& reset()
{
flags_ = store_type(0);
return *this;
}
FlagSet& reset(enum_type flag)
{
flags_ &= ~flag;
return *this;
}
FlagSet& flip()
{
flags_ = ~flags_;
return *this;
}
FlagSet& flip(enum_type flag)
{
flags_ ^= flag;
return *this;
}
size_t count() const
{
// http://www-graphics.stanford.edu/~seander/bithacks.html#CountBitsSetKernighan
store_type bits = flags_;
size_t total = 0;
for (; bits != 0; ++total)
{
bits &= bits - 1; // clear the least significant bit set
}
return total;
}
/*constexpr*/ size_t size() const // constexpr not supported in vs2010 yet
{
return sizeof(enum_type)*8;
}
bool test(enum_type flag) const
{
return (flags_ & flag) > 0;
}
bool any() const
{
return flags_ > 0;
}
bool none() const
{
return flags == 0;
}
private:
store_type flags_;
};
template<typename enumT>
FlagSet<enumT> operator & (const FlagSet<enumT>& lhs, const FlagSet<enumT>& rhs)
{
return FlagSet<enumT>(FlagSet<enumT>::store_type(lhs) & FlagSet<enumT>::store_type(rhs));
}
template<typename enumT>
FlagSet<enumT> operator | (const FlagSet<enumT>& lhs, const FlagSet<enumT>& rhs)
{
return FlagSet<enumT>(FlagSet<enumT>::store_type(lhs) | FlagSet<enumT>::store_type(rhs));
}
template<typename enumT>
FlagSet<enumT> operator ^ (const FlagSet<enumT>& lhs, const FlagSet<enumT>& rhs)
{
return FlagSet<enumT>(FlagSet<enumT>::store_type(lhs) ^ FlagSet<enumT>::store_type(rhs));
}
template <class charT, class traits, typename enumT>
std::basic_ostream<charT, traits> & operator << (std::basic_ostream<charT, traits>& os, const FlagSet<enumT>& flagSet)
{
return os << flagSet.to_string();
}
La interfaz se modela después std :: bitset. Mi objetivo era ser fiel al espíritu de C ++ de seguridad de tipo y una sobrecarga mínima (si la hay). Agradecería cualquier comentario sobre mi implementación.
Aquí hay un ejemplo mínimo:
#include <iostream>
enum KeyMod
{
Alt = 1 << 0, // 1
Shift = 1 << 1, // 2
Control = 1 << 2 // 4
};
void printState(const FlagSet<KeyMod>& keyMods)
{
std::cout << "Alt is " << (keyMods.test(Alt) ? "set" : "unset") << ".\n";
std::cout << "Shift is " << (keyMods.test(Shift) ? "set" : "unset") << ".\n";
std::cout << "Control is " << (keyMods.test(Control) ? "set" : "unset") << ".\n";
}
int main(int argc, char* argv[])
{
FlagSet<KeyMod> keyMods(Shift | Control);
printState(keyMods);
keyMods.set(Alt);
//keyMods.set(24); // error - an int is not a KeyMod value
keyMods.set(Shift);
keyMods.flip(Control);
printState(keyMods);
return 0;
}
Otros consejos
Puede sobrecargar operadores para los tipos de enumeración que devuelven el resultado mecanografiado adecuado.
inline FooFlags operator|(FooFlags a, FooFlags b) {
return static_cast<FooFlags>(+a | +b);
}
Cabe señalar que para ser teóricamente seguro, debe declarar manualmente el valor más alto posible para que el rango del tipo de enumeración se garantice que atrapará todas las combinaciones.
- En realidad, eso no es necesario: el rango de una enumeración siempre podrá atrapar toda la combinación, porque el valor positivo más alto del rango de una enumeración es siempre
(2^N)-1
Por el primeroN
poder representar al enumerador más alto. Ese valor tiene todos los bits 1.
Pensé que podría agregar una versión C ++ 11 para enum class
FooFlags operator|(FooFlags a, FooFlags b)
{
typedef std::underlying_type<FooFlags>::type enum_type;
return static_cast<FooFlags>(static_cast<enum_type>(a) | static_cast<enum_type>(b));
}
Si la versión C ++ 11 lo admite, supongo que esto sería un candidato principal para constexpr