Search code examples
c++enumsswitch-statementbinary-operators

c++ multiple enums in one function argument using bitwise or "|"


I recently came across some functions where you can pass multiple enums like this:

myFunction(One | Two);

Since I think this is a really elegant way I tried to implement something like that myself:

void myFunction(int _a){
    switch(_a){
        case One:
            cout<<"!!!!"<<endl;
            break;
        case Two:
            cout<<"?????"<<endl;
            break;
    }
}

now if I try to call the function with One | Two, I want that both switch cases get called. I am not really good with binary operators so I dont really know what to do. Any ideas would be great!

Thanks!


Solution

  • For that you have to make enums like :

    enum STATE {
      STATE_A = 1,
      STATE_B = 2,
      STATE_C = 4
    };
    

    i.e. enum element value should be in power of 2 to select valid case or if statement.

    So when you do like:

    void foo( int state) {
    
      if ( state & STATE_A ) {
        //  do something 
      }
    
      if ( state & STATE_B ) {
        //  do something 
      }
    
      if ( state & STATE_C ) {
        //  do something 
      }   
    }
    
    int main() {
      foo( STATE_A | STATE_B | STATE_C);
    }