原文：
https://www.alanzucconi.com/2015/07/26/enum-flags-and-bitwise-operators/
If you’re a game developer chances are you’re familiar with the need to describe different variations of an attribute. Whether it’s the type of an attack (melee, ice, fire, poison, …) or the state of an enemy AI (idle, alerted, chasing, attacking, resting, …) you can’t escape this. The most naive way of implementing this is simply by using constants:
public static int NONE = 0; public static int MELEE = 1; public static int FIRE = 2; public static int ICE = 3; public static int POISON = 4; public int attackType = NONE;
The downside is that you have no actual control over the values you can assign to attackType++.
The enum construct
Luckily, C# has a construct called enum (for enumeration) which has been specifically designed for these situations:
// Outside your class
public enum AttackType {
None,
Melee,
Fire,
Ice,
Poison
}
// Inside your class
public AttackType attackType = AttackType.None;
The definition of an enum creates a type which can support only a limited range or values. These values are given symbolic labels for clarity and are also returned as string when needed:
attackType = AttackType.Poison;
Debug.Log("Attack: " + attackType); # Prints "Attack: Poison"
Internally, every label has an integer value. Enums starts from zero and every new label is assigned the next integer number. Fire is two and so on. You can change that by explicitly changing the value of a label:
public enum AttackType {
None, // 0
Melee, // 1
Fire, // 2
Ice = 4, // 4
Poison // 5
}
Casting an enum to int will return its integer value. To be fair, enums are actual integers.
What makes enums even so interesting is the fact that they are automatically integrated in the Unity inspector. If a public field is an enum, it will conveniently appear like a dropdown menu:
Enums and Flags
The vast majority of developers use enums just as we’ve seen before. There’s much more we can do with them though. The first limitation is that standard enums can only hold a value at a time. What if we are using a melee weapon with a fire attack (a fire sword?!)? To solve this problem, enums can be decorated with ]. This allows them to be treated as bit masks, storing multiple values between them:
[Flags] public enum AttackType {
None = 0,
Melee = 1,
Fire = 2,
Ice = 4,
Poison = 8
}
// ...
public AttackType attackType = AttackType.Melee | AttackType.Fire;
In the example above, Fire values. We’ll see later how it is possible to retrieve these values. But first we need to understand how this is actually working. Enums are store as integers; when you have consecutive numbers, their bit representations look like this:
// Consecutive
public enum AttackType {
// Decimal // Binary
None = 0, // 000000
Melee = 1, // 000001
Fire = 2, // 000010
Ice = 3, // 000011
Poison = 4 // 000100
}
If we want to use 1 in its binary representation, and that they are all in different positions:
// Powers of two
[Flags] public enum AttackType {
// Decimal // Binary
None = 0, // 000000
Melee = 1, // 000001
Fire = 2, // 000010
Ice = 4, // 000100
Poison = 8 // 001000
}
At this stage, the variable Int32, it’s unwise to have an enum with more then 32 different labels.
To be fair, you can use enums even without ]. The only thing it does is allowing a nicer output ofenums when they are printed.
Bitwise operators
A bit mask is, essentially, an integer value in which several binary property (yes/no) are independently stored in its bit. In order to pack and unpack them we need some special operators. C# calls them bitwise operator, because they work on a bit to bit basis, ignoring carries unlikely addition and subtraction operators.
Bitwise OR
Setting a property is possible using the bitwise OR :
attackType = AttackType.Melee | AttackType.Fire; // OR attackType = AttackType.Melee; attackType |= AttackType.Fire;
What the bitwise OR does is setting the bit in the i-th position to 1.
// Label Binary Decimal // Melee: 000001 = 1 // Fire: 000010 = 2 // Melee | Fire: 000011 = 3
If you print Fire. This unfortunately doesn’t happen with the inspector; Unity doesn’t recognise the new type and simply leaves the field empty:
If you want the mixed type to appear, you’ll have to define it manually in the enum:
[Flags] public enum AttackType {
// Decimal // Binary
None = 0, // 000000
Melee = 1, // 000001
Fire = 2, // 000010
Ice = 4, // 000100
Poison = 8, // 001000
MeleeAndFire = Melee | Fire // 000011
}
Bitwise AND
The complementary operator to the bitwise OR is the bitwise AND. It works in the exact same way, with the exception that when applied with two integers it keeps only the bits which are set in both of them. While bitwise OR is used to set bits, bitwise AND is typically used to unpack property previously stores in an integer.
attackType = AttackType.Melee | AttackType.Fire; bool isIce = (attackType & AttackType.Ice) != 0;
Applying the bitwise AND between Ice; otherwise zero:
// Label Binary Decimal // Ice: 000100 = 4 // MeleeAndFire: 000011 = 3 // MeleeAndFire & Ice: 000000 = 0 // Fire: 000010 = 2 // MeleeAndFire: 000011 = 3 // MeleeAndFire & Fire: 000010 = 2
If bitwise operators are making your head spinning, .NET 4.0 has introduced the method HasFlag which can be conveniently used as follow:
attackType.HasFlag(AttackType.Ice);
Now, you need to be careful about the fact that 0 is always zero. A possible way to avoid this is to check against the original value:
bool isNone = (attackType & AttackType.None) == AttackType.None;
When it comes to always have a zero value in an enum.
Bitwise NOT
There is another useful bitwise operator, which is the bitwise NOT. What it does is simply inverting all the bits of an integer. This can be useful, for instance, to unset a bit. Let’s say we want our attack to stop being firey and become icy instead:
attackType = AttackType.Melee | AttackType.Fire attackType &= ~ AttackType.Fire; attackType |= AttackType.Ice;
By negating the property attackType, it will leave all the other bits unchanged and unset the fire property.
Bitwise XOR
After OR, AND and NOT, we cannot not mention the bitwise XOR. As the name suggest, it is used to xor bits in the same position of an integer variable. The xor (or exclusive or) of two binary values is true only if one or the other is true, but not both. This has a very important meaning for bit masks, since it allows to toggle a value.
attackType = AttackType.Melee | AttackType.Fire; attackType ^= AttackType.Fire; // Toggle fire attackType ^= AttackType.Ice; // Toggle ice
Bitwise shifts
The last two operators to work with bit masks are the bitwise shifts. Taken a number, they literally shift its bits right (>>) or left (<<). If you have a decimal number, let’s say “1” and you shift it of one position to the left, you’ll have “10”. Another shift and you’ll get “100”. If shifting one position in base ten is equivalent to multiply (or divide) by ten, shifting one position in base two is equivalent to multiply (or divide) by two. This is why bitwise shifts can be used to easily create powers of two:
[Flags] public enum AttackType {
// // Binary // Dec
None = 0, // 000000 0
Melee = 1 << 0, // 000001 1
Fire = 1 << 1, // 000010 2
Ice = 1 << 2, // 000100 4
Poison = 1 << 3, // 001000 8
}
Conclusion
This posts introduces enums, flags and bitwise operators. You can use the following functions to better work with them:
public static AttackType SetFlag (AttackType a, AttackType b)
{
return a | b;
}
public static AttackType UnsetFlag (AttackType a, AttackType b)
{
return a & (~b);
}
// Works with "None" as well
public static bool HasFlag (AttackType a, AttackType b)
{
return (a & b) == b;
}
public static AttackType ToogleFlag (AttackType a, AttackType b)
{
return a ^ b;
}
The technique described in this post is flexible enough to be adapted not just to attack types but also to inventory systems, item properties and finite state machines.
You should also check this other tutorial on how to extend the inspector for a more user friendly enum interface:
If instead you prefer a dropdown menu, you should check Unity’s EnumMaskField.