/
NSArray+Funjective.m
48 lines (41 loc) · 1.22 KB
/
NSArray+Funjective.m
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
//
// NSArray+Funjective.m
// funjective
//
// Created by alco on 23.04.12.
// Copyright (c) 2012 __MyCompanyName__. All rights reserved.
//
#import "NSArray+Funjective.h"
@implementation NSArray(Funjective)
- (void)each:(void (^)(id obj, unsigned index))block
{
[self enumerateObjectsUsingBlock:^(id obj, NSUInteger index, BOOL *stop) {
block(obj, index);
}];
}
- (NSArray *)map:(id (^)(id obj, unsigned index))block
{
NSMutableArray *result = [NSMutableArray arrayWithCapacity:[self count]];
[self enumerateObjectsUsingBlock:^(id obj, NSUInteger index, BOOL *stop) {
[result addObject:block(obj, index)];
}];
return result;
}
- (NSArray *)filter:(BOOL (^)(id obj, unsigned index))pred
{
NSMutableArray *result = [NSMutableArray arrayWithCapacity:[self count]];
[self enumerateObjectsUsingBlock:^(id obj, NSUInteger index, BOOL *stop) {
if (pred(obj, index))
[result addObject:obj];
}];
return result;
}
- (NSArray *)reverse
{
NSMutableArray *result = [NSMutableArray arrayWithCapacity:[self count]];
[self enumerateObjectsUsingBlock:^(id obj, NSUInteger index, BOOL *stop) {
[result insertObject:obj atIndex:0];
}];
return result;
}
@end