Какао: словарь с ключами перечисления?

Поскольку перечисления являются целыми числами, вы можете заключить перечисление в NSNumber. Когда вы добавляете / извлекаете что-то на / из карты, вы передаете перечисление конструктору NSNumber ...

Предполагая, что у вас есть перечисление вроде ...

enum ETest {
    FOO, BAR
};

Вы можете использовать его в NSDictionary вот так ...

NSMutableDictionary *dict = [[NSMutableDictionary alloc] init];
[dict setObject: @"Foo!" forKey:[NSNumber numberWithInt: FOO]];
NSLog(@"getting value for FOO -> %@", 
      [dict objectForKey: [NSNumber numberWithInt: FOO]]);  
[dict release]; 

С предложением VoidPointer может быть лучше использовать NSValue в тех случаях, когда перечисления оказываются не целыми (например, когда -fshort-enums находится в игре, что никогда не должно быть, поскольку вы, вероятно, нарушите совместимость с Foundation).

NSValue *value = [NSValue value: &myEnum withObjCType: @encode(enum ETest)];

Это не собирается здесь ничего добавлять, но дает вам общую технику «Я хочу использовать‹ имя не-ObjC-типа ›в классе коллекции».

Обратите внимание, что с помощью современных компиляторов вы можете указать перечислениям использовать фиксированный базовый тип. Это означает, что вы можете контролировать, какое хранилище используется для перечисления, но, поскольку приведенное выше решение является общим, оно все еще применяется, даже если вы это знаете.

Продолжая предложение Грэма Ли ...

Вы можете использовать objective-c category, чтобы добавить в NSMutableDictionary метод, который позволяет вам добавлять значение с ключом вашего типа, отличного от NSObject. Это избавляет ваш код от синтаксиса упаковки / развертывания.

Опять же, предполагая

enum ETest { FOO, BAR };

Во-первых, мы добавляем конструктор убеждения в NSValue:

@interface NSValue (valueWithETest)
+(NSValue*) valueWithETest:(enum ETest)etest; 
@end

@implementation NSValue (valueWithETest)
+(NSValue*) valueWithETest:(enum ETest)etest
{
    return [NSValue value: &etest withObjCType: @encode(enum ETest)];
}
@end

Затем мы добавим поддержку enum ETest в NSMutableDictionary.

@interface NSMutableDictionary (objectForETest)
-(void) setObject:(id)anObject forETest:(enum ETest)key;
-(id) objectForETest:(enum ETest)key;
@end

@implementation NSMutableDictionary (objectForETest)
-(void) setObject:(id)anObject forETest:(enum ETest)key
{
    [self setObject: anObject forKey:[NSValue valueWithETest:key]];
}
-(id) objectForETest:(enum ETest)key
{
    return [self objectForKey:[NSValue valueWithETest:key]];
}
@end

Таким образом, исходный пример можно преобразовать в

NSMutableDictionary *dict = [[NSMutableDictionary alloc] init];
[dict setObject: @"Bar!" forETest:BAR];
NSLog(@"getting value Bar -> %@", [dict objectForETest: BAR]);      
[dict release];

В зависимости от того, насколько часто вы используете перечисление для доступа к словарю, это может немного облегчить читаемость вашего кода.

перечисления не соответствуют NSCopying

Это преуменьшение; перечисления ни с чем не «согласуются», поскольку они не являются объектами; это примитивные значения C, которые взаимозаменяемы с целыми числами. Это настоящая причина, по которой их нельзя использовать в качестве ключей. Ключи и значения NSDictionary должны быть объектами. Но поскольку перечисления являются целыми числами, вы можете просто заключить их в NSNumber объекты. Это, наверное, самый простой вариант.

Другой вариант, если перечисления непрерывны от 0 до некоторого числа (т.е. вы не устанавливали никаких значений вручную), заключается в том, что вы можете использовать NSArray, где индекс представляет значение ключевого перечисления. (Любые «отсутствующие» записи должны быть заполнены NSNull.)

У категориального подхода есть свои собственные применения, но новые упакованные выражения (например, @(FOO)) должны позаботиться о преобразовании типов за вас. Он работает очень прозрачно, явно упаковывая перечисление при использовании его в качестве ключа.