iska
158 lines
3.2 KiB
Objective-C
158 lines
3.2 KiB
Objective-C
//
|
|
// HTMLOrderedDictionary.m
|
|
// HTMLKit
|
|
//
|
|
// Created by Iska on 14/03/15.
|
|
// Copyright (c) 2015 BrainCookie. All rights reserved.
|
|
//
|
|
|
|
#import "HTMLOrderedDictionary.h"
|
|
|
|
@interface HTMLOrderedDictionary ()
|
|
{
|
|
NSMutableDictionary *_dictionary;
|
|
NSMutableArray *_keys;
|
|
}
|
|
@end
|
|
|
|
@implementation HTMLOrderedDictionary
|
|
|
|
#pragma mark - Init
|
|
|
|
#pragma clang diagnostic push
|
|
#pragma clang diagnostic ignored "-Wobjc-designated-initializers"
|
|
- (instancetype)init
|
|
{
|
|
return [self initWithCapacity:0];
|
|
}
|
|
#pragma clang diagnostic pop
|
|
|
|
- (instancetype)initWithCapacity:(NSUInteger)capacity
|
|
{
|
|
self = [super init];
|
|
if (self) {
|
|
_dictionary = [[NSMutableDictionary alloc] initWithCapacity:capacity];
|
|
_keys = [[NSMutableArray alloc] initWithCapacity:capacity];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
- (instancetype)initWithObjects:(NSArray *)objects forKeys:(NSArray *)keys
|
|
{
|
|
self = [self initWithCapacity:objects.count];
|
|
if (self) {
|
|
[objects enumerateObjectsUsingBlock:^(id obj, NSUInteger idx, BOOL *stop) {
|
|
_dictionary[keys[idx]] = obj;
|
|
}];
|
|
}
|
|
return self;
|
|
}
|
|
|
|
#pragma mark - Access
|
|
|
|
- (id)objectForKey:(id)aKey
|
|
{
|
|
return _dictionary[aKey];
|
|
}
|
|
|
|
- (void)setObject:(id)anObject forKey:(id<NSCopying>)aKey
|
|
{
|
|
if (_dictionary[aKey] == nil) {
|
|
[_keys addObject:aKey];
|
|
}
|
|
_dictionary[aKey] = anObject;
|
|
}
|
|
|
|
- (void)removeObjectForKey:(id)aKey
|
|
{
|
|
[_keys removeObject:aKey];
|
|
[_dictionary removeObjectForKey:aKey];
|
|
}
|
|
|
|
- (NSUInteger)count
|
|
{
|
|
return _keys.count;
|
|
}
|
|
|
|
#pragma mark - Indexed Access
|
|
|
|
- (id)objectAtIndex:(NSUInteger)index
|
|
{
|
|
return _dictionary[_keys[index]];
|
|
}
|
|
|
|
- (void)setObject:(id)anObject forKey:(id<NSCopying>)aKey atIndex:(NSUInteger)index
|
|
{
|
|
if (_dictionary[aKey]) {
|
|
[_keys removeObject:aKey];
|
|
}
|
|
[_keys insertObject:aKey atIndex:index];
|
|
_dictionary[aKey] = anObject;
|
|
}
|
|
|
|
- (void)removeObjectAtIndex:(NSUInteger)index
|
|
{
|
|
if (_dictionary[_keys[index]]){
|
|
[_dictionary removeObjectForKey:_keys[index]];
|
|
[_keys removeObjectAtIndex:index];
|
|
}
|
|
}
|
|
|
|
- (void)replaceKeyValueAtIndex:(NSUInteger)index withObject:(id)anObject andKey:(id<NSCopying>)aKey
|
|
{
|
|
[_keys replaceObjectAtIndex:index withObject:aKey];
|
|
_dictionary[aKey] = anObject;
|
|
}
|
|
|
|
- (void)replaceKey:(id<NSCopying>)aKey withKey:(id<NSCopying>)newKey
|
|
{
|
|
id value = _dictionary[aKey];
|
|
if (value != nil) {
|
|
NSUInteger index = [_keys indexOfObject:aKey];
|
|
[_keys replaceObjectAtIndex:index withObject:newKey];
|
|
[_dictionary removeObjectForKey:aKey];
|
|
_dictionary[newKey] = value;
|
|
}
|
|
}
|
|
|
|
- (NSUInteger)indexOfKey:(id<NSCopying>)aKey
|
|
{
|
|
return [_keys indexOfObject:aKey];
|
|
}
|
|
|
|
#pragma mark - Subscript
|
|
|
|
- (id)objectAtIndexedSubscript:(NSUInteger)index
|
|
{
|
|
return _dictionary[_keys[index]];
|
|
}
|
|
|
|
- (void)setObject:(id)obj atIndexedSubscript:(NSUInteger)index
|
|
{
|
|
_dictionary[_keys[index]] = obj;
|
|
}
|
|
|
|
- (void)setObject:(id)obj forKeyedSubscript:(id<NSCopying>)key
|
|
{
|
|
[self setObject:obj forKey:key];
|
|
}
|
|
|
|
#pragma mark - Enumeration
|
|
|
|
- (NSEnumerator *)keyEnumerator
|
|
{
|
|
return _keys.objectEnumerator;
|
|
}
|
|
|
|
- (NSEnumerator *)reverseKeyEnumerator
|
|
{
|
|
return _keys.reverseObjectEnumerator;
|
|
}
|
|
|
|
- (NSUInteger)countByEnumeratingWithState:(NSFastEnumerationState *)state objects:(__unsafe_unretained id [])buffer count:(NSUInteger)len
|
|
{
|
|
return [_keys countByEnumeratingWithState:state objects:buffer count:len];
|
|
}
|
|
|
|
@end
|