blob: 5c789ff672854a16906ed37222865251612d93d1 (
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
88
89
90
91
92
93
94
95
96
97
98
99
|
/* 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)HB_allSatisfy:(BOOL (^)(id object))block
{
for (id object in self)
{
if (block(object) == NO)
{
return NO;
}
}
return YES;
}
- (BOOL)HB_containsWhere:(BOOL (^)(id object))block
{
for (id object in self)
{
if (block(object) == YES)
{
return YES;
}
}
return NO;
}
- (NSArray *)HB_filteredArrayUsingBlock:(BOOL (^)(id object))block
{
NSMutableArray *filteredArray = [NSMutableArray array];
for (id object in self)
{
if (block(object))
{
[filteredArray addObject:object];
}
}
return [filteredArray copy];
}
- (NSIndexSet *)HB_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 *)HB_unionWith:(NSIndexSet *)indexSet
{
NSMutableIndexSet *result = [self mutableCopy];
[result addIndexes:indexSet];
return result;
}
- (NSIndexSet *)HB_intersectionWith:(NSIndexSet *)indexSet
{
NSMutableIndexSet *result = [self mutableCopy];
[result addIndexes:indexSet];
[result removeIndexes:[self HB_symmetricDifferenceWith:indexSet]];
return result;
}
- (NSIndexSet *)HB_relativeComplementIn:(NSIndexSet *)universe
{
NSMutableIndexSet *complement = [universe mutableCopy];
[complement removeIndexes:self];
return complement;
}
- (NSIndexSet *)HB_symmetricDifferenceWith:(NSIndexSet *)indexSet
{
NSMutableIndexSet *a = [self mutableCopy];
NSMutableIndexSet *b = [indexSet mutableCopy];
[a removeIndexes:indexSet];
[b removeIndexes:self];
[a addIndexes:b];
return a;
}
@end
|
