Быстрое перечисление в категории NSArray для NSIntegers

поскольку я часто использую массивы NSInteger, я написал категорию для NSArray (и для NSMutableArray тоже), которая добавляет такие методы, как integerAtIndex:, arrayByAddingInteger: и т. д. Методы заботятся об обертывании / разворачивании NSInteger в объекте NSNumber.

Мне интересно, есть ли способ улучшить мою категорию, чтобы я мог сделать быстрое перечисление на NSIntegers. Хотелось бы уметь писать:

NSArray* arrayOfIntegers;
    .
    .
    .

for(NSInteger nextInteger in arrayOfIntegers)
    {
    }

....так что "nextInteger" извлекается из объекта NSNumber за кулисами. Могу я это сделать?

3 ответов


Я сомневаюсь, что есть очистить способ сделать это с помощью NSFastEnumeration, так как это сильно зависит от nextObject метод.

но вы можете сделать это по-другому, добавив метод блока для него:

@interface NSArray (Integers)
-(void)eachInteger:(void(^)(NSInteger))block;
@end

@implementation NSArray (Integers)
-(void)eachInteger:(void(^)(NSInteger))block {
  for (NSNumber *num in self) {
    block(num.integerValue);
  }
}
@end

таким образом, вы можете использовать его в коде подобным образом:

NSArray *arr = [NSArray arrayWithObjects:[NSNumber numberWithInt:23],
                                         [NSNumber numberWithInt:42],
                                         nil];
...
[arr eachInteger:^(NSInteger i) {
  NSLog(@"The int is %i", i);
}];
// =>
//    The int is 23
//    The int is 42

возможно, вы захотите взглянуть на категории NSArray на Lumumba Framework, который, оказывается, написали мной: D


этого точно нельзя сделать, но вы можете легко преобразовать свой NSNumber в NSInteger и использовать его позже. Вы даже можете написать для него макрос:

#define int_enum(var, arr, block) \
    for(NSNumber *__tmp in arr) { NSInteger var = [__tmp integerValue]; block }

использовать его как:

NSArray *array = // whatever;
int_enum(counter, array, {
    // things you want to do with `counter' as an NSInteger
});

Если вам действительно нравятся блоки, попробуйте следующее:

@interface NSArray(blockIteration)

@property (copy, nonatomic, readonly) void (^forEachObject)(void (^block)(NSArray *, id));
@property (copy, nonatomic, readonly) void (^forEachInt)(void (^block)(NSArray *, int));
@property (copy, nonatomic, readonly) void (^forEachDouble)(void (^block)(NSArray *, double));

@end

@implementation NSArray(blockIteration)

-(void (^)(void (^)(NSArray *, id))) forEachObject
{
    return [^(void (^block)(NSArray *, id)) {
        block = [block copy];
        for (id obj in self)
        {
            block(self, obj);
        }
    } copy];
}

-(void (^)(void (^)(NSArray *, int))) forEachInt
{
    return [^(void (^block)(NSArray *, int)) {
        block = [block copy];
        for (NSNumber *num in self)
        {
            block(self, [num intValue]);
        }
    } copy];
}

-(void (^)(void (^)(NSArray *, double))) forEachDouble
{
    return [^(void (^block)(NSArray *, double)) {
        block = [block copy];
        for (NSNumber *num in self)
        {
            block(self, [num doubleValue]);
        }
    } copy];
}


@end

int main()
{
    NSArray *array = [NSArray arrayWithObjects:@"Hello", @"World", @"This", @"Is", @"A", @"Test", nil];

    array.forEachObject(^(id arr, id obj) {
        NSLog(@"%@", obj);
    });
}

обратите внимание, что эта реализация зависит от ARC.