OrderedDictionary.m

//
//  OrderedDictionary.m
//  OrderedDictionary
//
//  Modified version (Added indexForKey/Value functions)
//
//  Created by Matt Gallagher on 19/12/08.
//  Copyright 2008 Matt Gallagher. All rights reserved.
//
//  This software is provided 'as-is', without any express or implied
//  warranty. In no event will the authors be held liable for any damages
//  arising from the use of this software. Permission is granted to anyone to
//  use this software for any purpose, including commercial applications, and to
//  alter it and redistribute it freely, subject to the following restrictions:
//
//  1. The origin of this software must not be misrepresented; you must not
//     claim that you wrote the original software. If you use this software
//     in a product, an acknowledgment in the product documentation would be
//     appreciated but is not required.
//  2. Altered source versions must be plainly marked as such, and must not be
//     misrepresented as being the original software.
//  3. This notice may not be removed or altered from any source
//     distribution.
//
 
#import "OrderedDictionary.h"
 
NSString *DescriptionForObject(NSObject *object, id locale, NSUInteger indent)
{
  NSString *objectString;
  if ([object isKindOfClass:[NSString class]])
  {
    objectString = (NSString *)[[object retain] autorelease];
  }
  else if ([object respondsToSelector:@selector(descriptionWithLocale:indent:)])
  {
    objectString = [(NSDictionary *)object descriptionWithLocale:locale indent:indent];
  }
  else if ([object respondsToSelector:@selector(descriptionWithLocale:)])
  {
    objectString = [(NSSet *)object descriptionWithLocale:locale];
  }
  else
  {
    objectString = [object description];
  }
  return objectString;
}
 
@implementation OrderedDictionary
 
- (void)initStorageWithCapacity:(NSUInteger)capacity
{
  dictionary = [[NSMutableDictionary alloc] initWithCapacity:capacity];
  array = [[NSMutableArray alloc] initWithCapacity:capacity];
}
 
- (id)init
{
  self = [super init];
  if (self != nil)
    [self initStorageWithCapacity:0];
  return self;
}
 
- (id)initWithCapacity:(NSUInteger)capacity
{
  self = [super init];
  if (self != nil)
    [self initStorageWithCapacity:capacity];
  return self;
}
 
- (void)dealloc
{
  [dictionary release];
  [array release];
  [super dealloc];
}
 
- (id)copy
{
  return [self mutableCopy];
}
 
- (void)setObject:(id)anObject forKey:(id)aKey
{
  if (![dictionary objectForKey:aKey])
  {
    [array addObject:aKey];
  }
  [dictionary setObject:anObject forKey:aKey];
}
 
- (void)removeObjectForKey:(id)aKey
{
  [dictionary removeObjectForKey:aKey];
  [array removeObject:aKey];
}
 
- (NSUInteger)count
{
  return [dictionary count];
}
 
- (id)objectForKey:(id)aKey
{
  return [dictionary objectForKey:aKey];
}
 
- (NSEnumerator *)keyEnumerator
{
  return [array objectEnumerator];
}
 
- (NSEnumerator *)reverseKeyEnumerator
{
  return [array reverseObjectEnumerator];
}
 
- (void)insertObject:(id)anObject forKey:(id)aKey atIndex:(NSUInteger)anIndex
{
  if ([dictionary objectForKey:aKey])
  {
    [self removeObjectForKey:aKey];
  }
  [array insertObject:aKey atIndex:anIndex];
  [dictionary setObject:anObject forKey:aKey];
}
 
- (id)keyAtIndex:(NSUInteger)anIndex
{
  return [array objectAtIndex:anIndex];
}
 
- (NSUInteger)indexForKey:(id)key
{
  return [array indexOfObject:key];
}
 
- (NSUInteger)indexForValue:(id)value
{
  NSArray *keys = [self allKeysForObject:value];
  if ([keys count] > 0)
  {
    return [self indexForKey:[keys objectAtIndex:0]];
  }
 
  return NSNotFound;
}
 
- (NSString *)descriptionWithLocale:(id)locale indent:(NSUInteger)level
{
  NSMutableString *indentString = [NSMutableString string];
  NSUInteger count = level;
  for (NSUInteger i = 0; i < count; i++)
  {
    [indentString appendFormat:@"    "];
  }
 
  NSMutableString *description = [NSMutableString string];
  [description appendFormat:@"%@{\n", indentString];
  for (NSObject *key in self)
  {
    [description appendFormat:@"%@    %@ = %@;\n", indentString,
                              DescriptionForObject(key, locale, level),
                              DescriptionForObject([self objectForKey:key], locale, level)];
  }
  [description appendFormat:@"%@}\n", indentString];
  return description;
}
 
@end