summaryrefslogtreecommitdiff
path: root/platform/darwin/src/NSCompoundPredicate+MGLAdditions.mm
blob: 19568b8033d572765e458ba05225525b3bf4ed26 (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
100
101
#import "NSCompoundPredicate+MGLAdditions.h"

#import "NSPredicate+MGLAdditions.h"
#import "NSExpression+MGLPrivateAdditions.h"

@implementation NSCompoundPredicate (MGLAdditions)

- (std::vector<mbgl::style::Filter>)mgl_subfilters
{
    std::vector<mbgl::style::Filter>filters;
    for (NSPredicate *predicate in self.subpredicates) {
        filters.push_back(predicate.mgl_filter);
    }
    return filters;
}

- (mbgl::style::Filter)mgl_filter
{
    switch (self.compoundPredicateType) {
        case NSNotPredicateType: {
            NSAssert(self.subpredicates.count <= 1, @"NOT predicate cannot have multiple subpredicates.");
            NSPredicate *subpredicate = self.subpredicates.firstObject;
            mbgl::style::Filter subfilter = subpredicate.mgl_filter;

            // Convert NOT(!= nil) to NotHasFilter.
            if (subfilter.is<mbgl::style::HasFilter>()) {
                auto hasFilter = subfilter.get<mbgl::style::HasFilter>();
                return mbgl::style::NotHasFilter { .key = hasFilter.key };
            }

            // Convert NOT(== nil) to HasFilter.
            if (subfilter.is<mbgl::style::NotHasFilter>()) {
                auto hasFilter = subfilter.get<mbgl::style::NotHasFilter>();
                return mbgl::style::HasFilter { .key = hasFilter.key };
            }

            // Convert NOT(IN) or NOT(CONTAINS) to NotInFilter.
            if (subfilter.is<mbgl::style::InFilter>()) {
                auto inFilter = subfilter.get<mbgl::style::InFilter>();
                mbgl::style::NotInFilter notInFilter;
                notInFilter.key = inFilter.key;
                notInFilter.values = inFilter.values;
                return notInFilter;
            }

            // Convert NOT(), NOT(AND), NOT(NOT), NOT(==), etc. into NoneFilter.
            mbgl::style::NoneFilter noneFilter;
            if (subfilter.is<mbgl::style::AnyFilter>()) {
                // Flatten NOT(OR).
                noneFilter.filters = subfilter.get<mbgl::style::AnyFilter>().filters;
            } else if (subpredicate) {
                noneFilter.filters = { subfilter };
            }
            return noneFilter;
        }
        case NSAndPredicateType: {
            mbgl::style::AllFilter filter;
            filter.filters = self.mgl_subfilters;
            return filter;
        }
        case NSOrPredicateType: {
            mbgl::style::AnyFilter filter;
            filter.filters = self.mgl_subfilters;
            return filter;
        }
    }

    [NSException raise:@"Compound predicate type not handled"
                format:@""];
    return {};
}

@end

@implementation NSCompoundPredicate (MGLExpressionAdditions)

- (id)mgl_jsonExpressionObject {
    switch (self.compoundPredicateType) {
        case NSNotPredicateType: {
            NSAssert(self.subpredicates.count <= 1, @"NOT predicate cannot have multiple subpredicates.");
            NSPredicate *subpredicate = self.subpredicates.firstObject;
            return @[@"!", subpredicate.mgl_jsonExpressionObject];
        }
            
        case NSAndPredicateType: {
            NSArray *subarrays = [self.subpredicates valueForKeyPath:@"mgl_jsonExpressionObject"];
            return [@[@"all"] arrayByAddingObjectsFromArray:subarrays];
        }
            
        case NSOrPredicateType: {
            NSArray *subarrays = [self.subpredicates valueForKeyPath:@"mgl_jsonExpressionObject"];
            return [@[@"any"] arrayByAddingObjectsFromArray:subarrays];
        }
    }
    
    [NSException raise:@"Compound predicate type not handled"
                format:@""];
    return nil;
}

@end