NSArray+pmap.m

@import Foundation.NSPointerArray;
#import "YOLO.ph"

@implementation NSArray (YOLOPMap)

- (NSArray *(^)(id (^)(id o)))pmap {
    return ^NSArray *(id (^block)(id))
    {
        NSPointerArray *aa = [NSPointerArray new];
        aa.count = self.count;
        [self enumerateObjectsWithOptions:NSEnumerationConcurrent usingBlock:^(id obj, NSUInteger idx, BOOL *stop) {
            id o = block(obj);
            [aa replacePointerAtIndex:idx withPointer:(__bridge_retained void *)o];
        }];

        [aa compact];
        NSArray *rv = aa.allObjects;

        for (int x = 0; x < aa.count; ++x)
            CFRelease([aa pointerAtIndex:x]);

        return rv;
    };
}

@end