aboutsummaryrefslogtreecommitdiffhomepage
path: root/Firestore/third_party/Immutable/FSTImmutableSortedDictionary.m
blob: 87c21a59018c105a2fbe88ab4917dc8e2400fc49 (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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
#import "Firestore/third_party/Immutable/FSTImmutableSortedDictionary.h"

#import "Firestore/third_party/Immutable/FSTArraySortedDictionary.h"
#import "Firestore/Source/Util/FSTClasses.h"
#import "Firestore/third_party/Immutable/FSTTreeSortedDictionary.h"

NS_ASSUME_NONNULL_BEGIN

const int kSortedDictionaryArrayToRBTreeSizeThreshold = 25;

@implementation FSTImmutableSortedDictionary

+ (FSTImmutableSortedDictionary *)dictionaryWithComparator:(NSComparator)comparator {
  return [[FSTArraySortedDictionary alloc] initWithComparator:comparator];
}

+ (FSTImmutableSortedDictionary *)dictionaryWithDictionary:(NSDictionary *)dictionary
                                                comparator:(NSComparator)comparator {
  if (dictionary.count <= kSortedDictionaryArrayToRBTreeSizeThreshold) {
    return [FSTArraySortedDictionary dictionaryWithDictionary:dictionary comparator:comparator];
  } else {
    return [FSTTreeSortedDictionary dictionaryWithDictionary:dictionary comparator:comparator];
  }
}

- (FSTImmutableSortedDictionary *)dictionaryBySettingObject:(id)aValue forKey:(id)aKey {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (FSTImmutableSortedDictionary *)dictionaryByRemovingObjectForKey:(id)aKey {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (BOOL)isEqual:(id)object {
  if (![object isKindOfClass:[FSTImmutableSortedDictionary class]]) {
    return NO;
  }

  // TODO(klimt): We could make this more efficient if we put the comparison inside the
  // implementations and short-circuit if they share the same tree node, for instance.
  FSTImmutableSortedDictionary *other = (FSTImmutableSortedDictionary *)object;
  if (self.count != other.count) {
    return NO;
  }
  __block BOOL isEqual = YES;
  [self enumerateKeysAndObjectsUsingBlock:^(id key, id value, BOOL *stop) {
    id otherValue = [other objectForKey:key];
    isEqual = isEqual && (value == otherValue || [value isEqual:otherValue]);
    *stop = !isEqual;
  }];
  return isEqual;
}

- (NSUInteger)hash {
  __block NSUInteger hash = 0;
  [self enumerateKeysAndObjectsUsingBlock:^(id key, id value, BOOL *stop) {
    hash = (hash * 31 + [key hash]) * 17 + [value hash];
  }];
  return hash;
}

- (NSString *)description {
  NSMutableString *str = [[NSMutableString alloc] init];
  __block BOOL first = YES;
  [str appendString:@"{ "];
  [self enumerateKeysAndObjectsUsingBlock:^(id key, id value, BOOL *stop) {
    if (!first) {
      [str appendString:@", "];
    }
    first = NO;
    [str appendString:[NSString stringWithFormat:@"%@: %@", key, value]];
  }];
  [str appendString:@" }"];
  return str;
}

- (nullable id)objectForKey:(id)key {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (id)objectForKeyedSubscript:(id)key {
  return [self objectForKey:key];
}

- (NSUInteger)indexOfKey:(id)key {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (BOOL)isEmpty {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (NSUInteger)count {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (id)minKey {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (id)maxKey {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (void)enumerateKeysAndObjectsUsingBlock:(void (^)(id, id, BOOL *))block {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (void)enumerateKeysAndObjectsReverse:(BOOL)reverse usingBlock:(void (^)(id, id, BOOL *))block {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (BOOL)containsKey:(id)key {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (NSEnumerator *)keyEnumerator {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (NSEnumerator *)keyEnumeratorFrom:(id)startKey {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (NSEnumerator *)keyEnumeratorFrom:(id)startKey to:(nullable id)endKey {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (NSEnumerator *)reverseKeyEnumerator {
  @throw FSTAbstractMethodException();  // NOLINT
}

- (NSEnumerator *)reverseKeyEnumeratorFrom:(id)startKey {
  @throw FSTAbstractMethodException();  // NOLINT
}

@end

NS_ASSUME_NONNULL_END