Cocoa: ¿diccionario con claves enum?

Question

Necesito crear un diccionario/HashMap donde los

• llaves son enumeraciones
• Los valores son algunos subclase de NSObject

NSDictionary no funcionará aquí (enumeraciones que no se ajusten a NSCopying)

Quizás podría utilizar un CFDictionaryRef aquí, pero me gustaría saber si hay alguna otra forma de lograr esto.

Answer 1

Dado que las enumeraciones son números enteros, puede ajustar la enumeración en un NSNumber. Cuando se agrega/retreive algo a/desde el mapa, se pasa la enumeración al constructor NSNumber ...

Suponiendo que tienes una enumeración como ...

enum ETest { 
    FOO, BAR 
}; 

Usted puede utilizarlo en un NSDictionary así ...

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

Answer 2

enums don't conform to NSCopying

este es un eufemismo; las enumeraciones no se "conforman" a nada ya que no son objetos; son valores primitivos de C que son intercambiables con enteros. Esa es la verdadera razón por la que no pueden usarse como llaves. Las claves y valores de NSDictionary deben ser objetos. Pero como las enumeraciones son números enteros, puedes envolverlas en objetos NSNumber. Esta es probablemente la opción más simple.

Otra opción, si las enumeraciones son contiguas desde 0 hasta un cierto número (es decir, no se establecieron valores manualmente), es que puede usar un NSArray donde el índice representa el valor de la clave enum. (Cualquier "perdidos" entradas tendrían que ser llenado con NSNull.)

Answer 3

Con la sugerencia de VoidPointer, puede ser mejor usar NSValue para aquellos momentos en enumeraciones resultan no ser enteros (como cuando -fshort-enums está en juego, que nunca debería ser como probablemente rompería la compatibilidad con Foundation).

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

eso no va a añadir mucho aquí, pero le da al general "Quiero usar < nombre de la no ObjC tipo > en una clase de colección" técnica.

Observe que con compiladores modernos puede decir enums to use a fixed underlying type. Esto significa que puede controlar qué almacenamiento se utiliza para la enumeración, pero como la solución anterior es general, aún se aplica incluso cuando usted sabe esto.

Answer 4

adicional que se extiende sobre la sugerencia de Graham Lee ...

Se podría utilizar un objetivo-c category con el fin de añadir un método para NSMutableDictionary que le permite añadir un valor con una clave de su tipo no NSObject . Esto mantiene su código libre de la sintaxis de envolver/desenvolver.

Una vez más, asumiendo

enum ETest { FOO, BAR }; 

En primer lugar, vamos a añadir un constructor convencer a 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 

A continuación vamos a añadir soporte 'enumeración eTest' a 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 

El ejemplo original se puede transformar a

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

Dependiendo de cuánto use su enumeración para acceder al diccionario, esto puede facilitar bastante la legibilidad de su código.

Answer 5

Me gusta el enfoque de categoría también y probablemente implemente esto también en mi caso. Con las nuevas expresiones de Literals/boxeo como se llame, ¿usarías @ (FOO) para encargarte del boxeo?

Creo que sí, y si es así, funciona de manera muy transparente al encasillar explícitamente la enumeración cuando se usa como clave.