Enum as Flag используя, устанавливая и сдвигая - PullRequest
3 голосов
/ 24 февраля 2009

У меня есть два флага:

[Flags]
enum Flags
{
  A = 1,
  B = 2
};

Я установил их так:

Mode = Flags.A | Flags.B; // default value
for(int i = 0; i < args.Length; i++) {
switch(args[i])
{
  case "--a":
  {
    if ((Mode & Flags.A) == Flags.A && (Mode & Flags.B) == Flags.B) 
     // both, default assume
    {
      Mode = Flags.A; // only A
    }
    else
    {
      Mode |= Flags.A; // append A
    }
    break;
  }
  case "--b":
  {
    if ((Mode & Flags.A) == Flags.A && (Mode & Flags.B) == Mode.B)
    {
      Mode = Flags.B;
    }
    else
    {
      Mode |= Flags.B;
    }
    break;
  }
} }

и используйте их позже так:

if((Mode & Flags.A) == Flags.A)
{
 //
}
if((Mode & Flags.B) == Flags.B)
{
 //
}

Главное резервирование: оба флага могут быть установлены. Или только один, в этом случае выполняется только одна часть кода.

Мой код хорош? Каков наилучший способ установки флагов?

Upd : Разве это менее уродливо, во-первых, как вы думаете?

Mode = 0; // default value now is empty
for(int i = 0; i < args.Length; i++) {
switch(args[i])
{
  case "--a":
  {
    Mode |= Flags.A;
    break;
  }
  case "--b":
  {
    Mode |= Flags.B;
    break;
  }
} }
if(Mode == 0)
{
  Mode = Flags.A | Flags.B; // if no parameters are given, setup both flags
}

Ответы [ 3 ]

8 голосов
/ 24 февраля 2009

Вот как я бы установил свои флаги:

Mode = 0;
for(int i = 0; i < args.Length; i++) {
    switch(args[i]) {
    case "--a":
        Mode |= Flags.A;
        break;
    case "--b":
        Mode |= Flags.B;
        break;
    }
}

Если оба флага включены по умолчанию, я думаю, что имеет смысл изменить параметры командной строки на что-то вроде --not-a и --not-b. Это будет отражать настройки по умолчанию и позволит вам избавиться от (Mode & Flags.A) == Flags.A && (Mode & Flags.B) == Flags.B, что довольно уродливо, ИМХО.

Тогда вы можете установить свои флаги так:

Mode = Flags.A | Flags.B;
for(int i = 0; i < args.Length; i++) {
    switch(args[i]) {
    case "--not-a":
        Mode &= ~Flags.A;
        break;
    case "--not-b":
        Mode &= ~Flags.B;
        break;
    }
}

Наконец, если у вас много флагов (вместо двух), может быть проще настроить перечисление следующим образом:

[Flags]
enum Flags
{
    A = 1,
    B = 1 << 1,
    C = 1 << 2,
    D = 1 << 3,
    E = 1 << 4,
    F = 1 << 5
};
2 голосов
/ 24 февраля 2009

Вы можете отключить «бит» с помощью следующего замечательного утверждения:

Mode &= ~Flags.A;

Я бы рекомендовал включить в ваше перечисление "нулевое" значение:

[Flags]
enum Flags
{
  Null = 0;
  A = 1,
  B = 2;
}

Это сделает вашу жизнь проще! : -)

1 голос
/ 24 февраля 2009

Вторая версия намного лучше - это именно то, что я бы сделал. Измените Mode == null на Mode == 0, хотя.

Mode = 0; // default value now is empty
for(int i = 0; i < args.Length; i++) {
    switch(args[i])
    {
        case "--a":
            Mode |= Flags.A;
            break;

        case "--b":
            Mode |= Flags.B;
            break;
    }
}

if(Mode == 0)
{
    Mode = Flags.A | Flags.B; // if no parameters are given, setup both flags
}
...