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
|
/*
Copyright (c) 2006 Benhur Stein
This file is part of Paj.
Paj is free software; you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the
Free Software Foundation; either version 2 of the License, or (at your
option) any later version.
Paj is distributed in the hope that it will be useful, but WITHOUT ANY
WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
for more details.
You should have received a copy of the GNU Lesser General Public License
along with Paj; if not, write to the Free Software Foundation, Inc.,
59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
*/
#include "NSArray+Additions.h"
#include "Macros.h"
#include <Foundation/NSEnumerator.h>
// Enumerator of objects in a range of indexes of an array
@interface ArrayRangeEnumerator : NSEnumerator
{
NSArray *array;
NSRange range;
unsigned nextIndex;
}
@end
@implementation ArrayRangeEnumerator
- (id)initWithArray:(NSArray *)anArray range:(NSRange)aRange
{
self = [super init];
if (self != nil) {
Assign(array, anArray);
range = aRange;
nextIndex = range.location;
int count = [array count];
if (NSMaxRange(range) >= count) {
range.length = count - range.location;
}
}
return self;
}
- (void)dealloc
{
Assign(array, nil);
[super dealloc];
}
- (id)nextObject
{
id nextObject;
if (NSLocationInRange(nextIndex, range)) {
nextObject = [array objectAtIndex:nextIndex];
nextIndex++;
} else {
nextObject = nil;
}
return nextObject;
}
@end
// Reverse enumerator of objects in a range of indexes of an array
@interface ReverseArrayRangeEnumerator : ArrayRangeEnumerator
@end
@implementation ReverseArrayRangeEnumerator
- (id)initWithArray:(NSArray *)anArray range:(NSRange)aRange
{
self = [super initWithArray:anArray range:aRange];
if (self != nil) {
nextIndex = NSMaxRange(range) - 1;
}
return self;
}
- (id)nextObject
{
id nextObject;
if (NSLocationInRange(nextIndex, range)) {
nextObject = [array objectAtIndex:nextIndex];
nextIndex--;
} else {
nextObject = nil;
}
return nextObject;
}
@end
@implementation NSArray (PajeAdditions)
- (NSEnumerator *)objectEnumeratorWithRange:(NSRange)range
{
return [[[ArrayRangeEnumerator alloc]
initWithArray:self range:range] autorelease];
}
- (NSEnumerator *)reverseObjectEnumeratorWithRange:(NSRange)range
{
return [[[ReverseArrayRangeEnumerator alloc]
initWithArray:self range:range] autorelease];
}
@end
|
