fix bug

杨锴

2025-04-16 09a372bc45fde16fd42257ab6f78b8deeecf720b

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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
/**
 * Tencent is pleased to support the open source community by making QMUI_iOS available.
 * Copyright (C) 2016-2021 THL A29 Limited, a Tencent company. All rights reserved.
 * Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License. You may obtain a copy of the License at
 * http://opensource.org/licenses/MIT
 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.
 */
 
//
//  QMUIOrderedDictionary.m
//  qmui
//
//  Created by QMUI Team on 16/7/21.
//
 
#import "QMUIOrderedDictionary.h"
 
@interface QMUIOrderedDictionary ()
 
@property(nonatomic, strong) NSMutableArray *mutableAllKeys;
@property(nonatomic, strong) NSMutableArray *mutableAllValues;
@property(nonatomic, strong) NSMutableDictionary *mutableDictionary;
@end
 
@implementation QMUIOrderedDictionary
 
- (instancetype)initWithKeysAndObjects:(id)firstKey, ... {
    if (self = [self init]) {
        
        if (firstKey) {
            [self.mutableAllKeys addObject:firstKey];
            
            va_list argumentList;
            va_start(argumentList, firstKey);
            id argument;
            NSInteger i = 1;
            while ((argument = va_arg(argumentList, id))) {
                if (i % 2 == 0) {
                    [self.mutableAllKeys addObject:argument];
                } else {
                    [self.mutableAllValues addObject:argument];
                }
                i++;
            }
            va_end(argumentList);
            
            [self.mutableAllKeys enumerateObjectsUsingBlock:^(id  _Nonnull key, NSUInteger idx, BOOL * _Nonnull stop) {
                [self.mutableDictionary setObject:self.mutableAllValues[idx] forKey:key];
            }];
        }
    }
    return self;
}
 
- (instancetype)init {
    if (self = [super init]) {
        self.mutableAllKeys = [[NSMutableArray alloc] init];
        self.mutableAllValues = [[NSMutableArray alloc] init];
        self.mutableDictionary = [[NSMutableDictionary alloc] init];
    }
    return self;
}
 
- (NSUInteger)count {
    return self.mutableDictionary.count;
}
 
- (NSArray *)allKeys {
    return self.mutableAllKeys.copy;
}
 
- (NSArray *)allValues {
    return self.mutableAllValues.copy;
}
 
- (void)setObject:(id)object forKey:(id)key {
    if ([self.mutableAllKeys containsObject:key]) {
        NSInteger index = [self.mutableAllKeys indexOfObject:key];
        [self.mutableAllValues replaceObjectAtIndex:index withObject:object];
        [self.mutableDictionary setObject:object forKey:key];
    } else {
        [self addObject:object forKey:key];
    }
}
 
- (void)addObject:(id)object forKey:(id)key {
    if (![self.mutableAllKeys containsObject:key]) {
        [self.mutableAllKeys addObject:key];
        [self.mutableAllValues addObject:object];
        [self.mutableDictionary setObject:object forKey:key];
    }
}
 
- (void)addObjects:(NSArray *)objects forKeys:(NSArray *)keys {
    if (objects.count == keys.count) {
        [keys enumerateObjectsUsingBlock:^(id  _Nonnull key, NSUInteger idx, BOOL * _Nonnull stop) {
            [self addObject:objects[idx] forKey:key];
        }];
    }
}
 
- (void)insertObject:(id)object forKey:(id)key atIndex:(NSInteger)index {
    if (![self.mutableAllKeys containsObject:key]) {
        [self.mutableAllKeys insertObject:key atIndex:index];
        [self.mutableAllValues insertObject:object atIndex:index];
        [self.mutableDictionary setObject:object forKey:key];
    }
}
 
- (void)insertObjects:(NSArray *)objects forKeys:(NSArray *)keys atIndex:(NSInteger)index {
    if (objects.count == keys.count) {
        __block NSInteger nextIndex = index;
        [keys enumerateObjectsUsingBlock:^(id  _Nonnull key, NSUInteger idx, BOOL * _Nonnull stop) {
            [self insertObject:objects[idx] forKey:key atIndex:nextIndex];
            nextIndex++;
        }];
    }
}
 
- (void)removeObject:(id)object forKey:(id)key {
    if ([self.mutableAllKeys containsObject:key]) {
        NSInteger index = [self.mutableAllKeys indexOfObject:key];
        [self removeObject:object atIndex:index];
    }
}
 
- (void)removeObject:(id)object atIndex:(NSInteger)index {
    if (index < self.allKeys.count) {
        [self.mutableDictionary removeObjectForKey:self.mutableAllKeys[index]];
        [self.mutableAllKeys removeObjectAtIndex:index];
        [self.mutableAllValues removeObjectAtIndex:index];
    }
}
 
- (id)objectForKey:(id)key {
    return [self.mutableDictionary objectForKey:key];
}
 
- (id)objectForKeyedSubscript:(id)key {
    return [self objectForKey:key];
}
 
- (id)objectAtIndex:(NSInteger)index {
    return [self.mutableDictionary objectForKey:self.mutableAllKeys[index]];
}
 
- (id)objectAtIndexedSubscript:(NSUInteger)idx {
    return [self objectAtIndex:idx];
}
 
- (NSString *)description {
    return [NSString stringWithFormat:@"%@, %@", [super description], self.mutableDictionary];
}
 
@end