blob: f24417632e46785a6dca429f3d615ad537192eb6 (
plain)
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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
/* NSArray+HBAdditions.m $
This file is part of the HandBrake source code.
Homepage: <http://handbrake.fr/>.
It may be used under the terms of the GNU General Public License. */
#import "NSArray+HBAdditions.h"
@implementation NSArray (HBAdditions)
- (BOOL)allSatisfy:(BOOL (^)(id object))block
{
for (id object in self)
{
if (block(object) == NO)
{
return NO;
}
}
return YES;
}
- (NSArray *)filteredArrayUsingBlock:(BOOL (^)(id object))block
{
NSMutableArray *filteredArray = [NSMutableArray array];
for (id object in self)
{
if (block(object))
{
[filteredArray addObject:object];
}
}
return [filteredArray copy];
}
- (NSIndexSet *)indexesOfObjectsUsingBlock:(BOOL (^)(id object))block
{
NSMutableIndexSet *indexes = [NSMutableIndexSet indexSet];
NSUInteger i = 0;
for (id object in self)
{
if (block(object))
{
[indexes addIndex:i];
}
i++;
}
return [indexes copy];
}
@end
@implementation NSIndexSet (HBAdditions)
- (NSIndexSet *)unionWith:(NSIndexSet *)indexSet
{
NSMutableIndexSet *result = [self mutableCopy];
[result addIndexes:indexSet];
return result;
}
- (NSIndexSet *)intersectionWith:(NSIndexSet *)indexSet
{
NSMutableIndexSet *result = [self mutableCopy];
[result addIndexes:indexSet];
[result removeIndexes:[self symmetricDifferenceWith:indexSet]];
return result;
}
- (NSIndexSet *)relativeComplementIn:(NSIndexSet *)universe
{
NSMutableIndexSet *complement = [universe mutableCopy];
[complement removeIndexes:self];
return complement;
}
- (NSIndexSet *)symmetricDifferenceWith:(NSIndexSet *)indexSet
{
NSMutableIndexSet *a = [self mutableCopy];
NSMutableIndexSet *b = [indexSet mutableCopy];
[a removeIndexes:indexSet];
[b removeIndexes:self];
[a addIndexes:b];
return a;
}
@end
|
