NSMutableArray+SKBlock - Initialize an array using a block

NSMutableArray+SKBlock.h

#import <Foundation/Foundation.h>

@interface NSMutableArray (SKBlock)

+(id)arrayWithObjectBlock:(id (^)(NSInteger index))block;
-(id)initWithObjectBlock:(id (^)(NSInteger index))block;

@end

NSMutableArray+SKBlock.m

#import "NSMutableArray+SKBlock.h"

@implementation NSMutableArray (SKBlock)

+(id)arrayWithObjectBlock:(id (^)(NSInteger index))block {
    return [[NSMutableArray alloc] initWithObjectBlock:block];
}

-(id)initWithObjectBlock:(id (^)(NSInteger index))block {
    NSInteger counter = 0;
    NSMutableArray *newArray = [NSMutableArray array];
    while (YES) {
        id obj = block(counter++);
        if (!obj)
            break;
        [newArray addObject:obj];
    }
    return newArray;
}

@end