You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
877 B
Objective-C
39 lines
877 B
Objective-C
//
|
|
// NSArray+Extension.m
|
|
// pEpiOSAdapter
|
|
//
|
|
// Created by Dirk Zimmermann on 03.05.17.
|
|
// Copyright © 2017 p≡p. All rights reserved.
|
|
//
|
|
|
|
#import "NSArray+Extension.h"
|
|
|
|
@implementation ArrayTake
|
|
|
|
- (instancetype)initWithElements:(NSArray * _Nonnull)elements rest:(NSArray * _Nonnull)rest
|
|
{
|
|
if (self = [super init]) {
|
|
_elements = elements;
|
|
_rest = rest;
|
|
}
|
|
return self;
|
|
}
|
|
|
|
@end
|
|
|
|
@implementation NSArray (Extension)
|
|
|
|
- (ArrayTake * _Nullable)takeOrNil:(NSInteger)count
|
|
{
|
|
if (self.count >= count) {
|
|
NSInteger restCount = self.count - count;
|
|
NSArray *elements = [self subarrayWithRange:NSMakeRange(0, count)];
|
|
NSArray *rest = [self subarrayWithRange:NSMakeRange(count, restCount)];
|
|
ArrayTake *taken = [[ArrayTake alloc] initWithElements:elements rest:rest];
|
|
return taken;
|
|
}
|
|
return nil;
|
|
}
|
|
|
|
@end
|