util/NSArray+Union.m at master · SciApps/util · GitHub

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
//
//  NSArray+Union.m
//  FinTech
//
//  Created by Bálint Róbert on 07/08/15.
//  Copyright (c) 2015 IncepTech Ltd. All rights reserved.
//

#import "NSArray+Union.h"

@implementation NSArray (Union)

- (NSArray *)arrayOfUnionObjectsWithDictionary:(NSDictionary *)dictionary useObjectsFromDictionary:(BOOL)dictionaryObjects {
    NSMutableArray *unificated = [[NSMutableArray alloc] initWithCapacity:MIN(self.count, dictionary.count)];

    // use the smaller container for linear search for best performance
    if (dictionary.count < self.count) {
        NSArray* objectList = dictionary.objectEnumerator.allObjects;

        for (id object in self) {

            NSUInteger idx = [objectList indexOfObject:object];

            if (idx != NSNotFound) {
                [unificated addObject:dictionaryObjects ? [objectList objectAtIndex:idx] : object];
            }
        }
    }
    else {

        for (id object in dictionary.objectEnumerator) {

            NSUInteger idx = [self indexOfObject:object];

            if (idx != NSNotFound) {
                [unificated addObject:dictionaryObjects ? object : [self objectAtIndex:idx]];
            }
        }
    }

    return unificated.copy;
}

- (NSArray *)arrayOfUnionObjectsWithArray:(NSArray *)array useObjectsFromSecondaryArray:(BOOL)secondaryArrayObjects {
    NSMutableArray *unificated = [[NSMutableArray alloc] initWithCapacity:MIN(self.count, array.count)];

    // use the smaller container for linear search for best performance
    if (array.count < self.count) {
        for (id object in self) {

            NSUInteger idx = [array indexOfObject:object];

            if (idx != NSNotFound) {
                [unificated addObject:secondaryArrayObjects ? [array objectAtIndex:idx] : object];
            }
        }
    }
    else {

        for (id object in array) {

            NSUInteger idx = [self indexOfObject:object];

            if (idx != NSNotFound) {
                [unificated addObject:secondaryArrayObjects ? object : [self objectAtIndex:idx]];
            }
        }
    }

    return unificated.copy;
}

- (NSArray *)arrayOfSubstractedObjectsWithArray:(NSArray *)array {
    if (array.count > self.count) {
#ifdef TestLog
        TestLog(@"%@ %@: using inverse substraction", NSStringFromClass(self.class), NSStringFromSelector(_cmd));
#endif
        return [array arrayOfSubstractedObjectsWithArray:self];
    }

    NSMutableArray *substracted = [[NSMutableArray alloc] initWithCapacity:self.count];

    for (id object in self) {

        NSUInteger idx = [array indexOfObject:object];

        if (idx == NSNotFound) {
            [substracted addObject:object];
        }
    }

    return substracted.copy;
}

@end