104 lines
2.3 KiB
Objective-C
104 lines
2.3 KiB
Objective-C
#import "FunctionalUtil.h"
|
|
#import "Constraints.h"
|
|
|
|
@implementation NSArray (FunctionalUtil)
|
|
-(bool) any:(int (^)(id item))predicate {
|
|
require(predicate != nil);
|
|
for (id e in self) {
|
|
if (predicate(e)) {
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
-(bool) all:(int (^)(id item))predicate {
|
|
require(predicate != nil);
|
|
for (id e in self) {
|
|
if (!predicate(e)) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
-(id) firstMatchingElseNil:(int (^)(id item))predicate {
|
|
require(predicate != nil);
|
|
for (id e in self) {
|
|
if (predicate(e)) {
|
|
return e;
|
|
}
|
|
}
|
|
return nil;
|
|
}
|
|
-(NSArray*) map:(id (^)(id item))projection {
|
|
require(projection != nil);
|
|
|
|
NSMutableArray* r = [NSMutableArray arrayWithCapacity:self.count];
|
|
for (id e in self) {
|
|
[r addObject:projection(e)];
|
|
}
|
|
return r;
|
|
}
|
|
-(NSArray*) filter:(int (^)(id item))predicate {
|
|
require(predicate != nil);
|
|
|
|
NSMutableArray* r = [NSMutableArray array];
|
|
for (id e in self) {
|
|
if (predicate(e)) {
|
|
[r addObject:e];
|
|
}
|
|
}
|
|
return r;
|
|
}
|
|
-(double) sumDouble {
|
|
double s = 0.0;
|
|
for (NSNumber* e in self) {
|
|
s += [e doubleValue];
|
|
}
|
|
return s;
|
|
}
|
|
-(NSUInteger) sumNSUInteger {
|
|
NSUInteger s = 0;
|
|
for (NSNumber* e in self) {
|
|
s += [e unsignedIntegerValue];
|
|
}
|
|
return s;
|
|
}
|
|
-(NSInteger) sumNSInteger {
|
|
NSInteger s = 0;
|
|
for (NSNumber* e in self) {
|
|
s += [e integerValue];
|
|
}
|
|
return s;
|
|
}
|
|
-(NSDictionary *)keyedBy:(id(^)(id value))keySelector {
|
|
require(keySelector != nil);
|
|
|
|
NSMutableDictionary *result = [NSMutableDictionary dictionary];
|
|
|
|
for (id value in self) {
|
|
result[keySelector(value)] = value;
|
|
}
|
|
checkOperation(result.count == self.count);
|
|
|
|
return result;
|
|
}
|
|
-(NSDictionary *)groupBy:(id(^)(id value))keySelector {
|
|
require(keySelector != nil);
|
|
|
|
NSMutableDictionary *result = [NSMutableDictionary dictionary];
|
|
|
|
for (id item in self) {
|
|
id key = keySelector(item);
|
|
|
|
NSMutableArray* group = result[key];
|
|
if (group == nil) {
|
|
group = [NSMutableArray array];
|
|
result[key] = group;
|
|
}
|
|
[group addObject:item];
|
|
}
|
|
|
|
return result;
|
|
}
|
|
@end
|