1324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver/* 2324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * [The "BSD license"] 3324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Copyright (c) 2011 Terence Parr and Alan Condit 4324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * All rights reserved. 5324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 6324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * Redistribution and use in source and binary forms, with or without 7324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * modification, are permitted provided that the following conditions 8324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * are met: 9324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 1. Redistributions of source code must retain the above copyright 10324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * notice, this list of conditions and the following disclaimer. 11324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 2. Redistributions in binary form must reproduce the above copyright 12324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * notice, this list of conditions and the following disclaimer in the 13324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * documentation and/or other materials provided with the distribution. 14324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 3. The name of the author may not be used to endorse or promote products 15324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * derived from this software without specific prior written permission. 16324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * 17324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 18324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 19324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 20324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 21324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 22324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 23324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 24324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 26324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver */ 28324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#import "AMutableArray.h" 29324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#import "ArrayIterator.h" 30324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#import "ANTLRRuntimeException.h" 31324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 32324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver@class AMutableArray; 33324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 34324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver@implementation ArrayIterator 35324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 36324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver@synthesize peekObj; 37324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver//@synthesize count; 38324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver@synthesize index; 39324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver@synthesize anArray; 40324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 41324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 42324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver+ (ArrayIterator *) newIterator:(NSArray *)array 43324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 44324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return [[ArrayIterator alloc] initWithArray:array]; 45324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 46324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 47324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver+ (ArrayIterator *) newIteratorForDictKey:(NSDictionary *)dict 48324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 49324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return [[ArrayIterator alloc] initWithDictKey:dict]; 50324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 51324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 52324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver+ (ArrayIterator *) newIteratorForDictObj:(NSDictionary *)dict 53324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 54324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return [[ArrayIterator alloc] initWithDictObj:dict]; 55324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 56324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 57324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver- (id) initWithArray:(NSArray *)array 58324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 59324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver self=[super init]; 60324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( self != nil ) { 61324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (![array isKindOfClass:[NSArray class]]) { 62324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @throw [NSException exceptionWithName:NSInvalidArgumentException 63324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver reason:[NSString stringWithFormat:@"ArrayIterator expecting NSArray class but got %@", [array className]] 64324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver userInfo:nil]; 65324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 66324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver anArray = [array retain]; 67324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#ifdef DONTUSENOMO 68324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver for (int i = 0; i < [array count]; i++) { 69324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver [anArray addObject:[array objectAtIndex:i]]; 70324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver count++; 71324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 72324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#endif 73324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver peekObj = nil; 74324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver count = [anArray count]; 75324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver index = 0; 76324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 77324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return self; 78324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 79324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 80324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver- (id) initWithDictKey:(NSDictionary *)dict 81324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 82324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver self=[super init]; 83324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( self != nil ) { 84324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (![dict isKindOfClass:[NSDictionary class]]) { 85324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @throw [NSException exceptionWithName:NSInvalidArgumentException 86324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver reason:[NSString stringWithFormat:@"ArrayIterator expecting NSDictionary class but got %@", [dict className]] 87324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver userInfo:nil]; 88324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 89324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver anArray = [[[dict keyEnumerator] allObjects] retain]; 90324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver peekObj = nil; 91324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver count = [anArray count]; 92324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver index = 0; 93324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 94324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return self; 95324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 96324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 97324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver- (id) initWithDictObj:(NSDictionary *)dict 98324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 99324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver self=[super init]; 100324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( self != nil ) { 101324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (![dict isKindOfClass:[NSDictionary class]]) { 102324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @throw [NSException exceptionWithName:NSInvalidArgumentException 103324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver reason:[NSString stringWithFormat:@"ArrayIterator expecting NSDictionary class but got %@", [dict className]] 104324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver userInfo:nil]; 105324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 106324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver anArray = [[[dict objectEnumerator] allObjects] retain]; 107324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver peekObj = nil; 108324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver count = [anArray count]; 109324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver index = 0; 110324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 111324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return self; 112324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 113324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 114324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver- (void)dealloc 115324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 116324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#ifdef DEBUG_DEALLOC 117324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver NSLog( @"called dealloc in ArrayIterator" ); 118324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver#endif 119324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( anArray ) [anArray release]; 120324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver [super dealloc]; 121324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 122324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 123324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver- (BOOL) hasNext 124324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 125324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( peekObj == nil ) { 126324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver peekObj = [self nextObject]; 127324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 128324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return ((peekObj) ? YES : NO); 129324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 130324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 131324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver- (NSObject *) nextObject 132324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 133324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver id obj = nil; 134324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( peekObj ) { 135324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver obj = peekObj; 136324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver peekObj = nil; 137324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return obj; 138324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 139324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( index >= count ) { 140324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return nil; 141324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 142324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( anArray ) { 143324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver obj = [anArray objectAtIndex:index++]; 144324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( index >= count ) { 145324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver [anArray release]; 146324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver anArray = nil; 147324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver index = 0; 148324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver count = 0; 149324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 150324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 151324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return obj; 152324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 153324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 154324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver- (NSArray *) allObjects 155324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 156324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if ( (count <= 0 || index >= count) && peekObj == nil ) return nil; 157324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver AMutableArray *theArray = [AMutableArray arrayWithCapacity:count]; 158324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver if (peekObj) { 159324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver [theArray addObject:peekObj]; 160324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver peekObj = nil; 161324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 162324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver for (int i = index; i < count; i++) { 163324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver [theArray addObject:[anArray objectAtIndex:i]]; 164324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver } 165324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return [NSArray arrayWithArray:(NSArray *)theArray]; 166324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 167324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 168324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver- (void) removeObjectAtIndex:(NSInteger)idx 169324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 170324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver @throw [ANTLRUnsupportedOperationException newException:@"Cant remove object from ArrayIterator"]; 171324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 172324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 173324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver- (NSInteger) count 174324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 175324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver return (index - count); 176324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 177324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 178324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver- (void) setCount:(NSInteger)cnt 179324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver{ 180324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver count = cnt; 181324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver} 182324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver 183324c4644fee44b9898524c09511bd33c3f12e2dfBen Gruver@end 184